blob: 46912853ffd96cc198ad0565cb30e0258ad8518b [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 Curtineb24d742010-04-12 17:16:38 +000010#include <Windows.h>
Benjamin Peterson2614cda2010-03-21 22:36:19 +000011#ifdef HAVE_PROCESS_H
Guido van Rossum644a12b1997-04-09 19:24:53 +000012#include <process.h>
13#endif
Benjamin Peterson2614cda2010-03-21 22:36:19 +000014#endif
Guido van Rossum644a12b1997-04-09 19:24:53 +000015
Benjamin Peterson2614cda2010-03-21 22:36:19 +000016#ifdef HAVE_SIGNAL_H
Guido van Rossum398d9fe1994-05-11 08:59:13 +000017#include <signal.h>
Benjamin Peterson2614cda2010-03-21 22:36:19 +000018#endif
19#ifdef HAVE_SYS_STAT_H
Christian Heimes5fb7c2a2007-12-24 08:52:31 +000020#include <sys/stat.h>
Benjamin Peterson2614cda2010-03-21 22:36:19 +000021#endif
Martin v. Löwis3bf3cc02008-03-24 14:05:07 +000022#ifdef HAVE_SYS_TIME_H
Martin v. Löwis823725e2008-03-24 13:39:54 +000023#include <sys/time.h>
Martin v. Löwis3bf3cc02008-03-24 14:05:07 +000024#endif
Christian Heimes5fb7c2a2007-12-24 08:52:31 +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 Pitrouf95a1b32010-05-09 15:52:27 +000037# define NSIG _NSIG /* For BSD/SysV */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000038# elif defined(_SIGMAX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000039# define NSIG (_SIGMAX + 1) /* For QNX */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000040# elif defined(SIGMAX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000041# define NSIG (SIGMAX + 1) /* For djgpp */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000042# else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +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 Pitrouf95a1b32010-05-09 15:52:27 +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
Christian Heimes5fb7c2a2007-12-24 08:52:31 +000089static sig_atomic_t wakeup_fd = -1;
90
Christian Heimesb76922a2007-12-11 01:06:40 +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öwis823725e2008-03-24 13:39:54 +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 Heimes1a8501c2008-10-02 19:56:01 +0000116Py_LOCAL_INLINE(double)
Martin v. Löwis823725e2008-03-24 13:39:54 +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 Pitrouf95a1b32010-05-09 15:52:27 +0000129 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000130
131 if(!(v = PyFloat_FromDouble(double_from_timeval(&iv->it_value)))) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000132 Py_DECREF(r);
133 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000134 }
135
136 PyTuple_SET_ITEM(r, 0, v);
137
138 if(!(v = PyFloat_FromDouble(double_from_timeval(&iv->it_interval)))) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000139 Py_DECREF(r);
140 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +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 Pitrouf95a1b32010-05-09 15:52:27 +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 Pitrouf95a1b32010-05-09 15:52:27 +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 Pitrouf95a1b32010-05-09 15:52:27 +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 Pitrouf95a1b32010-05-09 15:52:27 +0000179 /* See NOTES section above */
180 if (getpid() == main_pid) {
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000181#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +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 Pitrouf95a1b32010-05-09 15:52:27 +0000190 }
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000191#endif
Guido van Rossum602099a1994-09-14 13:32:22 +0000192#ifdef SIGCHLD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +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 Calderone6f137ca2010-05-09 03:18:57 +0000201#ifndef HAVE_SIGACTION
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000202 /* If the handler was not set up with sigaction, reinstall it. See
203 * Python/pythonrun.c for the implementation of PyOS_setsig which
204 * makes this true. See also issue8354. */
205 PyOS_setsig(sig_num, signal_handler);
Jean-Paul Calderone6f137ca2010-05-09 03:18:57 +0000206#endif
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 Pitrouf95a1b32010-05-09 15:52:27 +0000214 int t;
215 if (!PyArg_ParseTuple(args, "i:alarm", &t))
216 return NULL;
217 /* alarm() returns the number of seconds remaining */
218 return PyLong_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 Pitrouf95a1b32010-05-09 15:52:27 +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 Pitrouf95a1b32010-05-09 15:52:27 +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 Pitrouf95a1b32010-05-09 15:52:27 +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 Curtinef9efbd2010-08-06 19:27:32 +0000260#ifdef MS_WINDOWS
261 /* Validate that sig_num is one of the allowable signals */
Brian Curtinc734b312010-09-06 16:04:10 +0000262 switch (sig_num) {
263 case SIGABRT: break;
Brian Curtin9e88b5a2010-10-01 14:49:24 +0000264#ifdef SIGBREAK
265 /* Issue #10003: SIGBREAK is not documented as permitted, but works
266 and corresponds to CTRL_BREAK_EVENT. */
267 case SIGBREAK: break;
268#endif
Brian Curtinc734b312010-09-06 16:04:10 +0000269 case SIGFPE: break;
270 case SIGILL: break;
271 case SIGINT: break;
272 case SIGSEGV: break;
273 case SIGTERM: break;
274 default:
275 PyErr_SetString(PyExc_ValueError, "invalid signal value");
276 return NULL;
Brian Curtinef9efbd2010-08-06 19:27:32 +0000277 }
278#endif
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000279#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000280 if (PyThread_get_thread_ident() != main_thread) {
281 PyErr_SetString(PyExc_ValueError,
282 "signal only works in main thread");
283 return NULL;
284 }
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000285#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000286 if (sig_num < 1 || sig_num >= NSIG) {
287 PyErr_SetString(PyExc_ValueError,
288 "signal number out of range");
289 return NULL;
290 }
291 if (obj == IgnoreHandler)
292 func = SIG_IGN;
293 else if (obj == DefaultHandler)
294 func = SIG_DFL;
295 else if (!PyCallable_Check(obj)) {
296 PyErr_SetString(PyExc_TypeError,
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000297"signal handler must be signal.SIG_IGN, signal.SIG_DFL, or a callable object");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000298 return NULL;
299 }
300 else
301 func = signal_handler;
302 if (PyOS_setsig(sig_num, func) == SIG_ERR) {
303 PyErr_SetFromErrno(PyExc_RuntimeError);
304 return NULL;
305 }
306 old_handler = Handlers[sig_num].func;
307 Handlers[sig_num].tripped = 0;
308 Py_INCREF(obj);
309 Handlers[sig_num].func = obj;
310 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000311}
312
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000313PyDoc_STRVAR(signal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000314"signal(sig, action) -> action\n\
315\n\
316Set the action for the given signal. The action can be SIG_DFL,\n\
317SIG_IGN, or a callable Python object. The previous action is\n\
318returned. See getsignal() for possible return values.\n\
319\n\
320*** IMPORTANT NOTICE ***\n\
321A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000322the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000323
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000324
Guido van Rossume4485b01994-09-07 14:32:49 +0000325static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000326signal_getsignal(PyObject *self, PyObject *args)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000327{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000328 int sig_num;
329 PyObject *old_handler;
330 if (!PyArg_ParseTuple(args, "i:getsignal", &sig_num))
331 return NULL;
332 if (sig_num < 1 || sig_num >= NSIG) {
333 PyErr_SetString(PyExc_ValueError,
334 "signal number out of range");
335 return NULL;
336 }
337 old_handler = Handlers[sig_num].func;
338 Py_INCREF(old_handler);
339 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000340}
341
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000342PyDoc_STRVAR(getsignal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000343"getsignal(sig) -> action\n\
344\n\
345Return the current action for the given signal. The return value can be:\n\
346SIG_IGN -- if the signal is being ignored\n\
347SIG_DFL -- if the default action for the signal is in effect\n\
348None -- if an unknown handler is in effect\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000349anything else -- the callable Python object used as a handler");
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000350
Christian Heimes8640e742008-02-23 16:23:06 +0000351#ifdef HAVE_SIGINTERRUPT
352PyDoc_STRVAR(siginterrupt_doc,
353"siginterrupt(sig, flag) -> None\n\
354change system call restart behaviour: if flag is False, system calls\n\
355will be restarted when interrupted by signal sig, else system calls\n\
356will be interrupted.");
357
358static PyObject *
359signal_siginterrupt(PyObject *self, PyObject *args)
360{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000361 int sig_num;
362 int flag;
Christian Heimes8640e742008-02-23 16:23:06 +0000363
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000364 if (!PyArg_ParseTuple(args, "ii:siginterrupt", &sig_num, &flag))
365 return NULL;
366 if (sig_num < 1 || sig_num >= NSIG) {
367 PyErr_SetString(PyExc_ValueError,
368 "signal number out of range");
369 return NULL;
370 }
371 if (siginterrupt(sig_num, flag)<0) {
372 PyErr_SetFromErrno(PyExc_RuntimeError);
373 return NULL;
374 }
Christian Heimes8640e742008-02-23 16:23:06 +0000375
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000376 Py_INCREF(Py_None);
377 return Py_None;
Christian Heimes8640e742008-02-23 16:23:06 +0000378}
379
380#endif
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000381
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000382static PyObject *
383signal_set_wakeup_fd(PyObject *self, PyObject *args)
384{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000385 struct stat buf;
386 int fd, old_fd;
387 if (!PyArg_ParseTuple(args, "i:set_wakeup_fd", &fd))
388 return NULL;
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000389#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000390 if (PyThread_get_thread_ident() != main_thread) {
391 PyErr_SetString(PyExc_ValueError,
392 "set_wakeup_fd only works in main thread");
393 return NULL;
394 }
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000395#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000396 if (fd != -1 && fstat(fd, &buf) != 0) {
397 PyErr_SetString(PyExc_ValueError, "invalid fd");
398 return NULL;
399 }
400 old_fd = wakeup_fd;
401 wakeup_fd = fd;
402 return PyLong_FromLong(old_fd);
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000403}
404
405PyDoc_STRVAR(set_wakeup_fd_doc,
406"set_wakeup_fd(fd) -> fd\n\
407\n\
408Sets the fd to be written to (with '\\0') when a signal\n\
409comes in. A library can use this to wakeup select or poll.\n\
410The previous fd is returned.\n\
411\n\
412The fd must be non-blocking.");
413
414/* C API for the same, without all the error checking */
415int
416PySignal_SetWakeupFd(int fd)
417{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000418 int old_fd = wakeup_fd;
419 if (fd < 0)
420 fd = -1;
421 wakeup_fd = fd;
422 return old_fd;
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000423}
424
425
Martin v. Löwis823725e2008-03-24 13:39:54 +0000426#ifdef HAVE_SETITIMER
427static PyObject *
428signal_setitimer(PyObject *self, PyObject *args)
429{
430 double first;
431 double interval = 0;
432 int which;
433 struct itimerval new, old;
434
435 if(!PyArg_ParseTuple(args, "id|d:setitimer", &which, &first, &interval))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000436 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000437
438 timeval_from_double(first, &new.it_value);
439 timeval_from_double(interval, &new.it_interval);
440 /* Let OS check "which" value */
441 if (setitimer(which, &new, &old) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000442 PyErr_SetFromErrno(ItimerError);
443 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000444 }
445
446 return itimer_retval(&old);
447}
448
449PyDoc_STRVAR(setitimer_doc,
450"setitimer(which, seconds[, interval])\n\
451\n\
452Sets given itimer (one of ITIMER_REAL, ITIMER_VIRTUAL\n\
453or ITIMER_PROF) to fire after value seconds and after\n\
454that every interval seconds.\n\
455The itimer can be cleared by setting seconds to zero.\n\
456\n\
457Returns old values as a tuple: (delay, interval).");
458#endif
459
460
461#ifdef HAVE_GETITIMER
462static PyObject *
463signal_getitimer(PyObject *self, PyObject *args)
464{
465 int which;
466 struct itimerval old;
467
468 if (!PyArg_ParseTuple(args, "i:getitimer", &which))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000469 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000470
471 if (getitimer(which, &old) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000472 PyErr_SetFromErrno(ItimerError);
473 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000474 }
475
476 return itimer_retval(&old);
477}
478
479PyDoc_STRVAR(getitimer_doc,
480"getitimer(which)\n\
481\n\
482Returns current value of given itimer.");
483#endif
484
485
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000486/* List of functions defined in the module */
Barry Warsaw92971171997-01-03 00:14:25 +0000487static PyMethodDef signal_methods[] = {
Guido van Rossum1171ee61997-08-22 20:42:00 +0000488#ifdef HAVE_ALARM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000489 {"alarm", signal_alarm, METH_VARARGS, alarm_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000490#endif
Martin v. Löwis823725e2008-03-24 13:39:54 +0000491#ifdef HAVE_SETITIMER
492 {"setitimer", signal_setitimer, METH_VARARGS, setitimer_doc},
493#endif
494#ifdef HAVE_GETITIMER
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000495 {"getitimer", signal_getitimer, METH_VARARGS, getitimer_doc},
Martin v. Löwis823725e2008-03-24 13:39:54 +0000496#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000497 {"signal", signal_signal, METH_VARARGS, signal_doc},
498 {"getsignal", signal_getsignal, METH_VARARGS, getsignal_doc},
499 {"set_wakeup_fd", signal_set_wakeup_fd, METH_VARARGS, set_wakeup_fd_doc},
Christian Heimes8640e742008-02-23 16:23:06 +0000500#ifdef HAVE_SIGINTERRUPT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000501 {"siginterrupt", signal_siginterrupt, METH_VARARGS, siginterrupt_doc},
Christian Heimes8640e742008-02-23 16:23:06 +0000502#endif
Guido van Rossum1171ee61997-08-22 20:42:00 +0000503#ifdef HAVE_PAUSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000504 {"pause", (PyCFunction)signal_pause,
505 METH_NOARGS,pause_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000506#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000507 {"default_int_handler", signal_default_int_handler,
508 METH_VARARGS, default_int_handler_doc},
509 {NULL, NULL} /* sentinel */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000510};
511
Barry Warsaw92971171997-01-03 00:14:25 +0000512
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000513PyDoc_STRVAR(module_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000514"This module provides mechanisms to use signal handlers in Python.\n\
515\n\
516Functions:\n\
517\n\
518alarm() -- cause SIGALRM after a specified time [Unix only]\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000519setitimer() -- cause a signal (described below) after a specified\n\
520 float time and the timer may restart then [Unix only]\n\
521getitimer() -- get current value of timer [Unix only]\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000522signal() -- set the action for a given signal\n\
523getsignal() -- get the signal action for a given signal\n\
524pause() -- wait until a signal arrives [Unix only]\n\
525default_int_handler() -- default SIGINT handler\n\
526\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000527signal constants:\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000528SIG_DFL -- used to refer to the system default handler\n\
529SIG_IGN -- used to ignore the signal\n\
530NSIG -- number of defined signals\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000531SIGINT, SIGTERM, etc. -- signal numbers\n\
532\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000533itimer constants:\n\
534ITIMER_REAL -- decrements in real time, and delivers SIGALRM upon\n\
535 expiration\n\
536ITIMER_VIRTUAL -- decrements only when the process is executing,\n\
537 and delivers SIGVTALRM upon expiration\n\
538ITIMER_PROF -- decrements both when the process is executing and\n\
539 when the system is executing on behalf of the process.\n\
540 Coupled with ITIMER_VIRTUAL, this timer is usually\n\
541 used to profile the time spent by the application\n\
542 in user and kernel space. SIGPROF is delivered upon\n\
543 expiration.\n\
544\n\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000545*** IMPORTANT NOTICE ***\n\
546A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000547the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000548
Martin v. Löwis1a214512008-06-11 05:26:20 +0000549static struct PyModuleDef signalmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000550 PyModuleDef_HEAD_INIT,
551 "signal",
552 module_doc,
553 -1,
554 signal_methods,
555 NULL,
556 NULL,
557 NULL,
558 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +0000559};
560
Mark Hammondfe51c6d2002-08-02 02:27:13 +0000561PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +0000562PyInit_signal(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000563{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000564 PyObject *m, *d, *x;
565 int i;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000566
567#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000568 main_thread = PyThread_get_thread_ident();
569 main_pid = getpid();
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000570#endif
571
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000572 /* Create the module and add the functions */
573 m = PyModule_Create(&signalmodule);
574 if (m == NULL)
575 return NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000576
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000577 /* Add some symbolic constants to the module */
578 d = PyModule_GetDict(m);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000579
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000580 x = DefaultHandler = PyLong_FromVoidPtr((void *)SIG_DFL);
581 if (!x || PyDict_SetItemString(d, "SIG_DFL", x) < 0)
582 goto finally;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000583
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000584 x = IgnoreHandler = PyLong_FromVoidPtr((void *)SIG_IGN);
585 if (!x || PyDict_SetItemString(d, "SIG_IGN", x) < 0)
586 goto finally;
Barry Warsaw92971171997-01-03 00:14:25 +0000587
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000588 x = PyLong_FromLong((long)NSIG);
589 if (!x || PyDict_SetItemString(d, "NSIG", x) < 0)
590 goto finally;
591 Py_DECREF(x);
Barry Warsaw92971171997-01-03 00:14:25 +0000592
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000593 x = IntHandler = PyDict_GetItemString(d, "default_int_handler");
594 if (!x)
595 goto finally;
596 Py_INCREF(IntHandler);
Barry Warsaw92971171997-01-03 00:14:25 +0000597
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000598 Handlers[0].tripped = 0;
599 for (i = 1; i < NSIG; i++) {
600 void (*t)(int);
601 t = PyOS_getsig(i);
602 Handlers[i].tripped = 0;
603 if (t == SIG_DFL)
604 Handlers[i].func = DefaultHandler;
605 else if (t == SIG_IGN)
606 Handlers[i].func = IgnoreHandler;
607 else
608 Handlers[i].func = Py_None; /* None of our business */
609 Py_INCREF(Handlers[i].func);
610 }
611 if (Handlers[SIGINT].func == DefaultHandler) {
612 /* Install default int handler */
613 Py_INCREF(IntHandler);
614 Py_DECREF(Handlers[SIGINT].func);
615 Handlers[SIGINT].func = IntHandler;
616 old_siginthandler = PyOS_setsig(SIGINT, signal_handler);
617 }
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000618
619#ifdef SIGHUP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000620 x = PyLong_FromLong(SIGHUP);
621 PyDict_SetItemString(d, "SIGHUP", x);
622 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000623#endif
624#ifdef SIGINT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000625 x = PyLong_FromLong(SIGINT);
626 PyDict_SetItemString(d, "SIGINT", x);
627 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000628#endif
Tim Peters1ce3cf72001-10-01 17:58:40 +0000629#ifdef SIGBREAK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000630 x = PyLong_FromLong(SIGBREAK);
631 PyDict_SetItemString(d, "SIGBREAK", x);
632 Py_XDECREF(x);
Tim Peters1ce3cf72001-10-01 17:58:40 +0000633#endif
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000634#ifdef SIGQUIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000635 x = PyLong_FromLong(SIGQUIT);
636 PyDict_SetItemString(d, "SIGQUIT", x);
637 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000638#endif
639#ifdef SIGILL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000640 x = PyLong_FromLong(SIGILL);
641 PyDict_SetItemString(d, "SIGILL", x);
642 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000643#endif
644#ifdef SIGTRAP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000645 x = PyLong_FromLong(SIGTRAP);
646 PyDict_SetItemString(d, "SIGTRAP", x);
647 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000648#endif
649#ifdef SIGIOT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000650 x = PyLong_FromLong(SIGIOT);
651 PyDict_SetItemString(d, "SIGIOT", x);
652 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000653#endif
654#ifdef SIGABRT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000655 x = PyLong_FromLong(SIGABRT);
656 PyDict_SetItemString(d, "SIGABRT", x);
657 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000658#endif
659#ifdef SIGEMT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000660 x = PyLong_FromLong(SIGEMT);
661 PyDict_SetItemString(d, "SIGEMT", x);
662 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000663#endif
664#ifdef SIGFPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000665 x = PyLong_FromLong(SIGFPE);
666 PyDict_SetItemString(d, "SIGFPE", x);
667 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000668#endif
669#ifdef SIGKILL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000670 x = PyLong_FromLong(SIGKILL);
671 PyDict_SetItemString(d, "SIGKILL", x);
672 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000673#endif
674#ifdef SIGBUS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000675 x = PyLong_FromLong(SIGBUS);
676 PyDict_SetItemString(d, "SIGBUS", x);
677 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000678#endif
679#ifdef SIGSEGV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000680 x = PyLong_FromLong(SIGSEGV);
681 PyDict_SetItemString(d, "SIGSEGV", x);
682 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000683#endif
684#ifdef SIGSYS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000685 x = PyLong_FromLong(SIGSYS);
686 PyDict_SetItemString(d, "SIGSYS", x);
687 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000688#endif
689#ifdef SIGPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000690 x = PyLong_FromLong(SIGPIPE);
691 PyDict_SetItemString(d, "SIGPIPE", x);
692 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000693#endif
694#ifdef SIGALRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000695 x = PyLong_FromLong(SIGALRM);
696 PyDict_SetItemString(d, "SIGALRM", x);
697 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000698#endif
699#ifdef SIGTERM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000700 x = PyLong_FromLong(SIGTERM);
701 PyDict_SetItemString(d, "SIGTERM", x);
702 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000703#endif
704#ifdef SIGUSR1
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000705 x = PyLong_FromLong(SIGUSR1);
706 PyDict_SetItemString(d, "SIGUSR1", x);
707 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000708#endif
709#ifdef SIGUSR2
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000710 x = PyLong_FromLong(SIGUSR2);
711 PyDict_SetItemString(d, "SIGUSR2", x);
712 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000713#endif
714#ifdef SIGCLD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000715 x = PyLong_FromLong(SIGCLD);
716 PyDict_SetItemString(d, "SIGCLD", x);
717 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000718#endif
719#ifdef SIGCHLD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000720 x = PyLong_FromLong(SIGCHLD);
721 PyDict_SetItemString(d, "SIGCHLD", x);
722 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000723#endif
724#ifdef SIGPWR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000725 x = PyLong_FromLong(SIGPWR);
726 PyDict_SetItemString(d, "SIGPWR", x);
727 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000728#endif
729#ifdef SIGIO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000730 x = PyLong_FromLong(SIGIO);
731 PyDict_SetItemString(d, "SIGIO", x);
732 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000733#endif
734#ifdef SIGURG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000735 x = PyLong_FromLong(SIGURG);
736 PyDict_SetItemString(d, "SIGURG", x);
737 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000738#endif
739#ifdef SIGWINCH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000740 x = PyLong_FromLong(SIGWINCH);
741 PyDict_SetItemString(d, "SIGWINCH", x);
742 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000743#endif
744#ifdef SIGPOLL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000745 x = PyLong_FromLong(SIGPOLL);
746 PyDict_SetItemString(d, "SIGPOLL", x);
747 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000748#endif
749#ifdef SIGSTOP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000750 x = PyLong_FromLong(SIGSTOP);
751 PyDict_SetItemString(d, "SIGSTOP", x);
752 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000753#endif
754#ifdef SIGTSTP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000755 x = PyLong_FromLong(SIGTSTP);
756 PyDict_SetItemString(d, "SIGTSTP", x);
757 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000758#endif
759#ifdef SIGCONT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000760 x = PyLong_FromLong(SIGCONT);
761 PyDict_SetItemString(d, "SIGCONT", x);
762 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000763#endif
764#ifdef SIGTTIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000765 x = PyLong_FromLong(SIGTTIN);
766 PyDict_SetItemString(d, "SIGTTIN", x);
767 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000768#endif
769#ifdef SIGTTOU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000770 x = PyLong_FromLong(SIGTTOU);
771 PyDict_SetItemString(d, "SIGTTOU", x);
772 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000773#endif
774#ifdef SIGVTALRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000775 x = PyLong_FromLong(SIGVTALRM);
776 PyDict_SetItemString(d, "SIGVTALRM", x);
777 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000778#endif
779#ifdef SIGPROF
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000780 x = PyLong_FromLong(SIGPROF);
781 PyDict_SetItemString(d, "SIGPROF", x);
782 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000783#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000784#ifdef SIGXCPU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000785 x = PyLong_FromLong(SIGXCPU);
786 PyDict_SetItemString(d, "SIGXCPU", x);
787 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000788#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000789#ifdef SIGXFSZ
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000790 x = PyLong_FromLong(SIGXFSZ);
791 PyDict_SetItemString(d, "SIGXFSZ", x);
792 Py_XDECREF(x);
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000793#endif
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000794#ifdef SIGRTMIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000795 x = PyLong_FromLong(SIGRTMIN);
796 PyDict_SetItemString(d, "SIGRTMIN", x);
797 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000798#endif
799#ifdef SIGRTMAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000800 x = PyLong_FromLong(SIGRTMAX);
801 PyDict_SetItemString(d, "SIGRTMAX", x);
802 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000803#endif
Martin v. Löwis175af252002-01-12 11:43:25 +0000804#ifdef SIGINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000805 x = PyLong_FromLong(SIGINFO);
806 PyDict_SetItemString(d, "SIGINFO", x);
807 Py_XDECREF(x);
Martin v. Löwis175af252002-01-12 11:43:25 +0000808#endif
Martin v. Löwis823725e2008-03-24 13:39:54 +0000809
810#ifdef ITIMER_REAL
811 x = PyLong_FromLong(ITIMER_REAL);
812 PyDict_SetItemString(d, "ITIMER_REAL", x);
813 Py_DECREF(x);
814#endif
815#ifdef ITIMER_VIRTUAL
816 x = PyLong_FromLong(ITIMER_VIRTUAL);
817 PyDict_SetItemString(d, "ITIMER_VIRTUAL", x);
818 Py_DECREF(x);
819#endif
820#ifdef ITIMER_PROF
821 x = PyLong_FromLong(ITIMER_PROF);
822 PyDict_SetItemString(d, "ITIMER_PROF", x);
823 Py_DECREF(x);
824#endif
825
826#if defined (HAVE_SETITIMER) || defined (HAVE_GETITIMER)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000827 ItimerError = PyErr_NewException("signal.ItimerError",
828 PyExc_IOError, NULL);
Neal Norwitz2f99b242008-08-24 05:48:10 +0000829 if (ItimerError != NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000830 PyDict_SetItemString(d, "ItimerError", ItimerError);
Martin v. Löwis823725e2008-03-24 13:39:54 +0000831#endif
832
Brian Curtineb24d742010-04-12 17:16:38 +0000833#ifdef CTRL_C_EVENT
834 x = PyLong_FromLong(CTRL_C_EVENT);
835 PyDict_SetItemString(d, "CTRL_C_EVENT", x);
836 Py_DECREF(x);
837#endif
838
839#ifdef CTRL_BREAK_EVENT
840 x = PyLong_FromLong(CTRL_BREAK_EVENT);
841 PyDict_SetItemString(d, "CTRL_BREAK_EVENT", x);
842 Py_DECREF(x);
843#endif
844
Martin v. Löwis1a214512008-06-11 05:26:20 +0000845 if (PyErr_Occurred()) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000846 Py_DECREF(m);
847 m = NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +0000848 }
Barry Warsaw92971171997-01-03 00:14:25 +0000849
Barry Warsaw92971171997-01-03 00:14:25 +0000850 finally:
Martin v. Löwis1a214512008-06-11 05:26:20 +0000851 return m;
Guido van Rossum08c16611997-08-02 03:01:42 +0000852}
853
854static void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000855finisignal(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000856{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000857 int i;
858 PyObject *func;
Guido van Rossum08c16611997-08-02 03:01:42 +0000859
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000860 PyOS_setsig(SIGINT, old_siginthandler);
861 old_siginthandler = SIG_DFL;
Guido van Rossum08c16611997-08-02 03:01:42 +0000862
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000863 for (i = 1; i < NSIG; i++) {
864 func = Handlers[i].func;
865 Handlers[i].tripped = 0;
866 Handlers[i].func = NULL;
867 if (i != SIGINT && func != NULL && func != Py_None &&
868 func != DefaultHandler && func != IgnoreHandler)
869 PyOS_setsig(i, SIG_DFL);
870 Py_XDECREF(func);
871 }
Guido van Rossum08c16611997-08-02 03:01:42 +0000872
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000873 Py_XDECREF(IntHandler);
874 IntHandler = NULL;
875 Py_XDECREF(DefaultHandler);
876 DefaultHandler = NULL;
877 Py_XDECREF(IgnoreHandler);
878 IgnoreHandler = NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000879}
880
Barry Warsaw92971171997-01-03 00:14:25 +0000881
Barry Warsaw92971171997-01-03 00:14:25 +0000882/* Declared in pyerrors.h */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000883int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000884PyErr_CheckSignals(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000885{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000886 int i;
887 PyObject *f;
Barry Warsaw92971171997-01-03 00:14:25 +0000888
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000889 if (!is_tripped)
890 return 0;
Christian Heimesb76922a2007-12-11 01:06:40 +0000891
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000892#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000893 if (PyThread_get_thread_ident() != main_thread)
894 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000895#endif
Christian Heimesb76922a2007-12-11 01:06:40 +0000896
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000897 /*
898 * The is_tripped variable is meant to speed up the calls to
899 * PyErr_CheckSignals (both directly or via pending calls) when no
900 * signal has arrived. This variable is set to 1 when a signal arrives
901 * and it is set to 0 here, when we know some signals arrived. This way
902 * we can run the registered handlers with no signals blocked.
903 *
904 * NOTE: with this approach we can have a situation where is_tripped is
905 * 1 but we have no more signals to handle (Handlers[i].tripped
906 * is 0 for every signal i). This won't do us any harm (except
907 * we're gonna spent some cycles for nothing). This happens when
908 * we receive a signal i after we zero is_tripped and before we
909 * check Handlers[i].tripped.
910 */
911 is_tripped = 0;
Christian Heimesb76922a2007-12-11 01:06:40 +0000912
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000913 if (!(f = (PyObject *)PyEval_GetFrame()))
914 f = Py_None;
Christian Heimesb76922a2007-12-11 01:06:40 +0000915
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000916 for (i = 1; i < NSIG; i++) {
917 if (Handlers[i].tripped) {
918 PyObject *result = NULL;
919 PyObject *arglist = Py_BuildValue("(iO)", i, f);
920 Handlers[i].tripped = 0;
Barry Warsaw92971171997-01-03 00:14:25 +0000921
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000922 if (arglist) {
923 result = PyEval_CallObject(Handlers[i].func,
924 arglist);
925 Py_DECREF(arglist);
926 }
927 if (!result)
928 return -1;
Barry Warsaw92971171997-01-03 00:14:25 +0000929
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000930 Py_DECREF(result);
931 }
932 }
Christian Heimesb76922a2007-12-11 01:06:40 +0000933
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000934 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000935}
936
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000937
Barry Warsaw92971171997-01-03 00:14:25 +0000938/* Replacements for intrcheck.c functionality
939 * Declared in pyerrors.h
940 */
941void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000942PyErr_SetInterrupt(void)
Barry Warsaw92971171997-01-03 00:14:25 +0000943{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000944 is_tripped = 1;
945 Handlers[SIGINT].tripped = 1;
946 Py_AddPendingCall((int (*)(void *))PyErr_CheckSignals, NULL);
Barry Warsaw92971171997-01-03 00:14:25 +0000947}
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000948
949void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000950PyOS_InitInterrupts(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000951{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000952 PyObject *m = PyInit_signal();
953 if (m) {
Victor Stinner49d3f252010-10-17 01:24:53 +0000954 _PyImport_FixupBuiltin(m, "signal");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000955 Py_DECREF(m);
956 }
Guido van Rossum08c16611997-08-02 03:01:42 +0000957}
958
959void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000960PyOS_FiniInterrupts(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000961{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000962 finisignal();
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000963}
964
965int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000966PyOS_InterruptOccurred(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000967{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000968 if (Handlers[SIGINT].tripped) {
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000969#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000970 if (PyThread_get_thread_ident() != main_thread)
971 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000972#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000973 Handlers[SIGINT].tripped = 0;
974 return 1;
975 }
976 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000977}
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000978
979void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000980PyOS_AfterFork(void)
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000981{
982#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000983 PyEval_ReInitThreads();
984 main_thread = PyThread_get_thread_ident();
985 main_pid = getpid();
986 _PyImport_ReInitLock();
987 PyThread_ReInitTLS();
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000988#endif
989}