]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_spanningtree/treesocket2.cpp
Move MODENOTICE command to a command module
[user/henk/code/inspircd.git] / src / modules / m_spanningtree / treesocket2.cpp
index 93d99c2bad1fb58c1854934aa103e0ca40f822f0..918ce21645a88fe151448f26dd607e5ace7e5d83 100644 (file)
@@ -2,8 +2,8 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
- * See: http://www.inspircd.org/wiki/index.php/Credits
+ *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ * See: http://wiki.inspircd.org/Credits
  *
  * This program is free but copyrighted software; see
  *            the file COPYING for details.
  */
 
 #include "inspircd.h"
-#include "commands/cmd_whois.h"
-#include "commands/cmd_stats.h"
 #include "socket.h"
-#include "wildcard.h"
 #include "xline.h"
-#include "transport.h"
 #include "socketengine.h"
 
-#include "m_spanningtree/main.h"
-#include "m_spanningtree/utils.h"
-#include "m_spanningtree/treeserver.h"
-#include "m_spanningtree/link.h"
-#include "m_spanningtree/treesocket.h"
-#include "m_spanningtree/resolvers.h"
-#include "m_spanningtree/handshaketimer.h"
-
-/* $ModDep: m_spanningtree/timesynctimer.h m_spanningtree/resolvers.h m_spanningtree/main.h m_spanningtree/utils.h m_spanningtree/treeserver.h m_spanningtree/link.h m_spanningtree/treesocket.h */
-
-void TreeSocket::WriteLine(std::string line)
-{
-       Instance->Logs->Log("m_spanningtree",DEBUG, "S[%d] O %s", this->GetFd(), line.c_str());
-       line.append("\r\n");
-       this->Write(line);
-}
-
+#include "main.h"
+#include "utils.h"
+#include "treeserver.h"
+#include "link.h"
+#include "treesocket.h"
+#include "resolvers.h"
 
 /* Handle ERROR command */
-bool TreeSocket::Error(std::deque<std::string> &params)
+bool TreeSocket::Error(parameterlist &params)
 {
        if (params.size() < 1)
                return false;
-       this->Instance->SNO->WriteToSnoMask('l',"ERROR from %s: %s",(!InboundServerName.empty() ? InboundServerName.c_str() : myhost.c_str()),params[0].c_str());
+       ServerInstance->SNO->WriteGlobalSno('l',"ERROR from %s: %s",(!InboundServerName.empty() ? InboundServerName.c_str() : myhost.c_str()),params[0].c_str());
        /* we will return false to cause the socket to close. */
        return false;
 }
 
-void TreeSocket::Split(const std::string &line, std::deque<std::string> &n)
+void TreeSocket::Split(const std::string& line, std::string& prefix, std::string& command, parameterlist& params)
 {
-       n.clear();
        irc::tokenstream tokens(line);
+
+       if (!tokens.GetToken(prefix))
+               return;
+       
+       if (prefix[0] == ':')
+       {
+               prefix = prefix.substr(1);
+
+               if (prefix.empty())
+               {
+                       this->SendError("BUG (?) Empty prefix received: " + line);
+                       return;
+               }
+               if (!tokens.GetToken(command))
+               {
+                       this->SendError("BUG (?) Empty command received: " + line);
+                       return;
+               }
+       }
+       else
+       {
+               command = prefix;
+               prefix.clear();
+       }
+       if (command.empty())
+               this->SendError("BUG (?) Empty command received: " + line);
+
        std::string param;
        while (tokens.GetToken(param))
        {
-               n.push_back(param);
+               params.push_back(param);
        }
-       return;
 }
 
-bool TreeSocket::ProcessLine(std::string &line)
+void TreeSocket::ProcessLine(std::string &line)
 {
-       std::deque<std::string> params;
-       irc::string command;
        std::string prefix;
+       std::string command;
+       parameterlist params;
 
-       line = line.substr(0, line.find_first_of("\r\n"));
+       ServerInstance->Logs->Log("m_spanningtree",DEBUG, "S[%d] I %s", this->GetFd(), line.c_str());
 
-       if (line.empty())
-               return true;
+       Split(line, prefix, command, params);
 
-       Instance->Logs->Log("m_spanningtree",DEBUG, "S[%d] I %s", this->GetFd(), line.c_str());
-
-       this->Split(line.c_str(),params);
-       
-       if (params.empty())
-               return true;
-       
-       if ((params[0][0] == ':') && (params.size() > 1))
-       {
-               prefix = params[0].substr(1);
-               params.pop_front();
-               
-               if (prefix.empty())
-               {
-                       this->SendError("BUG (?) Empty prefix recieved.");
-                       return false;
-               }
-       }
-       
-       command = params[0].c_str();
-       params.pop_front();
+       if (command.empty())
+               return;
 
        switch (this->LinkState)
        {
@@ -117,27 +107,23 @@ bool TreeSocket::ProcessLine(std::string &line)
                        }
                        else if (command == "SERVER")
                        {
-                               return this->Inbound_Server(params);
+                               this->Inbound_Server(params);
                        }
                        else if (command == "ERROR")
                        {
-                               return this->Error(params);
+                               this->Error(params);
                        }
                        else if (command == "USER")
                        {
                                this->SendError("Client connections to this port are prohibited.");
-                               return false;
                        }
                        else if (command == "CAPAB")
                        {
-                               return this->Capab(params);
+                               this->Capab(params);
                        }
                        else
                        {
-                               // XXX ...wtf.
-                               irc::string error = "Invalid command in negotiation phase: " + command;
-                               this->SendError(assign(error));
-                               return false;
+                               this->SendError(std::string("Invalid command in negotiation phase: ") + command.c_str());
                        }
                break;
                case WAIT_AUTH_2:
@@ -154,65 +140,59 @@ bool TreeSocket::ProcessLine(std::string &line)
                                 * Both of these aren't allowable, so block them here. -- w
                                 */
                                this->SendError("You may not re-authenticate or commence netburst without sending BURST.");
-                               return true;
                        }
                        else if (command == "BURST")
                        {
                                if (params.size())
                                {
                                        time_t them = atoi(params[0].c_str());
-                                       time_t delta = them - Instance->Time();
+                                       time_t delta = them - ServerInstance->Time();
                                        if ((delta < -600) || (delta > 600))
                                        {
-                                               Instance->SNO->WriteToSnoMask('l',"\2ERROR\2: Your clocks are out by %d seconds (this is more than five minutes). Link aborted, \2PLEASE SYNC YOUR CLOCKS!\2",abs(delta));
-                                               SendError("Your clocks are out by "+ConvToStr(abs(delta))+" seconds (this is more than five minutes). Link aborted, PLEASE SYNC YOUR CLOCKS!");
-                                               return false;
+                                               ServerInstance->SNO->WriteGlobalSno('l',"\2ERROR\2: Your clocks are out by %d seconds (this is more than five minutes). Link aborted, \2PLEASE SYNC YOUR CLOCKS!\2",abs((long)delta));
+                                               SendError("Your clocks are out by "+ConvToStr(abs((long)delta))+" seconds (this is more than five minutes). Link aborted, PLEASE SYNC YOUR CLOCKS!");
+                                               return;
                                        }
                                        else if ((delta < -30) || (delta > 30))
                                        {
-                                               Instance->SNO->WriteToSnoMask('l',"\2WARNING\2: Your clocks are out by %d seconds. Please consider synching your clocks.", abs(delta));
+                                               ServerInstance->SNO->WriteGlobalSno('l',"\2WARNING\2: Your clocks are out by %d seconds. Please consider synching your clocks.", abs((long)delta));
                                        }
                                }
                                this->LinkState = CONNECTED;
-                               Link* lnk = Utils->FindLink(InboundServerName);
-
-                               Node = new TreeServer(this->Utils, this->Instance, InboundServerName, InboundDescription, InboundSID, Utils->TreeRoot, this, lnk ? lnk->Hidden : false);
 
-                               if (Node->DuplicateID())
+                               Utils->timeoutlist.erase(this);
+                               if (myautoconnect)
                                {
-                                       this->SendError("Server ID "+InboundSID+" already exists on the network!");
-                                       this->Instance->SNO->WriteToSnoMask('l',"Server \2"+InboundServerName+"\2 being introduced from \2" + prefix + "\2 denied, server ID already exists on the network. Closing link.");
-                                       return false;
+                                       myautoconnect->position = -1;
+                                       myautoconnect = NULL;
                                }
 
+                               Link* lnk = Utils->FindLink(InboundServerName);
+
+                               Node = new TreeServer(this->Utils, InboundServerName, InboundDescription, InboundSID, Utils->TreeRoot, this, lnk ? lnk->Hidden : false);
+
                                Utils->TreeRoot->AddChild(Node);
-                               params.clear();
-                               params.push_back(InboundServerName);
-                               params.push_back("*");
-                               params.push_back("1");
-                               params.push_back(InboundSID);
-                               params.push_back(":"+InboundDescription);
-                               Utils->DoOneToAllButSender(Instance->Config->GetSID(),"SERVER",params,InboundServerName);
+                               parameterlist sparams;
+                               sparams.push_back(InboundServerName);
+                               sparams.push_back("*");
+                               sparams.push_back("1");
+                               sparams.push_back(InboundSID);
+                               sparams.push_back(":"+InboundDescription);
+                               Utils->DoOneToAllButSender(ServerInstance->Config->GetSID(),"SERVER",sparams,InboundServerName);
+                               Utils->DoOneToAllButSender(prefix, "BURST", params, InboundServerName);
                                Node->bursting = true;
                                this->DoBurst(Node);
                        }
                        else if (command == "ERROR")
                        {
-                               return this->Error(params);
+                               this->Error(params);
                        }
                        else if (command == "CAPAB")
                        {
-                               return this->Capab(params);
+                               this->Capab(params);
                        }
 
                break;
-               case LISTENER:
-                       /*
-                        * This really shouldn't happen.
-                        */
-                       this->SendError("Internal error -- listening socket accepted its own descriptor!!!");
-                       return false;
-               break;
                case CONNECTING:
                        /*
                         * State CONNECTING:
@@ -223,401 +203,316 @@ bool TreeSocket::ProcessLine(std::string &line)
                        if (command == "SERVER")
                        {
                                // Our credentials have been accepted, send netburst. (this puts US into the CONNECTED state)
-                               return this->Outbound_Reply_Server(params);
+                               this->Outbound_Reply_Server(params);
                        }
                        else if (command == "ERROR")
                        {
-                               return this->Error(params);
+                               this->Error(params);
                        }
                        else if (command == "CAPAB")
                        {
-                               return this->Capab(params);
+                               this->Capab(params);
                        }
                break;
                case CONNECTED:
                        /*
-                       * State CONNECTED:
+                        * State CONNECTED:
                         *  Credentials have been exchanged, we've gotten their 'BURST' (or sent ours).
                         *  Anything from here on should be accepted a little more reasonably.
                         */
-                       if (!prefix.empty())
-                       {
-                               /*
-                                * Check for fake direction here, and drop any instances that are found.
-                                * What is fake direction? Imagine the following server setup:
-                                *    0AA <-> 0AB <-> 0AC
-                                * Fake direction would be 0AC sending a message to 0AB claiming to be from
-                                * 0AA, or something similar. Basically, a message taking a path that *cannot*
-                                * be correct.
-                                *
-                                * When would this be seen?
-                                * Well, hopefully never. It could be caused by race conditions, bugs, or
-                                * "miscreant" servers, though, so let's check anyway. -- w
-                                */
-                               std::string direction = prefix;
+                       this->ProcessConnectedLine(prefix, command, params);
+               break;
+       }
+}
 
-                               User *t = this->Instance->FindUUID(prefix);
-                               if (t)
-                               {
-                                       direction = t->server;
-                               }
+void TreeSocket::ProcessConnectedLine(std::string& prefix, std::string& command, parameterlist& params)
+{
+       User* who = ServerInstance->FindUUID(prefix);
+       std::string direction;
 
-                               TreeServer* route_back_again = Utils->BestRouteTo(direction);
-                               if ((!route_back_again) || (route_back_again->GetSocket() != this))
-                               {
-                                       if (route_back_again)
-                                               Instance->Logs->Log("m_spanningtree",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(Instance->Time() + Utils->PingFreq);
-                               route_back_again->SetPingFlag();
-                       }
-                       else
-                       {
-                               /*
-                                * Empty prefix from a server to server link:
-                                *  This is somewhat bad/naughty, so let's set the prefix
-                                *  to be the link that we got it from, so we don't break anything. -- w
-                                */
-                               TreeServer* n = Utils->FindServer(GetName());
-                               if (n)
-                                       prefix = n->GetID();
-                               else
-                                       prefix = GetName();
-                       }
+       if (who)
+       {
+               direction = who->server;
+       }
+       else
+       {
+               TreeServer* ServerSource = Utils->FindServer(prefix);
+               if (prefix.empty())
+                       ServerSource = Utils->FindServer(GetName());
 
-                       /*
-                        * First up, check for any malformed commands (e.g. MODE without a timestamp)
-                        * and rewrite commands where necessary (SVSMODE -> MODE for services). -- w
+               if (ServerSource)
+               {
+                       who = ServerSource->ServerUser;
+                       direction = prefix;
+               }
+               else
+               {
+                       /* It is important that we don't close the link here, unknown prefix can occur
+                        * due to various race conditions such as the KILL message for a user somehow
+                        * crossing the users QUIT further upstream from the server. Thanks jilles!
                         */
-                       if (command == "SVSMODE") // This isn't in an "else if" so we still force FMODE for changes on channels.
-                               command = "MODE";
+                       ServerInstance->Logs->Log("m_spanningtree", DEBUG, "Command '%s' from unknown prefix '%s'! Dropping entire command.",
+                               command.c_str(), prefix.c_str());
+                       return;
+               }
+       }
 
-                       if (command == "MODE")
-                       {
-                               if (params.size() >= 2)
-                               {
-                                       Channel* channel = Instance->FindChan(params[0]);
-                                       if (channel)
-                                       {
-                                               this->SendError("MODE may no longer be used on channels. Please use FMODE, with correct timestamp rules.");
-                                               return false;
-                                       }
-                               }
-                       }
+       // Make sure prefix is still good
+       prefix = who->uuid;
+
+       /*
+        * Check for fake direction here, and drop any instances that are found.
+        * What is fake direction? Imagine the following server setup:
+        *    0AA <-> 0AB <-> 0AC
+        * Fake direction would be 0AC sending a message to 0AB claiming to be from
+        * 0AA, or something similar. Basically, a message taking a path that *cannot*
+        * be correct.
+        *
+        * When would this be seen?
+        * Well, hopefully never. It could be caused by race conditions, bugs, or
+        * "miscreant" servers, though, so let's check anyway. -- w
+        *
+        * We also check here for totally invalid prefixes (prefixes that are neither
+        * a valid SID or a valid UUID, so that invalid UUID or SID never makes it
+        * to the higher level functions. -- B
+        */
+       TreeServer* route_back_again = Utils->BestRouteTo(direction);
+       if ((!route_back_again) || (route_back_again->GetSocket() != this))
+       {
+               if (route_back_again)
+                       ServerInstance->Logs->Log("m_spanningtree",DEBUG,"Protocol violation: Fake direction '%s' from connection '%s'",
+                               prefix.c_str(),this->GetName().c_str());
+               return;
+       }
 
+       /*
+        * First up, check for any malformed commands (e.g. MODE without a timestamp)
+        * and rewrite commands where necessary (SVSMODE -> MODE for services). -- w
+        */
+       if (command == "SVSMODE") // This isn't in an "else if" so we still force FMODE for changes on channels.
+               command = "MODE";
 
-                       /*
-                        * Now, check for (and parse) commands as appropriate. -- w
-                        */     
-               
-                       /* Find the server that this command originated from, used in the handlers below */
-                       TreeServer *ServerSource = Utils->FindServer(prefix);
-
-                       /* Find the link we just got this from so we don't bounce it back incorrectly */
-                       std::string sourceserv = this->myhost;
-                       if (!this->InboundServerName.empty())
-                       {
-                               sourceserv = this->InboundServerName;
-                       }
+       // TODO move all this into Commands
+       if (command == "UID")
+       {
+               this->ParseUID(prefix, params);
+       }
+       else if (command == "FJOIN")
+       {
+               this->ForceJoin(who,params);
+       }
+       else if (command == "STATS")
+       {
+               this->Stats(prefix, params);
+       }
+       else if (command == "MOTD")
+       {
+               this->Motd(prefix, params);
+       }
+       else if (command == "ADMIN")
+       {
+               this->Admin(prefix, params);
+       }
+       else if (command == "MAP")
+       {
+               Utils->Creator->HandleMap(params, who);
+       }
+       else if (command == "SERVER")
+       {
+               this->RemoteServer(prefix,params);
+       }
+       else if (command == "ERROR")
+       {
+               this->Error(params);
+       }
+       else if (command == "OPERTYPE")
+       {
+               this->OperType(prefix,params);
+       }
+       else if (command == "AWAY")
+       {
+               this->Away(prefix,params);
+       }
+       else if (command == "FMODE")
+       {
+               this->ForceMode(who,params);
+       }
+       else if (command == "FTOPIC")
+       {
+               this->ForceTopic(prefix,params);
+       }
+       else if (command == "METADATA")
+       {
+               this->MetaData(prefix,params);
+       }
+       else if (command == "PING")
+       {
+               this->LocalPing(prefix,params);
+       }
+       else if (command == "PONG")
+       {
+               TreeServer *s = Utils->FindServer(prefix);
+               if (s && s->bursting)
+               {
+                       ServerInstance->SNO->WriteGlobalSno('l',"Server \002%s\002 has not finished burst, forcing end of burst (send ENDBURST!)", prefix.c_str());
+                       s->FinishBurst();
+               }
+               this->LocalPong(prefix,params);
+       }
+       else if (command == "VERSION")
+       {
+               this->ServerVersion(prefix,params);
+       }
+       else if (command == "FHOST")
+       {
+               this->ChangeHost(prefix,params);
+       }
+       else if (command == "FNAME")
+       {
+               this->ChangeName(prefix,params);
+       }
+       else if (command == "FIDENT")
+       {
+               this->ChangeIdent(prefix,params);
+       }
+       else if (command == "ADDLINE")
+       {
+               this->AddLine(prefix,params);
+       }
+       else if (command == "DELLINE")
+       {
+               this->DelLine(prefix,params);
+       }
+       else if (command == "SVSNICK")
+       {
+               this->SVSNick(prefix,params);
+       }
+       else if (command == "SAVE")
+       {
+               this->ForceNick(prefix,params);
+       }
+       else if (command == "OPERQUIT")
+       {
+               this->OperQuit(prefix,params);
+       }
+       else if (command == "IDLE")
+       {
+               this->Whois(prefix,params);
+       }
+       else if (command == "PUSH")
+       {
+               this->Push(prefix,params);
+       }
+       else if (command == "TIME")
+       {
+               this->Time(prefix,params);
+       }
+       else if (command == "SVSJOIN")
+       {
+               this->ServiceJoin(prefix,params);
+       }
+       else if (command == "SVSPART")
+       {
+               this->ServicePart(prefix,params);
+       }
+       else if (command == "SQUIT")
+       {
+               if (params.size() == 2)
+               {
+                       this->Squit(Utils->FindServer(params[0]),params[1]);
+               }
+       }
+       else if (command == "SNONOTICE")
+       {
+               if (params.size() >= 2)
+               {
+                       ServerInstance->SNO->WriteToSnoMask(*(params[0].c_str()), "From " + who->nick + ": "+ params[1]);
+                       Utils->DoOneToAllButSender(prefix, command, params, prefix);
+               }
+       }
+       else if (command == "BURST")
+       {
+               // Set prefix server as bursting
+               TreeServer* ServerSource = Utils->FindServer(prefix);
+               if (!ServerSource)
+               {
+                       ServerInstance->SNO->WriteGlobalSno('l', "WTF: Got BURST from a non-server(?): %s", prefix.c_str());
+                       return;
+               }
 
-                       /*
-                        * XXX one of these days, this needs to be moved into class Commands.
-                        */
-                       if (command == "UID")
-                       {
-                               return this->ParseUID(prefix, params);
-                       }
-                       else if (command == "FJOIN")
-                       {
-                               return this->ForceJoin(prefix,params);
-                       }
-                       else if ((command == "NOTICE" || command == "PRIVMSG") && (Utils->IsServer(prefix)))
-                       {
-                               return this->ServerMessage(assign(command), prefix, params, sourceserv);
-                       }
-                       else if (command == "STATS")
-                       {
-                               return this->Stats(prefix, params);
-                       }
-                       else if (command == "MOTD")
-                       {
-                               return this->Motd(prefix, params);
-                       }
-                       else if (command == "KILL" && ServerSource)
-                       {
-                               // Kill from a server
-                               return this->RemoteKill(prefix,params);
-                       }
-                       else if (command == "MODULES")
-                       {
-                               return this->Modules(prefix, params);
-                       }
-                       else if (command == "ADMIN")
-                       {
-                               return this->Admin(prefix, params);
-                       }
-                       else if (command == "MAP")
-                       {
-                               User* user = Instance->FindNick(prefix);
-                               if (user)
-                               {
-                                       std::vector<std::string> p(params.begin(), params.end());
-                                       return Utils->Creator->HandleMap(p, user);
-                               }
-                       }
-                       else if (command == "SERVER")
-                       {
-                               return this->RemoteServer(prefix,params);
-                       }
-                       else if (command == "ERROR")
-                       {
-                               return this->Error(params);
-                       }
-                       else if (command == "OPERTYPE")
-                       {
-                               return this->OperType(prefix,params);
-                       }
-                       else if (command == "FMODE")
-                       {
-                               return this->ForceMode(prefix,params);
-                       }
-                       else if (command == "FTOPIC")
-                       {
-                               return this->ForceTopic(prefix,params);
-                       }
-                       else if (command == "METADATA")
-                       {
-                               return this->MetaData(prefix,params);
-                       }
-                       else if (command == "PING")
-                       {
-                               return this->LocalPing(prefix,params);
-                       }
-                       else if (command == "PONG")
-                       {
-                               return this->LocalPong(prefix,params);
-                       }
-                       else if (command == "VERSION")
-                       {
-                               return this->ServerVersion(prefix,params);
-                       }
-                       else if (command == "FHOST")
-                       {
-                               return this->ChangeHost(prefix,params);
-                       }
-                       else if (command == "FNAME")
-                       {
-                               return this->ChangeName(prefix,params);
-                       }
-                       else if (command == "ADDLINE")
-                       {
-                               return this->AddLine(prefix,params);
-                       }
-                       else if (command == "DELLINE")
-                       {
-                               return this->DelLine(prefix,params);
-                       }
-                       else if (command == "SVSNICK")
-                       {
-                               return this->ForceNick(prefix,params);
-                       }
-                       else if (command == "OPERQUIT")
-                       {
-                               return this->OperQuit(prefix,params);
-                       }
-                       else if (command == "IDLE")
-                       {
-                               return this->Whois(prefix,params);
-                       }
-                       else if (command == "PUSH")
-                       {
-                               return this->Push(prefix,params);
-                       }
-                       else if (command == "TIME")
-                       {
-                               return this->Time(prefix,params);
-                       }
-                       else if ((command == "KICK") && (Utils->IsServer(prefix)))
-                       {
-                               if (params.size() == 3)
-                               {
-                                       TreeServer* pf = Utils->FindServer(prefix);
-                                       User* user = this->Instance->FindNick(params[1]);
-                                       Channel* chan = this->Instance->FindChan(params[0]);
-                                       if (pf && user && chan)
-                                       {
-                                               if (!chan->ServerKickUser(user, params[2].c_str(), false, pf->GetName().c_str()))
-                                                       /* Yikes, the channels gone! */
-                                                       delete chan;
-                                       }
-                               }
+               ServerSource->bursting = true;
+               Utils->DoOneToAllButSender(prefix, command, params, prefix);
+       }
+       else if (command == "ENDBURST")
+       {
+               TreeServer* ServerSource = Utils->FindServer(prefix);
+               if (!ServerSource)
+               {
+                       ServerInstance->SNO->WriteGlobalSno('l', "WTF: Got ENDBURST from a non-server(?): %s", prefix.c_str());
+                       return;
+               }
 
-                               return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
-                       }
-                       else if (command == "SVSJOIN")
-                       {
-                               return this->ServiceJoin(prefix,params);
-                       }
-                       else if (command == "SVSPART")
-                       {
-                               return this->ServicePart(prefix,params);
-                       }
-                       else if (command == "SQUIT")
-                       {
-                               if (params.size() == 2)
-                               {
-                                       this->Squit(Utils->FindServer(params[0]),params[1]);
-                               }
-                               return true;
-                       }
-                       else if (command == "MODENOTICE")
-                       {
-                               if (params.size() >= 2)
-                               {
-                                       if (ServerSource)
-                                               Instance->Users->WriteMode(params[0].c_str(), WM_AND, "*** From %s: %s", (ServerSource ? ServerSource->GetName().c_str() : prefix.c_str()), params[1].c_str());
-                               }
-                               return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
-                       }
-                       else if (command == "SNONOTICE")
-                       {
-                               if (params.size() >= 2)
-                               {
-                                       Instance->SNO->WriteToSnoMask(*(params[0].c_str()), "From " + (ServerSource ? ServerSource->GetName().c_str() : prefix) + ": "+ params[1]);
-                                       return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
-                               }
+               ServerSource->FinishBurst();
+               Utils->DoOneToAllButSender(prefix, command, params, prefix);
+       }
+       else if (command == "ENCAP")
+       {
+               this->Encap(who, params);
+       }
+       else if (command == "NICK")
+       {
+               if (params.size() != 2)
+               {
+                       SendError("Protocol violation: NICK message without TS - :"+std::string(who->uuid)+" NICK "+params[0]);
+                       return;
+               }
 
-                       }
-                       else if (command == "BURST")
-                       {
-                               // Set prefix server as bursting
-                               if (!ServerSource)
-                               {
-                                       this->Instance->SNO->WriteToSnoMask('l', "WTF: Got BURST from a nonexistant server(?): %s", (ServerSource ? ServerSource->GetName().c_str() : prefix.c_str()));
-                                       return false;
-                               }
-                               
-                               ServerSource->bursting = true;
-                               return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
-                       }
-                       else if (command == "ENDBURST")
-                       {
-                               if (!ServerSource)
-                               {
-                                       this->Instance->SNO->WriteToSnoMask('l', "WTF: Got ENDBURST from a nonexistant server(?): %s", (ServerSource ? ServerSource->GetName().c_str() : prefix.c_str()));
-                                       return false;
-                               }
-                               
-                               ServerSource->FinishBurst();
-                               return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
-                       }
-                       else if (command == "ENCAP")
-                       {
-                               return this->Encap(prefix, params);
-                       }
-                       else if (command == "MODE")
-                       {
-                               // Server-prefix MODE.
-                               std::vector<std::string> modelist(params.begin(), params.end());
+               if (IS_SERVER(who))
+               {
+                       SendError("Protocol violation: Server changing nick");
+                       return;
+               }
 
-                               /* We don't support this for channel mode changes any more! */
-                               if (params.size() >= 1)
-                               {
-                                       if (Instance->FindChan(params[0]))
-                                       {
-                                               this->SendError("Protocol violation by '"+(ServerSource ? ServerSource->GetName().c_str() : prefix)+"'! MODE for channel mode changes is not supported by the InspIRCd 1.2 protocol. You must use FMODE to preserve channel timestamps.");
-                                               return false;
-                                       }
-                               }
-                                       
-                               // Insert into the parser
-                               this->Instance->SendMode(modelist, this->Instance->FakeClient);
-                               
-                               // Pass out to the network
-                               return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
-                       }
-                       else
+               /* Update timestamp on user when they change nicks */
+               who->age = atoi(params[1].c_str());
+
+               /*
+                * On nick messages, check that the nick doesnt already exist here.
+                * If it does, perform collision logic.
+                */
+               User* x = ServerInstance->FindNickOnly(params[0]);
+               if ((x) && (x != who))
+               {
+                       int collideret = 0;
+                       /* x is local, who is remote */
+                       collideret = this->DoCollision(x, who->age, who->ident, who->GetIPString(), who->uuid);
+                       if (collideret != 1)
                        {
                                /*
-                                * Not a special s2s command. Emulate the user doing it.
-                                * This saves us having a huge ugly command parser again.
+                                * Remote client lost, or both lost, parsing or passing on this
+                                * nickchange would be pointless, as the incoming client's server will
+                                * soon recieve SVSNICK to change its nick to its UID. :) -- w00t
                                 */
-                               User *who = this->Instance->FindUUID(prefix);
-
-                               if (!who)
-                               {
-                                       // this looks ugly because command is an irc::string
-                                       this->SendError("Command (" + std::string(command.c_str()) + ") from unknown prefix (" + prefix + ")! Dropping link.");
-                                       return false;
-                               }
-
-                               if (command == "NICK")
-                               {
-                                       if (params.size() != 2)
-                                       {
-                                               SendError("Protocol violation: NICK message without TS - :"+std::string(who->uuid)+" NICK "+params[0]);
-                                               return false;
-                                       }
-                                       /* Update timestamp on user when they change nicks */
-                                       who->age = atoi(params[1].c_str());
-
-                                       /*
-                                        * On nick messages, check that the nick doesnt already exist here.
-                                        * If it does, perform collision logic.
-                                        */
-                                       User* x = this->Instance->FindNickOnly(params[0]);
-                                       if ((x) && (x != who))
-                                       {
-                                               int collideret = 0;
-                                               /* x is local, who is remote */
-                                               collideret = this->DoCollision(x, who->age, who->ident.c_str(), who->GetIPString(), who->uuid.c_str());
-                                               if (collideret != 1)
-                                               {
-                                                       /*
-                                                        * Remote client lost, or both lost, parsing this nickchange would be
-                                                        * pointless, as the incoming client's server will soon recieve SVSNICK to
-                                                        * change its nick to its UID. :) -- w00t
-                                                        */
-                                                       return true;
-                                               }
-                                       }
-                               }
-                                       
-                               // its a user
-                               std::vector<std::string> strparams(params.begin(), params.end());
-
-                               switch (this->Instance->CallCommandHandler(command.c_str(), strparams, who))
-                               {
-                                       case CMD_INVALID:
-                                               /*
-                                                * XXX: command is irc::string, hence ugliness
-                                                */
-                                               this->SendError("Unrecognised or malformed command '" + std::string(command.c_str()) + "' -- possibly loaded mismatched modules");
-                                               return false;
-                                               break;
-                                       case CMD_FAILURE:
-                                               /*
-                                                * CMD_LOCALONLY is aliased to CMD_FAILURE, so this won't go out onto the network.
-                                                */
-                                               return true;
-                                               break;
-                                       default:
-                                               /* CMD_SUCCESS falls through here */
-                                               break;
-                               }
-
-                               return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
-
+                               return;
                        }
-                       return true;
-                       break; // end of state CONNECTED (phew).
+               }
+               who->ForceNickChange(params[0].c_str());
+               Utils->RouteCommand(route_back_again, command, params, who);
+       }
+       else
+       {
+               Command* cmd = ServerInstance->Parser->GetHandler(command);
+               CmdResult res = CMD_INVALID;
+               if (cmd && params.size() >= cmd->min_params)
+               {
+                       res = cmd->Handle(params, who);
+               }
+
+               if (res == CMD_INVALID)
+                       SendError("Unrecognised or malformed command '" + command + "' -- possibly loaded mismatched modules");
+               if (res == CMD_SUCCESS)
+                       Utils->RouteCommand(route_back_again, command, params, who);
        }
-       return true;
 }
 
 std::string TreeSocket::GetName()
@@ -634,15 +529,14 @@ void TreeSocket::OnTimeout()
 {
        if (this->LinkState == CONNECTING)
        {
-               Utils->Creator->RemoteMessage(NULL, "CONNECT: Connection to \002%s\002 timed out.", myhost.c_str());
-               Link* MyLink = Utils->FindLink(myhost);
-               if (MyLink)
-                       Utils->DoFailOver(MyLink);
+               ServerInstance->SNO->WriteGlobalSno('l', "CONNECT: Connection to \002%s\002 timed out.", myhost.c_str());
        }
 }
 
-void TreeSocket::OnClose()
+void TreeSocket::Close()
 {
+       this->BufferedSocket::Close();
+
        // Test fix for big fuckup
        if (this->LinkState != CONNECTED)
                return;
@@ -663,38 +557,10 @@ void TreeSocket::OnClose()
 
        if (!quitserver.empty())
        {
-               Utils->Creator->RemoteMessage(NULL,"Connection to '\2%s\2' failed.",quitserver.c_str());
-               time_t server_uptime = Instance->Time() - this->age;    
-               if (server_uptime)
-                       Utils->Creator->RemoteMessage(NULL,"Connection to '\2%s\2' was established for %s", quitserver.c_str(), Utils->Creator->TimeToStr(server_uptime).c_str());
-       }
-}
-
-int TreeSocket::OnIncomingConnection(int newsock, char* ip)
-{
-       bool found = false;
-
-       found = (std::find(Utils->ValidIPs.begin(), Utils->ValidIPs.end(), ip) != Utils->ValidIPs.end());
-       if (!found)
-       {
-               for (std::vector<std::string>::iterator i = Utils->ValidIPs.begin(); i != Utils->ValidIPs.end(); i++)
-               {
-                       if ((*i) == "*" || irc::sockets::MatchCIDR(ip, (*i).c_str()))
-                       {
-                               found = true;
-                               break;
-                       }
-               }
+               ServerInstance->SNO->WriteGlobalSno('l', "Connection to '\2%s\2' failed.",quitserver.c_str());
 
-               if (!found)
-               {
-                       Utils->Creator->RemoteMessage(NULL,"Server connection from %s denied (no link blocks with that IP address)", ip);
-                       Instance->SE->Close(newsock);
-                       return false;
-               }
+               time_t server_uptime = ServerInstance->Time() - this->age;
+               if (server_uptime)
+                               ServerInstance->SNO->WriteGlobalSno('l', "Connection to '\2%s\2' was established for %s", quitserver.c_str(), Utils->Creator->TimeToStr(server_uptime).c_str());
        }
-
-       TreeSocket* s = new TreeSocket(this->Utils, this->Instance, newsock, ip, this->Hook);
-       s = s; /* Whinge whinge whinge, thats all GCC ever does. */
-       return true;
 }