xref: /freebsd/usr.bin/netstat/main.c (revision e14ddd1f16e7e5788392c50de21ea7c927e0690c)
1 /*-
2  * Copyright (c) 1983, 1988, 1993
3  *	Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 4. Neither the name of the University nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  */
29 
30 #ifndef lint
31 char const copyright[] =
32 "@(#) Copyright (c) 1983, 1988, 1993\n\
33 	Regents of the University of California.  All rights reserved.\n";
34 #endif /* not lint */
35 
36 #if 0
37 #ifndef lint
38 static char sccsid[] = "@(#)main.c	8.4 (Berkeley) 3/1/94";
39 #endif /* not lint */
40 #endif
41 
42 #include <sys/cdefs.h>
43 __FBSDID("$FreeBSD$");
44 
45 #include <sys/param.h>
46 #include <sys/file.h>
47 #include <sys/protosw.h>
48 #include <sys/socket.h>
49 #include <sys/socketvar.h>
50 
51 #include <netinet/in.h>
52 
53 #ifdef NETGRAPH
54 #include <netgraph/ng_socket.h>
55 #endif
56 
57 #include <ctype.h>
58 #include <err.h>
59 #include <errno.h>
60 #include <kvm.h>
61 #include <limits.h>
62 #include <netdb.h>
63 #include <nlist.h>
64 #include <paths.h>
65 #include <stdint.h>
66 #include <stdio.h>
67 #include <stdlib.h>
68 #include <string.h>
69 #include <unistd.h>
70 #include "netstat.h"
71 
72 static struct nlist nl[] = {
73 #define	N_IFNET		0
74 	{ .n_name = "_ifnet" },
75 #define	N_RTSTAT	1
76 	{ .n_name = "_rtstat" },
77 #define	N_RTREE		2
78 	{ .n_name = "_rt_tables"},
79 #define	N_MRTSTAT	3
80 	{ .n_name = "_mrtstat" },
81 #define	N_MFCHASHTBL	4
82 	{ .n_name = "_mfchashtbl" },
83 #define	N_VIFTABLE	5
84 	{ .n_name = "_viftable" },
85 #define	N_IPX		6
86 	{ .n_name = "_ipxpcb_list"},
87 #define	N_IPXSTAT	7
88 	{ .n_name = "_ipxstat"},
89 #define	N_SPXSTAT	8
90 	{ .n_name = "_spx_istat"},
91 #define	N_DDPSTAT	9
92 	{ .n_name = "_ddpstat"},
93 #define	N_DDPCB		10
94 	{ .n_name = "_ddpcb"},
95 #define	N_NGSOCKS	11
96 	{ .n_name = "_ngsocklist"},
97 #define	N_IP6STAT	12
98 	{ .n_name = "_ip6stat" },
99 #define	N_ICMP6STAT	13
100 	{ .n_name = "_icmp6stat" },
101 #define	N_IPSECSTAT	14
102 	{ .n_name = "_ipsec4stat" },
103 #define	N_IPSEC6STAT	15
104 	{ .n_name = "_ipsec6stat" },
105 #define	N_PIM6STAT	16
106 	{ .n_name = "_pim6stat" },
107 #define	N_MRT6STAT	17
108 	{ .n_name = "_mrt6stat" },
109 #define	N_MF6CTABLE	18
110 	{ .n_name = "_mf6ctable" },
111 #define	N_MIF6TABLE	19
112 	{ .n_name = "_mif6table" },
113 #define	N_PFKEYSTAT	20
114 	{ .n_name = "_pfkeystat" },
115 #define	N_MBSTAT	21
116 	{ .n_name = "_mbstat" },
117 #define	N_MBTYPES	22
118 	{ .n_name = "_mbtypes" },
119 #define	N_NMBCLUSTERS	23
120 	{ .n_name = "_nmbclusters" },
121 #define	N_NMBUFS	24
122 	{ .n_name = "_nmbufs" },
123 #define	N_MBHI		25
124 	{ .n_name = "_mbuf_hiwm" },
125 #define	N_CLHI		26
126 	{ .n_name = "_clust_hiwm" },
127 #define	N_NCPUS		27
128 	{ .n_name = "_smp_cpus" },
129 #define	N_PAGESZ	28
130 	{ .n_name = "_pagesize" },
131 #define	N_MBPSTAT	29
132 	{ .n_name = "_mb_statpcpu" },
133 #define	N_RTTRASH	30
134 	{ .n_name = "_rttrash" },
135 #define	N_MBLO		31
136 	{ .n_name = "_mbuf_lowm" },
137 #define	N_CLLO		32
138 	{ .n_name = "_clust_lowm" },
139 #define	N_CARPSTAT	33
140 	{ .n_name = "_carpstats" },
141 #define	N_PFSYNCSTAT	34
142 	{ .n_name = "_pfsyncstats" },
143 #define	N_AHSTAT	35
144 	{ .n_name = "_ahstat" },
145 #define	N_ESPSTAT	36
146 	{ .n_name = "_espstat" },
147 #define	N_IPCOMPSTAT	37
148 	{ .n_name = "_ipcompstat" },
149 #define	N_TCPSTAT	38
150 	{ .n_name = "_tcpstatp" },
151 #define	N_UDPSTAT	39
152 	{ .n_name = "_udpstat" },
153 #define	N_IPSTAT	40
154 	{ .n_name = "_ipstatp" },
155 #define	N_ICMPSTAT	41
156 	{ .n_name = "_icmpstat" },
157 #define	N_IGMPSTAT	42
158 	{ .n_name = "_igmpstat" },
159 #define	N_PIMSTAT	43
160 	{ .n_name = "_pimstat" },
161 #define	N_TCBINFO	44
162 	{ .n_name = "_tcbinfo" },
163 #define	N_UDBINFO	45
164 	{ .n_name = "_udbinfo" },
165 #define	N_DIVCBINFO	46
166 	{ .n_name = "_divcbinfo" },
167 #define	N_RIPCBINFO	47
168 	{ .n_name = "_ripcbinfo" },
169 #define	N_UNP_COUNT	48
170 	{ .n_name = "_unp_count" },
171 #define	N_UNP_GENCNT	49
172 	{ .n_name = "_unp_gencnt" },
173 #define	N_UNP_DHEAD	50
174 	{ .n_name = "_unp_dhead" },
175 #define	N_UNP_SHEAD	51
176 	{ .n_name = "_unp_shead" },
177 #define	N_RIP6STAT	52
178 	{ .n_name = "_rip6stat" },
179 #define	N_SCTPSTAT	53
180 	{ .n_name = "_sctpstat" },
181 #define	N_MFCTABLESIZE	54
182 	{ .n_name = "_mfctablesize" },
183 #define N_ARPSTAT       55
184 	{ .n_name = "_arpstat" },
185 #define	N_UNP_SPHEAD	56
186 	{ .n_name = "unp_sphead" },
187 	{ .n_name = NULL },
188 };
189 
190 struct protox {
191 	int	pr_index;		/* index into nlist of cb head */
192 	int	pr_sindex;		/* index into nlist of stat block */
193 	u_char	pr_wanted;		/* 1 if wanted, 0 otherwise */
194 	void	(*pr_cblocks)(u_long, const char *, int, int);
195 					/* control blocks printing routine */
196 	void	(*pr_stats)(u_long, const char *, int, int);
197 					/* statistics printing routine */
198 	void	(*pr_istats)(char *);	/* per/if statistics printing routine */
199 	const char	*pr_name;		/* well-known name */
200 	int	pr_usesysctl;		/* non-zero if we use sysctl, not kvm */
201 	int	pr_protocol;
202 } protox[] = {
203 	{ N_TCBINFO,	N_TCPSTAT,	1,	protopr,
204 	  tcp_stats,	NULL,		"tcp",	1,	IPPROTO_TCP },
205 	{ N_UDBINFO,	N_UDPSTAT,	1,	protopr,
206 	  udp_stats,	NULL,		"udp",	1,	IPPROTO_UDP },
207 #ifdef SCTP
208 	{ -1,		N_SCTPSTAT,	1,	sctp_protopr,
209 	  sctp_stats,	NULL,		"sctp",	1,	IPPROTO_SCTP },
210 #endif
211 #ifdef SDP
212 	{ -1,		-1,		1,	protopr,
213 	 NULL,		NULL,		"sdp",	1,	IPPROTO_TCP },
214 #endif
215 	{ N_DIVCBINFO,	-1,		1,	protopr,
216 	  NULL,		NULL,		"divert", 1,	IPPROTO_DIVERT },
217 	{ N_RIPCBINFO,	N_IPSTAT,	1,	protopr,
218 	  ip_stats,	NULL,		"ip",	1,	IPPROTO_RAW },
219 	{ N_RIPCBINFO,	N_ICMPSTAT,	1,	protopr,
220 	  icmp_stats,	NULL,		"icmp",	1,	IPPROTO_ICMP },
221 	{ N_RIPCBINFO,	N_IGMPSTAT,	1,	protopr,
222 	  igmp_stats,	NULL,		"igmp",	1,	IPPROTO_IGMP },
223 #ifdef IPSEC
224 	{ -1,		N_IPSECSTAT,	1,	NULL,	/* keep as compat */
225 	  ipsec_stats,	NULL,		"ipsec", 0,	0},
226 	{ -1,		N_AHSTAT,	1,	NULL,
227 	  ah_stats,	NULL,		"ah",	0,	0},
228 	{ -1,		N_ESPSTAT,	1,	NULL,
229 	  esp_stats,	NULL,		"esp",	0,	0},
230 	{ -1,		N_IPCOMPSTAT,	1,	NULL,
231 	  ipcomp_stats,	NULL,		"ipcomp", 0,	0},
232 #endif
233 	{ N_RIPCBINFO,	N_PIMSTAT,	1,	protopr,
234 	  pim_stats,	NULL,		"pim",	1,	IPPROTO_PIM },
235 	{ -1,		N_CARPSTAT,	1,	NULL,
236 	  carp_stats,	NULL,		"carp",	1,	0 },
237 	{ -1,		N_PFSYNCSTAT,	1,	NULL,
238 	  pfsync_stats,	NULL,		"pfsync", 1,	0 },
239 	{ -1,		N_ARPSTAT,	1,	NULL,
240 	  arp_stats,	NULL,		"arp", 1,	0 },
241 	{ -1,		-1,		0,	NULL,
242 	  NULL,		NULL,		NULL,	0,	0 }
243 };
244 
245 #ifdef INET6
246 struct protox ip6protox[] = {
247 	{ N_TCBINFO,	N_TCPSTAT,	1,	protopr,
248 	  tcp_stats,	NULL,		"tcp",	1,	IPPROTO_TCP },
249 	{ N_UDBINFO,	N_UDPSTAT,	1,	protopr,
250 	  udp_stats,	NULL,		"udp",	1,	IPPROTO_UDP },
251 	{ N_RIPCBINFO,	N_IP6STAT,	1,	protopr,
252 	  ip6_stats,	ip6_ifstats,	"ip6",	1,	IPPROTO_RAW },
253 	{ N_RIPCBINFO,	N_ICMP6STAT,	1,	protopr,
254 	  icmp6_stats,	icmp6_ifstats,	"icmp6", 1,	IPPROTO_ICMPV6 },
255 #ifdef SDP
256 	{ -1,		-1,		1,	protopr,
257 	 NULL,		NULL,		"sdp",	1,	IPPROTO_TCP },
258 #endif
259 #ifdef IPSEC
260 	{ -1,		N_IPSEC6STAT,	1,	NULL,
261 	  ipsec_stats,	NULL,		"ipsec6", 0,	0 },
262 #endif
263 #ifdef notyet
264 	{ -1,		N_PIM6STAT,	1,	NULL,
265 	  pim6_stats,	NULL,		"pim6",	1,	0 },
266 #endif
267 	{ -1,		N_RIP6STAT,	1,	NULL,
268 	  rip6_stats,	NULL,		"rip6",	1,	0 },
269 	{ -1,		-1,		0,	NULL,
270 	  NULL,		NULL,		NULL,	0,	0 }
271 };
272 #endif /*INET6*/
273 
274 #ifdef IPSEC
275 struct protox pfkeyprotox[] = {
276 	{ -1,		N_PFKEYSTAT,	1,	NULL,
277 	  pfkey_stats,	NULL,		"pfkey", 0,	0 },
278 	{ -1,		-1,		0,	NULL,
279 	  NULL,		NULL,		NULL,	0,	0 }
280 };
281 #endif
282 
283 struct protox atalkprotox[] = {
284 	{ N_DDPCB,	N_DDPSTAT,	1,	atalkprotopr,
285 	  ddp_stats,	NULL,		"ddp",	0,	0 },
286 	{ -1,		-1,		0,	NULL,
287 	  NULL,		NULL,		NULL,	0,	0 }
288 };
289 #ifdef NETGRAPH
290 struct protox netgraphprotox[] = {
291 	{ N_NGSOCKS,	-1,		1,	netgraphprotopr,
292 	  NULL,		NULL,		"ctrl",	0,	0 },
293 	{ N_NGSOCKS,	-1,		1,	netgraphprotopr,
294 	  NULL,		NULL,		"data",	0,	0 },
295 	{ -1,		-1,		0,	NULL,
296 	  NULL,		NULL,		NULL,	0,	0 }
297 };
298 #endif
299 #ifdef IPX
300 struct protox ipxprotox[] = {
301 	{ N_IPX,	N_IPXSTAT,	1,	ipxprotopr,
302 	  ipx_stats,	NULL,		"ipx",	0,	0 },
303 	{ N_IPX,	N_SPXSTAT,	1,	ipxprotopr,
304 	  spx_stats,	NULL,		"spx",	0,	0 },
305 	{ -1,		-1,		0,	NULL,
306 	  NULL,		NULL,		0,	0,	0 }
307 };
308 #endif
309 
310 struct protox *protoprotox[] = {
311 					 protox,
312 #ifdef INET6
313 					 ip6protox,
314 #endif
315 #ifdef IPSEC
316 					 pfkeyprotox,
317 #endif
318 #ifdef IPX
319 					 ipxprotox,
320 #endif
321 					 atalkprotox, NULL };
322 
323 static void printproto(struct protox *, const char *);
324 static void usage(void);
325 static struct protox *name2protox(const char *);
326 static struct protox *knownname(const char *);
327 
328 static kvm_t *kvmd;
329 static char *nlistf = NULL, *memf = NULL;
330 
331 int	Aflag;		/* show addresses of protocol control block */
332 int	aflag;		/* show all sockets (including servers) */
333 int	Bflag;		/* show information about bpf consumers */
334 int	bflag;		/* show i/f total bytes in/out */
335 int	dflag;		/* show i/f dropped packets */
336 int	gflag;		/* show group (multicast) routing or stats */
337 int	hflag;		/* show counters in human readable format */
338 int	iflag;		/* show interfaces */
339 int	Lflag;		/* show size of listen queues */
340 int	mflag;		/* show memory stats */
341 int	noutputs = 0;	/* how much outputs before we exit */
342 int	numeric_addr;	/* show addresses numerically */
343 int	numeric_port;	/* show ports numerically */
344 static int pflag;	/* show given protocol */
345 int	Qflag;		/* show netisr information */
346 int	rflag;		/* show routing tables (or routing stats) */
347 int	sflag;		/* show protocol statistics */
348 int	Wflag;		/* wide display */
349 int	Tflag;		/* TCP Information */
350 int	xflag;		/* extra information, includes all socket buffer info */
351 int	zflag;		/* zero stats */
352 
353 int	interval;	/* repeat interval for i/f stats */
354 
355 char	*interface;	/* desired i/f for stats, or NULL for all i/fs */
356 int	unit;		/* unit number for above */
357 
358 int	af;		/* address family */
359 int	live;		/* true if we are examining a live system */
360 
361 int
362 main(int argc, char *argv[])
363 {
364 	struct protox *tp = NULL;  /* for printing cblocks & stats */
365 	int ch;
366 
367 	af = AF_UNSPEC;
368 
369 	while ((ch = getopt(argc, argv, "AaBbdf:ghI:iLlM:mN:np:Qq:rSTsuWw:xz"))
370 	    != -1)
371 		switch(ch) {
372 		case 'A':
373 			Aflag = 1;
374 			break;
375 		case 'a':
376 			aflag = 1;
377 			break;
378 		case 'B':
379 			Bflag = 1;
380 			break;
381 		case 'b':
382 			bflag = 1;
383 			break;
384 		case 'd':
385 			dflag = 1;
386 			break;
387 		case 'f':
388 			if (strcmp(optarg, "ipx") == 0)
389 				af = AF_IPX;
390 			else if (strcmp(optarg, "inet") == 0)
391 				af = AF_INET;
392 #ifdef INET6
393 			else if (strcmp(optarg, "inet6") == 0)
394 				af = AF_INET6;
395 #endif
396 #ifdef IPSEC
397 			else if (strcmp(optarg, "pfkey") == 0)
398 				af = PF_KEY;
399 #endif
400 			else if (strcmp(optarg, "unix") == 0)
401 				af = AF_UNIX;
402 			else if (strcmp(optarg, "atalk") == 0)
403 				af = AF_APPLETALK;
404 #ifdef NETGRAPH
405 			else if (strcmp(optarg, "ng") == 0
406 			    || strcmp(optarg, "netgraph") == 0)
407 				af = AF_NETGRAPH;
408 #endif
409 			else if (strcmp(optarg, "link") == 0)
410 				af = AF_LINK;
411 			else {
412 				errx(1, "%s: unknown address family", optarg);
413 			}
414 			break;
415 		case 'g':
416 			gflag = 1;
417 			break;
418 		case 'h':
419 			hflag = 1;
420 			break;
421 		case 'I': {
422 			char *cp;
423 
424 			iflag = 1;
425 			for (cp = interface = optarg; isalpha(*cp); cp++)
426 				continue;
427 			unit = atoi(cp);
428 			break;
429 		}
430 		case 'i':
431 			iflag = 1;
432 			break;
433 		case 'L':
434 			Lflag = 1;
435 			break;
436 		case 'M':
437 			memf = optarg;
438 			break;
439 		case 'm':
440 			mflag = 1;
441 			break;
442 		case 'N':
443 			nlistf = optarg;
444 			break;
445 		case 'n':
446 			numeric_addr = numeric_port = 1;
447 			break;
448 		case 'p':
449 			if ((tp = name2protox(optarg)) == NULL) {
450 				errx(1,
451 				     "%s: unknown or uninstrumented protocol",
452 				     optarg);
453 			}
454 			pflag = 1;
455 			break;
456 		case 'Q':
457 			Qflag = 1;
458 			break;
459 		case 'q':
460 			noutputs = atoi(optarg);
461 			if (noutputs != 0)
462 				noutputs++;
463 			break;
464 		case 'r':
465 			rflag = 1;
466 			break;
467 		case 's':
468 			++sflag;
469 			break;
470 		case 'S':
471 			numeric_addr = 1;
472 			break;
473 		case 'u':
474 			af = AF_UNIX;
475 			break;
476 		case 'W':
477 		case 'l':
478 			Wflag = 1;
479 			break;
480 		case 'w':
481 			interval = atoi(optarg);
482 			iflag = 1;
483 			break;
484 		case 'T':
485 			Tflag = 1;
486 			break;
487 		case 'x':
488 			xflag = 1;
489 			break;
490 		case 'z':
491 			zflag = 1;
492 			break;
493 		case '?':
494 		default:
495 			usage();
496 		}
497 	argv += optind;
498 	argc -= optind;
499 
500 #define	BACKWARD_COMPATIBILITY
501 #ifdef	BACKWARD_COMPATIBILITY
502 	if (*argv) {
503 		if (isdigit(**argv)) {
504 			interval = atoi(*argv);
505 			if (interval <= 0)
506 				usage();
507 			++argv;
508 			iflag = 1;
509 		}
510 		if (*argv) {
511 			nlistf = *argv;
512 			if (*++argv)
513 				memf = *argv;
514 		}
515 	}
516 #endif
517 
518 	/*
519 	 * Discard setgid privileges if not the running kernel so that bad
520 	 * guys can't print interesting stuff from kernel memory.
521 	 */
522 	live = (nlistf == NULL && memf == NULL);
523 	if (!live)
524 		setgid(getgid());
525 
526 	if (xflag && Tflag)
527 		errx(1, "-x and -T are incompatible, pick one.");
528 
529 	if (Bflag) {
530 		if (!live)
531 			usage();
532 		bpf_stats(interface);
533 		exit(0);
534 	}
535 	if (mflag) {
536 		if (!live) {
537 			if (kread(0, NULL, 0) == 0)
538 				mbpr(kvmd, nl[N_MBSTAT].n_value);
539 		} else
540 			mbpr(NULL, 0);
541 		exit(0);
542 	}
543 	if (Qflag) {
544 		if (!live) {
545 			if (kread(0, NULL, 0) == 0)
546 				netisr_stats(kvmd);
547 		} else
548 			netisr_stats(NULL);
549 		exit(0);
550 	}
551 #if 0
552 	/*
553 	 * Keep file descriptors open to avoid overhead
554 	 * of open/close on each call to get* routines.
555 	 */
556 	sethostent(1);
557 	setnetent(1);
558 #else
559 	/*
560 	 * This does not make sense any more with DNS being default over
561 	 * the files.  Doing a setXXXXent(1) causes a tcp connection to be
562 	 * used for the queries, which is slower.
563 	 */
564 #endif
565 	kread(0, NULL, 0);
566 	if (iflag && !sflag) {
567 		intpr(interval, nl[N_IFNET].n_value, NULL);
568 		exit(0);
569 	}
570 	if (rflag) {
571 		if (sflag)
572 			rt_stats(nl[N_RTSTAT].n_value, nl[N_RTTRASH].n_value);
573 		else
574 			routepr(nl[N_RTREE].n_value);
575 		exit(0);
576 	}
577 	if (gflag) {
578 		if (sflag) {
579 			if (af == AF_INET || af == AF_UNSPEC)
580 				mrt_stats(nl[N_MRTSTAT].n_value);
581 #ifdef INET6
582 			if (af == AF_INET6 || af == AF_UNSPEC)
583 				mrt6_stats(nl[N_MRT6STAT].n_value);
584 #endif
585 		} else {
586 			if (af == AF_INET || af == AF_UNSPEC)
587 				mroutepr(nl[N_MFCHASHTBL].n_value,
588 					 nl[N_MFCTABLESIZE].n_value,
589 					 nl[N_VIFTABLE].n_value);
590 #ifdef INET6
591 			if (af == AF_INET6 || af == AF_UNSPEC)
592 				mroute6pr(nl[N_MF6CTABLE].n_value,
593 					  nl[N_MIF6TABLE].n_value);
594 #endif
595 		}
596 		exit(0);
597 	}
598 
599 	if (tp) {
600 		printproto(tp, tp->pr_name);
601 		exit(0);
602 	}
603 	if (af == AF_INET || af == AF_UNSPEC)
604 		for (tp = protox; tp->pr_name; tp++)
605 			printproto(tp, tp->pr_name);
606 #ifdef INET6
607 	if (af == AF_INET6 || af == AF_UNSPEC)
608 		for (tp = ip6protox; tp->pr_name; tp++)
609 			printproto(tp, tp->pr_name);
610 #endif /*INET6*/
611 #ifdef IPSEC
612 	if (af == PF_KEY || af == AF_UNSPEC)
613 		for (tp = pfkeyprotox; tp->pr_name; tp++)
614 			printproto(tp, tp->pr_name);
615 #endif /*IPSEC*/
616 #ifdef IPX
617 	if (af == AF_IPX || af == AF_UNSPEC) {
618 		for (tp = ipxprotox; tp->pr_name; tp++)
619 			printproto(tp, tp->pr_name);
620 	}
621 #endif /* IPX */
622 	if (af == AF_APPLETALK || af == AF_UNSPEC)
623 		for (tp = atalkprotox; tp->pr_name; tp++)
624 			printproto(tp, tp->pr_name);
625 #ifdef NETGRAPH
626 	if (af == AF_NETGRAPH || af == AF_UNSPEC)
627 		for (tp = netgraphprotox; tp->pr_name; tp++)
628 			printproto(tp, tp->pr_name);
629 #endif /* NETGRAPH */
630 	if ((af == AF_UNIX || af == AF_UNSPEC) && !sflag)
631 		unixpr(nl[N_UNP_COUNT].n_value, nl[N_UNP_GENCNT].n_value,
632 		    nl[N_UNP_DHEAD].n_value, nl[N_UNP_SHEAD].n_value,
633 		    nl[N_UNP_SPHEAD].n_value);
634 	exit(0);
635 }
636 
637 /*
638  * Print out protocol statistics or control blocks (per sflag).
639  * If the interface was not specifically requested, and the symbol
640  * is not in the namelist, ignore this one.
641  */
642 static void
643 printproto(struct protox *tp, const char *name)
644 {
645 	void (*pr)(u_long, const char *, int, int);
646 	u_long off;
647 
648 	if (sflag) {
649 		if (iflag) {
650 			if (tp->pr_istats)
651 				intpr(interval, nl[N_IFNET].n_value,
652 				      tp->pr_istats);
653 			else if (pflag)
654 				printf("%s: no per-interface stats routine\n",
655 				    tp->pr_name);
656 			return;
657 		} else {
658 			pr = tp->pr_stats;
659 			if (!pr) {
660 				if (pflag)
661 					printf("%s: no stats routine\n",
662 					    tp->pr_name);
663 				return;
664 			}
665 			if (tp->pr_usesysctl && live)
666 				off = 0;
667 			else if (tp->pr_sindex < 0) {
668 				if (pflag)
669 					printf(
670 				    "%s: stats routine doesn't work on cores\n",
671 					    tp->pr_name);
672 				return;
673 			} else
674 				off = nl[tp->pr_sindex].n_value;
675 		}
676 	} else {
677 		pr = tp->pr_cblocks;
678 		if (!pr) {
679 			if (pflag)
680 				printf("%s: no PCB routine\n", tp->pr_name);
681 			return;
682 		}
683 		if (tp->pr_usesysctl && live)
684 			off = 0;
685 		else if (tp->pr_index < 0) {
686 			if (pflag)
687 				printf(
688 				    "%s: PCB routine doesn't work on cores\n",
689 				    tp->pr_name);
690 			return;
691 		} else
692 			off = nl[tp->pr_index].n_value;
693 	}
694 	if (pr != NULL && (off || (live && tp->pr_usesysctl) ||
695 	    af != AF_UNSPEC))
696 		(*pr)(off, name, af, tp->pr_protocol);
697 }
698 
699 static int
700 kvmd_init(void)
701 {
702 	char errbuf[_POSIX2_LINE_MAX];
703 
704 	if (kvmd != NULL)
705 		return (0);
706 
707 	kvmd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY, errbuf);
708 	setgid(getgid());
709 
710 	if (kvmd == NULL) {
711 		warnx("kvm not available: %s", errbuf);
712 		return (-1);
713 	}
714 
715 	if (kvm_nlist(kvmd, nl) < 0) {
716 		if (nlistf)
717 			errx(1, "%s: kvm_nlist: %s", nlistf,
718 			     kvm_geterr(kvmd));
719 		else
720 			errx(1, "kvm_nlist: %s", kvm_geterr(kvmd));
721 	}
722 
723 	if (nl[0].n_type == 0) {
724 		if (nlistf)
725 			errx(1, "%s: no namelist", nlistf);
726 		else
727 			errx(1, "no namelist");
728 	}
729 
730 	return (0);
731 }
732 
733 /*
734  * Read kernel memory, return 0 on success.
735  */
736 int
737 kread(u_long addr, void *buf, size_t size)
738 {
739 
740 	if (kvmd_init() < 0)
741 		return (-1);
742 
743 	if (!buf)
744 		return (0);
745 	if (kvm_read(kvmd, addr, buf, size) != (ssize_t)size) {
746 		warnx("%s", kvm_geterr(kvmd));
747 		return (-1);
748 	}
749 	return (0);
750 }
751 
752 /*
753  * Read an array of N counters in kernel memory into array of N uint64_t's.
754  */
755 int
756 kread_counters(u_long *addr, uint64_t *rval, size_t count)
757 {
758 
759 	if (kvmd_init() < 0)
760 		return (-1);
761 
762 	for (u_int i = 0; i < count; i++, addr++, rval++)
763 		*rval = kvm_counter_u64_fetch(kvmd, *addr);
764 
765 	return (0);
766 }
767 
768 const char *
769 plural(uintmax_t n)
770 {
771 	return (n != 1 ? "s" : "");
772 }
773 
774 const char *
775 plurales(uintmax_t n)
776 {
777 	return (n != 1 ? "es" : "");
778 }
779 
780 const char *
781 pluralies(uintmax_t n)
782 {
783 	return (n != 1 ? "ies" : "y");
784 }
785 
786 /*
787  * Find the protox for the given "well-known" name.
788  */
789 static struct protox *
790 knownname(const char *name)
791 {
792 	struct protox **tpp, *tp;
793 
794 	for (tpp = protoprotox; *tpp; tpp++)
795 		for (tp = *tpp; tp->pr_name; tp++)
796 			if (strcmp(tp->pr_name, name) == 0)
797 				return (tp);
798 	return (NULL);
799 }
800 
801 /*
802  * Find the protox corresponding to name.
803  */
804 static struct protox *
805 name2protox(const char *name)
806 {
807 	struct protox *tp;
808 	char **alias;			/* alias from p->aliases */
809 	struct protoent *p;
810 
811 	/*
812 	 * Try to find the name in the list of "well-known" names. If that
813 	 * fails, check if name is an alias for an Internet protocol.
814 	 */
815 	if ((tp = knownname(name)) != NULL)
816 		return (tp);
817 
818 	setprotoent(1);			/* make protocol lookup cheaper */
819 	while ((p = getprotoent()) != NULL) {
820 		/* assert: name not same as p->name */
821 		for (alias = p->p_aliases; *alias; alias++)
822 			if (strcmp(name, *alias) == 0) {
823 				endprotoent();
824 				return (knownname(p->p_name));
825 			}
826 	}
827 	endprotoent();
828 	return (NULL);
829 }
830 
831 static void
832 usage(void)
833 {
834 	(void)fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n",
835 "usage: netstat [-AaLnSTWx] [-f protocol_family | -p protocol]\n"
836 "               [-M core] [-N system]",
837 "       netstat -i | -I interface [-abdhnW] [-f address_family]\n"
838 "               [-M core] [-N system]",
839 "       netstat -w wait [-I interface] [-d] [-M core] [-N system] [-q howmany]",
840 "       netstat -s [-s] [-z] [-f protocol_family | -p protocol]\n"
841 "               [-M core] [-N system]",
842 "       netstat -i | -I interface -s [-f protocol_family | -p protocol]\n"
843 "               [-M core] [-N system]",
844 "       netstat -m [-M core] [-N system]",
845 "       netstat -B [-I interface]",
846 "       netstat -r [-AanW] [-f address_family] [-M core] [-N system]",
847 "       netstat -rs [-s] [-M core] [-N system]",
848 "       netstat -g [-W] [-f address_family] [-M core] [-N system]",
849 "       netstat -gs [-s] [-f address_family] [-M core] [-N system]",
850 "       netstat -Q");
851 	exit(1);
852 }
853