net-snmp/apps/snmpnetstat/main.c

577 lines
16 KiB
C

/* $OpenBSD: main.c,v 1.52 2005/02/10 14:25:08 itojun Exp $ */
/* $NetBSD: main.c,v 1.9 1996/05/07 02:55:02 thorpej Exp $ */
/*
* Copyright (c) 1983, 1988, 1993
* 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. 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, 1993\n\
Regents of the University of California. All rights reserved.\n";
#endif /* not lint */
#ifdef INHERITED_CODE
#ifndef lint
#if 0
static char sccsid[] = "from: @(#)main.c 8.4 (Berkeley) 3/1/94";
#else
static char *rcsid = "$OpenBSD: main.c,v 1.52 2005/02/10 14:25:08 itojun Exp $";
#endif
#endif /* not lint */
#endif
#include <net-snmp/net-snmp-config.h>
#include <net-snmp/net-snmp-includes.h>
#include <net-snmp/utilities.h>
#if HAVE_NETDB_H
#include <netdb.h>
#endif
#include "main.h"
#include "netstat.h"
#if HAVE_WINSOCK_H
#include "winstub.h"
#endif
int aflag; /* show all sockets (including servers) */
int bflag; /* show bytes instead of packets */
int dflag; /* show i/f dropped packets */
int gflag; /* show group (multicast) routing or stats */
int iflag; /* show interfaces */
int lflag; /* show routing table with use and ref */
int Lflag; /* Legacy mibs */
int mflag; /* show memory stats */
int nflag; /* show addresses numerically */
int oflag; /* Open/Net-BSD style octet output */
int pflag; /* show given protocol */
int qflag; /* only display non-zero values for output */
int rflag; /* show routing tables (or routing stats) */
int Sflag; /* show source address in routing table */
int sflag; /* show protocol statistics */
int tflag; /* show i/f watchdog timers */
int vflag; /* be verbose */
int interval; /* repeat interval for i/f stats */
char *intrface; /* desired i/f for stats, or NULL for all i/fs */
int af; /* address family */
int max_getbulk = 32; /* specifies the max-repeaters value to use with GETBULK requests */
char *progname = NULL;
const char *pname;
/*
* struct nlist nl[] - Omitted
*/
typedef void (stringfun)(const char*);
struct protox {
/* pr_index/pr_sindex - Omitted */
int pr_wanted; /* 1 if wanted, 0 otherwise */
stringfun *pr_cblocks; /* control blocks printing routine */
stringfun *pr_stats; /* statistics printing routine */
const char *pr_name; /* well-known name */
};
struct protox protox[] = {
{ 1, tcpprotopr, tcp_stats, "tcp" },
{ 1, udpprotopr, udp_stats, "udp" },
{ 1, (stringfun*)0, ip_stats, "ip" }, /* protopr Omitted */
{ 1, (stringfun*)0, icmp_stats, "icmp" },
/* igmp/ah/esp/ipencap/etherip/ipcomp/carp/pfsync/pim - Omitted */
{ 0, (stringfun*)0, (stringfun*)0, NULL }
};
struct protox ip6protox[] = {
{ 1, tcp6protopr, (stringfun*)0, "tcp6" },
{ 1, udp6protopr, (stringfun*)0, "udp6" },
{ 1, (stringfun*)0, ip6_stats, "ip6" },/* ip6protopr Omitted */
{ 1, (stringfun*)0, icmp6_stats, "icmp6" },
/* pim6/rip6 - Omitted */
{ 0, (stringfun*)0, (stringfun*)0, NULL }
};
struct protox ipxprotox[] = {
{ 1, tcpxprotopr, tcp_stats, "tcp" },
{ 1, udpxprotopr, udp_stats, "udp" },
{ 1, (stringfun*)0, ipx_stats, "ip" },/* ip6protopr Omitted */
{ 1, (stringfun*)0, ipx_stats, "ip6" },/* ip6protopr Omitted */
{ 1, (stringfun*)0, icmpx_stats, "icmp" },
{ 1, (stringfun*)0, icmpx_stats, "icmp6" },
{ 0, (stringfun*)0, (stringfun*)0, NULL }
};
/* {ipx,ns,atalk}protox Omitted */
struct protox *protoprotox[] = {
protox, ip6protox, NULL
};
static void printproto(struct protox *, const char *);
static void usage(void);
static struct protox *name2protox(const char *);
static struct protox *knownname(const char *);
netsnmp_session *ss;
struct protox *tp = NULL; /* for printing cblocks & stats */
static void
optProc( int argc, char *const *argv, int opt )
{
switch (opt) {
case 'C':
while (*optarg) {
switch (*optarg++) {
case 'a':
aflag = 1;
break;
case 'b':
bflag = 1;
break;
case 'd':
dflag = 1;
break;
case 'f':
if (!*optarg)
optarg = argv[optind++];
if (strcmp(optarg, "inet") == 0)
af = AF_INET;
else if (strcmp(optarg, "inet6") == 0)
af = AF_INET6;
/*
else if (strcmp(optarg, "local") == 0)
af = AF_LOCAL;
else if (strcmp(optarg, "unix") == 0)
af = AF_UNIX;
else if (strcmp(optarg, "ipx") == 0)
af = AF_IPX;
else if (strcmp(optarg, "ns") == 0)
af = AF_NS;
else if (strcmp(optarg, "encap") == 0)
af = PF_KEY;
else if (strcmp(optarg, "atalk") == 0)
af = AF_APPLETALK;
*/
else {
(void)fprintf(stderr,
"%s: %s: unknown address family\n",
progname, optarg);
exit(1);
}
return;
case 'g':
gflag = 1;
break;
case 'I':
iflag = 1;
if (!*optarg)
optarg = argv[optind++];
intrface = optarg;
return;
case 'i':
iflag = 1;
break;
case 'L':
Lflag = 1;
break;
/* case 'L': FreeBSD: Display listen queue lengths
NetBSD: Suppress link-level routes */
/* case 'l': OpenBSD: Wider IPv6 display
Linux: Listening sockets only
lflag = 1;
break;
case 'M': *BSD: Memory image
Linux: Masqueraded connections
memf = optarg;
break;
*/
case 'm':
mflag = 1;
break;
/* case 'N': *BSD: Kernel image
nlistf = optarg;
break;
*/
case 'n':
nflag = 1;
break;
case 'o':
oflag = 1;
break;
/* case 'P': NetBSD:
OpenBSD: dump PCB block */
case 'p':
if (!*optarg)
optarg = argv[optind++];
if ((tp = name2protox(optarg)) == NULL) {
(void)fprintf(stderr,
"%s: %s: unknown protocol\n",
progname, optarg);
exit(1);
}
pflag = 1;
pname = tp->pr_name;
return;
/* case 'q': NetBSD: IRQ information
OpenBSD: Suppress inactive I/Fs
qflag = 1;
break;
*/
case 'r':
rflag = 1;
break;
case 'R':
if (optind < argc) {
if (argv[optind]) {
max_getbulk = atoi(argv[optind]);
if (max_getbulk == 0) {
usage();
fprintf(stderr, "Bad -CR option: %s\n",
argv[optind]);
exit(1);
}
}
} else {
usage();
fprintf(stderr, "Bad -CR option: no argument given\n");
exit(1);
}
optind++;
break;
case 'S': /* FreeBSD:
NetBSD: Semi-numeric display
OpenBSD: Show route source selector */
Sflag = 1;
break;
case 's':
++sflag;
break;
/* case 't': FreeBSD:
OpenBSD: Display watchdog timers
tflag = 1;
break;
case 'u': OpenBSD: unix sockets only
af = AF_UNIX;
break;
*/
case 'v':
vflag = 1;
break;
case 'w':
if (!*optarg)
optarg = argv[optind++];
interval = atoi(optarg);
iflag = 1;
return;
case '?':
default:
usage();
exit(1);
}
}
break; /* End of '-Cx' switch */
/*
* Backward compatability for the main display modes
* (where this doesn't clash with standard SNMP flags)
*/
case 'i':
iflag = 1;
break;
case 'R':
rflag = 1; /* -r sets the retry count */
break;
case 's':
++sflag;
break;
}
}
int
main(int argc, char *argv[])
{
netsnmp_session session;
struct protoent *p;
char *cp;
int exit_code = 1;
SOCK_STARTUP;
af = AF_UNSPEC;
cp = strrchr( argv[0], '/' );
if (cp)
progname = cp+1;
else
progname = argv[0];
switch (snmp_parse_args( argc, argv, &session, "C:iRs", optProc)) {
case NETSNMP_PARSE_ARGS_ERROR:
goto out;
case NETSNMP_PARSE_ARGS_SUCCESS_EXIT:
exit_code = 0;
goto out;
case NETSNMP_PARSE_ARGS_ERROR_USAGE:
usage();
goto out;
default:
break;
}
/*
* Check argc vs optind ??
*/
argv += optind;
argc -= optind;
/*
* Open an SNMP session.
*/
ss = snmp_open(&session);
if (ss == NULL) {
/*
* diagnose snmp_open errors with the input netsnmp_session pointer
*/
snmp_sess_perror("snmpnetstat", &session);
goto out;
}
/*
* Omitted:
* Privilege handling
* "Backward Compatibility"
* Kernel namelis handling
*/
#if 0
if (mflag) {
/*
mbpr(nl[N_MBSTAT].n_value, nl[N_MBPOOL].n_value,
nl[N_MCLPOOL].n_value);
*/
exit_code = 0;
goto out;
}
if (pflag) {
printproto(tp, tp->pr_name);
exit_code = 0;
goto out;
}
#endif
/*
* Keep file descriptors open to avoid overhead
* of open/close on each call to get* routines.
*/
sethostent(1);
setnetent(1);
if (iflag) {
intpr(interval);
exit_code = 0;
goto out;
}
if (rflag) {
/*
if (sflag)
rt_stats();
else
*/
if (Lflag || routexpr(af) == 0) {
if (route4pr(af) == 0 && af == AF_INET) routepr();
route6pr(af);
}
exit_code = 0;
goto out;
}
/*
if (gflag) {
if (sflag) {
if (af == AF_INET || af == AF_UNSPEC)
mrt_stats(nl[N_MRTPROTO].n_value,
nl[N_MRTSTAT].n_value);
#ifdef NETSNMP_ENABLE_IPV6
if (af == AF_INET6 || af == AF_UNSPEC)
mrt6_stats(nl[N_MRT6PROTO].n_value,
nl[N_MRT6STAT].n_value);
#endif
}
else {
if (af == AF_INET || af == AF_UNSPEC)
mroutepr(nl[N_MRTPROTO].n_value,
nl[N_MFCHASHTBL].n_value,
nl[N_MFCHASH].n_value,
nl[N_VIFTABLE].n_value);
#ifdef NETSNMP_ENABLE_IPV6
if (af == AF_INET6 || af == AF_UNSPEC)
mroute6pr(nl[N_MRT6PROTO].n_value,
nl[N_MF6CTABLE].n_value,
nl[N_MIF6TABLE].n_value);
#endif
}
exit_code = 0;
goto out;
}
*/
setservent(1);
if (Lflag) {
switch (af) {
case AF_UNSPEC:
setprotoent(1);
/* ugh, this is O(MN) ... why do we do this? */
while ((p = getprotoent())) {
for (tp = protox; tp->pr_name; tp++)
if (strcmp(tp->pr_name, p->p_name) == 0)
if (tp->pr_name && tp->pr_wanted)
printproto(tp, p->p_name);
}
endprotoent();
break;
case AF_INET:
for (tp = protox; tp->pr_name; tp++)
printproto(tp, tp->pr_name);
/* FALL THROUGH */
case AF_INET6:
for (tp = ip6protox; tp->pr_name; tp++)
printproto(tp, tp->pr_name);
}
}
else {
for (tp = ipxprotox; tp->pr_name; tp++)
if (!pname || strcmp(pname,tp->pr_name) == 0)
printproto(tp, pname);
}
/*
if (af == AF_IPX || af == AF_UNSPEC)
for (tp = ipxprotox; tp->pr_name; tp++)
printproto(tp, tp->pr_name);
if (af == AF_NS || af == AF_UNSPEC)
for (tp = nsprotox; tp->pr_name; tp++)
printproto(tp, tp->pr_name);
if ((af == AF_UNIX || af == AF_UNSPEC) && !sflag)
unixpr(nl[N_UNIXSW].n_value);
if (af == AF_APPLETALK || af == AF_UNSPEC)
for (tp = atalkprotox; tp->pr_name; tp++)
printproto(tp, tp->pr_name);
*/
exit_code = 0;
out:
SOCK_CLEANUP;
return exit_code;
}
/*
* Print out protocol statistics or control blocks (per sflag).
* Namelist checks - Omitted
*/
static void
printproto(struct protox *tp, const char *name)
{
void (*pr)(const char *);
if (sflag) {
pr = tp->pr_stats;
} else {
pr = tp->pr_cblocks;
}
if (pr != NULL)
(*pr)(name);
}
/*
* Read kernel memory - Omitted
*/
const char *
plural(int n)
{
return (n != 1 ? "s" : "");
}
/*
* Find the protox for the given "well-known" name.
*/
static struct protox *
knownname(const 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.
*/
static struct protox *
name2protox(const 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())) {
/* netsnmp_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);
}
static void
usage(void)
{
(void)fprintf(stderr,
"usage: %s [snmp_opts] [-Canv] [-Cf address_family]\n", progname);
(void)fprintf(stderr,
" %s [snmp_opts] [-Cibodnv] [-CI interface] [-Cw wait]\n", progname);
(void)fprintf(stderr,
" %s [snmp_opts] [-Cs[s]] [-Cp protocol]\n", progname);
(void)fprintf(stderr,
" %s [snmp_opts] [-Crnv] [-Cf address_family]\n", progname);
}