src/modules/m_spanningtree/treesocket2.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
   6  * See: http://www.inspircd.org/wiki/index.php/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "configreader.h"
  15 #include "users.h"
  16 #include "channels.h"
  17 #include "modules.h"
  18 #include "commands/cmd_whois.h"
  19 #include "commands/cmd_stats.h"
  20 #include "socket.h"
  21 #include "inspircd.h"
  22 #include "wildcard.h"
  23 #include "xline.h"
  24 #include "transport.h"
  25 #include "socketengine.h"
  26
  27 #include "m_spanningtree/main.h"
  28 #include "m_spanningtree/utils.h"
  29 #include "m_spanningtree/treeserver.h"
  30 #include "m_spanningtree/link.h"
  31 #include "m_spanningtree/treesocket.h"
  32 #include "m_spanningtree/resolvers.h"
  33 #include "m_spanningtree/handshaketimer.h"
  34
  35 /* $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 */
  36
  37 static std::map<std::string, std::string> warned;       /* Server names that have had protocol violation warnings displayed for them */
  38
  39 int TreeSocket::WriteLine(std::string line)
  40 {
  41         Instance->Log(DEBUG, "S[%d] -> %s", this->GetFd(), line.c_str());
  42         line.append("\r\n");
  43         return this->Write(line);
  44 }
  45
  46
  47 /* Handle ERROR command */
  48 bool TreeSocket::Error(std::deque<std::string> &params)
  49 {
  50         if (params.size() < 1)
  51                 return false;
  52         this->Instance->SNO->WriteToSnoMask('l',"ERROR from %s: %s",(InboundServerName != "" ? InboundServerName.c_str() : myhost.c_str()),params[0].c_str());
  53         /* we will return false to cause the socket to close. */
  54         return false;
  55 }
  56
  57 bool TreeSocket::Modules(const std::string &prefix, std::deque<std::string> &params)
  58 {
  59         if (params.empty())
  60                 return true;
  61
  62         if (!this->Instance->MatchText(this->Instance->Config->ServerName, params[0]))
  63         {
  64                 /* Pass it on, not for us */
  65                 Utils->DoOneToOne(prefix, "MODULES", params, params[0]);
  66                 return true;
  67         }
  68
  69         char strbuf[MAXBUF];
  70         std::deque<std::string> par;
  71         par.push_back(prefix);
  72         par.push_back("");
  73
  74         userrec* source = this->Instance->FindNick(prefix);
  75         if (!source)
  76                 return true;
  77
  78         for (unsigned int i = 0; i < Instance->Config->module_names.size(); i++)
  79         {
  80                 Version V = Instance->modules[i]->GetVersion();
  81                 char modulename[MAXBUF];
  82                 char flagstate[MAXBUF];
  83                 *flagstate = 0;
  84                 if (V.Flags & VF_STATIC)
  85                         strlcat(flagstate,", static",MAXBUF);
  86                 if (V.Flags & VF_VENDOR)
  87                         strlcat(flagstate,", vendor",MAXBUF);
  88                 if (V.Flags & VF_COMMON)
  89                         strlcat(flagstate,", common",MAXBUF);
  90                 if (V.Flags & VF_SERVICEPROVIDER)
  91                         strlcat(flagstate,", service provider",MAXBUF);
  92                 if (!flagstate[0])
  93                         strcpy(flagstate,"  <no flags>");
  94                 strlcpy(modulename,Instance->Config->module_names[i].c_str(),256);
  95                 if (*source->oper)
  96                 {
  97                         snprintf(strbuf, MAXBUF, "::%s 900 %s :0x%08lx %d.%d.%d.%d %s (%s)",Instance->Config->ServerName,source->nick,(long unsigned int)Instance->modules[i],V.Major,V.Minor,V.Revision,V.Build,ServerConfig::CleanFilename(modulename),flagstate+2);
  98                 }
  99                 else
 100                 {
 101                         snprintf(strbuf, MAXBUF, "::%s 900 %s :%s",Instance->Config->ServerName,source->nick,ServerConfig::CleanFilename(modulename));
 102                 }
 103                 par[1] = strbuf;
 104                 Utils->DoOneToOne(Instance->Config->ServerName, "PUSH", par, source->server);
 105         }
 106         snprintf(strbuf, MAXBUF, "::%s 901 %s :End of MODULES list", Instance->Config->ServerName, source->nick);
 107         par[1] = strbuf;
 108         Utils->DoOneToOne(Instance->Config->ServerName, "PUSH", par, source->server);
 109         return true;
 110 }
 111
 112 /** remote MOTD. leet, huh? */
 113 bool TreeSocket::Motd(const std::string &prefix, std::deque<std::string> &params)
 114 {
 115         if (params.size() > 0)
 116         {
 117                 if (this->Instance->MatchText(this->Instance->Config->ServerName, params[0]))
 118                 {
 119                         /* It's for our server */
 120                         string_list results;
 121                         userrec* source = this->Instance->FindNick(prefix);
 122
 123                         if (source)
 124                         {
 125                                 std::deque<std::string> par;
 126                                 par.push_back(prefix);
 127                                 par.push_back("");
 128
 129                                 if (!Instance->Config->MOTD.size())
 130                                 {
 131                                         par[1] = std::string("::")+Instance->Config->ServerName+" 422 "+source->nick+" :Message of the day file is missing.";
 132                                         Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
 133                                         return true;
 134                                 }
 135
 136                                 par[1] = std::string("::")+Instance->Config->ServerName+" 375 "+source->nick+" :"+Instance->Config->ServerName+" message of the day";
 137                                 Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
 138
 139                                 for (unsigned int i = 0; i < Instance->Config->MOTD.size(); i++)
 140                                 {
 141                                         par[1] = std::string("::")+Instance->Config->ServerName+" 372 "+source->nick+" :- "+Instance->Config->MOTD[i];
 142                                         Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
 143                                 }
 144
 145                                 par[1] = std::string("::")+Instance->Config->ServerName+" 376 "+source->nick+" End of message of the day.";
 146                                 Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
 147                         }
 148                 }
 149                 else
 150                 {
 151                         /* Pass it on */
 152                         userrec* source = this->Instance->FindNick(prefix);
 153                         if (source)
 154                                 Utils->DoOneToOne(prefix, "MOTD", params, params[0]);
 155                 }
 156         }
 157         return true;
 158 }
 159
 160 /** remote ADMIN. leet, huh? */
 161 bool TreeSocket::Admin(const std::string &prefix, std::deque<std::string> &params)
 162 {
 163         if (params.size() > 0)
 164         {
 165                 if (this->Instance->MatchText(this->Instance->Config->ServerName, params[0]))
 166                 {
 167                         /* It's for our server */
 168                         string_list results;
 169                         userrec* source = this->Instance->FindNick(prefix);
 170                         if (source)
 171                         {
 172                                 std::deque<std::string> par;
 173                                 par.push_back(prefix);
 174                                 par.push_back("");
 175                                 par[1] = std::string("::")+Instance->Config->ServerName+" 256 "+source->nick+" :Administrative info for "+Instance->Config->ServerName;
 176                                 Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
 177                                 par[1] = std::string("::")+Instance->Config->ServerName+" 257 "+source->nick+" :Name     - "+Instance->Config->AdminName;
 178                                 Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
 179                                 par[1] = std::string("::")+Instance->Config->ServerName+" 258 "+source->nick+" :Nickname - "+Instance->Config->AdminNick;
 180                                 Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
 181                                 par[1] = std::string("::")+Instance->Config->ServerName+" 258 "+source->nick+" :E-Mail   - "+Instance->Config->AdminEmail;
 182                                 Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
 183                         }
 184                 }
 185                 else
 186                 {
 187                         /* Pass it on */
 188                         userrec* source = this->Instance->FindNick(prefix);
 189                         if (source)
 190                                 Utils->DoOneToOne(prefix, "ADMIN", params, params[0]);
 191                 }
 192         }
 193         return true;
 194 }
 195
 196 bool TreeSocket::Stats(const std::string &prefix, std::deque<std::string> &params)
 197 {
 198         /* Get the reply to a STATS query if it matches this servername,
 199          * and send it back as a load of PUSH queries
 200          */
 201         if (params.size() > 1)
 202         {
 203                 if (this->Instance->MatchText(this->Instance->Config->ServerName, params[1]))
 204                 {
 205                         /* It's for our server */
 206                         string_list results;
 207                         userrec* source = this->Instance->FindNick(prefix);
 208                         if (source)
 209                         {
 210                                 std::deque<std::string> par;
 211                                 par.push_back(prefix);
 212                                 par.push_back("");
 213                                 DoStats(this->Instance, *(params[0].c_str()), source, results);
 214                                 for (size_t i = 0; i < results.size(); i++)
 215                                 {
 216                                         par[1] = "::" + results[i];
 217                                         Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
 218                                 }
 219                         }
 220                 }
 221                 else
 222                 {
 223                         /* Pass it on */
 224                         userrec* source = this->Instance->FindNick(prefix);
 225                         if (source)
 226                                 Utils->DoOneToOne(prefix, "STATS", params, params[1]);
 227                 }
 228         }
 229         return true;
 230 }
 231
 232
 233 /** Because the core won't let users or even SERVERS set +o,
 234  * we use the OPERTYPE command to do this.
 235  */
 236 bool TreeSocket::OperType(const std::string &prefix, std::deque<std::string> &params)
 237 {
 238         if (params.size() != 1)
 239                 return true;
 240         std::string opertype = params[0];
 241         userrec* u = this->Instance->FindNick(prefix);
 242         if (u)
 243         {
 244                 u->modes[UM_OPERATOR] = 1;
 245                 this->Instance->all_opers.push_back(u);
 246                 strlcpy(u->oper,opertype.c_str(),NICKMAX-1);
 247                 Utils->DoOneToAllButSender(u->nick,"OPERTYPE",params,u->server);
 248                 this->Instance->SNO->WriteToSnoMask('o',"From %s: User %s (%s@%s) is now an IRC operator of type %s",u->server, u->nick,u->ident,u->host,irc::Spacify(opertype.c_str()));
 249         }
 250         return true;
 251 }
 252
 253 /** Because Andy insists that services-compatible servers must
 254  * implement SVSNICK and SVSJOIN, that's exactly what we do :p
 255  */
 256 bool TreeSocket::ForceNick(const std::string &prefix, std::deque<std::string> &params)
 257 {
 258         if (params.size() < 3)
 259                 return true;
 260
 261         userrec* u = this->Instance->FindNick(params[0]);
 262
 263         if (u)
 264         {
 265                 Utils->DoOneToAllButSender(prefix,"SVSNICK",params,prefix);
 266                 if (IS_LOCAL(u))
 267                 {
 268                         std::deque<std::string> par;
 269                         par.push_back(params[1]);
 270                         if (!u->ForceNickChange(params[1].c_str()))
 271                         {
 272                                 userrec::QuitUser(this->Instance, u, "Nickname collision");
 273                                 return true;
 274                         }
 275                         u->age = atoi(params[2].c_str());
 276                 }
 277         }
 278         return true;
 279 }
 280
 281 bool TreeSocket::OperQuit(const std::string &prefix, std::deque<std::string> &params)
 282 {
 283         if (params.size() < 1)
 284                 return true;
 285
 286         userrec* u = this->Instance->FindNick(prefix);
 287
 288         if (u)
 289         {
 290                 u->SetOperQuit(params[0]);
 291                 params[0] = ":" + params[0];
 292                 Utils->DoOneToAllButSender(prefix,"OPERQUIT",params,prefix);
 293         }
 294         return true;
 295 }
 296
 297 /*
 298  * Remote SQUIT (RSQUIT). Routing works similar to SVSNICK: Route it to the server that the target is connected to locally,
 299  * then let that server do the dirty work (squit it!). Example:
 300  * A -> B -> C -> D: oper on A squits D, A routes to B, B routes to C, C notices D connected locally, kills it. -- w00t
 301  */
 302 bool TreeSocket::RemoteSquit(const std::string &prefix, std::deque<std::string> &params)
 303 {
 304         /* ok.. :w00t RSQUIT jupe.barafranca.com :reason here */
 305         if (params.size() < 2)
 306                 return true;
 307
 308         TreeServer* s = Utils->FindServerMask(params[0]);
 309
 310         if (s)
 311         {
 312                 if (s == Utils->TreeRoot)
 313                 {
 314                         this->Instance->SNO->WriteToSnoMask('l',"What the fuck, I recieved a remote SQUIT for myself? :< (from %s", prefix.c_str());
 315                         return true;
 316                 }
 317
 318                 TreeSocket* sock = s->GetSocket();
 319
 320                 if (sock)
 321                 {
 322                         /* it's locally connected, KILL IT! */
 323                         Instance->SNO->WriteToSnoMask('l',"RSQUIT: Server \002%s\002 removed from network by %s: %s", params[0].c_str(), prefix.c_str(), params[1].c_str());
 324                         sock->Squit(s,"Server quit by " + prefix + ": " + params[1]);
 325                         Instance->SE->DelFd(sock);
 326                         sock->Close();
 327                         delete sock;
 328                 }
 329                 else
 330                 {
 331                         /* route the rsquit */
 332                         params[1] = ":" + params[1];
 333                         Utils->DoOneToOne(prefix, "RSQUIT", params, params[0]);
 334                 }
 335         }
 336         else
 337         {
 338                 /* mother fucker! it doesn't exist */
 339         }
 340
 341         return true;
 342 }
 343
 344 bool TreeSocket::ServiceJoin(const std::string &prefix, std::deque<std::string> &params)
 345 {
 346         if (params.size() < 2)
 347                 return true;
 348
 349         userrec* u = this->Instance->FindNick(params[0]);
 350
 351         if (u)
 352         {
 353                 /* only join if it's local, otherwise just pass it on! */
 354                 if (IS_LOCAL(u))
 355                         chanrec::JoinUser(this->Instance, u, params[1].c_str(), false, "", Instance->Time());
 356                 Utils->DoOneToAllButSender(prefix,"SVSJOIN",params,prefix);
 357         }
 358         return true;
 359 }
 360
 361 bool TreeSocket::RemoteRehash(const std::string &prefix, std::deque<std::string> &params)
 362 {
 363         if (params.size() < 1)
 364                 return false;
 365
 366         std::string servermask = params[0];
 367
 368         if (this->Instance->MatchText(this->Instance->Config->ServerName,servermask))
 369         {
 370                 this->Instance->SNO->WriteToSnoMask('l',"Remote rehash initiated by \002"+prefix+"\002.");
 371                 this->Instance->RehashServer();
 372                 Utils->ReadConfiguration(false);
 373                 InitializeDisabledCommands(Instance->Config->DisabledCommands, Instance);
 374         }
 375         Utils->DoOneToAllButSender(prefix,"REHASH",params,prefix);
 376         return true;
 377 }
 378
 379 bool TreeSocket::RemoteKill(const std::string &prefix, std::deque<std::string> &params)
 380 {
 381         if (params.size() != 2)
 382                 return true;
 383
 384         std::string nick = params[0];
 385         userrec* u = this->Instance->FindNick(prefix);
 386         userrec* who = this->Instance->FindNick(nick);
 387
 388         if (who)
 389         {
 390                 /* Prepend kill source, if we don't have one */
 391                 std::string sourceserv = prefix;
 392                 if (u)
 393                 {
 394                         sourceserv = u->server;
 395                 }
 396                 if (*(params[1].c_str()) != '[')
 397                 {
 398                         params[1] = "[" + sourceserv + "] Killed (" + params[1] +")";
 399                 }
 400                 std::string reason = params[1];
 401                 params[1] = ":" + params[1];
 402                 Utils->DoOneToAllButSender(prefix,"KILL",params,sourceserv);
 403                 who->Write(":%s KILL %s :%s (%s)", sourceserv.c_str(), who->nick, sourceserv.c_str(), reason.c_str());
 404                 userrec::QuitUser(this->Instance,who,reason);
 405         }
 406         return true;
 407 }
 408
 409 bool TreeSocket::LocalPong(const std::string &prefix, std::deque<std::string> &params)
 410 {
 411         if (params.size() < 1)
 412                 return true;
 413
 414         if (params.size() == 1)
 415         {
 416                 TreeServer* ServerSource = Utils->FindServer(prefix);
 417                 if (ServerSource)
 418                 {
 419                         ServerSource->SetPingFlag();
 420                         ServerSource->rtt = Instance->Time() - ServerSource->LastPing;
 421                 }
 422         }
 423         else
 424         {
 425                 std::string forwardto = params[1];
 426                 if (forwardto == this->Instance->Config->ServerName)
 427                 {
 428                         /*
 429                          * this is a PONG for us
 430                          * if the prefix is a user, check theyre local, and if they are,
 431                          * dump the PONG reply back to their fd. If its a server, do nowt.
 432                          * Services might want to send these s->s, but we dont need to yet.
 433                          */
 434                         userrec* u = this->Instance->FindNick(prefix);
 435                         if (u)
 436                         {
 437                                 u->WriteServ("PONG %s %s",params[0].c_str(),params[1].c_str());
 438                         }
 439                 }
 440                 else
 441                 {
 442                         // not for us, pass it on :)
 443                         Utils->DoOneToOne(prefix,"PONG",params,forwardto);
 444                 }
 445         }
 446
 447         return true;
 448 }
 449
 450 bool TreeSocket::MetaData(const std::string &prefix, std::deque<std::string> &params)
 451 {
 452         if (params.size() < 2)
 453                 return true;
 454         else if (params.size() < 3)
 455                 params.push_back("");
 456         TreeServer* ServerSource = Utils->FindServer(prefix);
 457         if (ServerSource)
 458         {
 459                 Utils->SetRemoteBursting(ServerSource, false);
 460
 461                 if (params[0] == "*")
 462                 {
 463                         FOREACH_MOD_I(this->Instance,I_OnDecodeMetaData,OnDecodeMetaData(TYPE_OTHER,NULL,params[1],params[2]));
 464                 }
 465                 else if (*(params[0].c_str()) == '#')
 466                 {
 467                         chanrec* c = this->Instance->FindChan(params[0]);
 468                         if (c)
 469                         {
 470                                 FOREACH_MOD_I(this->Instance,I_OnDecodeMetaData,OnDecodeMetaData(TYPE_CHANNEL,c,params[1],params[2]));
 471                         }
 472                 }
 473                 else if (*(params[0].c_str()) != '#')
 474                 {
 475                         userrec* u = this->Instance->FindNick(params[0]);
 476                         if (u)
 477                         {
 478                                 FOREACH_MOD_I(this->Instance,I_OnDecodeMetaData,OnDecodeMetaData(TYPE_USER,u,params[1],params[2]));
 479                         }
 480                 }
 481         }
 482
 483         params[2] = ":" + params[2];
 484         Utils->DoOneToAllButSender(prefix,"METADATA",params,prefix);
 485         return true;
 486 }
 487
 488 bool TreeSocket::ServerVersion(const std::string &prefix, std::deque<std::string> &params)
 489 {
 490         if (params.size() < 1)
 491                 return true;
 492
 493         TreeServer* ServerSource = Utils->FindServer(prefix);
 494
 495         if (ServerSource)
 496         {
 497                 ServerSource->SetVersion(params[0]);
 498         }
 499         params[0] = ":" + params[0];
 500         Utils->DoOneToAllButSender(prefix,"VERSION",params,prefix);
 501         return true;
 502 }
 503
 504 bool TreeSocket::ChangeHost(const std::string &prefix, std::deque<std::string> &params)
 505 {
 506         if (params.size() < 1)
 507                 return true;
 508         userrec* u = this->Instance->FindNick(prefix);
 509
 510         if (u)
 511         {
 512                 u->ChangeDisplayedHost(params[0].c_str());
 513                 Utils->DoOneToAllButSender(prefix,"FHOST",params,u->server);
 514         }
 515         return true;
 516 }
 517
 518 bool TreeSocket::AddLine(const std::string &prefix, std::deque<std::string> &params)
 519 {
 520         if (params.size() < 6)
 521                 return true;
 522         bool propogate = false;
 523         if (!this->bursting)
 524                 Utils->lines_to_apply = 0;
 525         switch (*(params[0].c_str()))
 526         {
 527                 case 'Z':
 528                         propogate = Instance->XLines->add_zline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
 529                         Instance->XLines->zline_set_creation_time(params[1].c_str(), atoi(params[3].c_str()));
 530                         if (propogate)
 531                                 Utils->lines_to_apply |= APPLY_ZLINES;
 532                 break;
 533                 case 'Q':
 534                         propogate = Instance->XLines->add_qline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
 535                         Instance->XLines->qline_set_creation_time(params[1].c_str(), atoi(params[3].c_str()));
 536                         if (propogate)
 537                                 Utils->lines_to_apply |= APPLY_QLINES;
 538                 break;
 539                 case 'E':
 540                         propogate = Instance->XLines->add_eline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
 541                         Instance->XLines->eline_set_creation_time(params[1].c_str(), atoi(params[3].c_str()));
 542                 break;
 543                 case 'G':
 544                         propogate = Instance->XLines->add_gline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
 545                         Instance->XLines->gline_set_creation_time(params[1].c_str(), atoi(params[3].c_str()));
 546                         if (propogate)
 547                                 Utils->lines_to_apply |= APPLY_GLINES;
 548                 break;
 549                 case 'K':
 550                         propogate = Instance->XLines->add_kline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
 551                         if (propogate)
 552                                 Utils->lines_to_apply |= APPLY_KLINES;
 553                 break;
 554                 default:
 555                         /* Just in case... */
 556                         this->Instance->SNO->WriteToSnoMask('x',"\2WARNING\2: Invalid xline type '"+params[0]+"' sent by server "+prefix+", ignored!");
 557                         propogate = false;
 558                 break;
 559         }
 560         /* Send it on its way */
 561         if (propogate)
 562         {
 563                 if (atoi(params[4].c_str()))
 564                 {
 565                         time_t c_requires_crap = ConvToInt(params[4]) + Instance->Time();
 566                         this->Instance->SNO->WriteToSnoMask('x',"%s Added %cLINE on %s to expire on %s (%s).",prefix.c_str(),*(params[0].c_str()),params[1].c_str(),Instance->TimeString(c_requires_crap).c_str(),params[5].c_str());
 567                 }
 568                 else
 569                 {
 570                         this->Instance->SNO->WriteToSnoMask('x',"%s Added permenant %cLINE on %s (%s).",prefix.c_str(),*(params[0].c_str()),params[1].c_str(),params[5].c_str());
 571                 }
 572                 params[5] = ":" + params[5];
 573                 Utils->DoOneToAllButSender(prefix,"ADDLINE",params,prefix);
 574         }
 575         if (!this->bursting)
 576         {
 577                 Instance->XLines->apply_lines(Utils->lines_to_apply);
 578                 Utils->lines_to_apply = 0;
 579         }
 580         return true;
 581 }
 582
 583 bool TreeSocket::ChangeName(const std::string &prefix, std::deque<std::string> &params)
 584 {
 585         if (params.size() < 1)
 586                 return true;
 587         userrec* u = this->Instance->FindNick(prefix);
 588         if (u)
 589         {
 590                 u->ChangeName(params[0].c_str());
 591                 params[0] = ":" + params[0];
 592                 Utils->DoOneToAllButSender(prefix,"FNAME",params,u->server);
 593         }
 594         return true;
 595 }
 596
 597 bool TreeSocket::Whois(const std::string &prefix, std::deque<std::string> &params)
 598 {
 599         if (params.size() < 1)
 600                 return true;
 601         userrec* u = this->Instance->FindNick(prefix);
 602         if (u)
 603         {
 604                 // an incoming request
 605                 if (params.size() == 1)
 606                 {
 607                         userrec* x = this->Instance->FindNick(params[0]);
 608                         if ((x) && (IS_LOCAL(x)))
 609                         {
 610                                 userrec* x = this->Instance->FindNick(params[0]);
 611                                 char signon[MAXBUF];
 612                                 char idle[MAXBUF];
 613                                 snprintf(signon, MAXBUF, "%lu", (unsigned long)x->signon);
 614                                 snprintf(idle, MAXBUF, "%lu", (unsigned long)abs((x->idle_lastmsg) - Instance->Time(true)));
 615                                 std::deque<std::string> par;
 616                                 par.push_back(prefix);
 617                                 par.push_back(signon);
 618                                 par.push_back(idle);
 619                                 // ours, we're done, pass it BACK
 620                                 Utils->DoOneToOne(params[0], "IDLE", par, u->server);
 621                         }
 622                         else
 623                         {
 624                                 // not ours pass it on
 625                                 if (x)
 626                                         Utils->DoOneToOne(prefix, "IDLE", params, x->server);
 627                         }
 628                 }
 629                 else if (params.size() == 3)
 630                 {
 631                         std::string who_did_the_whois = params[0];
 632                         userrec* who_to_send_to = this->Instance->FindNick(who_did_the_whois);
 633                         if ((who_to_send_to) && (IS_LOCAL(who_to_send_to)))
 634                         {
 635                                 // an incoming reply to a whois we sent out
 636                                 std::string nick_whoised = prefix;
 637                                 unsigned long signon = atoi(params[1].c_str());
 638                                 unsigned long idle = atoi(params[2].c_str());
 639                                 if ((who_to_send_to) && (IS_LOCAL(who_to_send_to)))
 640                                 {
 641                                         do_whois(this->Instance, who_to_send_to, u, signon, idle, nick_whoised.c_str());
 642                                 }
 643                         }
 644                         else
 645                         {
 646                                 // not ours, pass it on
 647                                 if (who_to_send_to)
 648                                         Utils->DoOneToOne(prefix, "IDLE", params, who_to_send_to->server);
 649                         }
 650                 }
 651         }
 652         return true;
 653 }
 654
 655 bool TreeSocket::Push(const std::string &prefix, std::deque<std::string> &params)
 656 {
 657         if (params.size() < 2)
 658                 return true;
 659         userrec* u = this->Instance->FindNick(params[0]);
 660         if (!u)
 661                 return true;
 662         if (IS_LOCAL(u))
 663         {
 664                 u->Write(params[1]);
 665         }
 666         else
 667         {
 668                 // continue the raw onwards
 669                 params[1] = ":" + params[1];
 670                 Utils->DoOneToOne(prefix,"PUSH",params,u->server);
 671         }
 672         return true;
 673 }
 674
 675 bool TreeSocket::HandleSetTime(const std::string &prefix, std::deque<std::string> &params)
 676 {
 677         if (!params.size() || !Utils->EnableTimeSync)
 678                 return true;
 679
 680         bool force = false;
 681
 682         if ((params.size() == 2) && (params[1] == "FORCE"))
 683                 force = true;
 684
 685         time_t them = atoi(params[0].c_str());
 686         time_t us = Instance->Time(false);
 687
 688         time_t diff = them - us;
 689
 690         Utils->DoOneToAllButSender(prefix, "TIMESET", params, prefix);
 691
 692         if (force || (them != us))
 693         {
 694                 time_t old = Instance->SetTimeDelta(diff);
 695                 Instance->Log(DEBUG, "TS (diff %d) from %s applied (old delta was %d)", diff, prefix.c_str(), old);
 696         }
 697
 698         return true;
 699 }
 700
 701 bool TreeSocket::Time(const std::string &prefix, std::deque<std::string> &params)
 702 {
 703         // :source.server TIME remote.server sendernick
 704         // :remote.server TIME source.server sendernick TS
 705         if (params.size() == 2)
 706         {
 707                 // someone querying our time?
 708                 if (this->Instance->Config->ServerName == params[0])
 709                 {
 710                         userrec* u = this->Instance->FindNick(params[1]);
 711                         if (u)
 712                         {
 713                                 params.push_back(ConvToStr(Instance->Time(false)));
 714                                 params[0] = prefix;
 715                                 Utils->DoOneToOne(this->Instance->Config->ServerName,"TIME",params,params[0]);
 716                         }
 717                 }
 718                 else
 719                 {
 720                         // not us, pass it on
 721                         userrec* u = this->Instance->FindNick(params[1]);
 722                         if (u)
 723                                 Utils->DoOneToOne(prefix,"TIME",params,params[0]);
 724                 }
 725         }
 726         else if (params.size() == 3)
 727         {
 728                 // a response to a previous TIME
 729                 userrec* u = this->Instance->FindNick(params[1]);
 730                 if ((u) && (IS_LOCAL(u)))
 731                 {
 732                         time_t rawtime = atol(params[2].c_str());
 733                         struct tm * timeinfo;
 734                         timeinfo = localtime(&rawtime);
 735                         char tms[26];
 736                         snprintf(tms,26,"%s",asctime(timeinfo));
 737                         tms[24] = 0;
 738                         u->WriteServ("391 %s %s :%s",u->nick,prefix.c_str(),tms);
 739                 }
 740                 else
 741                 {
 742                         if (u)
 743                                 Utils->DoOneToOne(prefix,"TIME",params,u->server);
 744                 }
 745         }
 746         return true;
 747 }
 748
 749 bool TreeSocket::LocalPing(const std::string &prefix, std::deque<std::string> &params)
 750 {
 751         if (params.size() < 1)
 752                 return true;
 753         if (params.size() == 1)
 754         {
 755                 std::string stufftobounce = params[0];
 756                 this->WriteLine(std::string(":")+this->Instance->Config->ServerName+" PONG "+stufftobounce);
 757                 return true;
 758         }
 759         else
 760         {
 761                 std::string forwardto = params[1];
 762                 if (forwardto == this->Instance->Config->ServerName)
 763                 {
 764                         // this is a ping for us, send back PONG to the requesting server
 765                         params[1] = params[0];
 766                         params[0] = forwardto;
 767                         Utils->DoOneToOne(forwardto,"PONG",params,params[1]);
 768                 }
 769                 else
 770                 {
 771                         // not for us, pass it on :)
 772                         Utils->DoOneToOne(prefix,"PING",params,forwardto);
 773                 }
 774                 return true;
 775         }
 776 }
 777
 778 /** TODO: This creates a total mess of output and needs to really use irc::modestacker.
 779  */
 780 bool TreeSocket::RemoveStatus(const std::string &prefix, std::deque<std::string> &params)
 781 {
 782         if (params.size() < 1)
 783                 return true;
 784         chanrec* c = Instance->FindChan(params[0]);
 785         if (c)
 786         {
 787                 for (char modeletter = 'A'; modeletter <= 'z'; modeletter++)
 788                 {
 789                         ModeHandler* mh = Instance->Modes->FindMode(modeletter, MODETYPE_CHANNEL);
 790                         if (mh)
 791                                 mh->RemoveMode(c);
 792                 }
 793         }
 794         return true;
 795 }
 796
 797 bool TreeSocket::RemoteServer(const std::string &prefix, std::deque<std::string> &params)
 798 {
 799         if (params.size() < 4)
 800                 return false;
 801         std::string servername = params[0];
 802         std::string password = params[1];
 803         // hopcount is not used for a remote server, we calculate this ourselves
 804         std::string description = params[3];
 805         TreeServer* ParentOfThis = Utils->FindServer(prefix);
 806         if (!ParentOfThis)
 807         {
 808                 this->SendError("Protocol error - Introduced remote server from unknown server "+prefix);
 809                 return false;
 810         }
 811         TreeServer* CheckDupe = Utils->FindServer(servername);
 812         if (CheckDupe)
 813         {
 814                 this->SendError("Server "+servername+" already exists!");
 815                 this->Instance->SNO->WriteToSnoMask('l',"Server \2"+servername+"\2 being introduced from \2" + prefix + "\2 denied, already exists. Closing link with " + prefix);
 816                 return false;
 817         }
 818         Link* lnk = Utils->FindLink(servername);
 819         TreeServer* Node = new TreeServer(this->Utils,this->Instance,servername,description,ParentOfThis,NULL, lnk ? lnk->Hidden : false);
 820         ParentOfThis->AddChild(Node);
 821         params[3] = ":" + params[3];
 822         Utils->SetRemoteBursting(Node, true);
 823         Utils->DoOneToAllButSender(prefix,"SERVER",params,prefix);
 824         this->Instance->SNO->WriteToSnoMask('l',"Server \002"+prefix+"\002 introduced server \002"+servername+"\002 ("+description+")");
 825         return true;
 826 }
 827
 828 bool TreeSocket::ComparePass(const std::string &ours, const std::string &theirs)
 829 {
 830         if ((!strncmp(ours.c_str(), "HMAC-SHA256:", 12)) || (!strncmp(theirs.c_str(), "HMAC-SHA256:", 12)))
 831         {
 832                 /* One or both of us specified hmac sha256, but we don't have sha256 module loaded!
 833                  * We can't allow this password as valid.
 834                  */
 835                 if (!Instance->FindModule("m_sha256.so") || !Utils->ChallengeResponse)
 836                                 return false;
 837                 else
 838                         /* Straight string compare of hashes */
 839                         return ours == theirs;
 840         }
 841         else
 842                 /* Straight string compare of plaintext */
 843                 return ours == theirs;
 844 }
 845
 846 bool TreeSocket::Outbound_Reply_Server(std::deque<std::string> &params)
 847 {
 848         if (params.size() < 4)
 849                 return false;
 850
 851         irc::string servername = params[0].c_str();
 852         std::string sname = params[0];
 853         std::string password = params[1];
 854         std::string description = params[3];
 855         int hops = atoi(params[2].c_str());
 856
 857         this->InboundServerName = sname;
 858         this->InboundDescription = description;
 859
 860         if (hops)
 861         {
 862                 this->SendError("Server too far away for authentication");
 863                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, server is too far away for authentication");
 864                 return false;
 865         }
 866
 867         for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
 868         {
 869                 if ((x->Name == servername) && ((ComparePass(this->MakePass(x->RecvPass,this->GetOurChallenge()),password)) || (x->RecvPass == password && (this->GetTheirChallenge().empty()))))
 870                 {
 871                         TreeServer* CheckDupe = Utils->FindServer(sname);
 872                         if (CheckDupe)
 873                         {
 874                                 this->SendError("Server "+sname+" already exists on server "+CheckDupe->GetParent()->GetName()+"!");
 875                                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName());
 876                                 return false;
 877                         }
 878                         // Begin the sync here. this kickstarts the
 879                         // other side, waiting in WAIT_AUTH_2 state,
 880                         // into starting their burst, as it shows
 881                         // that we're happy.
 882                         this->LinkState = CONNECTED;
 883                         // we should add the details of this server now
 884                         // to the servers tree, as a child of the root
 885                         // node.
 886                         TreeServer* Node = new TreeServer(this->Utils,this->Instance,sname,description,Utils->TreeRoot,this,x->Hidden);
 887                         Utils->TreeRoot->AddChild(Node);
 888                         params[3] = ":" + params[3];
 889                         Utils->DoOneToAllButSender(Utils->TreeRoot->GetName(),"SERVER",params,sname);
 890                         this->bursting = true;
 891                         this->DoBurst(Node);
 892                         return true;
 893                 }
 894         }
 895         this->SendError("Invalid credentials");
 896         this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, invalid link credentials");
 897         return false;
 898 }
 899
 900 bool TreeSocket::Inbound_Server(std::deque<std::string> &params)
 901 {
 902         if (params.size() < 4)
 903                 return false;
 904         irc::string servername = params[0].c_str();
 905         std::string sname = params[0];
 906         std::string password = params[1];
 907         std::string description = params[3];
 908         int hops = atoi(params[2].c_str());
 909
 910         this->InboundServerName = sname;
 911         this->InboundDescription = description;
 912
 913         if (hops)
 914         {
 915                 this->SendError("Server too far away for authentication");
 916                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, server is too far away for authentication");
 917                 return false;
 918         }
 919
 920         for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
 921         {
 922                 if ((x->Name == servername) && ((ComparePass(this->MakePass(x->RecvPass,this->GetOurChallenge()),password) || x->RecvPass == password && (this->GetTheirChallenge().empty()))))
 923                 {
 924                         /* First check for instances of the server that are waiting between the inbound and outbound SERVER command */
 925                         TreeSocket* CheckDupeSocket = Utils->FindBurstingServer(sname);
 926                         if (CheckDupeSocket)
 927                         {
 928                                 /* If we find one, we abort the link to prevent a race condition */
 929                                 this->SendError("Negotiation collision");
 930                                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, already exists in a negotiating state.");
 931                                 CheckDupeSocket->SendError("Negotiation collision");
 932                                 Instance->SE->DelFd(CheckDupeSocket);
 933                                 CheckDupeSocket->Close();
 934                                 delete CheckDupeSocket;
 935                                 return false;
 936                         }
 937                         /* Now check for fully initialized instances of the server */
 938                         TreeServer* CheckDupe = Utils->FindServer(sname);
 939                         if (CheckDupe)
 940                         {
 941                                 this->SendError("Server "+sname+" already exists on server "+CheckDupe->GetParent()->GetName()+"!");
 942                                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName());
 943                                 return false;
 944                         }
 945                         this->Instance->SNO->WriteToSnoMask('l',"Verified incoming server connection from \002"+sname+"\002["+(x->HiddenFromStats ? "<hidden>" : this->GetIP())+"] ("+description+")");
 946                         if (this->Hook)
 947                         {
 948                                 std::string name = InspSocketNameRequest((Module*)Utils->Creator, this->Hook).Send();
 949                                 this->Instance->SNO->WriteToSnoMask('l',"Connection from \2"+sname+"\2["+(x->HiddenFromStats ? "<hidden>" : this->GetIP())+"] using transport \2"+name+"\2");
 950                         }
 951
 952                         Utils->AddBurstingServer(sname,this);
 953
 954                         // this is good. Send our details: Our server name and description and hopcount of 0,
 955                         // along with the sendpass from this block.
 956                         this->WriteLine(std::string("SERVER ")+this->Instance->Config->ServerName+" "+this->MakePass(x->SendPass, this->GetTheirChallenge())+" 0 :"+this->Instance->Config->ServerDesc);
 957                         // move to the next state, we are now waiting for THEM.
 958                         this->LinkState = WAIT_AUTH_2;
 959                         return true;
 960                 }
 961         }
 962         this->SendError("Invalid credentials");
 963         this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, invalid link credentials");
 964         return false;
 965 }
 966
 967 void TreeSocket::Split(const std::string &line, std::deque<std::string> &n)
 968 {
 969         n.clear();
 970         irc::tokenstream tokens(line);
 971         std::string param;
 972         while (tokens.GetToken(param))
 973         {
 974                 if (!param.empty())
 975                         n.push_back(param);
 976         }
 977         return;
 978 }
 979
 980 bool TreeSocket::ProcessLine(std::string &line)
 981 {
 982         std::deque<std::string> params;
 983         irc::string command;
 984         std::string prefix;
 985
 986         line = line.substr(0, line.find_first_of("\r\n"));
 987
 988         if (line.empty())
 989                 return true;
 990
 991         Instance->Log(DEBUG, "S[%d] <- %s", this->GetFd(), line.c_str());
 992
 993         this->Split(line.c_str(),params);
 994
 995         if (params.empty())
 996                 return true;
 997
 998         if ((params[0][0] == ':') && (params.size() > 1))
 999         {
1000                 prefix = params[0].substr(1);
1001                 params.pop_front();
1002         }
1003         command = params[0].c_str();
1004         params.pop_front();
1005         switch (this->LinkState)
1006         {
1007                 TreeServer* Node;
1008
1009                 case WAIT_AUTH_1:
1010                         // Waiting for SERVER command from remote server. Server initiating
1011                         // the connection sends the first SERVER command, listening server
1012                         // replies with theirs if its happy, then if the initiator is happy,
1013                         // it starts to send its net sync, which starts the merge, otherwise
1014                         // it sends an ERROR.
1015                         if (command == "PASS")
1016                         {
1017                                 /* Silently ignored */
1018                         }
1019                         else if (command == "SERVER")
1020                         {
1021                                 return this->Inbound_Server(params);
1022                         }
1023                         else if (command == "ERROR")
1024                         {
1025                                 return this->Error(params);
1026                         }
1027                         else if (command == "USER")
1028                         {
1029                                 this->SendError("Client connections to this port are prohibited.");
1030                                 return false;
1031                         }
1032                         else if (command == "CAPAB")
1033                         {
1034                                 return this->Capab(params);
1035                         }
1036                         else if ((command == "U") || (command == "S"))
1037                         {
1038                                 this->SendError("Cannot use the old-style mesh linking protocol with m_spanningtree.so!");
1039                                 return false;
1040                         }
1041                         else
1042                         {
1043                                 irc::string error = "Invalid command in negotiation phase: " + command;
1044                                 this->SendError(assign(error));
1045                                 return false;
1046                         }
1047                 break;
1048                 case WAIT_AUTH_2:
1049                         // Waiting for start of other side's netmerge to say they liked our
1050                         // password.
1051                         if (command == "SERVER")
1052                         {
1053                                 // cant do this, they sent it to us in the WAIT_AUTH_1 state!
1054                                 // silently ignore.
1055                                 return true;
1056                         }
1057                         else if ((command == "U") || (command == "S"))
1058                         {
1059                                 this->SendError("Cannot use the old-style mesh linking protocol with m_spanningtree.so!");
1060                                 return false;
1061                         }
1062                         else if (command == "BURST")
1063                         {
1064                                 if (params.size() && Utils->EnableTimeSync)
1065                                 {
1066                                         bool we_have_delta = (Instance->Time(false) != Instance->Time(true));
1067                                         time_t them = atoi(params[0].c_str());
1068                                         time_t delta = them - Instance->Time(false);
1069                                         if ((delta < -300) || (delta > 300))
1070                                         {
1071                                                 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));
1072                                                 SendError("Your clocks are out by "+ConvToStr(abs(delta))+" seconds (this is more than ten minutes). Link aborted, PLEASE SYNC YOUR CLOCKS!");
1073                                                 return false;
1074                                         }
1075                                         else if ((delta < -30) || (delta > 30))
1076                                         {
1077                                                 Instance->SNO->WriteToSnoMask('l',"\2WARNING\2: Your clocks are out by %d seconds. Please consider synching your clocks.", abs(delta));
1078                                         }
1079
1080                                         if (!Utils->MasterTime && !we_have_delta)
1081                                         {
1082                                                 this->Instance->SetTimeDelta(delta);
1083                                                 // Send this new timestamp to any other servers
1084                                                 Utils->DoOneToMany(Utils->TreeRoot->GetName(), "TIMESET", params);
1085                                         }
1086                                 }
1087                                 this->LinkState = CONNECTED;
1088                                 Link* lnk = Utils->FindLink(InboundServerName);
1089                                 Node = new TreeServer(this->Utils,this->Instance, InboundServerName, InboundDescription, Utils->TreeRoot, this, lnk ? lnk->Hidden : false);
1090                                 Utils->DelBurstingServer(this);
1091                                 Utils->TreeRoot->AddChild(Node);
1092                                 params.clear();
1093                                 params.push_back(InboundServerName);
1094                                 params.push_back("*");
1095                                 params.push_back("1");
1096                                 params.push_back(":"+InboundDescription);
1097                                 Utils->DoOneToAllButSender(Utils->TreeRoot->GetName(),"SERVER",params,InboundServerName);
1098                                 this->bursting = true;
1099                                 this->DoBurst(Node);
1100                         }
1101                         else if (command == "ERROR")
1102                         {
1103                                 return this->Error(params);
1104                         }
1105                         else if (command == "CAPAB")
1106                         {
1107                                 return this->Capab(params);
1108                         }
1109
1110                 break;
1111                 case LISTENER:
1112                         this->SendError("Internal error -- listening socket accepted its own descriptor!!!");
1113                         return false;
1114                 break;
1115                 case CONNECTING:
1116                         if (command == "SERVER")
1117                         {
1118                                 // another server we connected to, which was in WAIT_AUTH_1 state,
1119                                 // has just sent us their credentials. If we get this far, theyre
1120                                 // happy with OUR credentials, and they are now in WAIT_AUTH_2 state.
1121                                 // if we're happy with this, we should send our netburst which
1122                                 // kickstarts the merge.
1123                                 return this->Outbound_Reply_Server(params);
1124                         }
1125                         else if (command == "ERROR")
1126                         {
1127                                 return this->Error(params);
1128                         }
1129                         else if (command == "CAPAB")
1130                         {
1131                                 return this->Capab(params);
1132                         }
1133                 break;
1134                 case CONNECTED:
1135                         // This is the 'authenticated' state, when all passwords
1136                         // have been exchanged and anything past this point is taken
1137                         // as gospel.
1138
1139                         if (prefix != "")
1140                         {
1141                                 std::string direction = prefix;
1142                                 userrec* t = this->Instance->FindNick(prefix);
1143                                 if (t)
1144                                 {
1145                                         direction = t->server;
1146                                 }
1147                                 TreeServer* route_back_again = Utils->BestRouteTo(direction);
1148                                 if ((!route_back_again) || (route_back_again->GetSocket() != this))
1149                                 {
1150                                         if (route_back_again)
1151                                                 Instance->Log(DEBUG,"Protocol violation: Fake direction in command '%s' from connection '%s'",line.c_str(),this->GetName().c_str());
1152                                         return true;
1153                                 }
1154                                 /* Fix by brain:
1155                                  * When there is activity on the socket, reset the ping counter so
1156                                  * that we're not wasting bandwidth pinging an active server.
1157                                  */
1158                                 route_back_again->SetNextPingTime(time(NULL) + 60);
1159                                 route_back_again->SetPingFlag();
1160                         }
1161
1162                         if ((command == "MODE") && (params.size() >= 2))
1163                         {
1164                                 chanrec* channel = Instance->FindChan(params[0]);
1165                                 if (channel)
1166                                 {
1167                                         userrec* x = Instance->FindNick(prefix);
1168                                         if (x)
1169                                         {
1170                                                 if (warned.find(x->server) == warned.end())
1171                                                 {
1172                                                         Instance->Log(DEFAULT,"WARNING: I revceived modes '%s' from another server '%s'. This is not compliant with InspIRCd. Please check that server for bugs.", params[1].c_str(), x->server);
1173                                                         Instance->SNO->WriteToSnoMask('d', "WARNING: The server %s is sending nonstandard modes: '%s MODE %s' where FMODE should be used, and may cause desyncs.", x->server, x->nick, params[1].c_str());
1174                                                         warned[x->server] = x->nick;
1175                                                 }
1176                                         }
1177                                 }
1178                         }
1179
1180                         if (command == "SVSMODE")
1181                         {
1182                                 /* Services expects us to implement
1183                                  * SVSMODE. In inspircd its the same as
1184                                  * MODE anyway.
1185                                  */
1186                                 command = "MODE";
1187                         }
1188                         std::string target = "";
1189                         /* Yes, know, this is a mess. Its reasonably fast though as we're
1190                          * working with std::string here.
1191                          */
1192                         if ((command == "NICK") && (params.size() >= 8))
1193                         {
1194                                 return this->IntroduceClient(prefix,params);
1195                         }
1196                         else if (command == "FJOIN")
1197                         {
1198                                 TreeServer* ServerSource = Utils->FindServer(prefix);
1199                                 if (ServerSource)
1200                                         Utils->SetRemoteBursting(ServerSource, false);
1201                                 return this->ForceJoin(prefix,params);
1202                         }
1203                         else if (command == "STATS")
1204                         {
1205                                 return this->Stats(prefix, params);
1206                         }
1207                         else if (command == "MOTD")
1208                         {
1209                                 return this->Motd(prefix, params);
1210                         }
1211                         else if (command == "MODULES")
1212                         {
1213                                 return this->Modules(prefix, params);
1214                         }
1215                         else if (command == "ADMIN")
1216                         {
1217                                 return this->Admin(prefix, params);
1218                         }
1219                         else if (command == "SERVER")
1220                         {
1221                                 return this->RemoteServer(prefix,params);
1222                         }
1223                         else if (command == "ERROR")
1224                         {
1225                                 return this->Error(params);
1226                         }
1227                         else if (command == "OPERTYPE")
1228                         {
1229                                 return this->OperType(prefix,params);
1230                         }
1231                         else if (command == "FMODE")
1232                         {
1233                                 TreeServer* ServerSource = Utils->FindServer(prefix);
1234                                 if (ServerSource)
1235                                         Utils->SetRemoteBursting(ServerSource, false);
1236                                 return this->ForceMode(prefix,params);
1237                         }
1238                         else if (command == "KILL")
1239                         {
1240                                 return this->RemoteKill(prefix,params);
1241                         }
1242                         else if (command == "FTOPIC")
1243                         {
1244                                 return this->ForceTopic(prefix,params);
1245                         }
1246                         else if (command == "REHASH")
1247                         {
1248                                 return this->RemoteRehash(prefix,params);
1249                         }
1250                         else if (command == "METADATA")
1251                         {
1252                                 return this->MetaData(prefix,params);
1253                         }
1254                         else if (command == "REMSTATUS")
1255                         {
1256                                 return this->RemoveStatus(prefix,params);
1257                         }
1258                         else if (command == "PING")
1259                         {
1260                                 if (prefix == "")
1261                                         prefix = this->GetName();
1262                                 /*
1263                                  * We just got a ping from a server that's bursting.
1264                                  * This can't be right, so set them to not bursting, and
1265                                  * apply their lines.
1266                                  */
1267                                 TreeServer* ServerSource = Utils->FindServer(prefix);
1268                                 if (ServerSource)
1269                                         Utils->SetRemoteBursting(ServerSource, false);
1270
1271                                 if (this->bursting)
1272                                 {
1273                                         this->bursting = false;
1274                                         Instance->XLines->apply_lines(Utils->lines_to_apply);
1275                                         Utils->lines_to_apply = 0;
1276                                 }
1277
1278                                 return this->LocalPing(prefix,params);
1279                         }
1280                         else if (command == "PONG")
1281                         {
1282                                 if (prefix == "")
1283                                         prefix = this->GetName();
1284                                 /*
1285                                  * We just got a pong from a server that's bursting.
1286                                  * This can't be right, so set them to not bursting, and
1287                                  * apply their lines.
1288                                  */
1289                                 TreeServer* ServerSource = Utils->FindServer(prefix);
1290                                 if (ServerSource)
1291                                         Utils->SetRemoteBursting(ServerSource, false);
1292
1293                                 if (this->bursting)
1294                                 {
1295                                         this->bursting = false;
1296                                         Instance->XLines->apply_lines(Utils->lines_to_apply);
1297                                         Utils->lines_to_apply = 0;
1298                                 }
1299
1300                                 return this->LocalPong(prefix,params);
1301                         }
1302                         else if (command == "VERSION")
1303                         {
1304                                 return this->ServerVersion(prefix,params);
1305                         }
1306                         else if (command == "FHOST")
1307                         {
1308                                 return this->ChangeHost(prefix,params);
1309                         }
1310                         else if (command == "FNAME")
1311                         {
1312                                 return this->ChangeName(prefix,params);
1313                         }
1314                         else if (command == "ADDLINE")
1315                         {
1316                                 TreeServer* ServerSource = Utils->FindServer(prefix);
1317                                 if (ServerSource)
1318                                         Utils->SetRemoteBursting(ServerSource, false);
1319                                 return this->AddLine(prefix,params);
1320                         }
1321                         else if (command == "SVSNICK")
1322                         {
1323                                 if (prefix == "")
1324                                 {
1325                                         prefix = this->GetName();
1326                                 }
1327                                 return this->ForceNick(prefix,params);
1328                         }
1329                         else if (command == "OPERQUIT")
1330                         {
1331                                 return this->OperQuit(prefix,params);
1332                         }
1333                         else if (command == "RSQUIT")
1334                         {
1335                                 return this->RemoteSquit(prefix, params);
1336                         }
1337                         else if (command == "IDLE")
1338                         {
1339                                 return this->Whois(prefix,params);
1340                         }
1341                         else if (command == "PUSH")
1342                         {
1343                                 return this->Push(prefix,params);
1344                         }
1345                         else if (command == "TIMESET")
1346                         {
1347                                 return this->HandleSetTime(prefix, params);
1348                         }
1349                         else if (command == "TIME")
1350                         {
1351                                 return this->Time(prefix,params);
1352                         }
1353                         else if ((command == "KICK") && (Utils->IsServer(prefix)))
1354                         {
1355                                 std::string sourceserv = this->myhost;
1356                                 if (params.size() == 3)
1357                                 {
1358                                         userrec* user = this->Instance->FindNick(params[1]);
1359                                         chanrec* chan = this->Instance->FindChan(params[0]);
1360                                         if (user && chan)
1361                                         {
1362                                                 if (!chan->ServerKickUser(user, params[2].c_str(), false))
1363                                                         /* Yikes, the channels gone! */
1364                                                         delete chan;
1365                                         }
1366                                 }
1367                                 if (this->InboundServerName != "")
1368                                 {
1369                                         sourceserv = this->InboundServerName;
1370                                 }
1371                                 return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
1372                         }
1373                         else if (command == "SVSJOIN")
1374                         {
1375                                 if (prefix == "")
1376                                 {
1377                                         prefix = this->GetName();
1378                                 }
1379                                 return this->ServiceJoin(prefix,params);
1380                         }
1381                         else if (command == "SQUIT")
1382                         {
1383                                 if (params.size() == 2)
1384                                 {
1385                                         this->Squit(Utils->FindServer(params[0]),params[1]);
1386                                 }
1387                                 return true;
1388                         }
1389                         else if (command == "OPERNOTICE")
1390                         {
1391                                 std::string sourceserv = this->myhost;
1392                                 if (this->InboundServerName != "")
1393                                         sourceserv = this->InboundServerName;
1394                                 if (params.size() >= 1)
1395                                         Instance->WriteOpers("*** From " + sourceserv + ": " + params[0]);
1396                                 return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
1397                         }
1398                         else if (command == "MODENOTICE")
1399                         {
1400                                 std::string sourceserv = this->myhost;
1401                                 if (this->InboundServerName != "")
1402                                         sourceserv = this->InboundServerName;
1403                                 if (params.size() >= 2)
1404                                 {
1405                                         Instance->WriteMode(params[0].c_str(), WM_AND, "*** From %s: %s", sourceserv.c_str(), params[1].c_str());
1406                                 }
1407                                 return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
1408                         }
1409                         else if (command == "SNONOTICE")
1410                         {
1411                                 std::string sourceserv = this->myhost;
1412                                 if (this->InboundServerName != "")
1413                                         sourceserv = this->InboundServerName;
1414                                 if (params.size() >= 2)
1415                                 {
1416                                         Instance->SNO->WriteToSnoMask(*(params[0].c_str()), "From " + sourceserv + ": "+ params[1]);
1417                                 }
1418                                 return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
1419                         }
1420                         else if (command == "ENDBURST")
1421                         {
1422                                 this->bursting = false;
1423                                 Instance->XLines->apply_lines(Utils->lines_to_apply);
1424                                 Utils->lines_to_apply = 0;
1425                                 std::string sourceserv = this->myhost;
1426                                 if (this->InboundServerName != "")
1427                                 {
1428                                         sourceserv = this->InboundServerName;
1429                                 }
1430                                 this->Instance->SNO->WriteToSnoMask('l',"Received end of netburst from \2%s\2",sourceserv.c_str());
1431
1432                                 Event rmode((char*)sourceserv.c_str(), (Module*)Utils->Creator, "new_server");
1433                                 rmode.Send(Instance);
1434
1435                                 return true;
1436                         }
1437                         else
1438                         {
1439                                 // not a special inter-server command.
1440                                 // Emulate the actual user doing the command,
1441                                 // this saves us having a huge ugly parser.
1442                                 userrec* who = this->Instance->FindNick(prefix);
1443                                 std::string sourceserv = this->myhost;
1444                                 if (this->InboundServerName != "")
1445                                 {
1446                                         sourceserv = this->InboundServerName;
1447                                 }
1448                                 if ((!who) && (command == "MODE"))
1449                                 {
1450                                         if (Utils->IsServer(prefix))
1451                                         {
1452                                                 const char* modelist[127];
1453                                                 for (size_t i = 0; i < params.size(); i++)
1454                                                         modelist[i] = params[i].c_str();
1455                                                 userrec* fake = new userrec(Instance);
1456                                                 fake->SetFd(FD_MAGIC_NUMBER);
1457                                                 this->Instance->SendMode(modelist, params.size(), fake);
1458
1459                                                 delete fake;
1460                                                 /* Hot potato! pass it on! */
1461                                                 return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
1462                                         }
1463                                 }
1464                                 if (who)
1465                                 {
1466                                         if ((command == "NICK") && (params.size() > 0))
1467                                         {
1468                                                 /* On nick messages, check that the nick doesnt
1469                                                  * already exist here. If it does, kill their copy,
1470                                                  * and our copy.
1471                                                  */
1472                                                 userrec* x = this->Instance->FindNick(params[0]);
1473                                                 if ((x) && (x != who))
1474                                                 {
1475                                                         std::deque<std::string> p;
1476                                                         p.push_back(params[0]);
1477                                                         p.push_back("Nickname collision ("+prefix+" -> "+params[0]+")");
1478                                                         Utils->DoOneToMany(this->Instance->Config->ServerName,"KILL",p);
1479                                                         p.clear();
1480                                                         p.push_back(prefix);
1481                                                         p.push_back("Nickname collision");
1482                                                         Utils->DoOneToMany(this->Instance->Config->ServerName,"KILL",p);
1483                                                         userrec::QuitUser(this->Instance,x,"Nickname collision ("+prefix+" -> "+params[0]+")");
1484                                                         userrec* y = this->Instance->FindNick(prefix);
1485                                                         if (y)
1486                                                         {
1487                                                                 userrec::QuitUser(this->Instance,y,"Nickname collision");
1488                                                         }
1489                                                         return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
1490                                                 }
1491                                         }
1492                                         // its a user
1493                                         target = who->server;
1494                                         const char* strparams[127];
1495                                         for (unsigned int q = 0; q < params.size(); q++)
1496                                         {
1497                                                 strparams[q] = params[q].c_str();
1498                                         }
1499                                         if(!this->Instance->CallCommandHandler(command.c_str(), strparams, params.size(), who))
1500                                         {
1501                                                 this->SendError("Unrecognised command '"+std::string(command.c_str())+"' -- possibly loaded mismatched modules");
1502                                                 return false;
1503                                         }
1504                                         else
1505                                                 return true;
1506                                 }
1507                                 else
1508                                 {
1509                                         // its not a user. Its either a server, or somethings screwed up.
1510                                         if (Utils->IsServer(prefix))
1511                                                 target = this->Instance->Config->ServerName;
1512                                         else
1513                                                 return true;
1514                                 }
1515                                 return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
1516
1517                         }
1518                         return true;
1519                 break;
1520         }
1521         return true;
1522 }
1523
1524 std::string TreeSocket::GetName()
1525 {
1526         std::string sourceserv = this->myhost;
1527         if (this->InboundServerName != "")
1528         {
1529                 sourceserv = this->InboundServerName;
1530         }
1531         return sourceserv;
1532 }
1533
1534 void TreeSocket::OnTimeout()
1535 {
1536         if (this->LinkState == CONNECTING)
1537         {
1538                 this->Instance->SNO->WriteToSnoMask('l',"CONNECT: Connection to \002"+myhost+"\002 timed out.");
1539                 Link* MyLink = Utils->FindLink(myhost);
1540                 if (MyLink)
1541                         Utils->DoFailOver(MyLink);
1542         }
1543 }
1544
1545 void TreeSocket::OnClose()
1546 {
1547         // Connection closed.
1548         // If the connection is fully up (state CONNECTED)
1549         // then propogate a netsplit to all peers.
1550         std::string quitserver = this->myhost;
1551         if (this->InboundServerName != "")
1552         {
1553                 quitserver = this->InboundServerName;
1554         }
1555         TreeServer* s = Utils->FindServer(quitserver);
1556         if (s)
1557         {
1558                 Squit(s,"Remote host closed the connection");
1559         }
1560
1561         if (quitserver != "")
1562         {
1563                 this->Instance->SNO->WriteToSnoMask('l',"Connection to '\2%s\2' failed.",quitserver.c_str());
1564                 time_t server_uptime = Instance->Time() - this->age;
1565                 if (server_uptime)
1566                         Instance->SNO->WriteToSnoMask('l',"Connection to '\2%s\2' was established for %s", quitserver.c_str(), Utils->Creator->TimeToStr(server_uptime).c_str());
1567         }
1568 }
1569
1570 int TreeSocket::OnIncomingConnection(int newsock, char* ip)
1571 {
1572         /* To prevent anyone from attempting to flood opers/DDoS by connecting to the server port,
1573          * or discovering if this port is the server port, we don't allow connections from any
1574          * IPs for which we don't have a link block.
1575          */
1576         bool found = false;
1577
1578         found = (std::find(Utils->ValidIPs.begin(), Utils->ValidIPs.end(), ip) != Utils->ValidIPs.end());
1579         if (!found)
1580         {
1581                 for (vector<std::string>::iterator i = Utils->ValidIPs.begin(); i != Utils->ValidIPs.end(); i++)
1582                         if (irc::sockets::MatchCIDR(ip, (*i).c_str()))
1583                                 found = true;
1584
1585                 if (!found)
1586                 {
1587                         this->Instance->SNO->WriteToSnoMask('l',"Server connection from %s denied (no link blocks with that IP address)", ip);
1588                         close(newsock);
1589                         return false;
1590                 }
1591         }
1592
1593         TreeSocket* s = new TreeSocket(this->Utils, this->Instance, newsock, ip, this->Hook);
1594         s = s; /* Whinge whinge whinge, thats all GCC ever does. */
1595         return true;
1596 }