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