1/*	$NetBSD: keyword.c,v 1.53 2009/10/21 21:11:57 rmind Exp $	*/
2
3/*-
4 * Copyright (c) 1990, 1993, 1994
5 *	The Regents of the University of California.  All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of the University nor the names of its contributors
16 *    may be used to endorse or promote products derived from this software
17 *    without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32#include <sys/cdefs.h>
33#ifndef lint
34#if 0
35static char sccsid[] = "@(#)keyword.c	8.5 (Berkeley) 4/2/94";
36#else
37__RCSID("$NetBSD: keyword.c,v 1.53 2009/10/21 21:11:57 rmind Exp $");
38#endif
39#endif /* not lint */
40
41#include <sys/param.h>
42#include <sys/time.h>
43#include <sys/lwp.h>
44#include <sys/proc.h>
45#include <sys/resource.h>
46#include <sys/sysctl.h>
47#include <sys/ucred.h>
48
49#include <err.h>
50#include <errno.h>
51#include <kvm.h>
52#include <stddef.h>
53#include <stdio.h>
54#include <stdlib.h>
55#include <string.h>
56#include <signal.h>
57
58#include "ps.h"
59
60static VAR *findvar(const char *);
61static int  vcmp(const void *, const void *);
62
63#if 0 	/* kernel doesn't calculate these */
64	PUVAR("idrss", "IDRSS", 0, p_uru_idrss, UINT64, PRIu64),
65	PUVAR("isrss", "ISRSS", 0, p_uru_isrss, UINT64, PRId64),
66	PUVAR("ixrss", "IXRSS", 0, p_uru_ixrss, UINT64, PRId64),
67	PUVAR("maxrss", "MAXRSS", 0, p_uru_maxrss, UINT64, PRIu64),
68#endif
69
70/* Compute offset in common structures. */
71#define	POFF(x)	offsetof(struct kinfo_proc2, x)
72#define	LOFF(x)	offsetof(struct kinfo_lwp, x)
73
74#define	UIDFMT	"u"
75#define	UID(n1, n2, of) \
76	{ .name = n1, .header = n2, .flag = 0, .oproc = pvar, \
77	  .off = POFF(of), .type = UINT32, .fmt = UIDFMT }
78#define	GID(n1, n2, off)	UID(n1, n2, off)
79
80#define	PIDFMT	"d"
81#define	PID(n1, n2, of) \
82	{ .name = n1, .header = n2, .flag = 0, .oproc = pvar, \
83	  .off = POFF(of), .type = INT32, .fmt = PIDFMT }
84
85#define	LVAR(n1, n2, fl, of, ty, fm) \
86	{ .name = n1, .header = n2, .flag = (fl) | LWP, .oproc = pvar, \
87	  .off = LOFF(of), .type = ty, .fmt = fm }
88#define	PVAR(n1, n2, fl, of, ty, fm) \
89	{ .name = n1, .header = n2, .flag = (fl) | 0, .oproc = pvar, \
90	  .off = POFF(of), .type = ty, .fmt = fm }
91#define	PUVAR(n1, n2, fl, of, ty, fm) \
92	{ .name = n1, .header = n2, .flag = (fl) | UAREA, .oproc = pvar, \
93	  .off = POFF(of), .type = ty, .fmt = fm }
94#define VAR3(n1, n2, fl) \
95	{ .name = n1, .header = n2, .flag = fl }
96#define VAR4(n1, n2, fl, op) \
97	{ .name = n1, .header = n2, .flag = fl, .oproc = op, }
98#define VAR6(n1, n2, fl, op, of, ty) \
99	{ .name = n1, .header = n2, .flag = fl, .oproc = op, \
100	  .off = of, .type = ty }
101
102/* NB: table must be sorted, in vi use:
103 *	:/^VAR/,/end_sort/! sort -t\" +1
104 * breaking long lines just makes the sort harder
105 *
106 * We support all the fields required by P1003.1-2004 (SUSv3), with
107 * the correct default headers, except for the "tty" field, where the
108 * standard says the header should be "TT", but we have "TTY".
109 */
110VAR var[] = {
111	VAR6("%cpu", "%CPU", 0, pcpu, 0, PCPU),
112	VAR6("%mem", "%MEM", 0, pmem, POFF(p_vm_rssize), INT32),
113	PVAR("acflag", "ACFLG", 0, p_acflag, USHORT, "x"),
114	VAR3("acflg", "acflag", ALIAS),
115	VAR3("args", "command", ALIAS),
116	VAR3("blocked", "sigmask", ALIAS),
117	VAR3("caught", "sigcatch", ALIAS),
118	VAR4("comm", "COMMAND", COMM|ARGV0|LJUST, command),
119	VAR4("command", "COMMAND", COMM|LJUST, command),
120	PVAR("cpu", "CPU", 0, p_estcpu, UINT, "u"),
121	VAR4("cpuid", "CPUID", LWP, cpuid),
122	VAR3("cputime", "time", ALIAS),
123	VAR6("ctime", "CTIME", 0, putimeval, POFF(p_uctime_sec), TIMEVAL),
124	GID("egid", "EGID", p_gid),
125	VAR4("egroup", "EGROUP", LJUST, gname),
126	VAR4("emul", "EMUL", LJUST, emul),
127	VAR6("etime", "ELAPSED", 0, elapsed, POFF(p_ustart_sec), TIMEVAL),
128	UID("euid", "EUID", p_uid),
129	VAR4("euser", "EUSER", LJUST, uname),
130	PVAR("f", "F", 0, p_flag, INT, "x"),
131	VAR3("flags", "f", ALIAS),
132	GID("gid", "GID", p_gid),
133	VAR4("group", "GROUP", LJUST, gname),
134	VAR4("groupnames", "GROUPNAMES", LJUST, groupnames),
135	VAR4("groups", "GROUPS", LJUST, groups),
136	/* holdcnt: unused, left for compat. */
137	LVAR("holdcnt", "HOLDCNT", 0, l_holdcnt, INT, "d"),
138	VAR3("ignored", "sigignore", ALIAS),
139	PUVAR("inblk", "INBLK", 0, p_uru_inblock, UINT64, PRIu64),
140	VAR3("inblock", "inblk", ALIAS),
141	PVAR("jobc", "JOBC", 0, p_jobc, SHORT, "d"),
142	PVAR("ktrace", "KTRACE", 0, p_traceflag, INT, "x"),
143/*XXX*/	PVAR("ktracep", "KTRACEP", 0, p_tracep, KPTR, PRIx64),
144	LVAR("laddr", "LADDR", 0, l_laddr, KPTR, PRIx64),
145	LVAR("lid", "LID", 0, l_lid, INT32, "d"),
146	VAR4("lim", "LIM", 0, maxrss),
147	VAR4("lname", "LNAME", LJUST|LWP, lname),
148	VAR4("login", "LOGIN", LJUST, logname),
149	VAR3("logname", "login", ALIAS),
150	VAR6("lstart", "STARTED", LJUST, lstarted, POFF(p_ustart_sec), UINT32),
151	VAR4("lstate", "STAT", LJUST|LWP, lstate),
152	PUVAR("majflt", "MAJFLT", 0, p_uru_majflt, UINT64, PRIu64),
153	PUVAR("minflt", "MINFLT", 0, p_uru_minflt, UINT64, PRIu64),
154	PUVAR("msgrcv", "MSGRCV", 0, p_uru_msgrcv, UINT64, PRIu64),
155	PUVAR("msgsnd", "MSGSND", 0, p_uru_msgsnd, UINT64, PRIu64),
156	VAR3("ni", "nice", ALIAS),
157	VAR6("nice", "NI", 0, pnice, POFF(p_nice), UCHAR),
158	PUVAR("nivcsw", "NIVCSW", 0, p_uru_nivcsw, UINT64, PRIu64),
159	PVAR("nlwp", "NLWP", 0, p_nlwps, UINT64, PRId64),
160	VAR3("nsignals", "nsigs", ALIAS),
161	PUVAR("nsigs", "NSIGS", 0, p_uru_nsignals, UINT64, PRIu64),
162	/* nswap: unused, left for compat. */
163	PUVAR("nswap", "NSWAP", 0, p_uru_nswap, UINT64, PRIu64),
164	PUVAR("nvcsw", "NVCSW", 0, p_uru_nvcsw, UINT64, PRIu64),
165/*XXX*/	LVAR("nwchan", "WCHAN", 0, l_wchan, KPTR, PRIx64),
166	PUVAR("oublk", "OUBLK", 0, p_uru_oublock, UINT64, PRIu64),
167	VAR3("oublock", "oublk", ALIAS),
168/*XXX*/	PVAR("p_ru", "P_RU", 0, p_ru, KPTR, PRIx64),
169/*XXX*/	PVAR("paddr", "PADDR", 0, p_paddr, KPTR, PRIx64),
170	PUVAR("pagein", "PAGEIN", 0, p_uru_majflt, UINT64, PRIu64),
171	VAR3("pcpu", "%cpu", ALIAS),
172	VAR3("pending", "sig", ALIAS),
173	PID("pgid", "PGID", p__pgid),
174	PID("pid", "PID", p_pid),
175	VAR3("pmem", "%mem", ALIAS),
176	PID("ppid", "PPID", p_ppid),
177	VAR4("pri", "PRI", LWP, pri),
178	LVAR("re", "RE", INF127, l_swtime, UINT, "u"),
179	GID("rgid", "RGID", p_rgid),
180	VAR4("rgroup", "RGROUP", LJUST, rgname),
181/*XXX*/	LVAR("rlink", "RLINK", 0, l_back, KPTR, PRIx64),
182	PVAR("rlwp", "RLWP", 0, p_nrlwps, UINT64, PRId64),
183	VAR6("rss", "RSS", 0, p_rssize, POFF(p_vm_rssize), INT32),
184	VAR3("rssize", "rsz", ALIAS),
185	VAR6("rsz", "RSZ", 0, rssize, POFF(p_vm_rssize), INT32),
186	UID("ruid", "RUID", p_ruid),
187	VAR4("ruser", "RUSER", LJUST, runame),
188	PVAR("sess", "SESS", 0, p_sess, KPTR24, PRIx64),
189	PID("sid", "SID", p_sid),
190	PVAR("sig", "PENDING", 0, p_siglist, SIGLIST, "s"),
191	PVAR("sigcatch", "CAUGHT", 0, p_sigcatch, SIGLIST, "s"),
192	PVAR("sigignore", "IGNORED", 0, p_sigignore, SIGLIST, "s"),
193	PVAR("sigmask", "BLOCKED", 0, p_sigmask, SIGLIST, "s"),
194	LVAR("sl", "SL", INF127, l_slptime, UINT, "u"),
195	VAR6("start", "STARTED", 0, started, POFF(p_ustart_sec), UINT32),
196	VAR3("stat", "state", ALIAS),
197	VAR4("state", "STAT", LJUST, state),
198	VAR6("stime", "STIME", 0, putimeval, POFF(p_ustime_sec), TIMEVAL),
199	GID("svgid", "SVGID", p_svgid),
200	VAR4("svgroup", "SVGROUP", LJUST, svgname),
201	UID("svuid", "SVUID", p_svuid),
202	VAR4("svuser", "SVUSER", LJUST, svuname),
203	/* "tdev" is UINT32, but we do this for sorting purposes */
204	VAR6("tdev", "TDEV", 0, tdev, POFF(p_tdev), INT32),
205	VAR6("time", "TIME", 0, cputime, 0, CPUTIME),
206	PID("tpgid", "TPGID", p_tpgid),
207	PVAR("tsess", "TSESS", 0, p_tsess, KPTR, PRIx64),
208	VAR6("tsiz", "TSIZ", 0, tsize, POFF(p_vm_tsize), INT32),
209	VAR6("tt", "TTY", LJUST, tname, POFF(p_tdev), INT32),
210	VAR6("tty", "TTY", LJUST, longtname, POFF(p_tdev), INT32),
211	LVAR("uaddr", "UADDR", 0, l_addr, KPTR, PRIx64),
212	VAR4("ucomm", "UCOMM", LJUST, ucomm),
213	UID("uid", "UID", p_uid),
214	LVAR("upr", "UPR", 0, l_usrpri, UCHAR, "u"),
215	VAR4("user", "USER", LJUST, uname),
216	VAR3("usrpri", "upr", ALIAS),
217	VAR6("utime", "UTIME", 0, putimeval, POFF(p_uutime_sec), TIMEVAL),
218	VAR3("vsize", "vsz", ALIAS),
219	VAR6("vsz", "VSZ", 0, vsize, 0, VSIZE),
220	VAR4("wchan", "WCHAN", LJUST|LWP, wchan),
221	PVAR("xstat", "XSTAT", 0, p_xstat, USHORT, "x"),
222/* "zzzz" end_sort */
223	{ .name = "" },
224};
225
226void
227showkey(void)
228{
229	VAR *v;
230	int i;
231	const char *p;
232	const char *sep;
233
234	i = 0;
235	sep = "";
236	for (v = var; *(p = v->name); ++v) {
237		int len = strlen(p);
238		if (termwidth && (i += len + 1) > termwidth) {
239			i = len;
240			sep = "\n";
241		}
242		(void)printf("%s%s", sep, p);
243		sep = " ";
244	}
245	(void)printf("\n");
246}
247
248/*
249 * Parse the string pp, and insert or append entries to the list
250 * referenced by listptr.  If pos in non-null and *pos is non-null, then
251 * *pos specifies where to insert (instead of appending).  If pos is
252 * non-null, then a new value is returned through *pos referring to the
253 * last item inserted.
254 */
255static void
256parsevarlist(const char *pp, struct varlist *listptr, struct varent **pos)
257{
258	char *p, *sp, *equalsp;
259
260	/* dup to avoid zapping arguments.  We will free sp later. */
261	p = sp = strdup(pp);
262
263	/*
264	 * Everything after the first '=' is part of a custom header.
265	 * Temporarily replace it with '\0' to simplify other code.
266	 */
267	equalsp = strchr(p, '=');
268	if (equalsp)
269	    *equalsp = '\0';
270
271#define	FMTSEP	" \t,\n"
272	while (p && *p) {
273		char *cp;
274		VAR *v;
275		struct varent *vent;
276
277		/*
278		 * skip separators before the first keyword, and
279		 * look for the separator after the keyword.
280		 */
281		for (cp = p; *cp != '\0'; cp++) {
282		    p = strpbrk(cp, FMTSEP);
283		    if (p != cp)
284			break;
285		}
286		if (*cp == '\0')
287		    break;
288		/*
289		 * Now cp points to the start of a keyword,
290		 * and p is NULL or points past the end of the keyword.
291		 *
292		 * Terminate the keyword with '\0', or reinstate the
293		 * '=' that was removed earlier, if appropriate.
294		 */
295		if (p) {
296			*p = '\0';
297			p++;
298		} else if (equalsp) {
299			*equalsp = '=';
300		}
301
302		/*
303		 * If findvar() likes the keyword or keyword=header,
304		 * add it to our list.  If findvar() doesn't like it,
305		 * it will print a warning, so we ignore it.
306		 */
307		if ((v = findvar(cp)) == NULL)
308			continue;
309		if ((vent = malloc(sizeof(struct varent))) == NULL)
310			err(1, NULL);
311		vent->var = v;
312		if (pos && *pos)
313		    SIMPLEQ_INSERT_AFTER(listptr, *pos, vent, next);
314		else {
315		    SIMPLEQ_INSERT_TAIL(listptr, vent, next);
316		}
317		if (pos)
318		    *pos = vent;
319	}
320 	free(sp);
321	if (SIMPLEQ_EMPTY(listptr))
322		errx(1, "no valid keywords");
323}
324
325void
326parsefmt(const char *p)
327{
328
329	parsevarlist(p, &displaylist, NULL);
330}
331
332void
333parsefmt_insert(const char *p, struct varent **pos)
334{
335
336	parsevarlist(p, &displaylist, pos);
337}
338
339void
340parsesort(const char *p)
341{
342
343	parsevarlist(p, &sortlist, NULL);
344}
345
346/* Search through a list for an entry with a specified name. */
347struct varent *
348varlist_find(struct varlist *list, const char *name)
349{
350	struct varent *vent;
351
352	SIMPLEQ_FOREACH(vent, list, next) {
353		if (strcmp(vent->var->name, name) == 0)
354			break;
355	}
356	return vent;
357}
358
359static VAR *
360findvar(const char *p)
361{
362	VAR *v;
363	char *hp;
364
365	hp = strchr(p, '=');
366	if (hp)
367		*hp++ = '\0';
368
369	v = bsearch(p, var, sizeof(var)/sizeof(VAR) - 1, sizeof(VAR), vcmp);
370	if (v && v->flag & ALIAS)
371		v = findvar(v->header);
372	if (!v) {
373		warnx("%s: keyword not found", p);
374		eval = 1;
375		return NULL;
376	}
377
378	if (v && hp) {
379		/*
380		 * Override the header.
381		 *
382		 * We need to copy the entry first, and override the
383		 * header in the copy, because the same field might be
384		 * used multiple times with different headers.  We also
385		 * need to strdup the header.
386		 */
387		struct var *newvar;
388		char *newheader;
389
390		if ((newvar = malloc(sizeof(struct var))) == NULL)
391			err(1, NULL);
392		if ((newheader = strdup(hp)) == NULL)
393			err(1, NULL);
394		memcpy(newvar, v, sizeof(struct var));
395		newvar->header = newheader;
396
397		/*
398		 * According to P1003.1-2004, if the header text is null,
399		 * such as -o user=, the field width will be at least as
400		 * wide as the default header text.
401		 */
402		if (*hp == '\0')
403			newvar->width = strlen(v->header);
404
405		v = newvar;
406	}
407	return v;
408}
409
410static int
411vcmp(const void *a, const void *b)
412{
413        return strcmp(a, ((const VAR *)b)->name);
414}
415