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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 通讯编程 > 查看源码
tora.cc
资源名称：ns-allinone-2.33.tar.gz　[点击查看]
上传用户：rrhhcc
上传日期：2015-12-11
资源大小：54129k
文件大小：16k
源码类别：

通讯编程

开发平台：

Visual C++

tora.cc：源码内容
  1. /* -*-  Mode:C++; c-basic-offset:8; tab-width:8; indent-tabs-mode:t -*- */
  2. /*
  3.  * Copyright (c) 1997 Regents of the University of California.
  4.  * All rights reserved.
  5.  *
  6.  * Redistribution and use in source and binary forms, with or without
  7.  * modification, are permitted provided that the following conditions
  8.  * are met:
  9.  * 1. Redistributions of source code must retain the above copyright
  10.  *    notice, this list of conditions and the following disclaimer.
  11.  * 2. Redistributions in binary form must reproduce the above copyright
  12.  *    notice, this list of conditions and the following disclaimer in the
  13.  *    documentation and/or other materials provided with the distribution.
  14.  * 3. All advertising materials mentioning features or use of this software
  15.  *    must display the following acknowledgement:
  16.  *      This product includes software developed by the Computer Systems
  17.  *      Engineering Group at Lawrence Berkeley Laboratory.
  18.  * 4. Neither the name of the University nor of the Laboratory may be used
  19.  *    to endorse or promote products derived from this software without
  20.  *    specific prior written permission.
  21.  *
  22.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  23.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  26.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  27.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  28.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  29.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  31.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32.  * SUCH DAMAGE.
  33.  *
  34.  * Ported from CMU/Monarch's code
  35.  *
  36.  * $Header: /cvsroot/nsnam/ns-2/tora/tora.cc,v 1.18 2005/08/28 23:23:03 tomh Exp $
  37.  */
  39. #include <agent.h>
  40. #include <random.h>
  41. #include <trace.h>
  43. #include <ll.h>
  44. #include <priqueue.h>
  45. #include <tora/tora_packet.h>
  46. #include <tora/tora.h>
  48. #define LOG(s)
  49. fprintf(stdout, "%s --- %s (index: %d, time: %f)n",
  50. __PRETTY_FUNCTION__, (s), index, Scheduler::instance().clock())
  52. // #define DEBUG
  53. #define CURRENT_TIME Scheduler::instance().clock()
  55. /* sec of spacing inserted between pkts when a bunch of packets are
  56.    dumped into the link layer all at once.  Allows arp time to resolve
  57.    dst, preventing the dumping of all but the last pkt on the floor */
  58. #define ARP_SEPARATION_DELAY 0.030
  59.   
  61. /* ======================================================================
  62.    TCL Hooks
  63.    ====================================================================== */
  64. int hdr_tora::offset_;
  65. static class TORAHeaderClass : public PacketHeaderClass {
  66. public:
  67.         TORAHeaderClass() : PacketHeaderClass("PacketHeader/TORA",
  68.       TORA_HDR_LEN) { 
  69. bind_offset(&hdr_tora::offset_);
  70. } 
  71. } class_toraAgent_hdr;
  73. static class toraAgentclass : public TclClass {
  74. public:
  75. toraAgentclass() : TclClass("Agent/TORA") {}
  76. TclObject* create(int argc, const char*const* argv) {
  77. assert(argc == 5);
  78. return (new toraAgent((nsaddr_t) atoi(argv[4])));
  79. }
  80. } class_toraAgent;
  83. /* ======================================================================
  84.    toraAgent Class Functions
  85.    ====================================================================== */
  86. toraAgent::toraAgent(nsaddr_t id) :
  87. rtAgent(id, PT_TORA),
  88. rqueue()
  89. {
  90. LIST_INIT(&dstlist);
  91. imepagent = 0;
  92. logtarget = 0;
  93. ifqueue = 0;
  94. }
  96. void
  97. toraAgent::reset()
  98. {
  99. Packet *p;
  100. while((p = rqueue.deque())) {
  101. drop(p,DROP_END_OF_SIMULATION);
  102. }
  103. }
  105. int
  106. toraAgent::command(int argc, const char*const* argv)
  107. {
  108. if(argc == 2) {
  109. Tcl& tcl = Tcl::instance();
  111. if(strncasecmp(argv[1], "id", 2) == 0) {
  112. tcl.resultf("%d", index);
  113. return TCL_OK;
  114. }
  115. }
  116. else if(argc == 3) {
  118. if(strcmp(argv[1], "log-target") == 0 || strcmp(argv[1], "tracetarget") == 0 ) {
  119. logtarget = (Trace*) TclObject::lookup(argv[2]);
  120. if(logtarget == 0)
  121. return TCL_ERROR;
  122. return TCL_OK;
  123. }
  124. else if(strcmp(argv[1], "drop-target") == 0) {
  125.         int stat = rqueue.command(argc,argv);
  126. if (stat != TCL_OK) return stat;
  127.                  return Agent::command(argc, argv);
  128. }
  129. else if(strcmp(argv[1], "if-queue") == 0) {
  130. ifqueue = (PriQueue*) TclObject::lookup(argv[2]);
  131. if(ifqueue == 0)
  132. return TCL_ERROR;
  133. return TCL_OK;
  134. }
  135. else if(strcmp(argv[1], "imep-agent") == 0) {
  136. imepagent = (imepAgent*) TclObject::lookup(argv[2]);
  137. if(imepagent == 0)
  138. return TCL_ERROR;
  139. imepagent->imepRegister((rtAgent*) this);
  140. return TCL_OK;
  141. }
  142. }
  143. return Agent::command(argc, argv);
  144. }
  147. /* ======================================================================
  148.    Destination Management Functions
  149.    ====================================================================== */
  150. TORADest*
  151. toraAgent::dst_find(nsaddr_t id)
  152. {
  153. TORADest* td = dstlist.lh_first;
  154. for( ; td; td = td->link.le_next) {
  155. if(td->index == id)
  156. return td;
  157. }
  158. return 0;
  159. }
  161. TORADest*
  162. toraAgent::dst_add(nsaddr_t id)
  163. {
  164. TORADest *td = new TORADest(id, this);
  165. assert(td);
  167. LIST_INSERT_HEAD(&dstlist, td, link);
  169. int *nblist = 0, nbcnt = 0;
  170. imepagent->imepGetBiLinks(nblist, nbcnt);
  172. for(int i = 0; i < nbcnt; i++)
  173. (void) td->nb_add(nblist[i]);
  175. if(nblist) delete[] nblist;
  177. return td;
  178. }
  180. void
  181. toraAgent::dst_dump()
  182. {
  183. TORADest *td = dstlist.lh_first;
  185. for( ; td; td = td->link.le_next)
  186. td->dump();
  187. }
  190. /* ======================================================================
  191.    Route Resolution
  192.    ====================================================================== */
  193. void
  194. toraAgent::forward(Packet *p, nsaddr_t nexthop, Time delay)
  195. {
  196.         struct hdr_cmn *ch = HDR_CMN(p);
  198. #ifdef TORA_DISALLOW_ROUTE_LOOP
  199.         if(nexthop == ch->prev_hop_) {
  200. log_route_loop(ch->prev_hop_, nexthop);
  201.                 drop(p, DROP_RTR_ROUTE_LOOP);
  202.                 return;
  203.         }
  204. #endif
  205.         ch->next_hop() = nexthop;
  206.         ch->prev_hop_ = ipaddr();
  207.         ch->addr_type() = NS_AF_INET;
  209. // change the packet direction to DOWN
  210. ch->direction() = hdr_cmn::DOWN;
  212. if (0.0 == delay) {
  213.   tora_output(p);
  214. } else {
  215.   Scheduler::instance().schedule(target_, p, delay);
  216. }
  217. }
  220. void
  221. toraAgent::rt_resolve(Packet *p)
  222. {
  223. struct hdr_ip *ih = HDR_IP(p);
  224. TORADest *td;
  225. TORANeighbor *tn;
  227. td = dst_find(ih->daddr());
  228. if(td == 0) {
  229. td = dst_add(ih->daddr());
  230. }
  232. tn = td->nb_find_next_hop();
  233. if(tn == 0) {
  234. rqueue.enque(p);
  236. trace("T %.9f _%d_ tora enq %d->%d",
  237.       Scheduler::instance().clock(), ipaddr(), 
  238.       ih->saddr(), ih->daddr());
  240. if(!td->rt_req)
  241.   { // if no QRY pending, then send one
  242.     sendQRY(ih->daddr());
  243.     td->time_tx_qry = CURRENT_TIME;
  244.     td->rt_req = 1;
  245.   }
  246. }
  247. else {
  248.                 forward(p, tn->index);
  249. }
  250. }
  253. /* ======================================================================
  254.    Incoming Packets
  255.    ====================================================================== */
  256. void
  257. toraAgent::recv(Packet *p, Handler *)
  258. {
  259. struct hdr_cmn *ch = HDR_CMN(p);
  260. struct hdr_ip *ih = HDR_IP(p);
  262. assert(initialized());
  264. if(ch->ptype() == PT_TORA) {
  265. recvTORA(p);
  266. return;
  267. }
  269.         /*
  270.          *  Must be a packet I'm originating...
  271.          */
  272. if(ih->saddr() == ipaddr() && ch->num_forwards() == 0) {
  273.                 /*
  274.                  *  Add the IP Header.
  275.                  */
  276.                 ch->size() += IP_HDR_LEN;
  277.                 
  278.                 ih->ttl_ = IP_DEF_TTL;
  279. }
  281. #ifdef TORA_DISALLOW_ROUTE_LOOP
  282.         /*
  283.          *  I received a packet that I sent.  Probably
  284.          *  a routing loop.
  285.          */
  286.         else if(ih->saddr() == ipaddr()) {
  287.                 drop(p, DROP_RTR_ROUTE_LOOP);
  288.                 return;
  289.         }
  290. #endif
  291. /*
  292.  *  Packet I'm forwarding...
  293.  */
  294. else {
  295. /*
  296.  *  Check the TTL.  If it is zero, then discard.
  297.  */
  298. if(--ih->ttl_ == 0) {
  299. drop(p, DROP_RTR_TTL);
  300. return;
  301. }
  302. }
  304. rt_resolve(p);
  305. }
  307. void
  308. toraAgent::recvTORA(Packet *p)
  309. {
  310. struct hdr_tora *th = HDR_TORA(p);
  311. TORADest *td;
  312. TORANeighbor *tn;
  314. /*
  315.  * Fix the source IP address.
  316.  */
  317. assert(HDR_IP (p)->sport() == RT_PORT);
  318. assert(HDR_IP (p)->dport() == RT_PORT);
  320. /*
  321.  * Incoming Packets.
  322.  */
  323. switch(th->th_type) {
  324. case TORATYPE_QRY:
  325. recvQRY(p);
  326. Packet::free(p);
  327. return; // don't add/change routing state
  329. case TORATYPE_UPD:
  330. log_recv_upd(p);
  331. recvUPD(p);
  332. break;
  334. case TORATYPE_CLR:
  335. log_recv_clr(p);
  336. recvCLR(p);
  337. break;
  339. default:
  340. fprintf(stderr,
  341. "%s: Invalid TORA type (%x)n",
  342. __PRETTY_FUNCTION__, th->th_type);
  343. exit(1);
  344. }
  346. if((td = dst_find(th->th_dst)) == 0) {
  347. Packet::free(p);
  348. return;
  349. }
  351. logNextHopChange(td);
  353. if((tn = td->nb_find_next_hop())) {
  354. Packet *p0;
  355. Time delay = 0.0;
  357. while((p0 = rqueue.deque(td->index))) {
  358.                         forward(p0, tn->index, delay);
  359. delay += ARP_SEPARATION_DELAY;
  360. }
  361. }
  362. Packet::free(p);
  363. }
  367. /*
  368.  *  IETF Draft - TORA Specification, section 3.7.6
  369.  */
  370. void
  371. toraAgent::recvQRY(Packet *p)
  372. {
  373. struct hdr_ip *ih = HDR_IP(p);
  374. struct hdr_tora_qry *qh = HDR_TORA_QRY(p);
  375. TORADest *td;
  376. TORANeighbor *tn;
  378. if(qh->tq_dst == ipaddr()) {
  379. #ifdef DEBUG
  380. fprintf(stderr, "node %d received `QRY` for itself.n", index);
  381. #endif
  382. return;
  383. }
  385. td = dst_find(qh->tq_dst);
  386. if(td == 0)
  387. td = dst_add(qh->tq_dst);
  389. if(td->rt_req) {
  390. return;
  391. }
  393. if(td->height.r == 0) { // II, A
  394. tn = td->nb_find(ih->saddr());
  396. if(tn && tn->time_act > td->time_upd) { // II, A, 1
  397. td->time_upd = Scheduler::instance().clock();
  398. sendUPD(td->index);
  399. }
  400. else { // II, A, 2
  401. }
  402. }
  403. else {
  404. tn = td->nb_find_min_height(0);
  406. if(tn) { // II, B, 1
  407. td->update_height(tn->height.tau,
  408.   tn->height.oid,
  409.   tn->height.r,
  410.   tn->height.delta + 1,
  411.   ipaddr());
  413. td->time_upd = Scheduler::instance().clock();
  415. sendUPD(td->index);
  416. }
  417. else {
  418. td->rt_req = 1;
  419. td->time_rt_req = CURRENT_TIME;
  421. if(td->num_active > 1) { // II, B, 1, a
  422. sendQRY(td->index);
  423. }
  424. else { // II, B, 1, b
  426. }
  427. }
  428. }
  429. }
  432. /*
  433.  *  IETF Draft - TORA Specification, section 3.7.7
  434.  */
  435. void
  436. toraAgent::recvUPD(Packet *p)
  437. {
  438. struct hdr_ip *ih = HDR_IP(p);
  439. struct hdr_tora_upd *uh = HDR_TORA_UPD(p);
  440. TORADest *td;
  441. TORANeighbor *tn;
  443. if(uh->tu_dst == ipaddr()) {
  444. return;
  445. }
  447. td = dst_find(uh->tu_dst);
  448. if(td == 0)
  449. td = dst_add(uh->tu_dst);
  451. tn = td->nb_find(ih->saddr());
  452. if(tn == 0) {
  453. /*
  454.  * update link status? -josh
  455.  */
  457.          // No, don't update linkstatus: it may be an update
  458.          // that was delayed in the IMEP layer for sequencing -dam
  459.          // no way at the TORA level to tell if we're connected...
  460.          trace("T %.9f _%d_ received `UPD` from non-neighbor %d",
  461.        CURRENT_TIME, ipaddr(), ih->saddr());  
  462. #ifdef DEBUG
  463. fprintf(stderr,
  464.                         "node %d received `UPD` from non-neighbor %dn",
  465. index, ih->src_);
  466. #endif
  467. return;
  468. }
  470. /*
  471.  *  Update height and link status for neighbor [j][k].
  472.  */
  473. td->update_height_nb(tn, uh);
  475. if(td->rt_req && tn->height.r == 0) { // I
  477. td->update_height(tn->height.tau,
  478.   tn->height.oid,
  479.   tn->height.r,
  480.   tn->height.delta + 1,
  481.   ipaddr());
  483. td->rt_req = 0;
  485. td->time_upd = Scheduler::instance().clock();
  487. sendUPD(td->index);
  488. }
  489. else if(td->num_down == 0) { // II
  490. if(td->num_up == 0) { // II, A
  491. if(td->height.isNull()) // II, A, 1
  492. return; // II, A, 1, a
  493. else {
  494. td->height.Null(); // II, A, 1, b 
  496. td->time_upd = Scheduler::instance().clock();
  498. sendUPD(td->index);
  499. }
  500. }
  501. else {
  502. if(td->nb_check_same_ref()) { // II, A, 2
  503. TORANeighbor *tn;
  505. if( (tn = td->nb_find_min_height(0)) ) { // II, A, 2, a
  506. td->update_height(tn->height.tau, // II, A, 2, a, i 
  507.   tn->height.oid,
  508.   1,
  509.   0,
  510.   ipaddr());
  511. td->time_upd = Scheduler::instance().clock();
  513. sendUPD(td->index);
  514. }
  515. else {
  516. if(td->height.oid == ipaddr()) { // II, A, 2, a, ii
  517. double temp_tau = td->height.tau; // II, A, 2, a, ii, x 
  518. nsaddr_t temp_oid = td->height.oid;
  520. td->height.Null();
  521. td->num_down = 0;
  522. td->num_up = 0;
  524. /*
  525.  *  For every active link n, if the neighbor connected
  526.  *  via link n is the destination j, set HT_NEIGH[j][n]=ZERO
  527.  *  and LNK_STAT[j][n] = DN.
  528.  *  Otherwise, set HT_NEIGH[j][n] = NULL and LNK_STAT[j][n] = UN.
  529.  */
  530. for(tn = td->nblist.lh_first; tn; tn = tn->link.le_next) {
  531. if(tn->index == td->index) {
  532. tn->height.Zero();
  533. tn->lnk_stat = LINK_DN;
  535. }
  536. else {
  537. tn->height.Null();
  538. tn->lnk_stat = LINK_UN;
  539. }
  540. }
  542. sendCLR(td->index, temp_tau, temp_oid);
  543. }
  544. else {
  545. td->update_height(Scheduler::instance().clock(), // II, A, 2, a, ii, y
  546.   ipaddr(),
  547.   0,
  548.   0,
  549.   ipaddr());
  550. td->rt_req = 0;
  551. td->time_upd = Scheduler::instance().clock();
  553. #ifdef DEBUG
  554. // under what circumstances does this rule fire?
  555. // seems like it will prevent the detection of 
  556. // partitions...??? -dam 8/24/98
  558. if (logtarget) 
  559.   {
  560.     sprintf(logtarget->pt_->buffer(), "T %.9f _%d_ rule IIA2a(ii)x fires %d",
  561.     Scheduler::instance().clock(), ipaddr(), td->index);
  562.     logtarget->pt_->dump();
  563.   }
  564. #endif
  565. sendUPD(td->index);
  566. }
  567. }
  569. }
  570. else {
  571. TORANeighbor *n = td->nb_find_max_height(); // II, A, 2, b
  572. assert(n);
  573. TORANeighbor *m = td->nb_find_min_nonnull_height(&n->height);
  574. assert(m);
  576. td->update_height(m->height.tau,
  577.   m->height.oid,
  578.   m->height.r,
  579.   m->height.delta - 1,
  580.   ipaddr());
  582. td->time_upd = Scheduler::instance().clock();
  584. sendUPD(td->index);
  585. }
  586. }
  587. }
  588. else { // II, B
  590. }
  591. }
  594. /*
  595.  *  IETF Draft - TORA Specification, section 3.7.8
  596.  */
  597. void
  598. toraAgent::recvCLR(Packet *p)
  599. {
  600. struct hdr_ip *ih = HDR_IP(p);
  601. struct hdr_tora_clr *th = HDR_TORA_CLR(p);
  602. TORADest *td;
  603. TORANeighbor *tn;
  605. if(th->tc_dst == ipaddr()) {
  606. return;
  607. }
  609. td = dst_find(th->tc_dst);
  610. if(td == 0)
  611. td = dst_add(th->tc_dst);
  612. assert(td);
  614. if(td->height.tau == th->tc_tau &&
  615.    td->height.oid == th->tc_oid &&
  616.    td->height.r == 1) { // I
  617. double temp_tau = td->height.tau;
  618. nsaddr_t temp_oid = td->height.oid;
  620. td->height.Null();
  621. td->num_up = 0;
  622. td->num_down = 0;
  624. for(tn = td->nblist.lh_first; tn; tn = tn->link.le_next) {
  625. if(tn->index == td->index) {
  626. tn->height.Zero();
  627. tn->lnk_stat = LINK_DN;
  628. }
  629. else {
  630. tn->height.Null();
  631. tn->lnk_stat = LINK_UN;
  632. }
  633. }
  634. if(td->num_active > 1) { // I, A
  635. sendCLR(td->index, temp_tau, temp_oid);
  636. }
  637. else { // I, B
  638. }
  639. }
  640. else {
  641. tn = td->nb_find(ih->saddr()); // II
  642. if(tn == 0) {
  643. /*
  644.  *  XXX - update link status?
  645.  */
  646.         trace("T %.9f _%d_ received `CLR` from non-neighbor %d",
  647.                CURRENT_TIME, index, ih->saddr());
  648. #ifdef DEBUG
  649. fprintf(stderr,
  650. "node %d received `CLR` from non-neighbor %dn",
  651. index, ih->src_);
  652. #endif
  653. return;
  654. }
  656. tn->height.Null();
  657. tn->lnk_stat = LINK_UN;
  659. for(tn = td->nblist.lh_first; tn; tn = tn->link.le_next) {
  660. if(tn->height.tau == th->tc_tau &&
  661.    tn->height.oid == th->tc_oid &&
  662.    tn->height.r == 1) {
  663. tn->height.Null();
  664. tn->lnk_stat = LINK_UN;
  665.                         }
  666. }
  667. if(td->num_down == 0) { // II, A
  668. if(td->num_up == 0) { // II, A, 1
  669. if(td->height.isNull()) { // II, A, 1, a
  670. }
  671. else {
  672. td->height.Null();
  673. td->time_upd = Scheduler::instance().clock();
  674. sendUPD(td->index);
  675. }
  676. }
  677. else {
  678. td->update_height(Scheduler::instance().clock(),
  679.   ipaddr(),
  680.   0,
  681.   0,
  682.   ipaddr());
  683. td->rt_req = 0;
  684. td->time_upd = Scheduler::instance().clock();
  685. sendUPD(td->index);
  686. }
  687. }
  688. else {
  689. // II, B
  690. }
  691. }
  692. }
  694. void
  695. toraAgent::trace(char* fmt, ...)
  696. {
  697.   va_list ap;
  698.   
  699.   if (!logtarget) return;
  701.   va_start(ap, fmt);
  702.   vsprintf(logtarget->pt_->buffer(), fmt, ap);
  703.   logtarget->pt_->dump();
  704.   va_end(ap);
  705. }