1/*	$NetBSD: main.c,v 1.81 2011/09/16 15:39:27 joerg Exp $	*/
2
3/*
4 * Copyright (c) 1983, 1988, 1993
5 *	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__COPYRIGHT("@(#) Copyright (c) 1983, 1988, 1993\
35 Regents of the University of California.  All rights reserved.");
36#endif /* not lint */
37
38#ifndef lint
39#if 0
40static char sccsid[] = "from: @(#)main.c	8.4 (Berkeley) 3/1/94";
41#else
42__RCSID("$NetBSD: main.c,v 1.81 2011/09/16 15:39:27 joerg Exp $");
43#endif
44#endif /* not lint */
45
46#include <sys/param.h>
47#include <sys/file.h>
48#include <sys/protosw.h>
49#include <sys/socket.h>
50
51#include <net/if.h>
52#include <netinet/in.h>
53
54#include <ctype.h>
55#include <err.h>
56#include <errno.h>
57#include <kvm.h>
58#include <limits.h>
59#include <netdb.h>
60#include <nlist.h>
61#include <paths.h>
62#include <stdio.h>
63#include <stdlib.h>
64#include <string.h>
65#include <unistd.h>
66#include "netstat.h"
67#include "prog_ops.h"
68
69struct nlist nl[] = {
70#define	N_MBSTAT	0
71	{ "_mbstat", 0, 0, 0, 0 },
72#define	N_IPSTAT	1
73	{ "_ipstat", 0, 0, 0, 0 },	/* not available via kvm */
74#define	N_TCBTABLE	2
75	{ "_tcbtable", 0, 0, 0, 0 },
76#define	N_TCPSTAT	3
77	{ "_tcpstat", 0, 0, 0, 0 },	/* not available via kvm */
78#define	N_UDBTABLE	4
79	{ "_udbtable", 0, 0, 0, 0 },
80#define	N_UDPSTAT	5
81	{ "_udpstat", 0, 0, 0, 0 },	/* not available via kvm */
82#define	N_IFNET		6
83	{ "_ifnet", 0, 0, 0, 0 },
84#define	N_IMP		7
85	{ "_imp_softc", 0, 0, 0, 0 },
86#define	N_ICMPSTAT	8
87	{ "_icmpstat", 0, 0, 0, 0 },	/* not available via kvm */
88#define	N_RTSTAT	9
89	{ "_rtstat", 0, 0, 0, 0 },
90#define	N_UNIXSW	10
91	{ "_unixsw", 0, 0, 0, 0 },
92#define	N_CLNPSTAT	11
93	{ "_clnp_stat", 0, 0, 0, 0 },
94#define	IN_NOTUSED	12
95	{ "_tp_inpcb", 0, 0, 0, 0 },
96#define	ISO_TP		13
97	{ "_tp_refinfo", 0, 0, 0, 0 },
98#define	N_TPSTAT	14
99	{ "_tp_stat", 0, 0, 0, 0 },
100#define	N_ESISSTAT	15
101	{ "_esis_stat", 0, 0, 0, 0 },
102#define N_NIMP		16
103	{ "_nimp", 0, 0, 0, 0 },
104#define N_RTREE		17
105	{ "_rt_tables", 0, 0, 0, 0 },
106#define N_CLTP		18
107	{ "_cltb", 0, 0, 0, 0 },
108#define N_CLTPSTAT	19
109	{ "_cltpstat", 0, 0, 0, 0 },
110#define	N_NFILE		20
111	{ "_nfile", 0, 0, 0, 0 },
112#define	N_FILE		21
113	{ "_file", 0, 0, 0, 0 },
114#define N_IGMPSTAT	22
115	{ "_igmpstat", 0, 0, 0, 0 },	/* not available via kvm */
116#define N_MRTPROTO	23
117	{ "_ip_mrtproto", 0, 0, 0, 0 },
118#define N_MRTSTAT	24
119	{ "_mrtstat", 0, 0, 0, 0 },
120#define N_MFCHASHTBL	25
121	{ "_mfchashtbl", 0, 0, 0, 0 },
122#define	N_MFCHASH	26
123	{ "_mfchash", 0, 0, 0, 0 },
124#define N_VIFTABLE	27
125	{ "_viftable", 0, 0, 0, 0 },
126#define N_MSIZE		28
127	{ "_msize", 0, 0, 0, 0 },
128#define N_MCLBYTES	29
129	{ "_mclbytes", 0, 0, 0, 0 },
130#define N_DDPSTAT	30
131	{ "_ddpstat", 0, 0, 0, 0 },	/* not available via kvm */
132#define N_DDPCB		31
133	{ "_ddpcb", 0, 0, 0, 0 },
134#define N_MBPOOL	32
135	{ "_mbpool", 0, 0, 0, 0 },
136#define N_MCLPOOL	33
137	{ "_mclpool", 0, 0, 0, 0 },
138#define N_DIVPCB	34
139	{ "_divcb", 0, 0, 0, 0 },
140#define N_DIVSTAT	35
141	{ "_divstat", 0, 0, 0, 0 },
142#define N_IP6STAT	36
143	{ "_ip6stat", 0, 0, 0, 0 },	/* not available via kvm */
144#define N_TCP6STAT	37
145	{ "_tcp6stat", 0, 0, 0, 0 },	/* not available via kvm */
146#define N_UDP6STAT	38
147	{ "_udp6stat", 0, 0, 0, 0 },	/* not available via kvm */
148#define N_ICMP6STAT	39
149	{ "_icmp6stat", 0, 0, 0, 0 },	/* not available via kvm */
150#define N_IPSECSTAT	40
151	{ "_ipsecstat", 0, 0, 0, 0 },	/* not available via kvm */
152#define N_IPSEC6STAT	41
153	{ "_ipsec6stat", 0, 0, 0, 0 },	/* not available via kvm */
154#define N_PIM6STAT	42
155	{ "_pim6stat", 0, 0, 0, 0 },	/* not available via kvm */
156#define N_MRT6PROTO	43
157	{ "_ip6_mrtproto", 0, 0, 0, 0 },
158#define N_MRT6STAT	44
159	{ "_mrt6stat", 0, 0, 0, 0 },
160#define N_MF6CTABLE	45
161	{ "_mf6ctable", 0, 0, 0, 0 },
162#define N_MIF6TABLE	46
163	{ "_mif6table", 0, 0, 0, 0 },
164#define N_PFKEYSTAT	47
165	{ "_pfkeystat", 0, 0, 0, 0 },	/* not available via kvm */
166#define N_ARPSTAT	48
167	{ "_arpstat", 0, 0, 0, 0 },	/* not available via kvm */
168#define N_RIP6STAT	49
169	{ "_rip6stat", 0, 0, 0, 0 },	/* not available via kvm */
170#define	N_ARPINTRQ	50
171	{ "_arpintrq", 0, 0, 0, 0 },
172#define	N_IPINTRQ	51
173	{ "_ipintrq", 0, 0, 0, 0 },
174#define	N_IP6INTRQ	52
175	{ "_ip6intrq", 0, 0, 0, 0 },
176#define	N_ATINTRQ1	53
177	{ "_atintrq1", 0, 0, 0, 0 },
178#define	N_ATINTRQ2	54
179	{ "_atintrq2", 0, 0, 0, 0 },
180#define	N_NSINTRQ	55
181	{ "_nsintrq", 0, 0, 0, 0 },
182#define	N_CLNLINTRQ	56
183	{ "_clnlintrq", 0, 0, 0, 0 },
184#define	N_LLCINTRQ	57
185	{ "_llcintrq", 0, 0, 0, 0 },
186#define	N_HDINTRQ	58
187	{ "_hdintrq", 0, 0, 0, 0 },
188#define	N_NATMINTRQ	59
189	{ "_natmintrq", 0, 0, 0, 0 },
190#define	N_PPPOEDISCINQ	61
191	{ "_ppoediscinq", 0, 0, 0, 0 },
192#define	N_PPPOEINQ	61
193	{ "_ppoeinq", 0, 0, 0, 0 },
194#define	N_PKINTRQ	62
195	{ "_pkintrq", 0, 0, 0, 0 },
196#define	N_HARDCLOCK_TICKS 63
197	{ "_hardclock_ticks", 0, 0, 0, 0 },
198#define N_PIMSTAT	64
199	{ "_pimstat", 0, 0, 0, 0 },
200#define N_CARPSTAT	65
201	{ "_carpstats", 0, 0, 0, 0 },	/* not available via kvm */
202#define N_PFSYNCSTAT	66
203	{ "_pfsyncstats", 0, 0, 0, 0},  /* not available via kvm */
204	{ "", 0, 0, 0, 0 },
205};
206
207struct protox {
208	u_char	pr_index;		/* index into nlist of cb head */
209	u_char	pr_sindex;		/* index into nlist of stat block */
210	u_char	pr_wanted;		/* 1 if wanted, 0 otherwise */
211	void	(*pr_cblocks)		/* control blocks printing routine */
212			__P((u_long, const char *));
213	void	(*pr_stats)		/* statistics printing routine */
214			__P((u_long, const char *));
215	void	(*pr_istats)
216			__P((const char *));	/* per/if statistics printing routine */
217	void	(*pr_dump)		/* PCB state dump routine */
218			__P((u_long, const char *, u_long));
219	const char *pr_name;		/* well-known name */
220} protox[] = {
221	{ N_TCBTABLE,	N_TCPSTAT,	1,	protopr,
222	  tcp_stats,	NULL,		tcp_dump,	"tcp" },
223	{ N_UDBTABLE,	N_UDPSTAT,	1,	protopr,
224	  udp_stats,	NULL,		0,	"udp" },
225	{ -1,		N_IPSTAT,	1,	0,
226	  ip_stats,	NULL,		0,	"ip" },
227	{ -1,		N_ICMPSTAT,	1,	0,
228	  icmp_stats,	NULL,		0,	"icmp" },
229	{ -1,		N_IGMPSTAT,	1,	0,
230	  igmp_stats,	NULL,		0,	"igmp" },
231	{ -1,		N_CARPSTAT,	1,	0,
232	  carp_stats,	NULL,		0,	"carp" },
233#ifdef IPSEC
234	{ -1,		N_IPSECSTAT,	1,	0,
235	  ipsec_switch,	NULL,		0,	"ipsec" },
236#endif
237	{ -1,		N_PIMSTAT,	1,	0,
238	  pim_stats,	NULL,		0,	"pim" },
239	{ -1,		N_PFSYNCSTAT,  1,  0,
240	  pfsync_stats,  NULL,		0,  "pfsync" },
241	{ -1,		-1,		0,	0,
242	  0,		NULL,		0,	0 }
243};
244
245#ifdef INET6
246struct protox ip6protox[] = {
247	{ -1,		N_IP6STAT,	1,	0,
248	  ip6_stats,	ip6_ifstats,	0,	"ip6" },
249	{ -1,		N_ICMP6STAT,	1,	0,
250	  icmp6_stats,	icmp6_ifstats,	0,	"icmp6" },
251#ifdef TCP6
252	{ N_TCBTABLE,	N_TCP6STAT,	1,	ip6protopr,
253	  tcp6_stats,	NULL,		tcp6_dump,	"tcp6" },
254#else
255	{ N_TCBTABLE,	N_TCP6STAT,	1,	ip6protopr,
256	  tcp_stats,	NULL,		tcp_dump,	"tcp6" },
257#endif
258	{ N_UDBTABLE,	N_UDP6STAT,	1,	ip6protopr,
259	  udp6_stats,	NULL,		0,	"udp6" },
260#ifdef IPSEC
261	{ -1,		N_IPSEC6STAT,	1,	0,
262	  ipsec_switch,	NULL,		0,	"ipsec6" },
263#endif
264	{ -1,		N_PIM6STAT,	1,	0,
265	  pim6_stats,	NULL,		0,	"pim6" },
266	{ -1,		N_RIP6STAT,	1,	0,
267	  rip6_stats,	NULL,		0,	"rip6" },
268	{ -1,		-1,		0,	0,
269	  0,		NULL,		0,	0 }
270};
271#endif
272
273struct protox arpprotox[] = {
274	{ -1,		N_ARPSTAT,	1,	0,
275	  arp_stats,	NULL,		0,	"arp" },
276	{ -1,		-1,		0,	0,
277	  0,		NULL,		0,	0 }
278};
279
280#ifdef IPSEC
281struct protox pfkeyprotox[] = {
282	{ -1,		N_PFKEYSTAT,	1,	0,
283	  pfkey_stats,	NULL,		0,	"pfkey" },
284	{ -1,		-1,		0,	0,
285	  0,		NULL,		0,	0 }
286};
287#endif
288
289#ifndef SMALL
290struct protox atalkprotox[] = {
291	{ N_DDPCB,	N_DDPSTAT,	1,	atalkprotopr,
292	  ddp_stats,	NULL,		0,	"ddp" },
293	{ -1,		-1,		0,	0,
294	  0,		NULL,		0,	NULL }
295};
296
297struct protox isoprotox[] = {
298	{ ISO_TP,	N_TPSTAT,	1,	iso_protopr,
299	  tp_stats,	NULL,		0,	"tp" },
300	{ N_CLTP,	N_CLTPSTAT,	1,	iso_protopr,
301	  cltp_stats,	NULL,		0,	"cltp" },
302	{ -1,		N_CLNPSTAT,	1,	 0,
303	  clnp_stats,	NULL,		0,	"clnp"},
304	{ -1,		N_ESISSTAT,	1,	 0,
305	  esis_stats,	NULL,		0,	"esis"},
306	{ -1,		-1,		0,	0,
307	  0,		NULL,		0,	0 }
308};
309#endif
310
311struct protox *protoprotox[] = { protox,
312#ifdef INET6
313				 ip6protox,
314#endif
315				 arpprotox,
316#ifdef IPSEC
317				 pfkeyprotox,
318#endif
319#ifndef SMALL
320				 atalkprotox,
321				 isoprotox,
322#endif
323				 NULL };
324
325const struct softintrq {
326	const char *siq_name;
327	int siq_index;
328} softintrq[] = {
329	{ "arpintrq", N_ARPINTRQ },
330	{ "ipintrq", N_IPINTRQ },
331	{ "ip6intrq", N_IP6INTRQ },
332	{ "atintrq1", N_ATINTRQ1 },
333	{ "atintrq2", N_ATINTRQ2 },
334	{ "clnlintrq", N_CLNLINTRQ },
335	{ "llcintrq", N_LLCINTRQ },
336	{ "hdintrq", N_HDINTRQ },
337	{ "natmintrq", N_NATMINTRQ },
338	{ "ppoediscinq", N_PPPOEDISCINQ },
339	{ "ppoeinq", N_PPPOEINQ },
340	{ "pkintrq", N_PKINTRQ },
341	{ NULL, -1 },
342};
343
344int main __P((int, char *[]));
345static void printproto __P((struct protox *, const char *));
346static void print_softintrq __P((void));
347__dead static void usage(void);
348static struct protox *name2protox __P((const char *));
349static struct protox *knownname __P((const char *));
350static void prepare(const char *, const char *, struct protox *tp);
351static kvm_t *prepare_kvmd(const char *, const char *, char *);
352
353static kvm_t *kvmd = NULL;
354gid_t egid;
355int interval;	/* repeat interval for i/f stats */
356static const char *nlistf = NULL, *memf = NULL;
357
358kvm_t *
359get_kvmd(void)
360{
361	char buf[_POSIX2_LINE_MAX];
362
363	if (kvmd != NULL)
364		return kvmd;
365	if ((kvmd = prepare_kvmd(nlistf, memf, buf)) == NULL)
366		errx(1, "kvm error: %s", buf);
367	return kvmd;
368}
369
370static kvm_t *
371prepare_kvmd(const char *nf, const char *mf, char *errbuf)
372{
373	kvm_t *k;
374
375	(void)setegid(egid);
376	k = kvm_openfiles(nf, mf, NULL, O_RDONLY, errbuf);
377	(void)setgid(getgid());
378	return k;
379}
380
381void
382prepare(const char *nf, const char *mf, struct protox *tp)
383{
384	char buf[_POSIX2_LINE_MAX];
385	/*
386	 * Try to figure out if we can use sysctl or not.
387	 */
388	if (nf != NULL && mf != NULL) {
389		/* Of course, we can't use sysctl with dumps. */
390		if (force_sysctl)
391			errx(EXIT_FAILURE, "can't use sysctl with dumps");
392
393		/* If we have -M and -N, we're not dealing with live memory. */
394		use_sysctl = 0;
395	} else if (qflag ||
396		   rflag ||
397		   iflag ||
398#ifndef SMALL
399		   gflag ||
400		   (pflag && tp->pr_sindex == N_TPSTAT) ||
401		   (pflag && tp->pr_sindex == N_CLTPSTAT) ||
402		   (pflag && tp->pr_sindex == N_CLNPSTAT) ||
403		   (pflag && tp->pr_sindex == N_ESISSTAT) ||
404#endif
405		   (pflag && tp->pr_sindex == N_PIMSTAT) ||
406		   Pflag) {
407		/* These flags are not yet supported via sysctl(3). */
408		use_sysctl = 0;
409	} else {
410		/* We can use sysctl(3). */
411		use_sysctl = 1;
412	}
413
414	if (force_sysctl && !use_sysctl) {
415		/* Let the user know what's about to happen. */
416		warnx("forcing sysctl usage even though it might not be "\
417		    "supported");
418		use_sysctl = 1;
419	}
420
421	kvmd = prepare_kvmd(nf, mf, buf);
422
423	if (!use_sysctl) {
424
425		if (kvmd == NULL)
426			errx(1, "kvm error: %s", buf);
427		if (kvm_nlist(kvmd, nl) < 0 || nl[0].n_type == 0) {
428			if (nf)
429				errx(1, "%s: no namelist", nf);
430			else
431				errx(1, "no namelist");
432		}
433	} else
434		(void)setgid(getgid());
435}
436
437int
438main(argc, argv)
439	int argc;
440	char *argv[];
441{
442	struct protoent *p;
443	struct protox *tp;	/* for printing cblocks & stats */
444	int ch;
445	char *cp;
446	u_long pcbaddr;
447
448	if (prog_init) {
449		if (prog_init() == -1)
450			err(1, "init failed");
451		force_sysctl = 1; /* cheap trick */
452	}
453
454	egid = getegid();
455	(void)setegid(getgid());
456	tp = NULL;
457	af = AF_UNSPEC;
458	pcbaddr = 0;
459
460	while ((ch = getopt(argc, argv,
461	    "AabBdf:ghI:LliM:mN:nP:p:qrsStTuVvw:X")) != -1)
462		switch (ch) {
463		case 'A':
464			Aflag = 1;
465			break;
466		case 'a':
467			aflag = 1;
468			break;
469		case 'b':
470			bflag = 1;
471			break;
472		case 'B':
473			Bflag = 1;
474			break;
475		case 'd':
476			dflag = 1;
477			break;
478		case 'f':
479			if (strcmp(optarg, "inet") == 0)
480				af = AF_INET;
481			else if (strcmp(optarg, "inet6") == 0)
482				af = AF_INET6;
483			else if (strcmp(optarg, "arp") == 0)
484				af = AF_ARP;
485			else if (strcmp(optarg, "pfkey") == 0)
486				af = PF_KEY;
487			else if (strcmp(optarg, "unix") == 0
488			    || strcmp(optarg, "local") == 0)
489				af = AF_LOCAL;
490			else if (strcmp(optarg, "iso") == 0)
491				af = AF_ISO;
492			else if (strcmp(optarg, "atalk") == 0)
493				af = AF_APPLETALK;
494			else if (strcmp(optarg, "mpls") == 0)
495				af = AF_MPLS;
496			else
497				errx(1, "%s: unknown address family",
498				    optarg);
499			break;
500#ifndef SMALL
501		case 'g':
502			gflag = 1;
503			break;
504#endif
505		case 'h':
506			hflag = 1;
507			break;
508		case 'I':
509			iflag = 1;
510			interface = optarg;
511			break;
512		case 'i':
513			iflag = 1;
514			break;
515		case 'L':
516			Lflag = 1;
517			break;
518		case 'l':
519			lflag = 1;
520			break;
521		case 'M':
522			memf = optarg;
523			break;
524		case 'm':
525			mflag = 1;
526			break;
527		case 'N':
528			nlistf = optarg;
529			break;
530		case 'n':
531			numeric_addr = numeric_port = nflag = 1;
532			break;
533		case 'P':
534			errno = 0;
535			pcbaddr = strtoul(optarg, &cp, 16);
536			if (*cp != '\0' || errno == ERANGE)
537				errx(1, "invalid PCB address %s",
538				    optarg);
539			Pflag = 1;
540			break;
541		case 'p':
542			if ((tp = name2protox(optarg)) == NULL)
543				errx(1, "%s: unknown or uninstrumented protocol",
544				    optarg);
545			pflag = 1;
546			break;
547		case 'q':
548			qflag = 1;
549			break;
550		case 'r':
551			rflag = 1;
552			break;
553		case 's':
554			++sflag;
555			break;
556		case 'S':
557			numeric_addr = 1;
558			break;
559		case 't':
560			tflag = 1;
561			break;
562		case 'T':
563			tagflag = 1;
564			break;
565		case 'u':
566			af = AF_LOCAL;
567			break;
568		case 'V':
569			Vflag++;
570			break;
571		case 'v':
572			vflag++;
573			break;
574		case 'w':
575			interval = atoi(optarg);
576			iflag = 1;
577			break;
578		case 'X':
579			force_sysctl = 1;
580			break;
581		case '?':
582		default:
583			usage();
584		}
585	argv += optind;
586	argc -= optind;
587
588#define	BACKWARD_COMPATIBILITY
589#ifdef	BACKWARD_COMPATIBILITY
590	if (*argv) {
591		if (isdigit((unsigned char)**argv)) {
592			interval = atoi(*argv);
593			if (interval <= 0)
594				usage();
595			++argv;
596			iflag = 1;
597		}
598		if (*argv) {
599			nlistf = *argv;
600			if (*++argv)
601				memf = *argv;
602		}
603	}
604#endif
605
606	prepare(nlistf, memf, tp);
607
608#ifndef SMALL
609	if (Bflag) {
610		if (sflag)
611			bpf_stats();
612		else
613			bpf_dump(interface);
614		exit(0);
615	}
616#endif
617
618	if (mflag) {
619		mbpr(nl[N_MBSTAT].n_value,  nl[N_MSIZE].n_value,
620		    nl[N_MCLBYTES].n_value, nl[N_MBPOOL].n_value,
621		    nl[N_MCLPOOL].n_value);
622		exit(0);
623	}
624	if (Pflag) {
625		if (tp == NULL) {
626			/* Default to TCP. */
627			tp = name2protox("tcp");
628		}
629		if (tp->pr_dump)
630			(*tp->pr_dump)(nl[tp->pr_index].n_value, tp->pr_name,
631			    pcbaddr);
632		else
633			printf("%s: no PCB dump routine\n", tp->pr_name);
634		exit(0);
635	}
636	if (pflag) {
637		if (iflag && tp->pr_istats)
638			intpr(interval, nl[N_IFNET].n_value, tp->pr_istats);
639		else if (tp->pr_stats)
640			(*tp->pr_stats)(nl[tp->pr_sindex].n_value,
641				tp->pr_name);
642		else
643			printf("%s: no stats routine\n", tp->pr_name);
644		exit(0);
645	}
646	if (qflag) {
647		print_softintrq();
648		exit(0);
649	}
650	/*
651	 * Keep file descriptors open to avoid overhead
652	 * of open/close on each call to get* routines.
653	 */
654	sethostent(1);
655	setnetent(1);
656	if (iflag) {
657		if (af != AF_UNSPEC)
658			goto protostat;
659
660		intpr(interval, nl[N_IFNET].n_value, NULL);
661		exit(0);
662	}
663	if (rflag) {
664		if (sflag)
665			rt_stats(use_sysctl ? 0 : nl[N_RTSTAT].n_value);
666		else {
667			if (use_sysctl)
668				p_rttables(af);
669			else
670				routepr(nl[N_RTREE].n_value);
671		}
672		exit(0);
673	}
674#ifndef SMALL
675	if (gflag) {
676		if (sflag) {
677			if (af == AF_INET || af == AF_UNSPEC)
678				mrt_stats(nl[N_MRTPROTO].n_value,
679					  nl[N_MRTSTAT].n_value);
680#ifdef INET6
681			if (af == AF_INET6 || af == AF_UNSPEC)
682				mrt6_stats(nl[N_MRT6PROTO].n_value,
683					   nl[N_MRT6STAT].n_value);
684#endif
685		}
686		else {
687			if (af == AF_INET || af == AF_UNSPEC)
688				mroutepr(nl[N_MRTPROTO].n_value,
689					 nl[N_MFCHASHTBL].n_value,
690					 nl[N_MFCHASH].n_value,
691					 nl[N_VIFTABLE].n_value);
692#ifdef INET6
693			if (af == AF_INET6 || af == AF_UNSPEC)
694				mroute6pr(nl[N_MRT6PROTO].n_value,
695					  nl[N_MF6CTABLE].n_value,
696					  nl[N_MIF6TABLE].n_value);
697#endif
698		}
699		exit(0);
700	}
701#endif
702  protostat:
703	if (af == AF_INET || af == AF_UNSPEC) {
704		setprotoent(1);
705		setservent(1);
706		/* ugh, this is O(MN) ... why do we do this? */
707		while ((p = getprotoent()) != NULL) {
708			for (tp = protox; tp->pr_name; tp++)
709				if (strcmp(tp->pr_name, p->p_name) == 0)
710					break;
711			if (tp->pr_name == 0 || tp->pr_wanted == 0)
712				continue;
713			printproto(tp, p->p_name);
714			tp->pr_wanted = 0;
715		}
716		endprotoent();
717		for (tp = protox; tp->pr_name; tp++)
718			if (tp->pr_wanted)
719				printproto(tp, tp->pr_name);
720	}
721#ifdef INET6
722	if (af == AF_INET6 || af == AF_UNSPEC)
723		for (tp = ip6protox; tp->pr_name; tp++)
724			printproto(tp, tp->pr_name);
725#endif
726	if (af == AF_ARP || af == AF_UNSPEC)
727		for (tp = arpprotox; tp->pr_name; tp++)
728			printproto(tp, tp->pr_name);
729#ifdef IPSEC
730	if (af == PF_KEY || af == AF_UNSPEC)
731		for (tp = pfkeyprotox; tp->pr_name; tp++)
732			printproto(tp, tp->pr_name);
733#endif
734#ifndef SMALL
735	if (af == AF_APPLETALK || af == AF_UNSPEC)
736		for (tp = atalkprotox; tp->pr_name; tp++)
737			printproto(tp, tp->pr_name);
738	if (af == AF_ISO || af == AF_UNSPEC)
739		for (tp = isoprotox; tp->pr_name; tp++)
740			printproto(tp, tp->pr_name);
741	if ((af == AF_LOCAL || af == AF_UNSPEC) && !sflag)
742		unixpr(nl[N_UNIXSW].n_value);
743#endif
744	exit(0);
745}
746
747/*
748 * Print out protocol statistics or control blocks (per sflag).
749 * If the interface was not specifically requested, and the symbol
750 * is not in the namelist, ignore this one.
751 */
752static void
753printproto(tp, name)
754	struct protox *tp;
755	const char *name;
756{
757	void (*pr) __P((u_long, const char *));
758	u_long off;
759
760	if (sflag) {
761		if (iflag) {
762			if (tp->pr_istats)
763				intpr(interval, nl[N_IFNET].n_value,
764				      tp->pr_istats);
765			return;
766		}
767		else {
768			pr = tp->pr_stats;
769			off = nl[tp->pr_sindex].n_value;
770		}
771	} else {
772		pr = tp->pr_cblocks;
773		off = nl[tp->pr_index].n_value;
774	}
775	if (pr != NULL && ((off || af != AF_UNSPEC) || use_sysctl)) {
776		(*pr)(off, name);
777	}
778}
779
780/*
781 * Print softintrq status.
782 */
783void
784print_softintrq()
785{
786	struct ifqueue intrq, *ifq = &intrq;
787	const struct softintrq *siq;
788	u_long off;
789
790	for (siq = softintrq; siq->siq_name != NULL; siq++) {
791		off = nl[siq->siq_index].n_value;
792		if (off == 0)
793			continue;
794
795		kread(off, (char *)ifq, sizeof(*ifq));
796		printf("%s:\n", siq->siq_name);
797		printf("\tqueue length: %d\n", ifq->ifq_len);
798		printf("\tmaximum queue length: %d\n", ifq->ifq_maxlen);
799		printf("\tpackets dropped: %d\n", ifq->ifq_drops);
800	}
801}
802
803/*
804 * Read kernel memory, return 0 on success.
805 */
806int
807kread(addr, buf, size)
808	u_long addr;
809	char *buf;
810	int size;
811{
812
813	if (kvm_read(kvmd, addr, buf, size) != size) {
814		warnx("%s", kvm_geterr(kvmd));
815		return (-1);
816	}
817	return (0);
818}
819
820const char *
821plural(n)
822	int n;
823{
824
825	return (n != 1 ? "s" : "");
826}
827
828const char *
829plurales(n)
830	int n;
831{
832
833	return (n != 1 ? "es" : "");
834}
835
836int
837get_hardticks(void)
838{
839	int hardticks;
840
841	kread(nl[N_HARDCLOCK_TICKS].n_value, (char *)&hardticks,
842	    sizeof(hardticks));
843	return (hardticks);
844}
845
846/*
847 * Find the protox for the given "well-known" name.
848 */
849static struct protox *
850knownname(name)
851	const char *name;
852{
853	struct protox **tpp, *tp;
854
855	for (tpp = protoprotox; *tpp; tpp++)
856		for (tp = *tpp; tp->pr_name; tp++)
857			if (strcmp(tp->pr_name, name) == 0)
858				return (tp);
859	return (NULL);
860}
861
862/*
863 * Find the protox corresponding to name.
864 */
865static struct protox *
866name2protox(name)
867	const char *name;
868{
869	struct protox *tp;
870	char **alias;			/* alias from p->aliases */
871	struct protoent *p;
872
873	/*
874	 * Try to find the name in the list of "well-known" names. If that
875	 * fails, check if name is an alias for an Internet protocol.
876	 */
877	if ((tp = knownname(name)) != NULL)
878		return (tp);
879
880	setprotoent(1);			/* make protocol lookup cheaper */
881	while ((p = getprotoent()) != NULL) {
882		/* assert: name not same as p->name */
883		for (alias = p->p_aliases; *alias; alias++)
884			if (strcmp(name, *alias) == 0) {
885				endprotoent();
886				return (knownname(p->p_name));
887			}
888	}
889	endprotoent();
890	return (NULL);
891}
892
893static void
894usage()
895{
896	const char *progname = getprogname();
897
898	(void)fprintf(stderr,
899"usage: %s [-Aan] [-f address_family] [-M core] [-N system]\n", progname);
900	(void)fprintf(stderr,
901"       %s [-bdgiLmnqrsSv] [-f address_family] [-M core] [-N system]\n",
902	progname);
903	(void)fprintf(stderr,
904"       %s [-dn] [-I interface] [-M core] [-N system] [-w wait]\n", progname);
905	(void)fprintf(stderr,
906"       %s [-p protocol] [-M core] [-N system]\n", progname);
907	(void)fprintf(stderr,
908"       %s [-p protocol] [-M core] [-N system] -P pcbaddr\n", progname);
909	(void)fprintf(stderr,
910"       %s [-p protocol] [-i] [-I Interface] \n", progname);
911	(void)fprintf(stderr,
912"       %s [-s] [-f address_family] [-i] [-I Interface]\n", progname);
913	(void)fprintf(stderr,
914"       %s [-s] [-B] [-I interface]\n", progname);
915	exit(1);
916}
917