rtpSessionTable.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 anytun.org <satp@wirdorange.org>
  15  *
  16  *  This program is free software; you can redistribute it and/or modify
  17  *  it under the terms of the GNU General Public License version 2
  18  *  as published by the Free Software Foundation.
  19  *
  20  *  This program is distributed in the hope that it will be useful,
  21  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  22  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  23  *  GNU General Public License for more details.
  24  *
  25  *  You should have received a copy of the GNU General Public License
  26  *  along with this program (see the file COPYING included with this
  27  *  distribution); if not, write to the Free Software Foundation, Inc.,
  28  *  59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  29  */
  30 #include "threadUtils.hpp"
  31 #include "datatypes.h"
  32
  33 #include "rtpSessionTable.h"
  34
  35 RtpSessionTable* RtpSessionTable::inst = NULL;
  36 Mutex RtpSessionTable::instMutex;
  37 RtpSessionTable& gRtpSessionTable = RtpSessionTable::instance();
  38
  39
  40 RtpSessionTable& RtpSessionTable::instance()
  41 {
  42   Lock lock(instMutex);
  43   static instanceCleaner c;
  44   if(!inst)
  45     inst = new RtpSessionTable();
  46
  47   return *inst;
  48 }
  49
  50 RtpSessionTable::RtpSessionTable()
  51 {
  52 }
  53
  54 RtpSessionTable::~RtpSessionTable()
  55 {
  56 }
  57
  58 void RtpSessionTable::addSession(const std::string & pref, const RtpSession & ses )
  59 {
  60   Lock lock(mutex_);
  61
  62
  63   std::pair<RtpSessionMap::iterator, bool> ret = map_.insert(RtpSessionMap::value_type(pref,ses));
  64   if(!ret.second)
  65   {
  66     map_.erase(ret.first);
  67     map_.insert(RtpSessionMap::value_type(pref,ses));
  68   }
  69 }
  70
  71
  72 void RtpSessionTable::delSession(const std::string & pref )
  73 {
  74   Lock lock(mutex_);
  75
  76   map_.erase(map_.find(pref));
  77 }
  78
  79 /*u_int16_t  RtpSessionTable::getRtpSession(const std::string & addr)
  80 {
  81         Lock lock(mutex_);
  82         if (map_.empty())
  83         return 0;
  84         NetworkPrefix prefix(addr,32);
  85         //TODO Routing algorithem isnt working!!!
  86         RoutingMap::iterator it = map_.lower_bound(prefix);
  87 //      it--;
  88         if (it!=map_.end())
  89                 return it->second;
  90         it=map_.begin();
  91         return it->second;
  92 }
  93 */
  94
  95 RtpSession& RtpSessionTable::getOrNewSessionUnlocked(const std::string & addr)
  96 {
  97   RtpSessionMap::iterator it = map_.find(addr);
  98   if(it!=map_.end())
  99     return it->second;
 100
 101   map_.insert(RtpSessionMap::value_type(addr, RtpSession()));
 102   it = map_.find(addr);
 103   return it->second;
 104 }
 105
 106 uint16_t RtpSessionTable::getCountUnlocked()
 107 {
 108         RtpSessionMap::iterator it = map_.begin();
 109         uint16_t routes=0;
 110         for (;it!=map_.end();++it)
 111                 routes++;
 112         return routes;
 113 }
 114
 115 RtpSessionMap::iterator RtpSessionTable::getBeginUnlocked()
 116 {
 117         return map_.begin();
 118 }
 119
 120 RtpSessionMap::iterator RtpSessionTable::getEndUnlocked()
 121 {
 122         return map_.end();
 123 }
 124
 125 void RtpSessionTable::clear()
 126 {
 127   Lock lock(mutex_);
 128         map_.clear();
 129 }
 130
 131 bool RtpSessionTable::empty()
 132 {
 133   Lock lock(mutex_);
 134         return map_.empty();
 135 }
 136
 137 Mutex& RtpSessionTable::getMutex()
 138 {
 139   return mutex_;
 140 }