Annotations

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

diff --git a/src/modules/m_conn_waitpong.cpp b/src/modules/m_conn_waitpong.cpp
index 804026adbae8c4de193747be55b07d70c972d02f..937577c5ec2443e67b0758a5374943d0a9eb48cf 100644 (file)
--- a/src/modules/m_conn_waitpong.cpp
+++ b/src/modules/m_conn_waitpong.cpp
@@ -4,10 +4,13 @@
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
-#include "helperfuncs.h"
+
+#include "inspircd.h"
 
 /* $ModDesc: Forces connecting clients to send a PONG message back to the server before they can complete their connection */
 
+
+
 char* RandString(unsigned int length)
 {
        unsigned char* tmp = new unsigned char[(length/4)*3];
@@ -28,23 +31,23 @@ char* RandString(unsigned int length)
 
 class ModuleWaitPong : public Module
 {
-       Server* Srv;
+       
        ConfigReader* Conf;
        
        bool sendsnotice;
        bool killonbadreply;
 
  public:
-       ModuleWaitPong(Server* Me)
+       ModuleWaitPong(InspIRCd* Me)
                : Module::Module(Me)
        {
-               Srv = Me;
+               
                OnRehash("");
        }
        
        virtual void OnRehash(const std::string &param)
        {
-               Conf = new ConfigReader;
+               Conf = new ConfigReader(ServerInstance);
                
                sendsnotice = Conf->ReadFlag("waitpong", "sendsnotice", 0);
                
@@ -68,19 +71,20 @@ class ModuleWaitPong : public Module
        {
                char* pingrpl = RandString(10);
                
-               Write(user->fd, "PING :%s", pingrpl);
+               user->Write("PING :%s", pingrpl);
                
                if(sendsnotice)
-                       WriteServ(user->fd, "NOTICE %s :*** If you are having problems connecting due to ping timeouts, please type /quote PONG %s or /raw PONG %s now.", user->nick, pingrpl, pingrpl);
+                       user->WriteServ("NOTICE %s :*** If you are having problems connecting due to ping timeouts, please type /quote PONG %s or /raw PONG %s now.", user->nick, pingrpl, pingrpl);
                        
                user->Extend("waitpong_pingstr", pingrpl);
        }
        
-       virtual int OnPreCommand(const std::string &command, char** parameters, int pcnt, userrec* user, bool validated)
+       virtual int OnPreCommand(const std::string &command, const char** parameters, int pcnt, userrec* user, bool validated)
        {
                if(command == "PONG")
                {
-                       char* pingrpl = user->GetExt("waitpong_pingstr");
+                       char* pingrpl;
+                       user->GetExt("waitpong_pingstr", pingrpl);
                        
                        if(pingrpl)
                        {
@@ -93,7 +97,7 @@ class ModuleWaitPong : public Module
                                else
                                {
                                        if(killonbadreply)
-                                               Srv->QuitUser(user, "Incorrect ping reply for registration");
+                                               userrec::QuitUser(ServerInstance, user, "Incorrect ping reply for registration");
                                        return 1;
                                }
                        }
@@ -104,12 +108,14 @@ class ModuleWaitPong : public Module
 
        virtual bool OnCheckReady(userrec* user)
        {
-               return (!user->GetExt("waitpong_pingstr"));
+               char* pingrpl;
+               return (!user->GetExt("waitpong_pingstr", pingrpl));
        }
        
        virtual void OnUserDisconnect(userrec* user)
        {
-               char* pingrpl = user->GetExt("waitpong_pingstr");
+               char* pingrpl;
+               user->GetExt("waitpong_pingstr", pingrpl);
 
                if(pingrpl)
                {
@@ -123,7 +129,8 @@ class ModuleWaitPong : public Module
                if(target_type == TYPE_USER)
                {
                        userrec* user = (userrec*)item;
-                       char* pingrpl = user->GetExt("waitpong_pingstr");
+                       char* pingrpl;
+                       user->GetExt("waitpong_pingstr", pingrpl);
                        
                        if(pingrpl)
                        {
@@ -155,7 +162,7 @@ class ModuleWaitPongFactory : public ModuleFactory
        {
        }
        
-       virtual Module * CreateModule(Server* Me)
+       virtual Module * CreateModule(InspIRCd* Me)
        {
                return new ModuleWaitPong(Me);
        }