1/*-
2 * Copyright (c) 2007, 2011 Robert N. M. Watson
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: head/usr.bin/procstat/procstat.c 227956 2011-11-24 20:54:06Z trociny $
27 */
28
29#include <sys/param.h>
30#include <sys/sysctl.h>
31#include <sys/user.h>
32
33#include <err.h>
34#include <libprocstat.h>
35#include <stdio.h>
36#include <stdlib.h>
37#include <sysexits.h>
38#include <unistd.h>
39
40#include "procstat.h"
41
42static int aflag, bflag, cflag, eflag, fflag, iflag, jflag, kflag, lflag, sflag;
43static int tflag, vflag, xflag;
44int hflag, nflag, Cflag;
45
46static void
47usage(void)
48{
49
50 fprintf(stderr, "usage: procstat [-h] [-C] [-M core] [-N system] "
51 "[-w interval] \n");
52 fprintf(stderr, " [-b | -c | -e | -f | -i | -j | -k | "
53 "-l | -s | -t | -v | -x] [-a | pid ...]\n");
54 exit(EX_USAGE);
55}
56
57static void
58procstat(struct procstat *prstat, struct kinfo_proc *kipp)
59{
60
61 if (bflag)
62 procstat_bin(kipp);
63 else if (cflag)
64 procstat_args(kipp);
65 else if (eflag)
66 procstat_env(kipp);
67 else if (fflag)
68 procstat_files(prstat, kipp);
69 else if (iflag)
70 procstat_sigs(prstat, kipp);
71 else if (jflag)
72 procstat_threads_sigs(prstat, kipp);
73 else if (kflag)
74 procstat_kstack(kipp, kflag);
75 else if (lflag)
76 procstat_rlimit(kipp);
77 else if (sflag)
78 procstat_cred(kipp);
79 else if (tflag)
80 procstat_threads(kipp);
81 else if (vflag)
82 procstat_vm(kipp);
83 else if (xflag)
84 procstat_auxv(kipp);
85 else
86 procstat_basic(kipp);
87}
88
89/*
90 * Sort processes first by pid and then tid.
91 */
92static int
93kinfo_proc_compare(const void *a, const void *b)
94{
95 int i;
96
97 i = ((const struct kinfo_proc *)a)->ki_pid -
98 ((const struct kinfo_proc *)b)->ki_pid;
99 if (i != 0)
100 return (i);
101 i = ((const struct kinfo_proc *)a)->ki_tid -
102 ((const struct kinfo_proc *)b)->ki_tid;
103 return (i);
104}
105
106void
107kinfo_proc_sort(struct kinfo_proc *kipp, int count)
108{
109
110 qsort(kipp, count, sizeof(*kipp), kinfo_proc_compare);
111}
112
113int
114main(int argc, char *argv[])
115{
116 int ch, interval, tmp;
117 int i;
118 struct kinfo_proc *p;
119 struct procstat *prstat;
120 long l;
121 pid_t pid;
122 char *dummy;
123 char *nlistf, *memf;
124 int cnt;
125
126 interval = 0;
127 memf = nlistf = NULL;
128 while ((ch = getopt(argc, argv, "CN:M:abcefijklhstvw:x")) != -1) {
129 switch (ch) {
130 case 'C':
131 Cflag++;
132 break;
133
134 case 'M':
135 memf = optarg;
136 break;
137 case 'N':
138 nlistf = optarg;
139 break;
140 case 'a':
141 aflag++;
142 break;
143
144 case 'b':
145 bflag++;
146 break;
147
148 case 'c':
149 cflag++;
150 break;
151
152 case 'e':
153 eflag++;
154 break;
155
156 case 'f':
157 fflag++;
158 break;
159
160 case 'i':
161 iflag++;
162 break;
163
164 case 'j':
165 jflag++;
166 break;
167
168 case 'k':
169 kflag++;
170 break;
171
172 case 'l':
173 lflag++;
174 break;
175
176 case 'n':
177 nflag++;
178 break;
179
180 case 'h':
181 hflag++;
182 break;
183
184 case 's':
185 sflag++;
186 break;
187
188 case 't':
189 tflag++;
190 break;
191
192 case 'v':
193 vflag++;
194 break;
195
196 case 'w':
197 l = strtol(optarg, &dummy, 10);
198 if (*dummy != '\0')
199 usage();
200 if (l < 1 || l > INT_MAX)
201 usage();
202 interval = l;
203 break;
204
205 case 'x':
206 xflag++;
207 break;
208
209 case '?':
210 default:
211 usage();
212 }
213
214 }
215 argc -= optind;
216 argv += optind;
217
218 /* We require that either 0 or 1 mode flags be set. */
219 tmp = bflag + cflag + eflag + fflag + (kflag ? 1 : 0) + lflag + sflag +
220 tflag + vflag + xflag;
221 if (!(tmp == 0 || tmp == 1))
222 usage();
223
224 /* We allow -k to be specified up to twice, but not more. */
225 if (kflag > 2)
226 usage();
227
228 /* Must specify either the -a flag or a list of pids. */
229 if (!(aflag == 1 && argc == 0) && !(aflag == 0 && argc > 0))
230 usage();
231
232 /* Only allow -C with -f. */
233 if (Cflag && !fflag)
234 usage();
235
236 if (memf != NULL)
237 prstat = procstat_open_kvm(nlistf, memf);
238 else
239 prstat = procstat_open_sysctl();
240 if (prstat == NULL)
241 errx(1, "procstat_open()");
242 do {
243 if (aflag) {
244 p = procstat_getprocs(prstat, KERN_PROC_PROC, 0, &cnt);
245 if (p == NULL)
246 errx(1, "procstat_getprocs()");
247 kinfo_proc_sort(p, cnt);
248 for (i = 0; i < cnt; i++) {
249 procstat(prstat, &p[i]);
250
251 /* Suppress header after first process. */
252 hflag = 1;
253 }
254 procstat_freeprocs(prstat, p);
255 }
256 for (i = 0; i < argc; i++) {
257 l = strtol(argv[i], &dummy, 10);
258 if (*dummy != '\0')
259 usage();
260 if (l < 0)
261 usage();
262 pid = l;
263
264 p = procstat_getprocs(prstat, KERN_PROC_PID, pid, &cnt);
265 if (p == NULL)
266 errx(1, "procstat_getprocs()");
267 if (cnt != 0)
268 procstat(prstat, p);
269 procstat_freeprocs(prstat, p);
270
271 /* Suppress header after first process. */
272 hflag = 1;
273 }
274 if (interval)
275 sleep(interval);
276 } while (interval);
277 procstat_close(prstat);
278 exit(0);
279}
2 * Copyright (c) 2007, 2011 Robert N. M. Watson
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: head/usr.bin/procstat/procstat.c 227956 2011-11-24 20:54:06Z trociny $
27 */
28
29#include <sys/param.h>
30#include <sys/sysctl.h>
31#include <sys/user.h>
32
33#include <err.h>
34#include <libprocstat.h>
35#include <stdio.h>
36#include <stdlib.h>
37#include <sysexits.h>
38#include <unistd.h>
39
40#include "procstat.h"
41
42static int aflag, bflag, cflag, eflag, fflag, iflag, jflag, kflag, lflag, sflag;
43static int tflag, vflag, xflag;
44int hflag, nflag, Cflag;
45
46static void
47usage(void)
48{
49
50 fprintf(stderr, "usage: procstat [-h] [-C] [-M core] [-N system] "
51 "[-w interval] \n");
52 fprintf(stderr, " [-b | -c | -e | -f | -i | -j | -k | "
53 "-l | -s | -t | -v | -x] [-a | pid ...]\n");
54 exit(EX_USAGE);
55}
56
57static void
58procstat(struct procstat *prstat, struct kinfo_proc *kipp)
59{
60
61 if (bflag)
62 procstat_bin(kipp);
63 else if (cflag)
64 procstat_args(kipp);
65 else if (eflag)
66 procstat_env(kipp);
67 else if (fflag)
68 procstat_files(prstat, kipp);
69 else if (iflag)
70 procstat_sigs(prstat, kipp);
71 else if (jflag)
72 procstat_threads_sigs(prstat, kipp);
73 else if (kflag)
74 procstat_kstack(kipp, kflag);
75 else if (lflag)
76 procstat_rlimit(kipp);
77 else if (sflag)
78 procstat_cred(kipp);
79 else if (tflag)
80 procstat_threads(kipp);
81 else if (vflag)
82 procstat_vm(kipp);
83 else if (xflag)
84 procstat_auxv(kipp);
85 else
86 procstat_basic(kipp);
87}
88
89/*
90 * Sort processes first by pid and then tid.
91 */
92static int
93kinfo_proc_compare(const void *a, const void *b)
94{
95 int i;
96
97 i = ((const struct kinfo_proc *)a)->ki_pid -
98 ((const struct kinfo_proc *)b)->ki_pid;
99 if (i != 0)
100 return (i);
101 i = ((const struct kinfo_proc *)a)->ki_tid -
102 ((const struct kinfo_proc *)b)->ki_tid;
103 return (i);
104}
105
106void
107kinfo_proc_sort(struct kinfo_proc *kipp, int count)
108{
109
110 qsort(kipp, count, sizeof(*kipp), kinfo_proc_compare);
111}
112
113int
114main(int argc, char *argv[])
115{
116 int ch, interval, tmp;
117 int i;
118 struct kinfo_proc *p;
119 struct procstat *prstat;
120 long l;
121 pid_t pid;
122 char *dummy;
123 char *nlistf, *memf;
124 int cnt;
125
126 interval = 0;
127 memf = nlistf = NULL;
128 while ((ch = getopt(argc, argv, "CN:M:abcefijklhstvw:x")) != -1) {
129 switch (ch) {
130 case 'C':
131 Cflag++;
132 break;
133
134 case 'M':
135 memf = optarg;
136 break;
137 case 'N':
138 nlistf = optarg;
139 break;
140 case 'a':
141 aflag++;
142 break;
143
144 case 'b':
145 bflag++;
146 break;
147
148 case 'c':
149 cflag++;
150 break;
151
152 case 'e':
153 eflag++;
154 break;
155
156 case 'f':
157 fflag++;
158 break;
159
160 case 'i':
161 iflag++;
162 break;
163
164 case 'j':
165 jflag++;
166 break;
167
168 case 'k':
169 kflag++;
170 break;
171
172 case 'l':
173 lflag++;
174 break;
175
176 case 'n':
177 nflag++;
178 break;
179
180 case 'h':
181 hflag++;
182 break;
183
184 case 's':
185 sflag++;
186 break;
187
188 case 't':
189 tflag++;
190 break;
191
192 case 'v':
193 vflag++;
194 break;
195
196 case 'w':
197 l = strtol(optarg, &dummy, 10);
198 if (*dummy != '\0')
199 usage();
200 if (l < 1 || l > INT_MAX)
201 usage();
202 interval = l;
203 break;
204
205 case 'x':
206 xflag++;
207 break;
208
209 case '?':
210 default:
211 usage();
212 }
213
214 }
215 argc -= optind;
216 argv += optind;
217
218 /* We require that either 0 or 1 mode flags be set. */
219 tmp = bflag + cflag + eflag + fflag + (kflag ? 1 : 0) + lflag + sflag +
220 tflag + vflag + xflag;
221 if (!(tmp == 0 || tmp == 1))
222 usage();
223
224 /* We allow -k to be specified up to twice, but not more. */
225 if (kflag > 2)
226 usage();
227
228 /* Must specify either the -a flag or a list of pids. */
229 if (!(aflag == 1 && argc == 0) && !(aflag == 0 && argc > 0))
230 usage();
231
232 /* Only allow -C with -f. */
233 if (Cflag && !fflag)
234 usage();
235
236 if (memf != NULL)
237 prstat = procstat_open_kvm(nlistf, memf);
238 else
239 prstat = procstat_open_sysctl();
240 if (prstat == NULL)
241 errx(1, "procstat_open()");
242 do {
243 if (aflag) {
244 p = procstat_getprocs(prstat, KERN_PROC_PROC, 0, &cnt);
245 if (p == NULL)
246 errx(1, "procstat_getprocs()");
247 kinfo_proc_sort(p, cnt);
248 for (i = 0; i < cnt; i++) {
249 procstat(prstat, &p[i]);
250
251 /* Suppress header after first process. */
252 hflag = 1;
253 }
254 procstat_freeprocs(prstat, p);
255 }
256 for (i = 0; i < argc; i++) {
257 l = strtol(argv[i], &dummy, 10);
258 if (*dummy != '\0')
259 usage();
260 if (l < 0)
261 usage();
262 pid = l;
263
264 p = procstat_getprocs(prstat, KERN_PROC_PID, pid, &cnt);
265 if (p == NULL)
266 errx(1, "procstat_getprocs()");
267 if (cnt != 0)
268 procstat(prstat, p);
269 procstat_freeprocs(prstat, p);
270
271 /* Suppress header after first process. */
272 hflag = 1;
273 }
274 if (interval)
275 sleep(interval);
276 } while (interval);
277 procstat_close(prstat);
278 exit(0);
279}