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 GPLv2, see file LICENSE in this source tree.
10  */
11 
12 #include "libbb.h"
13 
14 /* All known arches use small ints for signals */
15 smallint bb_got_signal;
16 
record_signo(int signo)17 void record_signo(int signo)
18 {
19 	bb_got_signal = signo;
20 }
21 
22 /* Saves 2 bytes on x86! Oh my... */
sigaction_set(int signum,const struct sigaction * act)23 int FAST_FUNC sigaction_set(int signum, const struct sigaction *act)
24 {
25 	return sigaction(signum, act, NULL);
26 }
27 
sigprocmask_allsigs(int how)28 int FAST_FUNC sigprocmask_allsigs(int how)
29 {
30 	sigset_t set;
31 	sigfillset(&set);
32 	return sigprocmask(how, &set, NULL);
33 }
34 
bb_signals(int sigs,void (* f)(int))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 
bb_signals_recursive_norestart(int sigs,void (* f)(int))50 void FAST_FUNC bb_signals_recursive_norestart(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 
sig_block(int sig)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 
sig_unblock(int sig)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 
wait_for_any_sig(void)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 */
kill_myself_with_sig(int sig)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(sig | 128); /* Should not reach it */
101 }
102 
signal_SA_RESTART_empty_mask(int sig,void (* handler)(int))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 
signal_no_SA_RESTART_empty_mask(int sig,void (* handler)(int))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 }
122