diff options
| -rw-r--r-- | include/modules.h | 3 | ||||
| -rw-r--r-- | src/modules.cpp | 6 | ||||
| -rw-r--r-- | src/modules/m_testcommand.cpp | 35 |
3 files changed, 29 insertions, 15 deletions
diff --git a/include/modules.h b/include/modules.h index a1d350b8f..f7b8bc53a 100644 --- a/include/modules.h +++ b/include/modules.h @@ -71,6 +71,7 @@ enum TargetTypeFlags { #include <typeinfo> #include "timer.h" #include "mode.h" +#include "dns.h" class Server; class ServerConfig; @@ -1431,6 +1432,8 @@ class Server : public Extensible virtual bool DelModeWatcher(ModeWatcher* mw); + virtual bool AddResolver(Resolver* r); + /** Adds a command to the command table. * This allows modules to add extra commands into the command table. You must place a function within your * module which is is of type handlerfunc: diff --git a/src/modules.cpp b/src/modules.cpp index 2297db53b..7214a9673 100644 --- a/src/modules.cpp +++ b/src/modules.cpp @@ -43,6 +43,7 @@ #include "typedefs.h" #include "modules.h" #include "command_parse.h" +#include "dns.h" extern ServerConfig *Config; extern InspIRCd* ServerInstance; @@ -606,6 +607,11 @@ bool Server::DelModeWatcher(ModeWatcher* mw) return ServerInstance->ModeGrok->DelModeWatcher(mw); } +bool Server::AddResolver(Resolver* r) +{ + return dns_add_class(r); +} + int Server::CountUsers(chanrec* c) { return usercount(c); diff --git a/src/modules/m_testcommand.cpp b/src/modules/m_testcommand.cpp index 473f2941d..c194811ec 100644 --- a/src/modules/m_testcommand.cpp +++ b/src/modules/m_testcommand.cpp @@ -20,9 +20,25 @@ using namespace std; #include "users.h" #include "channels.h" #include "modules.h" +#include "dns.h" /* $ModDesc: Povides a proof-of-concept test /WOOT command */ +class MyResolver : public Resolver +{ + MyResolver(const std::string &source, bool forward, const std::string &dnsserver = "") : Resolver(source, forward, dnsserver) { } + + virtual void OnLookupComplete(const std::string &result) + { + log(DEBUG,"*** RESOLVER COMPLETED LOOKUP, IP IS: '%s'",result.c_str()); + } + + virtual void OnError(ResolverError e) + { + log(DEBUG,"*** RESOLVER GOT ERROR: %d",e); + } +}; + static Server *Srv; class cmd_woot : public command_t @@ -35,21 +51,10 @@ class cmd_woot : public command_t void Handle (char **parameters, int pcnt, userrec *user) { - Srv->Log(DEBUG,"woot handler"); - // Here is a sample of how to send servermodes. Note that unless remote - // servers in your net are u:lined, they may reverse this, but its a - // quick and effective modehack. - - // NOTE: DO NOT CODE LIKE THIS!!! This has no checks and can send - // invalid or plain confusing mode changes, code some checking! - char* modes[3]; - modes[0] = "#chatspike"; - modes[1] = "+o"; - modes[2] = user->nick; - - // run the mode change, send numerics (such as "no such channel") back - // to "user". - Srv->SendMode(modes,3,user); + /* We dont have to worry about deleting 'r', the core will + * do it for us as required.*/ + MyResolver* r = new MyResolver("brainbox.ath.cx", true); + Srv->AddResolver(r); } }; |