queue.cc
上传用户:rrhhcc
上传日期:2015-12-11
资源大小:54129k
文件大小:6k
源码类别:

通讯编程

开发平台:

Visual C++

  1. /* -*- Mode:C++; c-basic-offset:8; tab-width:8; indent-tabs-mode:t -*- */
  2. /*
  3.  * Copyright (c) 1996-1997 The 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 Network Research
  17.  *  Group at Lawrence Berkeley National 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. #ifndef lint
  35. static const char rcsid[] =
  36.     "@(#) $Header: /cvsroot/nsnam/ns-2/queue/queue.cc,v 1.29 2004/10/28 01:22:48 sfloyd Exp $ (LBL)";
  37. #endif
  38. #include "queue.h"
  39. #include <math.h>
  40. #include <stdio.h>
  41. void PacketQueue::remove(Packet* target)
  42. {
  43. for (Packet *pp= 0, *p= head_; p; pp= p, p= p->next_) {
  44. if (p == target) {
  45. if (!pp) deque();
  46. else {
  47. if (p == tail_) 
  48. tail_= pp;
  49. pp->next_= p->next_;
  50. --len_;
  51. bytes_ -= hdr_cmn::access(p)->size();
  52. }
  53. return;
  54. }
  55. }
  56. fprintf(stderr, "PacketQueue:: remove() couldn't find targetn");
  57. abort();
  58. }
  59. /*
  60.  * Remove packet pkt located after packet prev on the queue.  Either p or prev
  61.  * could be NULL.  If prev is NULL then pkt must be the head of the queue.
  62.  */
  63. void PacketQueue::remove(Packet* pkt, Packet *prev) //XXX: screwy
  64. {
  65. if (pkt) {
  66. if (head_ == pkt)
  67. PacketQueue::deque(); /* decrements len_ internally */
  68. else {
  69. prev->next_ = pkt->next_;
  70. if (tail_ == pkt)
  71. tail_ = prev;
  72. --len_;
  73. bytes_ -= hdr_cmn::access(pkt)->size();
  74. }
  75. }
  76. return;
  77. }
  78. void QueueHandler::handle(Event*)
  79. {
  80. queue_.resume();
  81. }
  82. Queue::~Queue() {
  83. }
  84. Queue::Queue() : Connector(), blocked_(0), unblock_on_resume_(1), qh_(*this),
  85.  pq_(0), 
  86.  last_change_(0), /* temporarily NULL */
  87.  old_util_(0), period_begin_(0), cur_util_(0), buf_slot_(0),
  88.  util_buf_(NULL)
  89. {
  90. bind("limit_", &qlim_);
  91. bind("util_weight_", &util_weight_);
  92. bind_bool("blocked_", &blocked_);
  93. bind_bool("unblock_on_resume_", &unblock_on_resume_);
  94. bind("util_check_intv_", &util_check_intv_);
  95. bind("util_records_", &util_records_);
  96. if (util_records_ > 0) {
  97. util_buf_ = new double[util_records_];
  98. if (util_buf_ == NULL) {
  99. printf("Error allocating util_bufs!");
  100. util_records_ = 0;
  101. }
  102. for (int i = 0; i < util_records_; i++) {
  103. util_buf_[i] = 0;
  104. }
  105. }
  106. }
  107. void Queue::recv(Packet* p, Handler*)
  108. {
  109. double now = Scheduler::instance().clock();
  110. enque(p);
  111. if (!blocked_) {
  112. /*
  113.  * We're not blocked.  Get a packet and send it on.
  114.  * We perform an extra check because the queue
  115.  * might drop the packet even if it was
  116.  * previously empty!  (e.g., RED can do this.)
  117.  */
  118. p = deque();
  119. if (p != 0) {
  120. utilUpdate(last_change_, now, blocked_);
  121. last_change_ = now;
  122. blocked_ = 1;
  123. target_->recv(p, &qh_);
  124. }
  125. }
  126. }
  127. void Queue::utilUpdate(double int_begin, double int_end, int link_state) {
  128. double decay;
  129. decay = exp(-util_weight_ * (int_end - int_begin));
  130. old_util_ = link_state + (old_util_ - link_state) * decay;
  131. // PS: measuring peak utilization
  132. if (util_records_ == 0)
  133. return; // We don't track peak utilization
  134. double intv = int_end - int_begin;
  135. double tot_intv = int_begin - period_begin_;
  136. if (intv || tot_intv) {
  137. int guard = 0; // for protecting against long while loops 
  138. cur_util_ = (link_state * intv + cur_util_ * tot_intv) /
  139. (intv + tot_intv);
  140. while (tot_intv + intv > util_check_intv_ &&
  141.        guard++ < util_records_) {
  142. period_begin_ = int_end;
  143. util_buf_[buf_slot_] = cur_util_;
  144. buf_slot_ = (buf_slot_ + 1) % util_records_;
  145. cur_util_ = link_state;
  146. intv -= util_check_intv_;
  147. }
  148. }
  149. }
  150. double Queue::utilization(void) 
  151. {
  152. double now = Scheduler::instance().clock();
  153. utilUpdate(last_change_, now, blocked_);
  154. last_change_ = now;
  155. return old_util_;
  156. }
  157. double Queue::peak_utilization(void)
  158. {
  159. double now = Scheduler::instance().clock();
  160. double peak = 0;
  161. int i;
  162. // PS: if peak_utilization tracking is disabled,
  163. // return the weighed avg instead
  164. if (util_records_ == 0)
  165. return utilization();
  166. utilUpdate(last_change_, now, blocked_);
  167. last_change_ = now;
  168. for (i = 0; i < util_records_; i++) {
  169. if (util_buf_[i] > peak)
  170. peak = util_buf_[i];
  171. }
  172. return peak;
  173. }
  174. void Queue::updateStats(int queuesize)
  175. {
  176.         double now = Scheduler::instance().clock();
  177.         double newtime = now - total_time_;
  178.         if (newtime > 0.0) {
  179.                 double oldave = true_ave_;
  180.                 double oldtime = total_time_;
  181.                 double newtime = now - total_time_;
  182.                 true_ave_ = (oldtime * oldave + newtime * queuesize) /now;
  183.                 total_time_ = now;
  184.         }
  185. }
  186. void Queue::resume()
  187. {
  188. double now = Scheduler::instance().clock();
  189. Packet* p = deque();
  190. if (p != 0) {
  191. target_->recv(p, &qh_);
  192. } else {
  193. if (unblock_on_resume_) {
  194. utilUpdate(last_change_, now, blocked_);
  195. last_change_ = now;
  196. blocked_ = 0;
  197. }
  198. else {
  199. utilUpdate(last_change_, now, blocked_);
  200. last_change_ = now;
  201. blocked_ = 1;
  202. }
  203. }
  204. }
  205. void Queue::reset()
  206. {
  207. Packet* p;
  208. total_time_ = 0.0;
  209. true_ave_ = 0.0;
  210. while ((p = deque()) != 0)
  211. drop(p);
  212. }