Annotation of src/usr.bin/netstat/main.c, Revision 1.18
1.18 ! angelos 1: /* $OpenBSD: main.c,v 1.17 1999/03/01 01:28:13 d Exp $ */
1.2 deraadt 2: /* $NetBSD: main.c,v 1.9 1996/05/07 02:55:02 thorpej Exp $ */
1.1 deraadt 3:
4: /*
5: * Copyright (c) 1983, 1988, 1993
6: * Regents of the University of California. All rights reserved.
7: *
8: * Redistribution and use in source and binary forms, with or without
9: * modification, are permitted provided that the following conditions
10: * are met:
11: * 1. Redistributions of source code must retain the above copyright
12: * notice, this list of conditions and the following disclaimer.
13: * 2. Redistributions in binary form must reproduce the above copyright
14: * notice, this list of conditions and the following disclaimer in the
15: * documentation and/or other materials provided with the distribution.
16: * 3. All advertising materials mentioning features or use of this software
17: * must display the following acknowledgement:
18: * This product includes software developed by the University of
19: * California, Berkeley and its contributors.
20: * 4. Neither the name of the University nor the names of its contributors
21: * may be used to endorse or promote products derived from this software
22: * without specific prior written permission.
23: *
24: * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25: * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27: * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28: * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29: * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30: * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31: * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32: * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33: * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34: * SUCH DAMAGE.
35: */
36:
37: #ifndef lint
38: char copyright[] =
39: "@(#) Copyright (c) 1983, 1988, 1993\n\
40: Regents of the University of California. All rights reserved.\n";
41: #endif /* not lint */
42:
43: #ifndef lint
44: #if 0
45: static char sccsid[] = "from: @(#)main.c 8.4 (Berkeley) 3/1/94";
46: #else
1.18 ! angelos 47: static char *rcsid = "$OpenBSD: main.c,v 1.17 1999/03/01 01:28:13 d Exp $";
1.1 deraadt 48: #endif
49: #endif /* not lint */
50:
51: #include <sys/param.h>
52: #include <sys/file.h>
53: #include <sys/protosw.h>
54: #include <sys/socket.h>
55:
56: #include <netinet/in.h>
57:
58: #include <ctype.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 <stdio.h>
66: #include <stdlib.h>
67: #include <string.h>
68: #include <unistd.h>
69: #include "netstat.h"
70:
71: struct nlist nl[] = {
72: #define N_MBSTAT 0
73: { "_mbstat" },
74: #define N_IPSTAT 1
75: { "_ipstat" },
76: #define N_TCBTABLE 2
77: { "_tcbtable" },
78: #define N_TCPSTAT 3
79: { "_tcpstat" },
80: #define N_UDBTABLE 4
81: { "_udbtable" },
82: #define N_UDPSTAT 5
83: { "_udpstat" },
84: #define N_IFNET 6
85: { "_ifnet" },
86: #define N_IMP 7
87: { "_imp_softc" },
88: #define N_ICMPSTAT 8
89: { "_icmpstat" },
90: #define N_RTSTAT 9
91: { "_rtstat" },
92: #define N_UNIXSW 10
93: { "_unixsw" },
94: #define N_IDP 11
95: { "_nspcb"},
96: #define N_IDPSTAT 12
97: { "_idpstat"},
98: #define N_SPPSTAT 13
99: { "_spp_istat"},
100: #define N_NSERR 14
101: { "_ns_errstat"},
102: #define N_CLNPSTAT 15
103: { "_clnp_stat"},
104: #define IN_NOTUSED 16
105: { "_tp_inpcb" },
106: #define ISO_TP 17
107: { "_tp_refinfo" },
108: #define N_TPSTAT 18
109: { "_tp_stat" },
110: #define N_ESISSTAT 19
111: { "_esis_stat"},
112: #define N_NIMP 20
113: { "_nimp"},
114: #define N_RTREE 21
115: { "_rt_tables"},
116: #define N_CLTP 22
117: { "_cltb"},
118: #define N_CLTPSTAT 23
119: { "_cltpstat"},
120: #define N_NFILE 24
121: { "_nfile" },
122: #define N_FILE 25
123: { "_file" },
124: #define N_IGMPSTAT 26
125: { "_igmpstat" },
126: #define N_MRTPROTO 27
127: { "_ip_mrtproto" },
128: #define N_MRTSTAT 28
129: { "_mrtstat" },
130: #define N_MFCHASHTBL 29
131: { "_mfchashtbl" },
132: #define N_MFCHASH 30
133: { "_mfchash" },
134: #define N_VIFTABLE 31
135: { "_viftable" },
1.4 mickey 136: #define N_IPX 32
137: { "_ipxcbtable"},
138: #define N_IPXSTAT 33
139: { "_ipxstat"},
140: #define N_SPXSTAT 34
141: { "_spx_istat"},
142: #define N_IPXERR 35
143: { "_ipx_errstat"},
1.8 angelos 144: #define N_AHSTAT 36
145: { "_ahstat"},
146: #define N_ESPSTAT 37
147: { "_espstat"},
148: #define N_IP4STAT 38
149: { "_ip4stat"},
1.12 denny 150: #define N_DDPSTAT 39
151: { "_ddpstat"},
152: #define N_DDPCB 40
153: { "_ddpcb"},
1.18 ! angelos 154: #define N_ETHERIPSTAT 41
! 155: { "_etheripstat"},
1.11 millert 156: { ""},
1.1 deraadt 157: };
158:
159: struct protox {
160: u_char pr_index; /* index into nlist of cb head */
161: u_char pr_sindex; /* index into nlist of stat block */
162: u_char pr_wanted; /* 1 if wanted, 0 otherwise */
163: void (*pr_cblocks)(); /* control blocks printing routine */
164: void (*pr_stats)(); /* statistics printing routine */
165: char *pr_name; /* well-known name */
166: } protox[] = {
167: { N_TCBTABLE, N_TCPSTAT, 1, protopr,
168: tcp_stats, "tcp" },
169: { N_UDBTABLE, N_UDPSTAT, 1, protopr,
170: udp_stats, "udp" },
171: { -1, N_IPSTAT, 1, 0,
172: ip_stats, "ip" },
173: { -1, N_ICMPSTAT, 1, 0,
174: icmp_stats, "icmp" },
175: { -1, N_IGMPSTAT, 1, 0,
176: igmp_stats, "igmp" },
1.8 angelos 177: { -1, N_AHSTAT, 1, 0,
1.15 angelos 178: ah_stats, "ah" },
1.8 angelos 179: { -1, N_ESPSTAT, 1, 0,
1.15 angelos 180: esp_stats, "esp" },
1.8 angelos 181: { -1, N_IP4STAT, 1, 0,
1.9 angelos 182: ip4_stats, "ipencap" },
1.18 ! angelos 183: { -1, N_ETHERIPSTAT, 1, 0,
! 184: etherip_stats,"etherip" },
1.1 deraadt 185: { -1, -1, 0, 0,
186: 0, 0 }
187: };
188:
1.4 mickey 189: struct protox ipxprotox[] = {
190: { N_IPX, N_IPXSTAT, 1, ipxprotopr,
191: ipx_stats, "ipx" },
192: { N_IPX, N_SPXSTAT, 1, ipxprotopr,
193: spx_stats, "spx" },
194: { -1, N_IPXERR, 1, 0,
195: ipxerr_stats, "ipx_err" },
196: { -1, -1, 0, 0,
197: 0, 0 }
198: };
199:
1.1 deraadt 200: struct protox nsprotox[] = {
201: { N_IDP, N_IDPSTAT, 1, nsprotopr,
202: idp_stats, "idp" },
203: { N_IDP, N_SPPSTAT, 1, nsprotopr,
204: spp_stats, "spp" },
205: { -1, N_NSERR, 1, 0,
206: nserr_stats, "ns_err" },
207: { -1, -1, 0, 0,
208: 0, 0 }
209: };
210:
211: struct protox isoprotox[] = {
212: { ISO_TP, N_TPSTAT, 1, iso_protopr,
213: tp_stats, "tp" },
214: { N_CLTP, N_CLTPSTAT, 1, iso_protopr,
215: cltp_stats, "cltp" },
216: { -1, N_CLNPSTAT, 1, 0,
217: clnp_stats, "clnp"},
218: { -1, N_ESISSTAT, 1, 0,
219: esis_stats, "esis"},
220: { -1, -1, 0, 0,
221: 0, 0 }
222: };
223:
1.12 denny 224: struct protox atalkprotox[] = {
225: { N_DDPCB, N_DDPSTAT, 1, atalkprotopr,
226: ddp_stats, "ddp" },
227: { -1, -1, 0, 0,
228: 0, 0 }
229: };
230:
231: struct protox *protoprotox[] = { protox, ipxprotox, nsprotox, isoprotox, atalkprotox, NULL };
1.1 deraadt 232:
233: static void printproto __P((struct protox *, char *));
234: static void usage __P((void));
235: static struct protox *name2protox __P((char *));
236: static struct protox *knownname __P((char *));
237:
238: kvm_t *kvmd;
239:
240: int
241: main(argc, argv)
242: int argc;
243: char *argv[];
244: {
245: extern char *optarg;
246: extern int optind;
247: register struct protoent *p;
248: register struct protox *tp; /* for printing cblocks & stats */
249: int ch;
250: char *nlistf = NULL, *memf = NULL;
251: char buf[_POSIX2_LINE_MAX];
252:
253: af = AF_UNSPEC;
254:
1.17 d 255: while ((ch = getopt(argc, argv, "Aadf:gI:iM:mN:np:rstuvw:")) != -1)
1.1 deraadt 256: switch(ch) {
257: case 'A':
258: Aflag = 1;
259: break;
260: case 'a':
261: aflag = 1;
262: break;
263: case 'd':
264: dflag = 1;
265: break;
266: case 'f':
1.4 mickey 267: if (strcmp(optarg, "inet") == 0)
1.1 deraadt 268: af = AF_INET;
1.7 kstailey 269: else if (strcmp(optarg, "local") == 0)
270: af = AF_LOCAL;
1.1 deraadt 271: else if (strcmp(optarg, "unix") == 0)
272: af = AF_UNIX;
1.4 mickey 273: else if (strcmp(optarg, "ipx") == 0)
274: af = AF_IPX;
275: else if (strcmp(optarg, "ns") == 0)
276: af = AF_NS;
1.1 deraadt 277: else if (strcmp(optarg, "iso") == 0)
278: af = AF_ISO;
1.10 angelos 279: else if (strcmp(optarg, "encap") == 0)
1.16 angelos 280: af = PF_KEY;
1.12 denny 281: else if (strcmp(optarg, "atalk") == 0)
282: af = AF_APPLETALK;
1.1 deraadt 283: else {
284: (void)fprintf(stderr,
285: "%s: %s: unknown address family\n",
1.2 deraadt 286: __progname, optarg);
1.1 deraadt 287: exit(1);
288: }
289: break;
290: case 'g':
291: gflag = 1;
292: break;
1.2 deraadt 293: case 'I':
1.1 deraadt 294: iflag = 1;
1.2 deraadt 295: interface = optarg;
1.1 deraadt 296: break;
297: case 'i':
298: iflag = 1;
299: break;
300: case 'M':
301: memf = optarg;
302: break;
303: case 'm':
304: mflag = 1;
305: break;
306: case 'N':
307: nlistf = optarg;
308: break;
309: case 'n':
310: nflag = 1;
311: break;
312: case 'p':
313: if ((tp = name2protox(optarg)) == NULL) {
314: (void)fprintf(stderr,
315: "%s: %s: unknown or uninstrumented protocol\n",
1.2 deraadt 316: __progname, optarg);
1.1 deraadt 317: exit(1);
318: }
319: pflag = 1;
320: break;
321: case 'r':
322: rflag = 1;
323: break;
324: case 's':
325: ++sflag;
326: break;
327: case 't':
328: tflag = 1;
329: break;
330: case 'u':
331: af = AF_UNIX;
1.13 peter 332: break;
333: case 'v':
334: vflag = 1;
1.1 deraadt 335: break;
336: case 'w':
337: interval = atoi(optarg);
338: iflag = 1;
339: break;
340: case '?':
341: default:
342: usage();
343: }
344: argv += optind;
345: argc -= optind;
346:
347: #define BACKWARD_COMPATIBILITY
348: #ifdef BACKWARD_COMPATIBILITY
349: if (*argv) {
350: if (isdigit(**argv)) {
351: interval = atoi(*argv);
352: if (interval <= 0)
353: usage();
354: ++argv;
355: iflag = 1;
356: }
357: if (*argv) {
358: nlistf = *argv;
359: if (*++argv)
360: memf = *argv;
361: }
362: }
363: #endif
364:
365: /*
366: * Discard setgid privileges if not the running kernel so that bad
367: * guys can't print interesting stuff from kernel memory.
368: */
1.5 tholo 369: if (nlistf != NULL || memf != NULL) {
370: setegid(getgid());
1.1 deraadt 371: setgid(getgid());
1.5 tholo 372: }
1.1 deraadt 373:
1.2 deraadt 374: if ((kvmd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY,
375: buf)) == NULL) {
376: fprintf(stderr, "%s: kvm_open: %s\n", __progname, buf);
1.1 deraadt 377: exit(1);
378: }
1.14 deraadt 379: setegid(getgid());
380: setgid(getgid());
381:
1.1 deraadt 382: if (kvm_nlist(kvmd, nl) < 0 || nl[0].n_type == 0) {
383: if (nlistf)
1.2 deraadt 384: fprintf(stderr, "%s: %s: no namelist\n", __progname,
385: nlistf);
1.1 deraadt 386: else
1.2 deraadt 387: fprintf(stderr, "%s: no namelist\n", __progname);
1.1 deraadt 388: exit(1);
389: }
390: if (mflag) {
391: mbpr(nl[N_MBSTAT].n_value);
392: exit(0);
393: }
394: if (pflag) {
395: if (tp->pr_stats)
396: (*tp->pr_stats)(nl[tp->pr_sindex].n_value,
397: tp->pr_name);
398: else
399: printf("%s: no stats routine\n", tp->pr_name);
400: exit(0);
401: }
402: /*
403: * Keep file descriptors open to avoid overhead
404: * of open/close on each call to get* routines.
405: */
406: sethostent(1);
407: setnetent(1);
408: if (iflag) {
409: intpr(interval, nl[N_IFNET].n_value);
410: exit(0);
411: }
412: if (rflag) {
413: if (sflag)
414: rt_stats(nl[N_RTSTAT].n_value);
415: else
416: routepr(nl[N_RTREE].n_value);
417: exit(0);
418: }
419: if (gflag) {
420: if (sflag)
421: mrt_stats(nl[N_MRTPROTO].n_value,
422: nl[N_MRTSTAT].n_value);
423: else
424: mroutepr(nl[N_MRTPROTO].n_value,
425: nl[N_MFCHASHTBL].n_value,
426: nl[N_MFCHASH].n_value,
427: nl[N_VIFTABLE].n_value);
428: exit(0);
429: }
430: if (af == AF_INET || af == AF_UNSPEC) {
431: setprotoent(1);
432: setservent(1);
433: /* ugh, this is O(MN) ... why do we do this? */
1.11 millert 434: while ((p = getprotoent())) {
1.1 deraadt 435: for (tp = protox; tp->pr_name; tp++)
436: if (strcmp(tp->pr_name, p->p_name) == 0)
437: break;
438: if (tp->pr_name == 0 || tp->pr_wanted == 0)
439: continue;
440: printproto(tp, p->p_name);
441: }
442: endprotoent();
443: }
1.4 mickey 444: if (af == AF_IPX || af == AF_UNSPEC)
445: for (tp = ipxprotox; tp->pr_name; tp++)
446: printproto(tp, tp->pr_name);
1.1 deraadt 447: if (af == AF_NS || af == AF_UNSPEC)
448: for (tp = nsprotox; tp->pr_name; tp++)
449: printproto(tp, tp->pr_name);
450: if (af == AF_ISO || af == AF_UNSPEC)
451: for (tp = isoprotox; tp->pr_name; tp++)
452: printproto(tp, tp->pr_name);
453: if ((af == AF_UNIX || af == AF_UNSPEC) && !sflag)
454: unixpr(nl[N_UNIXSW].n_value);
1.12 denny 455: if (af == AF_APPLETALK || af == AF_UNSPEC)
456: for (tp = atalkprotox; tp->pr_name; tp++)
457: printproto(tp, tp->pr_name);
1.1 deraadt 458: exit(0);
459: }
460:
461: /*
462: * Print out protocol statistics or control blocks (per sflag).
463: * If the interface was not specifically requested, and the symbol
464: * is not in the namelist, ignore this one.
465: */
466: static void
467: printproto(tp, name)
468: register struct protox *tp;
469: char *name;
470: {
471: void (*pr)();
472: u_long off;
473:
474: if (sflag) {
475: pr = tp->pr_stats;
476: off = nl[tp->pr_sindex].n_value;
477: } else {
478: pr = tp->pr_cblocks;
479: off = nl[tp->pr_index].n_value;
480: }
481: if (pr != NULL && (off || af != AF_UNSPEC))
482: (*pr)(off, name);
483: }
484:
485: /*
486: * Read kernel memory, return 0 on success.
487: */
488: int
489: kread(addr, buf, size)
490: u_long addr;
491: char *buf;
492: int size;
493: {
494:
495: if (kvm_read(kvmd, addr, buf, size) != size) {
1.2 deraadt 496: (void)fprintf(stderr, "%s: %s\n", __progname,
1.1 deraadt 497: kvm_geterr(kvmd));
498: return (-1);
499: }
500: return (0);
501: }
502:
503: char *
504: plural(n)
505: int n;
506: {
507: return (n != 1 ? "s" : "");
508: }
509:
510: char *
511: plurales(n)
512: int n;
513: {
514: return (n != 1 ? "es" : "");
515: }
516:
517: /*
518: * Find the protox for the given "well-known" name.
519: */
520: static struct protox *
521: knownname(name)
522: char *name;
523: {
524: struct protox **tpp, *tp;
525:
526: for (tpp = protoprotox; *tpp; tpp++)
527: for (tp = *tpp; tp->pr_name; tp++)
528: if (strcmp(tp->pr_name, name) == 0)
529: return (tp);
530: return (NULL);
531: }
532:
533: /*
534: * Find the protox corresponding to name.
535: */
536: static struct protox *
537: name2protox(name)
538: char *name;
539: {
540: struct protox *tp;
541: char **alias; /* alias from p->aliases */
542: struct protoent *p;
543:
544: /*
545: * Try to find the name in the list of "well-known" names. If that
546: * fails, check if name is an alias for an Internet protocol.
547: */
1.11 millert 548: if ((tp = knownname(name)))
1.1 deraadt 549: return (tp);
550:
551: setprotoent(1); /* make protocol lookup cheaper */
1.11 millert 552: while ((p = getprotoent())) {
1.1 deraadt 553: /* assert: name not same as p->name */
554: for (alias = p->p_aliases; *alias; alias++)
555: if (strcmp(name, *alias) == 0) {
556: endprotoent();
557: return (knownname(p->p_name));
558: }
559: }
560: endprotoent();
561: return (NULL);
562: }
563:
564: static void
565: usage()
566: {
567: (void)fprintf(stderr,
1.2 deraadt 568: "usage: %s [-Aan] [-f address_family] [-M core] [-N system]\n", __progname);
1.1 deraadt 569: (void)fprintf(stderr,
1.17 d 570: " %s [-gimnrs] [-f address_family] [-M core] [-N system]\n", __progname);
1.1 deraadt 571: (void)fprintf(stderr,
1.2 deraadt 572: " %s [-n] [-I interface] [-M core] [-N system] [-w wait]\n", __progname);
1.1 deraadt 573: (void)fprintf(stderr,
1.2 deraadt 574: " %s [-M core] [-N system] [-p protocol]\n", __progname);
1.1 deraadt 575: exit(1);
576: }