[BACK]Return to main.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / netstat

Annotation of src/usr.bin/netstat/main.c, Revision 1.10

1.10    ! angelos     1: /*     $OpenBSD: main.c,v 1.9 1997/02/22 04:35:24 angelos 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.10    ! angelos    47: static char *rcsid = "$OpenBSD: main.c,v 1.9 1997/02/22 04:35:24 angelos 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.1       deraadt   150:        "",
                    151: };
                    152:
                    153: struct protox {
                    154:        u_char  pr_index;               /* index into nlist of cb head */
                    155:        u_char  pr_sindex;              /* index into nlist of stat block */
                    156:        u_char  pr_wanted;              /* 1 if wanted, 0 otherwise */
                    157:        void    (*pr_cblocks)();        /* control blocks printing routine */
                    158:        void    (*pr_stats)();          /* statistics printing routine */
                    159:        char    *pr_name;               /* well-known name */
                    160: } protox[] = {
                    161:        { N_TCBTABLE,   N_TCPSTAT,      1,      protopr,
                    162:          tcp_stats,    "tcp" },
                    163:        { N_UDBTABLE,   N_UDPSTAT,      1,      protopr,
                    164:          udp_stats,    "udp" },
                    165:        { -1,           N_IPSTAT,       1,      0,
                    166:          ip_stats,     "ip" },
                    167:        { -1,           N_ICMPSTAT,     1,      0,
                    168:          icmp_stats,   "icmp" },
                    169:        { -1,           N_IGMPSTAT,     1,      0,
                    170:          igmp_stats,   "igmp" },
1.8       angelos   171:        { -1,           N_AHSTAT,       1,      0,
                    172:          ah_stats,     "sipp-ah" },
                    173:        { -1,           N_ESPSTAT,      1,      0,
                    174:          esp_stats,    "sipp-esp" },
                    175:        { -1,           N_IP4STAT,      1,      0,
1.9       angelos   176:          ip4_stats,    "ipencap" },
1.1       deraadt   177:        { -1,           -1,             0,      0,
                    178:          0,            0 }
                    179: };
                    180:
1.4       mickey    181: struct protox ipxprotox[] = {
                    182:        { N_IPX,        N_IPXSTAT,      1,      ipxprotopr,
                    183:          ipx_stats,    "ipx" },
                    184:        { N_IPX,        N_SPXSTAT,      1,      ipxprotopr,
                    185:          spx_stats,    "spx" },
                    186:        { -1,           N_IPXERR,       1,      0,
                    187:          ipxerr_stats, "ipx_err" },
                    188:        { -1,           -1,             0,      0,
                    189:          0,            0 }
                    190: };
                    191:
1.1       deraadt   192: struct protox nsprotox[] = {
                    193:        { N_IDP,        N_IDPSTAT,      1,      nsprotopr,
                    194:          idp_stats,    "idp" },
                    195:        { N_IDP,        N_SPPSTAT,      1,      nsprotopr,
                    196:          spp_stats,    "spp" },
                    197:        { -1,           N_NSERR,        1,      0,
                    198:          nserr_stats,  "ns_err" },
                    199:        { -1,           -1,             0,      0,
                    200:          0,            0 }
                    201: };
                    202:
                    203: struct protox isoprotox[] = {
                    204:        { ISO_TP,       N_TPSTAT,       1,      iso_protopr,
                    205:          tp_stats,     "tp" },
                    206:        { N_CLTP,       N_CLTPSTAT,     1,      iso_protopr,
                    207:          cltp_stats,   "cltp" },
                    208:        { -1,           N_CLNPSTAT,     1,       0,
                    209:          clnp_stats,   "clnp"},
                    210:        { -1,           N_ESISSTAT,     1,       0,
                    211:          esis_stats,   "esis"},
                    212:        { -1,           -1,             0,      0,
                    213:          0,            0 }
                    214: };
                    215:
1.4       mickey    216: struct protox *protoprotox[] = { protox, ipxprotox, nsprotox, isoprotox, NULL };
1.1       deraadt   217:
                    218: static void printproto __P((struct protox *, char *));
                    219: static void usage __P((void));
                    220: static struct protox *name2protox __P((char *));
                    221: static struct protox *knownname __P((char *));
                    222:
                    223: kvm_t *kvmd;
                    224:
                    225: int
                    226: main(argc, argv)
                    227:        int argc;
                    228:        char *argv[];
                    229: {
                    230:        extern char *optarg;
                    231:        extern int optind;
                    232:        register struct protoent *p;
                    233:        register struct protox *tp;     /* for printing cblocks & stats */
                    234:        register char *cp;
                    235:        int ch;
                    236:        char *nlistf = NULL, *memf = NULL;
                    237:        char buf[_POSIX2_LINE_MAX];
                    238:
                    239:        af = AF_UNSPEC;
                    240:
1.6       millert   241:        while ((ch = getopt(argc, argv, "Aadf:ghI:iM:mN:np:rstuw:")) != -1)
1.1       deraadt   242:                switch(ch) {
                    243:                case 'A':
                    244:                        Aflag = 1;
                    245:                        break;
                    246:                case 'a':
                    247:                        aflag = 1;
                    248:                        break;
                    249:                case 'd':
                    250:                        dflag = 1;
                    251:                        break;
                    252:                case 'f':
1.4       mickey    253:                        if (strcmp(optarg, "inet") == 0)
1.1       deraadt   254:                                af = AF_INET;
1.7       kstailey  255:                        else if (strcmp(optarg, "local") == 0)
                    256:                                af = AF_LOCAL;
1.1       deraadt   257:                        else if (strcmp(optarg, "unix") == 0)
                    258:                                af = AF_UNIX;
1.4       mickey    259:                        else if (strcmp(optarg, "ipx") == 0)
                    260:                                af = AF_IPX;
                    261:                        else if (strcmp(optarg, "ns") == 0)
                    262:                                af = AF_NS;
1.1       deraadt   263:                        else if (strcmp(optarg, "iso") == 0)
                    264:                                af = AF_ISO;
1.10    ! angelos   265:                        else if (strcmp(optarg, "encap") == 0)
        !           266:                                af = AF_ENCAP;
1.1       deraadt   267:                        else {
                    268:                                (void)fprintf(stderr,
                    269:                                    "%s: %s: unknown address family\n",
1.2       deraadt   270:                                    __progname, optarg);
1.1       deraadt   271:                                exit(1);
                    272:                        }
                    273:                        break;
                    274:                case 'g':
                    275:                        gflag = 1;
                    276:                        break;
1.2       deraadt   277:                case 'I':
1.1       deraadt   278:                        iflag = 1;
1.2       deraadt   279:                        interface = optarg;
1.1       deraadt   280:                        break;
                    281:                case 'i':
                    282:                        iflag = 1;
                    283:                        break;
                    284:                case 'M':
                    285:                        memf = optarg;
                    286:                        break;
                    287:                case 'm':
                    288:                        mflag = 1;
                    289:                        break;
                    290:                case 'N':
                    291:                        nlistf = optarg;
                    292:                        break;
                    293:                case 'n':
                    294:                        nflag = 1;
                    295:                        break;
                    296:                case 'p':
                    297:                        if ((tp = name2protox(optarg)) == NULL) {
                    298:                                (void)fprintf(stderr,
                    299:                                    "%s: %s: unknown or uninstrumented protocol\n",
1.2       deraadt   300:                                    __progname, optarg);
1.1       deraadt   301:                                exit(1);
                    302:                        }
                    303:                        pflag = 1;
                    304:                        break;
                    305:                case 'r':
                    306:                        rflag = 1;
                    307:                        break;
                    308:                case 's':
                    309:                        ++sflag;
                    310:                        break;
                    311:                case 't':
                    312:                        tflag = 1;
                    313:                        break;
                    314:                case 'u':
                    315:                        af = AF_UNIX;
                    316:                        break;
                    317:                case 'w':
                    318:                        interval = atoi(optarg);
                    319:                        iflag = 1;
                    320:                        break;
                    321:                case '?':
                    322:                default:
                    323:                        usage();
                    324:                }
                    325:        argv += optind;
                    326:        argc -= optind;
                    327:
                    328: #define        BACKWARD_COMPATIBILITY
                    329: #ifdef BACKWARD_COMPATIBILITY
                    330:        if (*argv) {
                    331:                if (isdigit(**argv)) {
                    332:                        interval = atoi(*argv);
                    333:                        if (interval <= 0)
                    334:                                usage();
                    335:                        ++argv;
                    336:                        iflag = 1;
                    337:                }
                    338:                if (*argv) {
                    339:                        nlistf = *argv;
                    340:                        if (*++argv)
                    341:                                memf = *argv;
                    342:                }
                    343:        }
                    344: #endif
                    345:
                    346:        /*
                    347:         * Discard setgid privileges if not the running kernel so that bad
                    348:         * guys can't print interesting stuff from kernel memory.
                    349:         */
1.5       tholo     350:        if (nlistf != NULL || memf != NULL) {
                    351:                setegid(getgid());
1.1       deraadt   352:                setgid(getgid());
1.5       tholo     353:        }
1.1       deraadt   354:
1.2       deraadt   355:        if ((kvmd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY,
                    356:            buf)) == NULL) {
                    357:                fprintf(stderr, "%s: kvm_open: %s\n", __progname, buf);
1.1       deraadt   358:                exit(1);
                    359:        }
                    360:        if (kvm_nlist(kvmd, nl) < 0 || nl[0].n_type == 0) {
                    361:                if (nlistf)
1.2       deraadt   362:                        fprintf(stderr, "%s: %s: no namelist\n", __progname,
                    363:                            nlistf);
1.1       deraadt   364:                else
1.2       deraadt   365:                        fprintf(stderr, "%s: no namelist\n", __progname);
1.1       deraadt   366:                exit(1);
                    367:        }
                    368:        if (mflag) {
                    369:                mbpr(nl[N_MBSTAT].n_value);
                    370:                exit(0);
                    371:        }
                    372:        if (pflag) {
                    373:                if (tp->pr_stats)
                    374:                        (*tp->pr_stats)(nl[tp->pr_sindex].n_value,
                    375:                                tp->pr_name);
                    376:                else
                    377:                        printf("%s: no stats routine\n", tp->pr_name);
                    378:                exit(0);
                    379:        }
                    380:        /*
                    381:         * Keep file descriptors open to avoid overhead
                    382:         * of open/close on each call to get* routines.
                    383:         */
                    384:        sethostent(1);
                    385:        setnetent(1);
                    386:        if (iflag) {
                    387:                intpr(interval, nl[N_IFNET].n_value);
                    388:                exit(0);
                    389:        }
                    390:        if (rflag) {
                    391:                if (sflag)
                    392:                        rt_stats(nl[N_RTSTAT].n_value);
                    393:                else
                    394:                        routepr(nl[N_RTREE].n_value);
                    395:                exit(0);
                    396:        }
                    397:        if (gflag) {
                    398:                if (sflag)
                    399:                        mrt_stats(nl[N_MRTPROTO].n_value,
                    400:                            nl[N_MRTSTAT].n_value);
                    401:                else
                    402:                        mroutepr(nl[N_MRTPROTO].n_value,
                    403:                            nl[N_MFCHASHTBL].n_value,
                    404:                            nl[N_MFCHASH].n_value,
                    405:                            nl[N_VIFTABLE].n_value);
                    406:                exit(0);
                    407:        }
                    408:        if (af == AF_INET || af == AF_UNSPEC) {
                    409:                setprotoent(1);
                    410:                setservent(1);
                    411:                /* ugh, this is O(MN) ... why do we do this? */
                    412:                while (p = getprotoent()) {
                    413:                        for (tp = protox; tp->pr_name; tp++)
                    414:                                if (strcmp(tp->pr_name, p->p_name) == 0)
                    415:                                        break;
                    416:                        if (tp->pr_name == 0 || tp->pr_wanted == 0)
                    417:                                continue;
                    418:                        printproto(tp, p->p_name);
                    419:                }
                    420:                endprotoent();
                    421:        }
1.4       mickey    422:        if (af == AF_IPX || af == AF_UNSPEC)
                    423:                for (tp = ipxprotox; tp->pr_name; tp++)
                    424:                        printproto(tp, tp->pr_name);
1.1       deraadt   425:        if (af == AF_NS || af == AF_UNSPEC)
                    426:                for (tp = nsprotox; tp->pr_name; tp++)
                    427:                        printproto(tp, tp->pr_name);
                    428:        if (af == AF_ISO || af == AF_UNSPEC)
                    429:                for (tp = isoprotox; tp->pr_name; tp++)
                    430:                        printproto(tp, tp->pr_name);
                    431:        if ((af == AF_UNIX || af == AF_UNSPEC) && !sflag)
                    432:                unixpr(nl[N_UNIXSW].n_value);
                    433:        exit(0);
                    434: }
                    435:
                    436: /*
                    437:  * Print out protocol statistics or control blocks (per sflag).
                    438:  * If the interface was not specifically requested, and the symbol
                    439:  * is not in the namelist, ignore this one.
                    440:  */
                    441: static void
                    442: printproto(tp, name)
                    443:        register struct protox *tp;
                    444:        char *name;
                    445: {
                    446:        void (*pr)();
                    447:        u_long off;
                    448:
                    449:        if (sflag) {
                    450:                pr = tp->pr_stats;
                    451:                off = nl[tp->pr_sindex].n_value;
                    452:        } else {
                    453:                pr = tp->pr_cblocks;
                    454:                off = nl[tp->pr_index].n_value;
                    455:        }
                    456:        if (pr != NULL && (off || af != AF_UNSPEC))
                    457:                (*pr)(off, name);
                    458: }
                    459:
                    460: /*
                    461:  * Read kernel memory, return 0 on success.
                    462:  */
                    463: int
                    464: kread(addr, buf, size)
                    465:        u_long addr;
                    466:        char *buf;
                    467:        int size;
                    468: {
                    469:
                    470:        if (kvm_read(kvmd, addr, buf, size) != size) {
1.2       deraadt   471:                (void)fprintf(stderr, "%s: %s\n", __progname,
1.1       deraadt   472:                    kvm_geterr(kvmd));
                    473:                return (-1);
                    474:        }
                    475:        return (0);
                    476: }
                    477:
                    478: char *
                    479: plural(n)
                    480:        int n;
                    481: {
                    482:        return (n != 1 ? "s" : "");
                    483: }
                    484:
                    485: char *
                    486: plurales(n)
                    487:        int n;
                    488: {
                    489:        return (n != 1 ? "es" : "");
                    490: }
                    491:
                    492: /*
                    493:  * Find the protox for the given "well-known" name.
                    494:  */
                    495: static struct protox *
                    496: knownname(name)
                    497:        char *name;
                    498: {
                    499:        struct protox **tpp, *tp;
                    500:
                    501:        for (tpp = protoprotox; *tpp; tpp++)
                    502:                for (tp = *tpp; tp->pr_name; tp++)
                    503:                        if (strcmp(tp->pr_name, name) == 0)
                    504:                                return (tp);
                    505:        return (NULL);
                    506: }
                    507:
                    508: /*
                    509:  * Find the protox corresponding to name.
                    510:  */
                    511: static struct protox *
                    512: name2protox(name)
                    513:        char *name;
                    514: {
                    515:        struct protox *tp;
                    516:        char **alias;                   /* alias from p->aliases */
                    517:        struct protoent *p;
                    518:
                    519:        /*
                    520:         * Try to find the name in the list of "well-known" names. If that
                    521:         * fails, check if name is an alias for an Internet protocol.
                    522:         */
                    523:        if (tp = knownname(name))
                    524:                return (tp);
                    525:
                    526:        setprotoent(1);                 /* make protocol lookup cheaper */
                    527:        while (p = getprotoent()) {
                    528:                /* assert: name not same as p->name */
                    529:                for (alias = p->p_aliases; *alias; alias++)
                    530:                        if (strcmp(name, *alias) == 0) {
                    531:                                endprotoent();
                    532:                                return (knownname(p->p_name));
                    533:                        }
                    534:        }
                    535:        endprotoent();
                    536:        return (NULL);
                    537: }
                    538:
                    539: static void
                    540: usage()
                    541: {
                    542:        (void)fprintf(stderr,
1.2       deraadt   543: "usage: %s [-Aan] [-f address_family] [-M core] [-N system]\n", __progname);
1.1       deraadt   544:        (void)fprintf(stderr,
1.2       deraadt   545: "       %s [-ghimnrs] [-f address_family] [-M core] [-N system]\n", __progname);
1.1       deraadt   546:        (void)fprintf(stderr,
1.2       deraadt   547: "       %s [-n] [-I interface] [-M core] [-N system] [-w wait]\n", __progname);
1.1       deraadt   548:        (void)fprintf(stderr,
1.2       deraadt   549: "       %s [-M core] [-N system] [-p protocol]\n", __progname);
1.1       deraadt   550:        exit(1);
                    551: }