xref: /freebsd/usr.bin/procstat/procstat_sigs.c (revision 3157ba21)
1 /*-
2  * Copyright (c) 2010 Konstantin Belousov
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $FreeBSD$
27  */
28 
29 #include <sys/param.h>
30 #include <sys/sysctl.h>
31 #include <sys/user.h>
32 
33 #include <ctype.h>
34 #include <err.h>
35 #include <errno.h>
36 #include <signal.h>
37 #include <stdio.h>
38 #include <stdlib.h>
39 #include <string.h>
40 
41 #include "procstat.h"
42 
43 static void
44 procstat_print_signame(int sig)
45 {
46 	char name[12];
47 	int i;
48 
49 	if (!nflag && sig < sys_nsig) {
50 		strlcpy(name, sys_signame[sig], sizeof(name));
51 		for (i = 0; name[i] != 0; i++)
52 			name[i] = toupper(name[i]);
53 		printf("%-7s ", name);
54 	} else
55 		printf("%-7d ", sig);
56 }
57 
58 static void
59 procstat_print_sig(const sigset_t *set, int sig, char flag)
60 {
61 
62 	printf("%c", sigismember(set, sig) ? flag : '-');
63 }
64 
65 void
66 procstat_sigs(pid_t pid, struct kinfo_proc *kipp)
67 {
68 	int j;
69 
70 	if (!hflag)
71 		printf("%5s %-16s %-7s %4s\n", "PID", "COMM", "SIG", "FLAGS");
72 
73 	for (j = 1; j <= _SIG_MAXSIG; j++) {
74 		printf("%5d ", pid);
75 		printf("%-16s ", kipp->ki_comm);
76 		procstat_print_signame(j);
77 		printf(" ");
78 		procstat_print_sig(&kipp->ki_siglist, j, 'P');
79 		procstat_print_sig(&kipp->ki_sigignore, j, 'I');
80 		procstat_print_sig(&kipp->ki_sigcatch, j, 'C');
81 		printf("\n");
82 	}
83 }
84 
85 void
86 procstat_threads_sigs(pid_t pid, struct kinfo_proc *kipp)
87 {
88 	struct kinfo_proc *kip;
89 	int error, name[4], j;
90 	unsigned int i;
91 	size_t len;
92 
93 	if (!hflag)
94 		printf("%5s %6s %-16s %-7s %4s\n", "PID", "TID", "COMM",
95 		     "SIG", "FLAGS");
96 
97 	/*
98 	 * We need to re-query for thread information, so don't use *kipp.
99 	 */
100 	name[0] = CTL_KERN;
101 	name[1] = KERN_PROC;
102 	name[2] = KERN_PROC_PID | KERN_PROC_INC_THREAD;
103 	name[3] = pid;
104 
105 	len = 0;
106 	error = sysctl(name, 4, NULL, &len, NULL, 0);
107 	if (error < 0 && errno != ESRCH) {
108 		warn("sysctl: kern.proc.pid: %d", pid);
109 		return;
110 	}
111 	if (error < 0)
112 		return;
113 
114 	kip = malloc(len);
115 	if (kip == NULL)
116 		err(-1, "malloc");
117 
118 	if (sysctl(name, 4, kip, &len, NULL, 0) < 0) {
119 		warn("sysctl: kern.proc.pid: %d", pid);
120 		free(kip);
121 		return;
122 	}
123 
124 	kinfo_proc_sort(kip, len / sizeof(*kipp));
125 	for (i = 0; i < len / sizeof(*kipp); i++) {
126 		kipp = &kip[i];
127 		for (j = 1; j <= _SIG_MAXSIG; j++) {
128 			printf("%5d ", pid);
129 			printf("%6d ", kipp->ki_tid);
130 			printf("%-16s ", kipp->ki_comm);
131 			procstat_print_signame(j);
132 			printf(" ");
133 			procstat_print_sig(&kipp->ki_siglist, j, 'P');
134 			procstat_print_sig(&kipp->ki_sigmask, j, 'B');
135 			printf("\n");
136 		}
137 	}
138 	free(kip);
139 }
140