src/routingTree.hpp

   1 /*
   2  *  anytun
   3  *
   4  *  The secure anycast tunneling protocol (satp) defines a protocol used
   5  *  for communication between any combination of unicast and anycast
   6  *  tunnel endpoints.  It has less protocol overhead than IPSec in Tunnel
   7  *  mode and allows tunneling of every ETHER TYPE protocol (e.g.
   8  *  ethernet, ip, arp ...). satp directly includes cryptography and
   9  *  message authentication based on the methodes used by SRTP.  It is
  10  *  intended to deliver a generic, scaleable and secure solution for
  11  *  tunneling and relaying of packets of any protocol.
  12  *
  13  *
  14  *  Copyright (C) 2007-2009 Othmar Gsenger, Erwin Nindl,
  15  *                          Christian Pointner <satp@wirdorange.org>
  16  *
  17  *  This file is part of Anytun.
  18  *
  19  *  Anytun is free software: you can redistribute it and/or modify
  20  *  it under the terms of the GNU General Public License as published by
  21  *  the Free Software Foundation, either version 3 of the License, or
  22  *  any later version.
  23  *
  24  *  Anytun is distributed in the hope that it will be useful,
  25  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  26  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  27  *  GNU General Public License for more details.
  28  *
  29  *  You should have received a copy of the GNU General Public License
  30  *  along with anytun.  If not, see <http://www.gnu.org/licenses/>.
  31  */
  32
  33 #ifndef ANYTUN_routingTree_hpp_INCLUDED
  34 #define ANYTUN_routingTree_hpp_INCLUDED
  35
  36 #include "anytunError.h"
  37
  38 class RoutingTree
  39 {
  40
  41 public:
  42   template <class BinaryType>
  43   static void walk(BinaryType bytes ,RoutingTreeNode* node,uint8_t length,uint16_t mux) {
  44     for(int i=0; i<(length/8); i++) {
  45       if(!node->nodes_[bytes[i]]) {
  46         node->nodes_[bytes[i]] = new RoutingTreeNode;
  47       }
  48       node=node->nodes_[bytes[i]];
  49     }
  50     if(length%8) {
  51       unsigned char idx=0xff;
  52       idx <<=8-(length%8);
  53       idx &= bytes[length/8];
  54       unsigned char maxidx=0xff;
  55       maxidx>>=(length%8);
  56       maxidx|=idx;
  57       for(unsigned char i=idx; i<=maxidx; i++) {
  58         if(!node->nodes_[i]) {
  59           node->nodes_[i] = new RoutingTreeNode;
  60         }
  61         node->nodes_[i]->valid_=true;
  62         node->nodes_[i]->mux_=mux;
  63       }
  64     } else {
  65       node->valid_=true;
  66       node->mux_=mux;
  67     }
  68   }
  69
  70   template <class BinaryType>
  71   static uint16_t find(BinaryType bytes ,RoutingTreeNode& root) {
  72     bool valid=0;
  73     uint16_t mux=0;
  74     RoutingTreeNode* node = &root;
  75     if(root.valid_) {
  76       mux=root.mux_;
  77       valid=1;
  78     }
  79     for(size_t level=0; level<bytes.size(); level++) {
  80       if(node->nodes_[bytes[level]]) {
  81         node=node->nodes_[bytes[level]];
  82         if(node->valid_) {
  83           mux=node->mux_;
  84           valid=1;
  85         }
  86       } else {
  87         break;
  88       }
  89     }
  90     if(!valid) {
  91       AnytunError::throwErr() << "no route";
  92     }
  93     return mux;
  94   }
  95
  96 };
  97
  98 #endif