blob: a93caadb059ea6d915fe02f028c6c31875607ae2 [file] [log] [blame]
Guido van Rossum398d9fe1994-05-11 08:59:13 +00001
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002/* Signal module -- many thanks to Lance Ellinghaus */
Guido van Rossum398d9fe1994-05-11 08:59:13 +00003
Guido van Rossum644a12b1997-04-09 19:24:53 +00004/* XXX Signals should be recorded per thread, now we have thread state. */
5
Guido van Rossum602099a1994-09-14 13:32:22 +00006#include "Python.h"
Guido van Rossum398d9fe1994-05-11 08:59:13 +00007
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00008#ifdef MS_WINDOWS
Brian Curtineb24d742010-04-12 17:16:38 +00009#include <Windows.h>
Benjamin Peterson2614cda2010-03-21 22:36:19 +000010#ifdef HAVE_PROCESS_H
Guido van Rossum644a12b1997-04-09 19:24:53 +000011#include <process.h>
12#endif
Benjamin Peterson2614cda2010-03-21 22:36:19 +000013#endif
Guido van Rossum644a12b1997-04-09 19:24:53 +000014
Benjamin Peterson2614cda2010-03-21 22:36:19 +000015#ifdef HAVE_SIGNAL_H
Guido van Rossum398d9fe1994-05-11 08:59:13 +000016#include <signal.h>
Benjamin Peterson2614cda2010-03-21 22:36:19 +000017#endif
18#ifdef HAVE_SYS_STAT_H
Christian Heimes5fb7c2a2007-12-24 08:52:31 +000019#include <sys/stat.h>
Benjamin Peterson2614cda2010-03-21 22:36:19 +000020#endif
Martin v. Löwis3bf3cc02008-03-24 14:05:07 +000021#ifdef HAVE_SYS_TIME_H
Martin v. Löwis823725e2008-03-24 13:39:54 +000022#include <sys/time.h>
Martin v. Löwis3bf3cc02008-03-24 14:05:07 +000023#endif
Christian Heimes5fb7c2a2007-12-24 08:52:31 +000024
Guido van Rossumbb4ba121994-06-23 11:25:45 +000025#ifndef SIG_ERR
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +000026#define SIG_ERR ((PyOS_sighandler_t)(-1))
Guido van Rossumbb4ba121994-06-23 11:25:45 +000027#endif
28
Andrew MacIntyre7bf68332002-03-03 02:59:16 +000029#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000030#define NSIG 12
31#include <process.h>
32#endif
33
Guido van Rossum3bbc62e1995-01-02 19:30:30 +000034#ifndef NSIG
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000035# if defined(_NSIG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000036# define NSIG _NSIG /* For BSD/SysV */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000037# elif defined(_SIGMAX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000038# define NSIG (_SIGMAX + 1) /* For QNX */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000039# elif defined(SIGMAX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000040# define NSIG (SIGMAX + 1) /* For djgpp */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000041# else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000042# define NSIG 64 /* Use a reasonable default value */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000043# endif
Guido van Rossum3bbc62e1995-01-02 19:30:30 +000044#endif
45
46
Guido van Rossumbb4ba121994-06-23 11:25:45 +000047/*
48 NOTES ON THE INTERACTION BETWEEN SIGNALS AND THREADS
49
50 When threads are supported, we want the following semantics:
51
52 - only the main thread can set a signal handler
53 - any thread can get a signal handler
54 - signals are only delivered to the main thread
55
56 I.e. we don't support "synchronous signals" like SIGFPE (catching
57 this doesn't make much sense in Python anyway) nor do we support
58 signals as a means of inter-thread communication, since not all
59 thread implementations support that (at least our thread library
60 doesn't).
61
62 We still have the problem that in some implementations signals
63 generated by the keyboard (e.g. SIGINT) are delivered to all
64 threads (e.g. SGI), while in others (e.g. Solaris) such signals are
65 delivered to one random thread (an intermediate possibility would
Guido van Rossuma3c04b01995-01-12 11:29:01 +000066 be to deliver it to the main thread -- POSIX?). For now, we have
Guido van Rossumbb4ba121994-06-23 11:25:45 +000067 a working implementation that works in all three cases -- the
68 handler ignores signals if getpid() isn't the same as in the main
69 thread. XXX This is a hack.
70
Guido van Rossum9e8181b2000-09-19 00:46:46 +000071 GNU pth is a user-space threading library, and as such, all threads
72 run within the same process. In this case, if the currently running
73 thread is not the main_thread, send the signal to the main_thread.
Guido van Rossumbb4ba121994-06-23 11:25:45 +000074*/
75
76#ifdef WITH_THREAD
Guido van Rossum295b8e51997-06-06 21:16:41 +000077#include <sys/types.h> /* For pid_t */
Guido van Rossum49b56061998-10-01 20:42:43 +000078#include "pythread.h"
Guido van Rossumbb4ba121994-06-23 11:25:45 +000079static long main_thread;
80static pid_t main_pid;
81#endif
82
Barry Warsaw92971171997-01-03 00:14:25 +000083static struct {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000084 int tripped;
85 PyObject *func;
Barry Warsaw92971171997-01-03 00:14:25 +000086} Handlers[NSIG];
Guido van Rossum398d9fe1994-05-11 08:59:13 +000087
Christian Heimes5fb7c2a2007-12-24 08:52:31 +000088static sig_atomic_t wakeup_fd = -1;
89
Christian Heimesb76922a2007-12-11 01:06:40 +000090/* Speed up sigcheck() when none tripped */
91static volatile sig_atomic_t is_tripped = 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +000092
Barry Warsaw92971171997-01-03 00:14:25 +000093static PyObject *DefaultHandler;
94static PyObject *IgnoreHandler;
95static PyObject *IntHandler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +000096
Martin v. Löwisf58de1b2001-03-06 12:13:56 +000097/* On Solaris 8, gcc will produce a warning that the function
98 declaration is not a prototype. This is caused by the definition of
99 SIG_DFL as (void (*)())0; the correct declaration would have been
100 (void (*)(int))0. */
101
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000102static PyOS_sighandler_t old_siginthandler = SIG_DFL;
Guido van Rossum08c16611997-08-02 03:01:42 +0000103
Martin v. Löwis823725e2008-03-24 13:39:54 +0000104#ifdef HAVE_GETITIMER
105static PyObject *ItimerError;
106
107/* auxiliary functions for setitimer/getitimer */
108static void
109timeval_from_double(double d, struct timeval *tv)
110{
111 tv->tv_sec = floor(d);
112 tv->tv_usec = fmod(d, 1.0) * 1000000.0;
113}
114
Christian Heimes1a8501c2008-10-02 19:56:01 +0000115Py_LOCAL_INLINE(double)
Martin v. Löwis823725e2008-03-24 13:39:54 +0000116double_from_timeval(struct timeval *tv)
117{
118 return tv->tv_sec + (double)(tv->tv_usec / 1000000.0);
119}
120
121static PyObject *
122itimer_retval(struct itimerval *iv)
123{
124 PyObject *r, *v;
125
126 r = PyTuple_New(2);
127 if (r == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000128 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000129
130 if(!(v = PyFloat_FromDouble(double_from_timeval(&iv->it_value)))) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000131 Py_DECREF(r);
132 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000133 }
134
135 PyTuple_SET_ITEM(r, 0, v);
136
137 if(!(v = PyFloat_FromDouble(double_from_timeval(&iv->it_interval)))) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000138 Py_DECREF(r);
139 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000140 }
141
142 PyTuple_SET_ITEM(r, 1, v);
143
144 return r;
145}
146#endif
Barry Warsaw92971171997-01-03 00:14:25 +0000147
Guido van Rossume4485b01994-09-07 14:32:49 +0000148static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000149signal_default_int_handler(PyObject *self, PyObject *args)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000150{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000151 PyErr_SetNone(PyExc_KeyboardInterrupt);
152 return NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000153}
154
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000155PyDoc_STRVAR(default_int_handler_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000156"default_int_handler(...)\n\
157\n\
Michael W. Hudson24ec2112004-06-17 15:55:53 +0000158The default handler for SIGINT installed by Python.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000159It raises KeyboardInterrupt.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000160
Thomas Wouters0796b002000-07-22 23:49:30 +0000161
162static int
163checksignals_witharg(void * unused)
164{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000165 return PyErr_CheckSignals();
Thomas Wouters0796b002000-07-22 23:49:30 +0000166}
167
Tim Peters4f1b2082000-07-23 21:18:09 +0000168static void
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000169signal_handler(int sig_num)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000170{
Antoine Pitrou39a65912010-11-05 19:47:27 +0000171 int save_errno = errno;
172
173#if defined(WITH_THREAD) && defined(WITH_PTH)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000174 if (PyThread_get_thread_ident() != main_thread) {
175 pth_raise(*(pth_t *) main_thread, sig_num);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000176 }
Antoine Pitrou39a65912010-11-05 19:47:27 +0000177 else
Guido van Rossum9e8181b2000-09-19 00:46:46 +0000178#endif
Antoine Pitrou39a65912010-11-05 19:47:27 +0000179 {
180#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000181 /* See NOTES section above */
Antoine Pitrou39a65912010-11-05 19:47:27 +0000182 if (getpid() == main_pid)
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000183#endif
Antoine Pitrou39a65912010-11-05 19:47:27 +0000184 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000185 Handlers[sig_num].tripped = 1;
186 /* Set is_tripped after setting .tripped, as it gets
187 cleared in PyErr_CheckSignals() before .tripped. */
188 is_tripped = 1;
189 Py_AddPendingCall(checksignals_witharg, NULL);
190 if (wakeup_fd != -1)
191 write(wakeup_fd, "\0", 1);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000192 }
Antoine Pitrou39a65912010-11-05 19:47:27 +0000193
Jean-Paul Calderone6f137ca2010-05-09 03:18:57 +0000194#ifndef HAVE_SIGACTION
Antoine Pitrou39a65912010-11-05 19:47:27 +0000195#ifdef SIGCHLD
196 /* To avoid infinite recursion, this signal remains
197 reset until explicit re-instated.
198 Don't clear the 'func' field as it is our pointer
199 to the Python handler... */
200 if (sig_num != SIGCHLD)
201#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000202 /* If the handler was not set up with sigaction, reinstall it. See
203 * Python/pythonrun.c for the implementation of PyOS_setsig which
204 * makes this true. See also issue8354. */
205 PyOS_setsig(sig_num, signal_handler);
Jean-Paul Calderone6f137ca2010-05-09 03:18:57 +0000206#endif
Antoine Pitrou39a65912010-11-05 19:47:27 +0000207 }
208
209 /* Issue #10311: asynchronously executing signal handlers should not
210 mutate errno under the feet of unsuspecting C code. */
211 errno = save_errno;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000212}
Guido van Rossume4485b01994-09-07 14:32:49 +0000213
Guido van Rossum06d511d1995-03-10 15:13:48 +0000214
Guido van Rossum1171ee61997-08-22 20:42:00 +0000215#ifdef HAVE_ALARM
Guido van Rossume4485b01994-09-07 14:32:49 +0000216static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000217signal_alarm(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000218{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000219 int t;
220 if (!PyArg_ParseTuple(args, "i:alarm", &t))
221 return NULL;
222 /* alarm() returns the number of seconds remaining */
223 return PyLong_FromLong((long)alarm(t));
Guido van Rossumaa0f4c71994-08-23 13:49:37 +0000224}
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000225
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000226PyDoc_STRVAR(alarm_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000227"alarm(seconds)\n\
228\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000229Arrange for SIGALRM to arrive after the given number of seconds.");
Guido van Rossum06d511d1995-03-10 15:13:48 +0000230#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +0000231
Guido van Rossum1171ee61997-08-22 20:42:00 +0000232#ifdef HAVE_PAUSE
Guido van Rossuma597dde1995-01-10 20:56:29 +0000233static PyObject *
Neal Norwitz3a6f9782002-03-25 20:46:46 +0000234signal_pause(PyObject *self)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000235{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000236 Py_BEGIN_ALLOW_THREADS
237 (void)pause();
238 Py_END_ALLOW_THREADS
239 /* make sure that any exceptions that got raised are propagated
240 * back into Python
241 */
242 if (PyErr_CheckSignals())
243 return NULL;
Barry Warsaw92971171997-01-03 00:14:25 +0000244
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000245 Py_INCREF(Py_None);
246 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +0000247}
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000248PyDoc_STRVAR(pause_doc,
Barry Warsaw1ee36ff1998-07-21 22:41:18 +0000249"pause()\n\
250\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000251Wait until a signal arrives.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000252
Guido van Rossum06d511d1995-03-10 15:13:48 +0000253#endif
Guido van Rossume4485b01994-09-07 14:32:49 +0000254
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000255
Guido van Rossume4485b01994-09-07 14:32:49 +0000256static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000257signal_signal(PyObject *self, PyObject *args)
Guido van Rossume4485b01994-09-07 14:32:49 +0000258{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000259 PyObject *obj;
260 int sig_num;
261 PyObject *old_handler;
262 void (*func)(int);
263 if (!PyArg_ParseTuple(args, "iO:signal", &sig_num, &obj))
264 return NULL;
Brian Curtinef9efbd2010-08-06 19:27:32 +0000265#ifdef MS_WINDOWS
266 /* Validate that sig_num is one of the allowable signals */
Brian Curtinc734b312010-09-06 16:04:10 +0000267 switch (sig_num) {
268 case SIGABRT: break;
Brian Curtin9e88b5a2010-10-01 14:49:24 +0000269#ifdef SIGBREAK
270 /* Issue #10003: SIGBREAK is not documented as permitted, but works
271 and corresponds to CTRL_BREAK_EVENT. */
272 case SIGBREAK: break;
273#endif
Brian Curtinc734b312010-09-06 16:04:10 +0000274 case SIGFPE: break;
275 case SIGILL: break;
276 case SIGINT: break;
277 case SIGSEGV: break;
278 case SIGTERM: break;
279 default:
280 PyErr_SetString(PyExc_ValueError, "invalid signal value");
281 return NULL;
Brian Curtinef9efbd2010-08-06 19:27:32 +0000282 }
283#endif
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000284#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000285 if (PyThread_get_thread_ident() != main_thread) {
286 PyErr_SetString(PyExc_ValueError,
287 "signal only works in main thread");
288 return NULL;
289 }
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000290#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000291 if (sig_num < 1 || sig_num >= NSIG) {
292 PyErr_SetString(PyExc_ValueError,
293 "signal number out of range");
294 return NULL;
295 }
296 if (obj == IgnoreHandler)
297 func = SIG_IGN;
298 else if (obj == DefaultHandler)
299 func = SIG_DFL;
300 else if (!PyCallable_Check(obj)) {
301 PyErr_SetString(PyExc_TypeError,
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000302"signal handler must be signal.SIG_IGN, signal.SIG_DFL, or a callable object");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000303 return NULL;
304 }
305 else
306 func = signal_handler;
307 if (PyOS_setsig(sig_num, func) == SIG_ERR) {
308 PyErr_SetFromErrno(PyExc_RuntimeError);
309 return NULL;
310 }
311 old_handler = Handlers[sig_num].func;
312 Handlers[sig_num].tripped = 0;
313 Py_INCREF(obj);
314 Handlers[sig_num].func = obj;
315 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000316}
317
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000318PyDoc_STRVAR(signal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000319"signal(sig, action) -> action\n\
320\n\
321Set the action for the given signal. The action can be SIG_DFL,\n\
322SIG_IGN, or a callable Python object. The previous action is\n\
323returned. See getsignal() for possible return values.\n\
324\n\
325*** IMPORTANT NOTICE ***\n\
326A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000327the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000328
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000329
Guido van Rossume4485b01994-09-07 14:32:49 +0000330static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000331signal_getsignal(PyObject *self, PyObject *args)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000332{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000333 int sig_num;
334 PyObject *old_handler;
335 if (!PyArg_ParseTuple(args, "i:getsignal", &sig_num))
336 return NULL;
337 if (sig_num < 1 || sig_num >= NSIG) {
338 PyErr_SetString(PyExc_ValueError,
339 "signal number out of range");
340 return NULL;
341 }
342 old_handler = Handlers[sig_num].func;
343 Py_INCREF(old_handler);
344 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000345}
346
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000347PyDoc_STRVAR(getsignal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000348"getsignal(sig) -> action\n\
349\n\
350Return the current action for the given signal. The return value can be:\n\
351SIG_IGN -- if the signal is being ignored\n\
352SIG_DFL -- if the default action for the signal is in effect\n\
353None -- if an unknown handler is in effect\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000354anything else -- the callable Python object used as a handler");
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000355
Christian Heimes8640e742008-02-23 16:23:06 +0000356#ifdef HAVE_SIGINTERRUPT
357PyDoc_STRVAR(siginterrupt_doc,
358"siginterrupt(sig, flag) -> None\n\
359change system call restart behaviour: if flag is False, system calls\n\
360will be restarted when interrupted by signal sig, else system calls\n\
361will be interrupted.");
362
363static PyObject *
364signal_siginterrupt(PyObject *self, PyObject *args)
365{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000366 int sig_num;
367 int flag;
Christian Heimes8640e742008-02-23 16:23:06 +0000368
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000369 if (!PyArg_ParseTuple(args, "ii:siginterrupt", &sig_num, &flag))
370 return NULL;
371 if (sig_num < 1 || sig_num >= NSIG) {
372 PyErr_SetString(PyExc_ValueError,
373 "signal number out of range");
374 return NULL;
375 }
376 if (siginterrupt(sig_num, flag)<0) {
377 PyErr_SetFromErrno(PyExc_RuntimeError);
378 return NULL;
379 }
Christian Heimes8640e742008-02-23 16:23:06 +0000380
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000381 Py_INCREF(Py_None);
382 return Py_None;
Christian Heimes8640e742008-02-23 16:23:06 +0000383}
384
385#endif
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000386
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000387static PyObject *
388signal_set_wakeup_fd(PyObject *self, PyObject *args)
389{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000390 struct stat buf;
391 int fd, old_fd;
392 if (!PyArg_ParseTuple(args, "i:set_wakeup_fd", &fd))
393 return NULL;
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000394#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000395 if (PyThread_get_thread_ident() != main_thread) {
396 PyErr_SetString(PyExc_ValueError,
397 "set_wakeup_fd only works in main thread");
398 return NULL;
399 }
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000400#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000401 if (fd != -1 && fstat(fd, &buf) != 0) {
402 PyErr_SetString(PyExc_ValueError, "invalid fd");
403 return NULL;
404 }
405 old_fd = wakeup_fd;
406 wakeup_fd = fd;
407 return PyLong_FromLong(old_fd);
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000408}
409
410PyDoc_STRVAR(set_wakeup_fd_doc,
411"set_wakeup_fd(fd) -> fd\n\
412\n\
413Sets the fd to be written to (with '\\0') when a signal\n\
414comes in. A library can use this to wakeup select or poll.\n\
415The previous fd is returned.\n\
416\n\
417The fd must be non-blocking.");
418
419/* C API for the same, without all the error checking */
420int
421PySignal_SetWakeupFd(int fd)
422{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000423 int old_fd = wakeup_fd;
424 if (fd < 0)
425 fd = -1;
426 wakeup_fd = fd;
427 return old_fd;
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000428}
429
430
Martin v. Löwis823725e2008-03-24 13:39:54 +0000431#ifdef HAVE_SETITIMER
432static PyObject *
433signal_setitimer(PyObject *self, PyObject *args)
434{
435 double first;
436 double interval = 0;
437 int which;
438 struct itimerval new, old;
439
440 if(!PyArg_ParseTuple(args, "id|d:setitimer", &which, &first, &interval))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000441 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000442
443 timeval_from_double(first, &new.it_value);
444 timeval_from_double(interval, &new.it_interval);
445 /* Let OS check "which" value */
446 if (setitimer(which, &new, &old) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000447 PyErr_SetFromErrno(ItimerError);
448 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000449 }
450
451 return itimer_retval(&old);
452}
453
454PyDoc_STRVAR(setitimer_doc,
455"setitimer(which, seconds[, interval])\n\
456\n\
457Sets given itimer (one of ITIMER_REAL, ITIMER_VIRTUAL\n\
458or ITIMER_PROF) to fire after value seconds and after\n\
459that every interval seconds.\n\
460The itimer can be cleared by setting seconds to zero.\n\
461\n\
462Returns old values as a tuple: (delay, interval).");
463#endif
464
465
466#ifdef HAVE_GETITIMER
467static PyObject *
468signal_getitimer(PyObject *self, PyObject *args)
469{
470 int which;
471 struct itimerval old;
472
473 if (!PyArg_ParseTuple(args, "i:getitimer", &which))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000474 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000475
476 if (getitimer(which, &old) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000477 PyErr_SetFromErrno(ItimerError);
478 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000479 }
480
481 return itimer_retval(&old);
482}
483
484PyDoc_STRVAR(getitimer_doc,
485"getitimer(which)\n\
486\n\
487Returns current value of given itimer.");
488#endif
489
490
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000491/* List of functions defined in the module */
Barry Warsaw92971171997-01-03 00:14:25 +0000492static PyMethodDef signal_methods[] = {
Guido van Rossum1171ee61997-08-22 20:42:00 +0000493#ifdef HAVE_ALARM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000494 {"alarm", signal_alarm, METH_VARARGS, alarm_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000495#endif
Martin v. Löwis823725e2008-03-24 13:39:54 +0000496#ifdef HAVE_SETITIMER
497 {"setitimer", signal_setitimer, METH_VARARGS, setitimer_doc},
498#endif
499#ifdef HAVE_GETITIMER
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000500 {"getitimer", signal_getitimer, METH_VARARGS, getitimer_doc},
Martin v. Löwis823725e2008-03-24 13:39:54 +0000501#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000502 {"signal", signal_signal, METH_VARARGS, signal_doc},
503 {"getsignal", signal_getsignal, METH_VARARGS, getsignal_doc},
504 {"set_wakeup_fd", signal_set_wakeup_fd, METH_VARARGS, set_wakeup_fd_doc},
Christian Heimes8640e742008-02-23 16:23:06 +0000505#ifdef HAVE_SIGINTERRUPT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000506 {"siginterrupt", signal_siginterrupt, METH_VARARGS, siginterrupt_doc},
Christian Heimes8640e742008-02-23 16:23:06 +0000507#endif
Guido van Rossum1171ee61997-08-22 20:42:00 +0000508#ifdef HAVE_PAUSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000509 {"pause", (PyCFunction)signal_pause,
510 METH_NOARGS,pause_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000511#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000512 {"default_int_handler", signal_default_int_handler,
513 METH_VARARGS, default_int_handler_doc},
514 {NULL, NULL} /* sentinel */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000515};
516
Barry Warsaw92971171997-01-03 00:14:25 +0000517
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000518PyDoc_STRVAR(module_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000519"This module provides mechanisms to use signal handlers in Python.\n\
520\n\
521Functions:\n\
522\n\
523alarm() -- cause SIGALRM after a specified time [Unix only]\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000524setitimer() -- cause a signal (described below) after a specified\n\
525 float time and the timer may restart then [Unix only]\n\
526getitimer() -- get current value of timer [Unix only]\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000527signal() -- set the action for a given signal\n\
528getsignal() -- get the signal action for a given signal\n\
529pause() -- wait until a signal arrives [Unix only]\n\
530default_int_handler() -- default SIGINT handler\n\
531\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000532signal constants:\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000533SIG_DFL -- used to refer to the system default handler\n\
534SIG_IGN -- used to ignore the signal\n\
535NSIG -- number of defined signals\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000536SIGINT, SIGTERM, etc. -- signal numbers\n\
537\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000538itimer constants:\n\
539ITIMER_REAL -- decrements in real time, and delivers SIGALRM upon\n\
540 expiration\n\
541ITIMER_VIRTUAL -- decrements only when the process is executing,\n\
542 and delivers SIGVTALRM upon expiration\n\
543ITIMER_PROF -- decrements both when the process is executing and\n\
544 when the system is executing on behalf of the process.\n\
545 Coupled with ITIMER_VIRTUAL, this timer is usually\n\
546 used to profile the time spent by the application\n\
547 in user and kernel space. SIGPROF is delivered upon\n\
548 expiration.\n\
549\n\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000550*** IMPORTANT NOTICE ***\n\
551A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000552the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000553
Martin v. Löwis1a214512008-06-11 05:26:20 +0000554static struct PyModuleDef signalmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000555 PyModuleDef_HEAD_INIT,
556 "signal",
557 module_doc,
558 -1,
559 signal_methods,
560 NULL,
561 NULL,
562 NULL,
563 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +0000564};
565
Mark Hammondfe51c6d2002-08-02 02:27:13 +0000566PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +0000567PyInit_signal(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000568{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000569 PyObject *m, *d, *x;
570 int i;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000571
572#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000573 main_thread = PyThread_get_thread_ident();
574 main_pid = getpid();
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000575#endif
576
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000577 /* Create the module and add the functions */
578 m = PyModule_Create(&signalmodule);
579 if (m == NULL)
580 return NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000581
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000582 /* Add some symbolic constants to the module */
583 d = PyModule_GetDict(m);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000584
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000585 x = DefaultHandler = PyLong_FromVoidPtr((void *)SIG_DFL);
586 if (!x || PyDict_SetItemString(d, "SIG_DFL", x) < 0)
587 goto finally;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000588
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000589 x = IgnoreHandler = PyLong_FromVoidPtr((void *)SIG_IGN);
590 if (!x || PyDict_SetItemString(d, "SIG_IGN", x) < 0)
591 goto finally;
Barry Warsaw92971171997-01-03 00:14:25 +0000592
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000593 x = PyLong_FromLong((long)NSIG);
594 if (!x || PyDict_SetItemString(d, "NSIG", x) < 0)
595 goto finally;
596 Py_DECREF(x);
Barry Warsaw92971171997-01-03 00:14:25 +0000597
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000598 x = IntHandler = PyDict_GetItemString(d, "default_int_handler");
599 if (!x)
600 goto finally;
601 Py_INCREF(IntHandler);
Barry Warsaw92971171997-01-03 00:14:25 +0000602
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000603 Handlers[0].tripped = 0;
604 for (i = 1; i < NSIG; i++) {
605 void (*t)(int);
606 t = PyOS_getsig(i);
607 Handlers[i].tripped = 0;
608 if (t == SIG_DFL)
609 Handlers[i].func = DefaultHandler;
610 else if (t == SIG_IGN)
611 Handlers[i].func = IgnoreHandler;
612 else
613 Handlers[i].func = Py_None; /* None of our business */
614 Py_INCREF(Handlers[i].func);
615 }
616 if (Handlers[SIGINT].func == DefaultHandler) {
617 /* Install default int handler */
618 Py_INCREF(IntHandler);
619 Py_DECREF(Handlers[SIGINT].func);
620 Handlers[SIGINT].func = IntHandler;
621 old_siginthandler = PyOS_setsig(SIGINT, signal_handler);
622 }
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000623
624#ifdef SIGHUP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000625 x = PyLong_FromLong(SIGHUP);
626 PyDict_SetItemString(d, "SIGHUP", x);
627 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000628#endif
629#ifdef SIGINT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000630 x = PyLong_FromLong(SIGINT);
631 PyDict_SetItemString(d, "SIGINT", x);
632 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000633#endif
Tim Peters1ce3cf72001-10-01 17:58:40 +0000634#ifdef SIGBREAK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000635 x = PyLong_FromLong(SIGBREAK);
636 PyDict_SetItemString(d, "SIGBREAK", x);
637 Py_XDECREF(x);
Tim Peters1ce3cf72001-10-01 17:58:40 +0000638#endif
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000639#ifdef SIGQUIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000640 x = PyLong_FromLong(SIGQUIT);
641 PyDict_SetItemString(d, "SIGQUIT", x);
642 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000643#endif
644#ifdef SIGILL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000645 x = PyLong_FromLong(SIGILL);
646 PyDict_SetItemString(d, "SIGILL", x);
647 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000648#endif
649#ifdef SIGTRAP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000650 x = PyLong_FromLong(SIGTRAP);
651 PyDict_SetItemString(d, "SIGTRAP", x);
652 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000653#endif
654#ifdef SIGIOT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000655 x = PyLong_FromLong(SIGIOT);
656 PyDict_SetItemString(d, "SIGIOT", x);
657 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000658#endif
659#ifdef SIGABRT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000660 x = PyLong_FromLong(SIGABRT);
661 PyDict_SetItemString(d, "SIGABRT", x);
662 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000663#endif
664#ifdef SIGEMT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000665 x = PyLong_FromLong(SIGEMT);
666 PyDict_SetItemString(d, "SIGEMT", x);
667 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000668#endif
669#ifdef SIGFPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000670 x = PyLong_FromLong(SIGFPE);
671 PyDict_SetItemString(d, "SIGFPE", x);
672 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000673#endif
674#ifdef SIGKILL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000675 x = PyLong_FromLong(SIGKILL);
676 PyDict_SetItemString(d, "SIGKILL", x);
677 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000678#endif
679#ifdef SIGBUS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000680 x = PyLong_FromLong(SIGBUS);
681 PyDict_SetItemString(d, "SIGBUS", x);
682 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000683#endif
684#ifdef SIGSEGV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000685 x = PyLong_FromLong(SIGSEGV);
686 PyDict_SetItemString(d, "SIGSEGV", x);
687 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000688#endif
689#ifdef SIGSYS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000690 x = PyLong_FromLong(SIGSYS);
691 PyDict_SetItemString(d, "SIGSYS", x);
692 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000693#endif
694#ifdef SIGPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000695 x = PyLong_FromLong(SIGPIPE);
696 PyDict_SetItemString(d, "SIGPIPE", x);
697 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000698#endif
699#ifdef SIGALRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000700 x = PyLong_FromLong(SIGALRM);
701 PyDict_SetItemString(d, "SIGALRM", x);
702 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000703#endif
704#ifdef SIGTERM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000705 x = PyLong_FromLong(SIGTERM);
706 PyDict_SetItemString(d, "SIGTERM", x);
707 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000708#endif
709#ifdef SIGUSR1
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000710 x = PyLong_FromLong(SIGUSR1);
711 PyDict_SetItemString(d, "SIGUSR1", x);
712 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000713#endif
714#ifdef SIGUSR2
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000715 x = PyLong_FromLong(SIGUSR2);
716 PyDict_SetItemString(d, "SIGUSR2", x);
717 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000718#endif
719#ifdef SIGCLD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000720 x = PyLong_FromLong(SIGCLD);
721 PyDict_SetItemString(d, "SIGCLD", x);
722 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000723#endif
724#ifdef SIGCHLD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000725 x = PyLong_FromLong(SIGCHLD);
726 PyDict_SetItemString(d, "SIGCHLD", x);
727 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000728#endif
729#ifdef SIGPWR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000730 x = PyLong_FromLong(SIGPWR);
731 PyDict_SetItemString(d, "SIGPWR", x);
732 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000733#endif
734#ifdef SIGIO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000735 x = PyLong_FromLong(SIGIO);
736 PyDict_SetItemString(d, "SIGIO", x);
737 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000738#endif
739#ifdef SIGURG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000740 x = PyLong_FromLong(SIGURG);
741 PyDict_SetItemString(d, "SIGURG", x);
742 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000743#endif
744#ifdef SIGWINCH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000745 x = PyLong_FromLong(SIGWINCH);
746 PyDict_SetItemString(d, "SIGWINCH", x);
747 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000748#endif
749#ifdef SIGPOLL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000750 x = PyLong_FromLong(SIGPOLL);
751 PyDict_SetItemString(d, "SIGPOLL", x);
752 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000753#endif
754#ifdef SIGSTOP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000755 x = PyLong_FromLong(SIGSTOP);
756 PyDict_SetItemString(d, "SIGSTOP", x);
757 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000758#endif
759#ifdef SIGTSTP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000760 x = PyLong_FromLong(SIGTSTP);
761 PyDict_SetItemString(d, "SIGTSTP", x);
762 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000763#endif
764#ifdef SIGCONT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000765 x = PyLong_FromLong(SIGCONT);
766 PyDict_SetItemString(d, "SIGCONT", x);
767 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000768#endif
769#ifdef SIGTTIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000770 x = PyLong_FromLong(SIGTTIN);
771 PyDict_SetItemString(d, "SIGTTIN", x);
772 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000773#endif
774#ifdef SIGTTOU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000775 x = PyLong_FromLong(SIGTTOU);
776 PyDict_SetItemString(d, "SIGTTOU", x);
777 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000778#endif
779#ifdef SIGVTALRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000780 x = PyLong_FromLong(SIGVTALRM);
781 PyDict_SetItemString(d, "SIGVTALRM", x);
782 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000783#endif
784#ifdef SIGPROF
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000785 x = PyLong_FromLong(SIGPROF);
786 PyDict_SetItemString(d, "SIGPROF", x);
787 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000788#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000789#ifdef SIGXCPU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000790 x = PyLong_FromLong(SIGXCPU);
791 PyDict_SetItemString(d, "SIGXCPU", x);
792 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000793#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000794#ifdef SIGXFSZ
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000795 x = PyLong_FromLong(SIGXFSZ);
796 PyDict_SetItemString(d, "SIGXFSZ", x);
797 Py_XDECREF(x);
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000798#endif
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000799#ifdef SIGRTMIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000800 x = PyLong_FromLong(SIGRTMIN);
801 PyDict_SetItemString(d, "SIGRTMIN", x);
802 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000803#endif
804#ifdef SIGRTMAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000805 x = PyLong_FromLong(SIGRTMAX);
806 PyDict_SetItemString(d, "SIGRTMAX", x);
807 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000808#endif
Martin v. Löwis175af252002-01-12 11:43:25 +0000809#ifdef SIGINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000810 x = PyLong_FromLong(SIGINFO);
811 PyDict_SetItemString(d, "SIGINFO", x);
812 Py_XDECREF(x);
Martin v. Löwis175af252002-01-12 11:43:25 +0000813#endif
Martin v. Löwis823725e2008-03-24 13:39:54 +0000814
815#ifdef ITIMER_REAL
816 x = PyLong_FromLong(ITIMER_REAL);
817 PyDict_SetItemString(d, "ITIMER_REAL", x);
818 Py_DECREF(x);
819#endif
820#ifdef ITIMER_VIRTUAL
821 x = PyLong_FromLong(ITIMER_VIRTUAL);
822 PyDict_SetItemString(d, "ITIMER_VIRTUAL", x);
823 Py_DECREF(x);
824#endif
825#ifdef ITIMER_PROF
826 x = PyLong_FromLong(ITIMER_PROF);
827 PyDict_SetItemString(d, "ITIMER_PROF", x);
828 Py_DECREF(x);
829#endif
830
831#if defined (HAVE_SETITIMER) || defined (HAVE_GETITIMER)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000832 ItimerError = PyErr_NewException("signal.ItimerError",
833 PyExc_IOError, NULL);
Neal Norwitz2f99b242008-08-24 05:48:10 +0000834 if (ItimerError != NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000835 PyDict_SetItemString(d, "ItimerError", ItimerError);
Martin v. Löwis823725e2008-03-24 13:39:54 +0000836#endif
837
Brian Curtineb24d742010-04-12 17:16:38 +0000838#ifdef CTRL_C_EVENT
839 x = PyLong_FromLong(CTRL_C_EVENT);
840 PyDict_SetItemString(d, "CTRL_C_EVENT", x);
841 Py_DECREF(x);
842#endif
843
844#ifdef CTRL_BREAK_EVENT
845 x = PyLong_FromLong(CTRL_BREAK_EVENT);
846 PyDict_SetItemString(d, "CTRL_BREAK_EVENT", x);
847 Py_DECREF(x);
848#endif
849
Martin v. Löwis1a214512008-06-11 05:26:20 +0000850 if (PyErr_Occurred()) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000851 Py_DECREF(m);
852 m = NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +0000853 }
Barry Warsaw92971171997-01-03 00:14:25 +0000854
Barry Warsaw92971171997-01-03 00:14:25 +0000855 finally:
Martin v. Löwis1a214512008-06-11 05:26:20 +0000856 return m;
Guido van Rossum08c16611997-08-02 03:01:42 +0000857}
858
859static void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000860finisignal(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000861{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000862 int i;
863 PyObject *func;
Guido van Rossum08c16611997-08-02 03:01:42 +0000864
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000865 PyOS_setsig(SIGINT, old_siginthandler);
866 old_siginthandler = SIG_DFL;
Guido van Rossum08c16611997-08-02 03:01:42 +0000867
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000868 for (i = 1; i < NSIG; i++) {
869 func = Handlers[i].func;
870 Handlers[i].tripped = 0;
871 Handlers[i].func = NULL;
872 if (i != SIGINT && func != NULL && func != Py_None &&
873 func != DefaultHandler && func != IgnoreHandler)
874 PyOS_setsig(i, SIG_DFL);
875 Py_XDECREF(func);
876 }
Guido van Rossum08c16611997-08-02 03:01:42 +0000877
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000878 Py_XDECREF(IntHandler);
879 IntHandler = NULL;
880 Py_XDECREF(DefaultHandler);
881 DefaultHandler = NULL;
882 Py_XDECREF(IgnoreHandler);
883 IgnoreHandler = NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000884}
885
Barry Warsaw92971171997-01-03 00:14:25 +0000886
Barry Warsaw92971171997-01-03 00:14:25 +0000887/* Declared in pyerrors.h */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000888int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000889PyErr_CheckSignals(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000890{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000891 int i;
892 PyObject *f;
Barry Warsaw92971171997-01-03 00:14:25 +0000893
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000894 if (!is_tripped)
895 return 0;
Christian Heimesb76922a2007-12-11 01:06:40 +0000896
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000897#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000898 if (PyThread_get_thread_ident() != main_thread)
899 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000900#endif
Christian Heimesb76922a2007-12-11 01:06:40 +0000901
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000902 /*
903 * The is_tripped variable is meant to speed up the calls to
904 * PyErr_CheckSignals (both directly or via pending calls) when no
905 * signal has arrived. This variable is set to 1 when a signal arrives
906 * and it is set to 0 here, when we know some signals arrived. This way
907 * we can run the registered handlers with no signals blocked.
908 *
909 * NOTE: with this approach we can have a situation where is_tripped is
910 * 1 but we have no more signals to handle (Handlers[i].tripped
911 * is 0 for every signal i). This won't do us any harm (except
912 * we're gonna spent some cycles for nothing). This happens when
913 * we receive a signal i after we zero is_tripped and before we
914 * check Handlers[i].tripped.
915 */
916 is_tripped = 0;
Christian Heimesb76922a2007-12-11 01:06:40 +0000917
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000918 if (!(f = (PyObject *)PyEval_GetFrame()))
919 f = Py_None;
Christian Heimesb76922a2007-12-11 01:06:40 +0000920
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000921 for (i = 1; i < NSIG; i++) {
922 if (Handlers[i].tripped) {
923 PyObject *result = NULL;
924 PyObject *arglist = Py_BuildValue("(iO)", i, f);
925 Handlers[i].tripped = 0;
Barry Warsaw92971171997-01-03 00:14:25 +0000926
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000927 if (arglist) {
928 result = PyEval_CallObject(Handlers[i].func,
929 arglist);
930 Py_DECREF(arglist);
931 }
932 if (!result)
933 return -1;
Barry Warsaw92971171997-01-03 00:14:25 +0000934
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000935 Py_DECREF(result);
936 }
937 }
Christian Heimesb76922a2007-12-11 01:06:40 +0000938
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000939 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000940}
941
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000942
Barry Warsaw92971171997-01-03 00:14:25 +0000943/* Replacements for intrcheck.c functionality
944 * Declared in pyerrors.h
945 */
946void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000947PyErr_SetInterrupt(void)
Barry Warsaw92971171997-01-03 00:14:25 +0000948{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000949 is_tripped = 1;
950 Handlers[SIGINT].tripped = 1;
951 Py_AddPendingCall((int (*)(void *))PyErr_CheckSignals, NULL);
Barry Warsaw92971171997-01-03 00:14:25 +0000952}
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000953
954void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000955PyOS_InitInterrupts(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000956{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000957 PyObject *m = PyInit_signal();
958 if (m) {
Victor Stinner49d3f252010-10-17 01:24:53 +0000959 _PyImport_FixupBuiltin(m, "signal");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000960 Py_DECREF(m);
961 }
Guido van Rossum08c16611997-08-02 03:01:42 +0000962}
963
964void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000965PyOS_FiniInterrupts(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000966{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000967 finisignal();
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000968}
969
970int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000971PyOS_InterruptOccurred(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000972{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000973 if (Handlers[SIGINT].tripped) {
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000974#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000975 if (PyThread_get_thread_ident() != main_thread)
976 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000977#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000978 Handlers[SIGINT].tripped = 0;
979 return 1;
980 }
981 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000982}
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000983
984void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000985PyOS_AfterFork(void)
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000986{
987#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000988 PyEval_ReInitThreads();
989 main_thread = PyThread_get_thread_ident();
990 main_pid = getpid();
991 _PyImport_ReInitLock();
992 PyThread_ReInitTLS();
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000993#endif
994}