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

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

网络

开发平台：

Unix_Linux

ospf6_test.c：源码内容
  2. #include <zebra.h>
  3. #include <math.h>
  5. #include "log.h"
  6. #include "memory.h"
  7. #include "prefix.h"
  8. #include "table.h"
  9. #include "command.h"
  10. #include "vty.h"
  11. #include "thread.h"
  12. #include "linklist.h"
  13. #include "if.h"
  15. #include "ospf6_proto.h"
  16. #include "ospf6_lsa.h"
  17. #include "ospf6_lsdb.h"
  18. #include "ospf6_route.h"
  19. #include "ospf6_spf.h"
  21. #include "ospf6_top.h"
  22. #include "ospf6_area.h"
  23. #include "ospf6_interface.h"
  24. #include "ospf6_intra.h"
  26. extern char *malloc_options;
  28. struct thread_master *master;
  30. #define NUMBER_OF_TRIALS 1000000
  31. int N = 4;
  33. #define ROW(n) ((n) % N)
  34. #define COL(n) ((n) / N)
  36. #define IS_TOPOF(n) (COL(n) > 0)
  37. #define IS_BOTTOMOF(n) (COL(n) < N - 1)
  38. #define IS_LEFTOF(n) (ROW(n) > 0)
  39. #define IS_RIGHTOF(n) (ROW(n) < N - 1)
  40. #define TOPOF(n) (n - N)
  41. #define BOTTOMOF(n) (n + N)
  42. #define LEFTOF(n) (n - 1)
  43. #define RIGHTOF(n) (n + 1)
  45. #define IFINDEX_TOP    0
  46. #define IFINDEX_BOTTOM 1
  47. #define IFINDEX_LEFT   2
  48. #define IFINDEX_RIGHT  3
  50. struct in6_addr topaddr =
  51. {{{ 0xfe, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  52.     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, IFINDEX_TOP }}};
  53. struct in6_addr leftaddr =
  54. {{{ 0xfe, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  55.     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, IFINDEX_LEFT }}};
  56. struct in6_addr rightaddr =
  57. {{{ 0xfe, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  58.     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, IFINDEX_RIGHT }}};
  59. struct in6_addr bottomaddr =
  60. {{{ 0xfe, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  61.     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, IFINDEX_BOTTOM }}};
  63. void
  64. install_router_lsa (u_int32_t router_id, struct ospf6_lsdb *lsdb)
  65. {
  66.   char buffer[OSPF6_MAX_LSASIZE];
  67.   struct ospf6_lsa_header *lsa_header;
  68.   struct ospf6_lsa *lsa;
  69.   struct ospf6_router_lsa *router_lsa;
  70.   struct ospf6_router_lsdesc *lsdesc;
  72.   //printf ("install %dn", router_id);
  74.   lsa_header = (struct ospf6_lsa_header *) buffer;
  75.   router_lsa = (struct ospf6_router_lsa *)
  76.     ((caddr_t) lsa_header + sizeof (struct ospf6_lsa_header));
  77.   lsdesc = (struct ospf6_router_lsdesc *)
  78.     ((caddr_t) router_lsa + sizeof (struct ospf6_router_lsa));
  80.   if (IS_TOPOF (router_id))
  81.     {
  82.       lsdesc->type = OSPF6_ROUTER_LSDESC_POINTTOPOINT;
  83.       lsdesc->metric = htons (1);
  84.       lsdesc->interface_id = htonl (IFINDEX_TOP);
  85.       lsdesc->neighbor_interface_id = htonl (IFINDEX_BOTTOM);
  86.       lsdesc->neighbor_router_id = htonl (TOPOF (router_id));
  87.       lsdesc++;
  88.     }
  89.   if (IS_LEFTOF (router_id))
  90.     {
  91.       lsdesc->type = OSPF6_ROUTER_LSDESC_POINTTOPOINT;
  92.       lsdesc->metric = htons (1);
  93.       lsdesc->interface_id = htonl (IFINDEX_LEFT);
  94.       lsdesc->neighbor_interface_id = htonl (IFINDEX_RIGHT);
  95.       lsdesc->neighbor_router_id = htonl (LEFTOF (router_id));
  96.       lsdesc++;
  97.     }
  98.   if (IS_RIGHTOF (router_id))
  99.     {
  100.       lsdesc->type = OSPF6_ROUTER_LSDESC_POINTTOPOINT;
  101.       lsdesc->metric = htons (1);
  102.       lsdesc->interface_id = htonl (IFINDEX_RIGHT);
  103.       lsdesc->neighbor_interface_id = htonl (IFINDEX_LEFT);
  104.       lsdesc->neighbor_router_id = htonl (RIGHTOF (router_id));
  105.       lsdesc++;
  106.     }
  107.   if (IS_BOTTOMOF (router_id))
  108.     {
  109.       lsdesc->type = OSPF6_ROUTER_LSDESC_POINTTOPOINT;
  110.       lsdesc->metric = htons (1);
  111.       lsdesc->interface_id = htonl (IFINDEX_BOTTOM);
  112.       lsdesc->neighbor_interface_id = htonl (IFINDEX_TOP);
  113.       lsdesc->neighbor_router_id = htonl (BOTTOMOF (router_id));
  114.       lsdesc++;
  115.     }
  117.   OSPF6_OPT_SET (router_lsa->options, OSPF6_OPT_V6);
  118.   OSPF6_OPT_SET (router_lsa->options, OSPF6_OPT_E);
  119.   OSPF6_OPT_CLEAR (router_lsa->options, OSPF6_OPT_MC);
  120.   OSPF6_OPT_CLEAR (router_lsa->options, OSPF6_OPT_N);
  121.   OSPF6_OPT_SET (router_lsa->options, OSPF6_OPT_R);
  122.   OSPF6_OPT_CLEAR (router_lsa->options, OSPF6_OPT_DC);
  123.   UNSET_FLAG (router_lsa->bits, OSPF6_ROUTER_BIT_B);
  124.   UNSET_FLAG (router_lsa->bits, OSPF6_ROUTER_BIT_E);
  125.   UNSET_FLAG (router_lsa->bits, OSPF6_ROUTER_BIT_V);
  126.   UNSET_FLAG (router_lsa->bits, OSPF6_ROUTER_BIT_W);
  128.   lsa_header->age = 0;
  129.   lsa_header->type = htons (OSPF6_LSTYPE_ROUTER);
  130.   lsa_header->id = htonl (0);
  131.   lsa_header->adv_router = htonl (router_id);
  132.   lsa_header->seqnum = htonl (INITIAL_SEQUENCE_NUMBER);
  133.   lsa_header->length = htons ((caddr_t) lsdesc - (caddr_t) buffer);
  135.   lsa = ospf6_lsa_create (lsa_header);
  136.   ospf6_lsdb_add (lsa, lsdb);
  137. }
  139. static double
  140. diffsec (struct timeval *end, struct timeval *start)
  141. {
  142.   double time;
  143.   time = (end->tv_sec - start->tv_sec) * 1.0 +
  144.          (end->tv_usec - start->tv_usec) / 1000000.0;
  145.   if (time < 0.0)
  146.     fprintf (stderr, "time: %f end{%010lu,%010lu} start{%010lu,%010lu}n",
  147.              time, end->tv_sec, end->tv_usec, start->tv_sec, start->tv_usec);
  148.   return time;
  149. }
  151. void
  152. lsdb_lookup_try (int ntry, struct ospf6_lsdb *lsdb)
  153. {
  154.   struct ospf6_lsa *lsa;
  155.   struct timeval start, now;
  156.   double time, timesum, timesumsq;
  157.   double max, min, avg, variance, std;
  158.   int rets, retn;
  159.   int i;
  160.   u_int32_t router_id;
  162.   max = 0.0;
  163.   min = 999999999.0;
  164.   timesum = 0.0;
  165.   timesumsq = 0.0;
  166.   for (i = 0; i < ntry; i++)
  167.     {
  168.       router_id = random () % (N * N);
  170.       memset (&start, 0, sizeof (&start));
  171.       memset (&now, 0, sizeof (&now));
  173.       lsa = NULL;
  174.       rets = gettimeofday (&start, NULL);
  175.       lsa = ospf6_lsdb_lookup (htons (OSPF6_LSTYPE_ROUTER), htonl (0),
  176.                                htonl (router_id), lsdb);
  177.       retn = gettimeofday (&now, NULL);
  178.       assert (lsa);
  180.       if (rets != 0 || retn != 0)
  181.         {
  182.           fprintf (stderr, "gettimeofday failed: %sn", strerror (errno));
  183.           i--;
  184.           continue;
  185.         }
  187.       time = diffsec (&now, &start);
  188.       timesum += time;
  189.       timesumsq += time * time;
  190.       max = (time > max ? time : max);
  191.       min = (time < min ? time : min);
  192.     }
  194.   avg = timesum / ntry;
  195.   variance = (timesumsq - ntry * avg * avg) / ntry;
  196.   std = sqrt (variance);
  198.   printf ("test lsdb: #LSAs: %d #lookup: %d min: %.3f avg: %.3f max: %.3f stddev: %.3f (ms)n",
  199.           N * N, ntry, min * 1000.0, avg * 1000.0, max * 1000.0, std * 1000.0);
  200. }
  202. void
  203. install_link_lsdb (u_int32_t router_id, int ifindex)
  204. {
  205.   struct interface *ifp;
  206.   struct ospf6_interface *oi;
  207.   char *ifname = NULL;
  208.   u_int32_t peer = 0;
  209.   int peer_ifindex = 0;
  210.   char buffer[OSPF6_MAX_LSASIZE];
  211.   struct ospf6_lsa_header *lsa_header;
  212.   struct ospf6_link_lsa *link_lsa;
  213.   struct ospf6_lsa *lsa;
  214.   char *p;
  216.   if (ifindex == IFINDEX_TOP && IS_TOPOF (router_id))
  217.     {
  218.       ifname = "top";
  219.       peer = TOPOF (router_id);
  220.       peer_ifindex = IFINDEX_BOTTOM;
  221.     }
  222.   else if (ifindex == IFINDEX_LEFT && IS_LEFTOF (router_id))
  223.     {
  224.       ifname = "left";
  225.       peer = LEFTOF (router_id);
  226.       peer_ifindex = IFINDEX_RIGHT;
  227.     }
  228.   else if (ifindex == IFINDEX_RIGHT && IS_RIGHTOF (router_id))
  229.     {
  230.       ifname = "right";
  231.       peer = RIGHTOF (router_id);
  232.       peer_ifindex = IFINDEX_LEFT;
  233.     }
  234.   else if (ifindex == IFINDEX_BOTTOM && IS_BOTTOMOF (router_id))
  235.     {
  236.       ifname = "bottom";
  237.       peer = BOTTOMOF (router_id);
  238.       peer_ifindex = IFINDEX_TOP;
  239.     }
  240.   else
  241.     assert (0);
  243.   ifp = if_get_by_name (ifname);
  244.   ifp->ifindex = ifindex;
  246.   oi = malloc (sizeof (struct ospf6_interface));
  247.   oi->interface = ifp;
  248.   ifp->info = oi;
  250.   oi->lsdb = ospf6_lsdb_create (oi);
  252.   memset (buffer, 0, sizeof (buffer));
  253.   lsa_header = (struct ospf6_lsa_header *) buffer;
  254.   link_lsa = (struct ospf6_link_lsa *)
  255.     ((caddr_t) lsa_header + sizeof (struct ospf6_lsa_header));
  256.   p = ((caddr_t) link_lsa + sizeof (struct ospf6_link_lsa));
  258.   link_lsa->priority = 1;
  259.   link_lsa->options[0] = 0;
  260.   link_lsa->options[1] = 0;
  261.   link_lsa->options[2] = 0;
  262.   if (ifindex == IFINDEX_TOP)
  263.     link_lsa->linklocal_addr = topaddr;
  264.   else if (ifindex == IFINDEX_LEFT)
  265.     link_lsa->linklocal_addr = leftaddr;
  266.   else if (ifindex == IFINDEX_RIGHT)
  267.     link_lsa->linklocal_addr = rightaddr;
  268.   else
  269.     link_lsa->linklocal_addr = bottomaddr;
  270.   link_lsa->prefix_num = htonl (0);
  272.   lsa_header->age = 0;
  273.   lsa_header->type = htons (OSPF6_LSTYPE_LINK);
  274.   lsa_header->id = htonl (ifindex);
  275.   lsa_header->adv_router = htonl (router_id);
  276.   lsa_header->seqnum = htonl (INITIAL_SEQUENCE_NUMBER);
  277.   lsa_header->length = htons ((caddr_t) p - (caddr_t) buffer);
  279.   lsa = ospf6_lsa_create (lsa_header);
  280.   ospf6_lsdb_add (lsa, oi->lsdb);
  282.   /* peer's */
  283.   memset (buffer, 0, sizeof (buffer));
  284.   lsa_header = (struct ospf6_lsa_header *) buffer;
  285.   link_lsa = (struct ospf6_link_lsa *)
  286.     ((caddr_t) lsa_header + sizeof (struct ospf6_lsa_header));
  287.   p = ((caddr_t) link_lsa + sizeof (struct ospf6_link_lsa));
  289.   link_lsa->priority = 1;
  290.   link_lsa->options[0] = 0;
  291.   link_lsa->options[1] = 0;
  292.   link_lsa->options[2] = 0;
  293.   if (peer_ifindex == IFINDEX_TOP)
  294.     link_lsa->linklocal_addr = topaddr;
  295.   else if (peer_ifindex == IFINDEX_LEFT)
  296.     link_lsa->linklocal_addr = leftaddr;
  297.   else if (peer_ifindex == IFINDEX_RIGHT)
  298.     link_lsa->linklocal_addr = rightaddr;
  299.   else
  300.     link_lsa->linklocal_addr = bottomaddr;
  301.   link_lsa->prefix_num = htonl (0);
  303.   lsa_header->age = 0;
  304.   lsa_header->type = htons (OSPF6_LSTYPE_LINK);
  305.   lsa_header->id = htonl (peer_ifindex);
  306.   lsa_header->adv_router = htonl (peer);
  307.   lsa_header->seqnum = htonl (INITIAL_SEQUENCE_NUMBER);
  308.   lsa_header->length = htons ((caddr_t) p - (caddr_t) buffer);
  310.   lsa = ospf6_lsa_create (lsa_header);
  311.   ospf6_lsdb_add (lsa, oi->lsdb);
  313.   return;
  314. }
  316. void
  317. spf_try (u_int32_t router_id, struct ospf6_lsdb *lsdb)
  318. {
  319.   struct ospf6_area *oa;
  320.   struct ospf6_interface *oi;
  321.   struct timeval start, end;
  322.   double time;
  323.   int rets, rete;
  324.   listnode node;
  326.   oa = malloc (sizeof (struct ospf6_area));
  327.   snprintf (oa->name, sizeof (oa->name), "spf_test");
  328.   oa->lsdb = lsdb;
  329.   oa->spf_table = ospf6_route_table_create (); 
  331.   oa->if_list = list_new ();
  332.   if (IS_TOPOF (router_id))
  333.     {
  334.       install_link_lsdb (router_id, IFINDEX_TOP);
  335.       oi = ospf6_interface_lookup_by_ifindex (IFINDEX_TOP);
  336.       listnode_add (oa->if_list, oi);
  337.     }
  338.   if (IS_LEFTOF (router_id))
  339.     {
  340.       install_link_lsdb (router_id, IFINDEX_LEFT);
  341.       oi = ospf6_interface_lookup_by_ifindex (IFINDEX_LEFT);
  342.       listnode_add (oa->if_list, oi);
  343.     }
  344.   if (IS_RIGHTOF (router_id))
  345.     {
  346.       install_link_lsdb (router_id, IFINDEX_RIGHT);
  347.       oi = ospf6_interface_lookup_by_ifindex (IFINDEX_RIGHT);
  348.       listnode_add (oa->if_list, oi);
  349.     }
  350.   if (IS_BOTTOMOF (router_id))
  351.     {
  352.       install_link_lsdb (router_id, IFINDEX_BOTTOM);
  353.       oi = ospf6_interface_lookup_by_ifindex (IFINDEX_BOTTOM);
  354.       listnode_add (oa->if_list, oi);
  355.     }
  357.   rets = gettimeofday (&start, (struct timezone *) NULL);
  358.   ospf6_spf_calculation (htonl (router_id), oa->spf_table, oa);
  359.   rete = gettimeofday (&end, (struct timezone *) NULL);
  361.   if (rets != 0 || rete != 0)
  362.     fprintf (stderr, "gettimeofday failed: %sn", strerror (errno));
  363.   time = diffsec (&end, &start);
  365.   printf ("test spf: root: %lu #LSAs: %d #routes: %d time: %.3f (ms)n",
  366.           (unsigned long) router_id, N * N, oa->spf_table->count,
  367.           time * 1000.0);
  369.   for (node = listhead (oa->if_list); node; nextnode (node))
  370.     {
  371.       oi = (struct ospf6_interface *) getdata (node);
  372.       ospf6_lsdb_delete (oi->lsdb);
  373.       oi->interface->info = NULL;
  374.       free (oi);
  375.     }
  376.   list_delete (oa->if_list);
  378.   ospf6_route_table_delete (oa->spf_table);
  379.   free (oa);
  380. }
  382. int
  383. main (int argc, char **argv)
  384. {
  385.   struct ospf6_lsdb *lsdb;
  386.   u_int32_t router_id;
  387.   struct timeval start, now;
  388.   double timeadd;
  390. #if 0
  391.   malloc_options = "X";
  392. #endif
  394.   if (argc > 1)
  395.     N = atoi (argv[1]);
  397.   lsdb = ospf6_lsdb_create (NULL);
  399.   fprintf (stderr, "Installing %d virtual Router-LSA ... ", N * N);
  400.   fflush (stdout);
  401.   gettimeofday (&start, NULL);
  402.   for (router_id = 0; router_id < N * N; router_id++)
  403.     install_router_lsa (router_id, lsdb);
  404.   gettimeofday (&now, NULL);
  405.   timeadd = diffsec (&now, &start);
  406.   fprintf (stderr, "done (%.3f ms) n", timeadd * 1000.0);
  408.   gettimeofday (&now, NULL);
  409.   srandom (now.tv_sec);
  411.   //lsdb_lookup_try (NUMBER_OF_TRIALS, lsdb);
  413.   cmd_init (0);
  414.   vty_init ();
  415.   if_init ();
  416.   zlog_default = openzlog ("ospf6_test", ZLOG_STDERR, ZLOG_OSPF6,
  417.                            LOG_CONS|LOG_NDELAY|LOG_PERROR|LOG_PID,
  418.                            LOG_DAEMON);
  420.   OSPF6_DEBUG_SPF_OFF (OSPF6_DEBUG_SPF_PROCESS);
  421.   spf_try (0, lsdb);
  422.   spf_try ((N / 2) * (N + 1), lsdb);
  424.   return 0;
  425. }