1 /* $NetBSD: main.c,v 1.11 1997/04/03 04:46:48 christos 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 #ifndef lint 37 char copyright[] = 38 "@(#) 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 static char *rcsid = "$NetBSD: main.c,v 1.11 1997/04/03 04:46:48 christos 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 register struct protoent *p; 221 register struct protox *tp; /* for printing cblocks & stats */ 222 int ch; 223 char *nlistf = NULL, *memf = NULL; 224 char buf[_POSIX2_LINE_MAX]; 225 226 af = AF_UNSPEC; 227 228 while ((ch = getopt(argc, argv, "Aadf:ghI:iM:mN:np:rstuw:")) != EOF) 229 switch(ch) { 230 case 'A': 231 Aflag = 1; 232 break; 233 case 'a': 234 aflag = 1; 235 break; 236 case 'd': 237 dflag = 1; 238 break; 239 case 'f': 240 if (strcmp(optarg, "ns") == 0) 241 af = AF_NS; 242 else if (strcmp(optarg, "inet") == 0) 243 af = AF_INET; 244 else if (strcmp(optarg, "unix") == 0) 245 af = AF_UNIX; 246 else if (strcmp(optarg, "iso") == 0) 247 af = AF_ISO; 248 else if (strcmp(optarg, "atalk") == 0) 249 af = AF_APPLETALK; 250 else { 251 (void)fprintf(stderr, 252 "%s: %s: unknown address family\n", 253 __progname, optarg); 254 exit(1); 255 } 256 break; 257 case 'g': 258 gflag = 1; 259 break; 260 case 'I': 261 iflag = 1; 262 interface = optarg; 263 break; 264 case 'i': 265 iflag = 1; 266 break; 267 case 'M': 268 memf = optarg; 269 break; 270 case 'm': 271 mflag = 1; 272 break; 273 case 'N': 274 nlistf = optarg; 275 break; 276 case 'n': 277 nflag = 1; 278 break; 279 case 'p': 280 if ((tp = name2protox(optarg)) == NULL) { 281 (void)fprintf(stderr, 282 "%s: %s: unknown or uninstrumented protocol\n", 283 __progname, optarg); 284 exit(1); 285 } 286 pflag = 1; 287 break; 288 case 'r': 289 rflag = 1; 290 break; 291 case 's': 292 ++sflag; 293 break; 294 case 't': 295 tflag = 1; 296 break; 297 case 'u': 298 af = AF_UNIX; 299 break; 300 case 'w': 301 interval = atoi(optarg); 302 iflag = 1; 303 break; 304 case '?': 305 default: 306 usage(); 307 } 308 argv += optind; 309 argc -= optind; 310 311 #define BACKWARD_COMPATIBILITY 312 #ifdef BACKWARD_COMPATIBILITY 313 if (*argv) { 314 if (isdigit(**argv)) { 315 interval = atoi(*argv); 316 if (interval <= 0) 317 usage(); 318 ++argv; 319 iflag = 1; 320 } 321 if (*argv) { 322 nlistf = *argv; 323 if (*++argv) 324 memf = *argv; 325 } 326 } 327 #endif 328 329 /* 330 * Discard setgid privileges if not the running kernel so that bad 331 * guys can't print interesting stuff from kernel memory. 332 */ 333 if (nlistf != NULL || memf != NULL) 334 setgid(getgid()); 335 336 if ((kvmd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY, 337 buf)) == NULL) { 338 fprintf(stderr, "%s: kvm_open: %s\n", __progname, buf); 339 exit(1); 340 } 341 if (kvm_nlist(kvmd, nl) < 0 || nl[0].n_type == 0) { 342 if (nlistf) 343 fprintf(stderr, "%s: %s: no namelist\n", __progname, 344 nlistf); 345 else 346 fprintf(stderr, "%s: no namelist\n", __progname); 347 exit(1); 348 } 349 if (mflag) { 350 mbpr(nl[N_MBSTAT].n_value, nl[N_MSIZE].n_value, 351 nl[N_MCLBYTES].n_value); 352 exit(0); 353 } 354 if (pflag) { 355 if (tp->pr_stats) 356 (*tp->pr_stats)(nl[tp->pr_sindex].n_value, 357 tp->pr_name); 358 else 359 printf("%s: no stats routine\n", tp->pr_name); 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(nl[N_RTSTAT].n_value); 375 else 376 routepr(nl[N_RTREE].n_value); 377 exit(0); 378 } 379 if (gflag) { 380 if (sflag) 381 mrt_stats(nl[N_MRTPROTO].n_value, 382 nl[N_MRTSTAT].n_value); 383 else 384 mroutepr(nl[N_MRTPROTO].n_value, 385 nl[N_MFCHASHTBL].n_value, 386 nl[N_MFCHASH].n_value, 387 nl[N_VIFTABLE].n_value); 388 exit(0); 389 } 390 if (af == AF_INET || af == AF_UNSPEC) { 391 setprotoent(1); 392 setservent(1); 393 /* ugh, this is O(MN) ... why do we do this? */ 394 while ((p = getprotoent()) != NULL) { 395 for (tp = protox; tp->pr_name; tp++) 396 if (strcmp(tp->pr_name, p->p_name) == 0) 397 break; 398 if (tp->pr_name == 0 || tp->pr_wanted == 0) 399 continue; 400 printproto(tp, p->p_name); 401 } 402 endprotoent(); 403 } 404 if (af == AF_APPLETALK || af == AF_UNSPEC) 405 for (tp = atalkprotox; tp->pr_name; tp++) 406 printproto(tp, tp->pr_name); 407 if (af == AF_NS || af == AF_UNSPEC) 408 for (tp = nsprotox; tp->pr_name; tp++) 409 printproto(tp, tp->pr_name); 410 if (af == AF_ISO || af == AF_UNSPEC) 411 for (tp = isoprotox; tp->pr_name; tp++) 412 printproto(tp, tp->pr_name); 413 if ((af == AF_UNIX || af == AF_UNSPEC) && !sflag) 414 unixpr(nl[N_UNIXSW].n_value); 415 exit(0); 416 } 417 418 /* 419 * Print out protocol statistics or control blocks (per sflag). 420 * If the interface was not specifically requested, and the symbol 421 * is not in the namelist, ignore this one. 422 */ 423 static void 424 printproto(tp, name) 425 register struct protox *tp; 426 char *name; 427 { 428 void (*pr) __P((u_long, char *)); 429 u_long off; 430 431 if (sflag) { 432 pr = tp->pr_stats; 433 off = nl[tp->pr_sindex].n_value; 434 } else { 435 pr = tp->pr_cblocks; 436 off = nl[tp->pr_index].n_value; 437 } 438 if (pr != NULL && (off || af != AF_UNSPEC)) 439 (*pr)(off, name); 440 } 441 442 /* 443 * Read kernel memory, return 0 on success. 444 */ 445 int 446 kread(addr, buf, size) 447 u_long addr; 448 char *buf; 449 int size; 450 { 451 452 if (kvm_read(kvmd, addr, buf, size) != size) { 453 (void)fprintf(stderr, "%s: %s\n", __progname, 454 kvm_geterr(kvmd)); 455 return (-1); 456 } 457 return (0); 458 } 459 460 char * 461 plural(n) 462 int n; 463 { 464 return (n != 1 ? "s" : ""); 465 } 466 467 char * 468 plurales(n) 469 int n; 470 { 471 return (n != 1 ? "es" : ""); 472 } 473 474 /* 475 * Find the protox for the given "well-known" name. 476 */ 477 static struct protox * 478 knownname(name) 479 char *name; 480 { 481 struct protox **tpp, *tp; 482 483 for (tpp = protoprotox; *tpp; tpp++) 484 for (tp = *tpp; tp->pr_name; tp++) 485 if (strcmp(tp->pr_name, name) == 0) 486 return (tp); 487 return (NULL); 488 } 489 490 /* 491 * Find the protox corresponding to name. 492 */ 493 static struct protox * 494 name2protox(name) 495 char *name; 496 { 497 struct protox *tp; 498 char **alias; /* alias from p->aliases */ 499 struct protoent *p; 500 501 /* 502 * Try to find the name in the list of "well-known" names. If that 503 * fails, check if name is an alias for an Internet protocol. 504 */ 505 if ((tp = knownname(name)) != NULL) 506 return (tp); 507 508 setprotoent(1); /* make protocol lookup cheaper */ 509 while ((p = getprotoent()) != NULL) { 510 /* assert: name not same as p->name */ 511 for (alias = p->p_aliases; *alias; alias++) 512 if (strcmp(name, *alias) == 0) { 513 endprotoent(); 514 return (knownname(p->p_name)); 515 } 516 } 517 endprotoent(); 518 return (NULL); 519 } 520 521 static void 522 usage() 523 { 524 (void)fprintf(stderr, 525 "usage: %s [-Aan] [-f address_family] [-M core] [-N system]\n", __progname); 526 (void)fprintf(stderr, 527 " %s [-ghimnrs] [-f address_family] [-M core] [-N system]\n", __progname); 528 (void)fprintf(stderr, 529 " %s [-n] [-I interface] [-M core] [-N system] [-w wait]\n", __progname); 530 (void)fprintf(stderr, 531 " %s [-M core] [-N system] [-p protocol]\n", __progname); 532 exit(1); 533 } 534