Added some missing parameter checking in m_swhois

[user/henk/code/inspircd.git] / src / modules / m_testcommand.cpp

diff --git a/src/modules/m_testcommand.cpp b/src/modules/m_testcommand.cpp
index fd05b8009b4488d42e02a13290249a24b65ea25e..6df2db1dd300572d5aa8b2195a0a60eda6c82412 100644 (file)
--- a/src/modules/m_testcommand.cpp
+++ b/src/modules/m_testcommand.cpp
@@ -1,69 +1,105 @@
-#include <stdio.h>
+/*       +------------------------------------+
+ *       | Inspire Internet Relay Chat Daemon |
+ *       +------------------------------------+
+ *
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
+ *
+ * This program is free but copyrighted software; see
+ *            the file COPYING for details.
+ *
+ * ---------------------------------------------------
+ */
 
+#include "inspircd.h"
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
+#include "dns.h"
 
 /* $ModDesc: Povides a proof-of-concept test /WOOT command */
 
-Server *Srv;
-        
-
-void handle_woot(char **parameters, int pcnt, userrec *user)
-{
-       // this test command just accepts:
-       // /woot <text>
-       // and sends <text> to all opers with +s mode.
-       Srv->SendOpers(parameters[0]);
-}
-
-class ModuleTestCommand : public Module
+/** A test resolver class for IPV6
+ */
+class MyV6Resolver : public Resolver
 {
+       bool fw;
  public:
-       ModuleTestCommand()
+       MyV6Resolver(InspIRCd* Instance, Module* me, const std::string &source, bool forward, bool &cached) : Resolver(Instance, source, forward ? DNS_QUERY_AAAA : DNS_QUERY_PTR6, cached, me)
        {
-               Srv = new Server;
-               Srv->AddCommand("WOOT",handle_woot,0,1)
+               fw = forward;
        }
-       
-       virtual ~ModuleTestCommand()
+
+       virtual void OnLookupComplete(const std::string &result, unsigned int ttl, bool cached)
        {
-               delete Srv;
+               ServerInstance->Log(DEBUG,"*** RESOLVER COMPLETED %s LOOKUP, IP IS: '%s' TTL=%lu CACHED=%d",fw ? "FORWARD" : "REVERSE", result.c_str(), ttl, cached);
        }
-       
-       virtual Version GetVersion()
+
+       virtual void OnError(ResolverError e, const std::string &errormessage)
        {
-               return Version(1,0,0,0);
+               ServerInstance->Log(DEBUG,"*** RESOLVER GOT ERROR: %d: %s",e,errormessage.c_str());
        }
-       
-       virtual void OnUserConnect(userrec* user)
+};
+
+/** Handle /WOOT
+ */
+class cmd_woot : public command_t
+{
+       Module* Creator;
+ public:
+       /* Command 'woot', takes no parameters and needs no special modes */
+       cmd_woot (InspIRCd* Instance, Module* maker) : command_t(Instance,"WOOT", 0, 0), Creator(maker)
        {
+               this->source = "m_testcommand.so";
        }
 
-};
+       CmdResult Handle (const char** parameters, int pcnt, userrec *user)
+       {
+               /* We dont have to worry about deleting 'r', the core will
+                * do it for us as required.*/
+               try
+               {
+                       bool cached;
+                       MyV6Resolver* r = new MyV6Resolver(ServerInstance, Creator, "shake.stacken.kth.se", true, cached);
+                       ServerInstance->AddResolver(r, cached);
+                       r = new MyV6Resolver(ServerInstance, Creator, "2001:6b0:1:ea:202:a5ff:fecd:13a6", false, cached);
+                       ServerInstance->AddResolver(r, cached);
+               }
+               catch (ModuleException& e)
+               {
+                       ServerInstance->Log(DEBUG,"Danger, will robinson! There was an exception: %s",e.GetReason());
+               }
 
+               return CMD_FAILURE;
+       }
+};
 
-class ModuleTestCommandFactory : public ModuleFactory
+class ModuleTestCommand : public Module
 {
+       cmd_woot* newcommand;
  public:
-       ModuleTestCommandFactory()
+       ModuleTestCommand(InspIRCd* Me)
+               : Module(Me)
        {
+               
+               // Create a new command
+               newcommand = new cmd_woot(ServerInstance, this);
+               ServerInstance->AddCommand(newcommand);
        }
-       
-       ~ModuleTestCommandFactory()
+
+       void Implements(char* List)
        {
        }
-       
-       virtual Module * CreateModule()
+
+       virtual ~ModuleTestCommand()
        {
-               return new ModuleTestCommand;
+               delete newcommand;
        }
-       
-};
 
+       virtual Version GetVersion()
+       {
+               return Version(1, 1, 0, 0, VF_VENDOR, API_VERSION);
+       }
+};
 
-extern "C" void * init_module( void )
-{
-       return new ModuleTestCommandFactory;
-}
-
+MODULE_INIT(ModuleTestCommand)