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