1*52e706caSkn /* $OpenBSD: ps.c,v 1.80 2023/11/10 09:17:02 kn Exp $ */
2df930be7Sderaadt /* $NetBSD: ps.c,v 1.15 1995/05/18 20:33:25 mycroft Exp $ */
3df930be7Sderaadt
4df930be7Sderaadt /*-
5df930be7Sderaadt * Copyright (c) 1990, 1993, 1994
6df930be7Sderaadt * The Regents of the University of California. All rights reserved.
7df930be7Sderaadt *
8df930be7Sderaadt * Redistribution and use in source and binary forms, with or without
9df930be7Sderaadt * modification, are permitted provided that the following conditions
10df930be7Sderaadt * are met:
11df930be7Sderaadt * 1. Redistributions of source code must retain the above copyright
12df930be7Sderaadt * notice, this list of conditions and the following disclaimer.
13df930be7Sderaadt * 2. Redistributions in binary form must reproduce the above copyright
14df930be7Sderaadt * notice, this list of conditions and the following disclaimer in the
15df930be7Sderaadt * documentation and/or other materials provided with the distribution.
1629295d1cSmillert * 3. Neither the name of the University nor the names of its contributors
17df930be7Sderaadt * may be used to endorse or promote products derived from this software
18df930be7Sderaadt * without specific prior written permission.
19df930be7Sderaadt *
20df930be7Sderaadt * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21df930be7Sderaadt * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22df930be7Sderaadt * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23df930be7Sderaadt * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24df930be7Sderaadt * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25df930be7Sderaadt * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26df930be7Sderaadt * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27df930be7Sderaadt * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28df930be7Sderaadt * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29df930be7Sderaadt * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30df930be7Sderaadt * SUCH DAMAGE.
31df930be7Sderaadt */
32df930be7Sderaadt
33acd8f640Sderaadt #include <sys/param.h> /* NODEV */
34b9fc9a72Sderaadt #include <sys/types.h>
35acd8f640Sderaadt #include <sys/signal.h>
360adec678Sderaadt #include <sys/sysctl.h>
37df930be7Sderaadt #include <sys/time.h>
38df930be7Sderaadt #include <sys/resource.h>
39df930be7Sderaadt #include <sys/proc.h>
40df930be7Sderaadt #include <sys/stat.h>
41df930be7Sderaadt #include <sys/ioctl.h>
42df930be7Sderaadt
43df930be7Sderaadt #include <ctype.h>
44df930be7Sderaadt #include <err.h>
45df930be7Sderaadt #include <errno.h>
46df930be7Sderaadt #include <fcntl.h>
47df930be7Sderaadt #include <kvm.h>
48033e8ff0Sschwarze #include <locale.h>
49df930be7Sderaadt #include <nlist.h>
50df930be7Sderaadt #include <paths.h>
5152576792Salex #include <pwd.h>
52df930be7Sderaadt #include <stdio.h>
53df930be7Sderaadt #include <stdlib.h>
54df930be7Sderaadt #include <string.h>
55df930be7Sderaadt #include <unistd.h>
566763c4d8Sderaadt #include <limits.h>
57df930be7Sderaadt
58df930be7Sderaadt #include "ps.h"
59df930be7Sderaadt
609517852aSmpech extern char *__progname;
619517852aSmpech
6279f972fbSderaadt struct varent *vhead;
63df930be7Sderaadt
64df930be7Sderaadt int eval; /* exit value */
65df930be7Sderaadt int sumrusage; /* -S */
66df930be7Sderaadt int termwidth; /* width of screen (0 == infinity) */
67df930be7Sderaadt int totwidth; /* calculated width of requested variables */
68df930be7Sderaadt
696b176adcSmillert int needcomm, needenv, neednlist, commandonly;
70df930be7Sderaadt
714527c594Smillert enum sort { DEFAULT, SORTMEM, SORTCPU } sortby = DEFAULT;
72df930be7Sderaadt
73c72b5b24Smillert static char *kludge_oldps_options(char *);
74c72b5b24Smillert static int pscomp(const void *, const void *);
75c72b5b24Smillert static void scanvars(void);
7624ca0f25Sjob static void forest_sort(struct pinfo *, int);
77c72b5b24Smillert static void usage(void);
78df930be7Sderaadt
79df930be7Sderaadt char dfmt[] = "pid tt state time command";
80b9d9655cSjsing char tfmt[] = "pid tid tt state time command";
81df930be7Sderaadt char jfmt[] = "user pid ppid pgid sess jobc state tt time command";
82df930be7Sderaadt char lfmt[] = "uid pid ppid cpu pri nice vsz rss wchan state tt time command";
83df930be7Sderaadt char o1[] = "pid";
84df930be7Sderaadt char o2[] = "tt state time command";
85df930be7Sderaadt char ufmt[] = "user pid %cpu %mem vsz rss tt state start time command";
86df930be7Sderaadt char vfmt[] = "pid state time sl re pagein vsz rss lim tsiz %cpu %mem command";
87df930be7Sderaadt
88df930be7Sderaadt kvm_t *kd;
895877eea9Sart int kvm_sysctl_only;
90df930be7Sderaadt
91df930be7Sderaadt int
main(int argc,char * argv[])9228416801Sderaadt main(int argc, char *argv[])
93df930be7Sderaadt {
9424ca0f25Sjob struct kinfo_proc *kp;
9524ca0f25Sjob struct pinfo *pinfo;
96df930be7Sderaadt struct varent *vent;
97df930be7Sderaadt struct winsize ws;
98df930be7Sderaadt dev_t ttydev;
99df930be7Sderaadt pid_t pid;
100df930be7Sderaadt uid_t uid;
1011601eb99Sguenther int all, ch, flag, i, fmt, lineno, nentries;
102fc7b2cf1Sguenther int prtheader, showthreads, wflag, kflag, what, Uflag, xflg;
10324ca0f25Sjob int forest;
104c2e41e89Sguenther char *nlistf, *memf, *swapf, *cols, errbuf[_POSIX2_LINE_MAX];
105df930be7Sderaadt
106033e8ff0Sschwarze setlocale(LC_CTYPE, "");
107033e8ff0Sschwarze
1087220d8ecSbentley termwidth = 0;
1097220d8ecSbentley if ((cols = getenv("COLUMNS")) != NULL)
1107220d8ecSbentley termwidth = strtonum(cols, 1, INT_MAX, NULL);
111511be169Sbentley if (termwidth == 0 &&
112511be169Sbentley (ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) == 0 ||
113511be169Sbentley ioctl(STDERR_FILENO, TIOCGWINSZ, &ws) == 0 ||
114511be169Sbentley ioctl(STDIN_FILENO, TIOCGWINSZ, &ws) == 0) &&
1157220d8ecSbentley ws.ws_col > 0)
116df930be7Sderaadt termwidth = ws.ws_col - 1;
1177220d8ecSbentley if (termwidth == 0)
1187220d8ecSbentley termwidth = 79;
119df930be7Sderaadt
120df930be7Sderaadt if (argc > 1)
121df930be7Sderaadt argv[1] = kludge_oldps_options(argv[1]);
122df930be7Sderaadt
123fc7b2cf1Sguenther all = fmt = prtheader = showthreads = wflag = kflag = Uflag = xflg = 0;
124df930be7Sderaadt pid = -1;
125b9badb0aSotto uid = 0;
12624ca0f25Sjob forest = 0;
127df930be7Sderaadt ttydev = NODEV;
128df930be7Sderaadt memf = nlistf = swapf = NULL;
129df930be7Sderaadt while ((ch = getopt(argc, argv,
13024ca0f25Sjob "AaCcefgHhjkLlM:mN:O:o:p:rSTt:U:uvW:wx")) != -1)
131c9c5a36dSkevlo switch (ch) {
1329ea50c04Sokan case 'A':
1339ea50c04Sokan all = 1;
1349ea50c04Sokan xflg = 1;
1359ea50c04Sokan break;
136df930be7Sderaadt case 'a':
137df930be7Sderaadt all = 1;
138df930be7Sderaadt break;
1399ea50c04Sokan case 'C':
1404bdadadaSbluhm break; /* no-op */
141df930be7Sderaadt case 'c':
142df930be7Sderaadt commandonly = 1;
143df930be7Sderaadt break;
144df930be7Sderaadt case 'e': /* XXX set ufmt */
145df930be7Sderaadt needenv = 1;
146df930be7Sderaadt break;
14724ca0f25Sjob case 'f':
14824ca0f25Sjob forest = 1;
14924ca0f25Sjob break;
150df930be7Sderaadt case 'g':
151df930be7Sderaadt break; /* no-op */
152fc7b2cf1Sguenther case 'H':
153fc7b2cf1Sguenther showthreads = 1;
154fc7b2cf1Sguenther break;
155df930be7Sderaadt case 'h':
156df930be7Sderaadt prtheader = ws.ws_row > 5 ? ws.ws_row : 22;
157df930be7Sderaadt break;
158df930be7Sderaadt case 'j':
159df930be7Sderaadt parsefmt(jfmt);
160df930be7Sderaadt fmt = 1;
161df930be7Sderaadt jfmt[0] = '\0';
162df930be7Sderaadt break;
1633f2d366dSniklas case 'k':
1643f36dab3Sokan kflag = 1;
1653f2d366dSniklas break;
166df930be7Sderaadt case 'L':
167df930be7Sderaadt showkey();
168df930be7Sderaadt exit(0);
169df930be7Sderaadt case 'l':
170df930be7Sderaadt parsefmt(lfmt);
171df930be7Sderaadt fmt = 1;
172df930be7Sderaadt lfmt[0] = '\0';
173df930be7Sderaadt break;
174df930be7Sderaadt case 'M':
175df930be7Sderaadt memf = optarg;
176df930be7Sderaadt break;
177df930be7Sderaadt case 'm':
178df930be7Sderaadt sortby = SORTMEM;
179df930be7Sderaadt break;
180df930be7Sderaadt case 'N':
181df930be7Sderaadt nlistf = optarg;
182df930be7Sderaadt break;
183df930be7Sderaadt case 'O':
184df930be7Sderaadt parsefmt(o1);
185df930be7Sderaadt parsefmt(optarg);
186df930be7Sderaadt parsefmt(o2);
187df930be7Sderaadt o1[0] = o2[0] = '\0';
188df930be7Sderaadt fmt = 1;
189df930be7Sderaadt break;
190df930be7Sderaadt case 'o':
191df930be7Sderaadt parsefmt(optarg);
192df930be7Sderaadt fmt = 1;
193df930be7Sderaadt break;
194df930be7Sderaadt case 'p':
195df930be7Sderaadt pid = atol(optarg);
196df930be7Sderaadt xflg = 1;
197df930be7Sderaadt break;
198df930be7Sderaadt case 'r':
199df930be7Sderaadt sortby = SORTCPU;
200df930be7Sderaadt break;
201df930be7Sderaadt case 'S':
202df930be7Sderaadt sumrusage = 1;
203df930be7Sderaadt break;
204df930be7Sderaadt case 'T':
205df930be7Sderaadt if ((optarg = ttyname(STDIN_FILENO)) == NULL)
206df930be7Sderaadt errx(1, "stdin: not a terminal");
207df930be7Sderaadt /* FALLTHROUGH */
208df930be7Sderaadt case 't': {
209df930be7Sderaadt struct stat sb;
210b9fc9a72Sderaadt char *ttypath, pathbuf[PATH_MAX];
211df930be7Sderaadt
212df930be7Sderaadt if (strcmp(optarg, "co") == 0)
213df930be7Sderaadt ttypath = _PATH_CONSOLE;
2144075259fSderaadt else if (*optarg != '/') {
2154075259fSderaadt int r = snprintf(pathbuf, sizeof(pathbuf), "%s%s",
2164075259fSderaadt _PATH_TTY, optarg);
2174075259fSderaadt if (r < 0 || r > sizeof(pathbuf))
2184075259fSderaadt errx(1, "%s: too long\n", optarg);
2194075259fSderaadt ttypath = pathbuf;
2204075259fSderaadt } else
221df930be7Sderaadt ttypath = optarg;
222df930be7Sderaadt if (stat(ttypath, &sb) == -1)
223df930be7Sderaadt err(1, "%s", ttypath);
224df930be7Sderaadt if (!S_ISCHR(sb.st_mode))
225df930be7Sderaadt errx(1, "%s: not a terminal", ttypath);
226df930be7Sderaadt ttydev = sb.st_rdev;
227df930be7Sderaadt break;
228df930be7Sderaadt }
229*52e706caSkn case 'U': {
230*52e706caSkn int found = 0;
231*52e706caSkn
232*52e706caSkn if (uid_from_user(optarg, &uid) == 0)
233*52e706caSkn found = 1;
234*52e706caSkn else {
235*52e706caSkn const char *errstr;
236*52e706caSkn
237*52e706caSkn uid = strtonum(optarg, 0, UID_MAX, &errstr);
238*52e706caSkn if (errstr == NULL &&
239*52e706caSkn user_from_uid(uid, 1) != NULL)
240*52e706caSkn found = 1;
241*52e706caSkn }
242*52e706caSkn if (!found)
243*52e706caSkn errx(1, "%s: unknown user", optarg);
244b9badb0aSotto Uflag = xflg = 1;
24552576792Salex break;
246*52e706caSkn }
247df930be7Sderaadt case 'u':
248df930be7Sderaadt parsefmt(ufmt);
249df930be7Sderaadt sortby = SORTCPU;
250df930be7Sderaadt fmt = 1;
251df930be7Sderaadt ufmt[0] = '\0';
252df930be7Sderaadt break;
253df930be7Sderaadt case 'v':
254df930be7Sderaadt parsefmt(vfmt);
255df930be7Sderaadt sortby = SORTMEM;
256df930be7Sderaadt fmt = 1;
257df930be7Sderaadt vfmt[0] = '\0';
258df930be7Sderaadt break;
259df930be7Sderaadt case 'W':
260df930be7Sderaadt swapf = optarg;
261df930be7Sderaadt break;
262df930be7Sderaadt case 'w':
263df930be7Sderaadt if (wflag)
264df930be7Sderaadt termwidth = UNLIMITED;
265df930be7Sderaadt else if (termwidth < 131)
266df930be7Sderaadt termwidth = 131;
2673f36dab3Sokan wflag = 1;
268df930be7Sderaadt break;
269df930be7Sderaadt case 'x':
270df930be7Sderaadt xflg = 1;
271df930be7Sderaadt break;
272df930be7Sderaadt default:
273df930be7Sderaadt usage();
274df930be7Sderaadt }
275df930be7Sderaadt argc -= optind;
276df930be7Sderaadt argv += optind;
277df930be7Sderaadt
278df930be7Sderaadt #define BACKWARD_COMPATIBILITY
279df930be7Sderaadt #ifdef BACKWARD_COMPATIBILITY
280df930be7Sderaadt if (*argv) {
281df930be7Sderaadt nlistf = *argv;
282df930be7Sderaadt if (*++argv) {
283df930be7Sderaadt memf = *argv;
284df930be7Sderaadt if (*++argv)
285df930be7Sderaadt swapf = *argv;
286df930be7Sderaadt }
287df930be7Sderaadt }
288df930be7Sderaadt #endif
289df930be7Sderaadt
2905877eea9Sart if (nlistf == NULL && memf == NULL && swapf == NULL) {
2915877eea9Sart kd = kvm_openfiles(NULL, NULL, NULL, KVM_NO_FILES, errbuf);
2925877eea9Sart kvm_sysctl_only = 1;
2935877eea9Sart } else {
294df930be7Sderaadt kd = kvm_openfiles(nlistf, memf, swapf, O_RDONLY, errbuf);
2955877eea9Sart }
296f2812ea1Sguenther if (kd == NULL)
297df930be7Sderaadt errx(1, "%s", errbuf);
298df930be7Sderaadt
2995db3b398Sflorian if (unveil(_PATH_DEVDB, "r") == -1 && errno != ENOENT)
300bc5a8259Sbeck err(1, "unveil %s", _PATH_DEVDB);
3015db3b398Sflorian if (unveil(_PATH_DEV, "r") == -1 && errno != ENOENT)
302bc5a8259Sbeck err(1, "unveil %s", _PATH_DEV);
3032f5bf968Sderaadt if (swapf)
3042f5bf968Sderaadt if (unveil(swapf, "r") == -1)
305bc5a8259Sbeck err(1, "unveil %s", swapf);
3062f5bf968Sderaadt if (nlistf)
3072f5bf968Sderaadt if (unveil(nlistf, "r") == -1)
308bc5a8259Sbeck err(1, "unveil %s", nlistf);
3092f5bf968Sderaadt if (memf)
3102f5bf968Sderaadt if (unveil(memf, "r") == -1)
311bc5a8259Sbeck err(1, "unveil %s", memf);
312dcce26deSderaadt if (pledge("stdio rpath getpw ps", NULL) == -1)
31346a2cbd9Sderaadt err(1, "pledge");
3144120813fSderaadt
315b9d9655cSjsing if (!fmt) {
316b9d9655cSjsing if (showthreads)
317b9d9655cSjsing parsefmt(tfmt);
318b9d9655cSjsing else
319df930be7Sderaadt parsefmt(dfmt);
320b9d9655cSjsing }
321df930be7Sderaadt
32252576792Salex /* XXX - should be cleaner */
323ba86f864Sotto if (!all && ttydev == NODEV && pid == -1 && !Uflag) {
324df930be7Sderaadt uid = getuid();
325ba86f864Sotto Uflag = 1;
326ba86f864Sotto }
327df930be7Sderaadt
328df930be7Sderaadt /*
329df930be7Sderaadt * scan requested variables, noting what structures are needed,
33005ea949fSmillert * and adjusting header widths as appropriate.
331df930be7Sderaadt */
332df930be7Sderaadt scanvars();
3336b176adcSmillert
3346b176adcSmillert if (neednlist && !nlistread)
3356b176adcSmillert (void) donlist();
3366b176adcSmillert
337df930be7Sderaadt /*
338df930be7Sderaadt * get proc list
339df930be7Sderaadt */
340b9badb0aSotto if (Uflag) {
34190624e93Smillert what = KERN_PROC_UID;
34290624e93Smillert flag = uid;
343df930be7Sderaadt } else if (ttydev != NODEV) {
34490624e93Smillert what = KERN_PROC_TTY;
34590624e93Smillert flag = ttydev;
346df930be7Sderaadt } else if (pid != -1) {
34790624e93Smillert what = KERN_PROC_PID;
34890624e93Smillert flag = pid;
3493f2d366dSniklas } else if (kflag) {
35090624e93Smillert what = KERN_PROC_KTHREAD;
3513f2d366dSniklas flag = 0;
352df930be7Sderaadt } else {
35390624e93Smillert what = KERN_PROC_ALL;
354df930be7Sderaadt flag = 0;
355df930be7Sderaadt }
3566171375dSpirofti if (showthreads)
3576171375dSpirofti what |= KERN_PROC_SHOW_THREADS;
358970ac3e4Sderaadt
359df930be7Sderaadt /*
360df930be7Sderaadt * select procs
361df930be7Sderaadt */
3625027561dSguenther kp = kvm_getprocs(kd, what, flag, sizeof(*kp), &nentries);
363e6592a11Sangelos if (kp == NULL)
36490624e93Smillert errx(1, "%s", kvm_geterr(kd));
365f2812ea1Sguenther
366df930be7Sderaadt /*
367df930be7Sderaadt * print header
368df930be7Sderaadt */
369df930be7Sderaadt printheader();
370df930be7Sderaadt if (nentries == 0)
371a30d4cafSmichaels exit(1);
37224ca0f25Sjob
37324ca0f25Sjob if ((pinfo = calloc(nentries, sizeof(struct pinfo))) == NULL)
37424ca0f25Sjob err(1, NULL);
37590624e93Smillert for (i = 0; i < nentries; i++)
37624ca0f25Sjob pinfo[i].ki = &kp[i];
37724ca0f25Sjob qsort(pinfo, nentries, sizeof(struct pinfo), pscomp);
37824ca0f25Sjob
37924ca0f25Sjob if (forest)
38024ca0f25Sjob forest_sort(pinfo, nentries);
38124ca0f25Sjob
382df930be7Sderaadt /*
383df930be7Sderaadt * for each proc, call each variable output function.
384df930be7Sderaadt */
385df930be7Sderaadt for (i = lineno = 0; i < nentries; i++) {
38624ca0f25Sjob if (xflg == 0 && ((int)pinfo[i].ki->p_tdev == NODEV ||
38724ca0f25Sjob (pinfo[i].ki->p_psflags & PS_CONTROLT ) == 0))
388df930be7Sderaadt continue;
38924ca0f25Sjob if (showthreads && pinfo[i].ki->p_tid == -1)
390243c13beSpirofti continue;
391df930be7Sderaadt for (vent = vhead; vent; vent = vent->next) {
39224ca0f25Sjob (vent->var->oproc)(&pinfo[i], vent);
393df930be7Sderaadt if (vent->next != NULL)
394df930be7Sderaadt (void)putchar(' ');
395df930be7Sderaadt }
396df930be7Sderaadt (void)putchar('\n');
397df930be7Sderaadt if (prtheader && lineno++ == prtheader - 4) {
398df930be7Sderaadt (void)putchar('\n');
399df930be7Sderaadt printheader();
400df930be7Sderaadt lineno = 0;
401df930be7Sderaadt }
402df930be7Sderaadt }
403df930be7Sderaadt exit(eval);
404df930be7Sderaadt }
405df930be7Sderaadt
406df930be7Sderaadt static void
scanvars(void)40728416801Sderaadt scanvars(void)
408df930be7Sderaadt {
409df930be7Sderaadt struct varent *vent;
410df930be7Sderaadt VAR *v;
411df930be7Sderaadt int i;
412df930be7Sderaadt
413df930be7Sderaadt for (vent = vhead; vent; vent = vent->next) {
414df930be7Sderaadt v = vent->var;
415df930be7Sderaadt i = strlen(v->header);
416df930be7Sderaadt if (v->width < i)
417df930be7Sderaadt v->width = i;
418df930be7Sderaadt totwidth += v->width + 1; /* +1 for space */
419df930be7Sderaadt if (v->flag & COMM)
420df930be7Sderaadt needcomm = 1;
4216b176adcSmillert if (v->flag & NLIST)
4226b176adcSmillert neednlist = 1;
423df930be7Sderaadt }
424df930be7Sderaadt totwidth--;
425df930be7Sderaadt }
426df930be7Sderaadt
427df930be7Sderaadt static int
pscomp(const void * v1,const void * v2)42890624e93Smillert pscomp(const void *v1, const void *v2)
429df930be7Sderaadt {
43024ca0f25Sjob const struct pinfo *p1 = (const struct pinfo *)v1;
43124ca0f25Sjob const struct pinfo *p2 = (const struct pinfo *)v2;
43224ca0f25Sjob const struct kinfo_proc *kp1 = p1->ki;
43324ca0f25Sjob const struct kinfo_proc *kp2 = p2->ki;
434df930be7Sderaadt int i;
43590624e93Smillert #define VSIZE(k) ((k)->p_vm_dsize + (k)->p_vm_ssize + (k)->p_vm_tsize)
436df930be7Sderaadt
43790624e93Smillert if (sortby == SORTCPU && (i = getpcpu(kp2) - getpcpu(kp1)) != 0)
43890624e93Smillert return (i);
43990624e93Smillert if (sortby == SORTMEM && (i = VSIZE(kp2) - VSIZE(kp1)) != 0)
44090624e93Smillert return (i);
44190624e93Smillert if ((i = kp1->p_tdev - kp2->p_tdev) == 0 &&
44290624e93Smillert (i = kp1->p_ustart_sec - kp2->p_ustart_sec) == 0)
44390624e93Smillert i = kp1->p_ustart_usec - kp2->p_ustart_usec;
444fffbb510Smillert return (i);
445fffbb510Smillert }
446df930be7Sderaadt
447df930be7Sderaadt /*
448df930be7Sderaadt * ICK (all for getopt), would rather hide the ugliness
449df930be7Sderaadt * here than taint the main code.
450df930be7Sderaadt *
451df930be7Sderaadt * ps foo -> ps -foo
452df930be7Sderaadt * ps 34 -> ps -p34
453df930be7Sderaadt *
454df930be7Sderaadt * The old convention that 't' with no trailing tty arg means the users
455df930be7Sderaadt * tty, is only supported if argv[1] doesn't begin with a '-'. This same
456df930be7Sderaadt * feature is available with the option 'T', which takes no argument.
457df930be7Sderaadt */
458df930be7Sderaadt static char *
kludge_oldps_options(char * s)45928416801Sderaadt kludge_oldps_options(char *s)
460df930be7Sderaadt {
461df930be7Sderaadt size_t len;
462df930be7Sderaadt char *newopts, *ns, *cp;
463df930be7Sderaadt
464df930be7Sderaadt len = strlen(s);
465263704f2Sderaadt if ((newopts = ns = malloc(2 + len + 1)) == NULL)
466df930be7Sderaadt err(1, NULL);
467df930be7Sderaadt /*
468df930be7Sderaadt * options begin with '-'
469df930be7Sderaadt */
470df930be7Sderaadt if (*s != '-')
471df930be7Sderaadt *ns++ = '-'; /* add option flag */
472263704f2Sderaadt
473df930be7Sderaadt /*
474df930be7Sderaadt * gaze to end of argv[1]
475df930be7Sderaadt */
476df930be7Sderaadt cp = s + len - 1;
477df930be7Sderaadt /*
478df930be7Sderaadt * if last letter is a 't' flag with no argument (in the context
479df930be7Sderaadt * of the oldps options -- option string NOT starting with a '-' --
480df930be7Sderaadt * then convert to 'T' (meaning *this* terminal, i.e. ttyname(0)).
481df930be7Sderaadt */
482df930be7Sderaadt if (*cp == 't' && *s != '-')
483df930be7Sderaadt *cp = 'T';
484df930be7Sderaadt else {
485df930be7Sderaadt /*
486df930be7Sderaadt * otherwise check for trailing number, which *may* be a
487df930be7Sderaadt * pid.
488df930be7Sderaadt */
489cd3e3e8cSderaadt while (cp >= s && isdigit((unsigned char)*cp))
490df930be7Sderaadt --cp;
491df930be7Sderaadt }
492df930be7Sderaadt cp++;
493df930be7Sderaadt memmove(ns, s, (size_t)(cp - s)); /* copy up to trailing number */
494df930be7Sderaadt ns += cp - s;
495df930be7Sderaadt /*
496*52e706caSkn * if there's a trailing number, and not a preceding 'p' (pid),
497*52e706caSkn * 't' (tty) or 'U' (user) flag,
498*52e706caSkn * then assume it's a pid and insert a 'p' flag.
499df930be7Sderaadt */
500cd3e3e8cSderaadt if (isdigit((unsigned char)*cp) &&
501*52e706caSkn (cp == s || (cp[-1] != 't' && cp[-1] != 'p' && cp[-1] != 'U' &&
502764064c4Smickey (cp - 1 == s || cp[-2] != 't'))))
503df930be7Sderaadt *ns++ = 'p';
504263704f2Sderaadt /* and append the number */
505263704f2Sderaadt (void)strlcpy(ns, cp, newopts + len + 3 - ns);
506df930be7Sderaadt
507df930be7Sderaadt return (newopts);
508df930be7Sderaadt }
509df930be7Sderaadt
510df930be7Sderaadt static void
forest_sort(struct pinfo * ki,int items)51124ca0f25Sjob forest_sort(struct pinfo *ki, int items)
51224ca0f25Sjob {
51324ca0f25Sjob int dst, lvl, maxlvl, n, ndst, nsrc, siblings, src;
51424ca0f25Sjob unsigned char *path;
51524ca0f25Sjob struct pinfo kn;
51624ca0f25Sjob
51724ca0f25Sjob /*
51824ca0f25Sjob * First, sort the entries by forest, tracking the forest
51924ca0f25Sjob * depth in the level field.
52024ca0f25Sjob */
52124ca0f25Sjob src = 0;
52224ca0f25Sjob maxlvl = 0;
52324ca0f25Sjob while (src < items) {
52424ca0f25Sjob if (ki[src].level) {
52524ca0f25Sjob src++;
52624ca0f25Sjob continue;
52724ca0f25Sjob }
52824ca0f25Sjob for (nsrc = 1; src + nsrc < items; nsrc++)
52924ca0f25Sjob if (!ki[src + nsrc].level)
53024ca0f25Sjob break;
53124ca0f25Sjob
53224ca0f25Sjob for (dst = 0; dst < items; dst++) {
53324ca0f25Sjob if (ki[dst].ki->p_pid == ki[src].ki->p_pid)
53424ca0f25Sjob continue;
53524ca0f25Sjob if (ki[dst].ki->p_pid == ki[src].ki->p_ppid)
53624ca0f25Sjob break;
53724ca0f25Sjob }
53824ca0f25Sjob
53924ca0f25Sjob if (dst == items) {
54024ca0f25Sjob src += nsrc;
54124ca0f25Sjob continue;
54224ca0f25Sjob }
54324ca0f25Sjob
54424ca0f25Sjob for (ndst = 1; dst + ndst < items; ndst++)
54524ca0f25Sjob if (ki[dst + ndst].level <= ki[dst].level)
54624ca0f25Sjob break;
54724ca0f25Sjob
54824ca0f25Sjob for (n = src; n < src + nsrc; n++) {
54924ca0f25Sjob ki[n].level += ki[dst].level + 1;
55024ca0f25Sjob if (maxlvl < ki[n].level)
55124ca0f25Sjob maxlvl = ki[n].level;
55224ca0f25Sjob }
55324ca0f25Sjob
55424ca0f25Sjob while (nsrc) {
55524ca0f25Sjob if (src < dst) {
55624ca0f25Sjob kn = ki[src];
55724ca0f25Sjob memmove(ki + src, ki + src + 1,
55824ca0f25Sjob (dst - src + ndst - 1) * sizeof *ki);
55924ca0f25Sjob ki[dst + ndst - 1] = kn;
56024ca0f25Sjob nsrc--;
56124ca0f25Sjob dst--;
56224ca0f25Sjob ndst++;
56324ca0f25Sjob } else if (src != dst + ndst) {
56424ca0f25Sjob kn = ki[src];
56524ca0f25Sjob memmove(ki + dst + ndst + 1, ki + dst + ndst,
56624ca0f25Sjob (src - dst - ndst) * sizeof *ki);
56724ca0f25Sjob ki[dst + ndst] = kn;
56824ca0f25Sjob ndst++;
56924ca0f25Sjob nsrc--;
57024ca0f25Sjob src++;
57124ca0f25Sjob } else {
57224ca0f25Sjob ndst += nsrc;
57324ca0f25Sjob src += nsrc;
57424ca0f25Sjob nsrc = 0;
57524ca0f25Sjob }
57624ca0f25Sjob }
57724ca0f25Sjob }
57824ca0f25Sjob /*
57924ca0f25Sjob * Now populate prefix (instead of level) with the command
58024ca0f25Sjob * prefix used to show descendancies.
58124ca0f25Sjob */
58224ca0f25Sjob path = calloc(1, (maxlvl + 7) / 8);
58324ca0f25Sjob if (path == NULL)
58424ca0f25Sjob err(1, NULL);
58524ca0f25Sjob
58624ca0f25Sjob for (src = 0; src < items; src++) {
58724ca0f25Sjob if ((lvl = ki[src].level) == 0) {
58824ca0f25Sjob ki[src].prefix = NULL;
58924ca0f25Sjob continue;
59024ca0f25Sjob }
59124ca0f25Sjob
59224ca0f25Sjob if ((ki[src].prefix = malloc(lvl * 2 + 1)) == NULL)
59324ca0f25Sjob err(1, NULL);
59424ca0f25Sjob
59524ca0f25Sjob for (n = 0; n < lvl - 2; n++) {
59624ca0f25Sjob ki[src].prefix[n * 2] =
59724ca0f25Sjob path[n / 8] & 1 << (n % 8) ? '|' : ' ';
59824ca0f25Sjob ki[src].prefix[n * 2 + 1] = ' ';
59924ca0f25Sjob
60024ca0f25Sjob }
60124ca0f25Sjob if (n == lvl - 2) {
60224ca0f25Sjob /* Have I any more siblings? */
60324ca0f25Sjob for (siblings = 0, dst = src + 1; dst < items; dst++) {
60424ca0f25Sjob if (ki[dst].level > lvl)
60524ca0f25Sjob continue;
60624ca0f25Sjob if (ki[dst].level == lvl)
60724ca0f25Sjob siblings = 1;
60824ca0f25Sjob break;
60924ca0f25Sjob }
61024ca0f25Sjob if (siblings)
61124ca0f25Sjob path[n / 8] |= 1 << (n % 8);
61224ca0f25Sjob else
61324ca0f25Sjob path[n / 8] &= ~(1 << (n % 8));
61424ca0f25Sjob ki[src].prefix[n * 2] = siblings ? '|' : '`';
61524ca0f25Sjob ki[src].prefix[n * 2 + 1] = '-';
61624ca0f25Sjob n++;
61724ca0f25Sjob }
61824ca0f25Sjob strlcpy(ki[src].prefix + n * 2, "- ", (lvl - n) * 2 + 1);
61924ca0f25Sjob }
62024ca0f25Sjob free(path);
62124ca0f25Sjob }
62224ca0f25Sjob
62324ca0f25Sjob static void
usage(void)62428416801Sderaadt usage(void)
625df930be7Sderaadt {
62624ca0f25Sjob fprintf(stderr, "usage: %s [-AacefHhjkLlmrSTuvwx] [-M core] [-N system]"
62724ca0f25Sjob " [-O fmt] [-o fmt] [-p pid]\n", __progname);
628*52e706caSkn fprintf(stderr, "%-*s[-t tty] [-U user] [-W swap]\n",
62924ca0f25Sjob (int)strlen(__progname) + 8, "");
630df930be7Sderaadt exit(1);
631df930be7Sderaadt }
632