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

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

通讯编程

开发平台：

Visual C++

ident-tree.h：源码内容
  1. #ifndef ident_tree_h
  2. #define ident_tree_h
  4. #include <stdlib.h>
  5. #include <stdio.h>
  6. #include "address.h"
  7. #include "packet.h"
  8. #include "ip.h"
  9. #include "tcl.h"
  10. #include "math.h"
  12. #include "pushback-constants.h"
  14. class AggSpec;
  16. struct cluster {
  17.   int prefix_;
  18.   int bits_;
  19.   int count_;
  20. };
  22. class AggReturn {
  23.  public:
  24.   cluster * clusterList_;
  25.   double limit_;
  26.   int finalIndex_;
  27.   int totalCount_;
  29.   AggReturn(cluster * clusterList, double bottom, int finalIndex, int totalCount) {
  30.     clusterList_ = clusterList;
  31.     limit_ = bottom;
  32.     finalIndex_ = finalIndex;
  33.     totalCount_=totalCount;
  34.   }
  36.   ~AggReturn() {
  37.     free(clusterList_);
  38.   }
  39. };
  41. class DropHashTable {
  43.  public:
  44.   Tcl_HashTable * hashTable_;
  45.   int count_;
  46.   char key[200];   //generous space for key
  48.   DropHashTable():count_(0) {
  49.     hashTable_ = new Tcl_HashTable;
  50.     Tcl_InitHashTable(hashTable_, TCL_STRING_KEYS);
  51.   }
  52.   
  53.   void 
  54.   packetToKey(Packet *p) {
  55.     hdr_ip * iph = hdr_ip::access(p);
  56.     ns_addr_t src = iph->src();
  57.     ns_addr_t dst = iph->dst();
  58.     int fid = iph->flowid();
  59.     sprintf(key,"%d.%d %d.%d %d",src.addr_, src.port_, dst.addr_, dst.port_, fid);
  60.   }
  61.     
  62.   void 
  63.   insert(Packet *p, int size) {
  64.     count_+=size;
  66.     int new_entry;
  67.     long oldValue;
  68.     packetToKey(p);
  69.     Tcl_HashEntry* he = Tcl_CreateHashEntry(hashTable_, (char*) key, &new_entry);
  70.     if (new_entry) 
  71.       oldValue = 0;
  72.     else 
  73.       oldValue = (long)Tcl_GetHashValue(he);
  74.     
  75.     //printf("old value = %d", oldValue);
  76.     oldValue+=size;
  77.     Tcl_SetHashValue(he, oldValue);
  78.   }
  79.    
  80.   void 
  81.   traverse() {
  82.     printf("DropHash count = %dn", count_);
  83.     Tcl_HashSearch searchPtr;
  84.     Tcl_HashEntry * he = Tcl_FirstHashEntry(hashTable_, &searchPtr);
  85.     while (he != NULL) {
  86.       char * key = Tcl_GetHashKey(hashTable_, he);
  87.       long value = (long)Tcl_GetHashValue(he);
  88.       printf("%s = %ldn", key, value);
  89.       he = Tcl_NextHashEntry(&searchPtr);
  90.     }
  91.   }
  92.    
  93.   void
  94.   reset() {
  95.     count_=0;
  96.     Tcl_DeleteHashTable(hashTable_);
  97.     Tcl_InitHashTable(hashTable_, TCL_STRING_KEYS);
  98.   }
  100. };
  102.   
  103. class PrefixTree {
  105.  public:
  106.   int countArray[(1 << (NO_BITS+1)) - 1];
  107.     
  108.   PrefixTree();
  109.   void reset();
  110.   void traverse();
  111.   void registerDrop(int address, int size);
  112.   
  113.   static int getLastIndex() {return (1 << (NO_BITS+1)) - 2;}
  114.   static int getMaxAddress();
  115.   static int getBitI(int address, int i);
  116.   static int getIndexFromPrefix(int prefix, int noBits);
  117.   static int getIndexFromAddress(int address, int noBits);
  118.   static int getPrefixFromIndex(int index);
  119.   static int getNoBitsFromIndex(int index);
  120.   static int getFirstIndexOfBit(int noBits);
  121.   static int getLastIndexOfBit(int noBits);
  122.   static int getPrefixBits(int prefix, int noBits);
  124.   AggReturn * identifyAggregate(double arrRate, double linkBW);
  125.   void insertCluster(cluster * clusterList, int index, int count, int noBits);
  126.   void goDownCluster(cluster * clusterList, int index);
  127.   void sortCluster(cluster * clusterList, int lastIndex);
  128.   void swapCluster(cluster * clusterList, int id1, int id2);
  130.   AggReturn * calculateLowerBound(); 
  131. };
  133. class IdentStruct {
  135.  public:
  136.   PrefixTree * dstTree_;
  137.   PrefixTree * srcTree_;
  138.   DropHashTable* dropHash_;
  139.   
  140.   double lowerBound_;
  141.   void setLowerBound(double newBound, int averageIt);
  142.   AggReturn * calculateLowerBound();
  143.   
  144.   IdentStruct();
  145.   
  146.   void reset();
  147.   void traverse();
  148.   
  149.   void registerDrop(Packet *);
  150.   
  151.   static unsigned int getMaxAddress();
  152.   static int getBitI(unsigned int, int);
  153.  
  154.   AggReturn * identifyAggregate(double arrRate, double linkBW);
  155.   
  156. };
  158. #endif