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

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

通讯编程

开发平台：

Visual C++

dsr-priqueue.cc：源码内容
  1. /* -*- c++ -*-
  2.    priqueue.cc
  3.    
  4.    A simple priority queue with a remove packet function
  5.    $Id: dsr-priqueue.cc,v 1.2 2002/09/11 20:11:09 buchheim Exp $
  6.    */
  8. #include <object.h>
  9. #include <packet.h>
  10. #include <cmu-trace.h>
  11. #include <dsr-priqueue.h>
  13. #define PRIQUEUE_DEBUG 0
  15. static class CMUPriQueueClass : public TclClass {
  16. public:
  17.   CMUPriQueueClass() : TclClass("CMUPriQueue") {}
  18.   TclObject* create(int, const char*const*) {
  19.     return (new CMUPriQueue);
  20.   }
  21. } class_CMUPriQueue;
  23. /* ======================================================================
  24.    Priority Queue Handler
  25.    ====================================================================== */
  26. void CMUPriQueueHandler::handle(Event*)
  27. {
  28.         qh_ifq->prq_resume();
  29. }       
  31. /* ======================================================================
  32.    CMUPriQueue - public routines
  33.    ====================================================================== */
  34. CMUPriQueue::CMUPriQueue() : Connector(), prq_qh_(this)
  35. {
  36. int i;
  38. for(i = 0; i < IFQ_MAX; i++) {
  39. prq_snd_[i].ifq_head = prq_snd_[i].ifq_tail = 0;
  40. prq_snd_[i].ifq_len = 0;
  41. prq_snd_[i].ifq_maxlen = IFQ_MAXLEN;
  42. prq_snd_[i].ifq_drops = 0;
  44. last_ifqlen_[i] = 0;
  45. }
  46. prq_logtarget_ = 0; // no logging target by default
  47. prq_ipaddr_ = 0;
  48. prq_blocked_ = 0;
  50. bind("qlen_logthresh_", &qlen_logthresh_);
  51. bind("fw_logthresh_", &fw_logthresh_);
  52. assert(qlen_logthresh_ > 0 && fw_logthresh_ > 0);
  54. bzero(last_ifqlen_, sizeof(last_ifqlen_));
  55. stat_send_ = stat_recv_ = stat_blocked_ = 0;
  56. }
  58. int
  59. CMUPriQueue::command(int argc, const char*const* argv)
  60. {
  61. if (argc == 2 && strcasecmp(argv[1], "reset") == 0) {
  62. Terminate();
  63. //FALL-THROUGH to give parents a chance to reset
  64. } else if(argc == 3) {
  65. if(strcmp(argv[1], "logtarget") == 0) {
  66. prq_logtarget_ = (Trace*) TclObject::lookup(argv[2]);
  67. assert(prq_logtarget_);
  68. return (TCL_OK);
  69. }
  70. else if(strcmp(argv[1], "ipaddr") == 0) {
  71. prq_ipaddr_ = atoi(argv[2]);
  72. assert(prq_ipaddr_ > 0);
  73. return (TCL_OK);
  74. }
  75. }
  76. return Connector::command(argc, argv);
  77. }
  79. void
  80. CMUPriQueue::log_stats()
  81. {
  82. int q, s = 0, d = 0;
  84. assert(IFQ_MAX == 4);
  86. /*
  87.  * avoid logging useless information
  88.  */
  89. for(q = 0; q < IFQ_MAX; q++) {
  90. s += prq_snd_[q].ifq_len;
  91. d += (prq_snd_[q].ifq_len - last_ifqlen_[q]);
  92. }
  93. if(d < 0)
  94. d = -d;
  96. /*
  97.  * Only write an entry to the log file is the queue length
  98.  * or "delta change" over the last second meets the threshold
  99.  * level OR, if this node is generally busy (forwarding more
  100.  * than a certain number of packets per second).
  101.  */
  102. if(s >= qlen_logthresh_ || d >= qlen_logthresh_ ||
  103.    stat_recv_ > fw_logthresh_) {
  104. if (prq_logtarget_->pt_->tagged()) {
  105.    // using the new tagged trace format
  106.    trace("I "TIME_FORMAT" -prq:adr %d -prq:rx %d -prq:tx %d -prq:bl %d -prq:snd0 {%d %d} -prq:snd1 {%d %d} -prq:snd2 {%d %d} -prq:snd3 {%d %d}",
  107.       Scheduler::instance().clock(),
  108.       prq_ipaddr_,
  109.       stat_recv_, stat_send_, stat_blocked_,
  110.       prq_snd_[0].ifq_len, prq_snd_[0].ifq_len-last_ifqlen_[0],
  111.       prq_snd_[1].ifq_len, prq_snd_[1].ifq_len-last_ifqlen_[1],
  112.       prq_snd_[2].ifq_len, prq_snd_[2].ifq_len-last_ifqlen_[2],
  113.       prq_snd_[3].ifq_len, prq_snd_[3].ifq_len-last_ifqlen_[3]);
  114. } else {
  115.    trace("I %.9f _%d_ ifq rx %d tx %d bl %d [%d %d] [%d %d] [%d %d] [%d %d]",
  116.       Scheduler::instance().clock(),
  117.       prq_ipaddr_,
  118.       stat_recv_, stat_send_, stat_blocked_,
  119.       prq_snd_[0].ifq_len, prq_snd_[0].ifq_len-last_ifqlen_[0],
  120.       prq_snd_[1].ifq_len, prq_snd_[1].ifq_len-last_ifqlen_[1],
  121.       prq_snd_[2].ifq_len, prq_snd_[2].ifq_len-last_ifqlen_[2],
  122.       prq_snd_[3].ifq_len, prq_snd_[3].ifq_len-last_ifqlen_[3]);
  123. }
  124. }
  126. for(q = 0; q < IFQ_MAX; q++) {
  127. last_ifqlen_[q] = prq_snd_[q].ifq_len;
  128. }
  129. stat_send_ = stat_recv_ = stat_blocked_ = 0;
  130. }
  132. void
  133. CMUPriQueue::trace(char* fmt, ...)
  134. {
  135. va_list ap;
  136.   
  137. assert(prq_logtarget_);
  139. va_start(ap, fmt);
  140. vsprintf(prq_logtarget_->pt_->buffer(), fmt, ap);
  141. prq_logtarget_->pt_->dump();
  142. va_end(ap);
  143. }
  145. void
  146. CMUPriQueue::recv(Packet *p, Handler *)
  147. {
  148. #if PRIQUEUE_DEBUG > 0
  149. prq_validate();
  150. #endif
  151. stat_recv_++;
  152. if(prq_blocked_ == 1) {
  153. stat_blocked_++;
  154. } else {
  155. assert(prq_length() == 0);
  156. }
  157. prq_enqueue(p);
  158. #if PRIQUEUE_DEBUG > 0
  159. prq_validate();
  160. #endif
  161. }
  163. void
  164. CMUPriQueue::prq_resume()
  165. {
  166. Packet *p;
  168. assert(prq_blocked_);
  169. #if PRIQUEUE_DEBUG > 0
  170. prq_validate();
  171. #endif
  172. p = prq_dequeue();
  173. if (p != 0) {
  174. stat_send_++;
  175.                 target_->recv(p, &prq_qh_);
  176.         } else {
  177. prq_blocked_ = 0;
  178.         }
  179. }
  182. /*
  183.  * Called at the end of the simulation to purge the IFQ.
  184.  */
  185. void
  186. CMUPriQueue::Terminate()
  187. {
  188. Packet *p;
  189. while((p = prq_dequeue())) {
  190. drop(p, DROP_END_OF_SIMULATION);
  191. }
  192. }
  194. Packet*
  195. CMUPriQueue::prq_get_nexthop(nsaddr_t id)
  196. {
  197. int q;
  198. Packet *p, *pprev = 0;
  199. struct ifqueue *ifq;
  201. #if PRIQUEUE_DEBUG > 0
  202. prq_validate();
  203. #endif
  204. for(q = 0; q < IFQ_MAX; q++) {
  205. ifq = &prq_snd_[q];
  206. pprev = 0;
  207. for(p = ifq->ifq_head; p; p = p->next_) {
  208. struct hdr_cmn *ch = HDR_CMN(p);
  210. if(ch->next_hop() == id)
  211. break;
  212. pprev = p;
  213. }
  215. if(p) {
  216. if(p == ifq->ifq_head) {
  217. assert(pprev == 0);
  219. IF_DEQUEUE(ifq, p);
  220. /* don't increment drop counter */
  221. #if PRIQUEUE_DEBUG > 0
  222. prq_validate();
  223. #endif
  224. return p;
  225. } else {
  226. assert(pprev);
  227. pprev->next_ = p->next_;
  229. if(p == ifq->ifq_tail)
  230. ifq->ifq_tail = pprev;
  231. ifq->ifq_len--;
  233. #if PRIQUEUE_DEBUG > 0
  234. prq_validate();
  235. #endif
  236. p->next_ = 0;
  237. return p;
  238. }
  239. }
  240.      }
  242. return (Packet*) 0;
  243. }
  245. int
  246. CMUPriQueue::prq_isfull(Packet *p)
  247. {
  248. int q = prq_assign_queue(p);
  249. struct ifqueue *ifq = &prq_snd_[q];
  251. if(IF_QFULL(ifq))
  252. return 1;
  253. else
  254. return 0;
  255. }
  257. int
  258. CMUPriQueue::prq_length()
  259. {
  260. int q, tlen = 0;
  262. for(q = 0; q < IFQ_MAX; q++) {
  263. tlen += prq_snd_[q].ifq_len;
  264. }
  266. return tlen;
  267. }
  269. /* ======================================================================
  270.    CMUPriQueue - private routines
  271.    ====================================================================== */
  272. void
  273. CMUPriQueue::prq_enqueue(Packet *p)
  274. {
  275. int q = prq_assign_queue(p);
  276. struct ifqueue *ifq = &prq_snd_[q];
  278. if(IF_QFULL(ifq)) {
  279. IF_DROP(ifq);
  280. drop(p, DROP_IFQ_QFULL);
  281. return;
  282. }
  283. IF_ENQUEUE(ifq, p);
  285. /*
  286.  * Start queue if idle...
  287.  */
  288. if(prq_blocked_ == 0) {
  289. p = prq_dequeue();
  290. prq_blocked_ = 1;
  292. stat_send_++;
  293. target_->recv(p, &prq_qh_);
  294. }
  295. }
  297. Packet*
  298. CMUPriQueue::prq_dequeue(void)
  299. {
  300. Packet *p;
  301. int q;
  303. for(q = 0; q < IFQ_MAX; q++) {
  304. if(prq_snd_[q].ifq_len > 0) {
  305. IF_DEQUEUE(&prq_snd_[q], p);
  306. return p;
  307. } else {
  308. assert(prq_snd_[q].ifq_head == 0);
  309. }
  310. }
  311. return (Packet*) 0;
  312. }
  314. int
  315. CMUPriQueue::prq_assign_queue(Packet *p)
  316. {
  317.         struct hdr_cmn *ch = HDR_CMN(p);
  319. switch(ch->ptype()) {
  320. case PT_AODV:
  321. case PT_DSR:
  322. case PT_IMEP:
  323. case PT_MESSAGE: /* used by DSDV */
  324. case PT_TORA:
  325. return IFQ_RTPROTO;
  327. case PT_AUDIO:
  328. case PT_VIDEO:
  329. return IFQ_REALTIME;
  331. case PT_ACK:
  332. return IFQ_LOWDELAY;
  334. default:
  335. return IFQ_NORMAL;
  336. }
  337. }
  339. void
  340. CMUPriQueue::prq_validate()
  341. {
  342. int q, qlen;
  343. Packet *p;
  344. struct ifqueue *ifq;
  346. for(q = 0; q < IFQ_MAX; q++) {
  347. ifq = &prq_snd_[q];
  348. qlen = 0;
  350. if(ifq->ifq_head == 0) {
  351. assert(ifq->ifq_len == 0);
  352. assert(ifq->ifq_head == ifq->ifq_tail);
  353. } else {
  355. for(p = ifq->ifq_head; p; p = p->next_) {
  356. if(p->next_ == 0)
  357. assert(p == ifq->ifq_tail);
  358. qlen++;
  359. }
  361. assert(qlen == ifq->ifq_len);
  362. assert(qlen <= ifq->ifq_maxlen);
  363. }
  364. }
  365. }