1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <errno.h>
#include <sys/socket.h>
#include <sys/un.h>
#define MCONSOLE_MAGIC (0xcafebabe)
#define MCONSOLE_MAX_DATA (512)
#define MCONSOLE_VERSION 2
struct mconsole_notify {
unsigned long magic;
int version;
enum { MCONSOLE_SOCKET, MCONSOLE_PANIC, MCONSOLE_HANG } type;
int len;
char data[MCONSOLE_MAX_DATA];
};
static char *sock_name = NULL;
static void notify(int sig)
{
struct sockaddr_un target;
struct mconsole_notify packet;
int sock, len, n;
sock = socket(PF_UNIX, SOCK_DGRAM, 0);
if(sock < 0){
perror("socket");
exit(1);
}
target.sun_family = AF_UNIX;
strcpy(target.sun_path, sock_name);
packet.magic = MCONSOLE_MAGIC;
packet.version = MCONSOLE_VERSION;
packet.type = MCONSOLE_HANG;
packet.len = 0;
len = sizeof(packet) + packet.len - sizeof(packet.data);
n = sendto(sock, &packet, len, 0, (struct sockaddr *) &target,
sizeof(target));
if(n < 0)
perror("sendto");
exit(1);
}
static int tracing_pid = -1;
static void kill_tracer(int sig)
{
kill(tracing_pid, SIGINT);
sleep(1);
kill(tracing_pid, SIGINT);
sleep(1);
kill(tracing_pid, SIGKILL);
exit(1);
}
int main(int argc, char **argv)
{
void (*handler)(int);
int n;
char c, *end;
if(argc < 3) exit(1);
if(!strcmp(argv[1], "-pid")){
tracing_pid = strtol(argv[2], &end, 0);
if(*end != '\0') exit(1);
if(kill(tracing_pid, 0)) exit(1);
handler = kill_tracer;
}
else if(!strcmp(argv[1], "-mconsole")){
sock_name = argv[2];
handler = notify;
}
else exit(1);
if(signal(SIGALRM, handler)) exit(1);
if(write(1, &c, sizeof(c)) != sizeof(c)) exit(1);
while(1){
alarm(60);
n = read(0, &c, sizeof(c));
if(n == 0) exit(0);
else if(n < 0) exit(1);
}
}
|
