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

通讯编程

开发平台:

Visual C++

  1. /* -*-  Mode:C++; c-basic-offset:8; tab-width:8; indent-tabs-mode:t -*- */
  2. /*
  3.  * Copyright (c) 2000  International Computer Science Institute
  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 ACIRI, the AT&T 
  17.  *      Center for Internet Research at ICSI (the International Computer
  18.  *      Science Institute).
  19.  * 4. Neither the name of ACIRI nor of ICSI may be used
  20.  *    to endorse or promote products derived from this software without
  21.  *    specific prior written permission.
  22.  *
  23.  * THIS SOFTWARE IS PROVIDED BY ICSI AND CONTRIBUTORS ``AS IS'' AND
  24.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26.  * ARE DISCLAIMED.  IN NO EVENT SHALL ICSI OR CONTRIBUTORS BE LIABLE
  27.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  30.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33.  * SUCH DAMAGE.
  34.  *
  35.  */
  36. #ifndef ns_pushback_h
  37. #define ns_pushback_h
  38. #include "config.h"
  39. #include "packet.h"
  40. #include "ip.h"
  41. #include "agent.h"
  42. #include "address.h"
  43. #include "node.h"
  44. #include "route.h"
  45. #include "timer-handler.h"
  46. #include "pushback-constants.h"
  47. #include "pushback-message.h"
  48. class PushbackQueue;
  49. class PushbackEventList;
  50. class IdentStruct;
  51. class RateLimitSession;
  52. struct hdr_pushback {
  53.   
  54.   PushbackMessage * msg_;
  55.   
  56.   static int offset_;
  57.   inline static int& offset() {
  58.     return offset_;
  59.   }
  60.   inline static hdr_pushback * access(Packet *p) {
  61.     return (hdr_pushback *) p->access(offset_);
  62.   }
  63. };
  64. // a structure to store details of each link on the node.
  65. struct queue_rec {
  66.   PushbackQueue * pbq_;             //pointer to the queue object
  67.   IdentStruct * idTree_;              // this queues prefix tree
  68.   // other required variables go here
  69. };
  70. class PushbackEvent;
  71. class PushbackTimer;
  72.   
  73. class PushbackAgent : public Agent {
  74.  public:
  75.   PushbackAgent();
  76.   int command(int argc, const char*const* argv);
  77.   void reportDrop(int qid, Packet * p);
  78.   void timeout(PushbackEvent * event);
  79.   void identifyAggregate(int qid, double arrRate, double linkBW);
  80.   void resetDropLog(int qid);
  81.   void recv(Packet *p, Handler *);
  82.   void calculateLowerBound(int qid, double arrRate);
  83.   int last_index_;
  84.   int verbose_;
  85.   int intResult_;
  86.   int debugLevel;
  87.   char prnMsg[500];  //hopefully long enough
  88.   
  89.   static int mergerAccept(int count, int bits, int bitsDiff);
  90.   void printMsg(char *msg, int msgLevel);
  91.   Node * node_;
  92. protected:
  93.   int enable_pushback_;
  94.   queue_rec queue_list_[MAX_QUEUES];
  95.   double requiredLimit_;
  96.   
  97.   RouteLogic * rtLogic_;
  98.   PushbackTimer * timer_;
  99.   
  100.   void initialUpdate(RateLimitSession * rls);
  101.   void pushbackCheck(RateLimitSession * rls);
  102.   void pushbackStatus(RateLimitSession* rls);
  103.   void pushbackRefresh(int qid);
  104.   void pushbackCancel(RateLimitSession* rls);
  105.   void processPushbackRequest(PushbackRequestMessage * msg);
  106.   void processPushbackStatus(PushbackStatusMessage *msg);
  107.   void processPushbackRefresh(PushbackRefreshMessage *msg);
  108.   void processPushbackCancel(PushbackCancelMessage *msg);
  109.   void refreshUpstreamLimits(RateLimitSession * rls);
  110.   int getQID(int sender);
  111.   int checkQID(int qid);
  112.   void sendMsg(PushbackMessage * msg);
  113. };
  114. class PushbackEvent {
  115.  public:
  116.   double time_;
  117.   int eventID_;
  118.   RateLimitSession * rls_;
  119.   int qid_;
  120.   
  121.   PushbackEvent * next_;
  122.   
  123.   PushbackEvent(double delay, int eventID, RateLimitSession * rls) {
  124.     time_ = Scheduler::instance().clock() + delay;
  125.     eventID_ = eventID;
  126.     rls_ = rls;
  127.     qid_= -1; //rls->localQID_;
  128.     next_=NULL;
  129.   }
  130.   PushbackEvent(double delay, int eventID, int qid) {
  131.     time_ = Scheduler::instance().clock() + delay;
  132.     eventID_ = eventID;
  133.     rls_=NULL;
  134.     qid_=qid;
  135.     next_=NULL;
  136.   }
  137.   
  138.   void setSucc(PushbackEvent * event) {
  139.     next_=event;
  140.   }
  141.   static char * type(PushbackEvent * event) {
  142.     switch (event->eventID_) {
  143.     case PUSHBACK_CHECK_EVENT: return "CHECK";
  144.     case PUSHBACK_REFRESH_EVENT: return "REFRESH";
  145.     case PUSHBACK_STATUS_EVENT: return "STATUS";
  146.     case INITIAL_UPDATE_EVENT: return "INITIAL UPDATE";
  147.     default: return "UNKNOWN";
  148.     }
  149.   }
  150. };
  151. class PushbackTimer: public TimerHandler {
  152.   
  153.  public: 
  154.   PushbackEvent * firstEvent_;
  155.   
  156.   PushbackTimer(PushbackAgent * agent): firstEvent_(NULL) { agent_ = agent;}
  157.   void insert(PushbackEvent * event);
  158.   void cancelStatus(RateLimitSession * rls);
  159.   void removeEvents(RateLimitSession * rls);
  160.   int containsRefresh(int qid);
  161.  protected:
  162.   PushbackAgent * agent_;
  163.   virtual void expire(Event *e);
  164.   void schedule();
  165.   void sanityCheck();
  166. };
  167. #endif