blob: 5aeee4c06defcabf1d38d6eb687aeb67d8c9cb2d [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#include "intrcheck.h"
8
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00009#ifdef MS_WINDOWS
Brian Curtine5aa8862010-04-02 23:26:06 +000010#include <Windows.h>
Andrew M. Kuchling4b81bc72010-02-22 23:12:00 +000011#ifdef HAVE_PROCESS_H
Guido van Rossum644a12b1997-04-09 19:24:53 +000012#include <process.h>
13#endif
Andrew M. Kuchling4b81bc72010-02-22 23:12:00 +000014#endif
Guido van Rossum644a12b1997-04-09 19:24:53 +000015
Andrew M. Kuchling4b81bc72010-02-22 23:12:00 +000016#ifdef HAVE_SIGNAL_H
Guido van Rossum398d9fe1994-05-11 08:59:13 +000017#include <signal.h>
Andrew M. Kuchling4b81bc72010-02-22 23:12:00 +000018#endif
19#ifdef HAVE_SYS_STAT_H
Guido van Rossum02de8972007-12-19 19:41:06 +000020#include <sys/stat.h>
Andrew M. Kuchling4b81bc72010-02-22 23:12:00 +000021#endif
Martin v. Löwisb74d0842008-03-24 13:54:23 +000022#ifdef HAVE_SYS_TIME_H
Martin v. Löwisaef18b12008-03-24 13:31:16 +000023#include <sys/time.h>
Martin v. Löwisb74d0842008-03-24 13:54:23 +000024#endif
Guido van Rossum02de8972007-12-19 19:41:06 +000025
Guido van Rossumbb4ba121994-06-23 11:25:45 +000026#ifndef SIG_ERR
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +000027#define SIG_ERR ((PyOS_sighandler_t)(-1))
Guido van Rossumbb4ba121994-06-23 11:25:45 +000028#endif
29
Andrew MacIntyre7bf68332002-03-03 02:59:16 +000030#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000031#define NSIG 12
32#include <process.h>
33#endif
34
Guido van Rossum3bbc62e1995-01-02 19:30:30 +000035#ifndef NSIG
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000036# if defined(_NSIG)
Antoine Pitrouc83ea132010-05-09 14:46:46 +000037# define NSIG _NSIG /* For BSD/SysV */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000038# elif defined(_SIGMAX)
Antoine Pitrouc83ea132010-05-09 14:46:46 +000039# define NSIG (_SIGMAX + 1) /* For QNX */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000040# elif defined(SIGMAX)
Antoine Pitrouc83ea132010-05-09 14:46:46 +000041# define NSIG (SIGMAX + 1) /* For djgpp */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000042# else
Antoine Pitrouc83ea132010-05-09 14:46:46 +000043# define NSIG 64 /* Use a reasonable default value */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000044# endif
Guido van Rossum3bbc62e1995-01-02 19:30:30 +000045#endif
46
47
Guido van Rossumbb4ba121994-06-23 11:25:45 +000048/*
49 NOTES ON THE INTERACTION BETWEEN SIGNALS AND THREADS
50
51 When threads are supported, we want the following semantics:
52
53 - only the main thread can set a signal handler
54 - any thread can get a signal handler
55 - signals are only delivered to the main thread
56
57 I.e. we don't support "synchronous signals" like SIGFPE (catching
58 this doesn't make much sense in Python anyway) nor do we support
59 signals as a means of inter-thread communication, since not all
60 thread implementations support that (at least our thread library
61 doesn't).
62
63 We still have the problem that in some implementations signals
64 generated by the keyboard (e.g. SIGINT) are delivered to all
65 threads (e.g. SGI), while in others (e.g. Solaris) such signals are
66 delivered to one random thread (an intermediate possibility would
Guido van Rossuma3c04b01995-01-12 11:29:01 +000067 be to deliver it to the main thread -- POSIX?). For now, we have
Guido van Rossumbb4ba121994-06-23 11:25:45 +000068 a working implementation that works in all three cases -- the
69 handler ignores signals if getpid() isn't the same as in the main
70 thread. XXX This is a hack.
71
Guido van Rossum9e8181b2000-09-19 00:46:46 +000072 GNU pth is a user-space threading library, and as such, all threads
73 run within the same process. In this case, if the currently running
74 thread is not the main_thread, send the signal to the main_thread.
Guido van Rossumbb4ba121994-06-23 11:25:45 +000075*/
76
77#ifdef WITH_THREAD
Guido van Rossum295b8e51997-06-06 21:16:41 +000078#include <sys/types.h> /* For pid_t */
Guido van Rossum49b56061998-10-01 20:42:43 +000079#include "pythread.h"
Guido van Rossumbb4ba121994-06-23 11:25:45 +000080static long main_thread;
81static pid_t main_pid;
82#endif
83
Barry Warsaw92971171997-01-03 00:14:25 +000084static struct {
Antoine Pitrouc83ea132010-05-09 14:46:46 +000085 int tripped;
86 PyObject *func;
Barry Warsaw92971171997-01-03 00:14:25 +000087} Handlers[NSIG];
Guido van Rossum398d9fe1994-05-11 08:59:13 +000088
Guido van Rossum02de8972007-12-19 19:41:06 +000089static sig_atomic_t wakeup_fd = -1;
90
Guido van Rossum137c49c2007-12-10 23:00:12 +000091/* Speed up sigcheck() when none tripped */
92static volatile sig_atomic_t is_tripped = 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +000093
Barry Warsaw92971171997-01-03 00:14:25 +000094static PyObject *DefaultHandler;
95static PyObject *IgnoreHandler;
96static PyObject *IntHandler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +000097
Martin v. Löwisf58de1b2001-03-06 12:13:56 +000098/* On Solaris 8, gcc will produce a warning that the function
99 declaration is not a prototype. This is caused by the definition of
100 SIG_DFL as (void (*)())0; the correct declaration would have been
101 (void (*)(int))0. */
102
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000103static PyOS_sighandler_t old_siginthandler = SIG_DFL;
Guido van Rossum08c16611997-08-02 03:01:42 +0000104
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000105#ifdef HAVE_GETITIMER
106static PyObject *ItimerError;
107
108/* auxiliary functions for setitimer/getitimer */
109static void
110timeval_from_double(double d, struct timeval *tv)
111{
112 tv->tv_sec = floor(d);
113 tv->tv_usec = fmod(d, 1.0) * 1000000.0;
114}
115
Christian Heimes32a66a02008-10-02 19:47:50 +0000116Py_LOCAL_INLINE(double)
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000117double_from_timeval(struct timeval *tv)
118{
119 return tv->tv_sec + (double)(tv->tv_usec / 1000000.0);
120}
121
122static PyObject *
123itimer_retval(struct itimerval *iv)
124{
125 PyObject *r, *v;
126
127 r = PyTuple_New(2);
128 if (r == NULL)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000129 return NULL;
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000130
131 if(!(v = PyFloat_FromDouble(double_from_timeval(&iv->it_value)))) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000132 Py_DECREF(r);
133 return NULL;
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000134 }
135
136 PyTuple_SET_ITEM(r, 0, v);
137
138 if(!(v = PyFloat_FromDouble(double_from_timeval(&iv->it_interval)))) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000139 Py_DECREF(r);
140 return NULL;
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000141 }
142
143 PyTuple_SET_ITEM(r, 1, v);
144
145 return r;
146}
147#endif
Barry Warsaw92971171997-01-03 00:14:25 +0000148
Guido van Rossume4485b01994-09-07 14:32:49 +0000149static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000150signal_default_int_handler(PyObject *self, PyObject *args)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000151{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000152 PyErr_SetNone(PyExc_KeyboardInterrupt);
153 return NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000154}
155
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000156PyDoc_STRVAR(default_int_handler_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000157"default_int_handler(...)\n\
158\n\
Michael W. Hudson24ec2112004-06-17 15:55:53 +0000159The default handler for SIGINT installed by Python.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000160It raises KeyboardInterrupt.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000161
Thomas Wouters0796b002000-07-22 23:49:30 +0000162
163static int
164checksignals_witharg(void * unused)
165{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000166 return PyErr_CheckSignals();
Thomas Wouters0796b002000-07-22 23:49:30 +0000167}
168
Tim Peters4f1b2082000-07-23 21:18:09 +0000169static void
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000170signal_handler(int sig_num)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000171{
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000172#ifdef WITH_THREAD
Guido van Rossum9e8181b2000-09-19 00:46:46 +0000173#ifdef WITH_PTH
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000174 if (PyThread_get_thread_ident() != main_thread) {
175 pth_raise(*(pth_t *) main_thread, sig_num);
176 return;
177 }
Guido van Rossum9e8181b2000-09-19 00:46:46 +0000178#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000179 /* See NOTES section above */
180 if (getpid() == main_pid) {
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000181#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000182 Handlers[sig_num].tripped = 1;
183 /* Set is_tripped after setting .tripped, as it gets
184 cleared in PyErr_CheckSignals() before .tripped. */
185 is_tripped = 1;
186 Py_AddPendingCall(checksignals_witharg, NULL);
187 if (wakeup_fd != -1)
188 write(wakeup_fd, "\0", 1);
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000189#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000190 }
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000191#endif
Guido van Rossum602099a1994-09-14 13:32:22 +0000192#ifdef SIGCHLD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000193 if (sig_num == SIGCHLD) {
194 /* To avoid infinite recursion, this signal remains
195 reset until explicit re-instated.
196 Don't clear the 'func' field as it is our pointer
197 to the Python handler... */
198 return;
199 }
Guido van Rossum602099a1994-09-14 13:32:22 +0000200#endif
Jean-Paul Calderonee54ddf12010-05-08 20:06:02 +0000201#ifndef HAVE_SIGACTION
Antoine Pitrouc83ea132010-05-09 14:46:46 +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 Calderonee54ddf12010-05-08 20:06:02 +0000206#endif
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000207}
Guido van Rossume4485b01994-09-07 14:32:49 +0000208
Guido van Rossum06d511d1995-03-10 15:13:48 +0000209
Guido van Rossum1171ee61997-08-22 20:42:00 +0000210#ifdef HAVE_ALARM
Guido van Rossume4485b01994-09-07 14:32:49 +0000211static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000212signal_alarm(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000213{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000214 int t;
215 if (!PyArg_ParseTuple(args, "i:alarm", &t))
216 return NULL;
217 /* alarm() returns the number of seconds remaining */
218 return PyInt_FromLong((long)alarm(t));
Guido van Rossumaa0f4c71994-08-23 13:49:37 +0000219}
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000220
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000221PyDoc_STRVAR(alarm_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000222"alarm(seconds)\n\
223\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000224Arrange for SIGALRM to arrive after the given number of seconds.");
Guido van Rossum06d511d1995-03-10 15:13:48 +0000225#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +0000226
Guido van Rossum1171ee61997-08-22 20:42:00 +0000227#ifdef HAVE_PAUSE
Guido van Rossuma597dde1995-01-10 20:56:29 +0000228static PyObject *
Neal Norwitz3a6f9782002-03-25 20:46:46 +0000229signal_pause(PyObject *self)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000230{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000231 Py_BEGIN_ALLOW_THREADS
232 (void)pause();
233 Py_END_ALLOW_THREADS
234 /* make sure that any exceptions that got raised are propagated
235 * back into Python
236 */
237 if (PyErr_CheckSignals())
238 return NULL;
Barry Warsaw92971171997-01-03 00:14:25 +0000239
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000240 Py_INCREF(Py_None);
241 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +0000242}
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000243PyDoc_STRVAR(pause_doc,
Barry Warsaw1ee36ff1998-07-21 22:41:18 +0000244"pause()\n\
245\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000246Wait until a signal arrives.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000247
Guido van Rossum06d511d1995-03-10 15:13:48 +0000248#endif
Guido van Rossume4485b01994-09-07 14:32:49 +0000249
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000250
Guido van Rossume4485b01994-09-07 14:32:49 +0000251static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000252signal_signal(PyObject *self, PyObject *args)
Guido van Rossume4485b01994-09-07 14:32:49 +0000253{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000254 PyObject *obj;
255 int sig_num;
256 PyObject *old_handler;
257 void (*func)(int);
258 if (!PyArg_ParseTuple(args, "iO:signal", &sig_num, &obj))
259 return NULL;
Brian Curtin24af0e92010-08-06 19:41:01 +0000260#ifdef MS_WINDOWS
261 /* Validate that sig_num is one of the allowable signals */
Brian Curtin84263332010-09-06 16:17:50 +0000262 switch (sig_num) {
263 case SIGABRT: break;
264 case SIGFPE: break;
265 case SIGILL: break;
266 case SIGINT: break;
267 case SIGSEGV: break;
268 case SIGTERM: break;
269 default:
270 PyErr_SetString(PyExc_ValueError, "invalid signal value");
271 return NULL;
Brian Curtin24af0e92010-08-06 19:41:01 +0000272 }
273#endif
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000274#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000275 if (PyThread_get_thread_ident() != main_thread) {
276 PyErr_SetString(PyExc_ValueError,
277 "signal only works in main thread");
278 return NULL;
279 }
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000280#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000281 if (sig_num < 1 || sig_num >= NSIG) {
282 PyErr_SetString(PyExc_ValueError,
283 "signal number out of range");
284 return NULL;
285 }
286 if (obj == IgnoreHandler)
287 func = SIG_IGN;
288 else if (obj == DefaultHandler)
289 func = SIG_DFL;
290 else if (!PyCallable_Check(obj)) {
291 PyErr_SetString(PyExc_TypeError,
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000292"signal handler must be signal.SIG_IGN, signal.SIG_DFL, or a callable object");
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000293 return NULL;
294 }
295 else
296 func = signal_handler;
297 if (PyOS_setsig(sig_num, func) == SIG_ERR) {
298 PyErr_SetFromErrno(PyExc_RuntimeError);
299 return NULL;
300 }
301 old_handler = Handlers[sig_num].func;
302 Handlers[sig_num].tripped = 0;
303 Py_INCREF(obj);
304 Handlers[sig_num].func = obj;
305 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000306}
307
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000308PyDoc_STRVAR(signal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000309"signal(sig, action) -> action\n\
310\n\
311Set the action for the given signal. The action can be SIG_DFL,\n\
312SIG_IGN, or a callable Python object. The previous action is\n\
313returned. See getsignal() for possible return values.\n\
314\n\
315*** IMPORTANT NOTICE ***\n\
316A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000317the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000318
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000319
Guido van Rossume4485b01994-09-07 14:32:49 +0000320static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000321signal_getsignal(PyObject *self, PyObject *args)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000322{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000323 int sig_num;
324 PyObject *old_handler;
325 if (!PyArg_ParseTuple(args, "i:getsignal", &sig_num))
326 return NULL;
327 if (sig_num < 1 || sig_num >= NSIG) {
328 PyErr_SetString(PyExc_ValueError,
329 "signal number out of range");
330 return NULL;
331 }
332 old_handler = Handlers[sig_num].func;
333 Py_INCREF(old_handler);
334 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000335}
336
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000337PyDoc_STRVAR(getsignal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000338"getsignal(sig) -> action\n\
339\n\
340Return the current action for the given signal. The return value can be:\n\
341SIG_IGN -- if the signal is being ignored\n\
342SIG_DFL -- if the default action for the signal is in effect\n\
343None -- if an unknown handler is in effect\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000344anything else -- the callable Python object used as a handler");
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000345
Facundo Batista7e251e82008-02-23 15:07:35 +0000346#ifdef HAVE_SIGINTERRUPT
347PyDoc_STRVAR(siginterrupt_doc,
348"siginterrupt(sig, flag) -> None\n\
349change system call restart behaviour: if flag is False, system calls\n\
350will be restarted when interrupted by signal sig, else system calls\n\
351will be interrupted.");
352
353static PyObject *
354signal_siginterrupt(PyObject *self, PyObject *args)
355{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000356 int sig_num;
357 int flag;
Facundo Batista7e251e82008-02-23 15:07:35 +0000358
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000359 if (!PyArg_ParseTuple(args, "ii:siginterrupt", &sig_num, &flag))
360 return NULL;
361 if (sig_num < 1 || sig_num >= NSIG) {
362 PyErr_SetString(PyExc_ValueError,
363 "signal number out of range");
364 return NULL;
365 }
366 if (siginterrupt(sig_num, flag)<0) {
367 PyErr_SetFromErrno(PyExc_RuntimeError);
368 return NULL;
369 }
Facundo Batista7e251e82008-02-23 15:07:35 +0000370
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000371 Py_INCREF(Py_None);
372 return Py_None;
Facundo Batista7e251e82008-02-23 15:07:35 +0000373}
374
375#endif
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000376
Guido van Rossum02de8972007-12-19 19:41:06 +0000377static PyObject *
378signal_set_wakeup_fd(PyObject *self, PyObject *args)
379{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000380 struct stat buf;
381 int fd, old_fd;
382 if (!PyArg_ParseTuple(args, "i:set_wakeup_fd", &fd))
383 return NULL;
Guido van Rossum02de8972007-12-19 19:41:06 +0000384#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000385 if (PyThread_get_thread_ident() != main_thread) {
386 PyErr_SetString(PyExc_ValueError,
387 "set_wakeup_fd only works in main thread");
388 return NULL;
389 }
Guido van Rossum02de8972007-12-19 19:41:06 +0000390#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000391 if (fd != -1 && fstat(fd, &buf) != 0) {
392 PyErr_SetString(PyExc_ValueError, "invalid fd");
393 return NULL;
394 }
395 old_fd = wakeup_fd;
396 wakeup_fd = fd;
397 return PyLong_FromLong(old_fd);
Guido van Rossum02de8972007-12-19 19:41:06 +0000398}
399
400PyDoc_STRVAR(set_wakeup_fd_doc,
401"set_wakeup_fd(fd) -> fd\n\
402\n\
403Sets the fd to be written to (with '\\0') when a signal\n\
404comes in. A library can use this to wakeup select or poll.\n\
405The previous fd is returned.\n\
406\n\
407The fd must be non-blocking.");
408
409/* C API for the same, without all the error checking */
410int
411PySignal_SetWakeupFd(int fd)
412{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000413 int old_fd = wakeup_fd;
414 if (fd < 0)
415 fd = -1;
416 wakeup_fd = fd;
417 return old_fd;
Guido van Rossum02de8972007-12-19 19:41:06 +0000418}
419
420
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000421#ifdef HAVE_SETITIMER
422static PyObject *
423signal_setitimer(PyObject *self, PyObject *args)
424{
425 double first;
426 double interval = 0;
427 int which;
428 struct itimerval new, old;
429
430 if(!PyArg_ParseTuple(args, "id|d:setitimer", &which, &first, &interval))
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000431 return NULL;
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000432
433 timeval_from_double(first, &new.it_value);
434 timeval_from_double(interval, &new.it_interval);
435 /* Let OS check "which" value */
436 if (setitimer(which, &new, &old) != 0) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000437 PyErr_SetFromErrno(ItimerError);
438 return NULL;
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000439 }
440
441 return itimer_retval(&old);
442}
443
444PyDoc_STRVAR(setitimer_doc,
445"setitimer(which, seconds[, interval])\n\
446\n\
447Sets given itimer (one of ITIMER_REAL, ITIMER_VIRTUAL\n\
448or ITIMER_PROF) to fire after value seconds and after\n\
449that every interval seconds.\n\
450The itimer can be cleared by setting seconds to zero.\n\
451\n\
452Returns old values as a tuple: (delay, interval).");
453#endif
454
455
456#ifdef HAVE_GETITIMER
457static PyObject *
458signal_getitimer(PyObject *self, PyObject *args)
459{
460 int which;
461 struct itimerval old;
462
463 if (!PyArg_ParseTuple(args, "i:getitimer", &which))
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000464 return NULL;
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000465
466 if (getitimer(which, &old) != 0) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000467 PyErr_SetFromErrno(ItimerError);
468 return NULL;
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000469 }
470
471 return itimer_retval(&old);
472}
473
474PyDoc_STRVAR(getitimer_doc,
475"getitimer(which)\n\
476\n\
477Returns current value of given itimer.");
478#endif
479
480
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000481/* List of functions defined in the module */
Barry Warsaw92971171997-01-03 00:14:25 +0000482static PyMethodDef signal_methods[] = {
Guido van Rossum1171ee61997-08-22 20:42:00 +0000483#ifdef HAVE_ALARM
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000484 {"alarm", signal_alarm, METH_VARARGS, alarm_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000485#endif
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000486#ifdef HAVE_SETITIMER
487 {"setitimer", signal_setitimer, METH_VARARGS, setitimer_doc},
488#endif
489#ifdef HAVE_GETITIMER
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000490 {"getitimer", signal_getitimer, METH_VARARGS, getitimer_doc},
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000491#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000492 {"signal", signal_signal, METH_VARARGS, signal_doc},
493 {"getsignal", signal_getsignal, METH_VARARGS, getsignal_doc},
494 {"set_wakeup_fd", signal_set_wakeup_fd, METH_VARARGS, set_wakeup_fd_doc},
Facundo Batista7e251e82008-02-23 15:07:35 +0000495#ifdef HAVE_SIGINTERRUPT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000496 {"siginterrupt", signal_siginterrupt, METH_VARARGS, siginterrupt_doc},
Facundo Batista7e251e82008-02-23 15:07:35 +0000497#endif
Guido van Rossum1171ee61997-08-22 20:42:00 +0000498#ifdef HAVE_PAUSE
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000499 {"pause", (PyCFunction)signal_pause,
500 METH_NOARGS,pause_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000501#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000502 {"default_int_handler", signal_default_int_handler,
503 METH_VARARGS, default_int_handler_doc},
504 {NULL, NULL} /* sentinel */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000505};
506
Barry Warsaw92971171997-01-03 00:14:25 +0000507
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000508PyDoc_STRVAR(module_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000509"This module provides mechanisms to use signal handlers in Python.\n\
510\n\
511Functions:\n\
512\n\
513alarm() -- cause SIGALRM after a specified time [Unix only]\n\
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000514setitimer() -- cause a signal (described below) after a specified\n\
515 float time and the timer may restart then [Unix only]\n\
516getitimer() -- get current value of timer [Unix only]\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000517signal() -- set the action for a given signal\n\
518getsignal() -- get the signal action for a given signal\n\
519pause() -- wait until a signal arrives [Unix only]\n\
520default_int_handler() -- default SIGINT handler\n\
521\n\
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000522signal constants:\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000523SIG_DFL -- used to refer to the system default handler\n\
524SIG_IGN -- used to ignore the signal\n\
525NSIG -- number of defined signals\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000526SIGINT, SIGTERM, etc. -- signal numbers\n\
527\n\
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000528itimer constants:\n\
529ITIMER_REAL -- decrements in real time, and delivers SIGALRM upon\n\
530 expiration\n\
531ITIMER_VIRTUAL -- decrements only when the process is executing,\n\
532 and delivers SIGVTALRM upon expiration\n\
533ITIMER_PROF -- decrements both when the process is executing and\n\
534 when the system is executing on behalf of the process.\n\
535 Coupled with ITIMER_VIRTUAL, this timer is usually\n\
536 used to profile the time spent by the application\n\
537 in user and kernel space. SIGPROF is delivered upon\n\
538 expiration.\n\
539\n\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000540*** IMPORTANT NOTICE ***\n\
541A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000542the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000543
Mark Hammondfe51c6d2002-08-02 02:27:13 +0000544PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000545initsignal(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000546{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000547 PyObject *m, *d, *x;
548 int i;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000549
550#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000551 main_thread = PyThread_get_thread_ident();
552 main_pid = getpid();
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000553#endif
554
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000555 /* Create the module and add the functions */
556 m = Py_InitModule3("signal", signal_methods, module_doc);
557 if (m == NULL)
558 return;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000559
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000560 /* Add some symbolic constants to the module */
561 d = PyModule_GetDict(m);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000562
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000563 x = DefaultHandler = PyLong_FromVoidPtr((void *)SIG_DFL);
564 if (!x || PyDict_SetItemString(d, "SIG_DFL", x) < 0)
565 goto finally;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000566
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000567 x = IgnoreHandler = PyLong_FromVoidPtr((void *)SIG_IGN);
568 if (!x || PyDict_SetItemString(d, "SIG_IGN", x) < 0)
569 goto finally;
Barry Warsaw92971171997-01-03 00:14:25 +0000570
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000571 x = PyInt_FromLong((long)NSIG);
572 if (!x || PyDict_SetItemString(d, "NSIG", x) < 0)
573 goto finally;
574 Py_DECREF(x);
Barry Warsaw92971171997-01-03 00:14:25 +0000575
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000576 x = IntHandler = PyDict_GetItemString(d, "default_int_handler");
577 if (!x)
578 goto finally;
579 Py_INCREF(IntHandler);
Barry Warsaw92971171997-01-03 00:14:25 +0000580
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000581 Handlers[0].tripped = 0;
582 for (i = 1; i < NSIG; i++) {
583 void (*t)(int);
584 t = PyOS_getsig(i);
585 Handlers[i].tripped = 0;
586 if (t == SIG_DFL)
587 Handlers[i].func = DefaultHandler;
588 else if (t == SIG_IGN)
589 Handlers[i].func = IgnoreHandler;
590 else
591 Handlers[i].func = Py_None; /* None of our business */
592 Py_INCREF(Handlers[i].func);
593 }
594 if (Handlers[SIGINT].func == DefaultHandler) {
595 /* Install default int handler */
596 Py_INCREF(IntHandler);
597 Py_DECREF(Handlers[SIGINT].func);
598 Handlers[SIGINT].func = IntHandler;
599 old_siginthandler = PyOS_setsig(SIGINT, signal_handler);
600 }
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000601
602#ifdef SIGHUP
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000603 x = PyInt_FromLong(SIGHUP);
604 PyDict_SetItemString(d, "SIGHUP", x);
605 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000606#endif
607#ifdef SIGINT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000608 x = PyInt_FromLong(SIGINT);
609 PyDict_SetItemString(d, "SIGINT", x);
610 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000611#endif
Tim Peters1ce3cf72001-10-01 17:58:40 +0000612#ifdef SIGBREAK
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000613 x = PyInt_FromLong(SIGBREAK);
614 PyDict_SetItemString(d, "SIGBREAK", x);
615 Py_XDECREF(x);
Tim Peters1ce3cf72001-10-01 17:58:40 +0000616#endif
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000617#ifdef SIGQUIT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000618 x = PyInt_FromLong(SIGQUIT);
619 PyDict_SetItemString(d, "SIGQUIT", x);
620 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000621#endif
622#ifdef SIGILL
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000623 x = PyInt_FromLong(SIGILL);
624 PyDict_SetItemString(d, "SIGILL", x);
625 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000626#endif
627#ifdef SIGTRAP
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000628 x = PyInt_FromLong(SIGTRAP);
629 PyDict_SetItemString(d, "SIGTRAP", x);
630 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000631#endif
632#ifdef SIGIOT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000633 x = PyInt_FromLong(SIGIOT);
634 PyDict_SetItemString(d, "SIGIOT", x);
635 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000636#endif
637#ifdef SIGABRT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000638 x = PyInt_FromLong(SIGABRT);
639 PyDict_SetItemString(d, "SIGABRT", x);
640 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000641#endif
642#ifdef SIGEMT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000643 x = PyInt_FromLong(SIGEMT);
644 PyDict_SetItemString(d, "SIGEMT", x);
645 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000646#endif
647#ifdef SIGFPE
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000648 x = PyInt_FromLong(SIGFPE);
649 PyDict_SetItemString(d, "SIGFPE", x);
650 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000651#endif
652#ifdef SIGKILL
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000653 x = PyInt_FromLong(SIGKILL);
654 PyDict_SetItemString(d, "SIGKILL", x);
655 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000656#endif
657#ifdef SIGBUS
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000658 x = PyInt_FromLong(SIGBUS);
659 PyDict_SetItemString(d, "SIGBUS", x);
660 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000661#endif
662#ifdef SIGSEGV
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000663 x = PyInt_FromLong(SIGSEGV);
664 PyDict_SetItemString(d, "SIGSEGV", x);
665 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000666#endif
667#ifdef SIGSYS
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000668 x = PyInt_FromLong(SIGSYS);
669 PyDict_SetItemString(d, "SIGSYS", x);
670 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000671#endif
672#ifdef SIGPIPE
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000673 x = PyInt_FromLong(SIGPIPE);
674 PyDict_SetItemString(d, "SIGPIPE", x);
675 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000676#endif
677#ifdef SIGALRM
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000678 x = PyInt_FromLong(SIGALRM);
679 PyDict_SetItemString(d, "SIGALRM", x);
680 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000681#endif
682#ifdef SIGTERM
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000683 x = PyInt_FromLong(SIGTERM);
684 PyDict_SetItemString(d, "SIGTERM", x);
685 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000686#endif
687#ifdef SIGUSR1
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000688 x = PyInt_FromLong(SIGUSR1);
689 PyDict_SetItemString(d, "SIGUSR1", x);
690 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000691#endif
692#ifdef SIGUSR2
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000693 x = PyInt_FromLong(SIGUSR2);
694 PyDict_SetItemString(d, "SIGUSR2", x);
695 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000696#endif
697#ifdef SIGCLD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000698 x = PyInt_FromLong(SIGCLD);
699 PyDict_SetItemString(d, "SIGCLD", x);
700 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000701#endif
702#ifdef SIGCHLD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000703 x = PyInt_FromLong(SIGCHLD);
704 PyDict_SetItemString(d, "SIGCHLD", x);
705 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000706#endif
707#ifdef SIGPWR
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000708 x = PyInt_FromLong(SIGPWR);
709 PyDict_SetItemString(d, "SIGPWR", x);
710 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000711#endif
712#ifdef SIGIO
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000713 x = PyInt_FromLong(SIGIO);
714 PyDict_SetItemString(d, "SIGIO", x);
715 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000716#endif
717#ifdef SIGURG
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000718 x = PyInt_FromLong(SIGURG);
719 PyDict_SetItemString(d, "SIGURG", x);
720 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000721#endif
722#ifdef SIGWINCH
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000723 x = PyInt_FromLong(SIGWINCH);
724 PyDict_SetItemString(d, "SIGWINCH", x);
725 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000726#endif
727#ifdef SIGPOLL
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000728 x = PyInt_FromLong(SIGPOLL);
729 PyDict_SetItemString(d, "SIGPOLL", x);
730 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000731#endif
732#ifdef SIGSTOP
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000733 x = PyInt_FromLong(SIGSTOP);
734 PyDict_SetItemString(d, "SIGSTOP", x);
735 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000736#endif
737#ifdef SIGTSTP
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000738 x = PyInt_FromLong(SIGTSTP);
739 PyDict_SetItemString(d, "SIGTSTP", x);
740 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000741#endif
742#ifdef SIGCONT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000743 x = PyInt_FromLong(SIGCONT);
744 PyDict_SetItemString(d, "SIGCONT", x);
745 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000746#endif
747#ifdef SIGTTIN
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000748 x = PyInt_FromLong(SIGTTIN);
749 PyDict_SetItemString(d, "SIGTTIN", x);
750 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000751#endif
752#ifdef SIGTTOU
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000753 x = PyInt_FromLong(SIGTTOU);
754 PyDict_SetItemString(d, "SIGTTOU", x);
755 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000756#endif
757#ifdef SIGVTALRM
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000758 x = PyInt_FromLong(SIGVTALRM);
759 PyDict_SetItemString(d, "SIGVTALRM", x);
760 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000761#endif
762#ifdef SIGPROF
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000763 x = PyInt_FromLong(SIGPROF);
764 PyDict_SetItemString(d, "SIGPROF", x);
765 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000766#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000767#ifdef SIGXCPU
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000768 x = PyInt_FromLong(SIGXCPU);
769 PyDict_SetItemString(d, "SIGXCPU", x);
770 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000771#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000772#ifdef SIGXFSZ
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000773 x = PyInt_FromLong(SIGXFSZ);
774 PyDict_SetItemString(d, "SIGXFSZ", x);
775 Py_XDECREF(x);
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000776#endif
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000777#ifdef SIGRTMIN
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000778 x = PyInt_FromLong(SIGRTMIN);
779 PyDict_SetItemString(d, "SIGRTMIN", x);
780 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000781#endif
782#ifdef SIGRTMAX
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000783 x = PyInt_FromLong(SIGRTMAX);
784 PyDict_SetItemString(d, "SIGRTMAX", x);
785 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000786#endif
Martin v. Löwis175af252002-01-12 11:43:25 +0000787#ifdef SIGINFO
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000788 x = PyInt_FromLong(SIGINFO);
789 PyDict_SetItemString(d, "SIGINFO", x);
790 Py_XDECREF(x);
Martin v. Löwis175af252002-01-12 11:43:25 +0000791#endif
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000792
793#ifdef ITIMER_REAL
794 x = PyLong_FromLong(ITIMER_REAL);
795 PyDict_SetItemString(d, "ITIMER_REAL", x);
796 Py_DECREF(x);
797#endif
798#ifdef ITIMER_VIRTUAL
799 x = PyLong_FromLong(ITIMER_VIRTUAL);
800 PyDict_SetItemString(d, "ITIMER_VIRTUAL", x);
801 Py_DECREF(x);
802#endif
803#ifdef ITIMER_PROF
804 x = PyLong_FromLong(ITIMER_PROF);
805 PyDict_SetItemString(d, "ITIMER_PROF", x);
806 Py_DECREF(x);
807#endif
808
809#if defined (HAVE_SETITIMER) || defined (HAVE_GETITIMER)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000810 ItimerError = PyErr_NewException("signal.ItimerError",
811 PyExc_IOError, NULL);
Neal Norwitz18aa3882008-08-24 05:04:52 +0000812 if (ItimerError != NULL)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000813 PyDict_SetItemString(d, "ItimerError", ItimerError);
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000814#endif
815
Brian Curtine5aa8862010-04-02 23:26:06 +0000816#ifdef CTRL_C_EVENT
817 x = PyInt_FromLong(CTRL_C_EVENT);
818 PyDict_SetItemString(d, "CTRL_C_EVENT", x);
819 Py_DECREF(x);
820#endif
821
822#ifdef CTRL_BREAK_EVENT
823 x = PyInt_FromLong(CTRL_BREAK_EVENT);
824 PyDict_SetItemString(d, "CTRL_BREAK_EVENT", x);
825 Py_DECREF(x);
826#endif
827
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000828 if (!PyErr_Occurred())
Guido van Rossum08c16611997-08-02 03:01:42 +0000829 return;
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000830
831 /* Check for errors */
832 finally:
833 return;
Guido van Rossum08c16611997-08-02 03:01:42 +0000834}
835
836static void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000837finisignal(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000838{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000839 int i;
840 PyObject *func;
Guido van Rossum08c16611997-08-02 03:01:42 +0000841
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000842 PyOS_setsig(SIGINT, old_siginthandler);
843 old_siginthandler = SIG_DFL;
Guido van Rossum08c16611997-08-02 03:01:42 +0000844
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000845 for (i = 1; i < NSIG; i++) {
846 func = Handlers[i].func;
847 Handlers[i].tripped = 0;
848 Handlers[i].func = NULL;
849 if (i != SIGINT && func != NULL && func != Py_None &&
850 func != DefaultHandler && func != IgnoreHandler)
851 PyOS_setsig(i, SIG_DFL);
852 Py_XDECREF(func);
853 }
Guido van Rossum08c16611997-08-02 03:01:42 +0000854
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000855 Py_XDECREF(IntHandler);
856 IntHandler = NULL;
857 Py_XDECREF(DefaultHandler);
858 DefaultHandler = NULL;
859 Py_XDECREF(IgnoreHandler);
860 IgnoreHandler = NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000861}
862
Barry Warsaw92971171997-01-03 00:14:25 +0000863
Barry Warsaw92971171997-01-03 00:14:25 +0000864/* Declared in pyerrors.h */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000865int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000866PyErr_CheckSignals(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000867{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000868 int i;
869 PyObject *f;
Barry Warsaw92971171997-01-03 00:14:25 +0000870
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000871 if (!is_tripped)
872 return 0;
Guido van Rossum137c49c2007-12-10 23:00:12 +0000873
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000874#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000875 if (PyThread_get_thread_ident() != main_thread)
876 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000877#endif
Guido van Rossum137c49c2007-12-10 23:00:12 +0000878
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000879 /*
880 * The is_tripped variable is meant to speed up the calls to
881 * PyErr_CheckSignals (both directly or via pending calls) when no
882 * signal has arrived. This variable is set to 1 when a signal arrives
883 * and it is set to 0 here, when we know some signals arrived. This way
884 * we can run the registered handlers with no signals blocked.
885 *
886 * NOTE: with this approach we can have a situation where is_tripped is
887 * 1 but we have no more signals to handle (Handlers[i].tripped
888 * is 0 for every signal i). This won't do us any harm (except
889 * we're gonna spent some cycles for nothing). This happens when
890 * we receive a signal i after we zero is_tripped and before we
891 * check Handlers[i].tripped.
892 */
893 is_tripped = 0;
Guido van Rossum137c49c2007-12-10 23:00:12 +0000894
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000895 if (!(f = (PyObject *)PyEval_GetFrame()))
896 f = Py_None;
Guido van Rossum137c49c2007-12-10 23:00:12 +0000897
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000898 for (i = 1; i < NSIG; i++) {
899 if (Handlers[i].tripped) {
900 PyObject *result = NULL;
901 PyObject *arglist = Py_BuildValue("(iO)", i, f);
902 Handlers[i].tripped = 0;
Barry Warsaw92971171997-01-03 00:14:25 +0000903
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000904 if (arglist) {
905 result = PyEval_CallObject(Handlers[i].func,
906 arglist);
907 Py_DECREF(arglist);
908 }
909 if (!result)
910 return -1;
Barry Warsaw92971171997-01-03 00:14:25 +0000911
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000912 Py_DECREF(result);
913 }
914 }
Guido van Rossum137c49c2007-12-10 23:00:12 +0000915
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000916 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000917}
918
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000919
Barry Warsaw92971171997-01-03 00:14:25 +0000920/* Replacements for intrcheck.c functionality
921 * Declared in pyerrors.h
922 */
923void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000924PyErr_SetInterrupt(void)
Barry Warsaw92971171997-01-03 00:14:25 +0000925{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000926 is_tripped = 1;
927 Handlers[SIGINT].tripped = 1;
928 Py_AddPendingCall((int (*)(void *))PyErr_CheckSignals, NULL);
Barry Warsaw92971171997-01-03 00:14:25 +0000929}
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000930
931void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000932PyOS_InitInterrupts(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000933{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000934 initsignal();
935 _PyImport_FixupExtension("signal", "signal");
Guido van Rossum08c16611997-08-02 03:01:42 +0000936}
937
938void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000939PyOS_FiniInterrupts(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000940{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000941 finisignal();
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000942}
943
944int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000945PyOS_InterruptOccurred(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000946{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000947 if (Handlers[SIGINT].tripped) {
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000948#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000949 if (PyThread_get_thread_ident() != main_thread)
950 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000951#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000952 Handlers[SIGINT].tripped = 0;
953 return 1;
954 }
955 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000956}
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000957
958void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000959PyOS_AfterFork(void)
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000960{
961#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000962 PyEval_ReInitThreads();
963 main_thread = PyThread_get_thread_ident();
964 main_pid = getpid();
965 _PyImport_ReInitLock();
966 PyThread_ReInitTLS();
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000967#endif
968}