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 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-2008 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 version 3 as
  21  *  published by the Free Software Foundation.
  22  *
  23  *  Anytun is distributed in the hope that it will be useful,
  24  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  25  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  26  *  GNU General Public License for more details.
  27  *
  28  *  You should have received a copy of the GNU General Public License
  29  *  along with anytun.  If not, see <http://www.gnu.org/licenses/>.
  30  */
  31
  32 #ifndef _RESOLVER_H_
  33 #define _RESOLVER_H_
  34
  35 #include <queue>
  36 #include <boost/asio.hpp>
  37 #include <boost/function.hpp>
  38
  39 #include "datatypes.h"
  40 #include "threadUtils.hpp"
  41
  42 typedef boost::function<void (boost::asio::ip::udp::endpoint)> UdpResolveCallback;
  43 typedef boost::function<void (boost::asio::ip::tcp::endpoint)> TcpResolveCallback;
  44 typedef boost::function<void (std::runtime_error const&)> ErrorCallback;
  45
  46 template<class Proto>
  47 class ResolveHandler
  48 {
  49 public:
  50   ResolveHandler(const std::string& addr, const std::string& port, boost::function<void (boost::asio::ip::basic_endpoint<Proto>)> const& onResolve, ErrorCallback const& onError, ResolvAddrType r = ANY);
  51   void operator()(const boost::system::error_code& e, const boost::asio::ip::basic_resolver_iterator<Proto>);
  52
  53 private:
  54   std::string addr_;
  55   std::string port_;
  56   boost::function<void (boost::asio::ip::basic_endpoint<Proto>)> onResolve_;
  57   ErrorCallback onError_;
  58   ResolvAddrType resolv_addr_type_;
  59 };
  60
  61 typedef ResolveHandler<boost::asio::ip::udp> UdpResolveHandler;
  62 typedef ResolveHandler<boost::asio::ip::tcp> TcpResolveHandler;
  63
  64 class Resolver
  65 {
  66 public:
  67   static Resolver& instance();
  68
  69   void init();
  70   void run();
  71
  72   void resolveUdp(const std::string& addr, const std::string& port, UdpResolveCallback const& onResolve, ErrorCallback const& onError, ResolvAddrType r = ANY);
  73   void resolveTcp(const std::string& addr, const std::string& port, TcpResolveCallback const& onResolve, ErrorCallback const& onError, ResolvAddrType r = ANY);
  74
  75 private:
  76   Resolver();
  77   ~Resolver();
  78   Resolver(const Resolver &r);
  79   void operator=(const Resolver &r);
  80
  81   static Resolver* inst;
  82   static ::Mutex instMutex;
  83   class instanceCleaner {
  84     public: ~instanceCleaner() {
  85       if(Resolver::inst != 0)
  86         delete Resolver::inst;
  87     }
  88   };
  89   friend class instanceCleaner;
  90
  91   boost::asio::io_service io_service_;
  92   boost::asio::ip::udp::resolver udp_resolver_;
  93   boost::asio::ip::tcp::resolver tcp_resolver_;
  94   boost::thread* thread_;
  95 };
  96
  97 extern Resolver& gResolver;
  98
  99 #endif