zend_signal.c 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414
  1. /*
  2. +----------------------------------------------------------------------+
  3. | Zend Signal Handling |
  4. +----------------------------------------------------------------------+
  5. | Copyright (c) 2008 The PHP Group |
  6. +----------------------------------------------------------------------+
  7. | This source file is subject to version 3.01 of the PHP license, |
  8. | that is bundled with this package in the file LICENSE, and is |
  9. | available through the world-wide-web at the following url: |
  10. | http://www.php.net/license/3_01.txt |
  11. | If you did not receive a copy of the PHP license and are unable to |
  12. | obtain it through the world-wide-web, please send a note to |
  13. | license@php.net so we can mail you a copy immediately. |
  14. +----------------------------------------------------------------------+
  15. | Authors: Lucas Nealan <lucas@php.net> |
  16. | Arnaud Le Blanc <lbarnaud@php.net> |
  17. +----------------------------------------------------------------------+
  18. This software was contributed to PHP by Facebook Inc. in 2008.
  19. Future revisions and derivatives of this source code must acknowledge
  20. Facebook Inc. as the original contributor of this module by leaving
  21. this note intact in the source code.
  22. All other licensing and usage conditions are those of the PHP Group.
  23. */
  24. /* $Id$ */
  25. #define _GNU_SOURCE
  26. #include <string.h>
  27. #include "zend.h"
  28. #include "zend_globals.h"
  29. #ifdef HAVE_SIGNAL_H
  30. #include <signal.h>
  31. #endif
  32. #ifdef HAVE_UNISTD_H
  33. #include <unistd.h>
  34. #endif
  35. #ifdef ZEND_SIGNALS
  36. #include "zend_signal.h"
  37. #ifdef ZTS
  38. ZEND_API int zend_signal_globals_id;
  39. #else
  40. zend_signal_globals_t zend_signal_globals;
  41. #endif
  42. static void zend_signal_handler(int signo, siginfo_t *siginfo, void *context TSRMLS_DC);
  43. static int zend_signal_register(int signo, void (*handler)(int, siginfo_t*, void*) TSRMLS_DC);
  44. #ifdef __CYGWIN__
  45. #define TIMEOUT_SIG SIGALRM
  46. #else
  47. #define TIMEOUT_SIG SIGPROF
  48. #endif
  49. static int zend_sigs[] = { TIMEOUT_SIG, SIGHUP, SIGINT, SIGQUIT, SIGTERM, SIGUSR1, SIGUSR2 };
  50. #define SA_FLAGS_MASK ~(SA_NODEFER | SA_RESETHAND)
  51. /* True globals, written only at process startup */
  52. static zend_signal_entry_t global_orig_handlers[NSIG];
  53. static sigset_t global_sigmask;
  54. /* {{{ zend_signal_handler_defer
  55. * Blocks signals if in critical section */
  56. void zend_signal_handler_defer(int signo, siginfo_t *siginfo, void *context)
  57. {
  58. int errno_save = errno;
  59. zend_signal_queue_t *queue, *qtmp;
  60. TSRMLS_FETCH();
  61. if (SIGG(active)) {
  62. if (SIGG(depth) == 0) { /* try to handle signal */
  63. if (SIGG(blocked) != -1) { /* inverse */
  64. SIGG(blocked) = -1; /* signal is not blocked */
  65. }
  66. if (SIGG(running) == 0) {
  67. SIGG(running) = 1;
  68. zend_signal_handler(signo, siginfo, context TSRMLS_CC);
  69. queue = SIGG(phead);
  70. SIGG(phead) = NULL;
  71. while (queue) {
  72. zend_signal_handler(queue->zend_signal.signo, queue->zend_signal.siginfo, queue->zend_signal.context TSRMLS_CC);
  73. qtmp = queue->next;
  74. queue->next = SIGG(pavail);
  75. queue->zend_signal.signo = 0;
  76. SIGG(pavail) = queue;
  77. queue = qtmp;
  78. }
  79. SIGG(running) = 0;
  80. }
  81. } else { /* delay signal handling */
  82. SIGG(blocked) = 0; /* signal is blocked */
  83. if ((queue = SIGG(pavail))) { /* if none available it's simply forgotton */
  84. SIGG(pavail) = queue->next;
  85. queue->zend_signal.signo = signo;
  86. queue->zend_signal.siginfo = siginfo;
  87. queue->zend_signal.context = context;
  88. queue->next = NULL;
  89. if (SIGG(phead) && SIGG(ptail)) {
  90. SIGG(ptail)->next = queue;
  91. } else {
  92. SIGG(phead) = queue;
  93. }
  94. SIGG(ptail) = queue;
  95. }
  96. #if ZEND_DEBUG
  97. else { /* this may not be safe to do, but could work and be useful */
  98. zend_output_debug_string(0, "zend_signal: not enough queue storage, lost signal (%d)", signo);
  99. }
  100. #endif
  101. }
  102. } else {
  103. /* need to just run handler if we're inactive and getting a signal */
  104. zend_signal_handler(signo, siginfo, context TSRMLS_CC);
  105. }
  106. errno = errno_save;
  107. } /* }}} */
  108. /* {{{ zend_signal_handler_unblock
  109. * Handle deferred signal from HANDLE_UNBLOCK_ALARMS */
  110. ZEND_API void zend_signal_handler_unblock(TSRMLS_D)
  111. {
  112. zend_signal_queue_t *queue;
  113. zend_signal_t zend_signal;
  114. if (SIGG(active)) {
  115. SIGNAL_BEGIN_CRITICAL(); /* procmask to protect handler_defer as if it were called by the kernel */
  116. queue = SIGG(phead);
  117. SIGG(phead) = queue->next;
  118. zend_signal = queue->zend_signal;
  119. queue->next = SIGG(pavail);
  120. queue->zend_signal.signo = 0;
  121. SIGG(pavail) = queue;
  122. zend_signal_handler_defer(zend_signal.signo, zend_signal.siginfo, zend_signal.context);
  123. SIGNAL_END_CRITICAL();
  124. }
  125. }
  126. /* }}} */
  127. /* {{{ zend_signal_handler
  128. * Call the previously registered handler for a signal
  129. */
  130. static void zend_signal_handler(int signo, siginfo_t *siginfo, void *context TSRMLS_DC)
  131. {
  132. int errno_save = errno;
  133. struct sigaction sa = {{0}};
  134. sigset_t sigset;
  135. zend_signal_entry_t p_sig = SIGG(handlers)[signo-1];
  136. if (p_sig.handler == SIG_DFL) { /* raise default handler */
  137. if (sigaction(signo, NULL, &sa) == 0) {
  138. sa.sa_handler = SIG_DFL;
  139. sigemptyset(&sa.sa_mask);
  140. sigemptyset(&sigset);
  141. sigaddset(&sigset, signo);
  142. if (sigaction(signo, &sa, NULL) == 0) {
  143. /* throw away any blocked signals */
  144. sigprocmask(SIG_UNBLOCK, &sigset, NULL);
  145. raise(signo);
  146. }
  147. }
  148. } else if (p_sig.handler != SIG_IGN) { /* ignore SIG_IGN */
  149. if (p_sig.flags & SA_SIGINFO) {
  150. if (p_sig.flags & SA_RESETHAND) {
  151. SIGG(handlers)[signo-1].flags = 0;
  152. SIGG(handlers)[signo-1].handler = SIG_DFL;
  153. }
  154. (*(void (*)(int, siginfo_t*, void*))p_sig.handler)(signo, siginfo, context);
  155. } else {
  156. (*(void (*)(int))p_sig.handler)(signo);
  157. }
  158. }
  159. errno = errno_save;
  160. } /* }}} */
  161. /* {{{ zend_sigaction
  162. * Register a signal handler that will be deferred in critical sections */
  163. ZEND_API int zend_sigaction(int signo, const struct sigaction *act, struct sigaction *oldact TSRMLS_DC)
  164. {
  165. struct sigaction sa = {{0}};
  166. sigset_t sigset;
  167. if (oldact != NULL) {
  168. oldact->sa_flags = SIGG(handlers)[signo-1].flags;
  169. oldact->sa_handler = (void *) SIGG(handlers)[signo-1].handler;
  170. oldact->sa_mask = global_sigmask;
  171. }
  172. if (act != NULL) {
  173. SIGG(handlers)[signo-1].flags = act->sa_flags;
  174. if (act->sa_flags & SA_SIGINFO) {
  175. SIGG(handlers)[signo-1].handler = (void *) act->sa_sigaction;
  176. } else {
  177. SIGG(handlers)[signo-1].handler = (void *) act->sa_handler;
  178. }
  179. sa.sa_flags = SA_SIGINFO | (act->sa_flags & SA_FLAGS_MASK);
  180. sa.sa_sigaction = zend_signal_handler_defer;
  181. sa.sa_mask = global_sigmask;
  182. if (sigaction(signo, &sa, NULL) < 0) {
  183. zend_error(E_ERROR, "Error installing signal handler for %d", signo);
  184. }
  185. /* unsure this signal is not blocked */
  186. sigemptyset(&sigset);
  187. sigaddset(&sigset, signo);
  188. zend_sigprocmask(SIG_UNBLOCK, &sigset, NULL);
  189. }
  190. return SUCCESS;
  191. }
  192. /* }}} */
  193. /* {{{ zend_signal
  194. * Register a signal handler that will be deferred in critical sections */
  195. ZEND_API int zend_signal(int signo, void (*handler)(int) TSRMLS_DC)
  196. {
  197. struct sigaction sa = {{0}};
  198. sa.sa_flags = 0;
  199. sa.sa_handler = handler;
  200. sa.sa_mask = global_sigmask;
  201. return zend_sigaction(signo, &sa, NULL TSRMLS_CC);
  202. }
  203. /* }}} */
  204. /* {{{ zend_signal_register
  205. * Set a handler for a signal we want to defer.
  206. * Previously set handler must have been saved before.
  207. */
  208. static int zend_signal_register(int signo, void (*handler)(int, siginfo_t*, void*) TSRMLS_DC)
  209. {
  210. struct sigaction sa = {{0}};
  211. if (sigaction(signo, NULL, &sa) == 0) {
  212. if ((sa.sa_flags & SA_SIGINFO) && sa.sa_sigaction == handler) {
  213. return FAILURE;
  214. }
  215. SIGG(handlers)[signo-1].flags = sa.sa_flags;
  216. if (sa.sa_flags & SA_SIGINFO) {
  217. SIGG(handlers)[signo-1].handler = (void *)sa.sa_sigaction;
  218. } else {
  219. SIGG(handlers)[signo-1].handler = (void *)sa.sa_handler;
  220. }
  221. sa.sa_flags = SA_SIGINFO; /* we'll use a siginfo handler */
  222. sa.sa_sigaction = handler;
  223. sa.sa_mask = global_sigmask;
  224. if (sigaction(signo, &sa, NULL) < 0) {
  225. zend_error(E_ERROR, "Error installing signal handler for %d", signo);
  226. }
  227. return SUCCESS;
  228. }
  229. return FAILURE;
  230. } /* }}} */
  231. /* {{{ zend_signal_activate
  232. * Install our signal handlers, per request */
  233. void zend_signal_activate(TSRMLS_D)
  234. {
  235. int x;
  236. memcpy(&SIGG(handlers), &global_orig_handlers, sizeof(global_orig_handlers));
  237. for (x=0; x < sizeof(zend_sigs) / sizeof(*zend_sigs); x++) {
  238. zend_signal_register(zend_sigs[x], zend_signal_handler_defer TSRMLS_CC);
  239. }
  240. SIGG(active) = 1;
  241. SIGG(depth) = 0;
  242. } /* }}} */
  243. /* {{{ zend_signal_deactivate
  244. * */
  245. void zend_signal_deactivate(TSRMLS_D)
  246. {
  247. int x;
  248. struct sigaction sa = {{0}};
  249. if (SIGG(check)) {
  250. if (SIGG(depth) != 0) {
  251. zend_error(E_CORE_WARNING, "zend_signal: shutdown with non-zero blocking depth (%d)", SIGG(depth));
  252. }
  253. /* did anyone steal our installed handler */
  254. for (x=0; x < sizeof(zend_sigs) / sizeof(*zend_sigs); x++) {
  255. sigaction(zend_sigs[x], NULL, &sa);
  256. if (sa.sa_sigaction != zend_signal_handler_defer) {
  257. zend_error(E_CORE_WARNING, "zend_signal: handler was replaced for signal (%d) after startup", zend_sigs[x]);
  258. }
  259. }
  260. }
  261. SIGNAL_BEGIN_CRITICAL();
  262. SIGG(active) = 0;
  263. SIGG(running) = 0;
  264. SIGG(blocked) = -1;
  265. SIGG(depth) = 0;
  266. SIGNAL_END_CRITICAL();
  267. }
  268. /* }}} */
  269. static void zend_signal_globals_ctor(zend_signal_globals_t *zend_signal_globals TSRMLS_DC)
  270. {
  271. size_t x;
  272. memset(zend_signal_globals, 0, sizeof(*zend_signal_globals));
  273. zend_signal_globals->blocked = -1;
  274. for (x = 0; x < sizeof(zend_signal_globals->pstorage) / sizeof(*zend_signal_globals->pstorage); ++x) {
  275. zend_signal_queue_t *queue = &zend_signal_globals->pstorage[x];
  276. queue->zend_signal.signo = 0;
  277. queue->next = zend_signal_globals->pavail;
  278. zend_signal_globals->pavail = queue;
  279. }
  280. }
  281. static void zend_signal_globals_dtor(zend_signal_globals_t *zend_signal_globals TSRMLS_DC)
  282. {
  283. zend_signal_globals->blocked = -1;
  284. }
  285. /* {{{ zend_signal_startup
  286. * alloc zend signal globals */
  287. void zend_signal_startup()
  288. {
  289. int signo;
  290. struct sigaction sa = {{0}};
  291. #ifdef ZTS
  292. ts_allocate_id(&zend_signal_globals_id, sizeof(zend_signal_globals_t), (ts_allocate_ctor) zend_signal_globals_ctor, (ts_allocate_dtor) zend_signal_globals_dtor);
  293. #else
  294. zend_signal_globals_ctor(&zend_signal_globals);
  295. #endif
  296. /* Used to block signals during execution of signal handlers */
  297. sigfillset(&global_sigmask);
  298. sigdelset(&global_sigmask, SIGILL);
  299. sigdelset(&global_sigmask, SIGABRT);
  300. sigdelset(&global_sigmask, SIGFPE);
  301. sigdelset(&global_sigmask, SIGKILL);
  302. sigdelset(&global_sigmask, SIGSEGV);
  303. sigdelset(&global_sigmask, SIGCONT);
  304. sigdelset(&global_sigmask, SIGSTOP);
  305. sigdelset(&global_sigmask, SIGTSTP);
  306. sigdelset(&global_sigmask, SIGTTIN);
  307. sigdelset(&global_sigmask, SIGTTOU);
  308. #ifdef SIGBUS
  309. sigdelset(&global_sigmask, SIGBUS);
  310. #endif
  311. #ifdef SIGSYS
  312. sigdelset(&global_sigmask, SIGSYS);
  313. #endif
  314. #ifdef SIGTRAP
  315. sigdelset(&global_sigmask, SIGTRAP);
  316. #endif
  317. /* Save previously registered signal handlers into orig_handlers */
  318. memset(&global_orig_handlers, 0, sizeof(global_orig_handlers));
  319. for (signo = 1; signo < NSIG; ++signo) {
  320. if (sigaction(signo, NULL, &sa) == 0) {
  321. global_orig_handlers[signo-1].flags = sa.sa_flags;
  322. if (sa.sa_flags & SA_SIGINFO) {
  323. global_orig_handlers[signo-1].handler = (void *) sa.sa_sigaction;
  324. } else {
  325. global_orig_handlers[signo-1].handler = (void *) sa.sa_handler;
  326. }
  327. }
  328. }
  329. }
  330. /* }}} */
  331. /* {{{ zend_signal_shutdown
  332. * called by zend_shutdown */
  333. void zend_signal_shutdown(TSRMLS_D)
  334. {
  335. #ifndef ZTS
  336. zend_signal_globals_dtor(&zend_signal_globals);
  337. #endif
  338. }
  339. /* }}} */
  340. #endif /* ZEND_SIGNALS */
  341. /*
  342. * Local variables:
  343. * tab-width: 4
  344. * c-basic-offset: 4
  345. * indent-tabs-mode: t
  346. * End:
  347. * vim600: fdm=marker
  348. * vim: noet sw=4 ts=4
  349. */