blob: 86759787c4e7a26ade1986db6eff7938552d6f1e [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);
Brian Curtin24af0e92010-08-06 19:41:01 +0000258#ifdef MS_WINDOWS
259 int cur_sig, num_valid_sigs = 6;
260 static int valid_sigs[] = {SIGABRT, SIGFPE, SIGILL, SIGINT,
261 SIGSEGV, SIGTERM};
262 BOOL valid_sig = FALSE;
263#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000264 if (!PyArg_ParseTuple(args, "iO:signal", &sig_num, &obj))
265 return NULL;
Brian Curtin24af0e92010-08-06 19:41:01 +0000266#ifdef MS_WINDOWS
267 /* Validate that sig_num is one of the allowable signals */
268 for (cur_sig = 0; cur_sig < num_valid_sigs; cur_sig++)
269 valid_sig |= (sig_num == valid_sigs[cur_sig]);
270 if (!valid_sig) {
271 PyErr_SetString(PyExc_ValueError, "signal number out of range");
272 return NULL;
273 }
274#endif
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000275#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000276 if (PyThread_get_thread_ident() != main_thread) {
277 PyErr_SetString(PyExc_ValueError,
278 "signal only works in main thread");
279 return NULL;
280 }
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000281#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000282 if (sig_num < 1 || sig_num >= NSIG) {
283 PyErr_SetString(PyExc_ValueError,
284 "signal number out of range");
285 return NULL;
286 }
287 if (obj == IgnoreHandler)
288 func = SIG_IGN;
289 else if (obj == DefaultHandler)
290 func = SIG_DFL;
291 else if (!PyCallable_Check(obj)) {
292 PyErr_SetString(PyExc_TypeError,
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000293"signal handler must be signal.SIG_IGN, signal.SIG_DFL, or a callable object");
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000294 return NULL;
295 }
296 else
297 func = signal_handler;
298 if (PyOS_setsig(sig_num, func) == SIG_ERR) {
299 PyErr_SetFromErrno(PyExc_RuntimeError);
300 return NULL;
301 }
302 old_handler = Handlers[sig_num].func;
303 Handlers[sig_num].tripped = 0;
304 Py_INCREF(obj);
305 Handlers[sig_num].func = obj;
306 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000307}
308
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000309PyDoc_STRVAR(signal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000310"signal(sig, action) -> action\n\
311\n\
312Set the action for the given signal. The action can be SIG_DFL,\n\
313SIG_IGN, or a callable Python object. The previous action is\n\
314returned. See getsignal() for possible return values.\n\
315\n\
316*** IMPORTANT NOTICE ***\n\
317A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000318the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000319
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000320
Guido van Rossume4485b01994-09-07 14:32:49 +0000321static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000322signal_getsignal(PyObject *self, PyObject *args)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000323{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000324 int sig_num;
325 PyObject *old_handler;
326 if (!PyArg_ParseTuple(args, "i:getsignal", &sig_num))
327 return NULL;
328 if (sig_num < 1 || sig_num >= NSIG) {
329 PyErr_SetString(PyExc_ValueError,
330 "signal number out of range");
331 return NULL;
332 }
333 old_handler = Handlers[sig_num].func;
334 Py_INCREF(old_handler);
335 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000336}
337
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000338PyDoc_STRVAR(getsignal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000339"getsignal(sig) -> action\n\
340\n\
341Return the current action for the given signal. The return value can be:\n\
342SIG_IGN -- if the signal is being ignored\n\
343SIG_DFL -- if the default action for the signal is in effect\n\
344None -- if an unknown handler is in effect\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000345anything else -- the callable Python object used as a handler");
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000346
Facundo Batista7e251e82008-02-23 15:07:35 +0000347#ifdef HAVE_SIGINTERRUPT
348PyDoc_STRVAR(siginterrupt_doc,
349"siginterrupt(sig, flag) -> None\n\
350change system call restart behaviour: if flag is False, system calls\n\
351will be restarted when interrupted by signal sig, else system calls\n\
352will be interrupted.");
353
354static PyObject *
355signal_siginterrupt(PyObject *self, PyObject *args)
356{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000357 int sig_num;
358 int flag;
Facundo Batista7e251e82008-02-23 15:07:35 +0000359
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000360 if (!PyArg_ParseTuple(args, "ii:siginterrupt", &sig_num, &flag))
361 return NULL;
362 if (sig_num < 1 || sig_num >= NSIG) {
363 PyErr_SetString(PyExc_ValueError,
364 "signal number out of range");
365 return NULL;
366 }
367 if (siginterrupt(sig_num, flag)<0) {
368 PyErr_SetFromErrno(PyExc_RuntimeError);
369 return NULL;
370 }
Facundo Batista7e251e82008-02-23 15:07:35 +0000371
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000372 Py_INCREF(Py_None);
373 return Py_None;
Facundo Batista7e251e82008-02-23 15:07:35 +0000374}
375
376#endif
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000377
Guido van Rossum02de8972007-12-19 19:41:06 +0000378static PyObject *
379signal_set_wakeup_fd(PyObject *self, PyObject *args)
380{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000381 struct stat buf;
382 int fd, old_fd;
383 if (!PyArg_ParseTuple(args, "i:set_wakeup_fd", &fd))
384 return NULL;
Guido van Rossum02de8972007-12-19 19:41:06 +0000385#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000386 if (PyThread_get_thread_ident() != main_thread) {
387 PyErr_SetString(PyExc_ValueError,
388 "set_wakeup_fd only works in main thread");
389 return NULL;
390 }
Guido van Rossum02de8972007-12-19 19:41:06 +0000391#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000392 if (fd != -1 && fstat(fd, &buf) != 0) {
393 PyErr_SetString(PyExc_ValueError, "invalid fd");
394 return NULL;
395 }
396 old_fd = wakeup_fd;
397 wakeup_fd = fd;
398 return PyLong_FromLong(old_fd);
Guido van Rossum02de8972007-12-19 19:41:06 +0000399}
400
401PyDoc_STRVAR(set_wakeup_fd_doc,
402"set_wakeup_fd(fd) -> fd\n\
403\n\
404Sets the fd to be written to (with '\\0') when a signal\n\
405comes in. A library can use this to wakeup select or poll.\n\
406The previous fd is returned.\n\
407\n\
408The fd must be non-blocking.");
409
410/* C API for the same, without all the error checking */
411int
412PySignal_SetWakeupFd(int fd)
413{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000414 int old_fd = wakeup_fd;
415 if (fd < 0)
416 fd = -1;
417 wakeup_fd = fd;
418 return old_fd;
Guido van Rossum02de8972007-12-19 19:41:06 +0000419}
420
421
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000422#ifdef HAVE_SETITIMER
423static PyObject *
424signal_setitimer(PyObject *self, PyObject *args)
425{
426 double first;
427 double interval = 0;
428 int which;
429 struct itimerval new, old;
430
431 if(!PyArg_ParseTuple(args, "id|d:setitimer", &which, &first, &interval))
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000432 return NULL;
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000433
434 timeval_from_double(first, &new.it_value);
435 timeval_from_double(interval, &new.it_interval);
436 /* Let OS check "which" value */
437 if (setitimer(which, &new, &old) != 0) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000438 PyErr_SetFromErrno(ItimerError);
439 return NULL;
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000440 }
441
442 return itimer_retval(&old);
443}
444
445PyDoc_STRVAR(setitimer_doc,
446"setitimer(which, seconds[, interval])\n\
447\n\
448Sets given itimer (one of ITIMER_REAL, ITIMER_VIRTUAL\n\
449or ITIMER_PROF) to fire after value seconds and after\n\
450that every interval seconds.\n\
451The itimer can be cleared by setting seconds to zero.\n\
452\n\
453Returns old values as a tuple: (delay, interval).");
454#endif
455
456
457#ifdef HAVE_GETITIMER
458static PyObject *
459signal_getitimer(PyObject *self, PyObject *args)
460{
461 int which;
462 struct itimerval old;
463
464 if (!PyArg_ParseTuple(args, "i:getitimer", &which))
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000465 return NULL;
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000466
467 if (getitimer(which, &old) != 0) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000468 PyErr_SetFromErrno(ItimerError);
469 return NULL;
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000470 }
471
472 return itimer_retval(&old);
473}
474
475PyDoc_STRVAR(getitimer_doc,
476"getitimer(which)\n\
477\n\
478Returns current value of given itimer.");
479#endif
480
481
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000482/* List of functions defined in the module */
Barry Warsaw92971171997-01-03 00:14:25 +0000483static PyMethodDef signal_methods[] = {
Guido van Rossum1171ee61997-08-22 20:42:00 +0000484#ifdef HAVE_ALARM
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000485 {"alarm", signal_alarm, METH_VARARGS, alarm_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000486#endif
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000487#ifdef HAVE_SETITIMER
488 {"setitimer", signal_setitimer, METH_VARARGS, setitimer_doc},
489#endif
490#ifdef HAVE_GETITIMER
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000491 {"getitimer", signal_getitimer, METH_VARARGS, getitimer_doc},
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000492#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000493 {"signal", signal_signal, METH_VARARGS, signal_doc},
494 {"getsignal", signal_getsignal, METH_VARARGS, getsignal_doc},
495 {"set_wakeup_fd", signal_set_wakeup_fd, METH_VARARGS, set_wakeup_fd_doc},
Facundo Batista7e251e82008-02-23 15:07:35 +0000496#ifdef HAVE_SIGINTERRUPT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000497 {"siginterrupt", signal_siginterrupt, METH_VARARGS, siginterrupt_doc},
Facundo Batista7e251e82008-02-23 15:07:35 +0000498#endif
Guido van Rossum1171ee61997-08-22 20:42:00 +0000499#ifdef HAVE_PAUSE
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000500 {"pause", (PyCFunction)signal_pause,
501 METH_NOARGS,pause_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000502#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000503 {"default_int_handler", signal_default_int_handler,
504 METH_VARARGS, default_int_handler_doc},
505 {NULL, NULL} /* sentinel */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000506};
507
Barry Warsaw92971171997-01-03 00:14:25 +0000508
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000509PyDoc_STRVAR(module_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000510"This module provides mechanisms to use signal handlers in Python.\n\
511\n\
512Functions:\n\
513\n\
514alarm() -- cause SIGALRM after a specified time [Unix only]\n\
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000515setitimer() -- cause a signal (described below) after a specified\n\
516 float time and the timer may restart then [Unix only]\n\
517getitimer() -- get current value of timer [Unix only]\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000518signal() -- set the action for a given signal\n\
519getsignal() -- get the signal action for a given signal\n\
520pause() -- wait until a signal arrives [Unix only]\n\
521default_int_handler() -- default SIGINT handler\n\
522\n\
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000523signal constants:\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000524SIG_DFL -- used to refer to the system default handler\n\
525SIG_IGN -- used to ignore the signal\n\
526NSIG -- number of defined signals\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000527SIGINT, SIGTERM, etc. -- signal numbers\n\
528\n\
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000529itimer constants:\n\
530ITIMER_REAL -- decrements in real time, and delivers SIGALRM upon\n\
531 expiration\n\
532ITIMER_VIRTUAL -- decrements only when the process is executing,\n\
533 and delivers SIGVTALRM upon expiration\n\
534ITIMER_PROF -- decrements both when the process is executing and\n\
535 when the system is executing on behalf of the process.\n\
536 Coupled with ITIMER_VIRTUAL, this timer is usually\n\
537 used to profile the time spent by the application\n\
538 in user and kernel space. SIGPROF is delivered upon\n\
539 expiration.\n\
540\n\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000541*** IMPORTANT NOTICE ***\n\
542A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000543the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000544
Mark Hammondfe51c6d2002-08-02 02:27:13 +0000545PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000546initsignal(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000547{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000548 PyObject *m, *d, *x;
549 int i;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000550
551#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000552 main_thread = PyThread_get_thread_ident();
553 main_pid = getpid();
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000554#endif
555
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000556 /* Create the module and add the functions */
557 m = Py_InitModule3("signal", signal_methods, module_doc);
558 if (m == NULL)
559 return;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000560
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000561 /* Add some symbolic constants to the module */
562 d = PyModule_GetDict(m);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000563
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000564 x = DefaultHandler = PyLong_FromVoidPtr((void *)SIG_DFL);
565 if (!x || PyDict_SetItemString(d, "SIG_DFL", x) < 0)
566 goto finally;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000567
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000568 x = IgnoreHandler = PyLong_FromVoidPtr((void *)SIG_IGN);
569 if (!x || PyDict_SetItemString(d, "SIG_IGN", x) < 0)
570 goto finally;
Barry Warsaw92971171997-01-03 00:14:25 +0000571
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000572 x = PyInt_FromLong((long)NSIG);
573 if (!x || PyDict_SetItemString(d, "NSIG", x) < 0)
574 goto finally;
575 Py_DECREF(x);
Barry Warsaw92971171997-01-03 00:14:25 +0000576
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000577 x = IntHandler = PyDict_GetItemString(d, "default_int_handler");
578 if (!x)
579 goto finally;
580 Py_INCREF(IntHandler);
Barry Warsaw92971171997-01-03 00:14:25 +0000581
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000582 Handlers[0].tripped = 0;
583 for (i = 1; i < NSIG; i++) {
584 void (*t)(int);
585 t = PyOS_getsig(i);
586 Handlers[i].tripped = 0;
587 if (t == SIG_DFL)
588 Handlers[i].func = DefaultHandler;
589 else if (t == SIG_IGN)
590 Handlers[i].func = IgnoreHandler;
591 else
592 Handlers[i].func = Py_None; /* None of our business */
593 Py_INCREF(Handlers[i].func);
594 }
595 if (Handlers[SIGINT].func == DefaultHandler) {
596 /* Install default int handler */
597 Py_INCREF(IntHandler);
598 Py_DECREF(Handlers[SIGINT].func);
599 Handlers[SIGINT].func = IntHandler;
600 old_siginthandler = PyOS_setsig(SIGINT, signal_handler);
601 }
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000602
603#ifdef SIGHUP
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000604 x = PyInt_FromLong(SIGHUP);
605 PyDict_SetItemString(d, "SIGHUP", x);
606 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000607#endif
608#ifdef SIGINT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000609 x = PyInt_FromLong(SIGINT);
610 PyDict_SetItemString(d, "SIGINT", x);
611 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000612#endif
Tim Peters1ce3cf72001-10-01 17:58:40 +0000613#ifdef SIGBREAK
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000614 x = PyInt_FromLong(SIGBREAK);
615 PyDict_SetItemString(d, "SIGBREAK", x);
616 Py_XDECREF(x);
Tim Peters1ce3cf72001-10-01 17:58:40 +0000617#endif
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000618#ifdef SIGQUIT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000619 x = PyInt_FromLong(SIGQUIT);
620 PyDict_SetItemString(d, "SIGQUIT", x);
621 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000622#endif
623#ifdef SIGILL
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000624 x = PyInt_FromLong(SIGILL);
625 PyDict_SetItemString(d, "SIGILL", x);
626 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000627#endif
628#ifdef SIGTRAP
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000629 x = PyInt_FromLong(SIGTRAP);
630 PyDict_SetItemString(d, "SIGTRAP", x);
631 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000632#endif
633#ifdef SIGIOT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000634 x = PyInt_FromLong(SIGIOT);
635 PyDict_SetItemString(d, "SIGIOT", x);
636 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000637#endif
638#ifdef SIGABRT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000639 x = PyInt_FromLong(SIGABRT);
640 PyDict_SetItemString(d, "SIGABRT", x);
641 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000642#endif
643#ifdef SIGEMT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000644 x = PyInt_FromLong(SIGEMT);
645 PyDict_SetItemString(d, "SIGEMT", x);
646 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000647#endif
648#ifdef SIGFPE
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000649 x = PyInt_FromLong(SIGFPE);
650 PyDict_SetItemString(d, "SIGFPE", x);
651 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000652#endif
653#ifdef SIGKILL
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000654 x = PyInt_FromLong(SIGKILL);
655 PyDict_SetItemString(d, "SIGKILL", x);
656 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000657#endif
658#ifdef SIGBUS
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000659 x = PyInt_FromLong(SIGBUS);
660 PyDict_SetItemString(d, "SIGBUS", x);
661 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000662#endif
663#ifdef SIGSEGV
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000664 x = PyInt_FromLong(SIGSEGV);
665 PyDict_SetItemString(d, "SIGSEGV", x);
666 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000667#endif
668#ifdef SIGSYS
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000669 x = PyInt_FromLong(SIGSYS);
670 PyDict_SetItemString(d, "SIGSYS", x);
671 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000672#endif
673#ifdef SIGPIPE
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000674 x = PyInt_FromLong(SIGPIPE);
675 PyDict_SetItemString(d, "SIGPIPE", x);
676 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000677#endif
678#ifdef SIGALRM
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000679 x = PyInt_FromLong(SIGALRM);
680 PyDict_SetItemString(d, "SIGALRM", x);
681 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000682#endif
683#ifdef SIGTERM
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000684 x = PyInt_FromLong(SIGTERM);
685 PyDict_SetItemString(d, "SIGTERM", x);
686 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000687#endif
688#ifdef SIGUSR1
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000689 x = PyInt_FromLong(SIGUSR1);
690 PyDict_SetItemString(d, "SIGUSR1", x);
691 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000692#endif
693#ifdef SIGUSR2
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000694 x = PyInt_FromLong(SIGUSR2);
695 PyDict_SetItemString(d, "SIGUSR2", x);
696 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000697#endif
698#ifdef SIGCLD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000699 x = PyInt_FromLong(SIGCLD);
700 PyDict_SetItemString(d, "SIGCLD", x);
701 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000702#endif
703#ifdef SIGCHLD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000704 x = PyInt_FromLong(SIGCHLD);
705 PyDict_SetItemString(d, "SIGCHLD", x);
706 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000707#endif
708#ifdef SIGPWR
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000709 x = PyInt_FromLong(SIGPWR);
710 PyDict_SetItemString(d, "SIGPWR", x);
711 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000712#endif
713#ifdef SIGIO
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000714 x = PyInt_FromLong(SIGIO);
715 PyDict_SetItemString(d, "SIGIO", x);
716 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000717#endif
718#ifdef SIGURG
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000719 x = PyInt_FromLong(SIGURG);
720 PyDict_SetItemString(d, "SIGURG", x);
721 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000722#endif
723#ifdef SIGWINCH
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000724 x = PyInt_FromLong(SIGWINCH);
725 PyDict_SetItemString(d, "SIGWINCH", x);
726 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000727#endif
728#ifdef SIGPOLL
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000729 x = PyInt_FromLong(SIGPOLL);
730 PyDict_SetItemString(d, "SIGPOLL", x);
731 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000732#endif
733#ifdef SIGSTOP
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000734 x = PyInt_FromLong(SIGSTOP);
735 PyDict_SetItemString(d, "SIGSTOP", x);
736 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000737#endif
738#ifdef SIGTSTP
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000739 x = PyInt_FromLong(SIGTSTP);
740 PyDict_SetItemString(d, "SIGTSTP", x);
741 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000742#endif
743#ifdef SIGCONT
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000744 x = PyInt_FromLong(SIGCONT);
745 PyDict_SetItemString(d, "SIGCONT", x);
746 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000747#endif
748#ifdef SIGTTIN
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000749 x = PyInt_FromLong(SIGTTIN);
750 PyDict_SetItemString(d, "SIGTTIN", x);
751 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000752#endif
753#ifdef SIGTTOU
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000754 x = PyInt_FromLong(SIGTTOU);
755 PyDict_SetItemString(d, "SIGTTOU", x);
756 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000757#endif
758#ifdef SIGVTALRM
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000759 x = PyInt_FromLong(SIGVTALRM);
760 PyDict_SetItemString(d, "SIGVTALRM", x);
761 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000762#endif
763#ifdef SIGPROF
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000764 x = PyInt_FromLong(SIGPROF);
765 PyDict_SetItemString(d, "SIGPROF", x);
766 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000767#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000768#ifdef SIGXCPU
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000769 x = PyInt_FromLong(SIGXCPU);
770 PyDict_SetItemString(d, "SIGXCPU", x);
771 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000772#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000773#ifdef SIGXFSZ
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000774 x = PyInt_FromLong(SIGXFSZ);
775 PyDict_SetItemString(d, "SIGXFSZ", x);
776 Py_XDECREF(x);
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000777#endif
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000778#ifdef SIGRTMIN
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000779 x = PyInt_FromLong(SIGRTMIN);
780 PyDict_SetItemString(d, "SIGRTMIN", x);
781 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000782#endif
783#ifdef SIGRTMAX
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000784 x = PyInt_FromLong(SIGRTMAX);
785 PyDict_SetItemString(d, "SIGRTMAX", x);
786 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000787#endif
Martin v. Löwis175af252002-01-12 11:43:25 +0000788#ifdef SIGINFO
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000789 x = PyInt_FromLong(SIGINFO);
790 PyDict_SetItemString(d, "SIGINFO", x);
791 Py_XDECREF(x);
Martin v. Löwis175af252002-01-12 11:43:25 +0000792#endif
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000793
794#ifdef ITIMER_REAL
795 x = PyLong_FromLong(ITIMER_REAL);
796 PyDict_SetItemString(d, "ITIMER_REAL", x);
797 Py_DECREF(x);
798#endif
799#ifdef ITIMER_VIRTUAL
800 x = PyLong_FromLong(ITIMER_VIRTUAL);
801 PyDict_SetItemString(d, "ITIMER_VIRTUAL", x);
802 Py_DECREF(x);
803#endif
804#ifdef ITIMER_PROF
805 x = PyLong_FromLong(ITIMER_PROF);
806 PyDict_SetItemString(d, "ITIMER_PROF", x);
807 Py_DECREF(x);
808#endif
809
810#if defined (HAVE_SETITIMER) || defined (HAVE_GETITIMER)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000811 ItimerError = PyErr_NewException("signal.ItimerError",
812 PyExc_IOError, NULL);
Neal Norwitz18aa3882008-08-24 05:04:52 +0000813 if (ItimerError != NULL)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000814 PyDict_SetItemString(d, "ItimerError", ItimerError);
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000815#endif
816
Brian Curtine5aa8862010-04-02 23:26:06 +0000817#ifdef CTRL_C_EVENT
818 x = PyInt_FromLong(CTRL_C_EVENT);
819 PyDict_SetItemString(d, "CTRL_C_EVENT", x);
820 Py_DECREF(x);
821#endif
822
823#ifdef CTRL_BREAK_EVENT
824 x = PyInt_FromLong(CTRL_BREAK_EVENT);
825 PyDict_SetItemString(d, "CTRL_BREAK_EVENT", x);
826 Py_DECREF(x);
827#endif
828
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000829 if (!PyErr_Occurred())
Guido van Rossum08c16611997-08-02 03:01:42 +0000830 return;
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000831
832 /* Check for errors */
833 finally:
834 return;
Guido van Rossum08c16611997-08-02 03:01:42 +0000835}
836
837static void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000838finisignal(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000839{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000840 int i;
841 PyObject *func;
Guido van Rossum08c16611997-08-02 03:01:42 +0000842
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000843 PyOS_setsig(SIGINT, old_siginthandler);
844 old_siginthandler = SIG_DFL;
Guido van Rossum08c16611997-08-02 03:01:42 +0000845
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000846 for (i = 1; i < NSIG; i++) {
847 func = Handlers[i].func;
848 Handlers[i].tripped = 0;
849 Handlers[i].func = NULL;
850 if (i != SIGINT && func != NULL && func != Py_None &&
851 func != DefaultHandler && func != IgnoreHandler)
852 PyOS_setsig(i, SIG_DFL);
853 Py_XDECREF(func);
854 }
Guido van Rossum08c16611997-08-02 03:01:42 +0000855
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000856 Py_XDECREF(IntHandler);
857 IntHandler = NULL;
858 Py_XDECREF(DefaultHandler);
859 DefaultHandler = NULL;
860 Py_XDECREF(IgnoreHandler);
861 IgnoreHandler = NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000862}
863
Barry Warsaw92971171997-01-03 00:14:25 +0000864
Barry Warsaw92971171997-01-03 00:14:25 +0000865/* Declared in pyerrors.h */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000866int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000867PyErr_CheckSignals(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000868{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000869 int i;
870 PyObject *f;
Barry Warsaw92971171997-01-03 00:14:25 +0000871
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000872 if (!is_tripped)
873 return 0;
Guido van Rossum137c49c2007-12-10 23:00:12 +0000874
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000875#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000876 if (PyThread_get_thread_ident() != main_thread)
877 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000878#endif
Guido van Rossum137c49c2007-12-10 23:00:12 +0000879
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000880 /*
881 * The is_tripped variable is meant to speed up the calls to
882 * PyErr_CheckSignals (both directly or via pending calls) when no
883 * signal has arrived. This variable is set to 1 when a signal arrives
884 * and it is set to 0 here, when we know some signals arrived. This way
885 * we can run the registered handlers with no signals blocked.
886 *
887 * NOTE: with this approach we can have a situation where is_tripped is
888 * 1 but we have no more signals to handle (Handlers[i].tripped
889 * is 0 for every signal i). This won't do us any harm (except
890 * we're gonna spent some cycles for nothing). This happens when
891 * we receive a signal i after we zero is_tripped and before we
892 * check Handlers[i].tripped.
893 */
894 is_tripped = 0;
Guido van Rossum137c49c2007-12-10 23:00:12 +0000895
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000896 if (!(f = (PyObject *)PyEval_GetFrame()))
897 f = Py_None;
Guido van Rossum137c49c2007-12-10 23:00:12 +0000898
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000899 for (i = 1; i < NSIG; i++) {
900 if (Handlers[i].tripped) {
901 PyObject *result = NULL;
902 PyObject *arglist = Py_BuildValue("(iO)", i, f);
903 Handlers[i].tripped = 0;
Barry Warsaw92971171997-01-03 00:14:25 +0000904
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000905 if (arglist) {
906 result = PyEval_CallObject(Handlers[i].func,
907 arglist);
908 Py_DECREF(arglist);
909 }
910 if (!result)
911 return -1;
Barry Warsaw92971171997-01-03 00:14:25 +0000912
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000913 Py_DECREF(result);
914 }
915 }
Guido van Rossum137c49c2007-12-10 23:00:12 +0000916
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000917 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000918}
919
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000920
Barry Warsaw92971171997-01-03 00:14:25 +0000921/* Replacements for intrcheck.c functionality
922 * Declared in pyerrors.h
923 */
924void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000925PyErr_SetInterrupt(void)
Barry Warsaw92971171997-01-03 00:14:25 +0000926{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000927 is_tripped = 1;
928 Handlers[SIGINT].tripped = 1;
929 Py_AddPendingCall((int (*)(void *))PyErr_CheckSignals, NULL);
Barry Warsaw92971171997-01-03 00:14:25 +0000930}
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000931
932void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000933PyOS_InitInterrupts(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000934{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000935 initsignal();
936 _PyImport_FixupExtension("signal", "signal");
Guido van Rossum08c16611997-08-02 03:01:42 +0000937}
938
939void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000940PyOS_FiniInterrupts(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000941{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000942 finisignal();
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000943}
944
945int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000946PyOS_InterruptOccurred(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000947{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000948 if (Handlers[SIGINT].tripped) {
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000949#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000950 if (PyThread_get_thread_ident() != main_thread)
951 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000952#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000953 Handlers[SIGINT].tripped = 0;
954 return 1;
955 }
956 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000957}
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000958
959void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000960PyOS_AfterFork(void)
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000961{
962#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000963 PyEval_ReInitThreads();
964 main_thread = PyThread_get_thread_ident();
965 main_pid = getpid();
966 _PyImport_ReInitLock();
967 PyThread_ReInitTLS();
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000968#endif
969}