classifier-addr.cc
上传用户: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) 1996-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 MASH Research
  17.  *  Group at the University of California Berkeley.
  18.  * 4. Neither the name of the University nor of the Research Group may be
  19.  *    used 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/classifier/classifier-addr.cc,v 1.28 2006/02/21 15:20:17 mahrenho Exp $";
  37. #endif
  38. #include "classifier-addr.h"
  39. int AddressClassifier::classify(Packet *p) {
  40. hdr_ip* iph = hdr_ip::access(p);
  41. return mshift(iph->daddr());
  42. }
  43. static class AddressClassifierClass : public TclClass {
  44. public:
  45. AddressClassifierClass() : TclClass("Classifier/Addr") {}
  46. TclObject* create(int, const char*const*) {
  47. return (new AddressClassifier());
  48. }
  49. } class_address_classifier;
  50. /* added for mobileip code  Ya, 2/99*/
  51.  
  52. static class ReserveAddressClassifierClass : public TclClass {
  53. public:
  54.         ReserveAddressClassifierClass() : TclClass("Classifier/Addr/Reserve") {}
  55.         TclObject* create(int, const char*const*) {
  56.                 return (new ReserveAddressClassifier());
  57.         }
  58. } class_reserve_address_classifier;
  59.  
  60. int ReserveAddressClassifier::command(int argc, const char*const* argv)
  61. {
  62.         // Tcl& tcl = Tcl::instance();
  63.         if (argc == 3 && strcmp(argv[1],"reserve-port") == 0) {
  64.                 reserved_ = atoi(argv[2]);
  65.                 alloc((maxslot_ = reserved_ - 1));
  66.                 return(TCL_OK);
  67.         }
  68.         return (AddressClassifier::command(argc, argv));
  69. }
  70.  
  71. void ReserveAddressClassifier::clear(int slot)
  72. {
  73.         slot_[slot] = 0;
  74.         if (slot == maxslot_) {
  75.                 while (--maxslot_ >= reserved_ && slot_[maxslot_] == 0)
  76.                         ;
  77.         }
  78. }
  79.  
  80. int ReserveAddressClassifier::classify(Packet *p) {
  81. hdr_ip* iph = hdr_ip::access(p);
  82. return iph->dport();
  83. }
  84. int ReserveAddressClassifier::getnxt(NsObject *nullagent)
  85. {
  86.         int i;
  87.         for (i=reserved_; i < nslot_; i++)
  88.                 if (slot_[i]==0 || slot_[i]==nullagent)
  89.                         return i;
  90.         i=nslot_;
  91.         alloc(nslot_); 
  92.         return i;
  93. }
  94. static class BcastAddressClassifierClass : public TclClass {
  95. public:
  96.         BcastAddressClassifierClass() : TclClass("Classifier/Hash/Dest/Bcast") {}
  97.         TclObject* create(int, const char*const*) { 
  98.                 return (new BcastAddressClassifier());
  99.         }
  100. } class_bcast_address_classifier;
  101.  
  102. NsObject* BcastAddressClassifier::find(Packet* p)
  103. {
  104.         NsObject* node = NULL;
  105.         int cl = classify(p);
  106.         if (cl < 0 || cl >= nslot_ || (node = slot_[cl]) == 0) {
  107.                 if (cl == BCAST_ADDR_MASK) {
  108.                         // limited broadcast; assuming no such packet
  109.                         // would be delivered back to sender
  110.                         return bcast_recver_;
  111.                 }
  112. if (default_target_) 
  113. return default_target_;
  114.  
  115.                 /*
  116.                  * Sigh.  Can't pass the pkt out to tcl because it's
  117.                  * not an object.
  118.                  */
  119.                 Tcl::instance().evalf("%s no-slot %d", name(), cl);
  120.                 /*
  121.                  * Try again.  Maybe callback patched up the table.
  122.                  */
  123.                 cl = classify(p);
  124.                 if (cl < 0 || cl >= nslot_ || (node = slot_[cl]) == 0)
  125.                         return (NULL);
  126.         }
  127.  
  128.         return (node);
  129. }
  130.  
  131. int BcastAddressClassifier::command(int argc, const char*const* argv)
  132. {
  133.         // Tcl& tcl = Tcl::instance();
  134.         if (argc == 3 && strcmp(argv[1],"bcast-receiver") == 0) {
  135.                 bcast_recver_ = (NsObject*)TclObject::lookup(argv[2]);
  136.                 return(TCL_OK);
  137.         }
  138.         return (AddressClassifier::command(argc, argv));
  139. }