src/modules/m_spanningtree/resolvers.h

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
   5  *   Copyright (C) 2007 Craig Edwards <craigedwards@brainbox.cc>
   6  *
   7  * This file is part of InspIRCd.  InspIRCd is free software: you can
   8  * redistribute it and/or modify it under the terms of the GNU General Public
   9  * License as published by the Free Software Foundation, version 2.
  10  *
  11  * This program is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  13  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  14  * details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20
  21 #ifndef M_SPANNINGTREE_RESOLVERS_H
  22 #define M_SPANNINGTREE_RESOLVERS_H
  23
  24 #include "socket.h"
  25 #include "inspircd.h"
  26 #include "xline.h"
  27
  28 #include "utils.h"
  29 #include "link.h"
  30
  31 /** Handle resolving of server IPs for the cache
  32  */
  33 class SecurityIPResolver : public Resolver
  34 {
  35  private:
  36         reference<Link> MyLink;
  37         SpanningTreeUtilities* Utils;
  38         Module* mine;
  39         std::string host;
  40         QueryType query;
  41  public:
  42         SecurityIPResolver(Module* me, SpanningTreeUtilities* U, const std::string &hostname, Link* x, bool &cached, QueryType qt)
  43                 : Resolver(hostname, qt, cached, me), MyLink(x), Utils(U), mine(me), host(hostname), query(qt)
  44         {
  45         }
  46
  47         void OnLookupComplete(const std::string &result, unsigned int ttl, bool cached)
  48         {
  49                 Utils->ValidIPs.push_back(result);
  50         }
  51
  52         void OnError(ResolverError e, const std::string &errormessage)
  53         {
  54                 if (query == DNS_QUERY_AAAA)
  55                 {
  56                         bool cached;
  57                         SecurityIPResolver* res = new SecurityIPResolver(mine, Utils, host, MyLink, cached, DNS_QUERY_A);
  58                         ServerInstance->AddResolver(res, cached);
  59                         return;
  60                 }
  61                 ServerInstance->Logs->Log("m_spanningtree",DEFAULT,"Could not resolve IP associated with Link '%s': %s",
  62                         MyLink->Name.c_str(),errormessage.c_str());
  63         }
  64 };
  65
  66 /** This class is used to resolve server hostnames during /connect and autoconnect.
  67  * As of 1.1, the resolver system is seperated out from BufferedSocket, so we must do this
  68  * resolver step first ourselves if we need it. This is totally nonblocking, and will
  69  * callback to OnLookupComplete or OnError when completed. Once it has completed we
  70  * will have an IP address which we can then use to continue our connection.
  71  */
  72 class ServernameResolver : public Resolver
  73 {
  74  private:
  75         SpanningTreeUtilities* Utils;
  76         QueryType query;
  77         std::string host;
  78         reference<Link> MyLink;
  79         reference<Autoconnect> myautoconnect;
  80  public:
  81         ServernameResolver(SpanningTreeUtilities* Util, const std::string &hostname, Link* x, bool &cached, QueryType qt, Autoconnect* myac);
  82         void OnLookupComplete(const std::string &result, unsigned int ttl, bool cached);
  83         void OnError(ResolverError e, const std::string &errormessage);
  84 };
  85
  86 #endif