xref: /netbsd-src/usr.bin/netstat/main.c (revision 811e6386f8c5e4a3521c7003da29ec8673e344fa)
1 /*
2  * Copyright (c) 1983, 1988 Regents of the University of California.
3  * 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  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *	This product includes software developed by the University of
16  *	California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  */
33 
34 #ifndef lint
35 char copyright[] =
36 "@(#) Copyright (c) 1983, 1988 Regents of the University of California.\n\
37  All rights reserved.\n";
38 #endif /* not lint */
39 
40 #ifndef lint
41 static char sccsid[] = "@(#)main.c	5.23 (Berkeley) 7/1/91";
42 #endif /* not lint */
43 
44 #include <sys/param.h>
45 #include <sys/socket.h>
46 #include <sys/file.h>
47 #include <errno.h>
48 #include <netdb.h>
49 #include <nlist.h>
50 #include <kvm.h>
51 #include <stdio.h>
52 #include <ctype.h>
53 #include <stdlib.h>
54 #include <string.h>
55 #include <paths.h>
56 
57 struct nlist nl[] = {
58 #define	N_MBSTAT	0
59 	{ "_mbstat" },
60 #define	N_IPSTAT	1
61 	{ "_ipstat" },
62 #define	N_TCB		2
63 	{ "_tcb" },
64 #define	N_TCPSTAT	3
65 	{ "_tcpstat" },
66 #define	N_UDB		4
67 	{ "_udb" },
68 #define	N_UDPSTAT	5
69 	{ "_udpstat" },
70 #define	N_IFNET		6
71 	{ "_ifnet" },
72 #define	N_IMP		7
73 	{ "_imp_softc" },
74 #define	N_RTHOST	8
75 	{ "_rthost" },
76 #define	N_RTNET		9
77 	{ "_rtnet" },
78 #define	N_ICMPSTAT	10
79 	{ "_icmpstat" },
80 #define	N_RTSTAT	11
81 	{ "_rtstat" },
82 #define	N_FILEHEAD	12
83 	{ "_filehead" },
84 #define	N_FILES		13
85 	{ "_nfiles" },
86 #define	N_UNIXSW	14
87 	{ "_unixsw" },
88 #define N_RTHASHSIZE	15
89 	{ "_rthashsize" },
90 #define N_IDP		16
91 	{ "_nspcb"},
92 #define N_IDPSTAT	17
93 	{ "_idpstat"},
94 #define N_SPPSTAT	18
95 	{ "_spp_istat"},
96 #define N_NSERR		19
97 	{ "_ns_errstat"},
98 #define	N_CLNPSTAT	20
99 	{ "_clnp_stat"},
100 #define	IN_TP		21
101 	{ "_tp_inpcb" },
102 #define	ISO_TP		22
103 	{ "_tp_isopcb" },
104 #define	N_TPSTAT	23
105 	{ "_tp_stat" },
106 #define	N_ESISSTAT	24
107 	{ "_esis_stat"},
108 #define N_NIMP		25
109 	{ "_nimp"},
110 #define N_RTREE		26
111 	{ "_radix_node_head"},
112 #define N_CLTP		27
113 	{ "_cltb"},
114 #define N_CLTPSTAT	28
115 	{ "_cltpstat"},
116 	"",
117 };
118 
119 /* internet protocols */
120 extern	int protopr();
121 extern	int tcp_stats(), udp_stats(), ip_stats(), icmp_stats();
122 #ifdef NS
123 /* ns protocols */
124 extern	int nsprotopr();
125 extern	int spp_stats(), idp_stats(), nserr_stats();
126 #endif
127 #ifdef ISO
128 /* iso protocols */
129 extern	int iso_protopr();
130 extern	int tp_stats(), esis_stats(), clnp_stats(), cltp_stats();
131 #endif
132 
133 struct protox {
134 	u_char	pr_index;		/* index into nlist of cb head */
135 	u_char	pr_sindex;		/* index into nlist of stat block */
136 	u_char	pr_wanted;		/* 1 if wanted, 0 otherwise */
137 	int	(*pr_cblocks)();	/* control blocks printing routine */
138 	int	(*pr_stats)();		/* statistics printing routine */
139 	char	*pr_name;		/* well-known name */
140 } protox[] = {
141 	{ N_TCB,	N_TCPSTAT,	1,	protopr,
142 	  tcp_stats,	"tcp" },
143 	{ N_UDB,	N_UDPSTAT,	1,	protopr,
144 	  udp_stats,	"udp" },
145 #ifdef ISO
146 	{ IN_TP,	N_TPSTAT,	1,	protopr,
147 	  tp_stats,	"tpip" },
148 #endif
149 	{ -1,		N_IPSTAT,	1,	0,
150 	  ip_stats,	"ip" },
151 	{ -1,		N_ICMPSTAT,	1,	0,
152 	  icmp_stats,	"icmp" },
153 	{ -1,		-1,		0,	0,
154 	  0,		0 }
155 };
156 
157 #ifdef NS
158 struct protox nsprotox[] = {
159 	{ N_IDP,	N_IDPSTAT,	1,	nsprotopr,
160 	  idp_stats,	"idp" },
161 	{ N_IDP,	N_SPPSTAT,	1,	nsprotopr,
162 	  spp_stats,	"spp" },
163 	{ -1,		N_NSERR,	1,	0,
164 	  nserr_stats,	"ns_err" },
165 	{ -1,		-1,		0,	0,
166 	  0,		0 }
167 };
168 #endif
169 
170 #ifdef ISO
171 struct protox isoprotox[] = {
172 	{ ISO_TP,	N_TPSTAT,	1,	iso_protopr,
173 	  tp_stats,	"tp" },
174 	{ N_CLTP,	N_CLTPSTAT,	1,	iso_protopr,
175 	  cltp_stats,	"cltp" },
176 	{ -1,		N_CLNPSTAT,	1,	 0,
177 	  clnp_stats,	"clnp"},
178 	{ -1,		N_ESISSTAT,	1,	 0,
179 	  esis_stats,	"esis"},
180 	{ -1,		-1,		0,	0,
181 	  0,		0 }
182 };
183 #endif
184 
185 struct protox *protoprotox[] = { protox,
186 
187 #ifdef NS
188 nsprotox,
189 #endif
190 
191 #ifdef ISO
192 isoprotox,
193 #endif
194 
195 NULL };
196 
197 char	*vmunix = _PATH_UNIX;
198 char	*kmemf;
199 int	kmem;
200 int	kflag;
201 int	Aflag;
202 int	aflag;
203 int	hflag;
204 int	iflag;
205 int	mflag;
206 int	nflag;
207 int	pflag;
208 int	rflag;
209 int	sflag;
210 int	tflag;
211 int	dflag;
212 int	interval;
213 char	*interface;
214 int	unit;
215 
216 int	af = AF_UNSPEC;
217 
218 main(argc, argv)
219 	int argc;
220 	char **argv;
221 {
222 	extern char *optarg;
223 	extern int optind;
224 	register struct protoent *p;
225 	register struct protox *tp;	/* for printing cblocks & stats */
226 	struct protox *name2protox();	/* for -p */
227 	int ch;
228 	void usage();
229 
230 	while ((ch = getopt(argc, argv, "Aadf:hI:iM:mN:np:rstuw")) != EOF)
231 		switch((char)ch) {
232 		case 'A':
233 			Aflag = 1;
234 			break;
235 		case 'a':
236 			aflag = 1;
237 			break;
238 		case 'd':
239 			dflag = 1;
240 			break;
241 		case 'f':
242 			if (strcmp(optarg, "ns") == 0)
243 				af = AF_NS;
244 			else if (strcmp(optarg, "inet") == 0)
245 				af = AF_INET;
246 			else if (strcmp(optarg, "unix") == 0)
247 				af = AF_UNIX;
248 			else if (strcmp(optarg, "iso") == 0)
249 				af = AF_ISO;
250 			else {
251 				(void)fprintf(stderr,
252 				    "%s: unknown address family\n", optarg);
253 				exit(1);
254 			}
255 			break;
256 		case 'h':
257 			hflag = 1;
258 			break;
259 		case 'I': {
260 			char *cp;
261 
262 			iflag = 1;
263 			for (cp = interface = optarg; isalpha(*cp); cp++);
264 			unit = atoi(cp);
265 			*cp = '\0';
266 			break;
267 		}
268 		case 'i':
269 			iflag = 1;
270 			break;
271 		case 'M':
272 			kmemf = optarg;
273 			kflag = 1;
274 			break;
275 		case 'm':
276 			mflag = 1;
277 			break;
278 		case 'N':
279 			vmunix = optarg;
280 			break;
281 		case 'n':
282 			nflag = 1;
283 			break;
284 		case 'p':
285 			if ((tp = name2protox(optarg)) == NULL) {
286 				(void)fprintf(stderr,
287 				    "%s: unknown or uninstrumented protocol\n",
288 				    optarg);
289 				exit(1);
290 			}
291 			pflag = 1;
292 			break;
293 		case 'r':
294 			rflag = 1;
295 			break;
296 		case 's':
297 			sflag = 1;
298 			break;
299 		case 't':
300 			tflag = 1;
301 			break;
302 		case 'u':
303 			af = AF_UNIX;
304 			break;
305 		case 'w':
306 			interval = atoi(optarg);
307 			break;
308 		case '?':
309 		default:
310 			usage();
311 		}
312 	argv += optind;
313 	argc -= optind;
314 
315 #define	BACKWARD_COMPATIBILITY
316 #ifdef	BACKWARD_COMPATIBILITY
317 	if (*argv) {
318 		if (isdigit(**argv)) {
319 			interval = atoi(*argv);
320 			if (interval <= 0)
321 				usage();
322 			++argv;
323 			iflag = 1;
324 		}
325 		if (*argv) {
326 			vmunix = *argv;
327 			if (*++argv) {
328 				kmemf = *argv;
329 				kflag = 1;
330 			}
331 		}
332 	}
333 #endif
334 	if (kvm_openfiles(vmunix, kmemf, NULL) == -1) {
335 		fprintf(stderr, "netstat: kvm_openfiles: %s\n", kvm_geterr());
336 		exit(1);
337 	}
338 	if (kvm_nlist(nl) < 0 || nl[0].n_type == 0) {
339 		fprintf(stderr, "%s: no namelist\n", vmunix);
340 		exit(1);
341 	}
342 	if (mflag) {
343 		mbpr((off_t)nl[N_MBSTAT].n_value);
344 		exit(0);
345 	}
346 	if (pflag) {
347 		if (tp->pr_stats)
348 			(*tp->pr_stats)(nl[tp->pr_sindex].n_value,
349 				tp->pr_name);
350 		else
351 			printf("%s: no stats routine\n", tp->pr_name);
352 		exit(0);
353 	}
354 	if (hflag) {
355 #ifdef IMP
356 		hostpr(nl[N_IMP].n_value, nl[N_NIMP].n_value);
357 #else
358 		(void)fprintf(stderr, "netstat: IMP code not compiled in\n");
359 #endif
360 		exit(0);
361 	}
362 	/*
363 	 * Keep file descriptors open to avoid overhead
364 	 * of open/close on each call to get* routines.
365 	 */
366 	sethostent(1);
367 	setnetent(1);
368 	if (iflag) {
369 		intpr(interval, nl[N_IFNET].n_value);
370 		exit(0);
371 	}
372 	if (rflag) {
373 		if (sflag)
374 			rt_stats((off_t)nl[N_RTSTAT].n_value);
375 		else
376 			routepr((off_t)nl[N_RTHOST].n_value,
377 				(off_t)nl[N_RTNET].n_value,
378 				(off_t)nl[N_RTHASHSIZE].n_value,
379 				(off_t)nl[N_RTREE].n_value);
380 		exit(0);
381 	}
382     if (af == AF_INET || af == AF_UNSPEC) {
383 	setprotoent(1);
384 	setservent(1);
385 	while (p = getprotoent()) {
386 
387 		for (tp = protox; tp->pr_name; tp++)
388 			if (strcmp(tp->pr_name, p->p_name) == 0)
389 				break;
390 		if (tp->pr_name == 0 || tp->pr_wanted == 0)
391 			continue;
392 		if (sflag) {
393 			if (tp->pr_stats)
394 				(*tp->pr_stats)(nl[tp->pr_sindex].n_value,
395 					p->p_name);
396 		} else
397 			if (tp->pr_cblocks)
398 				(*tp->pr_cblocks)(nl[tp->pr_index].n_value,
399 					p->p_name);
400 	}
401 	endprotoent();
402     }
403     if (af == AF_NS || af == AF_UNSPEC) {
404 #ifdef NS
405 	for (tp = nsprotox; tp->pr_name; tp++) {
406 		if (sflag) {
407 			if (tp->pr_stats)
408 				(*tp->pr_stats)(nl[tp->pr_sindex].n_value,
409 					tp->pr_name);
410 		} else
411 			if (tp->pr_cblocks)
412 				(*tp->pr_cblocks)(nl[tp->pr_index].n_value,
413 					tp->pr_name);
414 	}
415 #endif
416     }
417     if (af == AF_ISO || af == AF_UNSPEC) {
418 #ifdef ISO
419 	for (tp = isoprotox; tp->pr_name; tp++) {
420 		if (sflag) {
421 			if (tp->pr_stats)
422 				(*tp->pr_stats)(nl[tp->pr_sindex].n_value,
423 					tp->pr_name);
424 		} else
425 			if (tp->pr_cblocks)
426 				(*tp->pr_cblocks)(nl[tp->pr_index].n_value,
427 					tp->pr_name);
428 	}
429 #endif
430     }
431     if ((af == AF_UNIX || af == AF_UNSPEC) && !sflag)
432 	    unixpr((off_t)nl[N_FILEHEAD].n_value, (off_t)nl[N_FILES].n_value,
433 		(struct protosw *)nl[N_UNIXSW].n_value);
434     if (af == AF_UNSPEC && sflag)
435 #ifdef IMP
436 	impstats(nl[N_IMP].n_value, nl[N_NIMP].n_value);
437 #endif
438     exit(0);
439 }
440 
441 char *
442 plural(n)
443 	int n;
444 {
445 	return (n != 1 ? "s" : "");
446 }
447 
448 /*
449  * Find the protox for the given "well-known" name.
450  */
451 struct protox *
452 knownname(name)
453 	char *name;
454 {
455 	struct protox **tpp, *tp;
456 
457 	for (tpp = protoprotox; *tpp; tpp++)
458 	    for (tp = *tpp; tp->pr_name; tp++)
459 		if (strcmp(tp->pr_name, name) == 0)
460 			return(tp);
461 	return(NULL);
462 }
463 
464 /*
465  * Find the protox corresponding to name.
466  */
467 struct protox *
468 name2protox(name)
469 	char *name;
470 {
471 	struct protox *tp;
472 	char **alias;			/* alias from p->aliases */
473 	struct protoent *p;
474 
475 	/*
476 	 * Try to find the name in the list of "well-known" names. If that
477 	 * fails, check if name is an alias for an Internet protocol.
478 	 */
479 	if (tp = knownname(name))
480 		return(tp);
481 
482 	setprotoent(1);			/* make protocol lookup cheaper */
483 	while (p = getprotoent()) {
484 		/* assert: name not same as p->name */
485 		for (alias = p->p_aliases; *alias; alias++)
486 			if (strcmp(name, *alias) == 0) {
487 				endprotoent();
488 				return(knownname(p->p_name));
489 			}
490 	}
491 	endprotoent();
492 	return(NULL);
493 }
494 
495 void
496 usage()
497 {
498 	(void)fprintf(stderr,
499 "usage: netstat [-Aan] [-f address_family] [-M core] [-N system]\n");
500 	(void)fprintf(stderr,
501 "               [-himnrs] [-f address_family] [-M core] [-N system]\n");
502 	(void)fprintf(stderr,
503 "               [-n] [-I interface] [-M core] [-N system] [-w wait]\n");
504 	(void)fprintf(stderr,
505 "               [-M core] [-N system] [-p protocol]\n");
506 	exit(1);
507 }
508