blob: 0e7fd65b04b9863fd872a8277a75711f49f45a84 [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{
Antoine Pitrou39a65912010-11-05 19:47:27 +0000172 int save_errno = errno;
173
174#if defined(WITH_THREAD) && defined(WITH_PTH)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000175 if (PyThread_get_thread_ident() != main_thread) {
176 pth_raise(*(pth_t *) main_thread, sig_num);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000177 }
Antoine Pitrou39a65912010-11-05 19:47:27 +0000178 else
Guido van Rossum9e8181b2000-09-19 00:46:46 +0000179#endif
Antoine Pitrou39a65912010-11-05 19:47:27 +0000180 {
181#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000182 /* See NOTES section above */
Antoine Pitrou39a65912010-11-05 19:47:27 +0000183 if (getpid() == main_pid)
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000184#endif
Antoine Pitrou39a65912010-11-05 19:47:27 +0000185 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000186 Handlers[sig_num].tripped = 1;
187 /* Set is_tripped after setting .tripped, as it gets
188 cleared in PyErr_CheckSignals() before .tripped. */
189 is_tripped = 1;
190 Py_AddPendingCall(checksignals_witharg, NULL);
191 if (wakeup_fd != -1)
192 write(wakeup_fd, "\0", 1);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000193 }
Antoine Pitrou39a65912010-11-05 19:47:27 +0000194
Jean-Paul Calderone6f137ca2010-05-09 03:18:57 +0000195#ifndef HAVE_SIGACTION
Antoine Pitrou39a65912010-11-05 19:47:27 +0000196#ifdef SIGCHLD
197 /* To avoid infinite recursion, this signal remains
198 reset until explicit re-instated.
199 Don't clear the 'func' field as it is our pointer
200 to the Python handler... */
201 if (sig_num != SIGCHLD)
202#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000203 /* If the handler was not set up with sigaction, reinstall it. See
204 * Python/pythonrun.c for the implementation of PyOS_setsig which
205 * makes this true. See also issue8354. */
206 PyOS_setsig(sig_num, signal_handler);
Jean-Paul Calderone6f137ca2010-05-09 03:18:57 +0000207#endif
Antoine Pitrou39a65912010-11-05 19:47:27 +0000208 }
209
210 /* Issue #10311: asynchronously executing signal handlers should not
211 mutate errno under the feet of unsuspecting C code. */
212 errno = save_errno;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000213}
Guido van Rossume4485b01994-09-07 14:32:49 +0000214
Guido van Rossum06d511d1995-03-10 15:13:48 +0000215
Guido van Rossum1171ee61997-08-22 20:42:00 +0000216#ifdef HAVE_ALARM
Guido van Rossume4485b01994-09-07 14:32:49 +0000217static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000218signal_alarm(PyObject *self, PyObject *args)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000219{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000220 int t;
221 if (!PyArg_ParseTuple(args, "i:alarm", &t))
222 return NULL;
223 /* alarm() returns the number of seconds remaining */
224 return PyLong_FromLong((long)alarm(t));
Guido van Rossumaa0f4c71994-08-23 13:49:37 +0000225}
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000226
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000227PyDoc_STRVAR(alarm_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000228"alarm(seconds)\n\
229\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000230Arrange for SIGALRM to arrive after the given number of seconds.");
Guido van Rossum06d511d1995-03-10 15:13:48 +0000231#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +0000232
Guido van Rossum1171ee61997-08-22 20:42:00 +0000233#ifdef HAVE_PAUSE
Guido van Rossuma597dde1995-01-10 20:56:29 +0000234static PyObject *
Neal Norwitz3a6f9782002-03-25 20:46:46 +0000235signal_pause(PyObject *self)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000236{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000237 Py_BEGIN_ALLOW_THREADS
238 (void)pause();
239 Py_END_ALLOW_THREADS
240 /* make sure that any exceptions that got raised are propagated
241 * back into Python
242 */
243 if (PyErr_CheckSignals())
244 return NULL;
Barry Warsaw92971171997-01-03 00:14:25 +0000245
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000246 Py_INCREF(Py_None);
247 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +0000248}
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000249PyDoc_STRVAR(pause_doc,
Barry Warsaw1ee36ff1998-07-21 22:41:18 +0000250"pause()\n\
251\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000252Wait until a signal arrives.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000253
Guido van Rossum06d511d1995-03-10 15:13:48 +0000254#endif
Guido van Rossume4485b01994-09-07 14:32:49 +0000255
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000256
Guido van Rossume4485b01994-09-07 14:32:49 +0000257static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000258signal_signal(PyObject *self, PyObject *args)
Guido van Rossume4485b01994-09-07 14:32:49 +0000259{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000260 PyObject *obj;
261 int sig_num;
262 PyObject *old_handler;
263 void (*func)(int);
264 if (!PyArg_ParseTuple(args, "iO:signal", &sig_num, &obj))
265 return NULL;
Brian Curtinef9efbd2010-08-06 19:27:32 +0000266#ifdef MS_WINDOWS
267 /* Validate that sig_num is one of the allowable signals */
Brian Curtinc734b312010-09-06 16:04:10 +0000268 switch (sig_num) {
269 case SIGABRT: break;
Brian Curtin9e88b5a2010-10-01 14:49:24 +0000270#ifdef SIGBREAK
271 /* Issue #10003: SIGBREAK is not documented as permitted, but works
272 and corresponds to CTRL_BREAK_EVENT. */
273 case SIGBREAK: break;
274#endif
Brian Curtinc734b312010-09-06 16:04:10 +0000275 case SIGFPE: break;
276 case SIGILL: break;
277 case SIGINT: break;
278 case SIGSEGV: break;
279 case SIGTERM: break;
280 default:
281 PyErr_SetString(PyExc_ValueError, "invalid signal value");
282 return NULL;
Brian Curtinef9efbd2010-08-06 19:27:32 +0000283 }
284#endif
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000285#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000286 if (PyThread_get_thread_ident() != main_thread) {
287 PyErr_SetString(PyExc_ValueError,
288 "signal only works in main thread");
289 return NULL;
290 }
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000291#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000292 if (sig_num < 1 || sig_num >= NSIG) {
293 PyErr_SetString(PyExc_ValueError,
294 "signal number out of range");
295 return NULL;
296 }
297 if (obj == IgnoreHandler)
298 func = SIG_IGN;
299 else if (obj == DefaultHandler)
300 func = SIG_DFL;
301 else if (!PyCallable_Check(obj)) {
302 PyErr_SetString(PyExc_TypeError,
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000303"signal handler must be signal.SIG_IGN, signal.SIG_DFL, or a callable object");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000304 return NULL;
305 }
306 else
307 func = signal_handler;
308 if (PyOS_setsig(sig_num, func) == SIG_ERR) {
309 PyErr_SetFromErrno(PyExc_RuntimeError);
310 return NULL;
311 }
312 old_handler = Handlers[sig_num].func;
313 Handlers[sig_num].tripped = 0;
314 Py_INCREF(obj);
315 Handlers[sig_num].func = obj;
316 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000317}
318
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000319PyDoc_STRVAR(signal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000320"signal(sig, action) -> action\n\
321\n\
322Set the action for the given signal. The action can be SIG_DFL,\n\
323SIG_IGN, or a callable Python object. The previous action is\n\
324returned. See getsignal() for possible return values.\n\
325\n\
326*** IMPORTANT NOTICE ***\n\
327A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000328the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000329
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000330
Guido van Rossume4485b01994-09-07 14:32:49 +0000331static PyObject *
Peter Schneider-Kampe89b1562000-07-10 12:04:18 +0000332signal_getsignal(PyObject *self, PyObject *args)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000333{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000334 int sig_num;
335 PyObject *old_handler;
336 if (!PyArg_ParseTuple(args, "i:getsignal", &sig_num))
337 return NULL;
338 if (sig_num < 1 || sig_num >= NSIG) {
339 PyErr_SetString(PyExc_ValueError,
340 "signal number out of range");
341 return NULL;
342 }
343 old_handler = Handlers[sig_num].func;
344 Py_INCREF(old_handler);
345 return old_handler;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000346}
347
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000348PyDoc_STRVAR(getsignal_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000349"getsignal(sig) -> action\n\
350\n\
351Return the current action for the given signal. The return value can be:\n\
352SIG_IGN -- if the signal is being ignored\n\
353SIG_DFL -- if the default action for the signal is in effect\n\
354None -- if an unknown handler is in effect\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000355anything else -- the callable Python object used as a handler");
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000356
Christian Heimes8640e742008-02-23 16:23:06 +0000357#ifdef HAVE_SIGINTERRUPT
358PyDoc_STRVAR(siginterrupt_doc,
359"siginterrupt(sig, flag) -> None\n\
360change system call restart behaviour: if flag is False, system calls\n\
361will be restarted when interrupted by signal sig, else system calls\n\
362will be interrupted.");
363
364static PyObject *
365signal_siginterrupt(PyObject *self, PyObject *args)
366{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000367 int sig_num;
368 int flag;
Christian Heimes8640e742008-02-23 16:23:06 +0000369
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000370 if (!PyArg_ParseTuple(args, "ii:siginterrupt", &sig_num, &flag))
371 return NULL;
372 if (sig_num < 1 || sig_num >= NSIG) {
373 PyErr_SetString(PyExc_ValueError,
374 "signal number out of range");
375 return NULL;
376 }
377 if (siginterrupt(sig_num, flag)<0) {
378 PyErr_SetFromErrno(PyExc_RuntimeError);
379 return NULL;
380 }
Christian Heimes8640e742008-02-23 16:23:06 +0000381
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000382 Py_INCREF(Py_None);
383 return Py_None;
Christian Heimes8640e742008-02-23 16:23:06 +0000384}
385
386#endif
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000387
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000388static PyObject *
389signal_set_wakeup_fd(PyObject *self, PyObject *args)
390{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000391 struct stat buf;
392 int fd, old_fd;
393 if (!PyArg_ParseTuple(args, "i:set_wakeup_fd", &fd))
394 return NULL;
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000395#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000396 if (PyThread_get_thread_ident() != main_thread) {
397 PyErr_SetString(PyExc_ValueError,
398 "set_wakeup_fd only works in main thread");
399 return NULL;
400 }
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000401#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000402 if (fd != -1 && fstat(fd, &buf) != 0) {
403 PyErr_SetString(PyExc_ValueError, "invalid fd");
404 return NULL;
405 }
406 old_fd = wakeup_fd;
407 wakeup_fd = fd;
408 return PyLong_FromLong(old_fd);
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000409}
410
411PyDoc_STRVAR(set_wakeup_fd_doc,
412"set_wakeup_fd(fd) -> fd\n\
413\n\
414Sets the fd to be written to (with '\\0') when a signal\n\
415comes in. A library can use this to wakeup select or poll.\n\
416The previous fd is returned.\n\
417\n\
418The fd must be non-blocking.");
419
420/* C API for the same, without all the error checking */
421int
422PySignal_SetWakeupFd(int fd)
423{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000424 int old_fd = wakeup_fd;
425 if (fd < 0)
426 fd = -1;
427 wakeup_fd = fd;
428 return old_fd;
Christian Heimes5fb7c2a2007-12-24 08:52:31 +0000429}
430
431
Martin v. Löwis823725e2008-03-24 13:39:54 +0000432#ifdef HAVE_SETITIMER
433static PyObject *
434signal_setitimer(PyObject *self, PyObject *args)
435{
436 double first;
437 double interval = 0;
438 int which;
439 struct itimerval new, old;
440
441 if(!PyArg_ParseTuple(args, "id|d:setitimer", &which, &first, &interval))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000442 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000443
444 timeval_from_double(first, &new.it_value);
445 timeval_from_double(interval, &new.it_interval);
446 /* Let OS check "which" value */
447 if (setitimer(which, &new, &old) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000448 PyErr_SetFromErrno(ItimerError);
449 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000450 }
451
452 return itimer_retval(&old);
453}
454
455PyDoc_STRVAR(setitimer_doc,
456"setitimer(which, seconds[, interval])\n\
457\n\
458Sets given itimer (one of ITIMER_REAL, ITIMER_VIRTUAL\n\
459or ITIMER_PROF) to fire after value seconds and after\n\
460that every interval seconds.\n\
461The itimer can be cleared by setting seconds to zero.\n\
462\n\
463Returns old values as a tuple: (delay, interval).");
464#endif
465
466
467#ifdef HAVE_GETITIMER
468static PyObject *
469signal_getitimer(PyObject *self, PyObject *args)
470{
471 int which;
472 struct itimerval old;
473
474 if (!PyArg_ParseTuple(args, "i:getitimer", &which))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000475 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000476
477 if (getitimer(which, &old) != 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000478 PyErr_SetFromErrno(ItimerError);
479 return NULL;
Martin v. Löwis823725e2008-03-24 13:39:54 +0000480 }
481
482 return itimer_retval(&old);
483}
484
485PyDoc_STRVAR(getitimer_doc,
486"getitimer(which)\n\
487\n\
488Returns current value of given itimer.");
489#endif
490
491
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000492/* List of functions defined in the module */
Barry Warsaw92971171997-01-03 00:14:25 +0000493static PyMethodDef signal_methods[] = {
Guido van Rossum1171ee61997-08-22 20:42:00 +0000494#ifdef HAVE_ALARM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000495 {"alarm", signal_alarm, METH_VARARGS, alarm_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000496#endif
Martin v. Löwis823725e2008-03-24 13:39:54 +0000497#ifdef HAVE_SETITIMER
498 {"setitimer", signal_setitimer, METH_VARARGS, setitimer_doc},
499#endif
500#ifdef HAVE_GETITIMER
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000501 {"getitimer", signal_getitimer, METH_VARARGS, getitimer_doc},
Martin v. Löwis823725e2008-03-24 13:39:54 +0000502#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000503 {"signal", signal_signal, METH_VARARGS, signal_doc},
504 {"getsignal", signal_getsignal, METH_VARARGS, getsignal_doc},
505 {"set_wakeup_fd", signal_set_wakeup_fd, METH_VARARGS, set_wakeup_fd_doc},
Christian Heimes8640e742008-02-23 16:23:06 +0000506#ifdef HAVE_SIGINTERRUPT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000507 {"siginterrupt", signal_siginterrupt, METH_VARARGS, siginterrupt_doc},
Christian Heimes8640e742008-02-23 16:23:06 +0000508#endif
Guido van Rossum1171ee61997-08-22 20:42:00 +0000509#ifdef HAVE_PAUSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000510 {"pause", (PyCFunction)signal_pause,
511 METH_NOARGS,pause_doc},
Guido van Rossum06d511d1995-03-10 15:13:48 +0000512#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000513 {"default_int_handler", signal_default_int_handler,
514 METH_VARARGS, default_int_handler_doc},
515 {NULL, NULL} /* sentinel */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000516};
517
Barry Warsaw92971171997-01-03 00:14:25 +0000518
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000519PyDoc_STRVAR(module_doc,
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000520"This module provides mechanisms to use signal handlers in Python.\n\
521\n\
522Functions:\n\
523\n\
524alarm() -- cause SIGALRM after a specified time [Unix only]\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000525setitimer() -- cause a signal (described below) after a specified\n\
526 float time and the timer may restart then [Unix only]\n\
527getitimer() -- get current value of timer [Unix only]\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000528signal() -- set the action for a given signal\n\
529getsignal() -- get the signal action for a given signal\n\
530pause() -- wait until a signal arrives [Unix only]\n\
531default_int_handler() -- default SIGINT handler\n\
532\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000533signal constants:\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000534SIG_DFL -- used to refer to the system default handler\n\
535SIG_IGN -- used to ignore the signal\n\
536NSIG -- number of defined signals\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000537SIGINT, SIGTERM, etc. -- signal numbers\n\
538\n\
Martin v. Löwis823725e2008-03-24 13:39:54 +0000539itimer constants:\n\
540ITIMER_REAL -- decrements in real time, and delivers SIGALRM upon\n\
541 expiration\n\
542ITIMER_VIRTUAL -- decrements only when the process is executing,\n\
543 and delivers SIGVTALRM upon expiration\n\
544ITIMER_PROF -- decrements both when the process is executing and\n\
545 when the system is executing on behalf of the process.\n\
546 Coupled with ITIMER_VIRTUAL, this timer is usually\n\
547 used to profile the time spent by the application\n\
548 in user and kernel space. SIGPROF is delivered upon\n\
549 expiration.\n\
550\n\n\
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000551*** IMPORTANT NOTICE ***\n\
552A signal handler function is called with two arguments:\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000553the first is the signal number, the second is the interrupted stack frame.");
Guido van Rossum1d8fb2d1998-06-28 16:54:49 +0000554
Martin v. Löwis1a214512008-06-11 05:26:20 +0000555static struct PyModuleDef signalmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000556 PyModuleDef_HEAD_INIT,
557 "signal",
558 module_doc,
559 -1,
560 signal_methods,
561 NULL,
562 NULL,
563 NULL,
564 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +0000565};
566
Mark Hammondfe51c6d2002-08-02 02:27:13 +0000567PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +0000568PyInit_signal(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000569{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000570 PyObject *m, *d, *x;
571 int i;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000572
573#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000574 main_thread = PyThread_get_thread_ident();
575 main_pid = getpid();
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000576#endif
577
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000578 /* Create the module and add the functions */
579 m = PyModule_Create(&signalmodule);
580 if (m == NULL)
581 return NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000582
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000583 /* Add some symbolic constants to the module */
584 d = PyModule_GetDict(m);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000585
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000586 x = DefaultHandler = PyLong_FromVoidPtr((void *)SIG_DFL);
587 if (!x || PyDict_SetItemString(d, "SIG_DFL", x) < 0)
588 goto finally;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000589
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000590 x = IgnoreHandler = PyLong_FromVoidPtr((void *)SIG_IGN);
591 if (!x || PyDict_SetItemString(d, "SIG_IGN", x) < 0)
592 goto finally;
Barry Warsaw92971171997-01-03 00:14:25 +0000593
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000594 x = PyLong_FromLong((long)NSIG);
595 if (!x || PyDict_SetItemString(d, "NSIG", x) < 0)
596 goto finally;
597 Py_DECREF(x);
Barry Warsaw92971171997-01-03 00:14:25 +0000598
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000599 x = IntHandler = PyDict_GetItemString(d, "default_int_handler");
600 if (!x)
601 goto finally;
602 Py_INCREF(IntHandler);
Barry Warsaw92971171997-01-03 00:14:25 +0000603
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000604 Handlers[0].tripped = 0;
605 for (i = 1; i < NSIG; i++) {
606 void (*t)(int);
607 t = PyOS_getsig(i);
608 Handlers[i].tripped = 0;
609 if (t == SIG_DFL)
610 Handlers[i].func = DefaultHandler;
611 else if (t == SIG_IGN)
612 Handlers[i].func = IgnoreHandler;
613 else
614 Handlers[i].func = Py_None; /* None of our business */
615 Py_INCREF(Handlers[i].func);
616 }
617 if (Handlers[SIGINT].func == DefaultHandler) {
618 /* Install default int handler */
619 Py_INCREF(IntHandler);
620 Py_DECREF(Handlers[SIGINT].func);
621 Handlers[SIGINT].func = IntHandler;
622 old_siginthandler = PyOS_setsig(SIGINT, signal_handler);
623 }
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000624
625#ifdef SIGHUP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000626 x = PyLong_FromLong(SIGHUP);
627 PyDict_SetItemString(d, "SIGHUP", x);
628 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000629#endif
630#ifdef SIGINT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000631 x = PyLong_FromLong(SIGINT);
632 PyDict_SetItemString(d, "SIGINT", x);
633 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000634#endif
Tim Peters1ce3cf72001-10-01 17:58:40 +0000635#ifdef SIGBREAK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000636 x = PyLong_FromLong(SIGBREAK);
637 PyDict_SetItemString(d, "SIGBREAK", x);
638 Py_XDECREF(x);
Tim Peters1ce3cf72001-10-01 17:58:40 +0000639#endif
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000640#ifdef SIGQUIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000641 x = PyLong_FromLong(SIGQUIT);
642 PyDict_SetItemString(d, "SIGQUIT", x);
643 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000644#endif
645#ifdef SIGILL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000646 x = PyLong_FromLong(SIGILL);
647 PyDict_SetItemString(d, "SIGILL", x);
648 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000649#endif
650#ifdef SIGTRAP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000651 x = PyLong_FromLong(SIGTRAP);
652 PyDict_SetItemString(d, "SIGTRAP", x);
653 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000654#endif
655#ifdef SIGIOT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000656 x = PyLong_FromLong(SIGIOT);
657 PyDict_SetItemString(d, "SIGIOT", x);
658 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000659#endif
660#ifdef SIGABRT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000661 x = PyLong_FromLong(SIGABRT);
662 PyDict_SetItemString(d, "SIGABRT", x);
663 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000664#endif
665#ifdef SIGEMT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000666 x = PyLong_FromLong(SIGEMT);
667 PyDict_SetItemString(d, "SIGEMT", x);
668 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000669#endif
670#ifdef SIGFPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000671 x = PyLong_FromLong(SIGFPE);
672 PyDict_SetItemString(d, "SIGFPE", x);
673 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000674#endif
675#ifdef SIGKILL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000676 x = PyLong_FromLong(SIGKILL);
677 PyDict_SetItemString(d, "SIGKILL", x);
678 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000679#endif
680#ifdef SIGBUS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000681 x = PyLong_FromLong(SIGBUS);
682 PyDict_SetItemString(d, "SIGBUS", x);
683 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000684#endif
685#ifdef SIGSEGV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000686 x = PyLong_FromLong(SIGSEGV);
687 PyDict_SetItemString(d, "SIGSEGV", x);
688 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000689#endif
690#ifdef SIGSYS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000691 x = PyLong_FromLong(SIGSYS);
692 PyDict_SetItemString(d, "SIGSYS", x);
693 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000694#endif
695#ifdef SIGPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000696 x = PyLong_FromLong(SIGPIPE);
697 PyDict_SetItemString(d, "SIGPIPE", x);
698 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000699#endif
700#ifdef SIGALRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000701 x = PyLong_FromLong(SIGALRM);
702 PyDict_SetItemString(d, "SIGALRM", x);
703 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000704#endif
705#ifdef SIGTERM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000706 x = PyLong_FromLong(SIGTERM);
707 PyDict_SetItemString(d, "SIGTERM", x);
708 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000709#endif
710#ifdef SIGUSR1
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000711 x = PyLong_FromLong(SIGUSR1);
712 PyDict_SetItemString(d, "SIGUSR1", x);
713 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000714#endif
715#ifdef SIGUSR2
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000716 x = PyLong_FromLong(SIGUSR2);
717 PyDict_SetItemString(d, "SIGUSR2", x);
718 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000719#endif
720#ifdef SIGCLD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000721 x = PyLong_FromLong(SIGCLD);
722 PyDict_SetItemString(d, "SIGCLD", x);
723 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000724#endif
725#ifdef SIGCHLD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000726 x = PyLong_FromLong(SIGCHLD);
727 PyDict_SetItemString(d, "SIGCHLD", x);
728 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000729#endif
730#ifdef SIGPWR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000731 x = PyLong_FromLong(SIGPWR);
732 PyDict_SetItemString(d, "SIGPWR", x);
733 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000734#endif
735#ifdef SIGIO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000736 x = PyLong_FromLong(SIGIO);
737 PyDict_SetItemString(d, "SIGIO", x);
738 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000739#endif
740#ifdef SIGURG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000741 x = PyLong_FromLong(SIGURG);
742 PyDict_SetItemString(d, "SIGURG", x);
743 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000744#endif
745#ifdef SIGWINCH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000746 x = PyLong_FromLong(SIGWINCH);
747 PyDict_SetItemString(d, "SIGWINCH", x);
748 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000749#endif
750#ifdef SIGPOLL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000751 x = PyLong_FromLong(SIGPOLL);
752 PyDict_SetItemString(d, "SIGPOLL", x);
753 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000754#endif
755#ifdef SIGSTOP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000756 x = PyLong_FromLong(SIGSTOP);
757 PyDict_SetItemString(d, "SIGSTOP", x);
758 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000759#endif
760#ifdef SIGTSTP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000761 x = PyLong_FromLong(SIGTSTP);
762 PyDict_SetItemString(d, "SIGTSTP", x);
763 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000764#endif
765#ifdef SIGCONT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000766 x = PyLong_FromLong(SIGCONT);
767 PyDict_SetItemString(d, "SIGCONT", x);
768 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000769#endif
770#ifdef SIGTTIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000771 x = PyLong_FromLong(SIGTTIN);
772 PyDict_SetItemString(d, "SIGTTIN", x);
773 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000774#endif
775#ifdef SIGTTOU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000776 x = PyLong_FromLong(SIGTTOU);
777 PyDict_SetItemString(d, "SIGTTOU", x);
778 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000779#endif
780#ifdef SIGVTALRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000781 x = PyLong_FromLong(SIGVTALRM);
782 PyDict_SetItemString(d, "SIGVTALRM", x);
783 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000784#endif
785#ifdef SIGPROF
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000786 x = PyLong_FromLong(SIGPROF);
787 PyDict_SetItemString(d, "SIGPROF", x);
788 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000789#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000790#ifdef SIGXCPU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000791 x = PyLong_FromLong(SIGXCPU);
792 PyDict_SetItemString(d, "SIGXCPU", x);
793 Py_XDECREF(x);
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000794#endif
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000795#ifdef SIGXFSZ
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000796 x = PyLong_FromLong(SIGXFSZ);
797 PyDict_SetItemString(d, "SIGXFSZ", x);
798 Py_XDECREF(x);
Barry Warsaw14ed5fb1996-12-16 20:24:22 +0000799#endif
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000800#ifdef SIGRTMIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000801 x = PyLong_FromLong(SIGRTMIN);
802 PyDict_SetItemString(d, "SIGRTMIN", x);
803 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000804#endif
805#ifdef SIGRTMAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000806 x = PyLong_FromLong(SIGRTMAX);
807 PyDict_SetItemString(d, "SIGRTMAX", x);
808 Py_XDECREF(x);
Anthony Baxterf37f37d2003-07-31 10:35:29 +0000809#endif
Martin v. Löwis175af252002-01-12 11:43:25 +0000810#ifdef SIGINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000811 x = PyLong_FromLong(SIGINFO);
812 PyDict_SetItemString(d, "SIGINFO", x);
813 Py_XDECREF(x);
Martin v. Löwis175af252002-01-12 11:43:25 +0000814#endif
Martin v. Löwis823725e2008-03-24 13:39:54 +0000815
816#ifdef ITIMER_REAL
817 x = PyLong_FromLong(ITIMER_REAL);
818 PyDict_SetItemString(d, "ITIMER_REAL", x);
819 Py_DECREF(x);
820#endif
821#ifdef ITIMER_VIRTUAL
822 x = PyLong_FromLong(ITIMER_VIRTUAL);
823 PyDict_SetItemString(d, "ITIMER_VIRTUAL", x);
824 Py_DECREF(x);
825#endif
826#ifdef ITIMER_PROF
827 x = PyLong_FromLong(ITIMER_PROF);
828 PyDict_SetItemString(d, "ITIMER_PROF", x);
829 Py_DECREF(x);
830#endif
831
832#if defined (HAVE_SETITIMER) || defined (HAVE_GETITIMER)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000833 ItimerError = PyErr_NewException("signal.ItimerError",
834 PyExc_IOError, NULL);
Neal Norwitz2f99b242008-08-24 05:48:10 +0000835 if (ItimerError != NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000836 PyDict_SetItemString(d, "ItimerError", ItimerError);
Martin v. Löwis823725e2008-03-24 13:39:54 +0000837#endif
838
Brian Curtineb24d742010-04-12 17:16:38 +0000839#ifdef CTRL_C_EVENT
840 x = PyLong_FromLong(CTRL_C_EVENT);
841 PyDict_SetItemString(d, "CTRL_C_EVENT", x);
842 Py_DECREF(x);
843#endif
844
845#ifdef CTRL_BREAK_EVENT
846 x = PyLong_FromLong(CTRL_BREAK_EVENT);
847 PyDict_SetItemString(d, "CTRL_BREAK_EVENT", x);
848 Py_DECREF(x);
849#endif
850
Martin v. Löwis1a214512008-06-11 05:26:20 +0000851 if (PyErr_Occurred()) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000852 Py_DECREF(m);
853 m = NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +0000854 }
Barry Warsaw92971171997-01-03 00:14:25 +0000855
Barry Warsaw92971171997-01-03 00:14:25 +0000856 finally:
Martin v. Löwis1a214512008-06-11 05:26:20 +0000857 return m;
Guido van Rossum08c16611997-08-02 03:01:42 +0000858}
859
860static void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000861finisignal(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000862{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000863 int i;
864 PyObject *func;
Guido van Rossum08c16611997-08-02 03:01:42 +0000865
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000866 PyOS_setsig(SIGINT, old_siginthandler);
867 old_siginthandler = SIG_DFL;
Guido van Rossum08c16611997-08-02 03:01:42 +0000868
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000869 for (i = 1; i < NSIG; i++) {
870 func = Handlers[i].func;
871 Handlers[i].tripped = 0;
872 Handlers[i].func = NULL;
873 if (i != SIGINT && func != NULL && func != Py_None &&
874 func != DefaultHandler && func != IgnoreHandler)
875 PyOS_setsig(i, SIG_DFL);
876 Py_XDECREF(func);
877 }
Guido van Rossum08c16611997-08-02 03:01:42 +0000878
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000879 Py_XDECREF(IntHandler);
880 IntHandler = NULL;
881 Py_XDECREF(DefaultHandler);
882 DefaultHandler = NULL;
883 Py_XDECREF(IgnoreHandler);
884 IgnoreHandler = NULL;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000885}
886
Barry Warsaw92971171997-01-03 00:14:25 +0000887
Barry Warsaw92971171997-01-03 00:14:25 +0000888/* Declared in pyerrors.h */
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000889int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000890PyErr_CheckSignals(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000891{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000892 int i;
893 PyObject *f;
Barry Warsaw92971171997-01-03 00:14:25 +0000894
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000895 if (!is_tripped)
896 return 0;
Christian Heimesb76922a2007-12-11 01:06:40 +0000897
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000898#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000899 if (PyThread_get_thread_ident() != main_thread)
900 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000901#endif
Christian Heimesb76922a2007-12-11 01:06:40 +0000902
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000903 /*
904 * The is_tripped variable is meant to speed up the calls to
905 * PyErr_CheckSignals (both directly or via pending calls) when no
906 * signal has arrived. This variable is set to 1 when a signal arrives
907 * and it is set to 0 here, when we know some signals arrived. This way
908 * we can run the registered handlers with no signals blocked.
909 *
910 * NOTE: with this approach we can have a situation where is_tripped is
911 * 1 but we have no more signals to handle (Handlers[i].tripped
912 * is 0 for every signal i). This won't do us any harm (except
913 * we're gonna spent some cycles for nothing). This happens when
914 * we receive a signal i after we zero is_tripped and before we
915 * check Handlers[i].tripped.
916 */
917 is_tripped = 0;
Christian Heimesb76922a2007-12-11 01:06:40 +0000918
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000919 if (!(f = (PyObject *)PyEval_GetFrame()))
920 f = Py_None;
Christian Heimesb76922a2007-12-11 01:06:40 +0000921
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000922 for (i = 1; i < NSIG; i++) {
923 if (Handlers[i].tripped) {
924 PyObject *result = NULL;
925 PyObject *arglist = Py_BuildValue("(iO)", i, f);
926 Handlers[i].tripped = 0;
Barry Warsaw92971171997-01-03 00:14:25 +0000927
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000928 if (arglist) {
929 result = PyEval_CallObject(Handlers[i].func,
930 arglist);
931 Py_DECREF(arglist);
932 }
933 if (!result)
934 return -1;
Barry Warsaw92971171997-01-03 00:14:25 +0000935
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000936 Py_DECREF(result);
937 }
938 }
Christian Heimesb76922a2007-12-11 01:06:40 +0000939
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000940 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000941}
942
Guido van Rossumd2cd7ad2000-09-16 16:35:28 +0000943
Barry Warsaw92971171997-01-03 00:14:25 +0000944/* Replacements for intrcheck.c functionality
945 * Declared in pyerrors.h
946 */
947void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000948PyErr_SetInterrupt(void)
Barry Warsaw92971171997-01-03 00:14:25 +0000949{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000950 is_tripped = 1;
951 Handlers[SIGINT].tripped = 1;
952 Py_AddPendingCall((int (*)(void *))PyErr_CheckSignals, NULL);
Barry Warsaw92971171997-01-03 00:14:25 +0000953}
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000954
955void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000956PyOS_InitInterrupts(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000957{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000958 PyObject *m = PyInit_signal();
959 if (m) {
Victor Stinner49d3f252010-10-17 01:24:53 +0000960 _PyImport_FixupBuiltin(m, "signal");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000961 Py_DECREF(m);
962 }
Guido van Rossum08c16611997-08-02 03:01:42 +0000963}
964
965void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000966PyOS_FiniInterrupts(void)
Guido van Rossum08c16611997-08-02 03:01:42 +0000967{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000968 finisignal();
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000969}
970
971int
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000972PyOS_InterruptOccurred(void)
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000973{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000974 if (Handlers[SIGINT].tripped) {
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000975#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000976 if (PyThread_get_thread_ident() != main_thread)
977 return 0;
Guido van Rossumbb4ba121994-06-23 11:25:45 +0000978#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000979 Handlers[SIGINT].tripped = 0;
980 return 1;
981 }
982 return 0;
Guido van Rossum398d9fe1994-05-11 08:59:13 +0000983}
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000984
985void
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000986PyOS_AfterFork(void)
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000987{
988#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000989 PyEval_ReInitThreads();
990 main_thread = PyThread_get_thread_ident();
991 main_pid = getpid();
992 _PyImport_ReInitLock();
993 PyThread_ReInitTLS();
Guido van Rossum359bcaa1997-11-14 22:24:28 +0000994#endif
995}