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

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

SNMP编程

开发平台：

Unix_Linux

udp.c：源码内容
  1. /*
  2.  *  UDP MIB group implementation - udp.c
  3.  *
  4.  */
  6. #include <net-snmp/net-snmp-config.h>
  7. #include "mibII_common.h"
  9. #ifdef HAVE_NETINET_UDP_H
  10. #include <netinet/udp.h>
  11. #endif
  12. #if HAVE_NETINET_UDP_VAR_H
  13. #include <netinet/udp_var.h>
  14. #endif
  16. #include <net-snmp/net-snmp-includes.h>
  17. #include <net-snmp/agent/net-snmp-agent-includes.h>
  18. #include <net-snmp/agent/auto_nlist.h>
  20. #include "util_funcs.h"
  22. #ifdef solaris2
  23. #include "kernel_sunos5.h"
  24. #else
  25. #include "kernel.h"
  26. #endif
  28. #ifdef linux
  29. #include "kernel_linux.h"
  30. #endif
  32. #ifdef cygwin
  33. #define WIN32
  34. #include <windows.h>
  35. #endif
  37. #ifdef hpux
  38. #include <sys/mib.h>
  39. #include <netinet/mib_kern.h>
  40. #endif                          /* hpux */
  42. #ifdef linux
  43. #include "tcp.h"
  44. #endif
  45. #include "udp.h"
  46. #include "udpTable.h"
  47. #include "sysORTable.h"
  49. #ifdef CAN_USE_SYSCTL
  50. #include <sys/sysctl.h>
  51. #endif
  53. #if HAVE_DMALLOC_H
  54. #include <dmalloc.h>
  55. #endif
  57. #ifndef MIB_STATS_CACHE_TIMEOUT
  58. #define MIB_STATS_CACHE_TIMEOUT 5
  59. #endif
  60. #ifndef UDP_STATS_CACHE_TIMEOUT
  61. #define UDP_STATS_CACHE_TIMEOUT MIB_STATS_CACHE_TIMEOUT
  62. #endif
  64.         /*********************
  65.  *
  66.  *  Kernel & interface information,
  67.  *   and internal forward declarations
  68.  *
  69.  *********************/
  72.         /*********************
  73.  *
  74.  *  Initialisation & common implementation functions
  75.  *
  76.  *********************/
  78. /*
  79.  * Define the OID pointer to the top of the mib tree that we're
  80.  * registering underneath, and the OID for the MIB module 
  81.  */
  82. oid             udp_oid[]               = { SNMP_OID_MIB2, 7 };
  83. oid             udp_module_oid[]        = { SNMP_OID_MIB2, 50 };
  85. void
  86. init_udp(void)
  87. {
  88.     netsnmp_handler_registration *reginfo;
  90.     /*
  91.      * register ourselves with the agent as a group of scalars...
  92.      */
  93.     DEBUGMSGTL(("mibII/udpScalar", "Initialising UDP scalar groupn"));
  94.     reginfo = netsnmp_create_handler_registration("udp", udp_handler,
  95.     udp_oid, OID_LENGTH(udp_oid), HANDLER_CAN_RONLY);
  96.     netsnmp_register_scalar_group(reginfo, UDPINDATAGRAMS, UDPOUTDATAGRAMS);
  98.     /*
  99.      * .... with a local cache
  100.      *    (except for HP-UX 11, which extracts objects individually)
  101.      */
  102. #ifndef hpux11
  103.     netsnmp_inject_handler( reginfo,
  104.     netsnmp_get_cache_handler(UDP_STATS_CACHE_TIMEOUT,
  105.     udp_load, udp_free,
  106. udp_oid, OID_LENGTH(udp_oid)));
  107. #endif
  109.     REGISTER_SYSOR_ENTRY(udp_module_oid,
  110.                          "The MIB module for managing UDP implementations");
  112. #ifdef UDPSTAT_SYMBOL
  113.     auto_nlist(UDPSTAT_SYMBOL, 0, 0);
  114. #endif
  115. #ifdef UDB_SYMBOL
  116.     auto_nlist(UDB_SYMBOL, 0, 0);
  117. #endif
  118. #ifdef solaris2
  119.     init_kernel_sunos5();
  120. #endif
  121. }
  124.         /*********************
  125.  *
  126.  *  System specific implementation functions
  127.  *
  128.  *********************/
  130. #ifdef hpux11
  131. #define UDP_STAT_STRUCTURE int
  132. #endif
  134. #ifdef linux
  135. #define UDP_STAT_STRUCTURE struct udp_mib
  136. #define USES_SNMP_DESIGNED_UDPSTAT
  137. #undef UDPSTAT_SYMBOL
  138. #endif
  140. #ifdef solaris2
  141. #define UDP_STAT_STRUCTURE mib2_udp_t
  142. #define USES_SNMP_DESIGNED_UDPSTAT
  143. #endif
  145. #ifdef WIN32
  146. #include <iphlpapi.h>
  147. #define UDP_STAT_STRUCTURE MIB_UDPSTATS
  148. #endif
  150. #ifdef HAVE_SYS_TCPIPSTATS_H
  151. #define UDP_STAT_STRUCTURE struct kna
  152. #define USES_TRADITIONAL_UDPSTAT
  153. #endif
  156. #if !defined(UDP_STAT_STRUCTURE)
  157. #define UDP_STAT_STRUCTURE struct udpstat
  158. #define USES_TRADITIONAL_UDPSTAT
  159. #endif
  161. UDP_STAT_STRUCTURE udpstat;
  165.         /*********************
  166.  *
  167.  *  System independent handler (mostly)
  168.  *
  169.  *********************/
  173. int
  174. udp_handler(netsnmp_mib_handler          *handler,
  175.             netsnmp_handler_registration *reginfo,
  176.             netsnmp_agent_request_info   *reqinfo,
  177.             netsnmp_request_info         *requests)
  178. {
  179.     netsnmp_request_info  *request;
  180.     netsnmp_variable_list *requestvb;
  181.     long     ret_value = -1;
  182.     oid      subid;
  183.     int      type = ASN_COUNTER;
  185.     /*
  186.      * The cached data should already have been loaded by the
  187.      *    cache handler, higher up the handler chain.
  188.      * But just to be safe, check this and load it manually if necessary
  189.      */
  190. #ifndef hpux11
  191.     if (!netsnmp_cache_is_valid(reqinfo, reginfo->handlerName)) {
  192.         netsnmp_assert("cache" == "valid"); /* always false */
  193.         udp_load( NULL, NULL ); /* XXX - check for failure */
  194.     }
  195. #endif
  198.     /*
  199.      * 
  200.      *
  201.      */
  202.     DEBUGMSGTL(("mibII/udpScalar", "Handler - mode %sn",
  203.                     se_find_label_in_slist("agent_mode", reqinfo->mode)));
  204.     switch (reqinfo->mode) {
  205.     case MODE_GET:
  206.         for (request=requests; request; request=request->next) {
  207.             requestvb = request->requestvb;
  208.             subid = requestvb->name[OID_LENGTH(udp_oid)];  /* XXX */
  209.             DEBUGMSGTL(( "mibII/udpScalar", "oid: "));
  210.             DEBUGMSGOID(("mibII/udpScalar", requestvb->name,
  211.                                             requestvb->name_length));
  212.             DEBUGMSG((   "mibII/udpScalar", "n"));
  214.             switch (subid) {
  215. #ifdef USES_SNMP_DESIGNED_UDPSTAT
  216.     case UDPINDATAGRAMS:
  217.         ret_value = udpstat.udpInDatagrams;
  218.         break;
  219.     case UDPNOPORTS:
  220. #ifdef solaris2
  221.         ret_value = udp_load(NULL, (void *)UDPNOPORTS);
  222. if (ret_value == -1) {
  223.             netsnmp_set_request_error(reqinfo, request, SNMP_NOSUCHOBJECT);
  224.             continue;
  225. }
  226.         break;
  227. #else
  228.         ret_value = udpstat.udpNoPorts;
  229.         break;
  230. #endif
  231.     case UDPOUTDATAGRAMS:
  232.         ret_value = udpstat.udpOutDatagrams;
  233.         break;
  234.     case UDPINERRORS:
  235.         ret_value = udpstat.udpInErrors;
  236.         break;
  237. #else /* USES_SNMP_DESIGNED_UDPSTAT */
  239. #ifdef USES_TRADITIONAL_UDPSTAT
  240. #ifdef HAVE_SYS_TCPIPSTATS_H
  241.     /*
  242.      * This actually reads statistics for *all* the groups together,
  243.      * so we need to isolate the UDP-specific bits.  
  244.      */
  245. #define udpstat          udpstat.udpstat
  246. #endif
  247.     case UDPINDATAGRAMS:
  248. #if STRUCT_UDPSTAT_HAS_UDPS_IPACKETS
  249.         ret_value = udpstat.udps_ipackets;
  250.         break;
  251. #else
  252.         netsnmp_set_request_error(reqinfo, request, SNMP_NOSUCHOBJECT);
  253.         continue;
  254. #endif
  256.     case UDPNOPORTS:
  257. #if STRUCT_UDPSTAT_HAS_UDPS_NOPORT
  258.         ret_value = udpstat.udps_noport;
  259.         break;
  260. #else
  261.         netsnmp_set_request_error(reqinfo, request, SNMP_NOSUCHOBJECT);
  262.         continue;
  263. #endif
  265.     case UDPOUTDATAGRAMS:
  266. #if STRUCT_UDPSTAT_HAS_UDPS_OPACKETS
  267.         ret_value = udpstat.udps_opackets;
  268.         break;
  269. #else
  270.         netsnmp_set_request_error(reqinfo, request, SNMP_NOSUCHOBJECT);
  271.         continue;
  272. #endif
  274.     case UDPINERRORS:
  275.         ret_value = udpstat.udps_hdrops + udpstat.udps_badsum +
  276. #ifdef STRUCT_UDPSTAT_HAS_UDPS_DISCARD
  277.             udpstat.udps_discard +
  278. #endif
  279. #ifdef STRUCT_UDPSTAT_HAS_UDPS_FULLSOCK
  280.             udpstat.udps_fullsock +
  281. #endif
  282.             udpstat.udps_badlen;
  283.         break;
  284. #ifdef HAVE_SYS_TCPIPSTATS_H
  285. #undef udpstat
  286. #endif
  287. #else /* USES_TRADITIONAL_UDPSTAT */
  289. #ifdef hpux11
  290.     case UDPINDATAGRAMS:
  291.     case UDPNOPORTS:
  292.     case UDPOUTDATAGRAMS:
  293.     case UDPINERRORS:
  294. /*
  295.  * This is a bit of a hack, to shoehorn the HP-UX 11
  296.  * single-object retrieval approach into the caching
  297.  * architecture.
  298.  */
  299. if (udp_load(NULL, (void*)subid) == -1 ) {
  300.             netsnmp_set_request_error(reqinfo, request, SNMP_NOSUCHOBJECT);
  301.             continue;
  302. }
  303.         ret_value = udpstat;
  304.         break;
  305. #else /* hpux11 */
  306. #ifdef WIN32
  307.     case UDPINDATAGRAMS:
  308.         ret_value = udpstat.dwInDatagrams;
  309.         break;
  310.     case UDPNOPORTS:
  311.         ret_value = udpstat.dwNoPorts;
  312.         break;
  313.     case UDPOUTDATAGRAMS:
  314.         ret_value = udpstat.dwOutDatagrams;
  315.         break;
  316.     case UDPINERRORS:
  317.         ret_value = udpstat.dwInErrors;
  318.         break;
  319. #endif /* WIN32 */
  320. #endif /* hpux11 */
  321. #endif /* USES_TRADITIONAL_UDPSTAT */
  322. #endif /* USES_SNMP_DESIGNED_UDPSTAT */
  324.     }
  325.     snmp_set_var_typed_value(request->requestvb, (u_char)type,
  326.              (u_char *)&ret_value, sizeof(ret_value));
  327. }
  328.         break;
  330.     case MODE_GETNEXT:
  331.     case MODE_GETBULK:
  332.     case MODE_SET_RESERVE1:
  333.     case MODE_SET_RESERVE2:
  334.     case MODE_SET_ACTION:
  335.     case MODE_SET_COMMIT:
  336.     case MODE_SET_FREE:
  337.     case MODE_SET_UNDO:
  338.         snmp_log(LOG_WARNING, "mibII/udp: Unsupported mode (%d)n",
  339.                                reqinfo->mode);
  340.         break;
  341.     default:
  342.         snmp_log(LOG_WARNING, "mibII/udp: Unrecognised mode (%d)n",
  343.                                reqinfo->mode);
  344.         break;
  345.     }
  347.     return SNMP_ERR_NOERROR;
  348. }
  351.         /*********************
  352.  *
  353.  *  Internal implementation functions
  354.  *
  355.  *********************/
  357. #ifdef hpux11
  358. int
  359. udp_load(netsnmp_cache *cache, void *vmagic)
  360. {
  361.     int             fd;
  362.     struct nmparms  p;
  363.     unsigned int    ulen;
  364.     int             ret;
  365.     int             magic = (int) vmagic;
  366.     
  367.     if ((fd = open_mib("/dev/ip", O_RDONLY, 0, NM_ASYNC_OFF)) < 0) {
  368.         DEBUGMSGTL(("mibII/udpScalar", "Failed to load UDP object %d (hpux11)n", magic));
  369.         return (-1);            /* error */
  370.     }
  372.     switch (magic) {
  373.     case UDPINDATAGRAMS:
  374.         p.objid = ID_udpInDatagrams;
  375.         break;
  376.     case UDPNOPORTS:
  377.         p.objid = ID_udpNoPorts;
  378.         break;
  379.     case UDPOUTDATAGRAMS:
  380.         p.objid = ID_udpOutDatagrams;
  381.         break;
  382.     case UDPINERRORS:
  383.         p.objid = ID_udpInErrors;
  384.         break;
  385.     default:
  386.         udpstat = 0;
  387.         close_mib(fd);
  388.         return -1;
  389.     }
  391.     p.buffer = (void *)&udpstat;
  392.     ulen = sizeof(UDP_STAT_STRUCTURE);
  393.     p.len = &ulen;
  394.     ret = get_mib_info(fd, &p);
  395.     close_mib(fd);
  397.     DEBUGMSGTL(("mibII/udpScalar", "%s UDP object %d (hpux11)n",
  398.                (ret < 0 ? "Failed to load" : "Loaded"),  magic));
  399.     return (ret);         /* 0: ok, < 0: error */
  400. }
  401. #else                           /* hpux11 */
  402. #ifdef linux
  403. int
  404. udp_load(netsnmp_cache *cache, void *vmagic)
  405. {
  406.     long ret_value = -1;
  408.     ret_value = linux_read_udp_stat(&udpstat);
  410.     if ( ret_value < 0 ) {
  411.         DEBUGMSGTL(("mibII/udpScalar", "Failed to load UDP scalar Group (linux)n"));
  412.     } else {
  413.         DEBUGMSGTL(("mibII/udpScalar", "Loaded UDP scalar Group (linux)n"));
  414.     }
  415.     return ret_value;
  416. }
  417. #else                           /* linux */
  418. #ifdef solaris2
  419. int
  420. udp_load(netsnmp_cache *cache, void *vmagic)
  421. {
  422.     long ret_value = -1;
  423.     int  magic = (int)vmagic;
  424.     mib2_ip_t ipstat;
  426.     /*
  427.      * udpNoPorts is actually implemented as part of the MIB_IP group
  428.      * so we need to retrieve this independently
  429.      */
  430.     if (magic == UDPNOPORTS) {
  431.         if (getMibstat
  432.             (MIB_IP, &ipstat, sizeof(mib2_ip_t), GET_FIRST,
  433.              &Get_everything, NULL) < 0) {
  434.             DEBUGMSGTL(("mibII/udpScalar", "Failed to load UDP object %d (solaris)n", magic));
  435.             return -1;
  436.         } else {
  437.             DEBUGMSGTL(("mibII/udpScalar", "Loaded UDP object %d (solaris)n", magic));
  438.             return ipstat.udpNoPorts;
  439.         }
  440.     }
  442.     /*
  443.      * Otherwise, retrieve the whole of the MIB_UDP group (and cache it)
  444.      */
  445.     ret_value = getMibstat(MIB_UDP, &udpstat, sizeof(mib2_udp_t),
  446.                            GET_FIRST, &Get_everything, NULL);
  448.     if ( ret_value < 0 ) {
  449.         DEBUGMSGTL(("mibII/udpScalar", "Failed to load UDP scalar Group (solaris)n"));
  450.     } else {
  451.         DEBUGMSGTL(("mibII/udpScalar", "Loaded UDP scalar Group (solaris)n"));
  452.     }
  453.     return ret_value;
  454. }
  455. #else                           /* solaris2 */
  456. #ifdef WIN32
  457. int
  458. udp_load(netsnmp_cache *cache, void *vmagic)
  459. {
  460.     long ret_value = -1;
  462.     ret_value = GetUdpStatistics(&udpstat);
  464.     if ( ret_value < 0 ) {
  465.         DEBUGMSGTL(("mibII/udpScalar", "Failed to load UDP scalar Group (win32)n"));
  466.     } else {
  467.         DEBUGMSGTL(("mibII/udpScalar", "Loaded UDP scalar Group (win32)n"));
  468.     }
  469.     return ret_value;
  470. }
  471. #else                           /* WIN32 */
  472. #if (defined(CAN_USE_SYSCTL) && defined(UDPCTL_STATS))
  473. int
  474. udp_load(netsnmp_cache *cache, void *vmagic)
  475. {
  476.     int     sname[4]  = { CTL_NET, PF_INET, IPPROTO_UDP, UDPCTL_STATS };
  477.     size_t  len       = sizeof(udpstat);
  478.     long    ret_value = -1;
  480.     ret_value = sysctl(sname, 4, &udpstat, &len, 0, 0);
  482.     if ( ret_value < 0 ) {
  483.         DEBUGMSGTL(("mibII/udpScalar", "Failed to load UDP scalar Group (sysctl)n"));
  484.     } else {
  485.         DEBUGMSGTL(("mibII/udpScalar", "Loaded UDP scalar Group (sysctl)n"));
  486.     }
  487.     return ret_value;
  488. }
  489. #else /* (defined(CAN_USE_SYSCTL) && defined(UDPCTL_STATS)) */
  490. #ifdef HAVE_SYS_TCPIPSTATS_H
  491. int
  492. udp_load(netsnmp_cache *cache, void *vmagic)
  493. {
  494.     long ret_value = -1;
  496.     ret_value = sysmp(MP_SAGET, MPSA_TCPIPSTATS, &udpstat, sizeof(udpstat));
  498.     if ( ret_value < 0 ) {
  499.         DEBUGMSGTL(("mibII/udpScalar", "Failed to load UDP scalar Group (tcpipstats)n"));
  500.     } else {
  501.         DEBUGMSGTL(("mibII/udpScalar", "Loaded UDP scalar Group (tcpipstats)n"));
  502.     }
  503.     return ret_value;
  504. }
  505. #else /* HAVE_SYS_TCPIPSTATS_H */
  506. #ifdef UDPSTAT_SYMBOL
  507. int
  508. udp_load(netsnmp_cache *cache, void *vmagic)
  509. {
  510.     long ret_value = -1;
  512.     if (auto_nlist(UDPSTAT_SYMBOL, (char *)&udpstat, sizeof(udpstat)))
  513.         ret_value = 0;
  515.     if ( ret_value < 0 ) {
  516.         DEBUGMSGTL(("mibII/udpScalar", "Failed to load UDP scalar Group (udpstat)n"));
  517.     } else {
  518.         DEBUGMSGTL(("mibII/udpScalar", "Loaded UDP scalar Group (udpstat)n"));
  519.     }
  520.     return ret_value;
  521. }
  522. #else /* UDPSTAT_SYMBOL */
  523. int
  524. udp_load(netsnmp_cache *cache, void *vmagic)
  525. {
  526.     long ret_value = -1;
  528.     DEBUGMSGTL(("mibII/udpScalar", "Failed to load UDP scalar Group (null)n"));
  529.     return ret_value;
  530. }
  531. #endif /* UDPSTAT_SYMBOL */
  532. #endif /* HAVE_SYS_TCPIPSTATS_H */
  533. #endif /* (defined(CAN_USE_SYSCTL) && defined(UDPCTL_STATS)) */
  534. #endif                          /* hpux11 */
  535. #endif                          /* linux */
  536. #endif                          /* solaris2 */
  537. #endif                          /* WIN32 */
  540. void
  541. udp_free(netsnmp_cache *cache, void *magic)
  542. {
  543.     memset(&udpstat, 0, sizeof(udpstat));
  544. }