]> CyberLeo.Net >> Repos - FreeBSD/stable/9.git/blob - usr.bin/procstat/procstat.c
MFC r227838, r227873, r228025, r228049, r228289, r228447, r230753:
[FreeBSD/stable/9.git] / usr.bin / procstat / procstat.c
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$
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
42 static int aflag, bflag, cflag, eflag, fflag, iflag, jflag, kflag, sflag, tflag;
43 static int vflag, xflag;
44 int     hflag, nflag, Cflag;
45
46 static void
47 usage(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             "-s | -t | -v | -x] [-a | pid ...]\n");
54         exit(EX_USAGE);
55 }
56
57 static void
58 procstat(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 (sflag)
76                 procstat_cred(kipp);
77         else if (tflag)
78                 procstat_threads(kipp);
79         else if (vflag)
80                 procstat_vm(kipp);
81         else if (xflag)
82                 procstat_auxv(kipp);
83         else
84                 procstat_basic(kipp);
85 }
86
87 /*
88  * Sort processes first by pid and then tid.
89  */
90 static int
91 kinfo_proc_compare(const void *a, const void *b)
92 {
93         int i;
94
95         i = ((const struct kinfo_proc *)a)->ki_pid -
96             ((const struct kinfo_proc *)b)->ki_pid;
97         if (i != 0)
98                 return (i);
99         i = ((const struct kinfo_proc *)a)->ki_tid -
100             ((const struct kinfo_proc *)b)->ki_tid;
101         return (i);
102 }
103
104 void
105 kinfo_proc_sort(struct kinfo_proc *kipp, int count)
106 {
107
108         qsort(kipp, count, sizeof(*kipp), kinfo_proc_compare);
109 }
110
111 int
112 main(int argc, char *argv[])
113 {
114         int ch, interval, tmp;
115         int i;
116         struct kinfo_proc *p;
117         struct procstat *prstat;
118         long l;
119         pid_t pid;
120         char *dummy;
121         char *nlistf, *memf;
122         int cnt;
123
124         interval = 0;
125         memf = nlistf = NULL;
126         while ((ch = getopt(argc, argv, "CN:M:abcefijkhstvw:x")) != -1) {
127                 switch (ch) {
128                 case 'C':
129                         Cflag++;
130                         break;
131
132                 case 'M':
133                         memf = optarg;
134                         break;
135                 case 'N':
136                         nlistf = optarg;
137                         break;
138                 case 'a':
139                         aflag++;
140                         break;
141
142                 case 'b':
143                         bflag++;
144                         break;
145
146                 case 'c':
147                         cflag++;
148                         break;
149
150                 case 'e':
151                         eflag++;
152                         break;
153
154                 case 'f':
155                         fflag++;
156                         break;
157
158                 case 'i':
159                         iflag++;
160                         break;
161
162                 case 'j':
163                         jflag++;
164                         break;
165
166                 case 'k':
167                         kflag++;
168                         break;
169
170                 case 'n':
171                         nflag++;
172                         break;
173
174                 case 'h':
175                         hflag++;
176                         break;
177
178                 case 's':
179                         sflag++;
180                         break;
181
182                 case 't':
183                         tflag++;
184                         break;
185
186                 case 'v':
187                         vflag++;
188                         break;
189
190                 case 'w':
191                         l = strtol(optarg, &dummy, 10);
192                         if (*dummy != '\0')
193                                 usage();
194                         if (l < 1 || l > INT_MAX)
195                                 usage();
196                         interval = l;
197                         break;
198
199                 case 'x':
200                         xflag++;
201                         break;
202
203                 case '?':
204                 default:
205                         usage();
206                 }
207
208         }
209         argc -= optind;
210         argv += optind;
211
212         /* We require that either 0 or 1 mode flags be set. */
213         tmp = bflag + cflag + eflag + fflag + (kflag ? 1 : 0) + sflag + tflag +
214             vflag + xflag;
215         if (!(tmp == 0 || tmp == 1))
216                 usage();
217
218         /* We allow -k to be specified up to twice, but not more. */
219         if (kflag > 2)
220                 usage();
221
222         /* Must specify either the -a flag or a list of pids. */
223         if (!(aflag == 1 && argc == 0) && !(aflag == 0 && argc > 0))
224                 usage();
225
226         /* Only allow -C with -f. */
227         if (Cflag && !fflag)
228                 usage();
229
230         if (memf != NULL)
231                 prstat = procstat_open_kvm(nlistf, memf);
232         else
233                 prstat = procstat_open_sysctl();
234         if (prstat == NULL)
235                 errx(1, "procstat_open()");
236         do {
237                 if (aflag) {
238                         p = procstat_getprocs(prstat, KERN_PROC_PROC, 0, &cnt);
239                         if (p == NULL)
240                                 errx(1, "procstat_getprocs()");
241                         kinfo_proc_sort(p, cnt);
242                         for (i = 0; i < cnt; i++) {
243                                 procstat(prstat, &p[i]);
244
245                                 /* Suppress header after first process. */
246                                 hflag = 1;
247                         }
248                         procstat_freeprocs(prstat, p);
249                 }
250                 for (i = 0; i < argc; i++) {
251                         l = strtol(argv[i], &dummy, 10);
252                         if (*dummy != '\0')
253                                 usage();
254                         if (l < 0)
255                                 usage();
256                         pid = l;
257
258                         p = procstat_getprocs(prstat, KERN_PROC_PID, pid, &cnt);
259                         if (p == NULL)
260                                 errx(1, "procstat_getprocs()");
261                         if (cnt != 0)
262                                 procstat(prstat, p);
263                         procstat_freeprocs(prstat, p);
264
265                         /* Suppress header after first process. */
266                         hflag = 1;
267                 }
268                 if (interval)
269                         sleep(interval);
270         } while (interval);
271         procstat_close(prstat);
272         exit(0);
273 }