1 #include <u.h>
2 #include <signal.h>
3 #define NOPLAN9DEFINES
4 #include <libc.h>
5 #include "9proc.h"
6 
7 extern char *_p9sigstr(int, char*);
8 
9 static struct {
10 	int sig;
11 	int restart;
12 } sigs[] = {
13 	SIGHUP, 0,
14 	SIGINT, 0,
15 	SIGQUIT, 0,
16 	SIGILL, 0,
17 	SIGTRAP, 0,
18 /*	SIGABRT,	*/
19 #ifdef SIGEMT
20 	SIGEMT, 0,
21 #endif
22 	SIGFPE, 0,
23 	SIGBUS, 0,
24 /*	SIGSEGV,	*/
25 	SIGCHLD, 1,
26 	SIGSYS, 0,
27 	SIGPIPE, 0,
28 	SIGALRM, 0,
29 	SIGTERM, 0,
30 	SIGTSTP, 1,
31 	SIGTTIN, 1,
32 	SIGTTOU, 1,
33 	SIGXCPU, 0,
34 	SIGXFSZ, 0,
35 	SIGVTALRM, 0,
36 	SIGUSR1, 0,
37 	SIGUSR2, 0,
38 #ifdef SIGINFO
39 	SIGINFO, 0,
40 #endif
41 };
42 
43 static void (*notifyf)(void*, char*);
44 
45 static void
46 notifysigf(int sig)
47 {
48 	int v;
49 	char tmp[64];
50 	Uproc *up;
51 
52 	up = _p9uproc(1);
53 	v = p9setjmp(up->notejb);
54 	if(v == 0 && notifyf)
55 		(*notifyf)(nil, _p9sigstr(sig, tmp));
56 	else if(v == 2){
57 if(0)print("HANDLED %d\n", sig);
58 		return;
59 	}
60 if(0)print("DEFAULT %d\n", sig);
61 	signal(sig, SIG_DFL);
62 	kill(getpid(), sig);
63 }
64 	
65 int
66 notify(void (*f)(void*, char*))
67 {
68 	int i;
69 	struct sigaction sa, osa;
70 
71 	_p9uproc(0);
72 	memset(&sa, 0, sizeof sa);
73 	if(f == 0)
74 		sa.sa_handler = SIG_DFL;
75 	else{
76 		notifyf = f;
77 		sa.sa_handler = notifysigf;
78 	}
79 	for(i=0; i<nelem(sigs); i++){
80 		/*
81 		 * If someone has already installed a handler,
82 		 * It's probably some ld preload nonsense,
83 		 * like pct (a SIGVTALRM-based profiler).
84 		 * Leave it alone.
85 		 */
86 		sigaction(sigs[i].sig, nil, &osa);
87 		if(osa.sa_handler != SIG_DFL)
88 			continue;
89 		sigemptyset(&sa.sa_mask);
90 		sigaddset(&sa.sa_mask, i);
91 		if(sigs[i].restart)
92 			sa.sa_flags |= SA_RESTART;
93 		else
94 			sa.sa_flags &= ~SA_RESTART;
95 		sigaction(sigs[i].sig, &sa, nil);
96 	}
97 	return 0;
98 }
99 
100 int
101 noted(int v)
102 {
103 	Uproc *up;
104 
105 	up = _p9uproc(1);
106 	p9longjmp(up->notejb, v==NCONT ? 2 : 1);
107 	abort();
108 	return 0;
109 }