Fixed compile problems... Move along please, nothing to see here..

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

diff --git a/src/modules/m_cloaking.cpp b/src/modules/m_cloaking.cpp
index 73b09b423d116541b0a576f7e185ee1013ac52c1..7bcfb0af273a86f541d385466bea8746e77db3a8 100644 (file)
--- a/src/modules/m_cloaking.cpp
+++ b/src/modules/m_cloaking.cpp
@@ -29,7 +29,10 @@ using namespace std;
 // the server->server link, and all encoding of hosts is
 // done locally on the server by this module.
 
-#include <stdio.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
@@ -64,7 +67,8 @@ class ModuleCloaking : public Module
        {
                byte *p = (byte *)buf;
        
-               do {
+               do
+               {
                        *buf++ = (word32)((unsigned)p[3] << 8 | p[2]) << 16 |
                                ((unsigned)p[1] << 8 | p[0]);
                        p += 4;
@@ -93,7 +97,8 @@ class ModuleCloaking : public Module
                        ctx->bytes[1]++;        /* Carry from low to high */
 
                t = 64 - (t & 0x3f);    /* Space available in ctx->in (at least 1) */
-               if ((unsigned)t > (unsigned)len) {
+               if ((unsigned)t > (unsigned)len)
+               {
                        memcpy((byte *)ctx->in + 64 - (unsigned)t, buf, len);
                        return;
                }
@@ -105,7 +110,8 @@ class ModuleCloaking : public Module
                len -= (unsigned)t;
 
                /* Process data in 64-byte chunks */
-               while (len >= 64) {
+               while (len >= 64)
+               {
                        memcpy(ctx->in, buf, 64);
                        byteSwap(ctx->in, 16);
                        xMD5Transform(ctx->buf, ctx->in);
@@ -128,7 +134,8 @@ class ModuleCloaking : public Module
                /* Bytes of padding needed to make 56 bytes (-8..55) */
                count = 56 - 1 - count;
 
-               if (count < 0) {        /* Padding forces an extra block */
+               if (count < 0)
+               {       /* Padding forces an extra block */
                        memset(p, 0, count+8);
                        byteSwap(ctx->in, 16);
                        xMD5Transform(ctx->buf, ctx->in);
@@ -264,10 +271,11 @@ class ModuleCloaking : public Module
        }
         
  public:
-       ModuleCloaking()
+       ModuleCloaking(Server* Me)
+               : Module::Module(Me)
        {
-               // We must create an instance of the Server class to work with
-               Srv = new Server;
+               // We must take a copy of the Server class to work with
+               Srv = Me;
                
                // we must create a new mode. Set the parameters so the
                // mode doesn't require oper, and is a client usermode
@@ -287,8 +295,6 @@ class ModuleCloaking : public Module
        
        virtual ~ModuleCloaking()
        {
-               // not really neccessary, but free it anyway
-               delete Srv;
        }
        
        virtual Version GetVersion()
@@ -397,9 +403,9 @@ class ModuleCloakingFactory : public ModuleFactory
        {
        }
        
-       virtual Module * CreateModule()
+       virtual Module * CreateModule(Server* Me)
        {
-               return new ModuleCloaking;
+               return new ModuleCloaking(Me);
        }
        
 };