/usr.bin/procstat/procstat.c
https://bitbucket.org/freebsd/freebsd-head/ · C · 279 lines · 203 code · 40 blank · 36 comment · 64 complexity · 68ce9ebd6f8bf014e5639e57a92daac2 MD5 · raw file
- /*-
- * Copyright (c) 2007, 2011 Robert N. M. Watson
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- * 1. Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- *
- * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- * SUCH DAMAGE.
- *
- * $FreeBSD$
- */
- #include <sys/param.h>
- #include <sys/sysctl.h>
- #include <sys/user.h>
- #include <err.h>
- #include <libprocstat.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <sysexits.h>
- #include <unistd.h>
- #include "procstat.h"
- static int aflag, bflag, cflag, eflag, fflag, iflag, jflag, kflag, lflag, sflag;
- static int tflag, vflag, xflag;
- int hflag, nflag, Cflag;
- static void
- usage(void)
- {
- fprintf(stderr, "usage: procstat [-h] [-C] [-M core] [-N system] "
- "[-w interval] \n");
- fprintf(stderr, " [-b | -c | -e | -f | -i | -j | -k | "
- "-l | -s | -t | -v | -x] [-a | pid ...]\n");
- exit(EX_USAGE);
- }
- static void
- procstat(struct procstat *prstat, struct kinfo_proc *kipp)
- {
- if (bflag)
- procstat_bin(kipp);
- else if (cflag)
- procstat_args(kipp);
- else if (eflag)
- procstat_env(kipp);
- else if (fflag)
- procstat_files(prstat, kipp);
- else if (iflag)
- procstat_sigs(prstat, kipp);
- else if (jflag)
- procstat_threads_sigs(prstat, kipp);
- else if (kflag)
- procstat_kstack(kipp, kflag);
- else if (lflag)
- procstat_rlimit(kipp);
- else if (sflag)
- procstat_cred(kipp);
- else if (tflag)
- procstat_threads(kipp);
- else if (vflag)
- procstat_vm(kipp);
- else if (xflag)
- procstat_auxv(kipp);
- else
- procstat_basic(kipp);
- }
- /*
- * Sort processes first by pid and then tid.
- */
- static int
- kinfo_proc_compare(const void *a, const void *b)
- {
- int i;
- i = ((const struct kinfo_proc *)a)->ki_pid -
- ((const struct kinfo_proc *)b)->ki_pid;
- if (i != 0)
- return (i);
- i = ((const struct kinfo_proc *)a)->ki_tid -
- ((const struct kinfo_proc *)b)->ki_tid;
- return (i);
- }
- void
- kinfo_proc_sort(struct kinfo_proc *kipp, int count)
- {
- qsort(kipp, count, sizeof(*kipp), kinfo_proc_compare);
- }
- int
- main(int argc, char *argv[])
- {
- int ch, interval, tmp;
- int i;
- struct kinfo_proc *p;
- struct procstat *prstat;
- long l;
- pid_t pid;
- char *dummy;
- char *nlistf, *memf;
- int cnt;
- interval = 0;
- memf = nlistf = NULL;
- while ((ch = getopt(argc, argv, "CN:M:abcefijklhstvw:x")) != -1) {
- switch (ch) {
- case 'C':
- Cflag++;
- break;
- case 'M':
- memf = optarg;
- break;
- case 'N':
- nlistf = optarg;
- break;
- case 'a':
- aflag++;
- break;
- case 'b':
- bflag++;
- break;
- case 'c':
- cflag++;
- break;
- case 'e':
- eflag++;
- break;
- case 'f':
- fflag++;
- break;
- case 'i':
- iflag++;
- break;
- case 'j':
- jflag++;
- break;
- case 'k':
- kflag++;
- break;
- case 'l':
- lflag++;
- break;
- case 'n':
- nflag++;
- break;
- case 'h':
- hflag++;
- break;
- case 's':
- sflag++;
- break;
- case 't':
- tflag++;
- break;
- case 'v':
- vflag++;
- break;
- case 'w':
- l = strtol(optarg, &dummy, 10);
- if (*dummy != '\0')
- usage();
- if (l < 1 || l > INT_MAX)
- usage();
- interval = l;
- break;
- case 'x':
- xflag++;
- break;
- case '?':
- default:
- usage();
- }
- }
- argc -= optind;
- argv += optind;
- /* We require that either 0 or 1 mode flags be set. */
- tmp = bflag + cflag + eflag + fflag + (kflag ? 1 : 0) + lflag + sflag +
- tflag + vflag + xflag;
- if (!(tmp == 0 || tmp == 1))
- usage();
- /* We allow -k to be specified up to twice, but not more. */
- if (kflag > 2)
- usage();
- /* Must specify either the -a flag or a list of pids. */
- if (!(aflag == 1 && argc == 0) && !(aflag == 0 && argc > 0))
- usage();
- /* Only allow -C with -f. */
- if (Cflag && !fflag)
- usage();
- if (memf != NULL)
- prstat = procstat_open_kvm(nlistf, memf);
- else
- prstat = procstat_open_sysctl();
- if (prstat == NULL)
- errx(1, "procstat_open()");
- do {
- if (aflag) {
- p = procstat_getprocs(prstat, KERN_PROC_PROC, 0, &cnt);
- if (p == NULL)
- errx(1, "procstat_getprocs()");
- kinfo_proc_sort(p, cnt);
- for (i = 0; i < cnt; i++) {
- procstat(prstat, &p[i]);
- /* Suppress header after first process. */
- hflag = 1;
- }
- procstat_freeprocs(prstat, p);
- }
- for (i = 0; i < argc; i++) {
- l = strtol(argv[i], &dummy, 10);
- if (*dummy != '\0')
- usage();
- if (l < 0)
- usage();
- pid = l;
- p = procstat_getprocs(prstat, KERN_PROC_PID, pid, &cnt);
- if (p == NULL)
- errx(1, "procstat_getprocs()");
- if (cnt != 0)
- procstat(prstat, p);
- procstat_freeprocs(prstat, p);
- /* Suppress header after first process. */
- hflag = 1;
- }
- if (interval)
- sleep(interval);
- } while (interval);
- procstat_close(prstat);
- exit(0);
- }