src/modules/m_spanningtree/server.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
   6  *
   7  * This file is part of InspIRCd.  InspIRCd is free software: you can
   8  * redistribute it and/or modify it under the terms of the GNU General Public
   9  * License as published by the Free Software Foundation, version 2.
  10  *
  11  * This program is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  13  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  14  * details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20
  21 #include "inspircd.h"
  22
  23 #include "utils.h"
  24 #include "link.h"
  25 #include "treeserver.h"
  26 #include "treesocket.h"
  27
  28 /* $ModDep: m_spanningtree/main.h m_spanningtree/utils.h m_spanningtree/treeserver.h m_spanningtree/treesocket.h m_spanningtree/link.h */
  29
  30 /*
  31  * Some server somewhere in the network introducing another server.
  32  *      -- w
  33  */
  34 bool TreeSocket::RemoteServer(const std::string &prefix, parameterlist &params)
  35 {
  36         if (params.size() < 5)
  37         {
  38                 SendError("Protocol error - Not enough parameters for SERVER command");
  39                 return false;
  40         }
  41
  42         std::string servername = params[0];
  43         // password is not used for a remote server
  44         // hopcount is not used (ever)
  45         std::string sid = params[3];
  46         std::string description = params[4];
  47         TreeServer* ParentOfThis = Utils->FindServer(prefix);
  48
  49         if (!ParentOfThis)
  50         {
  51                 this->SendError("Protocol error - Introduced remote server from unknown server "+prefix);
  52                 return false;
  53         }
  54         if (!InspIRCd::IsSID(sid))
  55         {
  56                 this->SendError("Invalid format server ID: "+sid+"!");
  57                 return false;
  58         }
  59         TreeServer* CheckDupe = Utils->FindServer(servername);
  60         if (CheckDupe)
  61         {
  62                 this->SendError("Server "+servername+" already exists!");
  63                 ServerInstance->SNO->WriteToSnoMask('L', "Server \2"+CheckDupe->GetName()+"\2 being introduced from \2" + ParentOfThis->GetName() + "\2 denied, already exists. Closing link with " + ParentOfThis->GetName());
  64                 return false;
  65         }
  66         CheckDupe = Utils->FindServer(sid);
  67         if (CheckDupe)
  68         {
  69                 this->SendError("Server ID "+sid+" already exists! You may want to specify the server ID for the server manually with <server:id> so they do not conflict.");
  70                 ServerInstance->SNO->WriteToSnoMask('L', "Server \2"+servername+"\2 being introduced from \2" + ParentOfThis->GetName() + "\2 denied, server ID already exists on the network. Closing link with " + ParentOfThis->GetName());
  71                 return false;
  72         }
  73
  74
  75         Link* lnk = Utils->FindLink(servername);
  76
  77         TreeServer *Node = new TreeServer(Utils, servername, description, sid, ParentOfThis,NULL, lnk ? lnk->Hidden : false);
  78
  79         ParentOfThis->AddChild(Node);
  80         params[4] = ":" + params[4];
  81         Utils->DoOneToAllButSender(prefix,"SERVER",params,prefix);
  82         ServerInstance->SNO->WriteToSnoMask('L', "Server \002"+ParentOfThis->GetName()+"\002 introduced server \002"+servername+"\002 ("+description+")");
  83         return true;
  84 }
  85
  86
  87 /*
  88  * This is used after the other side of a connection has accepted our credentials.
  89  * They are then introducing themselves to us, BEFORE either of us burst. -- w
  90  */
  91 bool TreeSocket::Outbound_Reply_Server(parameterlist &params)
  92 {
  93         if (params.size() < 5)
  94         {
  95                 SendError("Protocol error - Not enough parameters for SERVER command");
  96                 return false;
  97         }
  98
  99         irc::string servername = params[0].c_str();
 100         std::string sname = params[0];
 101         std::string password = params[1];
 102         std::string sid = params[3];
 103         std::string description = params[4];
 104
 105         this->SendCapabilities(2);
 106
 107         if (!ServerInstance->IsSID(sid))
 108         {
 109                 this->SendError("Invalid format server ID: "+sid+"!");
 110                 return false;
 111         }
 112
 113         for (std::vector<reference<Link> >::iterator i = Utils->LinkBlocks.begin(); i < Utils->LinkBlocks.end(); i++)
 114         {
 115                 Link* x = *i;
 116                 if (x->Name != servername && x->Name != "*") // open link allowance
 117                         continue;
 118
 119                 if (!ComparePass(*x, password))
 120                 {
 121                         ServerInstance->SNO->WriteToSnoMask('l',"Invalid password on link: %s", x->Name.c_str());
 122                         continue;
 123                 }
 124
 125                 TreeServer* CheckDupe = Utils->FindServer(sname);
 126                 if (CheckDupe)
 127                 {
 128                         std::string pname = CheckDupe->GetParent() ? CheckDupe->GetParent()->GetName() : "<ourself>";
 129                         SendError("Server "+sname+" already exists on server "+pname+"!");
 130                         ServerInstance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, already exists on server "+pname);
 131                         return false;
 132                 }
 133                 CheckDupe = Utils->FindServer(sid);
 134                 if (CheckDupe)
 135                 {
 136                         this->SendError("Server ID "+sid+" already exists on the network! You may want to specify the server ID for the server manually with <server:id> so they do not conflict.");
 137                         ServerInstance->SNO->WriteToSnoMask('l',"Server \2"+assign(servername)+"\2 being introduced denied, server ID already exists on the network. Closing link.");
 138                         return false;
 139                 }
 140
 141                 /*
 142                  * They're in WAIT_AUTH_2 (having accepted our credentials).
 143                  * Set our state to CONNECTED (since everything's peachy so far) and send our
 144                  * netburst to them, which will trigger their CONNECTED state, and BURST in reply.
 145                  *
 146                  * While we're at it, create a treeserver object so we know about them.
 147                  *   -- w
 148                  */
 149                 this->LinkState = CONNECTED;
 150
 151                 Utils->timeoutlist.erase(this);
 152                 linkID = sname;
 153
 154                 MyRoot = new TreeServer(Utils, sname, description, sid, Utils->TreeRoot, this, x->Hidden);
 155                 Utils->TreeRoot->AddChild(MyRoot);
 156                 this->DoBurst(MyRoot);
 157
 158                 params[4] = ":" + params[4];
 159
 160                 /* IMPORTANT: Take password/hmac hash OUT of here before we broadcast the introduction! */
 161                 params[1] = "*";
 162                 Utils->DoOneToAllButSender(ServerInstance->Config->GetSID(),"SERVER",params,sname);
 163
 164                 return true;
 165         }
 166
 167         this->SendError("Invalid credentials (check the other server's linking snomask for more information)");
 168         ServerInstance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, invalid link credentials");
 169         return false;
 170 }
 171
 172 bool TreeSocket::CheckDuplicate(const std::string& sname, const std::string& sid)
 173 {
 174         /* Check for fully initialized instances of the server by name */
 175         TreeServer* CheckDupe = Utils->FindServer(sname);
 176         if (CheckDupe)
 177         {
 178                 std::string pname = CheckDupe->GetParent() ? CheckDupe->GetParent()->GetName() : "<ourself>";
 179                 SendError("Server "+sname+" already exists on server "+pname+"!");
 180                 ServerInstance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, already exists on server "+pname);
 181                 return false;
 182         }
 183
 184         /* Check for fully initialized instances of the server by id */
 185         ServerInstance->Logs->Log(MODNAME, LOG_DEBUG, "Looking for dupe SID %s", sid.c_str());
 186         CheckDupe = Utils->FindServerID(sid);
 187
 188         if (CheckDupe)
 189         {
 190                 this->SendError("Server ID "+CheckDupe->GetID()+" already exists on server "+CheckDupe->GetName()+"! You may want to specify the server ID for the server manually with <server:id> so they do not conflict.");
 191                 ServerInstance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, server ID '"+CheckDupe->GetID()+
 192                                 "' already exists on server "+CheckDupe->GetName());
 193                 return false;
 194         }
 195
 196         return true;
 197 }
 198
 199 /*
 200  * Someone else is attempting to connect to us if this is called. Validate their credentials etc.
 201  *              -- w
 202  */
 203 bool TreeSocket::Inbound_Server(parameterlist &params)
 204 {
 205         if (params.size() < 5)
 206         {
 207                 SendError("Protocol error - Missing SID");
 208                 return false;
 209         }
 210
 211         irc::string servername = params[0].c_str();
 212         std::string sname = params[0];
 213         std::string password = params[1];
 214         std::string sid = params[3];
 215         std::string description = params[4];
 216
 217         this->SendCapabilities(2);
 218
 219         if (!ServerInstance->IsSID(sid))
 220         {
 221                 this->SendError("Invalid format server ID: "+sid+"!");
 222                 return false;
 223         }
 224
 225         for (std::vector<reference<Link> >::iterator i = Utils->LinkBlocks.begin(); i < Utils->LinkBlocks.end(); i++)
 226         {
 227                 Link* x = *i;
 228                 if (x->Name != servername && x->Name != "*") // open link allowance
 229                         continue;
 230
 231                 if (!ComparePass(*x, password))
 232                 {
 233                         ServerInstance->SNO->WriteToSnoMask('l',"Invalid password on link: %s", x->Name.c_str());
 234                         continue;
 235                 }
 236
 237                 if (!CheckDuplicate(sname, sid))
 238                         return false;
 239
 240                 ServerInstance->SNO->WriteToSnoMask('l',"Verified incoming server connection " + linkID + " ("+description+")");
 241
 242                 this->SendCapabilities(2);
 243
 244                 // Save these for later, so when they accept our credentials (indicated by BURST) we remember them
 245                 this->capab->hidden = x->Hidden;
 246                 this->capab->sid = sid;
 247                 this->capab->description = description;
 248                 this->capab->name = sname;
 249
 250                 // Send our details: Our server name and description and hopcount of 0,
 251                 // along with the sendpass from this block.
 252                 this->WriteLine("SERVER "+ServerInstance->Config->ServerName+" "+this->MakePass(x->SendPass, this->GetTheirChallenge())+" 0 "+ServerInstance->Config->GetSID()+" :"+ServerInstance->Config->ServerDesc);
 253
 254                 // move to the next state, we are now waiting for THEM.
 255                 this->LinkState = WAIT_AUTH_2;
 256                 return true;
 257         }
 258
 259         this->SendError("Invalid credentials");
 260         ServerInstance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, invalid link credentials");
 261         return false;
 262 }
 263