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