开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > SNMP编程 > 查看源码
main.c
资源名称：net-snmp-5.2.2.rar　[点击查看]
上传用户：wxp200602
上传日期：2007-10-30
资源大小：4028k
文件大小：12k
源码类别：

SNMP编程

开发平台：

Unix_Linux

main.c：源码内容
  1. /* $OpenBSD: main.c,v 1.52 2005/02/10 14:25:08 itojun Exp $ */
  2. /* $NetBSD: main.c,v 1.9 1996/05/07 02:55:02 thorpej Exp $ */
  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. Neither the name of the University nor the names of its contributors
  17.  *    may be used to endorse or promote products derived from this software
  18.  *    without specific prior written permission.
  19.  *
  20.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  21.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  22.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  23.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  24.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  25.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  26.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  28.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  29.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  30.  * SUCH DAMAGE.
  31.  */
  33. #ifndef lint
  34. char copyright[] =
  35. "@(#) Copyright (c) 1983, 1988, 1993n
  36. Regents of the University of California.  All rights reserved.n";
  37. #endif /* not lint */
  39. #ifdef  INHERITED_CODE
  40. #ifndef lint
  41. #if 0
  42. static char sccsid[] = "from: @(#)main.c 8.4 (Berkeley) 3/1/94";
  43. #else
  44. static char *rcsid = "$OpenBSD: main.c,v 1.52 2005/02/10 14:25:08 itojun Exp $";
  45. #endif
  46. #endif /* not lint */
  47. #endif
  49. #include <net-snmp/net-snmp-config.h>
  50. #include <net-snmp/net-snmp-includes.h>
  51. #include <net-snmp/utilities.h>
  53. #if HAVE_NETDB_H
  54. #include <netdb.h>
  55. #endif
  57. #include "main.h"
  58. #include "netstat.h"
  60. int Aflag; /* show addresses of protocol control block */
  61. int aflag; /* show all sockets (including servers) */
  62. int bflag; /* show bytes instead of packets */
  63. int dflag; /* show i/f dropped packets */
  64. int gflag; /* show group (multicast) routing or stats */
  65. int iflag; /* show interfaces */
  66. int lflag; /* show routing table with use and ref */
  67. int mflag; /* show memory stats */
  68. int nflag; /* show addresses numerically */
  69. int oflag; /* Open/Net-BSD style octet output */
  70. int pflag; /* show given protocol */
  71. int qflag; /* only display non-zero values for output */
  72. int rflag; /* show routing tables (or routing stats) */
  73. int Sflag; /* show source address in routing table */
  74. int sflag; /* show protocol statistics */
  75. int tflag; /* show i/f watchdog timers */
  76. int vflag; /* be verbose */
  78. int interval; /* repeat interval for i/f stats */
  79. char *intrface; /* desired i/f for stats, or NULL for all i/fs */
  80. int af; /* address family */
  82. char    *progname = NULL;
  84.     /*
  85.      * struct nlist nl[] - Omitted
  86.      */
  88. struct protox {
  89.         /* pr_index/pr_sindex - Omitted */ 
  90. int pr_wanted; /* 1 if wanted, 0 otherwise */
  91. void (*pr_cblocks)(const char *); /* control blocks printing routine */
  92. void (*pr_stats)(const char *); /* statistics printing routine */
  93.   const char *pr_name; /* well-known name */
  94. } protox[] = {
  95. { 1, tcpprotopr, tcp_stats, "tcp" },
  96. { 1, udpprotopr, udp_stats, "udp" },
  98. { 1, 0, ip_stats, "ip" }, /* protopr Omitted */
  99. { 1, 0, icmp_stats, "icmp" },
  100. /* igmp/ah/esp/ipencap/etherip/ipcomp/carp/pfsync/pim - Omitted */
  101. { 0, 0, 0, 0 }
  102. };
  104. struct protox ip6protox[] = {
  105. { 1, tcp6protopr, 0, "tcp6" },
  106. { 1, udp6protopr, 0, "udp6" },
  108. { 1, 0, ip6_stats, "ip6" }, /* ip6protopr Omitted */
  109. { 1, 0, icmp6_stats, "icmp6" },
  110. /* pim6/rip6 - Omitted */
  111. { 0, 0, 0, 0 }
  112. };
  114. /* {ipx,ns,atalk}protox Omitted */
  116. struct protox *protoprotox[] = {
  117. protox, ip6protox, NULL
  118. };
  120. static void printproto(struct protox *, const char *);
  121. static void usage(void);
  122. static struct protox *name2protox(const char *);
  123. static struct protox *knownname(const char *);
  125. netsnmp_session *ss;
  126. struct protox *tp = NULL; /* for printing cblocks & stats */
  128. static void
  129. optProc( int argc, char *const *argv, int opt )
  130. {
  131.     switch (opt) {
  132.     case 'C':
  133.         while (*optarg) {
  134.             switch (*optarg++) {
  135.     /* case 'A': *BSD:  display PCB addresses
  136. Linux: protocol family
  137. Aflag = 1;
  138. break;
  139.      */
  140. case 'a':
  141. aflag = 1;
  142. break;
  143. case 'b':
  144. bflag = 1;
  145. break;
  146. case 'd':
  147. dflag = 1;
  148. break;
  149. case 'f':
  150.                         if (!*optarg)
  151.                             optarg = argv[optind++];
  152. if (strcmp(optarg, "inet") == 0)
  153. af = AF_INET;
  154. else if (strcmp(optarg, "inet6") == 0)
  155. af = AF_INET6;
  156. /*
  157. else if (strcmp(optarg, "local") == 0)
  158. af = AF_LOCAL;
  159. else if (strcmp(optarg, "unix") == 0)
  160. af = AF_UNIX;
  161. else if (strcmp(optarg, "ipx") == 0)
  162. af = AF_IPX;
  163. else if (strcmp(optarg, "ns") == 0)
  164. af = AF_NS;
  165. else if (strcmp(optarg, "encap") == 0)
  166. af = PF_KEY;
  167. else if (strcmp(optarg, "atalk") == 0)
  168. af = AF_APPLETALK;
  169. */
  170. else {
  171. (void)fprintf(stderr,
  172.     "%s: %s: unknown address familyn",
  173.     progname, optarg);
  174. exit(1);
  175. }
  176. return;
  177. case 'g':
  178. gflag = 1;
  179. break;
  180. case 'I':
  181. iflag = 1;
  182.                         if (!*optarg)
  183.                             optarg = argv[optind++];
  184. intrface = optarg;
  185. return;
  186. case 'i':
  187. iflag = 1;
  188. break;
  189.     /*  case 'L': FreeBSD: Display listen queue lengths
  190. NetBSD:  Suppress link-level routes */
  191.     /* case 'l': OpenBSD: Wider IPv6 display
  192. Linux:   Listening sockets only
  193. lflag = 1;
  194. break;
  195. case 'M': *BSD:    Memory image
  196. Linux:   Masqueraded connections
  197. memf = optarg;
  198. break;
  199.      */
  200. case 'm':
  201. mflag = 1;
  202. break;
  203.     /* case 'N': *BSD:    Kernel image
  204. nlistf = optarg;
  205. break;
  206.      */
  207. case 'n':
  208. nflag = 1;
  209. break;
  210. case 'o':
  211. oflag = 1;
  212. break;
  213.     /*  case 'P': NetBSD:
  214. OpenBSD: dump PCB block */
  215. case 'p':
  216.                         if (!*optarg)
  217.                             optarg = argv[optind++];
  218. if ((tp = name2protox(optarg)) == NULL) {
  219. (void)fprintf(stderr,
  220.     "%s: %s: unknown protocoln",
  221.     progname, optarg);
  222. exit(1);
  223. }
  224. pflag = 1;
  225. return;
  226.     /* case 'q': NetBSD:  IRQ information
  227. OpenBSD: Suppress inactive I/Fs
  228. qflag = 1;
  229. break;
  230.      */
  231. case 'r':
  232. rflag = 1;
  233. break;
  234. case 'S':      /* FreeBSD:
  235. NetBSD:  Semi-numeric display
  236. OpenBSD: Show route source selector */
  237. Sflag = 1;
  238. break;
  239. case 's':
  240. ++sflag;
  241. break;
  242.      /* case 't': FreeBSD:
  243. OpenBSD: Display watchdog timers
  244. tflag = 1;
  245. break;
  246. case 'u': OpenBSD: unix sockets only
  247. af = AF_UNIX;
  248. break;
  249.       */
  250. case 'v':
  251. vflag = 1;
  252. break;
  253. case 'w':
  254.                         if (!*optarg)
  255.                             optarg = argv[optind++];
  256. interval = atoi(optarg);
  257. iflag = 1;
  258. return;
  259. case '?':
  260. default:
  261. usage();
  262.             }
  263.         }
  264.         break;   /* End of '-Cx' switch */
  266.     /*
  267.      *  Backward compatability for the main display modes
  268.      *    (where this doesn't clash with standard SNMP flags)
  269.      */
  270.     case 'i':
  271. iflag = 1;
  272. break;
  273.     case 'R':
  274. rflag = 1;    /* -r sets the retry count */
  275. break;
  276.     case 's':
  277. ++sflag;
  278. break;
  279.     }
  280. }
  282. int
  283. main(int argc, char *argv[])
  284. {
  285. netsnmp_session session;
  286. struct protoent *p;
  287.         char *cp;
  289. af = AF_UNSPEC;
  290.         cp = strrchr( argv[0], '/' );
  291.         if (cp)
  292.             progname = cp+1;
  293.         else
  294.             progname = argv[0];
  296. switch (snmp_parse_args( argc, argv, &session, "C:iRs", optProc)) {
  297. case -2:
  298.     exit(0);
  299. case -1:
  300.     usage();
  301.     exit(0);
  302. default:
  303.     break;
  304. }
  306.     /*
  307.      * Check argc vs optind ??
  308.      */
  309. argv += optind;
  310. argc -= optind;
  312.     /*
  313.      * Open an SNMP session.
  314.      */
  315.     SOCK_STARTUP;
  316.     ss = snmp_open(&session);
  317.     if (ss == NULL) {
  318.         /*
  319.          * diagnose snmp_open errors with the input netsnmp_session pointer 
  320.          */
  321.         snmp_sess_perror("snmpnetstat", &session);
  322.         SOCK_CLEANUP;
  323.         exit(1);
  324.     }
  326. /*
  327.  * Omitted:
  328.  *     Privilege handling
  329.  *    "Backward Compatibility"
  330.  *     Kernel namelis handling
  331.  */
  333. if (mflag) {
  334.             /*
  335. mbpr(nl[N_MBSTAT].n_value, nl[N_MBPOOL].n_value,
  336.     nl[N_MCLPOOL].n_value);
  337.              */
  338. exit(0);
  339. }
  340. if (pflag) {
  341. printproto(tp, tp->pr_name);
  342. exit(0);
  343. }
  344. /*
  345.  * Keep file descriptors open to avoid overhead
  346.  * of open/close on each call to get* routines.
  347.  */
  348. sethostent(1);
  349. setnetent(1);
  350. if (iflag) {
  351. intpr(interval);
  352. exit(0);
  353. }
  354. if (rflag) {
  355.              /*
  356. if (sflag)
  357. rt_stats();
  358. else
  359.               */
  360. routepr();
  361. exit(0);
  362. }
  363.      /*
  364. if (gflag) {
  365. if (sflag) {
  366. if (af == AF_INET || af == AF_UNSPEC)
  367. mrt_stats(nl[N_MRTPROTO].n_value,
  368.     nl[N_MRTSTAT].n_value);
  369. #ifdef INET6
  370. if (af == AF_INET6 || af == AF_UNSPEC)
  371. mrt6_stats(nl[N_MRT6PROTO].n_value,
  372.     nl[N_MRT6STAT].n_value);
  373. #endif
  374. }
  375. else {
  376. if (af == AF_INET || af == AF_UNSPEC)
  377. mroutepr(nl[N_MRTPROTO].n_value,
  378.     nl[N_MFCHASHTBL].n_value,
  379.     nl[N_MFCHASH].n_value,
  380.     nl[N_VIFTABLE].n_value);
  381. #ifdef INET6
  382. if (af == AF_INET6 || af == AF_UNSPEC)
  383. mroute6pr(nl[N_MRT6PROTO].n_value,
  384.     nl[N_MF6CTABLE].n_value,
  385.     nl[N_MIF6TABLE].n_value);
  386. #endif
  387. }
  388. exit(0);
  389. }
  390.      */
  391. if (af == AF_INET || af == AF_UNSPEC) {
  392. setprotoent(1);
  393. setservent(1);
  394. /* ugh, this is O(MN) ... why do we do this? */
  395. while ((p = getprotoent())) {
  396. for (tp = protox; tp->pr_name; tp++)
  397. if (strcmp(tp->pr_name, p->p_name) == 0)
  398. break;
  399. if (tp->pr_name == 0 || tp->pr_wanted == 0)
  400. continue;
  401. printproto(tp, p->p_name);
  402. }
  403. endprotoent();
  404. }
  405. if (af == AF_INET6 || af == AF_UNSPEC)
  406. for (tp = ip6protox; tp->pr_name; tp++)
  407. printproto(tp, tp->pr_name);
  408.     /*
  409. if (af == AF_IPX || af == AF_UNSPEC)
  410. for (tp = ipxprotox; tp->pr_name; tp++)
  411. printproto(tp, tp->pr_name);
  412. if (af == AF_NS || af == AF_UNSPEC)
  413. for (tp = nsprotox; tp->pr_name; tp++)
  414. printproto(tp, tp->pr_name);
  415. if ((af == AF_UNIX || af == AF_UNSPEC) && !sflag)
  416. unixpr(nl[N_UNIXSW].n_value);
  417. if (af == AF_APPLETALK || af == AF_UNSPEC)
  418. for (tp = atalkprotox; tp->pr_name; tp++)
  419. printproto(tp, tp->pr_name);
  420.      */
  421. exit(0);
  422. }
  424. /*
  425.  * Print out protocol statistics or control blocks (per sflag).
  426.  * Namelist checks - Omitted
  427.  */
  428. static void
  429. printproto(struct protox *tp, const char *name)
  430. {
  431. void (*pr)(const char *);
  433. if (sflag) {
  434. pr = tp->pr_stats;
  435. } else {
  436. pr = tp->pr_cblocks;
  437. }
  438. if (pr != NULL)
  439. (*pr)(name);
  440. }
  442. /*
  443.  * Read kernel memory - Omitted
  444.  */
  446. const char *
  447. plural(int n)
  448. {
  449. return (n != 1 ? "s" : "");
  450. }
  452. const char *
  453. plurales(int n)
  454. {
  455. return (n != 1 ? "es" : "");
  456. }
  458. /*
  459.  * Find the protox for the given "well-known" name.
  460.  */
  461. static struct protox *
  462. knownname(const char *name)
  463. {
  464. struct protox **tpp, *tp;
  466. for (tpp = protoprotox; *tpp; tpp++)
  467. for (tp = *tpp; tp->pr_name; tp++)
  468. if (strcmp(tp->pr_name, name) == 0)
  469. return (tp);
  470. return (NULL);
  471. }
  473. /*
  474.  * Find the protox corresponding to name.
  475.  */
  476. static struct protox *
  477. name2protox(const char *name)
  478. {
  479. struct protox *tp;
  480. char **alias; /* alias from p->aliases */
  481. struct protoent *p;
  483. /*
  484.  * Try to find the name in the list of "well-known" names. If that
  485.  * fails, check if name is an alias for an Internet protocol.
  486.  */
  487. if ((tp = knownname(name)))
  488. return (tp);
  490. setprotoent(1); /* make protocol lookup cheaper */
  491. while ((p = getprotoent())) {
  492. /* assert: name not same as p->name */
  493. for (alias = p->p_aliases; *alias; alias++)
  494. if (strcmp(name, *alias) == 0) {
  495. endprotoent();
  496. return (knownname(p->p_name));
  497. }
  498. }
  499. endprotoent();
  500. return (NULL);
  501. }
  503. static void
  504. usage(void)
  505. {
  506. (void)fprintf(stderr,
  507. "usage: %s [snmp_opts] [-Can] [-Cf address_family]n", progname);
  508. (void)fprintf(stderr,
  509. "       %s [snmp_opts] [-CbdgimnrSs] [-Cf address_family]n", progname);
  510. (void)fprintf(stderr,
  511. "       %s [snmp_opts] [-Cbdn] [-CI interface] [-Cw wait]n", progname);
  512. (void)fprintf(stderr,
  513. "       %s [snmp_opts] [-Cs] [-Cp protocol]n", progname);
  514. (void)fprintf(stderr,
  515. "       %s [snmp_opts] [-Ca] [-Cf address_family] [-Ci | -CI interface]n", progname);
  516. exit(1);
  517. }