procstat.c revision 310121
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: stable/10/usr.bin/procstat/procstat.c 310121 2016-12-15 16:52:17Z vangyzen $ 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 <string.h> 38#include <sysexits.h> 39#include <unistd.h> 40 41#include "procstat.h" 42 43static int aflag, bflag, cflag, eflag, fflag, iflag, jflag, kflag, lflag, rflag; 44static int sflag, tflag, vflag, xflag, Sflag; 45int hflag, nflag, Cflag, Hflag; 46 47static void 48usage(void) 49{ 50 51 fprintf(stderr, "usage: procstat [-CHhn] [-M core] [-N system] " 52 "[-w interval] \n"); 53 fprintf(stderr, " [-b | -c | -e | -f | -i | -j | -k | " 54 "-l | -r | -s | -S | -t | -v | -x]\n"); 55 fprintf(stderr, " [-a | pid | core ...]\n"); 56 exit(EX_USAGE); 57} 58 59static void 60procstat(struct procstat *prstat, struct kinfo_proc *kipp) 61{ 62 63 if (bflag) 64 procstat_bin(prstat, kipp); 65 else if (cflag) 66 procstat_args(prstat, kipp); 67 else if (eflag) 68 procstat_env(prstat, kipp); 69 else if (fflag) 70 procstat_files(prstat, kipp); 71 else if (iflag) 72 procstat_sigs(prstat, kipp); 73 else if (jflag) 74 procstat_threads_sigs(prstat, kipp); 75 else if (kflag) 76 procstat_kstack(prstat, kipp, kflag); 77 else if (lflag) 78 procstat_rlimit(prstat, kipp); 79 else if (rflag) 80 procstat_rusage(prstat, kipp); 81 else if (sflag) 82 procstat_cred(prstat, kipp); 83 else if (tflag) 84 procstat_threads(prstat, kipp); 85 else if (vflag) 86 procstat_vm(prstat, kipp); 87 else if (xflag) 88 procstat_auxv(prstat, kipp); 89 else if (Sflag) 90 procstat_cs(prstat, kipp); 91 else 92 procstat_basic(kipp); 93} 94 95/* 96 * Sort processes first by pid and then tid. 97 */ 98static int 99kinfo_proc_compare(const void *a, const void *b) 100{ 101 int i; 102 103 i = ((const struct kinfo_proc *)a)->ki_pid - 104 ((const struct kinfo_proc *)b)->ki_pid; 105 if (i != 0) 106 return (i); 107 i = ((const struct kinfo_proc *)a)->ki_tid - 108 ((const struct kinfo_proc *)b)->ki_tid; 109 return (i); 110} 111 112void 113kinfo_proc_sort(struct kinfo_proc *kipp, int count) 114{ 115 116 qsort(kipp, count, sizeof(*kipp), kinfo_proc_compare); 117} 118 119const char * 120kinfo_proc_thread_name(const struct kinfo_proc *kipp) 121{ 122 static char name[MAXCOMLEN+1]; 123 124 strlcpy(name, kipp->ki_tdname, sizeof(name)); 125 strlcat(name, kipp->ki_moretdname, sizeof(name)); 126 if (name[0] == '\0' || strcmp(kipp->ki_comm, name) == 0) { 127 name[0] = '-'; 128 name[1] = '\0'; 129 } 130 131 return (name); 132} 133 134int 135main(int argc, char *argv[]) 136{ 137 int ch, interval, tmp; 138 int i; 139 struct kinfo_proc *p; 140 struct procstat *prstat, *cprstat; 141 long l; 142 pid_t pid; 143 char *dummy; 144 char *nlistf, *memf; 145 int cnt; 146 147 interval = 0; 148 memf = nlistf = NULL; 149 while ((ch = getopt(argc, argv, "CHN:M:abcefijklhrsStvw:x")) != -1) { 150 switch (ch) { 151 case 'C': 152 Cflag++; 153 break; 154 155 case 'H': 156 Hflag++; 157 break; 158 159 case 'M': 160 memf = optarg; 161 break; 162 case 'N': 163 nlistf = optarg; 164 break; 165 case 'S': 166 Sflag++; 167 break; 168 case 'a': 169 aflag++; 170 break; 171 172 case 'b': 173 bflag++; 174 break; 175 176 case 'c': 177 cflag++; 178 break; 179 180 case 'e': 181 eflag++; 182 break; 183 184 case 'f': 185 fflag++; 186 break; 187 188 case 'i': 189 iflag++; 190 break; 191 192 case 'j': 193 jflag++; 194 break; 195 196 case 'k': 197 kflag++; 198 break; 199 200 case 'l': 201 lflag++; 202 break; 203 204 case 'n': 205 nflag++; 206 break; 207 208 case 'h': 209 hflag++; 210 break; 211 212 case 'r': 213 rflag++; 214 break; 215 216 case 's': 217 sflag++; 218 break; 219 220 case 't': 221 tflag++; 222 break; 223 224 case 'v': 225 vflag++; 226 break; 227 228 case 'w': 229 l = strtol(optarg, &dummy, 10); 230 if (*dummy != '\0') 231 usage(); 232 if (l < 1 || l > INT_MAX) 233 usage(); 234 interval = l; 235 break; 236 237 case 'x': 238 xflag++; 239 break; 240 241 case '?': 242 default: 243 usage(); 244 } 245 246 } 247 argc -= optind; 248 argv += optind; 249 250 /* We require that either 0 or 1 mode flags be set. */ 251 tmp = bflag + cflag + eflag + fflag + iflag + jflag + (kflag ? 1 : 0) + 252 lflag + rflag + sflag + tflag + vflag + xflag + Sflag; 253 if (!(tmp == 0 || tmp == 1)) 254 usage(); 255 256 /* We allow -k to be specified up to twice, but not more. */ 257 if (kflag > 2) 258 usage(); 259 260 /* Must specify either the -a flag or a list of pids. */ 261 if (!(aflag == 1 && argc == 0) && !(aflag == 0 && argc > 0)) 262 usage(); 263 264 /* Only allow -C with -f. */ 265 if (Cflag && !fflag) 266 usage(); 267 268 if (memf != NULL) 269 prstat = procstat_open_kvm(nlistf, memf); 270 else 271 prstat = procstat_open_sysctl(); 272 if (prstat == NULL) 273 errx(1, "procstat_open()"); 274 do { 275 if (aflag) { 276 p = procstat_getprocs(prstat, KERN_PROC_PROC, 0, &cnt); 277 if (p == NULL) 278 errx(1, "procstat_getprocs()"); 279 kinfo_proc_sort(p, cnt); 280 for (i = 0; i < cnt; i++) { 281 procstat(prstat, &p[i]); 282 283 /* Suppress header after first process. */ 284 hflag = 1; 285 } 286 procstat_freeprocs(prstat, p); 287 } 288 for (i = 0; i < argc; i++) { 289 l = strtol(argv[i], &dummy, 10); 290 if (*dummy == '\0') { 291 if (l < 0) 292 usage(); 293 pid = l; 294 295 p = procstat_getprocs(prstat, KERN_PROC_PID, pid, &cnt); 296 if (p == NULL) 297 errx(1, "procstat_getprocs()"); 298 if (cnt != 0) 299 procstat(prstat, p); 300 procstat_freeprocs(prstat, p); 301 } else { 302 cprstat = procstat_open_core(argv[i]); 303 if (cprstat == NULL) { 304 warnx("procstat_open()"); 305 continue; 306 } 307 p = procstat_getprocs(cprstat, KERN_PROC_PID, 308 -1, &cnt); 309 if (p == NULL) 310 errx(1, "procstat_getprocs()"); 311 if (cnt != 0) 312 procstat(cprstat, p); 313 procstat_freeprocs(cprstat, p); 314 procstat_close(cprstat); 315 } 316 /* Suppress header after first process. */ 317 hflag = 1; 318 } 319 if (interval) 320 sleep(interval); 321 } while (interval); 322 procstat_close(prstat); 323 exit(0); 324} 325