src/packetSource.h

   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_packetSource_h_INCLUDED
  34 #define ANYTUN_packetSource_h_INCLUDED
  35
  36 #include <boost/asio.hpp>
  37
  38 #include "datatypes.h"
  39 #include "threadUtils.hpp"
  40
  41 // TODO: fix this when other packetSource types are introduced
  42 typedef boost::asio::ip::udp::endpoint PacketSourceEndpoint;
  43
  44 class PacketSource
  45 {
  46 public:
  47   virtual ~PacketSource() {}
  48
  49   virtual u_int32_t recv(u_int8_t* buf, u_int32_t len, PacketSourceEndpoint& remote) = 0;
  50   virtual void send(u_int8_t* buf, u_int32_t len, PacketSourceEndpoint remote) = 0;
  51
  52   void waitUntilReady();
  53
  54 protected:
  55   Semaphore ready_sem_;
  56 };
  57
  58 class UDPPacketSource : public PacketSource
  59 {
  60 public:
  61   typedef boost::asio::ip::udp proto;
  62
  63   UDPPacketSource(std::string localaddr, std::string port);
  64
  65   u_int32_t recv(u_int8_t* buf, u_int32_t len, PacketSourceEndpoint& remote);
  66   void send(u_int8_t* buf, u_int32_t len, PacketSourceEndpoint remote);
  67
  68   void onResolve(const boost::asio::ip::udp::endpoint& e);
  69   void onError(const std::runtime_error& e);
  70
  71 private:
  72
  73   boost::asio::io_service io_service_;
  74   proto::socket sock_;
  75 };
  76
  77 #endif