NetBSD/usr.bin/netstat/main.c

574 lines
13 KiB
C
Raw Normal View History

/* $NetBSD: main.c,v 1.15 1998/07/06 07:50:19 mrg Exp $ */
1995-10-04 00:42:34 +03:00
1993-03-21 21:04:42 +03:00
/*
1994-05-13 12:08:09 +04:00
* Copyright (c) 1983, 1988, 1993
* Regents of the University of California. All rights reserved.
1993-03-21 21:04:42 +03:00
*
* 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.
*/
#include <sys/cdefs.h>
1993-03-21 21:04:42 +03:00
#ifndef lint
__COPYRIGHT("@(#) Copyright (c) 1983, 1988, 1993\n\
Regents of the University of California. All rights reserved.\n");
1993-03-21 21:04:42 +03:00
#endif /* not lint */
#ifndef lint
1995-10-04 00:42:34 +03:00
#if 0
static char sccsid[] = "from: @(#)main.c 8.4 (Berkeley) 3/1/94";
#else
__RCSID("$NetBSD: main.c,v 1.15 1998/07/06 07:50:19 mrg Exp $");
1995-10-04 00:42:34 +03:00
#endif
1993-03-21 21:04:42 +03:00
#endif /* not lint */
#include <sys/param.h>
#include <sys/file.h>
1994-05-13 12:08:09 +04:00
#include <sys/protosw.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <ctype.h>
#include <err.h>
1993-03-21 21:04:42 +03:00
#include <errno.h>
1994-05-13 12:08:09 +04:00
#include <kvm.h>
#include <limits.h>
1993-03-21 21:04:42 +03:00
#include <netdb.h>
#include <nlist.h>
1994-05-13 12:08:09 +04:00
#include <paths.h>
1993-03-21 21:04:42 +03:00
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
1994-05-13 12:08:09 +04:00
#include <unistd.h>
#include "netstat.h"
1993-03-21 21:04:42 +03:00
struct nlist nl[] = {
#define N_MBSTAT 0
{ "_mbstat" },
#define N_IPSTAT 1
{ "_ipstat" },
1995-06-12 07:03:07 +04:00
#define N_TCBTABLE 2
{ "_tcbtable" },
1993-03-21 21:04:42 +03:00
#define N_TCPSTAT 3
{ "_tcpstat" },
1995-06-12 07:03:07 +04:00
#define N_UDBTABLE 4
{ "_udbtable" },
1993-03-21 21:04:42 +03:00
#define N_UDPSTAT 5
{ "_udpstat" },
#define N_IFNET 6
{ "_ifnet" },
#define N_IMP 7
{ "_imp_softc" },
1994-05-13 12:08:09 +04:00
#define N_ICMPSTAT 8
1993-03-21 21:04:42 +03:00
{ "_icmpstat" },
1994-05-13 12:08:09 +04:00
#define N_RTSTAT 9
1993-03-21 21:04:42 +03:00
{ "_rtstat" },
1994-05-13 12:08:09 +04:00
#define N_UNIXSW 10
1993-03-21 21:04:42 +03:00
{ "_unixsw" },
1994-05-13 12:08:09 +04:00
#define N_IDP 11
1993-03-21 21:04:42 +03:00
{ "_nspcb"},
1994-05-13 12:08:09 +04:00
#define N_IDPSTAT 12
1993-03-21 21:04:42 +03:00
{ "_idpstat"},
1994-05-13 12:08:09 +04:00
#define N_SPPSTAT 13
1993-03-21 21:04:42 +03:00
{ "_spp_istat"},
1994-05-13 12:08:09 +04:00
#define N_NSERR 14
1993-03-21 21:04:42 +03:00
{ "_ns_errstat"},
1994-05-13 12:08:09 +04:00
#define N_CLNPSTAT 15
1993-03-21 21:04:42 +03:00
{ "_clnp_stat"},
1994-05-13 12:08:09 +04:00
#define IN_NOTUSED 16
1993-03-21 21:04:42 +03:00
{ "_tp_inpcb" },
1994-05-13 12:08:09 +04:00
#define ISO_TP 17
{ "_tp_refinfo" },
#define N_TPSTAT 18
1993-03-21 21:04:42 +03:00
{ "_tp_stat" },
1994-05-13 12:08:09 +04:00
#define N_ESISSTAT 19
1993-03-21 21:04:42 +03:00
{ "_esis_stat"},
1994-05-13 12:08:09 +04:00
#define N_NIMP 20
1993-03-21 21:04:42 +03:00
{ "_nimp"},
1994-05-13 12:08:09 +04:00
#define N_RTREE 21
{ "_rt_tables"},
#define N_CLTP 22
1993-03-21 21:04:42 +03:00
{ "_cltb"},
1994-05-13 12:08:09 +04:00
#define N_CLTPSTAT 23
1993-03-21 21:04:42 +03:00
{ "_cltpstat"},
1994-05-13 12:08:09 +04:00
#define N_NFILE 24
{ "_nfile" },
#define N_FILE 25
{ "_file" },
#define N_IGMPSTAT 26
{ "_igmpstat" },
1994-05-13 12:08:09 +04:00
#define N_MRTPROTO 27
{ "_ip_mrtproto" },
1994-05-13 12:08:09 +04:00
#define N_MRTSTAT 28
{ "_mrtstat" },
1995-06-12 07:03:07 +04:00
#define N_MFCHASHTBL 29
{ "_mfchashtbl" },
#define N_MFCHASH 30
{ "_mfchash" },
#define N_VIFTABLE 31
{ "_viftable" },
#define N_MSIZE 32
{ "_msize" },
#define N_MCLBYTES 33
{ "_mclbytes" },
#define N_DDPSTAT 34
{ "_ddpstat"},
#define N_DDPCB 35
{ "_ddpcb"},
{ "" },
1993-03-21 21:04:42 +03:00
};
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 */
void (*pr_cblocks) /* control blocks printing routine */
__P((u_long, char *));
void (*pr_stats) /* statistics printing routine */
__P((u_long, char *));
void (*pr_dump) /* PCB state dump routine */
__P((u_long));
1993-03-21 21:04:42 +03:00
char *pr_name; /* well-known name */
} protox[] = {
1995-06-12 07:03:07 +04:00
{ N_TCBTABLE, N_TCPSTAT, 1, protopr,
tcp_stats, tcp_dump, "tcp" },
1995-06-12 07:03:07 +04:00
{ N_UDBTABLE, N_UDPSTAT, 1, protopr,
udp_stats, 0, "udp" },
1993-03-21 21:04:42 +03:00
{ -1, N_IPSTAT, 1, 0,
ip_stats, 0, "ip" },
1993-03-21 21:04:42 +03:00
{ -1, N_ICMPSTAT, 1, 0,
icmp_stats, 0, "icmp" },
{ -1, N_IGMPSTAT, 1, 0,
igmp_stats, 0, "igmp" },
1993-03-21 21:04:42 +03:00
{ -1, -1, 0, 0,
0, 0, 0 }
1993-03-21 21:04:42 +03:00
};
struct protox atalkprotox[] = {
{ N_DDPCB, N_DDPSTAT, 1, atalkprotopr,
ddp_stats, 0, "ddp" },
{ -1, -1, 0, 0,
0, 0 }
};
1993-03-21 21:04:42 +03:00
struct protox nsprotox[] = {
{ N_IDP, N_IDPSTAT, 1, nsprotopr,
idp_stats, 0, "idp" },
1993-03-21 21:04:42 +03:00
{ N_IDP, N_SPPSTAT, 1, nsprotopr,
spp_stats, 0, "spp" },
1993-03-21 21:04:42 +03:00
{ -1, N_NSERR, 1, 0,
nserr_stats, 0, "ns_err" },
1993-03-21 21:04:42 +03:00
{ -1, -1, 0, 0,
0, 0 }
};
struct protox isoprotox[] = {
{ ISO_TP, N_TPSTAT, 1, iso_protopr,
tp_stats, 0, "tp" },
1993-03-21 21:04:42 +03:00
{ N_CLTP, N_CLTPSTAT, 1, iso_protopr,
cltp_stats, 0, "cltp" },
1993-03-21 21:04:42 +03:00
{ -1, N_CLNPSTAT, 1, 0,
clnp_stats, 0, "clnp"},
1993-03-21 21:04:42 +03:00
{ -1, N_ESISSTAT, 1, 0,
esis_stats, 0, "esis"},
1993-03-21 21:04:42 +03:00
{ -1, -1, 0, 0,
0, 0, 0 }
1993-03-21 21:04:42 +03:00
};
struct protox *protoprotox[] = { protox, atalkprotox,
nsprotox, isoprotox, NULL };
1993-03-21 21:04:42 +03:00
int main __P((int, char *[]));
1994-05-13 12:08:09 +04:00
static void printproto __P((struct protox *, char *));
static void usage __P((void));
static struct protox *name2protox __P((char *));
static struct protox *knownname __P((char *));
1993-03-21 21:04:42 +03:00
1994-05-13 12:08:09 +04:00
kvm_t *kvmd;
1993-03-21 21:04:42 +03:00
1994-05-13 12:08:09 +04:00
int
1993-03-21 21:04:42 +03:00
main(argc, argv)
int argc;
1994-05-13 12:08:09 +04:00
char *argv[];
1993-03-21 21:04:42 +03:00
{
extern char *optarg;
extern int optind;
struct protoent *p;
struct protox *tp; /* for printing cblocks & stats */
1993-03-21 21:04:42 +03:00
int ch;
1994-05-13 12:08:09 +04:00
char *nlistf = NULL, *memf = NULL;
char buf[_POSIX2_LINE_MAX], *cp;
u_long pcbaddr;
gid_t egid = getegid();
1994-05-13 12:08:09 +04:00
(void)setegid(getgid());
tp = NULL;
1994-05-13 12:08:09 +04:00
af = AF_UNSPEC;
pcbaddr = 0;
1993-03-21 21:04:42 +03:00
while ((ch = getopt(argc, argv, "Aabdf:ghI:iM:mN:nP:p:rstuw:")) != -1)
1994-05-13 12:08:09 +04:00
switch(ch) {
1993-03-21 21:04:42 +03:00
case 'A':
Aflag = 1;
break;
case 'a':
aflag = 1;
break;
case 'b':
bflag = 1;
break;
1993-03-21 21:04:42 +03:00
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 if (strcmp(optarg, "atalk") == 0)
af = AF_APPLETALK;
1993-03-21 21:04:42 +03:00
else {
(void)fprintf(stderr,
1994-05-13 12:08:09 +04:00
"%s: %s: unknown address family\n",
__progname, optarg);
1993-03-21 21:04:42 +03:00
exit(1);
}
break;
1994-05-13 12:08:09 +04:00
case 'g':
gflag = 1;
1993-03-21 21:04:42 +03:00
break;
case 'I':
1993-03-21 21:04:42 +03:00
iflag = 1;
interface = optarg;
1993-03-21 21:04:42 +03:00
break;
case 'i':
iflag = 1;
break;
case 'M':
1994-05-13 12:08:09 +04:00
memf = optarg;
1993-03-21 21:04:42 +03:00
break;
case 'm':
mflag = 1;
break;
case 'N':
1994-05-13 12:08:09 +04:00
nlistf = optarg;
1993-03-21 21:04:42 +03:00
break;
case 'n':
nflag = 1;
break;
case 'P':
pcbaddr = strtoul(optarg, &cp, 16);
if (*cp != '\0' || errno == ERANGE) {
(void)fprintf(stderr,
"%s: invalid PCB address %s\n",
__progname, optarg);
exit(1);
}
Pflag = 1;
break;
1993-03-21 21:04:42 +03:00
case 'p':
if ((tp = name2protox(optarg)) == NULL) {
(void)fprintf(stderr,
1994-05-13 12:08:09 +04:00
"%s: %s: unknown or uninstrumented protocol\n",
__progname, optarg);
1993-03-21 21:04:42 +03:00
exit(1);
}
pflag = 1;
break;
case 'r':
rflag = 1;
break;
case 's':
1994-05-13 12:08:09 +04:00
++sflag;
1993-03-21 21:04:42 +03:00
break;
case 't':
tflag = 1;
break;
case 'u':
af = AF_UNIX;
break;
case 'w':
interval = atoi(optarg);
1994-05-13 12:08:09 +04:00
iflag = 1;
1993-03-21 21:04:42 +03:00
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) {
1994-05-13 12:08:09 +04:00
nlistf = *argv;
if (*++argv)
memf = *argv;
1993-03-21 21:04:42 +03:00
}
}
#endif
1994-05-13 12:08:09 +04:00
/*
* Discard setgid privileges. If not the running kernel, we toss
* them away totally so that bad guys can't print interesting stuff
* from kernel memory, otherwise switch back to kmem for the
* duration of the kvm_openfiles() call.
1994-05-13 12:08:09 +04:00
*/
if (nlistf != NULL || memf != NULL)
(void)setgid(getgid());
else
(void)setegid(egid);
1994-05-13 12:08:09 +04:00
if ((kvmd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY,
buf)) == NULL)
errx(1, "%s", buf);
if (nlistf == NULL && memf == NULL)
(void)setgid(getgid());
1994-05-13 12:08:09 +04:00
if (kvm_nlist(kvmd, nl) < 0 || nl[0].n_type == 0) {
if (nlistf)
fprintf(stderr, "%s: %s: no namelist\n", __progname,
nlistf);
1994-05-13 12:08:09 +04:00
else
fprintf(stderr, "%s: no namelist\n", __progname);
1993-03-21 21:04:42 +03:00
exit(1);
}
if (mflag) {
mbpr(nl[N_MBSTAT].n_value, nl[N_MSIZE].n_value,
nl[N_MCLBYTES].n_value);
1993-03-21 21:04:42 +03:00
exit(0);
}
if (Pflag) {
if (tp == NULL) {
/* Default to TCP. */
tp = name2protox("tcp");
}
if (tp->pr_dump)
(*tp->pr_dump)(pcbaddr);
else
printf("%s: no PCB dump routine\n", tp->pr_name);
exit(0);
}
1993-03-21 21:04:42 +03:00
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);
}
/*
* 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)
1994-04-01 13:18:06 +04:00
rt_stats(nl[N_RTSTAT].n_value);
1993-03-21 21:04:42 +03:00
else
1994-05-13 12:08:09 +04:00
routepr(nl[N_RTREE].n_value);
1993-03-21 21:04:42 +03:00
exit(0);
}
1994-05-13 12:08:09 +04:00
if (gflag) {
if (sflag)
1994-04-01 13:18:06 +04:00
mrt_stats(nl[N_MRTPROTO].n_value,
1994-05-13 12:08:09 +04:00
nl[N_MRTSTAT].n_value);
else
1994-04-01 13:18:06 +04:00
mroutepr(nl[N_MRTPROTO].n_value,
1995-06-12 07:03:07 +04:00
nl[N_MFCHASHTBL].n_value,
nl[N_MFCHASH].n_value,
1994-05-13 12:08:09 +04:00
nl[N_VIFTABLE].n_value);
exit(0);
}
1994-05-13 12:08:09 +04:00
if (af == AF_INET || af == AF_UNSPEC) {
setprotoent(1);
setservent(1);
/* ugh, this is O(MN) ... why do we do this? */
while ((p = getprotoent()) != NULL) {
1994-05-13 12:08:09 +04:00
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;
printproto(tp, p->p_name);
}
endprotoent();
}
if (af == AF_APPLETALK || af == AF_UNSPEC)
for (tp = atalkprotox; tp->pr_name; tp++)
printproto(tp, tp->pr_name);
1994-05-13 12:08:09 +04:00
if (af == AF_NS || af == AF_UNSPEC)
for (tp = nsprotox; tp->pr_name; tp++)
printproto(tp, tp->pr_name);
if (af == AF_ISO || af == AF_UNSPEC)
for (tp = isoprotox; tp->pr_name; tp++)
printproto(tp, tp->pr_name);
if ((af == AF_UNIX || af == AF_UNSPEC) && !sflag)
unixpr(nl[N_UNIXSW].n_value);
exit(0);
}
1994-05-13 12:08:09 +04:00
/*
* Print out protocol statistics or control blocks (per sflag).
* If the interface was not specifically requested, and the symbol
* is not in the namelist, ignore this one.
*/
static void
printproto(tp, name)
struct protox *tp;
1994-05-13 12:08:09 +04:00
char *name;
{
void (*pr) __P((u_long, char *));
1994-05-13 12:08:09 +04:00
u_long off;
1993-03-21 21:04:42 +03:00
1994-05-13 12:08:09 +04:00
if (sflag) {
pr = tp->pr_stats;
off = nl[tp->pr_sindex].n_value;
} else {
pr = tp->pr_cblocks;
off = nl[tp->pr_index].n_value;
1993-03-21 21:04:42 +03:00
}
1994-05-13 12:08:09 +04:00
if (pr != NULL && (off || af != AF_UNSPEC))
(*pr)(off, name);
}
/*
* Read kernel memory, return 0 on success.
*/
int
kread(addr, buf, size)
u_long addr;
char *buf;
int size;
{
if (kvm_read(kvmd, addr, buf, size) != size) {
(void)fprintf(stderr, "%s: %s\n", __progname,
1994-05-13 12:08:09 +04:00
kvm_geterr(kvmd));
return (-1);
1993-03-21 21:04:42 +03:00
}
1994-05-13 12:08:09 +04:00
return (0);
1993-03-21 21:04:42 +03:00
}
char *
plural(n)
int n;
{
return (n != 1 ? "s" : "");
}
1994-05-13 12:08:09 +04:00
char *
plurales(n)
int n;
{
return (n != 1 ? "es" : "");
}
1993-03-21 21:04:42 +03:00
/*
* Find the protox for the given "well-known" name.
*/
1994-05-13 12:08:09 +04:00
static struct protox *
1993-03-21 21:04:42 +03:00
knownname(name)
char *name;
{
struct protox **tpp, *tp;
for (tpp = protoprotox; *tpp; tpp++)
1994-05-13 12:08:09 +04:00
for (tp = *tpp; tp->pr_name; tp++)
if (strcmp(tp->pr_name, name) == 0)
return (tp);
return (NULL);
1993-03-21 21:04:42 +03:00
}
/*
* Find the protox corresponding to name.
*/
1994-05-13 12:08:09 +04:00
static struct protox *
1993-03-21 21:04:42 +03:00
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)) != NULL)
1994-05-13 12:08:09 +04:00
return (tp);
1993-03-21 21:04:42 +03:00
setprotoent(1); /* make protocol lookup cheaper */
while ((p = getprotoent()) != NULL) {
1993-03-21 21:04:42 +03:00
/* assert: name not same as p->name */
for (alias = p->p_aliases; *alias; alias++)
if (strcmp(name, *alias) == 0) {
endprotoent();
1994-05-13 12:08:09 +04:00
return (knownname(p->p_name));
1993-03-21 21:04:42 +03:00
}
}
endprotoent();
1994-05-13 12:08:09 +04:00
return (NULL);
1993-03-21 21:04:42 +03:00
}
1994-05-13 12:08:09 +04:00
static void
1993-03-21 21:04:42 +03:00
usage()
{
(void)fprintf(stderr,
"usage: %s [-Aan] [-f address_family] [-M core] [-N system]\n", __progname);
1993-03-21 21:04:42 +03:00
(void)fprintf(stderr,
" %s [-ghimnrs] [-f address_family] [-M core] [-N system]\n", __progname);
1993-03-21 21:04:42 +03:00
(void)fprintf(stderr,
" %s [-n] [-I interface] [-M core] [-N system] [-w wait]\n", __progname);
1993-03-21 21:04:42 +03:00
(void)fprintf(stderr,
" %s [-M core] [-N system] [-p protocol]\n", __progname);
(void)fprintf(stderr,
" %s [-M core] [-N system] [-p protocol] -P pcbaddr\n", __progname);
1993-03-21 21:04:42 +03:00
exit(1);
}