blob: 42d8dd82218f80dc0092ad229d74ae08dfec5ca9 [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 Curtinb42a21d2010-08-06 19:52:50 +000010#include <Windows.h>
11#ifdef HAVE_PROCESS_H
Guido van Rossum644a12b1997-04-09 19:24:53 +000012#include <process.h>
13#endif
Brian Curtinb42a21d2010-08-06 19:52:50 +000014#endif
Guido van Rossum644a12b1997-04-09 19:24:53 +000015
Guido van Rossum398d9fe1994-05-11 08:59:13 +000016#include <signal.h>
17
Christian Heimes5fb7c2a2007-12-24 08:52:31 +000018#include <sys/stat.h>
Martin v. Löwis3bf3cc02008-03-24 14:05:07 +000019#ifdef HAVE_SYS_TIME_H
Martin v. Löwis823725e2008-03-24 13:39:54 +000020#include <sys/time.h>
Martin v. Löwis3bf3cc02008-03-24 14:05:07 +000021#endif
Christian Heimes5fb7c2a2007-12-24 08:52:31 +000022
Guido van Rossumbb4ba121994-06-23 11:25:45 +000023#ifndef SIG_ERR
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +000024#define SIG_ERR ((PyOS_sighandler_t)(-1))
Guido van Rossumbb4ba121994-06-23 11:25:45 +000025#endif
26
Andrew MacIntyre7bf68332002-03-03 02:59:16 +000027#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000028#define NSIG 12
29#include <process.h>
30#endif
31
Guido van Rossum3bbc62e1995-01-02 19:30:30 +000032#ifndef NSIG
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000033# if defined(_NSIG)
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +000034# define NSIG _NSIG /* For BSD/SysV */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000035# elif defined(_SIGMAX)
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +000036# define NSIG (_SIGMAX + 1) /* For QNX */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000037# elif defined(SIGMAX)
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +000038# define NSIG (SIGMAX + 1) /* For djgpp */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000039# else
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +000040# define NSIG 64 /* Use a reasonable default value */
Marc-André Lemburg8bcfb8a2000-07-04 14:17:33 +000041# endif
Guido van Rossum3bbc62e1995-01-02 19:30:30 +000042#endif
43
44
Guido van Rossumbb4ba121994-06-23 11:25:45 +000045/*
46 NOTES ON THE INTERACTION BETWEEN SIGNALS AND THREADS
47
48 When threads are supported, we want the following semantics:
49
50 - only the main thread can set a signal handler
51 - any thread can get a signal handler
52 - signals are only delivered to the main thread
53
54 I.e. we don't support "synchronous signals" like SIGFPE (catching
55 this doesn't make much sense in Python anyway) nor do we support
56 signals as a means of inter-thread communication, since not all
57 thread implementations support that (at least our thread library
58 doesn't).
59
60 We still have the problem that in some implementations signals
61 generated by the keyboard (e.g. SIGINT) are delivered to all
62 threads (e.g. SGI), while in others (e.g. Solaris) such signals are
63 delivered to one random thread (an intermediate possibility would
Guido van Rossuma3c04b01995-01-12 11:29:01 +000064 be to deliver it to the main thread -- POSIX?). For now, we have
Guido van Rossumbb4ba121994-06-23 11:25:45 +000065 a working implementation that works in all three cases -- the
66 handler ignores signals if getpid() isn't the same as in the main
67 thread. XXX This is a hack.
68
Guido van Rossum9e8181b2000-09-19 00:46:46 +000069 GNU pth is a user-space threading library, and as such, all threads
70 run within the same process. In this case, if the currently running
71 thread is not the main_thread, send the signal to the main_thread.
Guido van Rossumbb4ba121994-06-23 11:25:45 +000072*/
73
74#ifdef WITH_THREAD
Guido van Rossum295b8e51997-06-06 21:16:41 +000075#include <sys/types.h> /* For pid_t */
Guido van Rossum49b56061998-10-01 20:42:43 +000076#include "pythread.h"
Guido van Rossumbb4ba121994-06-23 11:25:45 +000077static long main_thread;
78static pid_t main_pid;
79#endif
80
Barry Warsaw92971171997-01-03 00:14:25 +000081static struct {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +000082 int tripped;
83 PyObject *func;
Barry Warsaw92971171997-01-03 00:14:25 +000084} Handlers[NSIG];
Guido van Rossum398d9fe1994-05-11 08:59:13 +000085
Christian Heimes5fb7c2a2007-12-24 08:52:31 +000086static sig_atomic_t wakeup_fd = -1;
87
Christian Heimesb76922a2007-12-11 01:06:40 +000088/* Speed up sigcheck() when none tripped */
89static volatile sig_atomic_t is_tripped = 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +000090
Barry Warsaw92971171997-01-03 00:14:25 +000091static PyObject *DefaultHandler;
92static PyObject *IgnoreHandler;
93static PyObject *IntHandler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +000094
Martin v. Löwisf58de1b2001-03-06 12:13:56 +000095/* On Solaris 8, gcc will produce a warning that the function
96 declaration is not a prototype. This is caused by the definition of
97 SIG_DFL as (void (*)())0; the correct declaration would have been
98 (void (*)(int))0. */
99
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000100static PyOS_sighandler_t old_siginthandler = SIG_DFL;
Guido van Rossum08c16611997-08-02 03:01:42 +0000101
Martin v. Löwis823725e2008-03-24 13:39:54 +0000102#ifdef HAVE_GETITIMER
103static PyObject *ItimerError;
104
105/* auxiliary functions for setitimer/getitimer */
106static void
107timeval_from_double(double d, struct timeval *tv)
108{
109 tv->tv_sec = floor(d);
110 tv->tv_usec = fmod(d, 1.0) * 1000000.0;
111}
112
Christian Heimes1a8501c2008-10-02 19:56:01 +0000113Py_LOCAL_INLINE(double)
Martin v. Löwis823725e2008-03-24 13:39:54 +0000114double_from_timeval(struct timeval *tv)
115{
116 return tv->tv_sec + (double)(tv->tv_usec / 1000000.0);
117}
118
119static PyObject *
120itimer_retval(struct itimerval *iv)
121{
122 PyObject *r, *v;
123
124 r = PyTuple_New(2);
125 if (r == NULL)
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000126 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000127
128 if(!(v = PyFloat_FromDouble(double_from_timeval(&iv->it_value)))) {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000129 Py_DECREF(r);
130 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000131 }
132
133 PyTuple_SET_ITEM(r, 0, v);
134
135 if(!(v = PyFloat_FromDouble(double_from_timeval(&iv->it_interval)))) {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000136 Py_DECREF(r);
137 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000138 }
139
140 PyTuple_SET_ITEM(r, 1, v);
141
142 return r;
143}
144#endif
Barry Warsaw92971171997-01-03 00:14:25 +0000145
Guido van Rossume4485b01994-09-07 14:32:49 +0000146static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000147signal_default_int_handler(PyObject *self, PyObject *args)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000148{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000149 PyErr_SetNone(PyExc_KeyboardInterrupt);
150 return NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000151}
152
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000153PyDoc_STRVAR(default_int_handler_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000154"default_int_handler(...)\n\
155\n\
Michael W. Hudson24ec2112004-06-17 15:55:53 +0000156The default handler for SIGINT installed by Python.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000157It raises KeyboardInterrupt.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000158
Thomas Wouters0796b002000-07-22 23:49:30 +0000159
160static int
161checksignals_witharg(void * unused)
162{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000163 return PyErr_CheckSignals();
Thomas Wouters0796b002000-07-22 23:49:30 +0000164}
165
Tim Peters4f1b2082000-07-23 21:18:09 +0000166static void
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000167signal_handler(int sig_num)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000168{
Antoine Pitrou8d46e422010-11-05 19:54:58 +0000169 int save_errno = errno;
170
171#if defined(WITH_THREAD) && defined(WITH_PTH)
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000172 if (PyThread_get_thread_ident() != main_thread) {
173 pth_raise(*(pth_t *) main_thread, sig_num);
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000174 }
Antoine Pitrou8d46e422010-11-05 19:54:58 +0000175 else
Guido van Rossum9e8181b2000-09-19 00:46:46 +0000176#endif
Antoine Pitrou8d46e422010-11-05 19:54:58 +0000177 {
178#ifdef WITH_THREAD
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000179 /* See NOTES section above */
Antoine Pitrou8d46e422010-11-05 19:54:58 +0000180 if (getpid() == main_pid)
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000181#endif
Antoine Pitrou8d46e422010-11-05 19:54:58 +0000182 {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000183 Handlers[sig_num].tripped = 1;
184 /* Set is_tripped after setting .tripped, as it gets
185 cleared in PyErr_CheckSignals() before .tripped. */
186 is_tripped = 1;
187 Py_AddPendingCall(checksignals_witharg, NULL);
188 if (wakeup_fd != -1)
189 write(wakeup_fd, "\0", 1);
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000190 }
Antoine Pitrou8d46e422010-11-05 19:54:58 +0000191
Jean-Paul Calderone9c39bc72010-05-09 03:25:16 +0000192#ifndef HAVE_SIGACTION
Antoine Pitrou8d46e422010-11-05 19:54:58 +0000193#ifdef 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 if (sig_num != SIGCHLD)
199#endif
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000200 /* If the handler was not set up with sigaction, reinstall it. See
201 * Python/pythonrun.c for the implementation of PyOS_setsig which
202 * makes this true. See also issue8354. */
203 PyOS_setsig(sig_num, signal_handler);
Jean-Paul Calderone9c39bc72010-05-09 03:25:16 +0000204#endif
Antoine Pitrou8d46e422010-11-05 19:54:58 +0000205 }
206
207 /* Issue #10311: asynchronously executing signal handlers should not
208 mutate errno under the feet of unsuspecting C code. */
209 errno = save_errno;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000210}
Guido van Rossume4485b01994-09-07 14:32:49 +0000211
Guido van Rossum06d511d1995-03-10 15:13:48 +0000212
Guido van Rossum1171ee61997-08-22 20:42:00 +0000213#ifdef HAVE_ALARM
Guido van Rossume4485b01994-09-07 14:32:49 +0000214static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000215signal_alarm(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000216{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000217 int t;
218 if (!PyArg_ParseTuple(args, "i:alarm", &t))
219 return NULL;
220 /* alarm() returns the number of seconds remaining */
221 return PyLong_FromLong((long)alarm(t));
Guido van Rossumaa0f4c71994-08-23 13:49:37 +0000222}
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000223
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000224PyDoc_STRVAR(alarm_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000225"alarm(seconds)\n\
226\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000227Arrange for SIGALRM to arrive after the given number of seconds.");
Guido van Rossum06d511d1995-03-10 15:13:48 +0000228#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +0000229
Guido van Rossum1171ee61997-08-22 20:42:00 +0000230#ifdef HAVE_PAUSE
Guido van Rossuma597dde1995-01-10 20:56:29 +0000231static PyObject *
Neal Norwitz3a6f9782002-03-25 20:46:46 +0000232signal_pause(PyObject *self)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000233{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000234 Py_BEGIN_ALLOW_THREADS
235 (void)pause();
236 Py_END_ALLOW_THREADS
237 /* make sure that any exceptions that got raised are propagated
238 * back into Python
239 */
240 if (PyErr_CheckSignals())
241 return NULL;
Barry Warsaw92971171997-01-03 00:14:25 +0000242
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000243 Py_INCREF(Py_None);
244 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +0000245}
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000246PyDoc_STRVAR(pause_doc,
Barry Warsaw1ee36ff1998-07-21 22:41:18 +0000247"pause()\n\
248\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000249Wait until a signal arrives.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000250
Guido van Rossum06d511d1995-03-10 15:13:48 +0000251#endif
Guido van Rossume4485b01994-09-07 14:32:49 +0000252
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000253
Guido van Rossume4485b01994-09-07 14:32:49 +0000254static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000255signal_signal(PyObject *self, PyObject *args)
Guido van Rossume4485b01994-09-07 14:32:49 +0000256{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000257 PyObject *obj;
258 int sig_num;
259 PyObject *old_handler;
260 void (*func)(int);
261 if (!PyArg_ParseTuple(args, "iO:signal", &sig_num, &obj))
262 return NULL;
Brian Curtin3f004b12010-08-06 19:34:52 +0000263#ifdef MS_WINDOWS
264 /* Validate that sig_num is one of the allowable signals */
Brian Curtin912443c2010-09-06 16:10:04 +0000265 switch (sig_num) {
266 case SIGABRT: break;
Brian Curtineccd4d92010-10-01 15:09:53 +0000267#ifdef SIGBREAK
268 /* Issue #10003: SIGBREAK is not documented as permitted, but works
269 and corresponds to CTRL_BREAK_EVENT. */
270 case SIGBREAK: break;
271#endif
Brian Curtin912443c2010-09-06 16:10:04 +0000272 case SIGFPE: break;
273 case SIGILL: break;
274 case SIGINT: break;
275 case SIGSEGV: break;
276 case SIGTERM: break;
277 default:
278 PyErr_SetString(PyExc_ValueError, "invalid signal value");
279 return NULL;
Brian Curtin3f004b12010-08-06 19:34:52 +0000280 }
281#endif
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000282#ifdef WITH_THREAD
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000283 if (PyThread_get_thread_ident() != main_thread) {
284 PyErr_SetString(PyExc_ValueError,
285 "signal only works in main thread");
286 return NULL;
287 }
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000288#endif
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000289 if (sig_num < 1 || sig_num >= NSIG) {
290 PyErr_SetString(PyExc_ValueError,
291 "signal number out of range");
292 return NULL;
293 }
294 if (obj == IgnoreHandler)
295 func = SIG_IGN;
296 else if (obj == DefaultHandler)
297 func = SIG_DFL;
298 else if (!PyCallable_Check(obj)) {
299 PyErr_SetString(PyExc_TypeError,
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000300"signal handler must be signal.SIG_IGN, signal.SIG_DFL, or a callable object");
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000301 return NULL;
302 }
303 else
304 func = signal_handler;
305 if (PyOS_setsig(sig_num, func) == SIG_ERR) {
306 PyErr_SetFromErrno(PyExc_RuntimeError);
307 return NULL;
308 }
309 old_handler = Handlers[sig_num].func;
310 Handlers[sig_num].tripped = 0;
311 Py_INCREF(obj);
312 Handlers[sig_num].func = obj;
313 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000314}
315
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000316PyDoc_STRVAR(signal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000317"signal(sig, action) -> action\n\
318\n\
319Set the action for the given signal. The action can be SIG_DFL,\n\
320SIG_IGN, or a callable Python object. The previous action is\n\
321returned. See getsignal() for possible return values.\n\
322\n\
323*** IMPORTANT NOTICE ***\n\
324A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000325the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000326
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000327
Guido van Rossume4485b01994-09-07 14:32:49 +0000328static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000329signal_getsignal(PyObject *self, PyObject *args)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000330{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000331 int sig_num;
332 PyObject *old_handler;
333 if (!PyArg_ParseTuple(args, "i:getsignal", &sig_num))
334 return NULL;
335 if (sig_num < 1 || sig_num >= NSIG) {
336 PyErr_SetString(PyExc_ValueError,
337 "signal number out of range");
338 return NULL;
339 }
340 old_handler = Handlers[sig_num].func;
341 Py_INCREF(old_handler);
342 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000343}
344
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000345PyDoc_STRVAR(getsignal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000346"getsignal(sig) -> action\n\
347\n\
348Return the current action for the given signal. The return value can be:\n\
349SIG_IGN -- if the signal is being ignored\n\
350SIG_DFL -- if the default action for the signal is in effect\n\
351None -- if an unknown handler is in effect\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000352anything else -- the callable Python object used as a handler");
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000353
Christian Heimes8640e742008-02-23 16:23:06 +0000354#ifdef HAVE_SIGINTERRUPT
355PyDoc_STRVAR(siginterrupt_doc,
356"siginterrupt(sig, flag) -> None\n\
357change system call restart behaviour: if flag is False, system calls\n\
358will be restarted when interrupted by signal sig, else system calls\n\
359will be interrupted.");
360
361static PyObject *
362signal_siginterrupt(PyObject *self, PyObject *args)
363{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000364 int sig_num;
365 int flag;
Christian Heimes8640e742008-02-23 16:23:06 +0000366
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000367 if (!PyArg_ParseTuple(args, "ii:siginterrupt", &sig_num, &flag))
368 return NULL;
369 if (sig_num < 1 || sig_num >= NSIG) {
370 PyErr_SetString(PyExc_ValueError,
371 "signal number out of range");
372 return NULL;
373 }
374 if (siginterrupt(sig_num, flag)<0) {
375 PyErr_SetFromErrno(PyExc_RuntimeError);
376 return NULL;
377 }
Christian Heimes8640e742008-02-23 16:23:06 +0000378
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000379 Py_INCREF(Py_None);
380 return Py_None;
Christian Heimes8640e742008-02-23 16:23:06 +0000381}
382
383#endif
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000384
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000385static PyObject *
386signal_set_wakeup_fd(PyObject *self, PyObject *args)
387{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000388 struct stat buf;
389 int fd, old_fd;
390 if (!PyArg_ParseTuple(args, "i:set_wakeup_fd", &fd))
391 return NULL;
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000392#ifdef WITH_THREAD
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000393 if (PyThread_get_thread_ident() != main_thread) {
394 PyErr_SetString(PyExc_ValueError,
395 "set_wakeup_fd only works in main thread");
396 return NULL;
397 }
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000398#endif
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000399 if (fd != -1 && fstat(fd, &buf) != 0) {
400 PyErr_SetString(PyExc_ValueError, "invalid fd");
401 return NULL;
402 }
403 old_fd = wakeup_fd;
404 wakeup_fd = fd;
405 return PyLong_FromLong(old_fd);
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000406}
407
408PyDoc_STRVAR(set_wakeup_fd_doc,
409"set_wakeup_fd(fd) -> fd\n\
410\n\
411Sets the fd to be written to (with '\\0') when a signal\n\
412comes in. A library can use this to wakeup select or poll.\n\
413The previous fd is returned.\n\
414\n\
415The fd must be non-blocking.");
416
417/* C API for the same, without all the error checking */
418int
419PySignal_SetWakeupFd(int fd)
420{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000421 int old_fd = wakeup_fd;
422 if (fd < 0)
423 fd = -1;
424 wakeup_fd = fd;
425 return old_fd;
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000426}
427
428
Martin v. Löwis823725e2008-03-24 13:39:54 +0000429#ifdef HAVE_SETITIMER
430static PyObject *
431signal_setitimer(PyObject *self, PyObject *args)
432{
433 double first;
434 double interval = 0;
435 int which;
436 struct itimerval new, old;
437
438 if(!PyArg_ParseTuple(args, "id|d:setitimer", &which, &first, &interval))
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000439 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000440
441 timeval_from_double(first, &new.it_value);
442 timeval_from_double(interval, &new.it_interval);
443 /* Let OS check "which" value */
444 if (setitimer(which, &new, &old) != 0) {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000445 PyErr_SetFromErrno(ItimerError);
446 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000447 }
448
449 return itimer_retval(&old);
450}
451
452PyDoc_STRVAR(setitimer_doc,
453"setitimer(which, seconds[, interval])\n\
454\n\
455Sets given itimer (one of ITIMER_REAL, ITIMER_VIRTUAL\n\
456or ITIMER_PROF) to fire after value seconds and after\n\
457that every interval seconds.\n\
458The itimer can be cleared by setting seconds to zero.\n\
459\n\
460Returns old values as a tuple: (delay, interval).");
461#endif
462
463
464#ifdef HAVE_GETITIMER
465static PyObject *
466signal_getitimer(PyObject *self, PyObject *args)
467{
468 int which;
469 struct itimerval old;
470
471 if (!PyArg_ParseTuple(args, "i:getitimer", &which))
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000472 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000473
474 if (getitimer(which, &old) != 0) {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000475 PyErr_SetFromErrno(ItimerError);
476 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000477 }
478
479 return itimer_retval(&old);
480}
481
482PyDoc_STRVAR(getitimer_doc,
483"getitimer(which)\n\
484\n\
485Returns current value of given itimer.");
486#endif
487
488
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000489/* List of functions defined in the module */
Barry Warsaw92971171997-01-03 00:14:25 +0000490static PyMethodDef signal_methods[] = {
Guido van Rossum1171ee61997-08-22 20:42:00 +0000491#ifdef HAVE_ALARM
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000492 {"alarm", signal_alarm, METH_VARARGS, alarm_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000493#endif
Martin v. Löwis823725e2008-03-24 13:39:54 +0000494#ifdef HAVE_SETITIMER
495 {"setitimer", signal_setitimer, METH_VARARGS, setitimer_doc},
496#endif
497#ifdef HAVE_GETITIMER
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000498 {"getitimer", signal_getitimer, METH_VARARGS, getitimer_doc},
Martin v. Löwis823725e2008-03-24 13:39:54 +0000499#endif
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000500 {"signal", signal_signal, METH_VARARGS, signal_doc},
501 {"getsignal", signal_getsignal, METH_VARARGS, getsignal_doc},
502 {"set_wakeup_fd", signal_set_wakeup_fd, METH_VARARGS, set_wakeup_fd_doc},
Christian Heimes8640e742008-02-23 16:23:06 +0000503#ifdef HAVE_SIGINTERRUPT
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000504 {"siginterrupt", signal_siginterrupt, METH_VARARGS, siginterrupt_doc},
Christian Heimes8640e742008-02-23 16:23:06 +0000505#endif
Guido van Rossum1171ee61997-08-22 20:42:00 +0000506#ifdef HAVE_PAUSE
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000507 {"pause", (PyCFunction)signal_pause,
508 METH_NOARGS,pause_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000509#endif
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000510 {"default_int_handler", signal_default_int_handler,
511 METH_VARARGS, default_int_handler_doc},
512 {NULL, NULL} /* sentinel */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000513};
514
Barry Warsaw92971171997-01-03 00:14:25 +0000515
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000516PyDoc_STRVAR(module_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000517"This module provides mechanisms to use signal handlers in Python.\n\
518\n\
519Functions:\n\
520\n\
521alarm() -- cause SIGALRM after a specified time [Unix only]\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000522setitimer() -- cause a signal (described below) after a specified\n\
523 float time and the timer may restart then [Unix only]\n\
524getitimer() -- get current value of timer [Unix only]\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000525signal() -- set the action for a given signal\n\
526getsignal() -- get the signal action for a given signal\n\
527pause() -- wait until a signal arrives [Unix only]\n\
528default_int_handler() -- default SIGINT handler\n\
529\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000530signal constants:\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000531SIG_DFL -- used to refer to the system default handler\n\
532SIG_IGN -- used to ignore the signal\n\
533NSIG -- number of defined signals\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000534SIGINT, SIGTERM, etc. -- signal numbers\n\
535\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000536itimer constants:\n\
537ITIMER_REAL -- decrements in real time, and delivers SIGALRM upon\n\
538 expiration\n\
539ITIMER_VIRTUAL -- decrements only when the process is executing,\n\
540 and delivers SIGVTALRM upon expiration\n\
541ITIMER_PROF -- decrements both when the process is executing and\n\
542 when the system is executing on behalf of the process.\n\
543 Coupled with ITIMER_VIRTUAL, this timer is usually\n\
544 used to profile the time spent by the application\n\
545 in user and kernel space. SIGPROF is delivered upon\n\
546 expiration.\n\
547\n\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000548*** IMPORTANT NOTICE ***\n\
549A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000550the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000551
Martin v. Löwis1a214512008-06-11 05:26:20 +0000552static struct PyModuleDef signalmodule = {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000553 PyModuleDef_HEAD_INIT,
554 "signal",
555 module_doc,
556 -1,
557 signal_methods,
558 NULL,
559 NULL,
560 NULL,
561 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +0000562};
563
Mark Hammondfe51c6d2002-08-02 02:27:13 +0000564PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +0000565PyInit_signal(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000566{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000567 PyObject *m, *d, *x;
568 int i;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000569
570#ifdef WITH_THREAD
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000571 main_thread = PyThread_get_thread_ident();
572 main_pid = getpid();
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000573#endif
574
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000575 /* Create the module and add the functions */
576 m = PyModule_Create(&signalmodule);
577 if (m == NULL)
578 return NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000579
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000580 /* Add some symbolic constants to the module */
581 d = PyModule_GetDict(m);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000582
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000583 x = DefaultHandler = PyLong_FromVoidPtr((void *)SIG_DFL);
584 if (!x || PyDict_SetItemString(d, "SIG_DFL", x) < 0)
585 goto finally;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000586
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000587 x = IgnoreHandler = PyLong_FromVoidPtr((void *)SIG_IGN);
588 if (!x || PyDict_SetItemString(d, "SIG_IGN", x) < 0)
589 goto finally;
Barry Warsaw92971171997-01-03 00:14:25 +0000590
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000591 x = PyLong_FromLong((long)NSIG);
592 if (!x || PyDict_SetItemString(d, "NSIG", x) < 0)
593 goto finally;
594 Py_DECREF(x);
Barry Warsaw92971171997-01-03 00:14:25 +0000595
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000596 x = IntHandler = PyDict_GetItemString(d, "default_int_handler");
597 if (!x)
598 goto finally;
599 Py_INCREF(IntHandler);
Barry Warsaw92971171997-01-03 00:14:25 +0000600
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000601 Handlers[0].tripped = 0;
602 for (i = 1; i < NSIG; i++) {
603 void (*t)(int);
604 t = PyOS_getsig(i);
605 Handlers[i].tripped = 0;
606 if (t == SIG_DFL)
607 Handlers[i].func = DefaultHandler;
608 else if (t == SIG_IGN)
609 Handlers[i].func = IgnoreHandler;
610 else
611 Handlers[i].func = Py_None; /* None of our business */
612 Py_INCREF(Handlers[i].func);
613 }
614 if (Handlers[SIGINT].func == DefaultHandler) {
615 /* Install default int handler */
616 Py_INCREF(IntHandler);
617 Py_DECREF(Handlers[SIGINT].func);
618 Handlers[SIGINT].func = IntHandler;
619 old_siginthandler = PyOS_setsig(SIGINT, signal_handler);
620 }
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000621
622#ifdef SIGHUP
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000623 x = PyLong_FromLong(SIGHUP);
624 PyDict_SetItemString(d, "SIGHUP", x);
625 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000626#endif
627#ifdef SIGINT
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000628 x = PyLong_FromLong(SIGINT);
629 PyDict_SetItemString(d, "SIGINT", x);
630 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000631#endif
Tim Peters1ce3cf72001-10-01 17:58:40 +0000632#ifdef SIGBREAK
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000633 x = PyLong_FromLong(SIGBREAK);
634 PyDict_SetItemString(d, "SIGBREAK", x);
635 Py_XDECREF(x);
Tim Peters1ce3cf72001-10-01 17:58:40 +0000636#endif
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000637#ifdef SIGQUIT
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000638 x = PyLong_FromLong(SIGQUIT);
639 PyDict_SetItemString(d, "SIGQUIT", x);
640 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000641#endif
642#ifdef SIGILL
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000643 x = PyLong_FromLong(SIGILL);
644 PyDict_SetItemString(d, "SIGILL", x);
645 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000646#endif
647#ifdef SIGTRAP
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000648 x = PyLong_FromLong(SIGTRAP);
649 PyDict_SetItemString(d, "SIGTRAP", x);
650 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000651#endif
652#ifdef SIGIOT
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000653 x = PyLong_FromLong(SIGIOT);
654 PyDict_SetItemString(d, "SIGIOT", x);
655 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000656#endif
657#ifdef SIGABRT
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000658 x = PyLong_FromLong(SIGABRT);
659 PyDict_SetItemString(d, "SIGABRT", x);
660 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000661#endif
662#ifdef SIGEMT
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000663 x = PyLong_FromLong(SIGEMT);
664 PyDict_SetItemString(d, "SIGEMT", x);
665 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000666#endif
667#ifdef SIGFPE
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000668 x = PyLong_FromLong(SIGFPE);
669 PyDict_SetItemString(d, "SIGFPE", x);
670 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000671#endif
672#ifdef SIGKILL
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000673 x = PyLong_FromLong(SIGKILL);
674 PyDict_SetItemString(d, "SIGKILL", x);
675 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000676#endif
677#ifdef SIGBUS
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000678 x = PyLong_FromLong(SIGBUS);
679 PyDict_SetItemString(d, "SIGBUS", x);
680 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000681#endif
682#ifdef SIGSEGV
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000683 x = PyLong_FromLong(SIGSEGV);
684 PyDict_SetItemString(d, "SIGSEGV", x);
685 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000686#endif
687#ifdef SIGSYS
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000688 x = PyLong_FromLong(SIGSYS);
689 PyDict_SetItemString(d, "SIGSYS", x);
690 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000691#endif
692#ifdef SIGPIPE
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000693 x = PyLong_FromLong(SIGPIPE);
694 PyDict_SetItemString(d, "SIGPIPE", x);
695 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000696#endif
697#ifdef SIGALRM
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000698 x = PyLong_FromLong(SIGALRM);
699 PyDict_SetItemString(d, "SIGALRM", x);
700 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000701#endif
702#ifdef SIGTERM
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000703 x = PyLong_FromLong(SIGTERM);
704 PyDict_SetItemString(d, "SIGTERM", x);
705 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000706#endif
707#ifdef SIGUSR1
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000708 x = PyLong_FromLong(SIGUSR1);
709 PyDict_SetItemString(d, "SIGUSR1", x);
710 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000711#endif
712#ifdef SIGUSR2
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000713 x = PyLong_FromLong(SIGUSR2);
714 PyDict_SetItemString(d, "SIGUSR2", x);
715 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000716#endif
717#ifdef SIGCLD
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000718 x = PyLong_FromLong(SIGCLD);
719 PyDict_SetItemString(d, "SIGCLD", x);
720 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000721#endif
722#ifdef SIGCHLD
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000723 x = PyLong_FromLong(SIGCHLD);
724 PyDict_SetItemString(d, "SIGCHLD", x);
725 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000726#endif
727#ifdef SIGPWR
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000728 x = PyLong_FromLong(SIGPWR);
729 PyDict_SetItemString(d, "SIGPWR", x);
730 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000731#endif
732#ifdef SIGIO
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000733 x = PyLong_FromLong(SIGIO);
734 PyDict_SetItemString(d, "SIGIO", x);
735 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000736#endif
737#ifdef SIGURG
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000738 x = PyLong_FromLong(SIGURG);
739 PyDict_SetItemString(d, "SIGURG", x);
740 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000741#endif
742#ifdef SIGWINCH
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000743 x = PyLong_FromLong(SIGWINCH);
744 PyDict_SetItemString(d, "SIGWINCH", x);
745 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000746#endif
747#ifdef SIGPOLL
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000748 x = PyLong_FromLong(SIGPOLL);
749 PyDict_SetItemString(d, "SIGPOLL", x);
750 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000751#endif
752#ifdef SIGSTOP
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000753 x = PyLong_FromLong(SIGSTOP);
754 PyDict_SetItemString(d, "SIGSTOP", x);
755 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000756#endif
757#ifdef SIGTSTP
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000758 x = PyLong_FromLong(SIGTSTP);
759 PyDict_SetItemString(d, "SIGTSTP", x);
760 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000761#endif
762#ifdef SIGCONT
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000763 x = PyLong_FromLong(SIGCONT);
764 PyDict_SetItemString(d, "SIGCONT", x);
765 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000766#endif
767#ifdef SIGTTIN
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000768 x = PyLong_FromLong(SIGTTIN);
769 PyDict_SetItemString(d, "SIGTTIN", x);
770 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000771#endif
772#ifdef SIGTTOU
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000773 x = PyLong_FromLong(SIGTTOU);
774 PyDict_SetItemString(d, "SIGTTOU", x);
775 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000776#endif
777#ifdef SIGVTALRM
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000778 x = PyLong_FromLong(SIGVTALRM);
779 PyDict_SetItemString(d, "SIGVTALRM", x);
780 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000781#endif
782#ifdef SIGPROF
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000783 x = PyLong_FromLong(SIGPROF);
784 PyDict_SetItemString(d, "SIGPROF", x);
785 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000786#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000787#ifdef SIGXCPU
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000788 x = PyLong_FromLong(SIGXCPU);
789 PyDict_SetItemString(d, "SIGXCPU", x);
790 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000791#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000792#ifdef SIGXFSZ
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000793 x = PyLong_FromLong(SIGXFSZ);
794 PyDict_SetItemString(d, "SIGXFSZ", x);
795 Py_XDECREF(x);
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000796#endif
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000797#ifdef SIGRTMIN
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000798 x = PyLong_FromLong(SIGRTMIN);
799 PyDict_SetItemString(d, "SIGRTMIN", x);
800 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000801#endif
802#ifdef SIGRTMAX
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000803 x = PyLong_FromLong(SIGRTMAX);
804 PyDict_SetItemString(d, "SIGRTMAX", x);
805 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000806#endif
Martin v. Löwis175af252002-01-12 11:43:25 +0000807#ifdef SIGINFO
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000808 x = PyLong_FromLong(SIGINFO);
809 PyDict_SetItemString(d, "SIGINFO", x);
810 Py_XDECREF(x);
Martin v. Löwis175af252002-01-12 11:43:25 +0000811#endif
Martin v. Löwis823725e2008-03-24 13:39:54 +0000812
813#ifdef ITIMER_REAL
814 x = PyLong_FromLong(ITIMER_REAL);
815 PyDict_SetItemString(d, "ITIMER_REAL", x);
816 Py_DECREF(x);
817#endif
818#ifdef ITIMER_VIRTUAL
819 x = PyLong_FromLong(ITIMER_VIRTUAL);
820 PyDict_SetItemString(d, "ITIMER_VIRTUAL", x);
821 Py_DECREF(x);
822#endif
823#ifdef ITIMER_PROF
824 x = PyLong_FromLong(ITIMER_PROF);
825 PyDict_SetItemString(d, "ITIMER_PROF", x);
826 Py_DECREF(x);
827#endif
828
829#if defined (HAVE_SETITIMER) || defined (HAVE_GETITIMER)
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000830 ItimerError = PyErr_NewException("signal.ItimerError",
831 PyExc_IOError, NULL);
Neal Norwitz2f99b242008-08-24 05:48:10 +0000832 if (ItimerError != NULL)
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000833 PyDict_SetItemString(d, "ItimerError", ItimerError);
Martin v. Löwis823725e2008-03-24 13:39:54 +0000834#endif
835
Martin v. Löwis1a214512008-06-11 05:26:20 +0000836 if (PyErr_Occurred()) {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000837 Py_DECREF(m);
838 m = NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +0000839 }
Barry Warsaw92971171997-01-03 00:14:25 +0000840
Barry Warsaw92971171997-01-03 00:14:25 +0000841 finally:
Martin v. Löwis1a214512008-06-11 05:26:20 +0000842 return m;
Guido van Rossum08c16611997-08-02 03:01:42 +0000843}
844
845static void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000846finisignal(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000847{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000848 int i;
849 PyObject *func;
Guido van Rossum08c16611997-08-02 03:01:42 +0000850
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000851 PyOS_setsig(SIGINT, old_siginthandler);
852 old_siginthandler = SIG_DFL;
Guido van Rossum08c16611997-08-02 03:01:42 +0000853
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000854 for (i = 1; i < NSIG; i++) {
855 func = Handlers[i].func;
856 Handlers[i].tripped = 0;
857 Handlers[i].func = NULL;
858 if (i != SIGINT && func != NULL && func != Py_None &&
859 func != DefaultHandler && func != IgnoreHandler)
860 PyOS_setsig(i, SIG_DFL);
861 Py_XDECREF(func);
862 }
Guido van Rossum08c16611997-08-02 03:01:42 +0000863
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000864 Py_XDECREF(IntHandler);
865 IntHandler = NULL;
866 Py_XDECREF(DefaultHandler);
867 DefaultHandler = NULL;
868 Py_XDECREF(IgnoreHandler);
869 IgnoreHandler = NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000870}
871
Barry Warsaw92971171997-01-03 00:14:25 +0000872
Barry Warsaw92971171997-01-03 00:14:25 +0000873/* Declared in pyerrors.h */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000874int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000875PyErr_CheckSignals(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000876{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000877 int i;
878 PyObject *f;
Barry Warsaw92971171997-01-03 00:14:25 +0000879
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000880 if (!is_tripped)
881 return 0;
Christian Heimesb76922a2007-12-11 01:06:40 +0000882
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000883#ifdef WITH_THREAD
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000884 if (PyThread_get_thread_ident() != main_thread)
885 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000886#endif
Christian Heimesb76922a2007-12-11 01:06:40 +0000887
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000888 /*
889 * The is_tripped variable is meant to speed up the calls to
890 * PyErr_CheckSignals (both directly or via pending calls) when no
891 * signal has arrived. This variable is set to 1 when a signal arrives
892 * and it is set to 0 here, when we know some signals arrived. This way
893 * we can run the registered handlers with no signals blocked.
894 *
895 * NOTE: with this approach we can have a situation where is_tripped is
896 * 1 but we have no more signals to handle (Handlers[i].tripped
897 * is 0 for every signal i). This won't do us any harm (except
898 * we're gonna spent some cycles for nothing). This happens when
899 * we receive a signal i after we zero is_tripped and before we
900 * check Handlers[i].tripped.
901 */
902 is_tripped = 0;
Christian Heimesb76922a2007-12-11 01:06:40 +0000903
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000904 if (!(f = (PyObject *)PyEval_GetFrame()))
905 f = Py_None;
Christian Heimesb76922a2007-12-11 01:06:40 +0000906
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000907 for (i = 1; i < NSIG; i++) {
908 if (Handlers[i].tripped) {
909 PyObject *result = NULL;
910 PyObject *arglist = Py_BuildValue("(iO)", i, f);
911 Handlers[i].tripped = 0;
Barry Warsaw92971171997-01-03 00:14:25 +0000912
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000913 if (arglist) {
914 result = PyEval_CallObject(Handlers[i].func,
915 arglist);
916 Py_DECREF(arglist);
917 }
918 if (!result)
919 return -1;
Barry Warsaw92971171997-01-03 00:14:25 +0000920
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000921 Py_DECREF(result);
922 }
923 }
Christian Heimesb76922a2007-12-11 01:06:40 +0000924
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000925 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000926}
927
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000928
Barry Warsaw92971171997-01-03 00:14:25 +0000929/* Replacements for intrcheck.c functionality
930 * Declared in pyerrors.h
931 */
932void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000933PyErr_SetInterrupt(void)
Barry Warsaw92971171997-01-03 00:14:25 +0000934{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000935 is_tripped = 1;
936 Handlers[SIGINT].tripped = 1;
937 Py_AddPendingCall((int (*)(void *))PyErr_CheckSignals, NULL);
Barry Warsaw92971171997-01-03 00:14:25 +0000938}
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000939
940void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000941PyOS_InitInterrupts(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000942{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000943 PyObject *m = PyInit_signal();
944 if (m) {
945 _PyImport_FixupExtension(m, "signal", "signal");
946 Py_DECREF(m);
947 }
Guido van Rossum08c16611997-08-02 03:01:42 +0000948}
949
950void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000951PyOS_FiniInterrupts(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000952{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000953 finisignal();
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000954}
955
956int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000957PyOS_InterruptOccurred(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000958{
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000959 if (Handlers[SIGINT].tripped) {
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000960#ifdef WITH_THREAD
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000961 if (PyThread_get_thread_ident() != main_thread)
962 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000963#endif
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000964 Handlers[SIGINT].tripped = 0;
965 return 1;
966 }
967 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000968}
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000969
970void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000971PyOS_AfterFork(void)
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000972{
973#ifdef WITH_THREAD
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000974 PyEval_ReInitThreads();
975 main_thread = PyThread_get_thread_ident();
976 main_pid = getpid();
977 _PyImport_ReInitLock();
978 PyThread_ReInitTLS();
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000979#endif
980}