]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_spanningtree.cpp
Missing function call
[user/henk/code/inspircd.git] / src / modules / m_spanningtree.cpp
index 6e2760e6256846834422919f3f20655885ae264f..e859ff8445fc4436da93ec5e9e6ba2930d5cf9d8 100644 (file)
@@ -2,7 +2,7 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  Inspire is copyright (C) 2002-2005 ChatSpike-Dev.
+ *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
  *                       E-mail:
  *                <brain@chatspike.net>
  *               <Craig@chatspike.net>
@@ -73,6 +73,8 @@ using namespace std;
 class ModuleSpanningTree;
 static ModuleSpanningTree* TreeProtocolModule;
 
+extern ServerConfig* Config;
+
 extern std::vector<Module*> modules;
 extern std::vector<ircd_module*> factory;
 extern int MODCOUNT;
@@ -95,8 +97,6 @@ extern int MODCOUNT;
 enum ServerState { LISTENER, CONNECTING, WAIT_AUTH_1, WAIT_AUTH_2, CONNECTED };
 
 /* We need to import these from the core for use in netbursts */
-/*typedef nspace::hash_map<std::string, userrec*, nspace::hash<string>, irc::StrHashComp> user_hash;
-typedef nspace::hash_map<std::string, chanrec*, nspace::hash<string>, irc::StrHashComp> chan_hash;*/
 extern user_hash clientlist;
 extern chan_hash chanlist;
 
@@ -124,6 +124,11 @@ bool DoOneToMany(std::string prefix, std::string command, std::deque<std::string
 bool DoOneToAllButSenderRaw(std::string data, std::string omit, std::string prefix, std::string command, std::deque<std::string> &params);
 void ReadConfiguration(bool rebind);
 
+/* Flatten links and /MAP for non-opers */
+bool FlatLinks;
+/* Hide U-Lined servers in /MAP and /LINKS */
+bool HideULines;
+
 /* Imported from xline.cpp for use during netburst */
 extern std::vector<KLine> klines;
 extern std::vector<GLine> glines;
@@ -191,6 +196,7 @@ class TreeServer
                UserCount = OperCount = 0;
                VersionString = Srv->GetVersion();
                Route = NULL;
+               Socket = NULL; /* Fix by brain */
                AddHashEntry();
        }
 
@@ -202,7 +208,7 @@ class TreeServer
        {
                VersionString = "";
                UserCount = OperCount = 0;
-               this->SetNextPingTime(time(NULL) + 60);
+               this->SetNextPingTime(time(NULL) + 120);
                this->SetPingFlag();
 
                /* find the 'route' for this server (e.g. the one directly connected
@@ -294,17 +300,17 @@ class TreeServer
 
        std::string GetName()
        {
-               return this->ServerName;
+               return ServerName;
        }
 
        std::string GetDesc()
        {
-               return this->ServerDesc;
+               return ServerDesc;
        }
 
        std::string GetVersion()
        {
-               return this->VersionString;
+               return VersionString;
        }
 
        void SetNextPingTime(time_t t)
@@ -315,7 +321,7 @@ class TreeServer
 
        time_t NextPingTime()
        {
-               return this->NextPing;
+               return NextPing;
        }
 
        bool AnsweredLastPing()
@@ -330,22 +336,32 @@ class TreeServer
 
        int GetUserCount()
        {
-               return this->UserCount;
+               return UserCount;
+       }
+
+       void AddUserCount()
+       {
+               UserCount++;
+       }
+
+       void DelUserCount()
+       {
+               UserCount--;
        }
 
        int GetOperCount()
        {
-               return this->OperCount;
+               return OperCount;
        }
 
        TreeSocket* GetSocket()
        {
-               return this->Socket;
+               return Socket;
        }
 
        TreeServer* GetParent()
        {
-               return this->Parent;
+               return Parent;
        }
 
        void SetVersion(std::string Version)
@@ -441,6 +457,7 @@ class Link
         unsigned long AutoConnect;
         time_t NextConnectTime;
         std::string EncryptionKey;
+        bool HiddenFromStats;
 };
 
 /* The usual stuff for inspircd modules,
@@ -514,6 +531,33 @@ bool IsServer(std::string ServerName)
        return (FindServer(ServerName) != NULL);
 }
 
+
+class cmd_rconnect : public command_t    
+{
+       Module* Creator;
+ public:
+         cmd_rconnect (Module* Callback) : command_t("RCONNECT", 'o', 2), Creator(Callback)
+         {
+                this->source = "m_spanningtree.so";
+         }
+                
+         void Handle (char **parameters, int pcnt, userrec *user)
+         {
+               WriteServ(user->fd,"NOTICE %s :*** RCONNECT: Sending remote connect to \002%s\002 to connect server \002%s\002.",user->nick,parameters[0],parameters[1]);
+                /* Is this aimed at our server? */
+                if (Srv->MatchText(Srv->GetServerName(),parameters[0]))
+                {
+                        /* Yes, initiate the given connect */
+                       WriteOpers("*** Remote CONNECT from %s matching \002%s\002, connecting server \002%s\002",user->nick,parameters[0],parameters[1]);
+                        char* para[1];
+                        para[0] = parameters[1];
+                        Creator->OnPreCommand("CONNECT", para, 1, user, true);
+                }
+         }
+};
+         
+
+
 /* Every SERVER connection inbound or outbound is represented by
  * an object of type TreeSocket.
  * TreeSockets, being inherited from InspSocket, can be tied into
@@ -538,7 +582,8 @@ class TreeSocket : public InspSocket
        time_t NextPing;
        bool LastPingWasGood;
        bool bursting;
-       AES* ctx;
+       AES* ctx_in;
+       AES* ctx_out;
        unsigned int keylength;
        
  public:
@@ -553,6 +598,8 @@ class TreeSocket : public InspSocket
        {
                myhost = host;
                this->LinkState = LISTENER;
+               this->ctx_in = NULL;
+               this->ctx_out = NULL;
        }
 
        TreeSocket(std::string host, int port, bool listening, unsigned long maxtime, std::string ServerName)
@@ -560,6 +607,8 @@ class TreeSocket : public InspSocket
        {
                myhost = ServerName;
                this->LinkState = CONNECTING;
+               this->ctx_in = NULL;
+               this->ctx_out = NULL;
        }
 
        /* When a listening socket gives us a new file descriptor,
@@ -570,6 +619,17 @@ class TreeSocket : public InspSocket
                : InspSocket(newfd, ip)
        {
                this->LinkState = WAIT_AUTH_1;
+               this->ctx_in = NULL;
+               this->ctx_out = NULL;
+               this->SendCapabilities();
+       }
+
+       ~TreeSocket()
+       {
+               if (ctx_in)
+                       delete ctx_in;
+               if (ctx_out)
+                       delete ctx_out;
        }
 
        void InitAES(std::string key,std::string SName)
@@ -577,19 +637,22 @@ class TreeSocket : public InspSocket
                if (key == "")
                        return;
 
-               ctx = new AES();
+               ctx_in = new AES();
+               ctx_out = new AES();
                log(DEBUG,"Initialized AES key %s",key.c_str());
                // key must be 16, 24, 32 etc bytes (multiple of 8)
                keylength = key.length();
                if (!(keylength == 16 || keylength == 24 || keylength == 32))
                {
-                       WriteOpers("\2ERROR\2: Key length for encryptionkey is not 16, 24 or 32 bytes in length!");
+                       WriteOpers("*** \2ERROR\2: Key length for encryptionkey is not 16, 24 or 32 bytes in length!");
                        log(DEBUG,"Key length not 16, 24 or 32 characters!");
                }
                else
                {
-                       WriteOpers("\2AES\2: Initialized %d bit encryption to server %s",keylength*8,SName.c_str());
-                       ctx->MakeKey(key.c_str(), "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\
+                       WriteOpers("*** \2AES\2: Initialized %d bit encryption to server %s",keylength*8,SName.c_str());
+                       ctx_in->MakeKey(key.c_str(), "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\
+                               \0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0", keylength, keylength);
+                       ctx_out->MakeKey(key.c_str(), "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\
                                \0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0", keylength, keylength);
                }
        }
@@ -604,12 +667,13 @@ class TreeSocket : public InspSocket
        {
                if (this->LinkState == CONNECTING)
                {
-                       Srv->SendOpers("*** Connection to "+myhost+"["+this->GetIP()+"] established.");
                        /* we do not need to change state here. */
                        for (std::vector<Link>::iterator x = LinkBlocks.begin(); x < LinkBlocks.end(); x++)
                        {
                                if (x->Name == this->myhost)
                                {
+                                       Srv->SendOpers("*** Connection to \2"+myhost+"\2["+(x->HiddenFromStats ? "<hidden>" : this->GetIP())+"] established.");
+                                       this->SendCapabilities();
                                        if (x->EncryptionKey != "")
                                        {
                                                if (!(x->EncryptionKey.length() == 16 || x->EncryptionKey.length() == 24 || x->EncryptionKey.length() == 32))
@@ -633,6 +697,7 @@ class TreeSocket : public InspSocket
                 * If that happens the connection hangs here until it's closed. Unlikely
                 * and rather harmless.
                 */
+               Srv->SendOpers("*** Connection to \2"+myhost+"\2 lost link tag(!)");
                return true;
        }
        
@@ -678,6 +743,56 @@ class TreeSocket : public InspSocket
                }
        }
 
+       std::string MyCapabilities()
+       {
+               ServerConfig* Config = Srv->GetConfig();
+               std::vector<std::string> modlist;
+               std::string capabilities = "";
+
+                for (int i = 0; i <= MODCOUNT; i++)
+                {
+                       if ((modules[i]->GetVersion().Flags & VF_STATIC) || (modules[i]->GetVersion().Flags & VF_COMMON))
+                               modlist.push_back(Config->module_names[i]);
+                }
+               sort(modlist.begin(),modlist.end());
+               for (unsigned int i = 0; i < modlist.size(); i++)
+               {
+                       if (i)
+                               capabilities = capabilities + ",";
+                       capabilities = capabilities + modlist[i];
+               }
+               return capabilities;
+       }
+       
+       void SendCapabilities()
+       {
+               this->WriteLine("CAPAB "+MyCapabilities());
+       }
+
+       bool Capab(std::deque<std::string> params)
+       {
+               if (params.size() != 1)
+               {
+                       this->WriteLine("ERROR :Invalid number of parameters for CAPAB");
+                       return false;
+               }
+               if (params[0] != this->MyCapabilities())
+               {
+                       std::string quitserver = this->myhost;
+                       if (this->InboundServerName != "")
+                       {
+                               quitserver = this->InboundServerName;
+                       }
+                       WriteOpers("*** \2ERROR\2: Server '%s' does not have the same set of modules loaded, cannot link!",quitserver.c_str());
+                       WriteOpers("*** Our networked module set is: '%s'",this->MyCapabilities().c_str());
+                       WriteOpers("*** Other server's networked module set is: '%s'",params[0].c_str());
+                       WriteOpers("*** These lists must match exactly on both servers. Please correct these errors, and try again.");
+                       this->WriteLine("ERROR :CAPAB mismatch; My capabilities: '"+this->MyCapabilities()+"'");
+                       return false;
+               }
+               return true;
+       }
+
        /* This function forces this server to quit, removing this server
         * and any users on it (and servers and users below that, etc etc).
         * It's very slow and pretty clunky, but luckily unless your network
@@ -714,7 +829,7 @@ class TreeSocket : public InspSocket
         */
        void Squit(TreeServer* Current,std::string reason)
        {
-               if (Current)
+               if ((Current) && (Current != TreeRoot))
                {
                        std::deque<std::string> params;
                        params.push_back(Current->GetName());
@@ -768,33 +883,42 @@ class TreeSocket : public InspSocket
        {
                if (params.size() != 4)
                        return true;
-               std::string channel = params[0];
                time_t ts = atoi(params[1].c_str());
-               std::string setby = params[2];
-               std::string topic = params[3];
+               std::string nsource = source;
 
-               chanrec* c = Srv->FindChannel(channel);
+               chanrec* c = Srv->FindChannel(params[0]);
                if (c)
                {
                        if ((ts >= c->topicset) || (!*c->topic))
                        {
                                std::string oldtopic = c->topic;
-                               strlcpy(c->topic,topic.c_str(),MAXTOPIC);
-                               strlcpy(c->setby,setby.c_str(),NICKMAX);
+                               strlcpy(c->topic,params[3].c_str(),MAXTOPIC);
+                               strlcpy(c->setby,params[2].c_str(),NICKMAX);
                                c->topicset = ts;
                                /* if the topic text is the same as the current topic,
                                 * dont bother to send the TOPIC command out, just silently
                                 * update the set time and set nick.
                                 */
-                               if (oldtopic != topic)
-                                       WriteChannelWithServ((char*)source.c_str(), c, "TOPIC %s :%s", c->name, c->topic);
+                               if (oldtopic != params[3])
+                               {
+                                       userrec* user = Srv->FindNick(source);
+                                       if (!user)
+                                       {
+                                               WriteChannelWithServ((char*)source.c_str(), c, "TOPIC %s :%s", c->name, c->topic);
+                                       }
+                                       else
+                                       {
+                                               WriteChannel(c, user, "TOPIC %s :%s", c->name, c->topic);
+                                               nsource = user->server;
+                                       }
+                               }
                        }
                        
                }
                
                /* all done, send it on its way */
                params[3] = ":" + params[3];
-               DoOneToAllButSender(source,"FTOPIC",params,source);
+               DoOneToAllButSender(source,"FTOPIC",params,nsource);
 
                return true;
        }
@@ -846,56 +970,61 @@ class TreeSocket : public InspSocket
                {
                        /* process one channel at a time, applying modes. */
                        char* usr = (char*)params[usernum].c_str();
-                       char permissions = *usr;
-                       switch (permissions)
+                       /* Safety check just to make sure someones not sent us an FJOIN full of spaces
+                        * (is this even possible?) */
+                       if (usr && *usr)
                        {
-                               case '@':
-                                       usr++;
-                                       mode_users[modectr++] = usr;
-                                       strlcat(modestring,"o",MAXBUF);
-                               break;
-                               case '%':
-                                       usr++;
-                                       mode_users[modectr++] = usr;
-                                       strlcat(modestring,"h",MAXBUF);
-                               break;
-                               case '+':
-                                       usr++;
-                                       mode_users[modectr++] = usr;
-                                       strlcat(modestring,"v",MAXBUF);
-                               break;
-                       }
-                       who = Srv->FindNick(usr);
-                       if (who)
-                       {
-                               Srv->JoinUserToChannel(who,channel,key);
-                               if (modectr >= (MAXMODES-1))
+                               char permissions = *usr;
+                               switch (permissions)
                                {
-                                       /* theres a mode for this user. push them onto the mode queue, and flush it
-                                        * if there are more than MAXMODES to go.
-                                        */
-                                       if ((ourTS >= TS) || (Srv->IsUlined(who->server)))
-                                       {
-                                               /* We also always let u-lined clients win, no matter what the TS value */
-                                               log(DEBUG,"Our our channel newer than theirs, accepting their modes");
-                                               Srv->SendMode(mode_users,modectr,who);
-                                       }
-                                       else
+                                       case '@':
+                                               usr++;
+                                               mode_users[modectr++] = usr;
+                                               strlcat(modestring,"o",MAXBUF);
+                                       break;
+                                       case '%':
+                                               usr++;
+                                               mode_users[modectr++] = usr;
+                                               strlcat(modestring,"h",MAXBUF);
+                                       break;
+                                       case '+':
+                                               usr++;
+                                               mode_users[modectr++] = usr;
+                                               strlcat(modestring,"v",MAXBUF);
+                                       break;
+                               }
+                               who = Srv->FindNick(usr);
+                               if (who)
+                               {
+                                       Srv->JoinUserToChannel(who,channel,key);
+                                       if (modectr >= (MAXMODES-1))
                                        {
-                                               log(DEBUG,"Their channel newer than ours, bouncing their modes");
-                                               /* bouncy bouncy! */
-                                               std::deque<std::string> params;
-                                               /* modes are now being UNSET... */
-                                               *mode_users[1] = '-';
-                                               for (unsigned int x = 0; x < modectr; x++)
+                                               /* theres a mode for this user. push them onto the mode queue, and flush it
+                                                * if there are more than MAXMODES to go.
+                                                */
+                                               if ((ourTS >= TS) || (Srv->IsUlined(who->server)))
                                                {
-                                                       params.push_back(mode_users[x]);
+                                                       /* We also always let u-lined clients win, no matter what the TS value */
+                                                       log(DEBUG,"Our our channel newer than theirs, accepting their modes");
+                                                       Srv->SendMode(mode_users,modectr,who);
                                                }
-                                               // tell everyone to bounce the modes. bad modes, bad!
-                                               DoOneToMany(Srv->GetServerName(),"FMODE",params);
+                                               else
+                                               {
+                                                       log(DEBUG,"Their channel newer than ours, bouncing their modes");
+                                                       /* bouncy bouncy! */
+                                                       std::deque<std::string> params;
+                                                       /* modes are now being UNSET... */
+                                                       *mode_users[1] = '-';
+                                                       for (unsigned int x = 0; x < modectr; x++)
+                                                       {
+                                                               params.push_back(mode_users[x]);
+                                                       }
+                                                       // tell everyone to bounce the modes. bad modes, bad!
+                                                       DoOneToMany(Srv->GetServerName(),"FMODE",params);
+                                               }
+                                               strcpy(mode_users[1],"+");
+                                               modectr = 2;
                                        }
-                                       strcpy(mode_users[1],"+");
-                                       modectr = 2;
                                }
                        }
                }
@@ -931,10 +1060,6 @@ class TreeSocket : public InspSocket
                        return true;
                // NICK age nick host dhost ident +modes ip :gecos
                //       0   1    2    3      4     5    6   7
-               std::string nick = params[1];
-               std::string host = params[2];
-               std::string dhost = params[3];
-               std::string ident = params[4];
                time_t age = atoi(params[0].c_str());
                std::string modes = params[5];
                while (*(modes.c_str()) == '+')
@@ -943,17 +1068,15 @@ class TreeSocket : public InspSocket
                        m++;
                        modes = m;
                }
-               std::string ip = params[6];
-               std::string gecos = params[7];
-               char* tempnick = (char*)nick.c_str();
-               log(DEBUG,"Introduce client %s!%s@%s",tempnick,ident.c_str(),host.c_str());
+               char* tempnick = (char*)params[1].c_str();
+               log(DEBUG,"Introduce client %s!%s@%s",tempnick,params[4].c_str(),params[2].c_str());
                
                user_hash::iterator iter;
                iter = clientlist.find(tempnick);
                if (iter != clientlist.end())
                {
                        // nick collision
-                       log(DEBUG,"Nick collision on %s!%s@%s: %lu %lu",tempnick,ident.c_str(),host.c_str(),(unsigned long)age,(unsigned long)iter->second->age);
+                       log(DEBUG,"Nick collision on %s!%s@%s: %lu %lu",tempnick,params[4].c_str(),params[2].c_str(),(unsigned long)age,(unsigned long)iter->second->age);
                        this->WriteLine(":"+Srv->GetServerName()+" KILL "+tempnick+" :Nickname collision");
                        return true;
                }
@@ -961,15 +1084,15 @@ class TreeSocket : public InspSocket
                clientlist[tempnick] = new userrec();
                clientlist[tempnick]->fd = FD_MAGIC_NUMBER;
                strlcpy(clientlist[tempnick]->nick, tempnick,NICKMAX);
-               strlcpy(clientlist[tempnick]->host, host.c_str(),160);
-               strlcpy(clientlist[tempnick]->dhost, dhost.c_str(),160);
+               strlcpy(clientlist[tempnick]->host, params[2].c_str(),160);
+               strlcpy(clientlist[tempnick]->dhost, params[3].c_str(),160);
                clientlist[tempnick]->server = (char*)FindServerNamePtr(source.c_str());
-               strlcpy(clientlist[tempnick]->ident, ident.c_str(),IDENTMAX);
-               strlcpy(clientlist[tempnick]->fullname, gecos.c_str(),MAXGECOS);
+               strlcpy(clientlist[tempnick]->ident, params[4].c_str(),IDENTMAX);
+               strlcpy(clientlist[tempnick]->fullname, params[7].c_str(),MAXGECOS);
                clientlist[tempnick]->registered = 7;
                clientlist[tempnick]->signon = age;
                strlcpy(clientlist[tempnick]->modes, modes.c_str(),53);
-               strlcpy(clientlist[tempnick]->ip,ip.c_str(),16);
+               inet_aton(params[6].c_str(),&clientlist[tempnick]->ip4);
 
                ucrec a;
                a.channel = NULL;
@@ -979,10 +1102,17 @@ class TreeSocket : public InspSocket
 
                if (!this->bursting)
                {
-                       WriteOpers("*** Client connecting at %s: %s!%s@%s [%s]",clientlist[tempnick]->server,clientlist[tempnick]->nick,clientlist[tempnick]->ident,clientlist[tempnick]->host,clientlist[tempnick]->ip);
+                       WriteOpers("*** Client connecting at %s: %s!%s@%s [%s]",clientlist[tempnick]->server,clientlist[tempnick]->nick,clientlist[tempnick]->ident,clientlist[tempnick]->host,(char*)inet_ntoa(clientlist[tempnick]->ip4));
                }
                params[7] = ":" + params[7];
                DoOneToAllButSender(source,"NICK",params,source);
+
+               // Increment the Source Servers User Count..
+               TreeServer* SourceServer = FindServer(source);
+               if (SourceServer) {
+                       SourceServer->AddUserCount();
+               }
+
                return true;
        }
 
@@ -995,10 +1125,10 @@ class TreeSocket : public InspSocket
                log(DEBUG,"Sending FJOINs to other server for %s",c->name);
                char list[MAXBUF];
                snprintf(list,MAXBUF,":%s FJOIN %s %lu",Srv->GetServerName().c_str(),c->name,(unsigned long)c->age);
-               std::vector<char*> *ulist = c->GetUsers();
-               for (unsigned int i = 0; i < ulist->size(); i++)
+               std::map<char*,char*> *ulist = c->GetUsers();
+               for (std::map<char*,char*>::iterator i = ulist->begin(); i != ulist->end(); i++)
                {
-                       char* o = (*ulist)[i];
+                       char* o = i->second;
                        userrec* otheruser = (userrec*)o;
                        strlcat(list," ",MAXBUF);
                        strlcat(list,cmode(otheruser,c),MAXBUF);
@@ -1021,45 +1151,46 @@ class TreeSocket : public InspSocket
        void SendXLines(TreeServer* Current)
        {
                char data[MAXBUF];
+               const char* sn = Srv->GetServerName().c_str();
                /* Yes, these arent too nice looking, but they get the job done */
                for (std::vector<ZLine>::iterator i = zlines.begin(); i != zlines.end(); i++)
                {
-                       snprintf(data,MAXBUF,":%s ADDLINE Z %s %s %lu %lu :%s",Srv->GetServerName().c_str(),i->ipaddr,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
+                       snprintf(data,MAXBUF,":%s ADDLINE Z %s %s %lu %lu :%s",sn,i->ipaddr,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
                        this->WriteLine(data);
                }
                for (std::vector<QLine>::iterator i = qlines.begin(); i != qlines.end(); i++)
                {
-                       snprintf(data,MAXBUF,":%s ADDLINE Q %s %s %lu %lu :%s",Srv->GetServerName().c_str(),i->nick,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
+                       snprintf(data,MAXBUF,":%s ADDLINE Q %s %s %lu %lu :%s",sn,i->nick,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
                        this->WriteLine(data);
                }
                for (std::vector<GLine>::iterator i = glines.begin(); i != glines.end(); i++)
                {
-                       snprintf(data,MAXBUF,":%s ADDLINE G %s %s %lu %lu :%s",Srv->GetServerName().c_str(),i->hostmask,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
+                       snprintf(data,MAXBUF,":%s ADDLINE G %s %s %lu %lu :%s",sn,i->hostmask,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
                        this->WriteLine(data);
                }
                for (std::vector<ELine>::iterator i = elines.begin(); i != elines.end(); i++)
                {
-                       snprintf(data,MAXBUF,":%s ADDLINE E %s %s %lu %lu :%s",Srv->GetServerName().c_str(),i->hostmask,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
+                       snprintf(data,MAXBUF,":%s ADDLINE E %s %s %lu %lu :%s",sn,i->hostmask,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
                        this->WriteLine(data);
                }
                for (std::vector<ZLine>::iterator i = pzlines.begin(); i != pzlines.end(); i++)
                {
-                       snprintf(data,MAXBUF,":%s ADDLINE Z %s %s %lu %lu :%s",Srv->GetServerName().c_str(),i->ipaddr,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
+                       snprintf(data,MAXBUF,":%s ADDLINE Z %s %s %lu %lu :%s",sn,i->ipaddr,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
                        this->WriteLine(data);
                }
                for (std::vector<QLine>::iterator i = pqlines.begin(); i != pqlines.end(); i++)
                {
-                       snprintf(data,MAXBUF,":%s ADDLINE Q %s %s %lu %lu :%s",Srv->GetServerName().c_str(),i->nick,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
+                       snprintf(data,MAXBUF,":%s ADDLINE Q %s %s %lu %lu :%s",sn,i->nick,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
                        this->WriteLine(data);
                }
                for (std::vector<GLine>::iterator i = pglines.begin(); i != pglines.end(); i++)
                {
-                       snprintf(data,MAXBUF,":%s ADDLINE G %s %s %lu %lu :%s",Srv->GetServerName().c_str(),i->hostmask,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
+                       snprintf(data,MAXBUF,":%s ADDLINE G %s %s %lu %lu :%s",sn,i->hostmask,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
                        this->WriteLine(data);
                }
                for (std::vector<ELine>::iterator i = pelines.begin(); i != pelines.end(); i++)
                {
-                       snprintf(data,MAXBUF,":%s ADDLINE E %s %s %lu %lu :%s",Srv->GetServerName().c_str(),i->hostmask,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
+                       snprintf(data,MAXBUF,":%s ADDLINE E %s %s %lu %lu :%s",sn,i->hostmask,i->source,(unsigned long)i->set_time,(unsigned long)i->duration,i->reason);
                        this->WriteLine(data);
                }
        }
@@ -1069,27 +1200,28 @@ class TreeSocket : public InspSocket
        {
                char data[MAXBUF];
                std::deque<std::string> list;
+               const char* sn = Srv->GetServerName().c_str();
                for (chan_hash::iterator c = chanlist.begin(); c != chanlist.end(); c++)
                {
                        SendFJoins(Current, c->second);
-                       snprintf(data,MAXBUF,":%s FMODE %s +%s",Srv->GetServerName().c_str(),c->second->name,chanmodes(c->second));
+                       snprintf(data,MAXBUF,":%s FMODE %s +%s",sn,c->second->name,chanmodes(c->second,true));
                        this->WriteLine(data);
                        if (*c->second->topic)
                        {
-                               snprintf(data,MAXBUF,":%s FTOPIC %s %lu %s :%s",Srv->GetServerName().c_str(),c->second->name,(unsigned long)c->second->topicset,c->second->setby,c->second->topic);
+                               snprintf(data,MAXBUF,":%s FTOPIC %s %lu %s :%s",sn,c->second->name,(unsigned long)c->second->topicset,c->second->setby,c->second->topic);
                                this->WriteLine(data);
                        }
                        for (BanList::iterator b = c->second->bans.begin(); b != c->second->bans.end(); b++)
                        {
-                               snprintf(data,MAXBUF,":%s FMODE %s +b %s",Srv->GetServerName().c_str(),c->second->name,b->data);
+                               snprintf(data,MAXBUF,":%s FMODE %s +b %s",sn,c->second->name,b->data);
                                this->WriteLine(data);
                        }
-                       FOREACH_MOD OnSyncChannel(c->second,(Module*)TreeProtocolModule,(void*)this);
+                       FOREACH_MOD(I_OnSyncChannel,OnSyncChannel(c->second,(Module*)TreeProtocolModule,(void*)this));
                        list.clear();
                        c->second->GetExtList(list);
                        for (unsigned int j = 0; j < list.size(); j++)
                        {
-                               FOREACH_MOD OnSyncChannelMetaData(c->second,(Module*)TreeProtocolModule,(void*)this,list[j]);
+                               FOREACH_MOD(I_OnSyncChannelMetaData,OnSyncChannelMetaData(c->second,(Module*)TreeProtocolModule,(void*)this,list[j]));
                        }
                }
        }
@@ -1103,18 +1235,22 @@ class TreeSocket : public InspSocket
                {
                        if (u->second->registered == 7)
                        {
-                               snprintf(data,MAXBUF,":%s NICK %lu %s %s %s %s +%s %s :%s",u->second->server,(unsigned long)u->second->age,u->second->nick,u->second->host,u->second->dhost,u->second->ident,u->second->modes,u->second->ip,u->second->fullname);
+                               snprintf(data,MAXBUF,":%s NICK %lu %s %s %s %s +%s %s :%s",u->second->server,(unsigned long)u->second->age,u->second->nick,u->second->host,u->second->dhost,u->second->ident,u->second->modes,(char*)inet_ntoa(u->second->ip4),u->second->fullname);
                                this->WriteLine(data);
-                               if (strchr(u->second->modes,'o'))
+                               if (*u->second->oper)
                                {
                                        this->WriteLine(":"+std::string(u->second->nick)+" OPERTYPE "+std::string(u->second->oper));
                                }
-                               FOREACH_MOD OnSyncUser(u->second,(Module*)TreeProtocolModule,(void*)this);
+                               if (*u->second->awaymsg)
+                               {
+                                       this->WriteLine(":"+std::string(u->second->nick)+" AWAY :"+std::string(u->second->awaymsg));
+                               }
+                               FOREACH_MOD(I_OnSyncUser,OnSyncUser(u->second,(Module*)TreeProtocolModule,(void*)this));
                                list.clear();
                                u->second->GetExtList(list);
                                for (unsigned int j = 0; j < list.size(); j++)
                                {
-                                       FOREACH_MOD OnSyncUserMetaData(u->second,(Module*)TreeProtocolModule,(void*)this,list[j]);
+                                       FOREACH_MOD(I_OnSyncUserMetaData,OnSyncUserMetaData(u->second,(Module*)TreeProtocolModule,(void*)this,list[j]));
                                }
                        }
                }
@@ -1138,6 +1274,7 @@ class TreeSocket : public InspSocket
                /* Send everything else (channel modes, xlines etc) */
                this->SendChannelModes(s);
                this->SendXLines(s);
+               FOREACH_MOD(I_OnSyncOtherMetaData,OnSyncOtherMetaData((Module*)TreeProtocolModule,(void*)this));
                this->WriteLine("ENDBURST");
                Srv->SendOpers("*** Finished bursting to \2"+s->GetName()+"\2.");
        }
@@ -1153,38 +1290,42 @@ class TreeSocket : public InspSocket
         virtual bool OnDataReady()
        {
                char* data = this->Read();
-               if (data)
+               /* Check that the data read is a valid pointer and it has some content */
+               if (data && *data)
                {
-                       this->in_buffer += data;
+                       this->in_buffer.append(data);
                        /* While there is at least one new line in the buffer,
                         * do something useful (we hope!) with it.
                         */
                        while (in_buffer.find("\n") != std::string::npos)
                        {
-                               char* line = (char*)in_buffer.c_str();
-                               std::string ret = "";
-                               while ((*line != '\n') && (strlen(line)))
-                               {
-                                       ret = ret + *line;
-                                       line++;
-                               }
-                               if ((*line == '\n') || (*line == '\r'))
-                                       line++;
-                               in_buffer = line;
+                               std::string ret = in_buffer.substr(0,in_buffer.find("\n")-1);
+                               in_buffer = in_buffer.substr(in_buffer.find("\n")+1,in_buffer.length()-in_buffer.find("\n"));
+                               if (ret.find("\r") != std::string::npos)
+                                       ret = in_buffer.substr(0,in_buffer.find("\r")-1);
                                /* Process this one, abort if it
                                 * didnt return true.
                                 */
-                               if (this->ctx)
+                               if (this->ctx_in)
                                {
                                        char out[1024];
                                        char result[1024];
+                                       memset(result,0,1024);
+                                       memset(out,0,1024);
                                        log(DEBUG,"Original string '%s'",ret.c_str());
-                                       int nbytes = from64tobits(out, ret.c_str(), 1024);
-                                       log(DEBUG,"m_spanningtree: decrypt %d bytes",nbytes);
-                                       ctx->Decrypt(out, result, nbytes, 0);
-                                       for (int t = 0; t < nbytes; t++)
-                                               if (result[t] == '\7') result[t] = 0;
-                                       ret = result;
+                                       /* ERROR + CAPAB is still allowed unencryped */
+                                       if ((ret.substr(0,7) != "ERROR :") && (ret.substr(0,6) != "CAPAB "))
+                                       {
+                                               int nbytes = from64tobits(out, ret.c_str(), 1024);
+                                               if ((nbytes > 0) && (nbytes < 1024))
+                                               {
+                                                       log(DEBUG,"m_spanningtree: decrypt %d bytes",nbytes);
+                                                       ctx_in->Decrypt(out, result, nbytes, 1);
+                                                       for (int t = 0; t < nbytes; t++)
+                                                               if (result[t] == '\7') result[t] = 0;
+                                                       ret = result;
+                                               }
+                                       }
                                }
                                if (!this->ProcessLine(ret))
                                {
@@ -1192,17 +1333,20 @@ class TreeSocket : public InspSocket
                                }
                        }
                }
+               /* EAGAIN returns an empty but non-NULL string, so this
+                * evaluates to TRUE for EAGAIN but to FALSE for EOF.
+                */
                return (data != NULL);
        }
 
        int WriteLine(std::string line)
        {
                log(DEBUG,"OUT: %s",line.c_str());
-               if (this->ctx)
+               if (this->ctx_out)
                {
                        log(DEBUG,"AES context");
-                       char result[1024];
-                       char result64[1024];
+                       char result[10240];
+                       char result64[10240];
                        if (this->keylength)
                        {
                                while (line.length() % this->keylength != 0)
@@ -1211,10 +1355,11 @@ class TreeSocket : public InspSocket
                                        line = line + "\7";
                                }
                        }
-                       ctx->Encrypt(line.c_str(), result, line.length(),0);
-                       to64frombits((unsigned char*)result64,
-                                       (unsigned char*)result,
-                                       line.length());
+                       unsigned int ll = line.length();
+                       log(DEBUG,"Plaintext line with padding = %d chars",ll);
+                       ctx_out->Encrypt(line.c_str(), result, ll, 1);
+                       log(DEBUG,"Encrypted.");
+                       to64frombits((unsigned char*)result64,(unsigned char*)result,ll);
                        line = result64;
                        log(DEBUG,"Encrypted: %s",line.c_str());
                        //int from64tobits(char *out, const char *in, int maxlen);
@@ -1351,20 +1496,24 @@ class TreeSocket : public InspSocket
                TreeServer* ServerSource = FindServer(prefix);
                if (ServerSource)
                {
-                       if (*(params[0].c_str()) == '#')
+                       if (params[0] == "*")
+                       {
+                               FOREACH_MOD(I_OnDecodeMetaData,OnDecodeMetaData(TYPE_OTHER,NULL,params[1],params[2]));
+                       }
+                       else if (*(params[0].c_str()) == '#')
                        {
                                chanrec* c = Srv->FindChannel(params[0]);
                                if (c)
                                {
-                                       FOREACH_MOD OnDecodeMetaData(TYPE_CHANNEL,c,params[1],params[2]);
+                                       FOREACH_MOD(I_OnDecodeMetaData,OnDecodeMetaData(TYPE_CHANNEL,c,params[1],params[2]));
                                }
                        }
-                       else
+                       else if (*(params[0].c_str()) != '#')
                        {
                                userrec* u = Srv->FindNick(params[0]);
                                if (u)
                                {
-                                       FOREACH_MOD OnDecodeMetaData(TYPE_USER,u,params[1],params[2]);
+                                       FOREACH_MOD(I_OnDecodeMetaData,OnDecodeMetaData(TYPE_USER,u,params[1],params[2]));
                                }
                        }
                }
@@ -1404,42 +1553,42 @@ class TreeSocket : public InspSocket
        {
                if (params.size() < 6)
                        return true;
-               std::string linetype = params[0]; /* Z, Q, E, G, K */
-               std::string mask = params[1]; /* Line type dependent */
-               std::string source = params[2]; /* may not be online or may be a server */
-               std::string settime = params[3]; /* EPOCH time set */
-               std::string duration = params[4]; /* Duration secs */
-               std::string reason = params[5];
 
-               switch (*(linetype.c_str()))
+               bool propogate = false;
+
+               switch (*(params[0].c_str()))
                {
                        case 'Z':
-                               add_zline(atoi(duration.c_str()), source.c_str(), reason.c_str(), mask.c_str());
-                               zline_set_creation_time((char*)mask.c_str(), atoi(settime.c_str()));
+                               propogate = add_zline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
+                               zline_set_creation_time((char*)params[1].c_str(), atoi(params[3].c_str()));
                        break;
                        case 'Q':
-                               add_qline(atoi(duration.c_str()), source.c_str(), reason.c_str(), mask.c_str());
-                               qline_set_creation_time((char*)mask.c_str(), atoi(settime.c_str()));
+                               propogate = add_qline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
+                               qline_set_creation_time((char*)params[1].c_str(), atoi(params[3].c_str()));
                        break;
                        case 'E':
-                               add_eline(atoi(duration.c_str()), source.c_str(), reason.c_str(), mask.c_str());
-                               eline_set_creation_time((char*)mask.c_str(), atoi(settime.c_str()));
+                               propogate = add_eline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
+                               eline_set_creation_time((char*)params[1].c_str(), atoi(params[3].c_str()));
                        break;
                        case 'G':
-                               add_gline(atoi(duration.c_str()), source.c_str(), reason.c_str(), mask.c_str());
-                               gline_set_creation_time((char*)mask.c_str(), atoi(settime.c_str()));
+                               propogate = add_gline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
+                               gline_set_creation_time((char*)params[1].c_str(), atoi(params[3].c_str()));
                        break;
                        case 'K':
-                               add_kline(atoi(duration.c_str()), source.c_str(), reason.c_str(), mask.c_str());
+                               propogate = add_kline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
                        break;
                        default:
                                /* Just in case... */
-                               Srv->SendOpers("*** \2WARNING\2: Invalid xline type '"+linetype+"' sent by server "+prefix+", ignored!");
+                               Srv->SendOpers("*** \2WARNING\2: Invalid xline type '"+params[0]+"' sent by server "+prefix+", ignored!");
+                               propogate = false;
                        break;
                }
                /* Send it on its way */
-               params[5] = ":" + params[5];
-               DoOneToAllButSender(prefix,"ADDLINE",params,prefix);
+               if (propogate)
+               {
+                       params[5] = ":" + params[5];
+                       DoOneToAllButSender(prefix,"ADDLINE",params,prefix);
+               }
                return true;
        }
 
@@ -1470,7 +1619,7 @@ class TreeSocket : public InspSocket
                        if (params.size() == 1)
                        {
                                userrec* x = Srv->FindNick(params[0]);
-                               if (x->fd > -1)
+                               if ((x) && (x->fd > -1))
                                {
                                        userrec* x = Srv->FindNick(params[0]);
                                        log(DEBUG,"Got IDLE");
@@ -1496,7 +1645,7 @@ class TreeSocket : public InspSocket
                        {
                                std::string who_did_the_whois = params[0];
                                userrec* who_to_send_to = Srv->FindNick(who_did_the_whois);
-                               if (who_to_send_to->fd > -1)
+                               if ((who_to_send_to) && (who_to_send_to->fd > -1))
                                {
                                        log(DEBUG,"Got final IDLE");
                                        // an incoming reply to a whois we sent out
@@ -1543,6 +1692,7 @@ class TreeSocket : public InspSocket
                if (CheckDupe)
                {
                        this->WriteLine("ERROR :Server "+servername+" already exists on server "+CheckDupe->GetParent()->GetName()+"!");
+                       Srv->SendOpers("*** Server connection from \2"+servername+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName());
                        return false;
                }
                TreeServer* Node = new TreeServer(servername,description,ParentOfThis,NULL);
@@ -1563,6 +1713,7 @@ class TreeSocket : public InspSocket
                if (hops)
                {
                        this->WriteLine("ERROR :Server too far away for authentication");
+                       Srv->SendOpers("*** Server connection from \2"+servername+"\2 denied, server is too far away for authentication");
                        return false;
                }
                std::string description = params[3];
@@ -1574,6 +1725,7 @@ class TreeSocket : public InspSocket
                                if (CheckDupe)
                                {
                                        this->WriteLine("ERROR :Server "+servername+" already exists on server "+CheckDupe->GetParent()->GetName()+"!");
+                                       Srv->SendOpers("*** Server connection from \2"+servername+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName());
                                        return false;
                                }
                                // Begin the sync here. this kickstarts the
@@ -1594,6 +1746,7 @@ class TreeSocket : public InspSocket
                        }
                }
                this->WriteLine("ERROR :Invalid credentials");
+               Srv->SendOpers("*** Server connection from \2"+servername+"\2 denied, invalid link credentials");
                return false;
        }
 
@@ -1607,6 +1760,7 @@ class TreeSocket : public InspSocket
                if (hops)
                {
                        this->WriteLine("ERROR :Server too far away for authentication");
+                       Srv->SendOpers("*** Server connection from \2"+servername+"\2 denied, server is too far away for authentication");
                        return false;
                }
                std::string description = params[3];
@@ -1618,9 +1772,20 @@ class TreeSocket : public InspSocket
                                if (CheckDupe)
                                {
                                        this->WriteLine("ERROR :Server "+servername+" already exists on server "+CheckDupe->GetParent()->GetName()+"!");
+                                       Srv->SendOpers("*** Server connection from \2"+servername+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName());
                                        return false;
                                }
-                               Srv->SendOpers("*** Verified incoming server connection from \002"+servername+"\002["+this->GetIP()+"] ("+description+")");
+                               /* If the config says this link is encrypted, but the remote side
+                                * hasnt bothered to send the AES command before SERVER, then we
+                                * boot them off as we MUST have this connection encrypted.
+                                */
+                               if ((x->EncryptionKey != "") && (!this->ctx_in))
+                               {
+                                       this->WriteLine("ERROR :This link requires AES encryption to be enabled. Plaintext connection refused.");
+                                       Srv->SendOpers("*** Server connection from \2"+servername+"\2 denied, remote server did not enable AES.");
+                                       return false;
+                               }
+                               Srv->SendOpers("*** Verified incoming server connection from \002"+servername+"\002["+(x->HiddenFromStats ? "<hidden>" : this->GetIP())+"] ("+description+")");
                                this->InboundServerName = servername;
                                this->InboundDescription = description;
                                // this is good. Send our details: Our server name and description and hopcount of 0,
@@ -1632,6 +1797,7 @@ class TreeSocket : public InspSocket
                        }
                }
                this->WriteLine("ERROR :Invalid credentials");
+               Srv->SendOpers("*** Server connection from \2"+servername+"\2 denied, invalid link credentials");
                return false;
        }
 
@@ -1694,6 +1860,7 @@ class TreeSocket : public InspSocket
                if (line == "")
                        return true;
                Srv->Log(DEBUG,"IN: "+line);
+               
                std::deque<std::string> params;
                this->Split(line,true,params);
                std::string command = "";
@@ -1714,7 +1881,7 @@ class TreeSocket : public InspSocket
                        params.pop_front();
                }
 
-               if ((!this->ctx) && (command == "AES"))
+               if ((!this->ctx_in) && (command == "AES"))
                {
                         std::string sserv = params[0];
                         for (std::vector<Link>::iterator x = LinkBlocks.begin(); x < LinkBlocks.end(); x++)
@@ -1726,9 +1893,9 @@ class TreeSocket : public InspSocket
                         }
                         return true;
                }
-               else if ((this->ctx) && (command == "AES"))
+               else if ((this->ctx_in) && (command == "AES"))
                {
-                       WriteOpers("\2AES\2: Encryption already enabled on this connection yet %s is trying to enable it twice!",params[0].c_str());
+                       WriteOpers("*** \2AES\2: Encryption already enabled on this connection yet %s is trying to enable it twice!",params[0].c_str());
                }
 
                switch (this->LinkState)
@@ -1741,7 +1908,11 @@ class TreeSocket : public InspSocket
                                // replies with theirs if its happy, then if the initiator is happy,
                                // it starts to send its net sync, which starts the merge, otherwise
                                // it sends an ERROR.
-                               if (command == "SERVER")
+                               if (command == "PASS")
+                               {
+                                       /* Silently ignored */
+                               }
+                               else if (command == "SERVER")
                                {
                                        return this->Inbound_Server(params);
                                }
@@ -1749,6 +1920,25 @@ class TreeSocket : public InspSocket
                                {
                                        return this->Error(params);
                                }
+                               else if (command == "USER")
+                               {
+                                       this->WriteLine("ERROR :Client connections to this port are prohibited.");
+                                       return false;
+                               }
+                               else if (command == "CAPAB")
+                               {
+                                       return this->Capab(params);
+                               }
+                               else if ((command == "U") || (command == "S"))
+                               {
+                                       this->WriteLine("ERROR :Cannot use the old-style mesh linking protocol with m_spanningtree.so!");
+                                       return false;
+                               }
+                               else
+                               {
+                                       this->WriteLine("ERROR :Invalid command in negotiation phase.");
+                                       return false;
+                               }
                        break;
                        case WAIT_AUTH_2:
                                // Waiting for start of other side's netmerge to say they liked our
@@ -1759,6 +1949,11 @@ class TreeSocket : public InspSocket
                                        // silently ignore.
                                        return true;
                                }
+                               else if ((command == "U") || (command == "S"))
+                               {
+                                       this->WriteLine("ERROR :Cannot use the old-style mesh linking protocol with m_spanningtree.so!");
+                                       return false;
+                               }
                                else if (command == "BURST")
                                {
                                        this->LinkState = CONNECTED;
@@ -1777,6 +1972,10 @@ class TreeSocket : public InspSocket
                                {
                                        return this->Error(params);
                                }
+                               else if (command == "CAPAB")
+                               {
+                                       return this->Capab(params);
+                               }
                                
                        break;
                        case LISTENER:
@@ -1815,16 +2014,16 @@ class TreeSocket : public InspSocket
                                        if ((!route_back_again) || (route_back_again->GetSocket() != this))
                                        {
                                                if (route_back_again)
-                                               {
-                                                       WriteOpers("Protocol violation: Fake direction in command '%s' from connection '%s'",line.c_str(),this->GetName().c_str());
-                                               }
-                                               else
-                                               {
-                                                       WriteOpers("Protocol violation: Invalid source '%s' in command '%s' from connection '%s'",direction.c_str(),line.c_str(),this->GetName().c_str());
-                                               }
-                                               
+                                                       log(DEBUG,"Protocol violation: Fake direction in command '%s' from connection '%s'",line.c_str(),this->GetName().c_str());
                                                return true;
                                        }
+
+                                       /* Fix by brain:
+                                        * When there is activity on the socket, reset the ping counter so
+                                        * that we're not wasting bandwidth pinging an active server.
+                                        */                     
+                                       route_back_again->SetNextPingTime(time(NULL) + 120);
+                                       route_back_again->SetPingFlag();
                                }
                                
                                if (command == "SVSMODE")
@@ -1949,6 +2148,32 @@ class TreeSocket : public InspSocket
                                        }
                                        if (who)
                                        {
+                                               if ((command == "NICK") && (params.size() > 0))
+                                               {
+                                                       /* On nick messages, check that the nick doesnt
+                                                        * already exist here. If it does, kill their copy,
+                                                        * and our copy.
+                                                        */
+                                                       userrec* x = Srv->FindNick(params[0]);
+                                                       if (x)
+                                                       {
+                                                               std::deque<std::string> p;
+                                                               p.push_back(params[0]);
+                                                               p.push_back("Nickname collision ("+prefix+" -> "+params[0]+")");
+                                                               DoOneToMany(Srv->GetServerName(),"KILL",p);
+                                                               p.clear();
+                                                               p.push_back(prefix);
+                                                               p.push_back("Nickname collision");
+                                                               DoOneToMany(Srv->GetServerName(),"KILL",p);
+                                                               Srv->QuitUser(x,"Nickname collision ("+prefix+" -> "+params[0]+")");
+                                                               userrec* y = Srv->FindNick(prefix);
+                                                               if (y)
+                                                               {
+                                                                       Srv->QuitUser(y,"Nickname collision");
+                                                               }
+                                                               return DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
+                                                       }
+                                               }
                                                // its a user
                                                target = who->server;
                                                char* strparams[127];
@@ -2013,6 +2238,7 @@ class TreeSocket : public InspSocket
                {
                        Squit(s,"Remote host closed the connection");
                }
+               WriteOpers("Server '\2%s\2' closed the connection.",quitserver.c_str());
        }
 
        virtual int OnIncomingConnection(int newsock, char* ip)
@@ -2038,11 +2264,10 @@ void AddThisServer(TreeServer* server, std::deque<TreeServer*> &list)
 // returns a list of DIRECT servernames for a specific channel
 void GetListOfServersForChannel(chanrec* c, std::deque<TreeServer*> &list)
 {
-       std::vector<char*> *ulist = c->GetUsers();
-       unsigned int ucount = ulist->size();
-       for (unsigned int i = 0; i < ucount; i++)
+       std::map<char*,char*> *ulist = c->GetUsers();
+       for (std::map<char*,char*>::iterator i = ulist->begin(); i != ulist->end(); i++)
        {
-               char* o = (*ulist)[i];
+               char* o = i->second;
                userrec* otheruser = (userrec*)o;
                if (otheruser->fd < 0)
                {
@@ -2061,6 +2286,11 @@ bool DoOneToAllButSenderRaw(std::string data, std::string omit, std::string pref
        {
                if ((params.size() >= 2) && (*(params[0].c_str()) != '$'))
                {
+                       /* Prefixes */
+                       if ((*(params[0].c_str()) == '@') || (*(params[0].c_str()) == '%') || (*(params[0].c_str()) == '+'))
+                       {
+                               params[0] = params[0].substr(1, params[0].length()-1);
+                       }
                        if (*(params[0].c_str()) != '#')
                        {
                                // special routing for private messages/notices
@@ -2215,6 +2445,8 @@ void ReadConfiguration(bool rebind)
                        }
                }
        }
+       FlatLinks = Conf->ReadFlag("options","flatlinks",0);
+       HideULines = Conf->ReadFlag("options","hideulines",0);
        LinkBlocks.clear();
        for (int j =0; j < Conf->Enumerate("link"); j++)
        {
@@ -2226,6 +2458,7 @@ void ReadConfiguration(bool rebind)
                L.RecvPass = Conf->ReadValue("link","recvpass",j);
                L.AutoConnect = Conf->ReadInteger("link","autoconnect",j,true);
                L.EncryptionKey =  Conf->ReadValue("link","encryptionkey",j);
+               L.HiddenFromStats = Conf->ReadFlag("link","hidden",j);
                L.NextConnectTime = time(NULL) + L.AutoConnect;
                /* Bugfix by brain, do not allow people to enter bad configurations */
                if ((L.RecvPass != "") && (L.SendPass != "") && (L.Name != "") && (L.Port))
@@ -2235,7 +2468,22 @@ void ReadConfiguration(bool rebind)
                }
                else
                {
-                       log(DEFAULT,"m_spanningtree: Invalid configuration for server '%s', ignored!",L.Name.c_str());
+                       if (L.RecvPass == "")
+                       {
+                               log(DEFAULT,"Invalid configuration for server '%s', recvpass not defined!",L.Name.c_str());
+                       }
+                       else if (L.SendPass == "")
+                       {
+                               log(DEFAULT,"Invalid configuration for server '%s', sendpass not defined!",L.Name.c_str());
+                       }
+                       else if (L.Name == "")
+                       {
+                               log(DEFAULT,"Invalid configuration, link tag without a name!");
+                       }
+                       else if (!L.Port)
+                       {
+                               log(DEFAULT,"Invalid configuration for server '%s', no port specified!",L.Name.c_str());
+                       }
                }
        }
        delete Conf;
@@ -2247,11 +2495,14 @@ class ModuleSpanningTree : public Module
        std::vector<TreeSocket*> Bindings;
        int line;
        int NumServers;
+       unsigned int max_local;
+       unsigned int max_global;
+       cmd_rconnect* command_rconnect;
 
  public:
 
        ModuleSpanningTree(Server* Me)
-               : Module::Module(Me)
+               : Module::Module(Me), max_local(0), max_global(0)
        {
                Srv = Me;
                Bindings.clear();
@@ -2260,6 +2511,9 @@ class ModuleSpanningTree : public Module
                TreeRoot = new TreeServer(Srv->GetServerName(),Srv->GetServerDescription());
 
                ReadConfiguration(true);
+
+               command_rconnect = new cmd_rconnect(this);
+               Srv->AddCommand(command_rconnect);
        }
 
        void ShowLinks(TreeServer* Current, userrec* user, int hops)
@@ -2271,9 +2525,19 @@ class ModuleSpanningTree : public Module
                }
                for (unsigned int q = 0; q < Current->ChildCount(); q++)
                {
-                       ShowLinks(Current->GetChild(q),user,hops+1);
+                       if ((HideULines) && (Srv->IsUlined(Current->GetName())))
+                       {
+                               if (*user->oper)
+                               {
+                                        ShowLinks(Current->GetChild(q),user,hops+1);
+                               }
+                       }
+                       else
+                       {
+                               ShowLinks(Current->GetChild(q),user,hops+1);
+                       }
                }
-               WriteServ(user->fd,"364 %s %s %s :%d %s",user->nick,Current->GetName().c_str(),Parent.c_str(),hops,Current->GetDesc().c_str());
+               WriteServ(user->fd,"364 %s %s %s :%d %s",user->nick,Current->GetName().c_str(),(FlatLinks && (!*user->oper)) ? Srv->GetServerName().c_str() : Parent.c_str(),(FlatLinks && (!*user->oper)) ? 0 : hops,Current->GetDesc().c_str());
        }
 
        int CountLocalServs()
@@ -2295,11 +2559,19 @@ class ModuleSpanningTree : public Module
 
        void HandleLusers(char** parameters, int pcnt, userrec* user)
        {
+               /* Only update these when someone wants to see them, more efficient */
+               if ((unsigned int)local_count() > max_local)
+                       max_local = local_count();
+               if (clientlist.size() > max_global)
+                       max_global = clientlist.size();
+
                WriteServ(user->fd,"251 %s :There are %d users and %d invisible on %d servers",user->nick,usercnt()-usercount_invisible(),usercount_invisible(),this->CountServs());
                WriteServ(user->fd,"252 %s %d :operator(s) online",user->nick,usercount_opers());
                WriteServ(user->fd,"253 %s %d :unknown connections",user->nick,usercount_unknown());
                WriteServ(user->fd,"254 %s %d :channels formed",user->nick,chancount());
                WriteServ(user->fd,"254 %s :I have %d clients and %d servers",user->nick,local_count(),this->CountLocalServs());
+               WriteServ(user->fd,"265 %s :Current Local Users: %d  Max: %d",user->nick,local_count(),max_local);
+               WriteServ(user->fd,"266 %s :Current Global Users: %d  Max: %d",user->nick,clientlist.size(),max_global);
                return;
        }
 
@@ -2313,11 +2585,46 @@ class ModuleSpanningTree : public Module
                        {
                                matrix[line][t] = ' ';
                        }
-                       strlcpy(&matrix[line][depth],Current->GetName().c_str(),80);
+
+                       // For Aligning, we need to work out exactly how deep this thing is, and produce
+                       // a 'Spacer' String to compensate.
+                       char spacer[40];
+
+                       memset(spacer,' ',40);
+                       if ((40 - Current->GetName().length() - depth) > 1) {
+                               spacer[40 - Current->GetName().length() - depth] = '\0';
+                       }
+                       else
+                       {
+                               spacer[5] = '\0';
+                       }
+
+                       float percent;
+                       char text[80];
+                       if (clientlist.size() == 0) {
+                               // If there are no users, WHO THE HELL DID THE /MAP?!?!?!
+                               percent = 0;
+                       }
+                       else
+                       {
+                               percent = ((float)Current->GetUserCount() / (float)clientlist.size()) * 100;
+                       }
+                       snprintf(text, 80, "%s %s%5d [%5.2f%%]", Current->GetName().c_str(), spacer, Current->GetUserCount(), percent);
+                       strlcpy(&matrix[line][depth],text,80);
                        line++;
                        for (unsigned int q = 0; q < Current->ChildCount(); q++)
                        {
-                               ShowMap(Current->GetChild(q),user,depth+2,matrix);
+                               if ((HideULines) && (Srv->IsUlined(Current->GetChild(q)->GetName())))
+                               {
+                                       if (*user->oper)
+                                       {
+                                               ShowMap(Current->GetChild(q),user,(FlatLinks && (!*user->oper)) ? depth : depth+2,matrix);
+                                       }
+                               }
+                               else
+                               {
+                                       ShowMap(Current->GetChild(q),user,(FlatLinks && (!*user->oper)) ? depth : depth+2,matrix);
+                               }
                        }
                }
        }
@@ -2388,12 +2695,18 @@ class ModuleSpanningTree : public Module
                TreeServer* s = FindServerMask(parameters[0]);
                if (s)
                {
+                       if (s == TreeRoot)
+                       {
+                                WriteServ(user->fd,"NOTICE %s :*** SQUIT: Foolish mortal, you cannot make a server SQUIT itself! (%s matches local server name)",user->nick,parameters[0]);
+                               return 1;
+                       }
                        TreeSocket* sock = s->GetSocket();
                        if (sock)
                        {
+                               log(DEBUG,"Splitting server %s",s->GetName().c_str());
                                WriteOpers("*** SQUIT: Server \002%s\002 removed from network by %s",parameters[0],user->nick);
                                sock->Squit(s,"Server quit by "+std::string(user->nick)+"!"+std::string(user->ident)+"@"+std::string(user->host));
-                               sock->Close();
+                               Srv->RemoveSocket(sock);
                        }
                        else
                        {
@@ -2437,22 +2750,23 @@ class ModuleSpanningTree : public Module
                        TreeSocket* sock = serv->GetSocket();
                        if (sock)
                        {
-                               if (curtime >= serv->NextPingTime())
-                               {
-                                       if (serv->AnsweredLastPing())
-                                       {
-                                               sock->WriteLine(":"+Srv->GetServerName()+" PING "+serv->GetName());
-                                               serv->SetNextPingTime(curtime + 60);
-                                       }
-                                       else
-                                       {
-                                               // they didnt answer, boot them
-                                               WriteOpers("*** Server \002%s\002 pinged out",serv->GetName().c_str());
-                                               sock->Squit(serv,"Ping timeout");
-                                               sock->Close();
-                                               return;
-                                       }
-                               }
+                                if (curtime >= serv->NextPingTime())
+                                {               
+                                        if (serv->AnsweredLastPing())
+                                        {               
+                                                sock->WriteLine(":"+Srv->GetServerName()+" PING "+serv->GetName());
+                                                serv->SetNextPingTime(curtime + 120);
+                                        }                       
+                                        else            
+                                        {       
+                                                // they didnt answer, boot them
+                                                WriteOpers("*** Server \002%s\002 pinged out",serv->GetName().c_str());
+                                                sock->Squit(serv,"Ping timeout");
+                                               Srv->RemoveSocket(sock);
+                                                return;
+                                        }
+                                }
+
                        }
                }
        }
@@ -2471,7 +2785,15 @@ class ModuleSpanningTree : public Module
                                        // an autoconnected server is not connected. Check if its time to connect it
                                        WriteOpers("*** AUTOCONNECT: Auto-connecting server \002%s\002 (%lu seconds until next attempt)",x->Name.c_str(),x->AutoConnect);
                                        TreeSocket* newsocket = new TreeSocket(x->IPAddr,x->Port,false,10,x->Name);
-                                       Srv->AddSocket(newsocket);
+                                       if (newsocket->GetState() != I_ERROR)
+                                       {
+                                               Srv->AddSocket(newsocket);
+                                       }
+                                       else
+                                       {
+                                               WriteOpers("*** AUTOCONNECT: Error autoconnecting \002%s\002.",x->Name.c_str());
+                                               delete newsocket;
+                                       }
                                }
                        }
                }
@@ -2485,6 +2807,25 @@ class ModuleSpanningTree : public Module
                {
                        std::string Version = found->GetVersion();
                        WriteServ(user->fd,"351 %s :%s",user->nick,Version.c_str());
+                       if (found == TreeRoot)
+                       {
+                               std::stringstream out(Config->data005);
+                               std::string token = "";
+                               std::string line5 = "";
+                               int token_counter = 0;
+                               while (!out.eof())
+                               {
+                                       out >> token;
+                                       line5 = line5 + token + " ";   
+                                       token_counter++;
+                                       if ((token_counter >= 13) || (out.eof() == true))
+                                       {
+                                               WriteServ(user->fd,"005 %s %s:are supported by this server",user->nick,line5.c_str());
+                                               line5 = "";
+                                               token_counter = 0;
+                                       }
+                               }
+                       }
                }
                else
                {
@@ -2502,9 +2843,17 @@ class ModuleSpanningTree : public Module
                                TreeServer* CheckDupe = FindServer(x->Name);
                                if (!CheckDupe)
                                {
-                                       WriteServ(user->fd,"NOTICE %s :*** CONNECT: Connecting to server: \002%s\002 (%s:%d)",user->nick,x->Name.c_str(),x->IPAddr.c_str(),x->Port);
+                                       WriteServ(user->fd,"NOTICE %s :*** CONNECT: Connecting to server: \002%s\002 (%s:%d)",user->nick,x->Name.c_str(),(x->HiddenFromStats ? "<hidden>" : x->IPAddr.c_str()),x->Port);
                                        TreeSocket* newsocket = new TreeSocket(x->IPAddr,x->Port,false,10,x->Name);
-                                       Srv->AddSocket(newsocket);
+                                       if (newsocket->GetState() != I_ERROR)
+                                       {
+                                               Srv->AddSocket(newsocket);
+                                       }
+                                       else
+                                       {
+                                               WriteServ(user->fd,"NOTICE %s :*** CONNECT: Error connecting \002%s\002.",user->nick,x->Name.c_str());
+                                               delete newsocket;
+                                       }
                                        return 1;
                                }
                                else
@@ -2518,24 +2867,28 @@ class ModuleSpanningTree : public Module
                return 1;
        }
 
-       virtual bool HandleStats(char ** parameters, int pcnt, userrec* user)
+       virtual int OnStats(char statschar, userrec* user)
        {
-               if (*parameters[0] == 'c')
+               if (statschar == 'c')
                {
                        for (unsigned int i = 0; i < LinkBlocks.size(); i++)
                        {
-                               WriteServ(user->fd,"213 %s C *@%s * %s %d 0 M",user->nick,LinkBlocks[i].IPAddr.c_str(),LinkBlocks[i].Name.c_str(),LinkBlocks[i].Port);
+                               WriteServ(user->fd,"213 %s C *@%s * %s %d 0 %c%c%c",user->nick,(LinkBlocks[i].HiddenFromStats ? "<hidden>" : LinkBlocks[i].IPAddr).c_str(),LinkBlocks[i].Name.c_str(),LinkBlocks[i].Port,(LinkBlocks[i].EncryptionKey != "" ? 'e' : '-'),(LinkBlocks[i].AutoConnect ? 'a' : '-'),'s');
                                WriteServ(user->fd,"244 %s H * * %s",user->nick,LinkBlocks[i].Name.c_str());
                        }
-                       WriteServ(user->fd,"219 %s %s :End of /STATS report",user->nick,parameters[0]);
-                       WriteOpers("*** Notice: Stats '%s' requested by %s (%s@%s)",parameters[0],user->nick,user->ident,user->host);
-                       return true;
+                       WriteServ(user->fd,"219 %s %c :End of /STATS report",user->nick,statschar);
+                       WriteOpers("*** Notice: Stats '%c' requested by %s (%s@%s)",statschar,user->nick,user->ident,user->host);
+                       return 1;
                }
-               return false;
+               return 0;
        }
 
-       virtual int OnPreCommand(std::string command, char **parameters, int pcnt, userrec *user)
+       virtual int OnPreCommand(std::string command, char **parameters, int pcnt, userrec *user, bool validated)
        {
+               /* If the command doesnt appear to be valid, we dont want to mess with it. */
+               if (!validated)
+                       return 0;
+
                if (command == "CONNECT")
                {
                        return this->HandleConnect(parameters,pcnt,user);
@@ -2544,10 +2897,6 @@ class ModuleSpanningTree : public Module
                {
                        return this->HandleSquit(parameters,pcnt,user);
                }
-               else if (command == "STATS")
-               {
-                       return this->HandleStats(parameters,pcnt,user);
-               }
                else if (command == "MAP")
                {
                        this->HandleMap(parameters,pcnt,user);
@@ -2596,6 +2945,7 @@ class ModuleSpanningTree : public Module
                                        params.push_back(std::string(parameters[j]));
                                }
                        }
+                       log(DEBUG,"Globally route '%s'",command.c_str());
                        DoOneToMany(user->nick,command,params);
                }
                return 0;
@@ -2639,7 +2989,7 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       virtual void OnUserNotice(userrec* user, void* dest, int target_type, std::string text)
+       virtual void OnUserNotice(userrec* user, void* dest, int target_type, std::string text, char status)
        {
                if (target_type == TYPE_USER)
                {
@@ -2658,6 +3008,9 @@ class ModuleSpanningTree : public Module
                        if (user->fd > -1)
                        {
                                chanrec *c = (chanrec*)dest;
+                               std::string cname = c->name;
+                               if (status)
+                                       cname = status + cname;
                                std::deque<TreeServer*> list;
                                GetListOfServersForChannel(c,list);
                                unsigned int ucount = list.size();
@@ -2665,13 +3018,13 @@ class ModuleSpanningTree : public Module
                                {
                                        TreeSocket* Sock = list[i]->GetSocket();
                                        if (Sock)
-                                               Sock->WriteLine(":"+std::string(user->nick)+" NOTICE "+std::string(c->name)+" :"+text);
+                                               Sock->WriteLine(":"+std::string(user->nick)+" NOTICE "+cname+" :"+text);
                                }
                        }
                }
        }
 
-       virtual void OnUserMessage(userrec* user, void* dest, int target_type, std::string text)
+       virtual void OnUserMessage(userrec* user, void* dest, int target_type, std::string text, char status)
        {
                if (target_type == TYPE_USER)
                {
@@ -2692,6 +3045,9 @@ class ModuleSpanningTree : public Module
                        if (user->fd > -1)
                        {
                                chanrec *c = (chanrec*)dest;
+                               std::string cname = c->name;
+                               if (status)
+                                       cname = status + cname;
                                std::deque<TreeServer*> list;
                                GetListOfServersForChannel(c,list);
                                unsigned int ucount = list.size();
@@ -2699,7 +3055,7 @@ class ModuleSpanningTree : public Module
                                {
                                        TreeSocket* Sock = list[i]->GetSocket();
                                        if (Sock)
-                                               Sock->WriteLine(":"+std::string(user->nick)+" PRIVMSG "+std::string(c->name)+" :"+text);
+                                               Sock->WriteLine(":"+std::string(user->nick)+" PRIVMSG "+cname+" :"+text);
                                }
                        }
                }
@@ -2764,12 +3120,14 @@ class ModuleSpanningTree : public Module
                DoOneToMany(user->nick,"FNAME",params);
        }
 
-       virtual void OnUserPart(userrec* user, chanrec* channel)
+       virtual void OnUserPart(userrec* user, chanrec* channel, std::string partmessage)
        {
                if (user->fd > -1)
                {
                        std::deque<std::string> params;
                        params.push_back(channel->name);
+                       if (partmessage != "")
+                               params.push_back(":"+partmessage);
                        DoOneToMany(user->nick,"PART",params);
                }
        }
@@ -2787,9 +3145,16 @@ class ModuleSpanningTree : public Module
                        params.push_back(user->dhost);
                        params.push_back(user->ident);
                        params.push_back("+"+std::string(user->modes));
-                       params.push_back(user->ip);
+                       params.push_back((char*)inet_ntoa(user->ip4));
                        params.push_back(":"+std::string(user->fullname));
                        DoOneToMany(Srv->GetServerName(),"NICK",params);
+
+                       // User is Local, change needs to be reflected!
+                       TreeServer* SourceServer = FindServer(user->server);
+                       if (SourceServer) {
+                               SourceServer->AddUserCount();
+                       }
+
                }
        }
 
@@ -2801,6 +3166,12 @@ class ModuleSpanningTree : public Module
                        params.push_back(":"+reason);
                        DoOneToMany(user->nick,"QUIT",params);
                }
+               // Regardless, We need to modify the user Counts..
+               TreeServer* SourceServer = FindServer(user->server);
+               if (SourceServer) {
+                       SourceServer->DelUserCount();
+               }
+
        }
 
        virtual void OnUserPostNick(userrec* user, std::string oldnick)
@@ -2815,7 +3186,7 @@ class ModuleSpanningTree : public Module
 
        virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, std::string reason)
        {
-               if (source->fd > -1)
+                if ((source) && (source->fd > -1))
                {
                        std::deque<std::string> params;
                        params.push_back(chan->name);
@@ -2952,6 +3323,26 @@ class ModuleSpanningTree : public Module
                }
        }
 
+       virtual void OnSetAway(userrec* user)
+       {
+               if (IS_LOCAL(user))
+               {
+                       std::deque<std::string> params;
+                       params.push_back(":"+std::string(user->awaymsg));
+                       DoOneToMany(user->nick,"AWAY",params);
+               }
+       }
+
+       virtual void OnCancelAway(userrec* user)
+       {
+               if (IS_LOCAL(user))
+               {
+                       std::deque<std::string> params;
+                       params.clear();
+                       DoOneToMany(user->nick,"AWAY",params);
+               }
+       }
+
        virtual void ProtoSendMode(void* opaque, int target_type, void* target, std::string modeline)
        {
                TreeSocket* s = (TreeSocket*)opaque;
@@ -2980,7 +3371,11 @@ class ModuleSpanningTree : public Module
                                userrec* u = (userrec*)target;
                                s->WriteLine(":"+Srv->GetServerName()+" METADATA "+u->nick+" "+extname+" :"+extdata);
                        }
-                       else
+                       else if (target_type == TYPE_OTHER)
+                       {
+                               s->WriteLine(":"+Srv->GetServerName()+" METADATA * "+extname+" :"+extdata);
+                       }
+                       else if (target_type == TYPE_CHANNEL)
                        {
                                chanrec* c = (chanrec*)target;
                                s->WriteLine(":"+Srv->GetServerName()+" METADATA "+c->name+" "+extname+" :"+extdata);
@@ -2988,6 +3383,18 @@ class ModuleSpanningTree : public Module
                }
        }
 
+       virtual void OnEvent(Event* event)
+       {
+               if (event->GetEventID() == "send_metadata")
+               {
+                       std::deque<std::string>* params = (std::deque<std::string>*)event->GetData();
+                       if (params->size() < 3)
+                               return;
+                       (*params)[2] = ":" + (*params)[2];
+                       DoOneToMany(Srv->GetServerName(),"METADATA",*params);
+               }
+       }
+
        virtual ~ModuleSpanningTree()
        {
        }
@@ -2996,6 +3403,29 @@ class ModuleSpanningTree : public Module
        {
                return Version(1,0,0,0,VF_STATIC|VF_VENDOR);
        }
+
+       void Implements(char* List)
+       {
+               List[I_OnPreCommand] = List[I_OnGetServerDescription] = List[I_OnUserInvite] = List[I_OnPostLocalTopicChange] = 1;
+               List[I_OnWallops] = List[I_OnUserNotice] = List[I_OnUserMessage] = List[I_OnBackgroundTimer] = 1;
+               List[I_OnUserJoin] = List[I_OnChangeHost] = List[I_OnChangeName] = List[I_OnUserPart] = List[I_OnUserConnect] = 1;
+               List[I_OnUserQuit] = List[I_OnUserPostNick] = List[I_OnUserKick] = List[I_OnRemoteKill] = List[I_OnRehash] = 1;
+               List[I_OnOper] = List[I_OnAddGLine] = List[I_OnAddZLine] = List[I_OnAddQLine] = List[I_OnAddELine] = 1;
+               List[I_OnDelGLine] = List[I_OnDelZLine] = List[I_OnDelQLine] = List[I_OnDelELine] = List[I_ProtoSendMode] = List[I_OnMode] = 1;
+               List[I_OnStats] = List[I_ProtoSendMetaData] = List[I_OnEvent] = List[I_OnSetAway] = List[I_OnCancelAway] = 1;
+       }
+
+       /* It is IMPORTANT that m_spanningtree is the last module in the chain
+        * so that any activity it sees is FINAL, e.g. we arent going to send out
+        * a NICK message before m_cloaking has finished putting the +x on the user,
+        * etc etc.
+        * Therefore, we return PRIORITY_LAST to make sure we end up at the END of
+        * the module call queue.
+        */
+       Priority Prioritize()
+       {
+               return PRIORITY_LAST;
+       }
 };