blob: b1ee890bc55d750b2923472982e20b104864a622 [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)
37# define NSIG _NSIG /* For BSD/SysV */
38# elif defined(_SIGMAX)
39# define NSIG (_SIGMAX + 1) /* For QNX */
40# elif defined(SIGMAX)
41# define NSIG (SIGMAX + 1) /* For djgpp */
42# else
43# define NSIG 64 /* Use a reasonable default value */
44# 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 {
85 int tripped;
86 PyObject *func;
87} 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)
129 return NULL;
130
131 if(!(v = PyFloat_FromDouble(double_from_timeval(&iv->it_value)))) {
132 Py_DECREF(r);
133 return NULL;
134 }
135
136 PyTuple_SET_ITEM(r, 0, v);
137
138 if(!(v = PyFloat_FromDouble(double_from_timeval(&iv->it_interval)))) {
139 Py_DECREF(r);
140 return NULL;
141 }
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{
Guido van Rossume4485b01994-09-07 14:32:49 +0000152 PyErr_SetNone(PyExc_KeyboardInterrupt);
Barry Warsaw92971171997-01-03 00:14:25 +0000153 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{
166 return PyErr_CheckSignals();
167}
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
174 if (PyThread_get_thread_ident() != main_thread) {
175 pth_raise(*(pth_t *) main_thread, sig_num);
176 return;
177 }
178#endif
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000179 /* See NOTES section above */
180 if (getpid() == main_pid) {
181#endif
Barry Warsaw92971171997-01-03 00:14:25 +0000182 Handlers[sig_num].tripped = 1;
Guido van Rossum137c49c2007-12-10 23:00:12 +0000183 /* Set is_tripped after setting .tripped, as it gets
184 cleared in PyErr_CheckSignals() before .tripped. */
185 is_tripped = 1;
Thomas Wouters0796b002000-07-22 23:49:30 +0000186 Py_AddPendingCall(checksignals_witharg, NULL);
Guido van Rossum02de8972007-12-19 19:41:06 +0000187 if (wakeup_fd != -1)
188 write(wakeup_fd, "\0", 1);
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000189#ifdef WITH_THREAD
190 }
191#endif
Guido van Rossum602099a1994-09-14 13:32:22 +0000192#ifdef SIGCHLD
193 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... */
Tim Peters4f1b2082000-07-23 21:18:09 +0000198 return;
Guido van Rossum602099a1994-09-14 13:32:22 +0000199 }
200#endif
Jean-Paul Calderonee54ddf12010-05-08 20:06:02 +0000201#ifndef HAVE_SIGACTION
202 /* 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. */
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000205 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{
214 int t;
Neal Norwitzba3a16c2002-03-31 15:27:00 +0000215 if (!PyArg_ParseTuple(args, "i:alarm", &t))
Barry Warsaw92971171997-01-03 00:14:25 +0000216 return NULL;
Guido van Rossume4485b01994-09-07 14:32:49 +0000217 /* alarm() returns the number of seconds remaining */
Fred Drakedff3a372001-07-19 21:29:49 +0000218 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{
Guido van Rossuma597dde1995-01-10 20:56:29 +0000231 Py_BEGIN_ALLOW_THREADS
Barry Warsaw92971171997-01-03 00:14:25 +0000232 (void)pause();
Guido van Rossuma597dde1995-01-10 20:56:29 +0000233 Py_END_ALLOW_THREADS
Barry Warsaw92971171997-01-03 00:14:25 +0000234 /* make sure that any exceptions that got raised are propagated
235 * back into Python
236 */
237 if (PyErr_CheckSignals())
238 return NULL;
239
Guido van Rossume4485b01994-09-07 14:32:49 +0000240 Py_INCREF(Py_None);
241 return Py_None;
242}
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{
254 PyObject *obj;
255 int sig_num;
256 PyObject *old_handler;
Tim Peters4f1b2082000-07-23 21:18:09 +0000257 void (*func)(int);
Neal Norwitzba3a16c2002-03-31 15:27:00 +0000258 if (!PyArg_ParseTuple(args, "iO:signal", &sig_num, &obj))
Barry Warsaw92971171997-01-03 00:14:25 +0000259 return NULL;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000260#ifdef WITH_THREAD
Guido van Rossum65d5b571998-12-21 19:32:43 +0000261 if (PyThread_get_thread_ident() != main_thread) {
Guido van Rossumf4b012a1995-03-16 15:43:29 +0000262 PyErr_SetString(PyExc_ValueError,
263 "signal only works in main thread");
Barry Warsaw92971171997-01-03 00:14:25 +0000264 return NULL;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000265 }
266#endif
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000267 if (sig_num < 1 || sig_num >= NSIG) {
Guido van Rossumf4b012a1995-03-16 15:43:29 +0000268 PyErr_SetString(PyExc_ValueError,
269 "signal number out of range");
Barry Warsaw92971171997-01-03 00:14:25 +0000270 return NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000271 }
Barry Warsaw92971171997-01-03 00:14:25 +0000272 if (obj == IgnoreHandler)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000273 func = SIG_IGN;
Barry Warsaw92971171997-01-03 00:14:25 +0000274 else if (obj == DefaultHandler)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000275 func = SIG_DFL;
Guido van Rossumf4b012a1995-03-16 15:43:29 +0000276 else if (!PyCallable_Check(obj)) {
Guido van Rossume4485b01994-09-07 14:32:49 +0000277 PyErr_SetString(PyExc_TypeError,
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000278"signal handler must be signal.SIG_IGN, signal.SIG_DFL, or a callable object");
Barry Warsaw92971171997-01-03 00:14:25 +0000279 return NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000280 }
281 else
Barry Warsaw92971171997-01-03 00:14:25 +0000282 func = signal_handler;
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000283 if (PyOS_setsig(sig_num, func) == SIG_ERR) {
Guido van Rossume4485b01994-09-07 14:32:49 +0000284 PyErr_SetFromErrno(PyExc_RuntimeError);
Barry Warsaw92971171997-01-03 00:14:25 +0000285 return NULL;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000286 }
Barry Warsaw92971171997-01-03 00:14:25 +0000287 old_handler = Handlers[sig_num].func;
288 Handlers[sig_num].tripped = 0;
Guido van Rossume4485b01994-09-07 14:32:49 +0000289 Py_INCREF(obj);
Barry Warsaw92971171997-01-03 00:14:25 +0000290 Handlers[sig_num].func = obj;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000291 return old_handler;
292}
293
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000294PyDoc_STRVAR(signal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000295"signal(sig, action) -> action\n\
296\n\
297Set the action for the given signal. The action can be SIG_DFL,\n\
298SIG_IGN, or a callable Python object. The previous action is\n\
299returned. See getsignal() for possible return values.\n\
300\n\
301*** IMPORTANT NOTICE ***\n\
302A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000303the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000304
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000305
Guido van Rossume4485b01994-09-07 14:32:49 +0000306static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000307signal_getsignal(PyObject *self, PyObject *args)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000308{
309 int sig_num;
Guido van Rossume4485b01994-09-07 14:32:49 +0000310 PyObject *old_handler;
Neal Norwitzba3a16c2002-03-31 15:27:00 +0000311 if (!PyArg_ParseTuple(args, "i:getsignal", &sig_num))
Barry Warsaw92971171997-01-03 00:14:25 +0000312 return NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000313 if (sig_num < 1 || sig_num >= NSIG) {
Guido van Rossumf4b012a1995-03-16 15:43:29 +0000314 PyErr_SetString(PyExc_ValueError,
315 "signal number out of range");
Barry Warsaw92971171997-01-03 00:14:25 +0000316 return NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000317 }
Barry Warsaw92971171997-01-03 00:14:25 +0000318 old_handler = Handlers[sig_num].func;
Guido van Rossume4485b01994-09-07 14:32:49 +0000319 Py_INCREF(old_handler);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000320 return old_handler;
321}
322
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000323PyDoc_STRVAR(getsignal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000324"getsignal(sig) -> action\n\
325\n\
326Return the current action for the given signal. The return value can be:\n\
327SIG_IGN -- if the signal is being ignored\n\
328SIG_DFL -- if the default action for the signal is in effect\n\
329None -- if an unknown handler is in effect\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000330anything else -- the callable Python object used as a handler");
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000331
Facundo Batista7e251e82008-02-23 15:07:35 +0000332#ifdef HAVE_SIGINTERRUPT
333PyDoc_STRVAR(siginterrupt_doc,
334"siginterrupt(sig, flag) -> None\n\
335change system call restart behaviour: if flag is False, system calls\n\
336will be restarted when interrupted by signal sig, else system calls\n\
337will be interrupted.");
338
339static PyObject *
340signal_siginterrupt(PyObject *self, PyObject *args)
341{
342 int sig_num;
343 int flag;
344
345 if (!PyArg_ParseTuple(args, "ii:siginterrupt", &sig_num, &flag))
346 return NULL;
347 if (sig_num < 1 || sig_num >= NSIG) {
348 PyErr_SetString(PyExc_ValueError,
349 "signal number out of range");
350 return NULL;
351 }
352 if (siginterrupt(sig_num, flag)<0) {
353 PyErr_SetFromErrno(PyExc_RuntimeError);
354 return NULL;
355 }
356
357 Py_INCREF(Py_None);
358 return Py_None;
359}
360
361#endif
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000362
Guido van Rossum02de8972007-12-19 19:41:06 +0000363static PyObject *
364signal_set_wakeup_fd(PyObject *self, PyObject *args)
365{
366 struct stat buf;
367 int fd, old_fd;
368 if (!PyArg_ParseTuple(args, "i:set_wakeup_fd", &fd))
369 return NULL;
370#ifdef WITH_THREAD
371 if (PyThread_get_thread_ident() != main_thread) {
372 PyErr_SetString(PyExc_ValueError,
373 "set_wakeup_fd only works in main thread");
374 return NULL;
375 }
376#endif
377 if (fd != -1 && fstat(fd, &buf) != 0) {
378 PyErr_SetString(PyExc_ValueError, "invalid fd");
379 return NULL;
380 }
381 old_fd = wakeup_fd;
382 wakeup_fd = fd;
383 return PyLong_FromLong(old_fd);
384}
385
386PyDoc_STRVAR(set_wakeup_fd_doc,
387"set_wakeup_fd(fd) -> fd\n\
388\n\
389Sets the fd to be written to (with '\\0') when a signal\n\
390comes in. A library can use this to wakeup select or poll.\n\
391The previous fd is returned.\n\
392\n\
393The fd must be non-blocking.");
394
395/* C API for the same, without all the error checking */
396int
397PySignal_SetWakeupFd(int fd)
398{
399 int old_fd = wakeup_fd;
400 if (fd < 0)
401 fd = -1;
402 wakeup_fd = fd;
403 return old_fd;
404}
405
406
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000407#ifdef HAVE_SETITIMER
408static PyObject *
409signal_setitimer(PyObject *self, PyObject *args)
410{
411 double first;
412 double interval = 0;
413 int which;
414 struct itimerval new, old;
415
416 if(!PyArg_ParseTuple(args, "id|d:setitimer", &which, &first, &interval))
417 return NULL;
418
419 timeval_from_double(first, &new.it_value);
420 timeval_from_double(interval, &new.it_interval);
421 /* Let OS check "which" value */
422 if (setitimer(which, &new, &old) != 0) {
423 PyErr_SetFromErrno(ItimerError);
424 return NULL;
425 }
426
427 return itimer_retval(&old);
428}
429
430PyDoc_STRVAR(setitimer_doc,
431"setitimer(which, seconds[, interval])\n\
432\n\
433Sets given itimer (one of ITIMER_REAL, ITIMER_VIRTUAL\n\
434or ITIMER_PROF) to fire after value seconds and after\n\
435that every interval seconds.\n\
436The itimer can be cleared by setting seconds to zero.\n\
437\n\
438Returns old values as a tuple: (delay, interval).");
439#endif
440
441
442#ifdef HAVE_GETITIMER
443static PyObject *
444signal_getitimer(PyObject *self, PyObject *args)
445{
446 int which;
447 struct itimerval old;
448
449 if (!PyArg_ParseTuple(args, "i:getitimer", &which))
450 return NULL;
451
452 if (getitimer(which, &old) != 0) {
453 PyErr_SetFromErrno(ItimerError);
454 return NULL;
455 }
456
457 return itimer_retval(&old);
458}
459
460PyDoc_STRVAR(getitimer_doc,
461"getitimer(which)\n\
462\n\
463Returns current value of given itimer.");
464#endif
465
466
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000467/* List of functions defined in the module */
Barry Warsaw92971171997-01-03 00:14:25 +0000468static PyMethodDef signal_methods[] = {
Guido van Rossum1171ee61997-08-22 20:42:00 +0000469#ifdef HAVE_ALARM
Neal Norwitzba3a16c2002-03-31 15:27:00 +0000470 {"alarm", signal_alarm, METH_VARARGS, alarm_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000471#endif
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000472#ifdef HAVE_SETITIMER
473 {"setitimer", signal_setitimer, METH_VARARGS, setitimer_doc},
474#endif
475#ifdef HAVE_GETITIMER
476 {"getitimer", signal_getitimer, METH_VARARGS, getitimer_doc},
477#endif
Neal Norwitzba3a16c2002-03-31 15:27:00 +0000478 {"signal", signal_signal, METH_VARARGS, signal_doc},
479 {"getsignal", signal_getsignal, METH_VARARGS, getsignal_doc},
Guido van Rossum02de8972007-12-19 19:41:06 +0000480 {"set_wakeup_fd", signal_set_wakeup_fd, METH_VARARGS, set_wakeup_fd_doc},
Facundo Batista7e251e82008-02-23 15:07:35 +0000481#ifdef HAVE_SIGINTERRUPT
482 {"siginterrupt", signal_siginterrupt, METH_VARARGS, siginterrupt_doc},
483#endif
Guido van Rossum1171ee61997-08-22 20:42:00 +0000484#ifdef HAVE_PAUSE
Neil Schemenauer78662cf2002-03-28 21:04:14 +0000485 {"pause", (PyCFunction)signal_pause,
486 METH_NOARGS,pause_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000487#endif
Guido van Rossum02de8972007-12-19 19:41:06 +0000488 {"default_int_handler", signal_default_int_handler,
Neal Norwitzba3a16c2002-03-31 15:27:00 +0000489 METH_VARARGS, default_int_handler_doc},
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000490 {NULL, NULL} /* sentinel */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000491};
492
Barry Warsaw92971171997-01-03 00:14:25 +0000493
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000494PyDoc_STRVAR(module_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000495"This module provides mechanisms to use signal handlers in Python.\n\
496\n\
497Functions:\n\
498\n\
499alarm() -- cause SIGALRM after a specified time [Unix only]\n\
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000500setitimer() -- cause a signal (described below) after a specified\n\
501 float time and the timer may restart then [Unix only]\n\
502getitimer() -- get current value of timer [Unix only]\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000503signal() -- set the action for a given signal\n\
504getsignal() -- get the signal action for a given signal\n\
505pause() -- wait until a signal arrives [Unix only]\n\
506default_int_handler() -- default SIGINT handler\n\
507\n\
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000508signal constants:\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000509SIG_DFL -- used to refer to the system default handler\n\
510SIG_IGN -- used to ignore the signal\n\
511NSIG -- number of defined signals\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000512SIGINT, SIGTERM, etc. -- signal numbers\n\
513\n\
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000514itimer constants:\n\
515ITIMER_REAL -- decrements in real time, and delivers SIGALRM upon\n\
516 expiration\n\
517ITIMER_VIRTUAL -- decrements only when the process is executing,\n\
518 and delivers SIGVTALRM upon expiration\n\
519ITIMER_PROF -- decrements both when the process is executing and\n\
520 when the system is executing on behalf of the process.\n\
521 Coupled with ITIMER_VIRTUAL, this timer is usually\n\
522 used to profile the time spent by the application\n\
523 in user and kernel space. SIGPROF is delivered upon\n\
524 expiration.\n\
525\n\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000526*** IMPORTANT NOTICE ***\n\
527A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000528the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000529
Mark Hammondfe51c6d2002-08-02 02:27:13 +0000530PyMODINIT_FUNC
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000531initsignal(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000532{
Guido van Rossume4485b01994-09-07 14:32:49 +0000533 PyObject *m, *d, *x;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000534 int i;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000535
536#ifdef WITH_THREAD
Guido van Rossum65d5b571998-12-21 19:32:43 +0000537 main_thread = PyThread_get_thread_ident();
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000538 main_pid = getpid();
539#endif
540
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000541 /* Create the module and add the functions */
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000542 m = Py_InitModule3("signal", signal_methods, module_doc);
Neal Norwitz1ac754f2006-01-19 06:09:39 +0000543 if (m == NULL)
544 return;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000545
546 /* Add some symbolic constants to the module */
Guido van Rossume4485b01994-09-07 14:32:49 +0000547 d = PyModule_GetDict(m);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000548
Guido van Rossum276fa432000-06-30 23:04:18 +0000549 x = DefaultHandler = PyLong_FromVoidPtr((void *)SIG_DFL);
Barry Warsaw92971171997-01-03 00:14:25 +0000550 if (!x || PyDict_SetItemString(d, "SIG_DFL", x) < 0)
551 goto finally;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000552
Guido van Rossum276fa432000-06-30 23:04:18 +0000553 x = IgnoreHandler = PyLong_FromVoidPtr((void *)SIG_IGN);
Barry Warsaw92971171997-01-03 00:14:25 +0000554 if (!x || PyDict_SetItemString(d, "SIG_IGN", x) < 0)
555 goto finally;
556
557 x = PyInt_FromLong((long)NSIG);
558 if (!x || PyDict_SetItemString(d, "NSIG", x) < 0)
559 goto finally;
Guido van Rossum08c16611997-08-02 03:01:42 +0000560 Py_DECREF(x);
Barry Warsaw92971171997-01-03 00:14:25 +0000561
562 x = IntHandler = PyDict_GetItemString(d, "default_int_handler");
563 if (!x)
564 goto finally;
Guido van Rossum08c16611997-08-02 03:01:42 +0000565 Py_INCREF(IntHandler);
Barry Warsaw92971171997-01-03 00:14:25 +0000566
567 Handlers[0].tripped = 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000568 for (i = 1; i < NSIG; i++) {
Tim Peters4f1b2082000-07-23 21:18:09 +0000569 void (*t)(int);
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000570 t = PyOS_getsig(i);
Barry Warsaw92971171997-01-03 00:14:25 +0000571 Handlers[i].tripped = 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000572 if (t == SIG_DFL)
Barry Warsaw92971171997-01-03 00:14:25 +0000573 Handlers[i].func = DefaultHandler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000574 else if (t == SIG_IGN)
Barry Warsaw92971171997-01-03 00:14:25 +0000575 Handlers[i].func = IgnoreHandler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000576 else
Barry Warsaw92971171997-01-03 00:14:25 +0000577 Handlers[i].func = Py_None; /* None of our business */
578 Py_INCREF(Handlers[i].func);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000579 }
Barry Warsaw92971171997-01-03 00:14:25 +0000580 if (Handlers[SIGINT].func == DefaultHandler) {
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000581 /* Install default int handler */
Guido van Rossum08c16611997-08-02 03:01:42 +0000582 Py_INCREF(IntHandler);
Barry Warsaw92971171997-01-03 00:14:25 +0000583 Py_DECREF(Handlers[SIGINT].func);
584 Handlers[SIGINT].func = IntHandler;
Neal Norwitz41785152002-06-13 21:42:51 +0000585 old_siginthandler = PyOS_setsig(SIGINT, signal_handler);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000586 }
587
588#ifdef SIGHUP
Guido van Rossume4485b01994-09-07 14:32:49 +0000589 x = PyInt_FromLong(SIGHUP);
590 PyDict_SetItemString(d, "SIGHUP", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000591 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000592#endif
593#ifdef SIGINT
Guido van Rossume4485b01994-09-07 14:32:49 +0000594 x = PyInt_FromLong(SIGINT);
595 PyDict_SetItemString(d, "SIGINT", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000596 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000597#endif
Tim Peters1ce3cf72001-10-01 17:58:40 +0000598#ifdef SIGBREAK
599 x = PyInt_FromLong(SIGBREAK);
600 PyDict_SetItemString(d, "SIGBREAK", x);
601 Py_XDECREF(x);
602#endif
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000603#ifdef SIGQUIT
Guido van Rossume4485b01994-09-07 14:32:49 +0000604 x = PyInt_FromLong(SIGQUIT);
605 PyDict_SetItemString(d, "SIGQUIT", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000606 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000607#endif
608#ifdef SIGILL
Guido van Rossume4485b01994-09-07 14:32:49 +0000609 x = PyInt_FromLong(SIGILL);
610 PyDict_SetItemString(d, "SIGILL", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000611 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000612#endif
613#ifdef SIGTRAP
Guido van Rossume4485b01994-09-07 14:32:49 +0000614 x = PyInt_FromLong(SIGTRAP);
615 PyDict_SetItemString(d, "SIGTRAP", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000616 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000617#endif
618#ifdef SIGIOT
Guido van Rossume4485b01994-09-07 14:32:49 +0000619 x = PyInt_FromLong(SIGIOT);
620 PyDict_SetItemString(d, "SIGIOT", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000621 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000622#endif
623#ifdef SIGABRT
Guido van Rossume4485b01994-09-07 14:32:49 +0000624 x = PyInt_FromLong(SIGABRT);
625 PyDict_SetItemString(d, "SIGABRT", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000626 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000627#endif
628#ifdef SIGEMT
Guido van Rossume4485b01994-09-07 14:32:49 +0000629 x = PyInt_FromLong(SIGEMT);
630 PyDict_SetItemString(d, "SIGEMT", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000631 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000632#endif
633#ifdef SIGFPE
Guido van Rossume4485b01994-09-07 14:32:49 +0000634 x = PyInt_FromLong(SIGFPE);
635 PyDict_SetItemString(d, "SIGFPE", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000636 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000637#endif
638#ifdef SIGKILL
Guido van Rossume4485b01994-09-07 14:32:49 +0000639 x = PyInt_FromLong(SIGKILL);
640 PyDict_SetItemString(d, "SIGKILL", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000641 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000642#endif
643#ifdef SIGBUS
Guido van Rossume4485b01994-09-07 14:32:49 +0000644 x = PyInt_FromLong(SIGBUS);
645 PyDict_SetItemString(d, "SIGBUS", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000646 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000647#endif
648#ifdef SIGSEGV
Guido van Rossume4485b01994-09-07 14:32:49 +0000649 x = PyInt_FromLong(SIGSEGV);
650 PyDict_SetItemString(d, "SIGSEGV", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000651 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000652#endif
653#ifdef SIGSYS
Guido van Rossume4485b01994-09-07 14:32:49 +0000654 x = PyInt_FromLong(SIGSYS);
655 PyDict_SetItemString(d, "SIGSYS", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000656 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000657#endif
658#ifdef SIGPIPE
Guido van Rossume4485b01994-09-07 14:32:49 +0000659 x = PyInt_FromLong(SIGPIPE);
660 PyDict_SetItemString(d, "SIGPIPE", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000661 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000662#endif
663#ifdef SIGALRM
Guido van Rossume4485b01994-09-07 14:32:49 +0000664 x = PyInt_FromLong(SIGALRM);
665 PyDict_SetItemString(d, "SIGALRM", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000666 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000667#endif
668#ifdef SIGTERM
Guido van Rossume4485b01994-09-07 14:32:49 +0000669 x = PyInt_FromLong(SIGTERM);
670 PyDict_SetItemString(d, "SIGTERM", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000671 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000672#endif
673#ifdef SIGUSR1
Guido van Rossume4485b01994-09-07 14:32:49 +0000674 x = PyInt_FromLong(SIGUSR1);
675 PyDict_SetItemString(d, "SIGUSR1", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000676 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000677#endif
678#ifdef SIGUSR2
Guido van Rossume4485b01994-09-07 14:32:49 +0000679 x = PyInt_FromLong(SIGUSR2);
680 PyDict_SetItemString(d, "SIGUSR2", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000681 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000682#endif
683#ifdef SIGCLD
Guido van Rossume4485b01994-09-07 14:32:49 +0000684 x = PyInt_FromLong(SIGCLD);
685 PyDict_SetItemString(d, "SIGCLD", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000686 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000687#endif
688#ifdef SIGCHLD
Guido van Rossume4485b01994-09-07 14:32:49 +0000689 x = PyInt_FromLong(SIGCHLD);
690 PyDict_SetItemString(d, "SIGCHLD", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000691 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000692#endif
693#ifdef SIGPWR
Guido van Rossume4485b01994-09-07 14:32:49 +0000694 x = PyInt_FromLong(SIGPWR);
695 PyDict_SetItemString(d, "SIGPWR", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000696 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000697#endif
698#ifdef SIGIO
Guido van Rossume4485b01994-09-07 14:32:49 +0000699 x = PyInt_FromLong(SIGIO);
700 PyDict_SetItemString(d, "SIGIO", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000701 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000702#endif
703#ifdef SIGURG
Guido van Rossume4485b01994-09-07 14:32:49 +0000704 x = PyInt_FromLong(SIGURG);
705 PyDict_SetItemString(d, "SIGURG", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000706 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000707#endif
708#ifdef SIGWINCH
Guido van Rossume4485b01994-09-07 14:32:49 +0000709 x = PyInt_FromLong(SIGWINCH);
710 PyDict_SetItemString(d, "SIGWINCH", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000711 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000712#endif
713#ifdef SIGPOLL
Guido van Rossume4485b01994-09-07 14:32:49 +0000714 x = PyInt_FromLong(SIGPOLL);
715 PyDict_SetItemString(d, "SIGPOLL", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000716 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000717#endif
718#ifdef SIGSTOP
Guido van Rossume4485b01994-09-07 14:32:49 +0000719 x = PyInt_FromLong(SIGSTOP);
720 PyDict_SetItemString(d, "SIGSTOP", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000721 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000722#endif
723#ifdef SIGTSTP
Guido van Rossume4485b01994-09-07 14:32:49 +0000724 x = PyInt_FromLong(SIGTSTP);
725 PyDict_SetItemString(d, "SIGTSTP", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000726 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000727#endif
728#ifdef SIGCONT
Guido van Rossume4485b01994-09-07 14:32:49 +0000729 x = PyInt_FromLong(SIGCONT);
730 PyDict_SetItemString(d, "SIGCONT", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000731 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000732#endif
733#ifdef SIGTTIN
Guido van Rossume4485b01994-09-07 14:32:49 +0000734 x = PyInt_FromLong(SIGTTIN);
735 PyDict_SetItemString(d, "SIGTTIN", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000736 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000737#endif
738#ifdef SIGTTOU
Guido van Rossume4485b01994-09-07 14:32:49 +0000739 x = PyInt_FromLong(SIGTTOU);
740 PyDict_SetItemString(d, "SIGTTOU", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000741 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000742#endif
743#ifdef SIGVTALRM
Guido van Rossume4485b01994-09-07 14:32:49 +0000744 x = PyInt_FromLong(SIGVTALRM);
745 PyDict_SetItemString(d, "SIGVTALRM", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000746 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000747#endif
748#ifdef SIGPROF
Guido van Rossume4485b01994-09-07 14:32:49 +0000749 x = PyInt_FromLong(SIGPROF);
750 PyDict_SetItemString(d, "SIGPROF", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000751 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000752#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000753#ifdef SIGXCPU
754 x = PyInt_FromLong(SIGXCPU);
755 PyDict_SetItemString(d, "SIGXCPU", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000756 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000757#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000758#ifdef SIGXFSZ
759 x = PyInt_FromLong(SIGXFSZ);
760 PyDict_SetItemString(d, "SIGXFSZ", x);
Barry Warsaw73a75eb1997-01-09 23:50:28 +0000761 Py_XDECREF(x);
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000762#endif
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000763#ifdef SIGRTMIN
764 x = PyInt_FromLong(SIGRTMIN);
765 PyDict_SetItemString(d, "SIGRTMIN", x);
766 Py_XDECREF(x);
767#endif
768#ifdef SIGRTMAX
769 x = PyInt_FromLong(SIGRTMAX);
770 PyDict_SetItemString(d, "SIGRTMAX", x);
771 Py_XDECREF(x);
772#endif
Martin v. Löwis175af252002-01-12 11:43:25 +0000773#ifdef SIGINFO
774 x = PyInt_FromLong(SIGINFO);
775 PyDict_SetItemString(d, "SIGINFO", x);
776 Py_XDECREF(x);
777#endif
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000778
779#ifdef ITIMER_REAL
780 x = PyLong_FromLong(ITIMER_REAL);
781 PyDict_SetItemString(d, "ITIMER_REAL", x);
782 Py_DECREF(x);
783#endif
784#ifdef ITIMER_VIRTUAL
785 x = PyLong_FromLong(ITIMER_VIRTUAL);
786 PyDict_SetItemString(d, "ITIMER_VIRTUAL", x);
787 Py_DECREF(x);
788#endif
789#ifdef ITIMER_PROF
790 x = PyLong_FromLong(ITIMER_PROF);
791 PyDict_SetItemString(d, "ITIMER_PROF", x);
792 Py_DECREF(x);
793#endif
794
795#if defined (HAVE_SETITIMER) || defined (HAVE_GETITIMER)
796 ItimerError = PyErr_NewException("signal.ItimerError",
797 PyExc_IOError, NULL);
Neal Norwitz18aa3882008-08-24 05:04:52 +0000798 if (ItimerError != NULL)
799 PyDict_SetItemString(d, "ItimerError", ItimerError);
Martin v. Löwisaef18b12008-03-24 13:31:16 +0000800#endif
801
Brian Curtine5aa8862010-04-02 23:26:06 +0000802#ifdef CTRL_C_EVENT
803 x = PyInt_FromLong(CTRL_C_EVENT);
804 PyDict_SetItemString(d, "CTRL_C_EVENT", x);
805 Py_DECREF(x);
806#endif
807
808#ifdef CTRL_BREAK_EVENT
809 x = PyInt_FromLong(CTRL_BREAK_EVENT);
810 PyDict_SetItemString(d, "CTRL_BREAK_EVENT", x);
811 Py_DECREF(x);
812#endif
813
Barry Warsaw92971171997-01-03 00:14:25 +0000814 if (!PyErr_Occurred())
815 return;
816
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000817 /* Check for errors */
Barry Warsaw92971171997-01-03 00:14:25 +0000818 finally:
Guido van Rossum08c16611997-08-02 03:01:42 +0000819 return;
820}
821
822static void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000823finisignal(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000824{
825 int i;
Guido van Rossum7ff20ac1997-11-03 21:53:55 +0000826 PyObject *func;
Guido van Rossum08c16611997-08-02 03:01:42 +0000827
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000828 PyOS_setsig(SIGINT, old_siginthandler);
Guido van Rossum7ff20ac1997-11-03 21:53:55 +0000829 old_siginthandler = SIG_DFL;
Guido van Rossum08c16611997-08-02 03:01:42 +0000830
831 for (i = 1; i < NSIG; i++) {
Guido van Rossum7ff20ac1997-11-03 21:53:55 +0000832 func = Handlers[i].func;
Guido van Rossum08c16611997-08-02 03:01:42 +0000833 Handlers[i].tripped = 0;
Guido van Rossum08c16611997-08-02 03:01:42 +0000834 Handlers[i].func = NULL;
Guido van Rossum7ff20ac1997-11-03 21:53:55 +0000835 if (i != SIGINT && func != NULL && func != Py_None &&
836 func != DefaultHandler && func != IgnoreHandler)
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000837 PyOS_setsig(i, SIG_DFL);
Guido van Rossum7ff20ac1997-11-03 21:53:55 +0000838 Py_XDECREF(func);
Guido van Rossum08c16611997-08-02 03:01:42 +0000839 }
840
841 Py_XDECREF(IntHandler);
842 IntHandler = NULL;
843 Py_XDECREF(DefaultHandler);
844 DefaultHandler = NULL;
845 Py_XDECREF(IgnoreHandler);
846 IgnoreHandler = NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000847}
848
Barry Warsaw92971171997-01-03 00:14:25 +0000849
Barry Warsaw92971171997-01-03 00:14:25 +0000850/* Declared in pyerrors.h */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000851int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000852PyErr_CheckSignals(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000853{
854 int i;
Guido van Rossume4485b01994-09-07 14:32:49 +0000855 PyObject *f;
Barry Warsaw92971171997-01-03 00:14:25 +0000856
857 if (!is_tripped)
Guido van Rossum6299d1e1994-05-31 12:51:13 +0000858 return 0;
Guido van Rossum137c49c2007-12-10 23:00:12 +0000859
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000860#ifdef WITH_THREAD
Guido van Rossum65d5b571998-12-21 19:32:43 +0000861 if (PyThread_get_thread_ident() != main_thread)
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000862 return 0;
863#endif
Guido van Rossum137c49c2007-12-10 23:00:12 +0000864
865 /*
Gregory P. Smithf4666422009-08-12 17:02:37 +0000866 * The is_tripped variable is meant to speed up the calls to
Guido van Rossum137c49c2007-12-10 23:00:12 +0000867 * PyErr_CheckSignals (both directly or via pending calls) when no
868 * signal has arrived. This variable is set to 1 when a signal arrives
869 * and it is set to 0 here, when we know some signals arrived. This way
870 * we can run the registered handlers with no signals blocked.
871 *
872 * NOTE: with this approach we can have a situation where is_tripped is
873 * 1 but we have no more signals to handle (Handlers[i].tripped
874 * is 0 for every signal i). This won't do us any harm (except
875 * we're gonna spent some cycles for nothing). This happens when
876 * we receive a signal i after we zero is_tripped and before we
877 * check Handlers[i].tripped.
878 */
879 is_tripped = 0;
880
Guido van Rossum6297a7a2003-02-19 15:53:17 +0000881 if (!(f = (PyObject *)PyEval_GetFrame()))
Guido van Rossume4485b01994-09-07 14:32:49 +0000882 f = Py_None;
Guido van Rossum137c49c2007-12-10 23:00:12 +0000883
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000884 for (i = 1; i < NSIG; i++) {
Barry Warsaw92971171997-01-03 00:14:25 +0000885 if (Handlers[i].tripped) {
886 PyObject *result = NULL;
887 PyObject *arglist = Py_BuildValue("(iO)", i, f);
888 Handlers[i].tripped = 0;
889
890 if (arglist) {
891 result = PyEval_CallObject(Handlers[i].func,
892 arglist);
Guido van Rossume4485b01994-09-07 14:32:49 +0000893 Py_DECREF(arglist);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000894 }
Barry Warsaw92971171997-01-03 00:14:25 +0000895 if (!result)
Guido van Rossumfcdd0e41997-01-21 06:13:09 +0000896 return -1;
Barry Warsaw92971171997-01-03 00:14:25 +0000897
898 Py_DECREF(result);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000899 }
900 }
Guido van Rossum137c49c2007-12-10 23:00:12 +0000901
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000902 return 0;
903}
904
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000905
Barry Warsaw92971171997-01-03 00:14:25 +0000906/* Replacements for intrcheck.c functionality
907 * Declared in pyerrors.h
908 */
909void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000910PyErr_SetInterrupt(void)
Barry Warsaw92971171997-01-03 00:14:25 +0000911{
Guido van Rossum137c49c2007-12-10 23:00:12 +0000912 is_tripped = 1;
Barry Warsaw92971171997-01-03 00:14:25 +0000913 Handlers[SIGINT].tripped = 1;
Thomas Wouters334fb892000-07-25 12:56:38 +0000914 Py_AddPendingCall((int (*)(void *))PyErr_CheckSignals, NULL);
Barry Warsaw92971171997-01-03 00:14:25 +0000915}
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000916
917void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000918PyOS_InitInterrupts(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000919{
920 initsignal();
Guido van Rossum08c16611997-08-02 03:01:42 +0000921 _PyImport_FixupExtension("signal", "signal");
922}
923
924void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000925PyOS_FiniInterrupts(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000926{
927 finisignal();
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000928}
929
930int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000931PyOS_InterruptOccurred(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000932{
Barry Warsaw92971171997-01-03 00:14:25 +0000933 if (Handlers[SIGINT].tripped) {
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000934#ifdef WITH_THREAD
Guido van Rossum65d5b571998-12-21 19:32:43 +0000935 if (PyThread_get_thread_ident() != main_thread)
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000936 return 0;
937#endif
Barry Warsaw92971171997-01-03 00:14:25 +0000938 Handlers[SIGINT].tripped = 0;
Guido van Rossum6299d1e1994-05-31 12:51:13 +0000939 return 1;
940 }
941 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000942}
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000943
944void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000945PyOS_AfterFork(void)
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000946{
947#ifdef WITH_THREAD
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000948 PyEval_ReInitThreads();
Guido van Rossum65d5b571998-12-21 19:32:43 +0000949 main_thread = PyThread_get_thread_ident();
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000950 main_pid = getpid();
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +0000951 _PyImport_ReInitLock();
Benjamin Peterson114f7e52008-06-13 00:09:47 +0000952 PyThread_ReInitTLS();
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000953#endif
954}