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