NetBSD/usr.bin/netstat/main.c

510 lines
11 KiB
C

/*
* Copyright (c) 1983, 1988 Regents of the University of California.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by the University of
* California, Berkeley and its contributors.
* 4. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#ifndef lint
char copyright[] =
"@(#) Copyright (c) 1983, 1988 Regents of the University of California.\n\
All rights reserved.\n";
#endif /* not lint */
#ifndef lint
/*static char sccsid[] = "from: @(#)main.c 5.23 (Berkeley) 7/1/91";*/
static char rcsid[] = "$Id: main.c,v 1.3 1993/12/22 04:00:31 cgd Exp $";
#endif /* not lint */
#include <sys/param.h>
#include <sys/socket.h>
#include <sys/file.h>
#include <errno.h>
#include <netdb.h>
#include <nlist.h>
#include <kvm.h>
#include <stdio.h>
#include <ctype.h>
#include <stdlib.h>
#include <string.h>
#include <paths.h>
struct nlist nl[] = {
#define N_MBSTAT 0
{ "_mbstat" },
#define N_IPSTAT 1
{ "_ipstat" },
#define N_TCB 2
{ "_tcb" },
#define N_TCPSTAT 3
{ "_tcpstat" },
#define N_UDB 4
{ "_udb" },
#define N_UDPSTAT 5
{ "_udpstat" },
#define N_IFNET 6
{ "_ifnet" },
#define N_IMP 7
{ "_imp_softc" },
#define N_RTHOST 8
{ "_rthost" },
#define N_RTNET 9
{ "_rtnet" },
#define N_ICMPSTAT 10
{ "_icmpstat" },
#define N_RTSTAT 11
{ "_rtstat" },
#define N_FILEHEAD 12
{ "_filehead" },
#define N_FILES 13
{ "_nfiles" },
#define N_UNIXSW 14
{ "_unixsw" },
#define N_RTHASHSIZE 15
{ "_rthashsize" },
#define N_IDP 16
{ "_nspcb"},
#define N_IDPSTAT 17
{ "_idpstat"},
#define N_SPPSTAT 18
{ "_spp_istat"},
#define N_NSERR 19
{ "_ns_errstat"},
#define N_CLNPSTAT 20
{ "_clnp_stat"},
#define IN_TP 21
{ "_tp_inpcb" },
#define ISO_TP 22
{ "_tp_isopcb" },
#define N_TPSTAT 23
{ "_tp_stat" },
#define N_ESISSTAT 24
{ "_esis_stat"},
#define N_NIMP 25
{ "_nimp"},
#define N_RTREE 26
{ "_radix_node_head"},
#define N_CLTP 27
{ "_cltb"},
#define N_CLTPSTAT 28
{ "_cltpstat"},
"",
};
/* internet protocols */
extern int protopr();
extern int tcp_stats(), udp_stats(), ip_stats(), icmp_stats();
#ifdef NS
/* ns protocols */
extern int nsprotopr();
extern int spp_stats(), idp_stats(), nserr_stats();
#endif
#ifdef ISO
/* iso protocols */
extern int iso_protopr();
extern int tp_stats(), esis_stats(), clnp_stats(), cltp_stats();
#endif
struct protox {
u_char pr_index; /* index into nlist of cb head */
u_char pr_sindex; /* index into nlist of stat block */
u_char pr_wanted; /* 1 if wanted, 0 otherwise */
int (*pr_cblocks)(); /* control blocks printing routine */
int (*pr_stats)(); /* statistics printing routine */
char *pr_name; /* well-known name */
} protox[] = {
{ N_TCB, N_TCPSTAT, 1, protopr,
tcp_stats, "tcp" },
{ N_UDB, N_UDPSTAT, 1, protopr,
udp_stats, "udp" },
#ifdef ISO
{ IN_TP, N_TPSTAT, 1, protopr,
tp_stats, "tpip" },
#endif
{ -1, N_IPSTAT, 1, 0,
ip_stats, "ip" },
{ -1, N_ICMPSTAT, 1, 0,
icmp_stats, "icmp" },
{ -1, -1, 0, 0,
0, 0 }
};
#ifdef NS
struct protox nsprotox[] = {
{ N_IDP, N_IDPSTAT, 1, nsprotopr,
idp_stats, "idp" },
{ N_IDP, N_SPPSTAT, 1, nsprotopr,
spp_stats, "spp" },
{ -1, N_NSERR, 1, 0,
nserr_stats, "ns_err" },
{ -1, -1, 0, 0,
0, 0 }
};
#endif
#ifdef ISO
struct protox isoprotox[] = {
{ ISO_TP, N_TPSTAT, 1, iso_protopr,
tp_stats, "tp" },
{ N_CLTP, N_CLTPSTAT, 1, iso_protopr,
cltp_stats, "cltp" },
{ -1, N_CLNPSTAT, 1, 0,
clnp_stats, "clnp"},
{ -1, N_ESISSTAT, 1, 0,
esis_stats, "esis"},
{ -1, -1, 0, 0,
0, 0 }
};
#endif
struct protox *protoprotox[] = { protox,
#ifdef NS
nsprotox,
#endif
#ifdef ISO
isoprotox,
#endif
NULL };
char *vmunix = _PATH_UNIX;
char *kmemf;
int kmem;
int kflag;
int Aflag;
int aflag;
int hflag;
int iflag;
int mflag;
int nflag;
int pflag;
int rflag;
int sflag;
int tflag;
int dflag;
int interval;
char *interface;
int unit;
int af = AF_UNSPEC;
main(argc, argv)
int argc;
char **argv;
{
extern char *optarg;
extern int optind;
register struct protoent *p;
register struct protox *tp; /* for printing cblocks & stats */
struct protox *name2protox(); /* for -p */
int ch;
void usage();
while ((ch = getopt(argc, argv, "Aadf:hI:iM:mN:np:rstuw:")) != EOF)
switch((char)ch) {
case 'A':
Aflag = 1;
break;
case 'a':
aflag = 1;
break;
case 'd':
dflag = 1;
break;
case 'f':
if (strcmp(optarg, "ns") == 0)
af = AF_NS;
else if (strcmp(optarg, "inet") == 0)
af = AF_INET;
else if (strcmp(optarg, "unix") == 0)
af = AF_UNIX;
else if (strcmp(optarg, "iso") == 0)
af = AF_ISO;
else {
(void)fprintf(stderr,
"%s: unknown address family\n", optarg);
exit(1);
}
break;
case 'h':
hflag = 1;
break;
case 'I': {
char *cp;
iflag = 1;
for (cp = interface = optarg; isalpha(*cp); cp++);
unit = atoi(cp);
*cp = '\0';
break;
}
case 'i':
iflag = 1;
break;
case 'M':
kmemf = optarg;
kflag = 1;
break;
case 'm':
mflag = 1;
break;
case 'N':
vmunix = optarg;
break;
case 'n':
nflag = 1;
break;
case 'p':
if ((tp = name2protox(optarg)) == NULL) {
(void)fprintf(stderr,
"%s: unknown or uninstrumented protocol\n",
optarg);
exit(1);
}
pflag = 1;
break;
case 'r':
rflag = 1;
break;
case 's':
sflag = 1;
break;
case 't':
tflag = 1;
break;
case 'u':
af = AF_UNIX;
break;
case 'w':
interval = atoi(optarg);
iflag = 1;
break;
case '?':
default:
usage();
}
argv += optind;
argc -= optind;
#define BACKWARD_COMPATIBILITY
#ifdef BACKWARD_COMPATIBILITY
if (*argv) {
if (isdigit(**argv)) {
interval = atoi(*argv);
if (interval <= 0)
usage();
++argv;
iflag = 1;
}
if (*argv) {
vmunix = *argv;
if (*++argv) {
kmemf = *argv;
kflag = 1;
}
}
}
#endif
if (kvm_openfiles(vmunix, kmemf, NULL) == -1) {
fprintf(stderr, "netstat: kvm_openfiles: %s\n", kvm_geterr());
exit(1);
}
if (kvm_nlist(nl) < 0 || nl[0].n_type == 0) {
fprintf(stderr, "%s: no namelist\n", vmunix);
exit(1);
}
if (mflag) {
mbpr((off_t)nl[N_MBSTAT].n_value);
exit(0);
}
if (pflag) {
if (tp->pr_stats)
(*tp->pr_stats)(nl[tp->pr_sindex].n_value,
tp->pr_name);
else
printf("%s: no stats routine\n", tp->pr_name);
exit(0);
}
if (hflag) {
#ifdef IMP
hostpr(nl[N_IMP].n_value, nl[N_NIMP].n_value);
#else
(void)fprintf(stderr, "netstat: IMP code not compiled in\n");
#endif
exit(0);
}
/*
* Keep file descriptors open to avoid overhead
* of open/close on each call to get* routines.
*/
sethostent(1);
setnetent(1);
if (iflag) {
intpr(interval, nl[N_IFNET].n_value);
exit(0);
}
if (rflag) {
if (sflag)
rt_stats((off_t)nl[N_RTSTAT].n_value);
else
routepr((off_t)nl[N_RTHOST].n_value,
(off_t)nl[N_RTNET].n_value,
(off_t)nl[N_RTHASHSIZE].n_value,
(off_t)nl[N_RTREE].n_value);
exit(0);
}
if (af == AF_INET || af == AF_UNSPEC) {
setprotoent(1);
setservent(1);
while (p = getprotoent()) {
for (tp = protox; tp->pr_name; tp++)
if (strcmp(tp->pr_name, p->p_name) == 0)
break;
if (tp->pr_name == 0 || tp->pr_wanted == 0)
continue;
if (sflag) {
if (tp->pr_stats)
(*tp->pr_stats)(nl[tp->pr_sindex].n_value,
p->p_name);
} else
if (tp->pr_cblocks)
(*tp->pr_cblocks)(nl[tp->pr_index].n_value,
p->p_name);
}
endprotoent();
}
if (af == AF_NS || af == AF_UNSPEC) {
#ifdef NS
for (tp = nsprotox; tp->pr_name; tp++) {
if (sflag) {
if (tp->pr_stats)
(*tp->pr_stats)(nl[tp->pr_sindex].n_value,
tp->pr_name);
} else
if (tp->pr_cblocks)
(*tp->pr_cblocks)(nl[tp->pr_index].n_value,
tp->pr_name);
}
#endif
}
if (af == AF_ISO || af == AF_UNSPEC) {
#ifdef ISO
for (tp = isoprotox; tp->pr_name; tp++) {
if (sflag) {
if (tp->pr_stats)
(*tp->pr_stats)(nl[tp->pr_sindex].n_value,
tp->pr_name);
} else
if (tp->pr_cblocks)
(*tp->pr_cblocks)(nl[tp->pr_index].n_value,
tp->pr_name);
}
#endif
}
if ((af == AF_UNIX || af == AF_UNSPEC) && !sflag)
unixpr((off_t)nl[N_FILEHEAD].n_value, (off_t)nl[N_FILES].n_value,
(struct protosw *)nl[N_UNIXSW].n_value);
if (af == AF_UNSPEC && sflag)
#ifdef IMP
impstats(nl[N_IMP].n_value, nl[N_NIMP].n_value);
#endif
exit(0);
}
char *
plural(n)
int n;
{
return (n != 1 ? "s" : "");
}
/*
* Find the protox for the given "well-known" name.
*/
struct protox *
knownname(name)
char *name;
{
struct protox **tpp, *tp;
for (tpp = protoprotox; *tpp; tpp++)
for (tp = *tpp; tp->pr_name; tp++)
if (strcmp(tp->pr_name, name) == 0)
return(tp);
return(NULL);
}
/*
* Find the protox corresponding to name.
*/
struct protox *
name2protox(name)
char *name;
{
struct protox *tp;
char **alias; /* alias from p->aliases */
struct protoent *p;
/*
* Try to find the name in the list of "well-known" names. If that
* fails, check if name is an alias for an Internet protocol.
*/
if (tp = knownname(name))
return(tp);
setprotoent(1); /* make protocol lookup cheaper */
while (p = getprotoent()) {
/* assert: name not same as p->name */
for (alias = p->p_aliases; *alias; alias++)
if (strcmp(name, *alias) == 0) {
endprotoent();
return(knownname(p->p_name));
}
}
endprotoent();
return(NULL);
}
void
usage()
{
(void)fprintf(stderr,
"usage: netstat [-Aan] [-f address_family] [-M core] [-N system]\n");
(void)fprintf(stderr,
" [-himnrs] [-f address_family] [-M core] [-N system]\n");
(void)fprintf(stderr,
" [-n] [-I interface] [-M core] [-N system] [-w wait]\n");
(void)fprintf(stderr,
" [-M core] [-N system] [-p protocol]\n");
exit(1);
}