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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Linux/Unix编程 > 网络 > 查看源码
zebra_snmp.c
资源名称：zebra-0.95a.tar.gz　[点击查看]
上传用户：xiaozhuqw
上传日期：2009-11-15
资源大小：1338k
文件大小：15k
源码类别：

网络

开发平台：

Unix_Linux

zebra_snmp.c：源码内容
  1. /* FIB SNMP.
  2.  * Copyright (C) 1999 Kunihiro Ishiguro
  3.  *
  4.  * This file is part of GNU Zebra.
  5.  *
  6.  * GNU Zebra is free software; you can redistribute it and/or modify it
  7.  * under the terms of the GNU General Public License as published by the
  8.  * Free Software Foundation; either version 2, or (at your option) any
  9.  * later version.
  10.  *
  11.  * GNU Zebra is distributed in the hope that it will be useful, but
  12.  * WITHOUT ANY WARRANTY; without even the implied warranty of
  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14.  * General Public License for more details.
  15.  *
  16.  * You should have received a copy of the GNU General Public License
  17.  * along with GNU Zebra; see the file COPYING.  If not, write to the Free
  18.  * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  19.  * 02111-1307, USA.  
  20.  */
  22. #include <zebra.h>
  24. #ifdef HAVE_SNMP
  26. #ifdef HAVE_NETSNMP
  27. #include <net-snmp/net-snmp-config.h>
  28. #endif /* HAVE_NETSNMP */
  30. #include <asn1.h>
  31. #include <snmp.h>
  32. #include <snmp_impl.h>
  34. #include "if.h"
  35. #include "log.h"
  36. #include "prefix.h"
  37. #include "command.h"
  38. #include "smux.h"
  39. #include "table.h"
  41. #include "zebra/rib.h"
  43. #define IPFWMIB 1,3,6,1,2,1,4,24
  44. #define ZEBRAOID 1,3,6,1,4,1,3317,1,2,1
  46. /* ipForwardTable */
  47. #define IPFORWARDDEST                         1
  48. #define IPFORWARDMASK                         2
  49. #define IPFORWARDPOLICY                       3
  50. #define IPFORWARDNEXTHOP                      4
  51. #define IPFORWARDIFINDEX                      5
  52. #define IPFORWARDTYPE                         6
  53. #define IPFORWARDPROTO                        7
  54. #define IPFORWARDAGE                          8
  55. #define IPFORWARDINFO                         9
  56. #define IPFORWARDNEXTHOPAS                   10
  57. #define IPFORWARDMETRIC1                     11
  58. #define IPFORWARDMETRIC2                     12
  59. #define IPFORWARDMETRIC3                     13
  60. #define IPFORWARDMETRIC4                     14
  61. #define IPFORWARDMETRIC5                     15
  63. /* ipCidrRouteTable */
  64. #define IPCIDRROUTEDEST                       1
  65. #define IPCIDRROUTEMASK                       2
  66. #define IPCIDRROUTETOS                        3
  67. #define IPCIDRROUTENEXTHOP                    4
  68. #define IPCIDRROUTEIFINDEX                    5
  69. #define IPCIDRROUTETYPE                       6
  70. #define IPCIDRROUTEPROTO                      7
  71. #define IPCIDRROUTEAGE                        8
  72. #define IPCIDRROUTEINFO                       9
  73. #define IPCIDRROUTENEXTHOPAS                 10
  74. #define IPCIDRROUTEMETRIC1                   11
  75. #define IPCIDRROUTEMETRIC2                   12
  76. #define IPCIDRROUTEMETRIC3                   13
  77. #define IPCIDRROUTEMETRIC4                   14
  78. #define IPCIDRROUTEMETRIC5                   15
  79. #define IPCIDRROUTESTATUS                    16
  81. #define INTEGER32 ASN_INTEGER
  82. #define GAUGE32 ASN_GAUGE
  83. #define ENUMERATION ASN_INTEGER
  84. #define ROWSTATUS ASN_INTEGER
  85. #define IPADDRESS ASN_IPADDRESS
  86. #define OBJECTIDENTIFIER ASN_OBJECT_ID
  88. oid ipfw_oid [] = { IPFWMIB };
  89. oid zebra_oid [] = { ZEBRAOID };
  91. /* Hook functions. */
  92. u_char * ipFwNumber ();
  93. u_char * ipFwTable ();
  94. u_char * ipCidrNumber ();
  95. u_char * ipCidrTable ();
  97. struct variable zebra_variables[] = 
  98.   {
  99.     {0, GAUGE32, RONLY, ipFwNumber, 1, {1}},
  100.     {IPFORWARDDEST, IPADDRESS, RONLY, ipFwTable, 3, {2, 1, 1}},
  101.     {IPFORWARDMASK, IPADDRESS, RONLY, ipFwTable, 3, {2, 1, 2}},
  102.     {IPFORWARDPOLICY, INTEGER32, RONLY, ipFwTable, 3, {2, 1, 3}},
  103.     {IPFORWARDNEXTHOP, IPADDRESS, RONLY, ipFwTable, 3, {2, 1, 4}},
  104.     {IPFORWARDIFINDEX, INTEGER32, RONLY, ipFwTable, 3, {2, 1, 5}},
  105.     {IPFORWARDTYPE, ENUMERATION, RONLY, ipFwTable, 3, {2, 1, 6}},
  106.     {IPFORWARDPROTO, ENUMERATION, RONLY, ipFwTable, 3, {2, 1, 7}},
  107.     {IPFORWARDAGE, INTEGER32, RONLY, ipFwTable, 3, {2, 1, 8}},
  108.     {IPFORWARDINFO, OBJECTIDENTIFIER, RONLY, ipFwTable, 3, {2, 1, 9}},
  109.     {IPFORWARDNEXTHOPAS, INTEGER32, RONLY, ipFwTable, 3, {2, 1, 10}},
  110.     {IPFORWARDMETRIC1, INTEGER32, RONLY, ipFwTable, 3, {2, 1, 11}},
  111.     {IPFORWARDMETRIC2, INTEGER32, RONLY, ipFwTable, 3, {2, 1, 12}},
  112.     {IPFORWARDMETRIC3, INTEGER32, RONLY, ipFwTable, 3, {2, 1, 13}},
  113.     {IPFORWARDMETRIC4, INTEGER32, RONLY, ipFwTable, 3, {2, 1, 14}},
  114.     {IPFORWARDMETRIC5, INTEGER32, RONLY, ipFwTable, 3, {2, 1, 15}},
  115.     {0, GAUGE32, RONLY, ipCidrNumber, 1, {3}},
  116.     {IPCIDRROUTEDEST, IPADDRESS, RONLY, ipCidrTable, 3, {4, 1, 1}},
  117.     {IPCIDRROUTEMASK, IPADDRESS, RONLY, ipCidrTable, 3, {4, 1, 2}},
  118.     {IPCIDRROUTETOS, INTEGER32, RONLY, ipCidrTable, 3, {4, 1, 3}},
  119.     {IPCIDRROUTENEXTHOP, IPADDRESS, RONLY, ipCidrTable, 3, {4, 1, 4}},
  120.     {IPCIDRROUTEIFINDEX, INTEGER32, RONLY, ipCidrTable, 3, {4, 1, 5}},
  121.     {IPCIDRROUTETYPE, ENUMERATION, RONLY, ipCidrTable, 3, {4, 1, 6}},
  122.     {IPCIDRROUTEPROTO, ENUMERATION, RONLY, ipCidrTable, 3, {4, 1, 7}},
  123.     {IPCIDRROUTEAGE, INTEGER32, RONLY, ipCidrTable, 3, {4, 1, 8}},
  124.     {IPCIDRROUTEINFO, OBJECTIDENTIFIER, RONLY, ipCidrTable, 3, {4, 1, 9}},
  125.     {IPCIDRROUTENEXTHOPAS, INTEGER32, RONLY, ipCidrTable, 3, {4, 1, 10}},
  126.     {IPCIDRROUTEMETRIC1, INTEGER32, RONLY, ipCidrTable, 3, {4, 1, 11}},
  127.     {IPCIDRROUTEMETRIC2, INTEGER32, RONLY, ipCidrTable, 3, {4, 1, 12}},
  128.     {IPCIDRROUTEMETRIC3, INTEGER32, RONLY, ipCidrTable, 3, {4, 1, 13}},
  129.     {IPCIDRROUTEMETRIC4, INTEGER32, RONLY, ipCidrTable, 3, {4, 1, 14}},
  130.     {IPCIDRROUTEMETRIC5, INTEGER32, RONLY, ipCidrTable, 3, {4, 1, 15}},
  131.     {IPCIDRROUTESTATUS, ROWSTATUS, RONLY, ipCidrTable, 3, {4, 1, 16}}
  132.   };
  135. u_char *
  136. ipFwNumber (struct variable *v, oid objid[], size_t *objid_len,
  137.     int exact, size_t *val_len, WriteMethod **write_method)
  138. {
  139.   static int result;
  140.   struct route_table *table;
  141.   struct route_node *rn;
  142.   struct rib *rib;
  144.   if (smux_header_generic(v, objid, objid_len, exact, val_len, write_method) == MATCH_FAILED)
  145.     return NULL;
  147.   table = vrf_table (AFI_IP, SAFI_UNICAST, 0);
  148.   if (! table)
  149.     return NULL;
  151.   /* Return number of routing entries. */
  152.   result = 0;
  153.   for (rn = route_top (table); rn; rn = route_next (rn))
  154.     for (rib = rn->info; rib; rib = rib->next)
  155.       result++;
  157.   return (u_char *)&result;
  158. }
  160. u_char *
  161. ipCidrNumber (struct variable *v, oid objid[], size_t *objid_len,
  162.       int exact, size_t *val_len, WriteMethod **write_method)
  163. {
  164.   static int result;
  165.   struct route_table *table;
  166.   struct route_node *rn;
  167.   struct rib *rib;
  169.   if (smux_header_generic(v, objid, objid_len, exact, val_len, write_method) == MATCH_FAILED)
  170.     return NULL;
  172.   table = vrf_table (AFI_IP, SAFI_UNICAST, 0);
  173.   if (! table)
  174.     return 0;
  176.   /* Return number of routing entries. */
  177.   result = 0;
  178.   for (rn = route_top (table); rn; rn = route_next (rn))
  179.     for (rib = rn->info; rib; rib = rib->next)
  180.       result++;
  182.   return (u_char *)&result;
  183. }
  185. int
  186. in_addr_cmp(u_char *p1, u_char *p2)
  187. {
  188.   int i;
  190.   for (i=0; i<4; i++)
  191.     {
  192.       if (*p1 < *p2)
  193.         return -1;
  194.       if (*p1 > *p2)
  195.         return 1;
  196.       p1++; p2++;
  197.     }
  198.   return 0;
  199. }
  201. int 
  202. in_addr_add(u_char *p, int num)
  203. {
  204.   int i, ip0;
  206.   ip0 = *p;
  207.   p += 4;
  208.   for (i = 3; 0 <= i; i--) {
  209.     p--;
  210.     if (*p + num > 255) {
  211.       *p += num;
  212.       num = 1;
  213.     } else {
  214.       *p += num;
  215.       return 1;
  216.     }
  217.   }
  218.   if (ip0 > *p) {
  219.     /* ip + num > 0xffffffff */
  220.     return 0;
  221.   }
  222.   
  223.   return 1;
  224. }
  226. int proto_trans(int type)
  227. {
  228.   switch (type)
  229.     {
  230.     case ZEBRA_ROUTE_SYSTEM:
  231.       return 1; /* other */
  232.     case ZEBRA_ROUTE_KERNEL:
  233.       return 1; /* other */
  234.     case ZEBRA_ROUTE_CONNECT:
  235.       return 2; /* local interface */
  236.     case ZEBRA_ROUTE_STATIC:
  237.       return 3; /* static route */
  238.     case ZEBRA_ROUTE_RIP:
  239.       return 8; /* rip */
  240.     case ZEBRA_ROUTE_RIPNG:
  241.       return 1; /* shouldn't happen */
  242.     case ZEBRA_ROUTE_OSPF:
  243.       return 13; /* ospf */
  244.     case ZEBRA_ROUTE_OSPF6:
  245.       return 1; /* shouldn't happen */
  246.     case ZEBRA_ROUTE_BGP:
  247.       return 14; /* bgp */
  248.     default:
  249.       return 1; /* other */
  250.     }
  251. }
  253. void
  254. check_replace(struct route_node *np2, struct rib *rib2, 
  255.               struct route_node **np, struct rib **rib)
  256. {
  257.   int proto, proto2;
  259.   if (!*np)
  260.     {
  261.       *np = np2;
  262.       *rib = rib2;
  263.       return;
  264.     }
  266.   if (in_addr_cmp(&(*np)->p.u.prefix, &np2->p.u.prefix) < 0)
  267.     return;
  268.   if (in_addr_cmp(&(*np)->p.u.prefix, &np2->p.u.prefix) > 0)
  269.     {
  270.       *np = np2;
  271.       *rib = rib2;
  272.       return;
  273.     }
  275.   proto = proto_trans((*rib)->type);
  276.   proto2 = proto_trans(rib2->type);
  278.   if (proto2 > proto)
  279.     return;
  280.   if (proto2 < proto)
  281.     {
  282.       *np = np2;
  283.       *rib = rib2;
  284.       return;
  285.     }
  287.   if (in_addr_cmp((u_char *)&(*rib)->nexthop->gate.ipv4, 
  288.                   (u_char *)&rib2->nexthop->gate.ipv4) <= 0)
  289.     return;
  291.   *np = np2;
  292.   *rib = rib2;
  293.   return;
  294. }
  296. void
  297. get_fwtable_route_node(struct variable *v, oid objid[], size_t *objid_len, 
  298.        int exact, struct route_node **np, struct rib **rib)
  299. {
  300.   struct in_addr dest;
  301.   struct route_table *table;
  302.   struct route_node *np2;
  303.   struct rib *rib2;
  304.   int proto;
  305.   int policy;
  306.   struct in_addr nexthop;
  307.   u_char *pnt;
  308.   int i;
  310.   /* Init index variables */
  312.   pnt = (u_char *) &dest;
  313.   for (i = 0; i < 4; i++)
  314.     *pnt++ = 0;
  316.   pnt = (u_char *) &nexthop;
  317.   for (i = 0; i < 4; i++)
  318.     *pnt++ = 0;
  320.   proto = 0;
  321.   policy = 0;
  322.  
  323.   /* Init return variables */
  325.   *np = NULL;
  326.   *rib = NULL;
  328.   /* Short circuit exact matches of wrong length */
  330.   if (exact && (*objid_len != v->namelen + 10))
  331.     return;
  333.   table = vrf_table (AFI_IP, SAFI_UNICAST, 0);
  334.   if (! table)
  335.     return;
  337.   /* Get INDEX information out of OID.
  338.    * ipForwardDest, ipForwardProto, ipForwardPolicy, ipForwardNextHop
  339.    */
  341.   if (*objid_len > v->namelen)
  342.     oid2in_addr (objid + v->namelen, MIN(4, *objid_len - v->namelen), &dest);
  344.   if (*objid_len > v->namelen + 4)
  345.     proto = objid[v->namelen + 4];
  347.   if (*objid_len > v->namelen + 5)
  348.     policy = objid[v->namelen + 5];
  350.   if (*objid_len > v->namelen + 6)
  351.     oid2in_addr (objid + v->namelen + 6, MIN(4, *objid_len - v->namelen - 6),
  352.  &nexthop);
  354.   /* Apply GETNEXT on not exact search */
  356.   if (!exact && (*objid_len >= v->namelen + 10))
  357.     {
  358.       if (! in_addr_add((u_char *) &nexthop, 1)) 
  359.         return;
  360.     }
  362.   /* For exact: search matching entry in rib table. */
  364.   if (exact)
  365.     {
  366.       if (policy) /* Not supported (yet?) */
  367.         return;
  368.       for (*np = route_top (table); *np; *np = route_next (*np))
  369. {
  370.   if (!in_addr_cmp(&(*np)->p.u.prefix, (u_char *)&dest))
  371.     {
  372.       for (*rib = (*np)->info; *rib; *rib = (*rib)->next)
  373.         {
  374.   if (!in_addr_cmp((u_char *)&(*rib)->nexthop->gate.ipv4,
  375.    (u_char *)&nexthop))
  376.     if (proto == proto_trans((*rib)->type))
  377.       return;
  378. }
  379.     }
  380. }
  381.       return;
  382.     }
  384.   /* Search next best entry */
  386.   for (np2 = route_top (table); np2; np2 = route_next (np2))
  387.     {
  389.       /* Check destination first */
  390.       if (in_addr_cmp(&np2->p.u.prefix, (u_char *)&dest) > 0)
  391.         for (rib2 = np2->info; rib2; rib2 = rib2->next)
  392.   check_replace(np2, rib2, np, rib);
  394.       if (in_addr_cmp(&np2->p.u.prefix, (u_char *)&dest) == 0)
  395.         { /* have to look at each rib individually */
  396.           for (rib2 = np2->info; rib2; rib2 = rib2->next)
  397.     {
  398.       int proto2, policy2;
  400.       proto2 = proto_trans(rib2->type);
  401.       policy2 = 0;
  403.       if ((policy < policy2)
  404.   || ((policy == policy2) && (proto < proto2))
  405.   || ((policy == policy2) && (proto == proto2)
  406.       && (in_addr_cmp((u_char *)&rib2->nexthop->gate.ipv4,
  407.       (u_char *) &nexthop) >= 0)
  408.       ))
  409. check_replace(np2, rib2, np, rib);
  410.     }
  411. }
  412.     }
  414.   if (!*rib)
  415.     return;
  417.   policy = 0;
  418.   proto = proto_trans((*rib)->type);
  420.   *objid_len = v->namelen + 10;
  421.   pnt = (u_char *) &(*np)->p.u.prefix;
  422.   for (i = 0; i < 4; i++)
  423.     objid[v->namelen + i] = *pnt++;
  425.   objid[v->namelen + 4] = proto;
  426.   objid[v->namelen + 5] = policy;
  428.   {
  429.     struct nexthop *nexthop;
  431.     nexthop = (*rib)->nexthop;
  432.     if (nexthop)
  433.       {
  434. pnt = (u_char *) &nexthop->gate.ipv4;
  435. for (i = 0; i < 4; i++)
  436.   objid[i + v->namelen + 6] = *pnt++;
  437.       }
  438.   }
  440.   return;
  441. }
  443. u_char *
  444. ipFwTable (struct variable *v, oid objid[], size_t *objid_len,
  445.    int exact, size_t *val_len, WriteMethod **write_method)
  446. {
  447.   struct route_node *np;
  448.   struct rib *rib;
  449.   static int result;
  450.   static int resarr[2];
  451.   static struct in_addr netmask;
  452.   struct nexthop *nexthop;
  454.   get_fwtable_route_node(v, objid, objid_len, exact, &np, &rib);
  455.   if (!np)
  456.     return NULL;
  458.   nexthop = rib->nexthop;
  459.   if (! nexthop)
  460.     return NULL;
  462.   switch (v->magic)
  463.     {
  464.     case IPFORWARDDEST:
  465.       *val_len = 4;
  466.       return &np->p.u.prefix;
  467.       break;
  468.     case IPFORWARDMASK:
  469.       masklen2ip(np->p.prefixlen, &netmask);
  470.       *val_len = 4;
  471.       return (u_char *)&netmask;
  472.       break;
  473.     case IPFORWARDPOLICY:
  474.       result = 0;
  475.       *val_len  = sizeof(int);
  476.       return (u_char *)&result;
  477.       break;
  478.     case IPFORWARDNEXTHOP:
  479.       *val_len = 4;
  480.       return (u_char *)&nexthop->gate.ipv4;
  481.       break;
  482.     case IPFORWARDIFINDEX:
  483.       *val_len = sizeof(int);
  484.       return (u_char *)&nexthop->ifindex;
  485.       break;
  486.     case IPFORWARDTYPE:
  487.       if (nexthop->type == NEXTHOP_TYPE_IFINDEX
  488.   || nexthop->type == NEXTHOP_TYPE_IFNAME)
  489.         result = 3;
  490.       else
  491.         result = 4;
  492.       *val_len  = sizeof(int);
  493.       return (u_char *)&result;
  494.       break;
  495.     case IPFORWARDPROTO:
  496.       result = proto_trans(rib->type);
  497.       *val_len  = sizeof(int);
  498.       return (u_char *)&result;
  499.       break;
  500.     case IPFORWARDAGE:
  501.       result = 0;
  502.       *val_len  = sizeof(int);
  503.       return (u_char *)&result;
  504.       break;
  505.     case IPFORWARDINFO:
  506.       resarr[0] = 0;
  507.       resarr[1] = 0;
  508.       *val_len  = 2 * sizeof(int);
  509.       return (u_char *)resarr;
  510.       break;
  511.     case IPFORWARDNEXTHOPAS:
  512.       result = -1;
  513.       *val_len  = sizeof(int);
  514.       return (u_char *)&result;
  515.       break;
  516.     case IPFORWARDMETRIC1:
  517.       result = 0;
  518.       *val_len  = sizeof(int);
  519.       return (u_char *)&result;
  520.       break;
  521.     case IPFORWARDMETRIC2:
  522.       result = 0;
  523.       *val_len  = sizeof(int);
  524.       return (u_char *)&result;
  525.       break;
  526.     case IPFORWARDMETRIC3:
  527.       result = 0;
  528.       *val_len  = sizeof(int);
  529.       return (u_char *)&result;
  530.       break;
  531.     case IPFORWARDMETRIC4:
  532.       result = 0;
  533.       *val_len  = sizeof(int);
  534.       return (u_char *)&result;
  535.       break;
  536.     case IPFORWARDMETRIC5:
  537.       result = 0;
  538.       *val_len  = sizeof(int);
  539.       return (u_char *)&result;
  540.       break;
  541.     default:
  542.       return NULL;
  543.       break;
  544.     }  
  545.   return NULL;
  546. }
  548. u_char *
  549. ipCidrTable (struct variable *v, oid objid[], size_t *objid_len,
  550.      int exact, size_t *val_len, WriteMethod **write_method)
  551. {
  552.   switch (v->magic)
  553.     {
  554.     case IPCIDRROUTEDEST:
  555.       break;
  556.     default:
  557.       return NULL;
  558.       break;
  559.     }  
  560.   return NULL;
  561. }
  563. void
  564. zebra_snmp_init ()
  565. {
  566.   smux_init (zebra_oid, sizeof (zebra_oid) / sizeof (oid));
  567.   REGISTER_MIB("mibII/ipforward", zebra_variables, variable, ipfw_oid);
  568.   smux_start ();
  569. }
  570. #endif /* HAVE_SNMP */