src/modules/m_spanningtree/treesocket2.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2010 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15 #include "socket.h"
  16 #include "xline.h"
  17 #include "socketengine.h"
  18
  19 #include "main.h"
  20 #include "utils.h"
  21 #include "treeserver.h"
  22 #include "link.h"
  23 #include "treesocket.h"
  24 #include "resolvers.h"
  25
  26 /* Handle ERROR command */
  27 void TreeSocket::Error(parameterlist &params)
  28 {
  29         std::string msg = params.size() ? params[0] : "";
  30         SetError("received ERROR " + msg);
  31 }
  32
  33 void TreeSocket::Split(const std::string& line, std::string& prefix, std::string& command, parameterlist& params)
  34 {
  35         irc::tokenstream tokens(line);
  36
  37         if (!tokens.GetToken(prefix))
  38                 return;
  39
  40         if (prefix[0] == ':')
  41         {
  42                 prefix = prefix.substr(1);
  43
  44                 if (prefix.empty())
  45                 {
  46                         this->SendError("BUG (?) Empty prefix received: " + line);
  47                         return;
  48                 }
  49                 if (!tokens.GetToken(command))
  50                 {
  51                         this->SendError("BUG (?) Empty command received: " + line);
  52                         return;
  53                 }
  54         }
  55         else
  56         {
  57                 command = prefix;
  58                 prefix.clear();
  59         }
  60         if (command.empty())
  61                 this->SendError("BUG (?) Empty command received: " + line);
  62
  63         std::string param;
  64         while (tokens.GetToken(param))
  65         {
  66                 params.push_back(param);
  67         }
  68 }
  69
  70 void TreeSocket::ProcessLine(std::string &line)
  71 {
  72         std::string prefix;
  73         std::string command;
  74         parameterlist params;
  75
  76         ServerInstance->Logs->Log("m_spanningtree",DEBUG, "S[%d] I %s", this->GetFd(), line.c_str());
  77
  78         Split(line, prefix, command, params);
  79
  80         if (command.empty())
  81                 return;
  82
  83         switch (this->LinkState)
  84         {
  85                 TreeServer* Node;
  86
  87                 case WAIT_AUTH_1:
  88                         /*
  89                          * State WAIT_AUTH_1:
  90                          *  Waiting for SERVER command from remote server. Server initiating
  91                          *  the connection sends the first SERVER command, listening server
  92                          *  replies with theirs if its happy, then if the initiator is happy,
  93                          *  it starts to send its net sync, which starts the merge, otherwise
  94                          *  it sends an ERROR.
  95                          */
  96                         if (command == "PASS")
  97                         {
  98                                 /*
  99                                  * Ignore this silently. Some services packages insist on sending PASS, even
 100                                  * when it is not required (i.e. by us). We have to ignore this here, otherwise
 101                                  * as it's an unknown command (effectively), it will cause the connection to be
 102                                  * closed, which probably isn't what people want. -- w00t
 103                                  */
 104                         }
 105                         else if (command == "SERVER")
 106                         {
 107                                 this->Inbound_Server(params);
 108                         }
 109                         else if (command == "ERROR")
 110                         {
 111                                 this->Error(params);
 112                         }
 113                         else if (command == "USER")
 114                         {
 115                                 this->SendError("Client connections to this port are prohibited.");
 116                         }
 117                         else if (command == "CAPAB")
 118                         {
 119                                 this->Capab(params);
 120                         }
 121                         else
 122                         {
 123                                 this->SendError(std::string("Invalid command in negotiation phase: ") + command.c_str());
 124                         }
 125                 break;
 126                 case WAIT_AUTH_2:
 127                         /*
 128                          * State WAIT_AUTH_2:
 129                          *  We have sent SERVER to the other side of the connection. Now we're waiting for them to start BURST.
 130                          *  The other option at this stage of things, of course, is for them to close our connection thanks
 131                          *  to invalid credentials.. -- w
 132                          */
 133                         if (command == "SERVER")
 134                         {
 135                                 /*
 136                                  * Connection is either attempting to re-auth itself (stupid) or sending netburst without sending BURST.
 137                                  * Both of these aren't allowable, so block them here. -- w
 138                                  */
 139                                 this->SendError("You may not re-authenticate or commence netburst without sending BURST.");
 140                         }
 141                         else if (command == "BURST")
 142                         {
 143                                 if (params.size())
 144                                 {
 145                                         time_t them = atoi(params[0].c_str());
 146                                         time_t delta = them - ServerInstance->Time();
 147                                         if ((delta < -600) || (delta > 600))
 148                                         {
 149                                                 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));
 150                                                 SendError("Your clocks are out by "+ConvToStr(abs((long)delta))+" seconds (this is more than five minutes). Link aborted, PLEASE SYNC YOUR CLOCKS!");
 151                                                 return;
 152                                         }
 153                                         else if ((delta < -30) || (delta > 30))
 154                                         {
 155                                                 ServerInstance->SNO->WriteGlobalSno('l',"\2WARNING\2: Your clocks are out by %d seconds. Please consider synching your clocks.", abs((long)delta));
 156                                         }
 157                                 }
 158                                 this->LinkState = CONNECTED;
 159
 160                                 Utils->timeoutlist.erase(this);
 161                                 if (myautoconnect)
 162                                 {
 163                                         myautoconnect->position = -1;
 164                                         myautoconnect = NULL;
 165                                 }
 166
 167                                 Link* lnk = Utils->FindLink(InboundServerName);
 168
 169                                 Node = new TreeServer(this->Utils, InboundServerName, InboundDescription, InboundSID, Utils->TreeRoot, this, lnk ? lnk->Hidden : false);
 170
 171                                 Utils->TreeRoot->AddChild(Node);
 172                                 parameterlist sparams;
 173                                 sparams.push_back(InboundServerName);
 174                                 sparams.push_back("*");
 175                                 sparams.push_back("1");
 176                                 sparams.push_back(InboundSID);
 177                                 sparams.push_back(":"+InboundDescription);
 178                                 Utils->DoOneToAllButSender(ServerInstance->Config->GetSID(),"SERVER",sparams,InboundServerName);
 179                                 Utils->DoOneToAllButSender(prefix, "BURST", params, InboundServerName);
 180                                 Node->bursting = true;
 181                                 this->DoBurst(Node);
 182                         }
 183                         else if (command == "ERROR")
 184                         {
 185                                 this->Error(params);
 186                         }
 187                         else if (command == "CAPAB")
 188                         {
 189                                 this->Capab(params);
 190                         }
 191
 192                 break;
 193                 case CONNECTING:
 194                         /*
 195                          * State CONNECTING:
 196                          *  We're connecting (OUTGOING) to another server. They are in state WAIT_AUTH_1 until they verify
 197                          *  our credentials, when they proceed into WAIT_AUTH_2 and send SERVER to us. We then send BURST
 198                          *  + our netburst, which will put them into CONNECTED state. -- w
 199                          */
 200                         if (command == "SERVER")
 201                         {
 202                                 // Our credentials have been accepted, send netburst. (this puts US into the CONNECTED state)
 203                                 this->Outbound_Reply_Server(params);
 204                         }
 205                         else if (command == "ERROR")
 206                         {
 207                                 this->Error(params);
 208                         }
 209                         else if (command == "CAPAB")
 210                         {
 211                                 this->Capab(params);
 212                         }
 213                 break;
 214                 case CONNECTED:
 215                         /*
 216                          * State CONNECTED:
 217                          *  Credentials have been exchanged, we've gotten their 'BURST' (or sent ours).
 218                          *  Anything from here on should be accepted a little more reasonably.
 219                          */
 220                         this->ProcessConnectedLine(prefix, command, params);
 221                 break;
 222                 case DYING:
 223                 break;
 224         }
 225 }
 226
 227 void TreeSocket::ProcessConnectedLine(std::string& prefix, std::string& command, parameterlist& params)
 228 {
 229         User* who = ServerInstance->FindUUID(prefix);
 230         std::string direction;
 231
 232         if (who)
 233         {
 234                 direction = who->server;
 235         }
 236         else
 237         {
 238                 TreeServer* ServerSource = Utils->FindServer(prefix);
 239                 if (prefix.empty())
 240                         ServerSource = Utils->FindServer(GetName());
 241
 242                 if (ServerSource)
 243                 {
 244                         who = ServerSource->ServerUser;
 245                         direction = prefix;
 246                 }
 247                 else
 248                 {
 249                         /* It is important that we don't close the link here, unknown prefix can occur
 250                          * due to various race conditions such as the KILL message for a user somehow
 251                          * crossing the users QUIT further upstream from the server. Thanks jilles!
 252                          */
 253                         ServerInstance->Logs->Log("m_spanningtree", DEBUG, "Command '%s' from unknown prefix '%s'! Dropping entire command.",
 254                                 command.c_str(), prefix.c_str());
 255                         return;
 256                 }
 257         }
 258
 259         // Make sure prefix is still good
 260         prefix = who->uuid;
 261
 262         /*
 263          * Check for fake direction here, and drop any instances that are found.
 264          * What is fake direction? Imagine the following server setup:
 265          *    0AA <-> 0AB <-> 0AC
 266          * Fake direction would be 0AC sending a message to 0AB claiming to be from
 267          * 0AA, or something similar. Basically, a message taking a path that *cannot*
 268          * be correct.
 269          *
 270          * When would this be seen?
 271          * Well, hopefully never. It could be caused by race conditions, bugs, or
 272          * "miscreant" servers, though, so let's check anyway. -- w
 273          *
 274          * We also check here for totally invalid prefixes (prefixes that are neither
 275          * a valid SID or a valid UUID, so that invalid UUID or SID never makes it
 276          * to the higher level functions. -- B
 277          */
 278         TreeServer* route_back_again = Utils->BestRouteTo(direction);
 279         if ((!route_back_again) || (route_back_again->GetSocket() != this))
 280         {
 281                 if (route_back_again)
 282                         ServerInstance->Logs->Log("m_spanningtree",DEBUG,"Protocol violation: Fake direction '%s' from connection '%s'",
 283                                 prefix.c_str(),this->GetName().c_str());
 284                 return;
 285         }
 286
 287         /*
 288          * First up, check for any malformed commands (e.g. MODE without a timestamp)
 289          * and rewrite commands where necessary (SVSMODE -> MODE for services). -- w
 290          */
 291         if (command == "SVSMODE") // This isn't in an "else if" so we still force FMODE for changes on channels.
 292                 command = "MODE";
 293
 294         // TODO move all this into Commands
 295         if (command == "MAP")
 296         {
 297                 Utils->Creator->HandleMap(params, who);
 298         }
 299         else if (command == "SERVER")
 300         {
 301                 this->RemoteServer(prefix,params);
 302         }
 303         else if (command == "ERROR")
 304         {
 305                 this->Error(params);
 306         }
 307         else if (command == "AWAY")
 308         {
 309                 this->Away(prefix,params);
 310         }
 311         else if (command == "PING")
 312         {
 313                 this->LocalPing(prefix,params);
 314         }
 315         else if (command == "PONG")
 316         {
 317                 TreeServer *s = Utils->FindServer(prefix);
 318                 if (s && s->bursting)
 319                 {
 320                         ServerInstance->SNO->WriteGlobalSno('l',"Server \002%s\002 has not finished burst, forcing end of burst (send ENDBURST!)", prefix.c_str());
 321                         s->FinishBurst();
 322                 }
 323                 this->LocalPong(prefix,params);
 324         }
 325         else if (command == "VERSION")
 326         {
 327                 this->ServerVersion(prefix,params);
 328         }
 329         else if (command == "ADDLINE")
 330         {
 331                 this->AddLine(prefix,params);
 332         }
 333         else if (command == "DELLINE")
 334         {
 335                 this->DelLine(prefix,params);
 336         }
 337         else if (command == "SAVE")
 338         {
 339                 this->ForceNick(prefix,params);
 340         }
 341         else if (command == "OPERQUIT")
 342         {
 343                 this->OperQuit(prefix,params);
 344         }
 345         else if (command == "IDLE")
 346         {
 347                 this->Whois(prefix,params);
 348         }
 349         else if (command == "PUSH")
 350         {
 351                 this->Push(prefix,params);
 352         }
 353         else if (command == "SQUIT")
 354         {
 355                 if (params.size() == 2)
 356                 {
 357                         this->Squit(Utils->FindServer(params[0]),params[1]);
 358                 }
 359         }
 360         else if (command == "SNONOTICE")
 361         {
 362                 if (params.size() >= 2)
 363                 {
 364                         ServerInstance->SNO->WriteToSnoMask(*(params[0].c_str()), "From " + who->nick + ": "+ params[1]);
 365                         Utils->DoOneToAllButSender(prefix, command, params, prefix);
 366                 }
 367         }
 368         else if (command == "BURST")
 369         {
 370                 // Set prefix server as bursting
 371                 TreeServer* ServerSource = Utils->FindServer(prefix);
 372                 if (!ServerSource)
 373                 {
 374                         ServerInstance->SNO->WriteGlobalSno('l', "WTF: Got BURST from a non-server(?): %s", prefix.c_str());
 375                         return;
 376                 }
 377
 378                 ServerSource->bursting = true;
 379                 Utils->DoOneToAllButSender(prefix, command, params, prefix);
 380         }
 381         else if (command == "ENDBURST")
 382         {
 383                 TreeServer* ServerSource = Utils->FindServer(prefix);
 384                 if (!ServerSource)
 385                 {
 386                         ServerInstance->SNO->WriteGlobalSno('l', "WTF: Got ENDBURST from a non-server(?): %s", prefix.c_str());
 387                         return;
 388                 }
 389
 390                 ServerSource->FinishBurst();
 391                 Utils->DoOneToAllButSender(prefix, command, params, prefix);
 392         }
 393         else if (command == "ENCAP")
 394         {
 395                 this->Encap(who, params);
 396         }
 397         else if (command == "NICK")
 398         {
 399                 if (params.size() != 2)
 400                 {
 401                         SendError("Protocol violation: NICK message without TS - :"+std::string(who->uuid)+" NICK "+params[0]);
 402                         return;
 403                 }
 404
 405                 if (IS_SERVER(who))
 406                 {
 407                         SendError("Protocol violation: Server changing nick");
 408                         return;
 409                 }
 410
 411                 /* Update timestamp on user when they change nicks */
 412                 who->age = atoi(params[1].c_str());
 413
 414                 /*
 415                  * On nick messages, check that the nick doesnt already exist here.
 416                  * If it does, perform collision logic.
 417                  */
 418                 User* x = ServerInstance->FindNickOnly(params[0]);
 419                 if ((x) && (x != who))
 420                 {
 421                         int collideret = 0;
 422                         /* x is local, who is remote */
 423                         collideret = this->DoCollision(x, who->age, who->ident, who->GetIPString(), who->uuid);
 424                         if (collideret != 1)
 425                         {
 426                                 /*
 427                                  * Remote client lost, or both lost, parsing or passing on this
 428                                  * nickchange would be pointless, as the incoming client's server will
 429                                  * soon recieve SVSNICK to change its nick to its UID. :) -- w00t
 430                                  */
 431                                 return;
 432                         }
 433                 }
 434                 who->ForceNickChange(params[0].c_str());
 435                 Utils->RouteCommand(route_back_again, command, params, who);
 436         }
 437         else
 438         {
 439                 Command* cmd = ServerInstance->Parser->GetHandler(command);
 440                 CmdResult res = CMD_INVALID;
 441                 if (cmd && params.size() >= cmd->min_params)
 442                 {
 443                         res = cmd->Handle(params, who);
 444                 }
 445
 446                 if (res == CMD_INVALID)
 447                         SendError("Unrecognised or malformed command '" + command + "' -- possibly loaded mismatched modules");
 448                 if (res == CMD_SUCCESS)
 449                         Utils->RouteCommand(route_back_again, command, params, who);
 450         }
 451 }
 452
 453 std::string TreeSocket::GetName()
 454 {
 455         std::string sourceserv = this->myhost;
 456         if (!this->InboundServerName.empty())
 457         {
 458                 sourceserv = this->InboundServerName;
 459         }
 460         return sourceserv;
 461 }
 462
 463 void TreeSocket::OnTimeout()
 464 {
 465         ServerInstance->SNO->WriteGlobalSno('l', "CONNECT: Connection to \002%s\002 timed out.", myhost.c_str());
 466 }
 467
 468 void TreeSocket::Close()
 469 {
 470         this->BufferedSocket::Close();
 471         SetError("Remote host closed connection");
 472
 473         // Connection closed.
 474         // If the connection is fully up (state CONNECTED)
 475         // then propogate a netsplit to all peers.
 476         std::string quitserver = this->myhost;
 477         if (!this->InboundServerName.empty())
 478         {
 479                 quitserver = this->InboundServerName;
 480         }
 481         TreeServer* s = Utils->FindServer(quitserver);
 482         if (s && s->GetSocket() == this)
 483         {
 484                 Squit(s,getError());
 485         }
 486
 487         if (!quitserver.empty())
 488         {
 489                 ServerInstance->SNO->WriteGlobalSno('l', "Connection to '\2%s\2' failed.",quitserver.c_str());
 490
 491                 time_t server_uptime = ServerInstance->Time() - this->age;
 492                 if (server_uptime)
 493                         ServerInstance->SNO->WriteGlobalSno('l', "Connection to '\2%s\2' was established for %s", quitserver.c_str(), Utils->Creator->TimeToStr(server_uptime).c_str());
 494         }
 495 }