libbb/signals.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Utility routines.
   4  *
   5  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
   6  * Copyright (C) 2006 Rob Landley
   7  * Copyright (C) 2006 Denys Vlasenko
   8  *
   9  * Licensed under GPL version 2, see file LICENSE in this tarball for details.
  10  */
  11
  12 #include "libbb.h"
  13
  14 /* All known arches use small ints for signals */
  15 smallint bb_got_signal;
  16
  17 void record_signo(int signo)
  18 {
  19         bb_got_signal = signo;
  20 }
  21
  22 /* Saves 2 bytes on x86! Oh my... */
  23 int FAST_FUNC sigaction_set(int signum, const struct sigaction *act)
  24 {
  25         return sigaction(signum, act, NULL);
  26 }
  27
  28 int FAST_FUNC sigprocmask_allsigs(int how)
  29 {
  30         sigset_t set;
  31         sigfillset(&set);
  32         return sigprocmask(how, &set, NULL);
  33 }
  34
  35 void FAST_FUNC bb_signals(int sigs, void (*f)(int))
  36 {
  37         int sig_no = 0;
  38         int bit = 1;
  39
  40         while (sigs) {
  41                 if (sigs & bit) {
  42                         sigs &= ~bit;
  43                         signal(sig_no, f);
  44                 }
  45                 sig_no++;
  46                 bit <<= 1;
  47         }
  48 }
  49
  50 void FAST_FUNC bb_signals_recursive(int sigs, void (*f)(int))
  51 {
  52         int sig_no = 0;
  53         int bit = 1;
  54         struct sigaction sa;
  55
  56         memset(&sa, 0, sizeof(sa));
  57         sa.sa_handler = f;
  58         /*sa.sa_flags = 0;*/
  59         /*sigemptyset(&sa.sa_mask); - hope memset did it*/
  60
  61         while (sigs) {
  62                 if (sigs & bit) {
  63                         sigs &= ~bit;
  64                         sigaction_set(sig_no, &sa);
  65                 }
  66                 sig_no++;
  67                 bit <<= 1;
  68         }
  69 }
  70
  71 void FAST_FUNC sig_block(int sig)
  72 {
  73         sigset_t ss;
  74         sigemptyset(&ss);
  75         sigaddset(&ss, sig);
  76         sigprocmask(SIG_BLOCK, &ss, NULL);
  77 }
  78
  79 void FAST_FUNC sig_unblock(int sig)
  80 {
  81         sigset_t ss;
  82         sigemptyset(&ss);
  83         sigaddset(&ss, sig);
  84         sigprocmask(SIG_UNBLOCK, &ss, NULL);
  85 }
  86
  87 void FAST_FUNC wait_for_any_sig(void)
  88 {
  89         sigset_t ss;
  90         sigemptyset(&ss);
  91         sigsuspend(&ss);
  92 }
  93
  94 /* Assuming the sig is fatal */
  95 void FAST_FUNC kill_myself_with_sig(int sig)
  96 {
  97         signal(sig, SIG_DFL);
  98         sig_unblock(sig);
  99         raise(sig);
 100         _exit(EXIT_FAILURE); /* Should not reach it */
 101 }
 102
 103 void FAST_FUNC signal_SA_RESTART_empty_mask(int sig, void (*handler)(int))
 104 {
 105         struct sigaction sa;
 106         memset(&sa, 0, sizeof(sa));
 107         /*sigemptyset(&sa.sa_mask);*/
 108         sa.sa_flags = SA_RESTART;
 109         sa.sa_handler = handler;
 110         sigaction_set(sig, &sa);
 111 }
 112
 113 void FAST_FUNC signal_no_SA_RESTART_empty_mask(int sig, void (*handler)(int))
 114 {
 115         struct sigaction sa;
 116         memset(&sa, 0, sizeof(sa));
 117         /*sigemptyset(&sa.sa_mask);*/
 118         /*sa.sa_flags = 0;*/
 119         sa.sa_handler = handler;
 120         sigaction_set(sig, &sa);
 121 }