src/resolver.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 methods 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-2014 Markus Grüneis, 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  *  In addition, as a special exception, the copyright holders give
  33  *  permission to link the code of portions of this program with the
  34  *  OpenSSL library under certain conditions as described in each
  35  *  individual source file, and distribute linked combinations
  36  *  including the two.
  37  *  You must obey the GNU General Public License in all respects
  38  *  for all of the code used other than OpenSSL.  If you modify
  39  *  file(s) with this exception, you may extend this exception to your
  40  *  version of the file(s), but you are not obligated to do so.  If you
  41  *  do not wish to do so, delete this exception statement from your
  42  *  version.  If you delete this exception statement from all source
  43  *  files in the program, then also delete it here.
  44  */
  45
  46 #ifndef ANYTUN_resolver_h_INCLUDED
  47 #define ANYTUN_resolver_h_INCLUDED
  48
  49 #include <queue>
  50 #include <boost/asio.hpp>
  51 #include <boost/function.hpp>
  52
  53 #include "datatypes.h"
  54 #include "threadUtils.hpp"
  55
  56 typedef boost::function<void (boost::asio::ip::udp::resolver::iterator)> UdpResolveCallback;
  57 typedef boost::function<void (boost::asio::ip::tcp::resolver::iterator)> TcpResolveCallback;
  58 typedef boost::function<void (std::runtime_error const&)> ErrorCallback;
  59
  60 template<class Proto>
  61 class ResolveHandler
  62 {
  63 public:
  64   ResolveHandler(const std::string& addr, const std::string& port, boost::function<void (boost::asio::ip::basic_resolver_iterator<Proto>)> const& onResolve, ErrorCallback const& onError, ResolvAddrType r = ANY);
  65   void operator()(const boost::system::error_code& e, boost::asio::ip::basic_resolver_iterator<Proto>);
  66
  67 private:
  68   std::string addr_;
  69   std::string port_;
  70   boost::function<void (const boost::asio::ip::basic_resolver_iterator<Proto>)> onResolve_;
  71   ErrorCallback onError_;
  72   ResolvAddrType resolv_addr_type_;
  73 };
  74
  75 typedef ResolveHandler<boost::asio::ip::udp> UdpResolveHandler;
  76 typedef ResolveHandler<boost::asio::ip::tcp> TcpResolveHandler;
  77
  78 class Resolver
  79 {
  80 public:
  81   static Resolver& instance();
  82
  83   void init();
  84   void run();
  85
  86   void resolveUdp(const std::string& addr, const std::string& port, UdpResolveCallback const& onResolve, ErrorCallback const& onError, ResolvAddrType r = ANY);
  87   void resolveTcp(const std::string& addr, const std::string& port, TcpResolveCallback const& onResolve, ErrorCallback const& onError, ResolvAddrType r = ANY);
  88
  89 private:
  90   Resolver();
  91   ~Resolver();
  92   Resolver(const Resolver& r);
  93   void operator=(const Resolver& r);
  94
  95   boost::asio::io_service io_service_;
  96   boost::asio::ip::udp::resolver udp_resolver_;
  97   boost::asio::ip::tcp::resolver tcp_resolver_;
  98   boost::thread* thread_;
  99 };
 100
 101 extern Resolver& gResolver;
 102
 103 #endif