src/seqWindow.cpp

   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 #include "threadUtils.hpp"
  34 #include "datatypes.h"
  35 #include <cstring>
  36 #include "seqWindow.h"
  37
  38 SeqWindowElement::SeqWindowElement()
  39 {
  40   window_ = NULL;
  41   pos_ = 0;
  42   max_ = 0;
  43 }
  44
  45 SeqWindowElement::~SeqWindowElement()
  46 {
  47   if(window_)
  48     delete[] window_;
  49 }
  50
  51 void SeqWindowElement::init(window_size_t w, seq_nr_t m)
  52 {
  53   if(window_)
  54     delete[] window_;
  55   window_ = new u_int8_t[w];
  56   memset(window_, 0, w);
  57   pos_ = 0;
  58   max_ = m;
  59   window_[pos_] = 1;
  60 }
  61
  62 SeqWindow::SeqWindow(window_size_t w) : window_size_(w)
  63 {
  64 }
  65
  66 SeqWindow::~SeqWindow()
  67 {
  68 }
  69
  70 bool SeqWindow::checkAndAdd(sender_id_t sender, seq_nr_t seq_nr)
  71 {
  72   Lock lock(mutex_);
  73   if (!window_size_)
  74     return false;
  75
  76   SenderMap::iterator s = sender_.find(sender);
  77   if(s == sender_.end()) {
  78     sender_[sender].init(window_size_, seq_nr);
  79     return false;
  80   }
  81
  82   int shifted = 0;
  83   if(s->second.max_ < window_size_) {
  84     s->second.max_ += SEQ_NR_MAX/2;
  85     seq_nr += SEQ_NR_MAX/2;
  86     shifted = 1;
  87   }
  88   else if(s->second.max_ > (SEQ_NR_MAX - window_size_)) {
  89     s->second.max_ -= SEQ_NR_MAX/2;
  90     seq_nr -= SEQ_NR_MAX/2;
  91     shifted = 2;
  92   }
  93
  94   seq_nr_t min = s->second.max_ - window_size_ + 1;
  95   if(seq_nr < min || seq_nr == s->second.max_) {
  96     if(shifted == 1)
  97       s->second.max_ -= SEQ_NR_MAX/2;
  98     else if(shifted == 2)
  99       s->second.max_ += SEQ_NR_MAX/2;
 100     return true;
 101   }
 102
 103   if(seq_nr > s->second.max_) {
 104     seq_nr_t diff = seq_nr - s->second.max_;
 105     if(diff >= window_size_)
 106       diff = window_size_;
 107
 108     window_size_t new_pos = s->second.pos_ + diff;
 109
 110     if(new_pos >= window_size_) {
 111       new_pos -= window_size_;
 112
 113       if(s->second.pos_ < window_size_ - 1)
 114         memset(&(s->second.window_[s->second.pos_ + 1]), 0, window_size_ - s->second.pos_ - 1);
 115
 116       memset(s->second.window_, 0, new_pos);
 117     }
 118     else {
 119       memset(&(s->second.window_[s->second.pos_ + 1]), 0, diff);
 120     }
 121     s->second.pos_ = new_pos;
 122     s->second.window_[s->second.pos_] = 1;
 123     s->second.max_ = seq_nr;
 124
 125     if(shifted == 1)
 126       s->second.max_ -= SEQ_NR_MAX/2;
 127     else if(shifted == 2)
 128       s->second.max_ += SEQ_NR_MAX/2;
 129
 130     return false;
 131   }
 132
 133   seq_nr_t diff = s->second.max_ - seq_nr;
 134   window_size_t pos = diff > s->second.pos_ ? s->second.pos_ + window_size_ : s->second.pos_;
 135   pos -= diff;
 136
 137   if(shifted == 1)
 138     s->second.max_ -= SEQ_NR_MAX/2;
 139   else if(shifted == 2)
 140     s->second.max_ += SEQ_NR_MAX/2;
 141
 142   int ret = s->second.window_[pos];
 143   s->second.window_[pos] = 1;
 144
 145   if(ret)
 146     return true;
 147
 148   return false;
 149 }
 150
 151 void SeqWindow::clear(sender_id_t sender)
 152 {
 153   Lock lock(mutex_);
 154   sender_.erase(sender);
 155 }
 156
 157 void SeqWindow::clear()
 158 {
 159   Lock lock(mutex_);
 160   sender_.clear();
 161 }