src/modules/m_spanningtree.cpp

   1 /*   +------------------------------------+
   2  *   | Inspire Internet Relay Chat Daemon |
   3  *   +------------------------------------+
   4  *
   5  *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
   6  *                     E-mail:
   7  *                <brain@chatspike.net>
   8  *                <Craig@chatspike.net>
   9  *
  10  * Written by Craig Edwards, Craig McLure, and others.
  11  * This program is free but copyrighted software; see
  12  *            the file COPYING for details.
  13  *
  14  * ---------------------------------------------------
  15  */
  16
  17 /* $ModDesc: Povides a spanning tree server link protocol */
  18
  19 #include "configreader.h"
  20 #include "users.h"
  21 #include "channels.h"
  22 #include "modules.h"
  23 #include "commands/cmd_whois.h"
  24 #include "commands/cmd_stats.h"
  25 #include "socket.h"
  26 #include "inspircd.h"
  27 #include "wildcard.h"
  28 #include "xline.h"
  29 #include "cull_list.h"
  30 #include "aes.h"
  31
  32 /** If you make a change which breaks the protocol, increment this.
  33  * If you  completely change the protocol, completely change the number.
  34  */
  35 const long ProtocolVersion = 1101;
  36
  37 /*
  38  * The server list in InspIRCd is maintained as two structures
  39  * which hold the data in different ways. Most of the time, we
  40  * want to very quicky obtain three pieces of information:
  41  *
  42  * (1) The information on a server
  43  * (2) The information on the server we must send data through
  44  *     to actually REACH the server we're after
  45  * (3) Potentially, the child/parent objects of this server
  46  *
  47  * The InspIRCd spanning protocol provides easy access to these
  48  * by storing the data firstly in a recursive structure, where
  49  * each item references its parent item, and a dynamic list
  50  * of child items, and another structure which stores the items
  51  * hashed, linearly. This means that if we want to find a server
  52  * by name quickly, we can look it up in the hash, avoiding
  53  * any O(n) lookups. If however, during a split or sync, we want
  54  * to apply an operation to a server, and any of its child objects
  55  * we can resort to recursion to walk the tree structure.
  56  * Any socket can have one of five states at any one time.
  57  * The LISTENER state indicates a socket which is listening
  58  * for connections. It cannot receive data itself, only incoming
  59  * sockets.
  60  * The CONNECTING state indicates an outbound socket which is
  61  * waiting to be writeable.
  62  * The WAIT_AUTH_1 state indicates the socket is outbound and
  63  * has successfully connected, but has not yet sent and received
  64  * SERVER strings.
  65  * The WAIT_AUTH_2 state indicates that the socket is inbound
  66  * (allocated by a LISTENER) but has not yet sent and received
  67  * SERVER strings.
  68  * The CONNECTED state represents a fully authorized, fully
  69  * connected server.
  70  */
  71 enum ServerState { LISTENER, CONNECTING, WAIT_AUTH_1, WAIT_AUTH_2, CONNECTED };
  72
  73 /* Foward declarations */
  74 class TreeServer;
  75 class TreeSocket;
  76 class Link;
  77 class ModuleSpanningTree;
  78
  79 /* This hash_map holds the hash equivalent of the server
  80  * tree, used for rapid linear lookups.
  81  */
  82 typedef nspace::hash_map<std::string, TreeServer*, nspace::hash<string>, irc::StrHashComp> server_hash;
  83
  84 /** Contains helper functions and variables for this module,
  85  * and keeps them out of the global namespace
  86  */
  87 class SpanningTreeUtilities
  88 {
  89  private:
  90         /** Creator server
  91          */
  92         InspIRCd* ServerInstance;
  93  public:
  94         /** Creator module
  95          */
  96         ModuleSpanningTree* Creator;
  97         /** Flatten links and /MAP for non-opers
  98          */
  99         bool FlatLinks;
 100         /** Hide U-Lined servers in /MAP and /LINKS
 101          */
 102         bool HideULines;
 103         /** Announce TS changes to channels on merge
 104          */
 105         bool AnnounceTSChange;
 106         /** Socket bindings for listening sockets
 107          */
 108         std::vector<TreeSocket*> Bindings;
 109         /** This variable represents the root of the server tree
 110          */
 111         TreeServer *TreeRoot;
 112         /** IPs allowed to link to us
 113          */
 114         std::vector<std::string> ValidIPs;
 115         /** Hash of currently connected servers by name
 116          */
 117         server_hash serverlist;
 118         /** Holds the data from the <link> tags in the conf
 119          */
 120         std::vector<Link> LinkBlocks;
 121
 122         /** Initialise utility class
 123          */
 124         SpanningTreeUtilities(InspIRCd* Instance, ModuleSpanningTree* Creator);
 125         /** Destroy class and free listeners etc
 126          */
 127         ~SpanningTreeUtilities();
 128         /** Send a message from this server to one other local or remote
 129          */
 130         bool DoOneToOne(std::string prefix, std::string command, std::deque<std::string> &params, std::string target);
 131         /** Send a message from this server to all but one other, local or remote
 132          */
 133         bool DoOneToAllButSender(std::string prefix, std::string command, std::deque<std::string> &params, std::string omit);
 134         /** Send a message from this server to all but one other, local or remote
 135          */
 136         bool DoOneToAllButSender(const char* prefix, const char* command, std::deque<std::string> &params, std::string omit);
 137         /** Send a message from this server to all others
 138          */
 139         bool DoOneToMany(std::string prefix, std::string command, std::deque<std::string> &params);
 140         /** Send a message from this server to all others
 141          */
 142         bool DoOneToMany(const char* prefix, const char* command, std::deque<std::string> &params);
 143         /** Send a message from this server to all others, without doing any processing on the command (e.g. send it as-is with colons and all)
 144          */
 145         bool DoOneToAllButSenderRaw(std::string data, std::string omit, std::string prefix, irc::string command, std::deque<std::string> &params);
 146         /** Read the spanningtree module's tags from the config file
 147          */
 148         void ReadConfiguration(bool rebind);
 149         /** Add a server to the server list for GetListOfServersForChannel
 150          */
 151         void AddThisServer(TreeServer* server, std::deque<TreeServer*> &list);
 152         /** Compile a list of servers which contain members of channel c
 153          */
 154         void GetListOfServersForChannel(chanrec* c, std::deque<TreeServer*> &list);
 155         /** Find a server by name
 156          */
 157         TreeServer* FindServer(std::string ServerName);
 158         /** Find a route to a server by name
 159          */
 160         TreeServer* BestRouteTo(std::string ServerName);
 161         /** Find a server by glob mask
 162          */
 163         TreeServer* FindServerMask(std::string ServerName);
 164         /** Returns true if this is a server name we recognise
 165          */
 166         bool IsServer(std::string ServerName);
 167         /** Attempt to connect to the failover link of link x
 168          */
 169         void DoFailOver(Link* x);
 170         /** Find a link tag from a server name
 171          */
 172         Link* FindLink(const std::string& name);
 173 };
 174
 175
 176 /** Each server in the tree is represented by one class of
 177  * type TreeServer. A locally connected TreeServer can
 178  * have a class of type TreeSocket associated with it, for
 179  * remote servers, the TreeSocket entry will be NULL.
 180  * Each server also maintains a pointer to its parent
 181  * (NULL if this server is ours, at the top of the tree)
 182  * and a pointer to its "Route" (see the comments in the
 183  * constructors below), and also a dynamic list of pointers
 184  * to its children which can be iterated recursively
 185  * if required. Creating or deleting objects of type
 186  i* TreeServer automatically maintains the hash_map of
 187  * TreeServer items, deleting and inserting them as they
 188  * are created and destroyed.
 189  */
 190 class TreeServer : public classbase
 191 {
 192         InspIRCd* ServerInstance;               /* Creator */
 193         SpanningTreeUtilities* Utils;           /* Utility class */
 194         TreeServer* Parent;                     /* Parent entry */
 195         TreeServer* Route;                      /* Route entry */
 196         std::vector<TreeServer*> Children;      /* List of child objects */
 197         irc::string ServerName;                 /* Server's name */
 198         std::string ServerDesc;                 /* Server's description */
 199         std::string VersionString;              /* Version string or empty string */
 200         int UserCount;                          /* Not used in this version */
 201         int OperCount;                          /* Not used in this version */
 202         TreeSocket* Socket;                     /* For directly connected servers this points at the socket object */
 203         time_t NextPing;                        /* After this time, the server should be PINGed*/
 204         bool LastPingWasGood;                   /* True if the server responded to the last PING with a PONG */
 205
 206  public:
 207
 208         /** We don't use this constructor. Its a dummy, and won't cause any insertion
 209          * of the TreeServer into the hash_map. See below for the two we DO use.
 210          */
 211         TreeServer(SpanningTreeUtilities* Utils, InspIRCd* Instance) : ServerInstance(Instance)
 212         {
 213                 Parent = NULL;
 214                 ServerName = "";
 215                 ServerDesc = "";
 216                 VersionString = "";
 217                 UserCount = OperCount = 0;
 218                 VersionString = ServerInstance->GetVersionString();
 219         }
 220
 221         /** We use this constructor only to create the 'root' item, Utils->TreeRoot, which
 222          * represents our own server. Therefore, it has no route, no parent, and
 223          * no socket associated with it. Its version string is our own local version.
 224          */
 225         TreeServer(SpanningTreeUtilities* Utils, InspIRCd* Instance, std::string Name, std::string Desc) : ServerInstance(Instance), ServerName(Name.c_str()), ServerDesc(Desc)
 226         {
 227                 Parent = NULL;
 228                 VersionString = "";
 229                 UserCount = OperCount = 0;
 230                 VersionString = ServerInstance->GetVersionString();
 231                 Route = NULL;
 232                 Socket = NULL; /* Fix by brain */
 233                 AddHashEntry();
 234         }
 235
 236         /** When we create a new server, we call this constructor to initialize it.
 237          * This constructor initializes the server's Route and Parent, and sets up
 238          * its ping counters so that it will be pinged one minute from now.
 239          */
 240         TreeServer(SpanningTreeUtilities* Utils, InspIRCd* Instance, std::string Name, std::string Desc, TreeServer* Above, TreeSocket* Sock)
 241                 : ServerInstance(Instance), Parent(Above), ServerName(Name.c_str()), ServerDesc(Desc), Socket(Sock)
 242         {
 243                 VersionString = "";
 244                 UserCount = OperCount = 0;
 245                 this->SetNextPingTime(time(NULL) + 60);
 246                 this->SetPingFlag();
 247
 248                 /* find the 'route' for this server (e.g. the one directly connected
 249                  * to the local server, which we can use to reach it)
 250                  *
 251                  * In the following example, consider we have just added a TreeServer
 252                  * class for server G on our network, of which we are server A.
 253                  * To route traffic to G (marked with a *) we must send the data to
 254                  * B (marked with a +) so this algorithm initializes the 'Route'
 255                  * value to point at whichever server traffic must be routed through
 256                  * to get here. If we were to try this algorithm with server B,
 257                  * the Route pointer would point at its own object ('this').
 258                  *
 259                  *              A
 260                  *             / \
 261                  *          + B   C
 262                  *           / \   \
 263                  *          D   E   F
 264                  *         /         \
 265                  *      * G           H
 266                  *
 267                  * We only run this algorithm when a server is created, as
 268                  * the routes remain constant while ever the server exists, and
 269                  * do not need to be re-calculated.
 270                  */
 271
 272                 Route = Above;
 273                 if (Route == Utils->TreeRoot)
 274                 {
 275                         Route = this;
 276                 }
 277                 else
 278                 {
 279                         while (this->Route->GetParent() != Utils->TreeRoot)
 280                         {
 281                                 this->Route = Route->GetParent();
 282                         }
 283                 }
 284
 285                 /* Because recursive code is slow and takes a lot of resources,
 286                  * we store two representations of the server tree. The first
 287                  * is a recursive structure where each server references its
 288                  * children and its parent, which is used for netbursts and
 289                  * netsplits to dump the whole dataset to the other server,
 290                  * and the second is used for very fast lookups when routing
 291                  * messages and is instead a hash_map, where each item can
 292                  * be referenced by its server name. The AddHashEntry()
 293                  * call below automatically inserts each TreeServer class
 294                  * into the hash_map as it is created. There is a similar
 295                  * maintainance call in the destructor to tidy up deleted
 296                  * servers.
 297                  */
 298
 299                 this->AddHashEntry();
 300         }
 301
 302         int QuitUsers(const std::string &reason)
 303         {
 304                 ServerInstance->Log(DEBUG,"Removing all users from server %s",this->ServerName.c_str());
 305                 const char* reason_s = reason.c_str();
 306                 std::vector<userrec*> time_to_die;
 307                 for (user_hash::iterator n = ServerInstance->clientlist.begin(); n != ServerInstance->clientlist.end(); n++)
 308                 {
 309                         if (!strcmp(n->second->server, this->ServerName.c_str()))
 310                         {
 311                                 time_to_die.push_back(n->second);
 312                         }
 313                 }
 314                 for (std::vector<userrec*>::iterator n = time_to_die.begin(); n != time_to_die.end(); n++)
 315                 {
 316                         userrec* a = (userrec*)*n;
 317                         ServerInstance->Log(DEBUG,"Kill %s fd=%d",a->nick,a->GetFd());
 318                         if (!IS_LOCAL(a))
 319                                 userrec::QuitUser(ServerInstance,a,reason_s);
 320                 }
 321                 return time_to_die.size();
 322         }
 323
 324         /** This method is used to add the structure to the
 325          * hash_map for linear searches. It is only called
 326          * by the constructors.
 327          */
 328         void AddHashEntry()
 329         {
 330                 server_hash::iterator iter = Utils->serverlist.find(this->ServerName.c_str());
 331                 if (iter == Utils->serverlist.end())
 332                         Utils->serverlist[this->ServerName.c_str()] = this;
 333         }
 334
 335         /** This method removes the reference to this object
 336          * from the hash_map which is used for linear searches.
 337          * It is only called by the default destructor.
 338          */
 339         void DelHashEntry()
 340         {
 341                 server_hash::iterator iter = Utils->serverlist.find(this->ServerName.c_str());
 342                 if (iter != Utils->serverlist.end())
 343                         Utils->serverlist.erase(iter);
 344         }
 345
 346         /** These accessors etc should be pretty self-
 347          * explanitory.
 348          */
 349         TreeServer* GetRoute()
 350         {
 351                 return Route;
 352         }
 353
 354         std::string GetName()
 355         {
 356                 return ServerName.c_str();
 357         }
 358
 359         std::string GetDesc()
 360         {
 361                 return ServerDesc;
 362         }
 363
 364         std::string GetVersion()
 365         {
 366                 return VersionString;
 367         }
 368
 369         void SetNextPingTime(time_t t)
 370         {
 371                 this->NextPing = t;
 372                 LastPingWasGood = false;
 373         }
 374
 375         time_t NextPingTime()
 376         {
 377                 return NextPing;
 378         }
 379
 380         bool AnsweredLastPing()
 381         {
 382                 return LastPingWasGood;
 383         }
 384
 385         void SetPingFlag()
 386         {
 387                 LastPingWasGood = true;
 388         }
 389
 390         int GetUserCount()
 391         {
 392                 return UserCount;
 393         }
 394
 395         void AddUserCount()
 396         {
 397                 UserCount++;
 398         }
 399
 400         void DelUserCount()
 401         {
 402                 UserCount--;
 403         }
 404
 405         int GetOperCount()
 406         {
 407                 return OperCount;
 408         }
 409
 410         TreeSocket* GetSocket()
 411         {
 412                 return Socket;
 413         }
 414
 415         TreeServer* GetParent()
 416         {
 417                 return Parent;
 418         }
 419
 420         void SetVersion(std::string Version)
 421         {
 422                 VersionString = Version;
 423         }
 424
 425         unsigned int ChildCount()
 426         {
 427                 return Children.size();
 428         }
 429
 430         TreeServer* GetChild(unsigned int n)
 431         {
 432                 if (n < Children.size())
 433                 {
 434                         /* Make sure they  cant request
 435                          * an out-of-range object. After
 436                          * all we know what these programmer
 437                          * types are like *grin*.
 438                          */
 439                         return Children[n];
 440                 }
 441                 else
 442                 {
 443                         return NULL;
 444                 }
 445         }
 446
 447         void AddChild(TreeServer* Child)
 448         {
 449                 Children.push_back(Child);
 450         }
 451
 452         bool DelChild(TreeServer* Child)
 453         {
 454                 for (std::vector<TreeServer*>::iterator a = Children.begin(); a < Children.end(); a++)
 455                 {
 456                         if (*a == Child)
 457                         {
 458                                 Children.erase(a);
 459                                 return true;
 460                         }
 461                 }
 462                 return false;
 463         }
 464
 465         /** Removes child nodes of this node, and of that node, etc etc.
 466          * This is used during netsplits to automatically tidy up the
 467          * server tree. It is slow, we don't use it for much else.
 468          */
 469         bool Tidy()
 470         {
 471                 bool stillchildren = true;
 472                 while (stillchildren)
 473                 {
 474                         stillchildren = false;
 475                         for (std::vector<TreeServer*>::iterator a = Children.begin(); a < Children.end(); a++)
 476                         {
 477                                 TreeServer* s = (TreeServer*)*a;
 478                                 s->Tidy();
 479                                 Children.erase(a);
 480                                 DELETE(s);
 481                                 stillchildren = true;
 482                                 break;
 483                         }
 484                 }
 485                 return true;
 486         }
 487
 488         ~TreeServer()
 489         {
 490                 /* We'd better tidy up after ourselves, eh? */
 491                 this->DelHashEntry();
 492         }
 493 };
 494
 495 /** The Link class might as well be a struct,
 496  * but this is C++ and we don't believe in structs (!).
 497  * It holds the entire information of one <link>
 498  * tag from the main config file. We maintain a list
 499  * of them, and populate the list on rehash/load.
 500  */
 501 class Link : public classbase
 502 {
 503  public:
 504         irc::string Name;
 505         std::string IPAddr;
 506         int Port;
 507         std::string SendPass;
 508         std::string RecvPass;
 509         unsigned long AutoConnect;
 510         time_t NextConnectTime;
 511         std::string EncryptionKey;
 512         bool HiddenFromStats;
 513         irc::string FailOver;
 514 };
 515
 516 /** Yay for fast searches!
 517  * This is hundreds of times faster than recursion
 518  * or even scanning a linked list, especially when
 519  * there are more than a few servers to deal with.
 520  * (read as: lots).
 521  */
 522 TreeServer* SpanningTreeUtilities::FindServer(std::string ServerName)
 523 {
 524         server_hash::iterator iter;
 525         iter = serverlist.find(ServerName.c_str());
 526         if (iter != serverlist.end())
 527         {
 528                 return iter->second;
 529         }
 530         else
 531         {
 532                 return NULL;
 533         }
 534 }
 535
 536 /** Returns the locally connected server we must route a
 537  * message through to reach server 'ServerName'. This
 538  * only applies to one-to-one and not one-to-many routing.
 539  * See the comments for the constructor of TreeServer
 540  * for more details.
 541  */
 542 TreeServer* SpanningTreeUtilities::BestRouteTo(std::string ServerName)
 543 {
 544         if (ServerName.c_str() == TreeRoot->GetName())
 545                 return NULL;
 546         TreeServer* Found = FindServer(ServerName);
 547         if (Found)
 548         {
 549                 return Found->GetRoute();
 550         }
 551         else
 552         {
 553                 return NULL;
 554         }
 555 }
 556
 557 /** Find the first server matching a given glob mask.
 558  * Theres no find-using-glob method of hash_map [awwww :-(]
 559  * so instead, we iterate over the list using an iterator
 560  * and match each one until we get a hit. Yes its slow,
 561  * deal with it.
 562  */
 563 TreeServer* SpanningTreeUtilities::FindServerMask(std::string ServerName)
 564 {
 565         for (server_hash::iterator i = serverlist.begin(); i != serverlist.end(); i++)
 566         {
 567                 if (match(i->first.c_str(),ServerName.c_str()))
 568                         return i->second;
 569         }
 570         return NULL;
 571 }
 572
 573 /* A convenient wrapper that returns true if a server exists */
 574 bool SpanningTreeUtilities::IsServer(std::string ServerName)
 575 {
 576         return (FindServer(ServerName) != NULL);
 577 }
 578
 579
 580 /** Handle /RCONNECT
 581  */
 582 class cmd_rconnect : public command_t
 583 {
 584         Module* Creator;
 585         SpanningTreeUtilities* Utils;
 586  public:
 587         cmd_rconnect (InspIRCd* Instance, Module* Callback, SpanningTreeUtilities* Util) : command_t(Instance, "RCONNECT", 'o', 2), Creator(Callback), Utils(Util)
 588         {
 589                 this->source = "m_spanningtree.so";
 590                 syntax = "<remote-server-mask> <servermask>";
 591         }
 592
 593         CmdResult Handle (const char** parameters, int pcnt, userrec *user)
 594         {
 595                 user->WriteServ("NOTICE %s :*** RCONNECT: Sending remote connect to \002%s\002 to connect server \002%s\002.",user->nick,parameters[0],parameters[1]);
 596                 /* Is this aimed at our server? */
 597                 if (ServerInstance->MatchText(ServerInstance->Config->ServerName,parameters[0]))
 598                 {
 599                         /* Yes, initiate the given connect */
 600                         ServerInstance->SNO->WriteToSnoMask('l',"Remote CONNECT from %s matching \002%s\002, connecting server \002%s\002",user->nick,parameters[0],parameters[1]);
 601                         const char* para[1];
 602                         para[0] = parameters[1];
 603                         std::string original_command = std::string("CONNECT ") + parameters[1];
 604                         Creator->OnPreCommand("CONNECT", para, 1, user, true, original_command);
 605
 606                         return CMD_SUCCESS;
 607                 }
 608
 609                 return CMD_FAILURE;
 610         }
 611 };
 612
 613
 614
 615 /** Every SERVER connection inbound or outbound is represented by
 616  * an object of type TreeSocket.
 617  * TreeSockets, being inherited from InspSocket, can be tied into
 618  * the core socket engine, and we cn therefore receive activity events
 619  * for them, just like activex objects on speed. (yes really, that
 620  * is a technical term!) Each of these which relates to a locally
 621  * connected server is assocated with it, by hooking it onto a
 622  * TreeSocket class using its constructor. In this way, we can
 623  * maintain a list of servers, some of which are directly connected,
 624  * some of which are not.
 625  */
 626 class TreeSocket : public InspSocket
 627 {
 628         SpanningTreeUtilities* Utils;
 629         std::string myhost;
 630         std::string in_buffer;
 631         ServerState LinkState;
 632         std::string InboundServerName;
 633         std::string InboundDescription;
 634         int num_lost_users;
 635         int num_lost_servers;
 636         time_t NextPing;
 637         bool LastPingWasGood;
 638         bool bursting;
 639         AES* ctx_in;
 640         AES* ctx_out;
 641         unsigned int keylength;
 642         std::string ModuleList;
 643         std::map<std::string,std::string> CapKeys;
 644
 645  public:
 646
 647         /** Because most of the I/O gubbins are encapsulated within
 648          * InspSocket, we just call the superclass constructor for
 649          * most of the action, and append a few of our own values
 650          * to it.
 651          */
 652         TreeSocket(SpanningTreeUtilities* Util, InspIRCd* SI, std::string host, int port, bool listening, unsigned long maxtime)
 653                 : InspSocket(SI, host, port, listening, maxtime), Utils(Util)
 654         {
 655                 myhost = host;
 656                 this->LinkState = LISTENER;
 657                 this->ctx_in = NULL;
 658                 this->ctx_out = NULL;
 659         }
 660
 661         TreeSocket(SpanningTreeUtilities* Util, InspIRCd* SI, std::string host, int port, bool listening, unsigned long maxtime, std::string ServerName)
 662                 : InspSocket(SI, host, port, listening, maxtime), Utils(Util)
 663         {
 664                 myhost = ServerName;
 665                 this->LinkState = CONNECTING;
 666                 this->ctx_in = NULL;
 667                 this->ctx_out = NULL;
 668         }
 669
 670         /** When a listening socket gives us a new file descriptor,
 671          * we must associate it with a socket without creating a new
 672          * connection. This constructor is used for this purpose.
 673          */
 674         TreeSocket(SpanningTreeUtilities* Util, InspIRCd* SI, int newfd, char* ip)
 675                 : InspSocket(SI, newfd, ip), Utils(Util)
 676         {
 677                 this->LinkState = WAIT_AUTH_1;
 678                 this->ctx_in = NULL;
 679                 this->ctx_out = NULL;
 680                 this->SendCapabilities();
 681         }
 682
 683         ~TreeSocket()
 684         {
 685                 if (ctx_in)
 686                         DELETE(ctx_in);
 687                 if (ctx_out)
 688                         DELETE(ctx_out);
 689         }
 690
 691         void InitAES(std::string key,std::string SName)
 692         {
 693                 if (key == "")
 694                         return;
 695
 696                 ctx_in = new AES();
 697                 ctx_out = new AES();
 698                 Instance->Log(DEBUG,"Initialized AES key %s",key.c_str());
 699                 // key must be 16, 24, 32 etc bytes (multiple of 8)
 700                 keylength = key.length();
 701                 if (!(keylength == 16 || keylength == 24 || keylength == 32))
 702                 {
 703                         this->Instance->SNO->WriteToSnoMask('l',"\2ERROR\2: Key length for encryptionkey is not 16, 24 or 32 bytes in length!");
 704                         Instance->Log(DEBUG,"Key length not 16, 24 or 32 characters!");
 705                 }
 706                 else
 707                 {
 708                         if (this->GetState() != I_ERROR)
 709                         {
 710                                 this->Instance->SNO->WriteToSnoMask('l',"\2AES\2: Initialized %d bit encryption to server %s",keylength*8,SName.c_str());
 711                                 ctx_in->MakeKey(key.c_str(), "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\
 712                                         \0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0", keylength, keylength);
 713                                 ctx_out->MakeKey(key.c_str(), "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\
 714                                         \0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0", keylength, keylength);
 715                         }
 716                 }
 717         }
 718
 719         /** When an outbound connection finishes connecting, we receive
 720          * this event, and must send our SERVER string to the other
 721          * side. If the other side is happy, as outlined in the server
 722          * to server docs on the inspircd.org site, the other side
 723          * will then send back its own server string.
 724          */
 725         virtual bool OnConnected()
 726         {
 727                 if (this->LinkState == CONNECTING)
 728                 {
 729                         /* we do not need to change state here. */
 730                         for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
 731                         {
 732                                 if (x->Name == this->myhost)
 733                                 {
 734                                         this->Instance->SNO->WriteToSnoMask('l',"Connection to \2"+myhost+"\2["+(x->HiddenFromStats ? "<hidden>" : this->GetIP())+"] started.");
 735                                         this->SendCapabilities();
 736                                         if (x->EncryptionKey != "")
 737                                         {
 738                                                 if (!(x->EncryptionKey.length() == 16 || x->EncryptionKey.length() == 24 || x->EncryptionKey.length() == 32))
 739                                                 {
 740                                                         this->Instance->SNO->WriteToSnoMask('l',"\2WARNING\2: Your encryption key is NOT 16, 24 or 32 characters in length, encryption will \2NOT\2 be enabled.");
 741                                                 }
 742                                                 else
 743                                                 {
 744                                                         this->WriteLine(std::string("AES ")+this->Instance->Config->ServerName);
 745                                                         this->InitAES(x->EncryptionKey,x->Name.c_str());
 746                                                 }
 747                                         }
 748                                         /* found who we're supposed to be connecting to, send the neccessary gubbins. */
 749                                         this->WriteLine(std::string("SERVER ")+this->Instance->Config->ServerName+" "+x->SendPass+" 0 :"+this->Instance->Config->ServerDesc);
 750                                         return true;
 751                                 }
 752                         }
 753                 }
 754                 /* There is a (remote) chance that between the /CONNECT and the connection
 755                  * being accepted, some muppet has removed the <link> block and rehashed.
 756                  * If that happens the connection hangs here until it's closed. Unlikely
 757                  * and rather harmless.
 758                  */
 759                 this->Instance->SNO->WriteToSnoMask('l',"Connection to \2"+myhost+"\2 lost link tag(!)");
 760                 return true;
 761         }
 762
 763         virtual void OnError(InspSocketError e)
 764         {
 765                 /* We don't handle this method, because all our
 766                  * dirty work is done in OnClose() (see below)
 767                  * which is still called on error conditions too.
 768                  */
 769                 if (e == I_ERR_CONNECT)
 770                 {
 771                         this->Instance->SNO->WriteToSnoMask('l',"Connection failed: Connection to \002"+myhost+"\002 refused");
 772                         Link* MyLink = Utils->FindLink(myhost);
 773                         if (MyLink)
 774                                 Utils->DoFailOver(MyLink);
 775                 }
 776         }
 777
 778         virtual int OnDisconnect()
 779         {
 780                 /* For the same reason as above, we don't
 781                  * handle OnDisconnect()
 782                  */
 783                 return true;
 784         }
 785
 786         /** Recursively send the server tree with distances as hops.
 787          * This is used during network burst to inform the other server
 788          * (and any of ITS servers too) of what servers we know about.
 789          * If at any point any of these servers already exist on the other
 790          * end, our connection may be terminated. The hopcounts given
 791          * by this function are relative, this doesn't matter so long as
 792          * they are all >1, as all the remote servers re-calculate them
 793          * to be relative too, with themselves as hop 0.
 794          */
 795         void SendServers(TreeServer* Current, TreeServer* s, int hops)
 796         {
 797                 char command[1024];
 798                 for (unsigned int q = 0; q < Current->ChildCount(); q++)
 799                 {
 800                         TreeServer* recursive_server = Current->GetChild(q);
 801                         if (recursive_server != s)
 802                         {
 803                                 snprintf(command,1024,":%s SERVER %s * %d :%s",Current->GetName().c_str(),recursive_server->GetName().c_str(),hops,recursive_server->GetDesc().c_str());
 804                                 this->WriteLine(command);
 805                                 this->WriteLine(":"+recursive_server->GetName()+" VERSION :"+recursive_server->GetVersion());
 806                                 /* down to next level */
 807                                 this->SendServers(recursive_server, s, hops+1);
 808                         }
 809                 }
 810         }
 811
 812         std::string MyCapabilities()
 813         {
 814                 std::vector<std::string> modlist;
 815                 std::string capabilities = "";
 816
 817                 for (int i = 0; i <= this->Instance->GetModuleCount(); i++)
 818                 {
 819                         if (this->Instance->modules[i]->GetVersion().Flags & VF_COMMON)
 820                                 modlist.push_back(this->Instance->Config->module_names[i]);
 821                 }
 822                 sort(modlist.begin(),modlist.end());
 823                 for (unsigned int i = 0; i < modlist.size(); i++)
 824                 {
 825                         if (i)
 826                                 capabilities = capabilities + ",";
 827                         capabilities = capabilities + modlist[i];
 828                 }
 829                 return capabilities;
 830         }
 831
 832         void SendCapabilities()
 833         {
 834                 irc::commasepstream modulelist(MyCapabilities());
 835
 836                 this->WriteLine("CAPAB START");
 837
 838                 /* Send module names, split at 509 length */
 839                 std::string item = "*";
 840                 std::string line = "CAPAB MODULES ";
 841                 while ((item = modulelist.GetToken()) != "")
 842                 {
 843                         if (line.length() + item.length() + 1 > 509)
 844                         {
 845                                 this->WriteLine(line);
 846                                 line = "CAPAB MODULES ";
 847                         }
 848
 849                         if (line != "CAPAB MODULES ")
 850                                 line.append(",");
 851
 852                         line.append(item);
 853                 }
 854                 if (line != "CAPAB MODULES ")
 855                         this->WriteLine(line);
 856
 857                 int ip6 = 0;
 858                 int ip6support = 0;
 859 #ifdef IPV6
 860                 ip6 = 1;
 861 #endif
 862 #ifdef SUPPORT_IP6LINKS
 863                 ip6support = 1;
 864 #endif
 865                 this->WriteLine("CAPAB CAPABILITIES :NICKMAX="+ConvToStr(NICKMAX)+" HALFOP="+ConvToStr(this->Instance->Config->AllowHalfop)+" CHANMAX="+ConvToStr(CHANMAX)+" MAXMODES="+ConvToStr(MAXMODES)+" IDENTMAX="+ConvToStr(IDENTMAX)+" MAXQUIT="+ConvToStr(MAXQUIT)+" MAXTOPIC="+ConvToStr(MAXTOPIC)+" MAXKICK="+ConvToStr(MAXKICK)+" MAXGECOS="+ConvToStr(MAXGECOS)+" MAXAWAY="+ConvToStr(MAXAWAY)+" IP6NATIVE="+ConvToStr(ip6)+" IP6SUPPORT="+ConvToStr(ip6support)+" PROTOCOL="+ConvToStr(ProtocolVersion));
 866
 867                 this->WriteLine("CAPAB END");
 868         }
 869
 870         /* Check a comma seperated list for an item */
 871         bool HasItem(const std::string &list, const std::string &item)
 872         {
 873                 irc::commasepstream seplist(list);
 874
 875                 std::string item2 = "*";
 876                 while ((item2 = seplist.GetToken()) != "")
 877                 {
 878                         if (item2 == item)
 879                                 return true;
 880                 }
 881
 882                 return false;
 883         }
 884
 885         /* Isolate and return the elements that are different between two comma seperated lists */
 886         std::string ListDifference(const std::string &one, const std::string &two)
 887         {
 888                 irc::commasepstream list_one(one);
 889                 std::string item = "*";
 890                 std::string result = "";
 891                 while ((item = list_one.GetToken()) != "")
 892                 {
 893                         if (!HasItem(two, item))
 894                         {
 895                                 result.append(" ");
 896                                 result.append(item);
 897                         }
 898                 }
 899                 return result;
 900         }
 901
 902         bool Capab(std::deque<std::string> params)
 903         {
 904                 if (params.size() < 1)
 905                 {
 906                         this->WriteLine("ERROR :Invalid number of parameters for CAPAB - Mismatched version");
 907                         return false;
 908                 }
 909
 910                 if (params[0] == "START")
 911                 {
 912                         this->ModuleList = "";
 913                         this->CapKeys.clear();
 914                 }
 915                 else if (params[0] == "END")
 916                 {
 917                         std::string reason = "";
 918                         int ip6support = 0;
 919 #ifdef SUPPORT_IP6LINKS
 920                         ip6support = 1;
 921 #endif
 922                         /* Compare ModuleList and check CapKeys...
 923                          * Maybe this could be tidier? -- Brain
 924                          */
 925                         if ((this->ModuleList != this->MyCapabilities()) && (this->ModuleList.length()))
 926                         {
 927                                 std::string diff = ListDifference(this->ModuleList, this->MyCapabilities());
 928                                 if (!diff.length())
 929                                 {
 930                                         diff = "your server:" + ListDifference(this->MyCapabilities(), this->ModuleList);
 931                                 }
 932                                 else
 933                                 {
 934                                         diff = "this server:" + diff;
 935                                 }
 936                                 if (diff.length() == 12)
 937                                         reason = "Module list in CAPAB is not alphabetically ordered, cannot compare lists.";
 938                                 else
 939                                         reason = "Modules loaded on these servers are not correctly matched, these modules are not loaded on " + diff;
 940                         }
 941
 942                         if (((this->CapKeys.find("IP6SUPPORT") == this->CapKeys.end()) && (ip6support)) || ((this->CapKeys.find("IP6SUPPORT") != this->CapKeys.end()) && (this->CapKeys.find("IP6SUPPORT")->second != ConvToStr(ip6support))))
 943                                 reason = "We don't both support linking to IPV6 servers";
 944
 945                         if (((this->CapKeys.find("IP6NATIVE") != this->CapKeys.end()) && (this->CapKeys.find("IP6NATIVE")->second == "1")) && (!ip6support))
 946                                 reason = "The remote server is IPV6 native, and we don't support linking to IPV6 servers";
 947
 948                         if (((this->CapKeys.find("NICKMAX") == this->CapKeys.end()) || ((this->CapKeys.find("NICKMAX") != this->CapKeys.end()) && (this->CapKeys.find("NICKMAX")->second != ConvToStr(NICKMAX)))))
 949                                 reason = "Maximum nickname lengths differ or remote nickname length not specified";
 950
 951                         if (((this->CapKeys.find("PROTOCOL") == this->CapKeys.end()) || ((this->CapKeys.find("PROTOCOL") != this->CapKeys.end()) && (this->CapKeys.find("PROTOCOL")->second != ConvToStr(ProtocolVersion)))))
 952                         {
 953                                 if (this->CapKeys.find("PROTOCOL") != this->CapKeys.end())
 954                                 {
 955                                         reason = "Mismatched protocol versions "+this->CapKeys.find("PROTOCOL")->second+" and "+ConvToStr(ProtocolVersion);
 956                                 }
 957                                 else
 958                                 {
 959                                         reason = "Protocol version not specified";
 960                                 }
 961                         }
 962
 963                         if (((this->CapKeys.find("HALFOP") == this->CapKeys.end()) && (Instance->Config->AllowHalfop)) || ((this->CapKeys.find("HALFOP") != this->CapKeys.end()) && (this->CapKeys.find("HALFOP")->second != ConvToStr(Instance->Config->AllowHalfop))))
 964                                 reason = "We don't both have halfop support enabled/disabled identically";
 965
 966                         if (((this->CapKeys.find("IDENTMAX") == this->CapKeys.end()) || ((this->CapKeys.find("IDENTMAX") != this->CapKeys.end()) && (this->CapKeys.find("IDENTMAX")->second != ConvToStr(IDENTMAX)))))
 967                                 reason = "Maximum ident lengths differ or remote ident length not specified";
 968
 969                         if (((this->CapKeys.find("CHANMAX") == this->CapKeys.end()) || ((this->CapKeys.find("CHANMAX") != this->CapKeys.end()) && (this->CapKeys.find("CHANMAX")->second != ConvToStr(CHANMAX)))))
 970                                 reason = "Maximum channel lengths differ or remote channel length not specified";
 971
 972                         if (((this->CapKeys.find("MAXMODES") == this->CapKeys.end()) || ((this->CapKeys.find("MAXMODES") != this->CapKeys.end()) && (this->CapKeys.find("MAXMODES")->second != ConvToStr(MAXMODES)))))
 973                                 reason = "Maximum modes per line differ or remote modes per line not specified";
 974
 975                         if (((this->CapKeys.find("MAXQUIT") == this->CapKeys.end()) || ((this->CapKeys.find("MAXQUIT") != this->CapKeys.end()) && (this->CapKeys.find("MAXQUIT")->second != ConvToStr(MAXQUIT)))))
 976                                 reason = "Maximum quit lengths differ or remote quit length not specified";
 977
 978                         if (((this->CapKeys.find("MAXTOPIC") == this->CapKeys.end()) || ((this->CapKeys.find("MAXTOPIC") != this->CapKeys.end()) && (this->CapKeys.find("MAXTOPIC")->second != ConvToStr(MAXTOPIC)))))
 979                                 reason = "Maximum topic lengths differ or remote topic length not specified";
 980
 981                         if (((this->CapKeys.find("MAXKICK") == this->CapKeys.end()) || ((this->CapKeys.find("MAXKICK") != this->CapKeys.end()) && (this->CapKeys.find("MAXKICK")->second != ConvToStr(MAXKICK)))))
 982                                 reason = "Maximum kick lengths differ or remote kick length not specified";
 983
 984                         if (((this->CapKeys.find("MAXGECOS") == this->CapKeys.end()) || ((this->CapKeys.find("MAXGECOS") != this->CapKeys.end()) && (this->CapKeys.find("MAXGECOS")->second != ConvToStr(MAXGECOS)))))
 985                                 reason = "Maximum GECOS (fullname) lengths differ or remote GECOS length not specified";
 986
 987                         if (((this->CapKeys.find("MAXAWAY") == this->CapKeys.end()) || ((this->CapKeys.find("MAXAWAY") != this->CapKeys.end()) && (this->CapKeys.find("MAXAWAY")->second != ConvToStr(MAXAWAY)))))
 988                                 reason = "Maximum awaymessage lengths differ or remote awaymessage length not specified";
 989
 990                         if (reason.length())
 991                         {
 992                                 this->WriteLine("ERROR :CAPAB negotiation failed: "+reason);
 993                                 return false;
 994                         }
 995                 }
 996                 else if ((params[0] == "MODULES") && (params.size() == 2))
 997                 {
 998                         if (!this->ModuleList.length())
 999                         {
1000                                 this->ModuleList.append(params[1]);
1001                         }
1002                         else
1003                         {
1004                                 this->ModuleList.append(",");
1005                                 this->ModuleList.append(params[1]);
1006                         }
1007                 }
1008                 else if ((params[0] == "CAPABILITIES") && (params.size() == 2))
1009                 {
1010                         irc::tokenstream capabs(params[1]);
1011                         std::string item = "*";
1012                         while ((item = capabs.GetToken()) != "")
1013                         {
1014                                 /* Process each key/value pair */
1015                                 std::string::size_type equals = item.rfind('=');
1016                                 if (equals != std::string::npos)
1017                                 {
1018                                         std::string var = item.substr(0, equals);
1019                                         std::string value = item.substr(equals+1, item.length());
1020                                         this->Instance->Log(DEBUG,"Key='%s' Value='%s'",var.c_str(),value.c_str());
1021                                         CapKeys[var] = value;
1022                                 }
1023                         }
1024                 }
1025
1026                 return true;
1027         }
1028
1029         /** This function forces this server to quit, removing this server
1030          * and any users on it (and servers and users below that, etc etc).
1031          * It's very slow and pretty clunky, but luckily unless your network
1032          * is having a REAL bad hair day, this function shouldnt be called
1033          * too many times a month ;-)
1034          */
1035         void SquitServer(std::string &from, TreeServer* Current)
1036         {
1037                 /* recursively squit the servers attached to 'Current'.
1038                  * We're going backwards so we don't remove users
1039                  * while we still need them ;)
1040                  */
1041                 for (unsigned int q = 0; q < Current->ChildCount(); q++)
1042                 {
1043                         TreeServer* recursive_server = Current->GetChild(q);
1044                         this->SquitServer(from,recursive_server);
1045                 }
1046                 /* Now we've whacked the kids, whack self */
1047                 num_lost_servers++;
1048                 num_lost_users += Current->QuitUsers(from);
1049         }
1050
1051         /** This is a wrapper function for SquitServer above, which
1052          * does some validation first and passes on the SQUIT to all
1053          * other remaining servers.
1054          */
1055         void Squit(TreeServer* Current,std::string reason)
1056         {
1057                 if ((Current) && (Current != Utils->TreeRoot))
1058                 {
1059                         std::deque<std::string> params;
1060                         params.push_back(Current->GetName());
1061                         params.push_back(":"+reason);
1062                         Utils->DoOneToAllButSender(Current->GetParent()->GetName(),"SQUIT",params,Current->GetName());
1063                         if (Current->GetParent() == Utils->TreeRoot)
1064                         {
1065                                 this->Instance->WriteOpers("Server \002"+Current->GetName()+"\002 split: "+reason);
1066                         }
1067                         else
1068                         {
1069                                 this->Instance->WriteOpers("Server \002"+Current->GetName()+"\002 split from server \002"+Current->GetParent()->GetName()+"\002 with reason: "+reason);
1070                         }
1071                         num_lost_servers = 0;
1072                         num_lost_users = 0;
1073                         std::string from = Current->GetParent()->GetName()+" "+Current->GetName();
1074                         SquitServer(from, Current);
1075                         Current->Tidy();
1076                         Current->GetParent()->DelChild(Current);
1077                         DELETE(Current);
1078                         this->Instance->WriteOpers("Netsplit complete, lost \002%d\002 users on \002%d\002 servers.", num_lost_users, num_lost_servers);
1079                 }
1080                 else
1081                 {
1082                         Instance->Log(DEFAULT,"Squit from unknown server");
1083                 }
1084         }
1085
1086         /** FMODE command - server mode with timestamp checks */
1087         bool ForceMode(std::string source, std::deque<std::string> &params)
1088         {
1089                 /* Chances are this is a 1.0 FMODE without TS */
1090                 if (params.size() < 3)
1091                 {
1092                         this->WriteLine("ERROR :Version 1.0 FMODE sent to version 1.1 server");
1093                         return false;
1094                 }
1095
1096                 bool smode = false;
1097                 std::string sourceserv;
1098
1099                 /* Are we dealing with an FMODE from a user, or from a server? */
1100                 userrec* who = this->Instance->FindNick(source);
1101                 if (who)
1102                 {
1103                         /* FMODE from a user, set sourceserv to the users server name */
1104                         sourceserv = who->server;
1105                 }
1106                 else
1107                 {
1108                         /* FMODE from a server, create a fake user to receive mode feedback */
1109                         who = new userrec(this->Instance);
1110                         who->SetFd(FD_MAGIC_NUMBER);
1111                         smode = true;           /* Setting this flag tells us we should free the userrec later */
1112                         sourceserv = source;    /* Set sourceserv to the actual source string */
1113                 }
1114                 const char* modelist[64];
1115                 time_t TS = 0;
1116                 int n = 0;
1117                 memset(&modelist,0,sizeof(modelist));
1118                 for (unsigned int q = 0; (q < params.size()) && (q < 64); q++)
1119                 {
1120                         if (q == 1)
1121                         {
1122                                 /* The timestamp is in this position.
1123                                  * We don't want to pass that up to the
1124                                  * server->client protocol!
1125                                  */
1126                                 TS = atoi(params[q].c_str());
1127                         }
1128                         else
1129                         {
1130                                 /* Everything else is fine to append to the modelist */
1131                                 modelist[n++] = params[q].c_str();
1132                         }
1133
1134                 }
1135                 /* Extract the TS value of the object, either userrec or chanrec */
1136                 userrec* dst = this->Instance->FindNick(params[0]);
1137                 chanrec* chan = NULL;
1138                 time_t ourTS = 0;
1139                 if (dst)
1140                 {
1141                         ourTS = dst->age;
1142                 }
1143                 else
1144                 {
1145                         chan = this->Instance->FindChan(params[0]);
1146                         if (chan)
1147                         {
1148                                 ourTS = chan->age;
1149                         }
1150                         else
1151                                 /* Oops, channel doesnt exist! */
1152                                 return true;
1153                 }
1154
1155                 /* TS is equal: Merge the mode changes, use voooodoooooo on modes
1156                  * with parameters.
1157                  */
1158                 if (TS == ourTS)
1159                 {
1160                         Instance->Log(DEBUG,"Entering TS equality check");
1161                         ModeHandler* mh = NULL;
1162                         unsigned long paramptr = 3;
1163                         std::string to_bounce = "";
1164                         std::string to_keep = "";
1165                         std::vector<std::string> params_to_keep;
1166                         std::string params_to_bounce = "";
1167                         bool adding = true;
1168                         char cur_change = 1;
1169                         char old_change = 0;
1170                         char old_bounce_change = 0;
1171                         /* Merge modes, basically do special stuff to mode with params */
1172                         for (std::string::iterator x = params[2].begin(); x != params[2].end(); x++)
1173                         {
1174                                 switch (*x)
1175                                 {
1176                                         case '-':
1177                                                 adding = false;
1178                                         break;
1179                                         case '+':
1180                                                 adding = true;
1181                                         break;
1182                                         default:
1183                                                 if (adding)
1184                                                 {
1185                                                         /* We only care about whats being set,
1186                                                          * not whats being unset
1187                                                          */
1188                                                         mh = this->Instance->Modes->FindMode(*x, chan ? MODETYPE_CHANNEL : MODETYPE_USER);
1189
1190                                                         if ((mh) && (mh->GetNumParams(adding) > 0) && (!mh->IsListMode()))
1191                                                         {
1192                                                                 /* We only want to do special things to
1193                                                                  * modes with parameters, we are going to rewrite
1194                                                                  * those parameters
1195                                                                  */
1196                                                                 ModePair ret;
1197                                                                 adding ? cur_change = '+' : cur_change = '-';
1198
1199                                                                 ret = mh->ModeSet(smode ? NULL : who, dst, chan, params[paramptr]);
1200
1201                                                                 /* The mode is set here, check which we should keep */
1202                                                                 if (ret.first)
1203                                                                 {
1204                                                                         bool which_to_keep = mh->CheckTimeStamp(TS, ourTS, params[paramptr], ret.second, chan);
1205
1206                                                                         if (which_to_keep == true)
1207                                                                         {
1208                                                                                 /* Keep ours, bounce theirs:
1209                                                                                  * Send back ours to them and
1210                                                                                  * drop their mode changs
1211                                                                                  */
1212                                                                                 adding ? cur_change = '+' : cur_change = '-';
1213                                                                                 if (cur_change != old_bounce_change)
1214                                                                                         to_bounce += cur_change;
1215                                                                                 to_bounce += *x;
1216                                                                                 old_bounce_change = cur_change;
1217
1218                                                                                 if ((mh->GetNumParams(adding) > 0) && (paramptr < params.size()))
1219                                                                                         params_to_bounce.append(" ").append(ret.second);
1220                                                                         }
1221                                                                         else
1222                                                                         {
1223                                                                                 /* Keep theirs: Accept their mode change,
1224                                                                                  * do nothing else
1225                                                                                  */
1226                                                                                 adding ? cur_change = '+' : cur_change = '-';
1227                                                                                 if (cur_change != old_change)
1228                                                                                         to_keep += cur_change;
1229                                                                                 to_keep += *x;
1230                                                                                 old_change = cur_change;
1231
1232                                                                                 if ((mh->GetNumParams(adding) > 0) && (paramptr < params.size()))
1233                                                                                         params_to_keep.push_back(params[paramptr]);
1234                                                                         }
1235                                                                 }
1236                                                                 else
1237                                                                 {
1238                                                                         /* Mode isnt set here, we want it */
1239                                                                         adding ? cur_change = '+' : cur_change = '-';
1240                                                                         if (cur_change != old_change)
1241                                                                                 to_keep += cur_change;
1242                                                                         to_keep += *x;
1243                                                                         old_change = cur_change;
1244
1245                                                                         if ((mh->GetNumParams(adding) > 0) && (paramptr < params.size()))
1246                                                                                 params_to_keep.push_back(params[paramptr]);
1247                                                                 }
1248
1249                                                                 paramptr++;
1250                                                         }
1251                                                         else
1252                                                         {
1253                                                                 mh = this->Instance->Modes->FindMode(*x, chan ? MODETYPE_CHANNEL : MODETYPE_USER);
1254
1255                                                                 if (mh)
1256                                                                 {
1257                                                                         adding ? cur_change = '+' : cur_change = '-';
1258
1259                                                                         /* Just keep this, safe to merge with no checks
1260                                                                          * it has no parameters
1261                                                                          */
1262
1263                                                                         if (cur_change != old_change)
1264                                                                                 to_keep += cur_change;
1265                                                                         to_keep += *x;
1266                                                                         old_change = cur_change;
1267
1268                                                                         if ((mh->GetNumParams(adding) > 0) && (paramptr < params.size()))
1269                                                                         {
1270                                                                                 Instance->Log(DEBUG,"Mode removal %d %d",adding, mh->GetNumParams(adding));
1271                                                                                 params_to_keep.push_back(params[paramptr++]);
1272                                                                         }
1273                                                                 }
1274                                                         }
1275                                                 }
1276                                                 else
1277                                                 {
1278                                                         mh = this->Instance->Modes->FindMode(*x, chan ? MODETYPE_CHANNEL : MODETYPE_USER);
1279
1280                                                         if (mh)
1281                                                         {
1282                                                                 /* Taking a mode away */
1283                                                                 adding ? cur_change = '+' : cur_change = '-';
1284
1285                                                                 if (cur_change != old_change)
1286                                                                         to_keep += cur_change;
1287                                                                 to_keep += *x;
1288                                                                 old_change = cur_change;
1289
1290                                                                 if ((mh->GetNumParams(adding) > 0) && (paramptr < params.size()))
1291                                                                         params_to_keep.push_back(params[paramptr++]);
1292                                                         }
1293                                                 }
1294                                         break;
1295                                 }
1296                         }
1297
1298                         if (to_bounce.length())
1299                         {
1300                                 std::deque<std::string> newparams;
1301                                 newparams.push_back(params[0]);
1302                                 newparams.push_back(ConvToStr(ourTS));
1303                                 newparams.push_back(to_bounce+params_to_bounce);
1304                                 Instance->Log(DEBUG,"BOUNCE BACK: %s",(to_bounce+params_to_bounce).c_str());
1305                                 Utils->DoOneToOne(this->Instance->Config->ServerName,"FMODE",newparams,sourceserv);
1306                         }
1307
1308                         if (to_keep.length())
1309                         {
1310                                 unsigned int n = 2;
1311                                 unsigned int q = 0;
1312                                 modelist[0] = params[0].c_str();
1313                                 modelist[1] = to_keep.c_str();
1314
1315                                 if (params_to_keep.size() > 0)
1316                                 {
1317                                         for (q = 0; (q < params_to_keep.size()) && (q < 64); q++)
1318                                         {
1319                                                 Instance->Log(DEBUG,"KEEP Item %d of %d: %s", q, params_to_keep.size(), params_to_keep[q].c_str());
1320                                                 modelist[n++] = params_to_keep[q].c_str();
1321                                         }
1322                                 }
1323
1324                                 if (smode)
1325                                 {
1326                                         Instance->Log(DEBUG,"Send mode");
1327                                         this->Instance->SendMode(modelist, n, who);
1328                                 }
1329                                 else
1330                                 {
1331                                         Instance->Log(DEBUG,"Send mode client");
1332                                         this->Instance->CallCommandHandler("MODE", modelist, n, who);
1333                                 }
1334
1335                                 /* HOT POTATO! PASS IT ON! */
1336                                 Utils->DoOneToAllButSender(source,"FMODE",params,sourceserv);
1337                         }
1338                 }
1339                 else
1340                 /* U-lined servers always win regardless of their TS */
1341                 if ((TS > ourTS) && (!this->Instance->ULine(source.c_str())))
1342                 {
1343                         /* Bounce the mode back to its sender.* We use our lower TS, so the other end
1344                          * SHOULD accept it, if its clock is right.
1345                          *
1346                          * NOTE: We should check that we arent bouncing anything thats already set at this end.
1347                          * If we are, bounce +ourmode to 'reinforce' it. This prevents desyncs.
1348                          * e.g. They send +l 50, we have +l 10 set. rather than bounce -l 50, we bounce +l 10.
1349                          *
1350                          * Thanks to jilles for pointing out this one-hell-of-an-issue before i even finished
1351                          * writing the code. It took me a while to come up with this solution.
1352                          *
1353                          * XXX: BE SURE YOU UNDERSTAND THIS CODE FULLY BEFORE YOU MESS WITH IT.
1354                          */
1355
1356                         std::deque<std::string> newparams;      /* New parameter list we send back */
1357                         newparams.push_back(params[0]);         /* Target, user or channel */
1358                         newparams.push_back(ConvToStr(ourTS));  /* Timestamp value of the target */
1359                         newparams.push_back("");                /* This contains the mode string. For now
1360                                                                  * it's empty, we fill it below.
1361                                                                  */
1362
1363                         /* Intelligent mode bouncing. Don't just invert, reinforce any modes which are already
1364                          * set to avoid a desync here.
1365                          */
1366                         std::string modebounce = "";
1367                         bool adding = true;
1368                         unsigned int t = 3;
1369                         ModeHandler* mh = NULL;
1370                         char cur_change = 1;
1371                         char old_change = 0;
1372                         for (std::string::iterator x = params[2].begin(); x != params[2].end(); x++)
1373                         {
1374                                 /* Iterate over all mode chars in the sent set */
1375                                 switch (*x)
1376                                 {
1377                                         /* Adding or subtracting modes? */
1378                                         case '-':
1379                                                 adding = false;
1380                                         break;
1381                                         case '+':
1382                                                 adding = true;
1383                                         break;
1384                                         default:
1385                                                 /* Find the mode handler for this mode */
1386                                                 mh = this->Instance->Modes->FindMode(*x, chan ? MODETYPE_CHANNEL : MODETYPE_USER);
1387
1388                                                 /* Got a mode handler?
1389                                                  * This also prevents us bouncing modes we have no handler for.
1390                                                  */
1391                                                 if (mh)
1392                                                 {
1393                                                         ModePair ret;
1394                                                         std::string p = "";
1395
1396                                                         /* Does the mode require a parameter right now?
1397                                                          * If it does, fetch it if we can
1398                                                          */
1399                                                         if ((mh->GetNumParams(adding) > 0) && (t < params.size()))
1400                                                                 p = params[t++];
1401
1402                                                         /* Call the ModeSet method to determine if its set with the
1403                                                          * given parameter here or not.
1404                                                          */
1405                                                         ret = mh->ModeSet(smode ? NULL : who, dst, chan, p);
1406
1407                                                         /* XXX: Really. Dont ask.
1408                                                          * Determine from if its set combined with what the current
1409                                                          * 'state' is (adding or not) as to wether we should 'invert'
1410                                                          * or 'reinforce' the mode change
1411                                                          */
1412                                                         (!ret.first ? (adding ? cur_change = '-' : cur_change = '+') : (!adding ? cur_change = '-' : cur_change = '+'));
1413
1414                                                         /* Quickly determine if we have 'flipped' from + to -,
1415                                                          * or - to +, to prevent unneccessary +/- chars in the
1416                                                          * output string that waste bandwidth
1417                                                          */
1418                                                         if (cur_change != old_change)
1419                                                                 modebounce += cur_change;
1420                                                         old_change = cur_change;
1421
1422                                                         /* Add the mode character to the output string */
1423                                                         modebounce += mh->GetModeChar();
1424
1425                                                         /* We got a parameter back from ModeHandler::ModeSet,
1426                                                          * are we supposed to be sending one out right now?
1427                                                          */
1428                                                         if (ret.second.length())
1429                                                         {
1430                                                                 if (mh->GetNumParams(cur_change == '+') > 0)
1431                                                                         /* Yes we're supposed to be sending out
1432                                                                          * the parameter. Make sure it goes
1433                                                                          */
1434                                                                         newparams.push_back(ret.second);
1435                                                         }
1436
1437                                                 }
1438                                         break;
1439                                 }
1440                         }
1441
1442                         /* Update the parameters for FMODE with the new 'bounced' string */
1443                         newparams[2] = modebounce;
1444                         /* Only send it back the way it came, no need to send it anywhere else */
1445                         Utils->DoOneToOne(this->Instance->Config->ServerName,"FMODE",newparams,sourceserv);
1446                         Instance->Log(DEBUG,"FMODE bounced intelligently, our TS less than theirs and the other server is NOT a uline.");
1447                 }
1448                 else
1449                 {
1450                         Instance->Log(DEBUG,"Allow modes, TS lower for sender");
1451                         /* The server was ulined, but something iffy is up with the TS.
1452                          * Sound the alarm bells!
1453                          */
1454                         if ((this->Instance->ULine(sourceserv.c_str())) && (TS > ourTS))
1455                         {
1456                                 this->Instance->WriteOpers("\2WARNING!\2 U-Lined server '%s' has bad TS for '%s' (accepted change): \2SYNC YOUR CLOCKS\2 to avoid this notice",sourceserv.c_str(),params[0].c_str());
1457                         }
1458                         /* Allow the mode, route it to either server or user command handling */
1459                         if (smode)
1460                                 this->Instance->SendMode(modelist,n,who);
1461                         else
1462                                 this->Instance->CallCommandHandler("MODE", modelist, n, who);
1463
1464                         /* HOT POTATO! PASS IT ON! */
1465                         Utils->DoOneToAllButSender(source,"FMODE",params,sourceserv);
1466                 }
1467                 /* Are we supposed to free the userrec? */
1468                 if (smode)
1469                         DELETE(who);
1470
1471                 return true;
1472         }
1473
1474         /** FTOPIC command */
1475         bool ForceTopic(std::string source, std::deque<std::string> &params)
1476         {
1477                 if (params.size() != 4)
1478                         return true;
1479                 time_t ts = atoi(params[1].c_str());
1480                 std::string nsource = source;
1481
1482                 chanrec* c = this->Instance->FindChan(params[0]);
1483                 if (c)
1484                 {
1485                         if ((ts >= c->topicset) || (!*c->topic))
1486                         {
1487                                 std::string oldtopic = c->topic;
1488                                 strlcpy(c->topic,params[3].c_str(),MAXTOPIC);
1489                                 strlcpy(c->setby,params[2].c_str(),NICKMAX-1);
1490                                 c->topicset = ts;
1491                                 /* if the topic text is the same as the current topic,
1492                                  * dont bother to send the TOPIC command out, just silently
1493                                  * update the set time and set nick.
1494                                  */
1495                                 if (oldtopic != params[3])
1496                                 {
1497                                         userrec* user = this->Instance->FindNick(source);
1498                                         if (!user)
1499                                         {
1500                                                 c->WriteChannelWithServ(source.c_str(), "TOPIC %s :%s", c->name, c->topic);
1501                                         }
1502                                         else
1503                                         {
1504                                                 c->WriteChannel(user, "TOPIC %s :%s", c->name, c->topic);
1505                                                 nsource = user->server;
1506                                         }
1507                                         /* all done, send it on its way */
1508                                         params[3] = ":" + params[3];
1509                                         Utils->DoOneToAllButSender(source,"FTOPIC",params,nsource);
1510                                 }
1511                         }
1512
1513                 }
1514
1515                 return true;
1516         }
1517
1518         /** FJOIN, similar to TS6 SJOIN, but not quite. */
1519         bool ForceJoin(std::string source, std::deque<std::string> &params)
1520         {
1521                 /* 1.1 FJOIN works as follows:
1522                  *
1523                  * Each FJOIN is sent along with a timestamp, and the side with the lowest
1524                  * timestamp 'wins'. From this point on we will refer to this side as the
1525                  * winner. The side with the higher timestamp loses, from this point on we
1526                  * will call this side the loser or losing side. This should be familiar to
1527                  * anyone who's dealt with dreamforge or TS6 before.
1528                  *
1529                  * When two sides of a split heal and this occurs, the following things
1530                  * will happen:
1531                  *
1532                  * If the timestamps are exactly equal, both sides merge their privilages
1533                  * and users, as in InspIRCd 1.0 and ircd2.8. The channels have not been
1534                  * re-created during a split, this is safe to do.
1535                  *
1536                  *
1537                  * If the timestamps are NOT equal, the losing side removes all privilage
1538                  * modes from all of its users that currently exist in the channel, before
1539                  * introducing new users into the channel which are listed in the FJOIN
1540                  * command's parameters. This means, all modes +ohv, and privilages added
1541                  * by modules, such as +qa. The losing side then LOWERS its timestamp value
1542                  * of the channel to match that of the winning side, and the modes of the
1543                  * users of the winning side are merged in with the losing side. The loser
1544                  * then sends out a set of FMODE commands which 'confirm' that it just
1545                  * removed all privilage modes from its existing users, which allows for
1546                  * services packages to still work correctly without needing to know the
1547                  * timestamping rules which InspIRCd follows. In TS6 servers this is always
1548                  * a problem, and services packages must contain code which explicitly
1549                  * behaves as TS6 does, removing ops from the losing side of a split where
1550                  * neccessary within its internal records, as this state information is
1551                  * not explicitly echoed out in that protocol.
1552                  *
1553                  * The winning side on the other hand will ignore all user modes from the
1554                  * losing side, so only its own modes get applied. Life is simple for those
1555                  * who succeed at internets. :-)
1556                  *
1557                  * NOTE: Unlike TS6 and dreamforge and other protocols which have SJOIN,
1558                  * FJOIN does not contain the simple-modes such as +iklmnsp. Why not,
1559                  * you ask? Well, quite simply because we don't need to. They'll be sent
1560                  * after the FJOIN by FMODE, and FMODE is timestamped, so in the event
1561                  * the losing side sends any modes for the channel which shouldnt win,
1562                  * they wont as their timestamp will be too high :-)
1563                  */
1564
1565                 if (params.size() < 3)
1566                         return true;
1567
1568                 char first[MAXBUF];             /* The first parameter of the mode command */
1569                 char modestring[MAXBUF];        /* The mode sequence (2nd parameter) of the mode command */
1570                 char* mode_users[127];          /* The values used by the mode command */
1571                 memset(&mode_users,0,sizeof(mode_users));       /* Initialize mode parameters */
1572                 mode_users[0] = first;          /* Set this up to be our on-stack value */
1573                 mode_users[1] = modestring;     /* Same here as above */
1574                 strcpy(modestring,"+");         /* Initialize the mode sequence to just '+' */
1575                 unsigned int modectr = 2;       /* Pointer to the third mode parameter (e.g. the one after the +-sequence) */
1576
1577                 userrec* who = NULL;                    /* User we are currently checking */
1578                 std::string channel = params[0];        /* Channel name, as a string */
1579                 time_t TS = atoi(params[1].c_str());    /* Timestamp given to us for remote side */
1580
1581                 /* Try and find the channel */
1582                 chanrec* chan = this->Instance->FindChan(channel);
1583
1584                 /* Initialize channel name in the mode parameters */
1585                 strlcpy(mode_users[0],channel.c_str(),MAXBUF);
1586
1587                 /* default TS is a high value, which if we dont have this
1588                  * channel will let the other side apply their modes.
1589                  */
1590                 time_t ourTS = time(NULL)+600;
1591
1592                 /* Does this channel exist? if it does, get its REAL timestamp */
1593                 if (chan)
1594                         ourTS = chan->age;
1595
1596                 /* In 1.1, if they have the newer channel, we immediately clear
1597                  * all status modes from our users. We then accept their modes.
1598                  * If WE have the newer channel its the other side's job to do this.
1599                  * Note that this causes the losing server to send out confirming
1600                  * FMODE lines.
1601                  */
1602                 if (ourTS > TS)
1603                 {
1604                         std::deque<std::string> param_list;
1605
1606                         if (chan)
1607                                 chan->age = TS;
1608
1609                         /* Lower the TS here */
1610                         if (Utils->AnnounceTSChange && chan)
1611                                 chan->WriteChannelWithServ(Instance->Config->ServerName,
1612                                 "TS for %s changed from %lu to %lu", chan->name, ourTS, TS);
1613                         ourTS = TS;
1614
1615                         param_list.push_back(channel);
1616                         /* Zap all the privilage modes on our side */
1617                         this->RemoveStatus(Instance->Config->ServerName, param_list);
1618                 }
1619
1620                 /* Put the final parameter of the FJOIN into a tokenstream ready to split it */
1621                 irc::tokenstream users(params[2]);
1622                 std::string item = "*";
1623
1624                 /* do this first, so our mode reversals are correctly received by other servers
1625                  * if there is a TS collision.
1626                  */
1627                 params[2] = ":" + params[2];
1628                 Utils->DoOneToAllButSender(source,"FJOIN",params,source);
1629
1630                 /* Now, process every 'prefixes,nick' pair */
1631                 while (item != "")
1632                 {
1633                         /* Find next user */
1634                         item = users.GetToken();
1635
1636                         const char* usr = item.c_str();
1637
1638                         /* Safety check just to make sure someones not sent us an FJOIN full of spaces
1639                          * (is this even possible?) */
1640                         if (usr && *usr)
1641                         {
1642                                 const char* permissions = usr;
1643                                 int ntimes = 0;
1644                                 char* nm = new char[MAXBUF];
1645                                 char* tnm = nm;
1646
1647                                 /* Iterate through all the prefix values, convert them from prefixes
1648                                  * to mode letters, and append them to the mode sequence
1649                                  */
1650                                 while ((*permissions) && (*permissions != ',') && (ntimes < MAXBUF))
1651                                 {
1652                                         ModeHandler* mh = Instance->Modes->FindPrefix(*permissions);
1653                                         if (mh)
1654                                         {
1655                                                 /* This is a valid prefix */
1656                                                 ntimes++;
1657                                                 *tnm++ = mh->GetModeChar();
1658                                         }
1659                                         else
1660                                         {
1661                                                 /* Not a valid prefix...
1662                                                  * danger bill bobbertson! (that's will robinsons older brother ;-) ...)
1663                                                  */
1664                                                 this->Instance->WriteOpers("ERROR: We received a user with an unknown prefix '%c'. Closed connection to avoid a desync.",*permissions);
1665                                                 this->WriteLine(std::string("ERROR :Invalid prefix '")+(*permissions)+"' in FJOIN");
1666                                                 return false;
1667                                         }
1668                                         usr++;
1669                                         permissions++;
1670                                 }
1671
1672                                 /* Null terminate modes */
1673                                 *tnm = 0;
1674                                 /* Advance past the comma, to the nick */
1675                                 usr++;
1676
1677                                 /* Check the user actually exists */
1678                                 who = this->Instance->FindNick(usr);
1679                                 if (who)
1680                                 {
1681                                         /* Did they get any modes? How many times? */
1682                                         strlcat(modestring, nm, MAXBUF);
1683                                         for (int k = 0; k < ntimes; k++)
1684                                                 mode_users[modectr++] = strdup(usr);
1685
1686                                         /* Free temporary buffer used for mode sequence */
1687                                         delete[] nm;
1688
1689                                         /* Check that the user's 'direction' is correct
1690                                          * based on the server sending the FJOIN. We must
1691                                          * check each nickname in turn, because the origin of
1692                                          * the FJOIN may be different to the origin of the nicks
1693                                          * in the command itself.
1694                                          */
1695                                         TreeServer* route_back_again = Utils->BestRouteTo(who->server);
1696                                         if ((!route_back_again) || (route_back_again->GetSocket() != this))
1697                                         {
1698                                                 /* Oh dear oh dear. */
1699                                                 Instance->Log(DEBUG,"Fake direction in FJOIN, user '%s'",who->nick);
1700                                                 continue;
1701                                         }
1702                                         /* Finally, we can actually place the user into the channel.
1703                                          * We're sure its right. Final answer, phone a friend.
1704                                          */
1705                                         chanrec::JoinUser(this->Instance, who, channel.c_str(), true, "");
1706
1707                                         /* Have we already queued up MAXMODES modes with parameters
1708                                          * (+qaohv) ready to be sent to the server?
1709                                          */
1710                                         if (modectr >= (MAXMODES-1))
1711                                         {
1712                                                 /* Only actually give the users any status if we lost
1713                                                  * the FJOIN or drew (equal timestamps).
1714                                                  * It isn't actually possible for ourTS to be > TS here,
1715                                                  * only possible to actually have ourTS == TS, or
1716                                                  * ourTS < TS, because if we lost, we already lowered
1717                                                  * our TS above before we entered this loop. We only
1718                                                  * check >= as a safety measure, in case someone stuffed
1719                                                  * up. If someone DID stuff up, it was most likely me.
1720                                                  * Note: I do not like baseball bats in the face...
1721                                                  */
1722                                                 if (ourTS >= TS)
1723                                                 {
1724                                                         Instance->Log(DEBUG,"Our our channel newer than theirs, accepting their modes");
1725                                                         this->Instance->SendMode((const char**)mode_users,modectr,who);
1726
1727                                                         /* Something stuffed up, and for some reason, the timestamp is
1728                                                          * NOT lowered right now and should be. Lower it. Usually this
1729                                                          * code won't be executed, doubtless someone will remove it some
1730                                                          * day soon.
1731                                                          */
1732                                                         if (ourTS > TS)
1733                                                         {
1734                                                                 Instance->Log(DEFAULT,"Channel TS for %s changed from %lu to %lu",chan->name,ourTS,TS);
1735                                                                 chan->age = TS;
1736                                                                 ourTS = TS;
1737                                                         }
1738                                                 }
1739
1740                                                 /* Reset all this back to defaults, and
1741                                                  * free any ram we have left allocated.
1742                                                  */
1743                                                 strcpy(mode_users[1],"+");
1744                                                 for (unsigned int f = 2; f < modectr; f++)
1745                                                         free(mode_users[f]);
1746                                                 modectr = 2;
1747                                         }
1748                                 }
1749                                 else
1750                                 {
1751                                         /* Remember to free this */
1752                                         delete[] nm;
1753                                         /* If we got here, there's a nick in FJOIN which doesnt exist on this server.
1754                                          * We don't try to process the nickname here (that WOULD cause a segfault because
1755                                          * we'd be playing with null pointers) however, we DO pass the nickname on, just
1756                                          * in case somehow we're desynched, so that other users which might be able to see
1757                                          * the nickname get their fair chance to process it.
1758                                          */
1759                                         Instance->Log(SPARSE,"Warning! Invalid user in FJOIN to channel %s IGNORED", channel.c_str());
1760                                         continue;
1761                                 }
1762                         }
1763                 }
1764
1765                 /* there werent enough modes built up to flush it during FJOIN,
1766                  * or, there are a number left over. flush them out.
1767                  */
1768                 if ((modectr > 2) && (who) && (chan))
1769                 {
1770                         if (ourTS >= TS)
1771                         {
1772                                 /* Our channel is newer than theirs. Evil deeds must be afoot. */
1773                                 this->Instance->SendMode((const char**)mode_users,modectr,who);
1774                                 /* Yet again, we can't actually get a true value here, if everything else
1775                                  * is working as it should.
1776                                  */
1777                                 if (ourTS > TS)
1778                                 {
1779                                         Instance->Log(DEFAULT,"Channel TS for %s changed from %lu to %lu",chan->name,ourTS,TS);
1780                                         chan->age = TS;
1781                                         ourTS = TS;
1782                                 }
1783                         }
1784
1785                         /* Free anything we have left to free */
1786                         for (unsigned int f = 2; f < modectr; f++)
1787                                 free(mode_users[f]);
1788                 }
1789
1790                 /* All done. That wasnt so bad was it, you can wipe
1791                  * the sweat from your forehead now. :-)
1792                  */
1793                 return true;
1794         }
1795
1796         /** NICK command */
1797         bool IntroduceClient(std::string source, std::deque<std::string> &params)
1798         {
1799                 if (params.size() < 8)
1800                         return true;
1801                 if (params.size() > 8)
1802                 {
1803                         this->WriteLine(std::string(":")+this->Instance->Config->ServerName+" KILL "+params[1]+" :Invalid client introduction ("+params[1]+"?)");
1804                         return true;
1805                 }
1806                 // NICK age nick host dhost ident +modes ip :gecos
1807                 //       0    1   2     3     4      5   6     7
1808                 time_t age = atoi(params[0].c_str());
1809
1810                 /* This used to have a pretty craq'y loop doing the same thing,
1811                  * now we just let the STL do the hard work (more efficiently)
1812                  */
1813                 std::string::size_type pos_after_plus = params[5].find_first_not_of('+');
1814                 if (pos_after_plus != std::string::npos)
1815                         params[5] = params[5].substr(pos_after_plus);
1816
1817                 const char* tempnick = params[1].c_str();
1818                 Instance->Log(DEBUG,"Introduce client %s!%s@%s",tempnick,params[4].c_str(),params[2].c_str());
1819
1820                 user_hash::iterator iter = this->Instance->clientlist.find(tempnick);
1821
1822                 if (iter != this->Instance->clientlist.end())
1823                 {
1824                         // nick collision
1825                         Instance->Log(DEBUG,"Nick collision on %s!%s@%s: %lu %lu",tempnick,params[4].c_str(),params[2].c_str(),(unsigned long)age,(unsigned long)iter->second->age);
1826                         this->WriteLine(std::string(":")+this->Instance->Config->ServerName+" KILL "+tempnick+" :Nickname collision");
1827                         userrec::QuitUser(this->Instance, iter->second, "Nickname collision");
1828                         return true;
1829                 }
1830
1831                 userrec* _new = new userrec(this->Instance);
1832                 this->Instance->clientlist[tempnick] = _new;
1833                 _new->SetFd(FD_MAGIC_NUMBER);
1834                 strlcpy(_new->nick, tempnick,NICKMAX-1);
1835                 strlcpy(_new->host, params[2].c_str(),63);
1836                 strlcpy(_new->dhost, params[3].c_str(),63);
1837                 _new->server = this->Instance->FindServerNamePtr(source.c_str());
1838                 strlcpy(_new->ident, params[4].c_str(),IDENTMAX);
1839                 strlcpy(_new->fullname, params[7].c_str(),MAXGECOS);
1840                 _new->registered = REG_ALL;
1841                 _new->signon = age;
1842
1843                 for (std::string::iterator v = params[5].begin(); v != params[5].end(); v++)
1844                         _new->modes[(*v)-65] = 1;
1845
1846 #ifdef SUPPORT_IP6LINKS
1847                 if (params[6].find_first_of(":") != std::string::npos)
1848                         _new->SetSockAddr(AF_INET6, params[6].c_str(), 0);
1849                 else
1850 #endif
1851                         _new->SetSockAddr(AF_INET, params[6].c_str(), 0);
1852
1853                 this->Instance->SNO->WriteToSnoMask('C',"Client connecting at %s: %s!%s@%s [%s]",_new->server,_new->nick,_new->ident,_new->host, _new->GetIPString());
1854
1855                 params[7] = ":" + params[7];
1856                 Utils->DoOneToAllButSender(source,"NICK",params,source);
1857
1858                 // Increment the Source Servers User Count..
1859                 TreeServer* SourceServer = Utils->FindServer(source);
1860                 if (SourceServer)
1861                 {
1862                         Instance->Log(DEBUG,"Found source server of %s",_new->nick);
1863                         SourceServer->AddUserCount();
1864                 }
1865
1866                 FOREACH_MOD_I(Instance,I_OnPostConnect,OnPostConnect(_new));
1867
1868                 return true;
1869         }
1870
1871         /** Send one or more FJOINs for a channel of users.
1872          * If the length of a single line is more than 480-NICKMAX
1873          * in length, it is split over multiple lines.
1874          */
1875         void SendFJoins(TreeServer* Current, chanrec* c)
1876         {
1877                 Instance->Log(DEBUG,"Sending FJOINs to other server for %s",c->name);
1878                 char list[MAXBUF];
1879                 std::string individual_halfops = std::string(":")+this->Instance->Config->ServerName+" FMODE "+c->name+" "+ConvToStr(c->age);
1880
1881                 size_t dlen, curlen;
1882                 dlen = curlen = snprintf(list,MAXBUF,":%s FJOIN %s %lu",this->Instance->Config->ServerName,c->name,(unsigned long)c->age);
1883                 int numusers = 0;
1884                 char* ptr = list + dlen;
1885
1886                 CUList *ulist = c->GetUsers();
1887                 std::vector<userrec*> specific_halfop;
1888                 std::vector<userrec*> specific_voice;
1889                 std::string modes = "";
1890                 std::string params = "";
1891
1892                 for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
1893                 {
1894                         // The first parameter gets a : before it
1895                         size_t ptrlen = snprintf(ptr, MAXBUF, " %s%s,%s", !numusers ? ":" : "", c->GetAllPrefixChars(i->second), i->second->nick);
1896
1897                         curlen += ptrlen;
1898                         ptr += ptrlen;
1899
1900                         numusers++;
1901
1902                         if (curlen > (480-NICKMAX))
1903                         {
1904                                 this->WriteLine(list);
1905                                 dlen = curlen = snprintf(list,MAXBUF,":%s FJOIN %s %lu",this->Instance->Config->ServerName,c->name,(unsigned long)c->age);
1906                                 ptr = list + dlen;
1907                                 ptrlen = 0;
1908                                 numusers = 0;
1909                         }
1910                 }
1911
1912                 if (numusers)
1913                         this->WriteLine(list);
1914
1915                 for (BanList::iterator b = c->bans.begin(); b != c->bans.end(); b++)
1916                 {
1917                         modes.append("b");
1918                         params.append(b->data).append(" ");
1919                 }
1920                 this->WriteLine(std::string(":")+this->Instance->Config->ServerName+" FMODE "+c->name+" "+ConvToStr(c->age)+" +"+c->ChanModes(true)+modes+" "+params);
1921         }
1922
1923         /** Send G, Q, Z and E lines */
1924         void SendXLines(TreeServer* Current)
1925         {
1926                 char data[MAXBUF];
1927                 std::string n = this->Instance->Config->ServerName;
1928                 const char* sn = n.c_str();
1929                 int iterations = 0;
1930                 /* Yes, these arent too nice looking, but they get the job done */
1931                 for (std::vector<ZLine*>::iterator i = Instance->XLines->zlines.begin(); i != Instance->XLines->zlines.end(); i++, iterations++)
1932                 {
1933                         snprintf(data,MAXBUF,":%s ADDLINE Z %s %s %lu %lu :%s",sn,(*i)->ipaddr,(*i)->source,(unsigned long)(*i)->set_time,(unsigned long)(*i)->duration,(*i)->reason);
1934                         this->WriteLine(data);
1935                 }
1936                 for (std::vector<QLine*>::iterator i = Instance->XLines->qlines.begin(); i != Instance->XLines->qlines.end(); i++, iterations++)
1937                 {
1938                         snprintf(data,MAXBUF,":%s ADDLINE Q %s %s %lu %lu :%s",sn,(*i)->nick,(*i)->source,(unsigned long)(*i)->set_time,(unsigned long)(*i)->duration,(*i)->reason);
1939                         this->WriteLine(data);
1940                 }
1941                 for (std::vector<GLine*>::iterator i = Instance->XLines->glines.begin(); i != Instance->XLines->glines.end(); i++, iterations++)
1942                 {
1943                         snprintf(data,MAXBUF,":%s ADDLINE G %s %s %lu %lu :%s",sn,(*i)->hostmask,(*i)->source,(unsigned long)(*i)->set_time,(unsigned long)(*i)->duration,(*i)->reason);
1944                         this->WriteLine(data);
1945                 }
1946                 for (std::vector<ELine*>::iterator i = Instance->XLines->elines.begin(); i != Instance->XLines->elines.end(); i++, iterations++)
1947                 {
1948                         snprintf(data,MAXBUF,":%s ADDLINE E %s %s %lu %lu :%s",sn,(*i)->hostmask,(*i)->source,(unsigned long)(*i)->set_time,(unsigned long)(*i)->duration,(*i)->reason);
1949                         this->WriteLine(data);
1950                 }
1951                 for (std::vector<ZLine*>::iterator i = Instance->XLines->pzlines.begin(); i != Instance->XLines->pzlines.end(); i++, iterations++)
1952                 {
1953                         snprintf(data,MAXBUF,":%s ADDLINE Z %s %s %lu %lu :%s",sn,(*i)->ipaddr,(*i)->source,(unsigned long)(*i)->set_time,(unsigned long)(*i)->duration,(*i)->reason);
1954                         this->WriteLine(data);
1955                 }
1956                 for (std::vector<QLine*>::iterator i = Instance->XLines->pqlines.begin(); i != Instance->XLines->pqlines.end(); i++, iterations++)
1957                 {
1958                         snprintf(data,MAXBUF,":%s ADDLINE Q %s %s %lu %lu :%s",sn,(*i)->nick,(*i)->source,(unsigned long)(*i)->set_time,(unsigned long)(*i)->duration,(*i)->reason);
1959                         this->WriteLine(data);
1960                 }
1961                 for (std::vector<GLine*>::iterator i = Instance->XLines->pglines.begin(); i != Instance->XLines->pglines.end(); i++, iterations++)
1962                 {
1963                         snprintf(data,MAXBUF,":%s ADDLINE G %s %s %lu %lu :%s",sn,(*i)->hostmask,(*i)->source,(unsigned long)(*i)->set_time,(unsigned long)(*i)->duration,(*i)->reason);
1964                         this->WriteLine(data);
1965                 }
1966                 for (std::vector<ELine*>::iterator i = Instance->XLines->pelines.begin(); i != Instance->XLines->pelines.end(); i++, iterations++)
1967                 {
1968                         snprintf(data,MAXBUF,":%s ADDLINE E %s %s %lu %lu :%s",sn,(*i)->hostmask,(*i)->source,(unsigned long)(*i)->set_time,(unsigned long)(*i)->duration,(*i)->reason);
1969                         this->WriteLine(data);
1970                 }
1971         }
1972
1973         /** Send channel modes and topics */
1974         void SendChannelModes(TreeServer* Current)
1975         {
1976                 char data[MAXBUF];
1977                 std::deque<std::string> list;
1978                 int iterations = 0;
1979                 std::string n = this->Instance->Config->ServerName;
1980                 const char* sn = n.c_str();
1981                 for (chan_hash::iterator c = this->Instance->chanlist.begin(); c != this->Instance->chanlist.end(); c++, iterations++)
1982                 {
1983                         SendFJoins(Current, c->second);
1984                         if (*c->second->topic)
1985                         {
1986                                 snprintf(data,MAXBUF,":%s FTOPIC %s %lu %s :%s",sn,c->second->name,(unsigned long)c->second->topicset,c->second->setby,c->second->topic);
1987                                 this->WriteLine(data);
1988                         }
1989                         FOREACH_MOD_I(this->Instance,I_OnSyncChannel,OnSyncChannel(c->second,(Module*)Utils->Creator,(void*)this));
1990                         list.clear();
1991                         c->second->GetExtList(list);
1992                         for (unsigned int j = 0; j < list.size(); j++)
1993                         {
1994                                 FOREACH_MOD_I(this->Instance,I_OnSyncChannelMetaData,OnSyncChannelMetaData(c->second,(Module*)Utils->Creator,(void*)this,list[j]));
1995                         }
1996                 }
1997         }
1998
1999         /** send all users and their oper state/modes */
2000         void SendUsers(TreeServer* Current)
2001         {
2002                 char data[MAXBUF];
2003                 std::deque<std::string> list;
2004                 int iterations = 0;
2005                 for (user_hash::iterator u = this->Instance->clientlist.begin(); u != this->Instance->clientlist.end(); u++, iterations++)
2006                 {
2007                         if (u->second->registered == REG_ALL)
2008                         {
2009                                 snprintf(data,MAXBUF,":%s NICK %lu %s %s %s %s +%s %s :%s",u->second->server,(unsigned long)u->second->age,u->second->nick,u->second->host,u->second->dhost,u->second->ident,u->second->FormatModes(),u->second->GetIPString(),u->second->fullname);
2010                                 this->WriteLine(data);
2011                                 if (*u->second->oper)
2012                                 {
2013                                         this->WriteLine(":"+std::string(u->second->nick)+" OPERTYPE "+std::string(u->second->oper));
2014                                 }
2015                                 if (*u->second->awaymsg)
2016                                 {
2017                                         this->WriteLine(":"+std::string(u->second->nick)+" AWAY :"+std::string(u->second->awaymsg));
2018                                 }
2019                                 FOREACH_MOD_I(this->Instance,I_OnSyncUser,OnSyncUser(u->second,(Module*)Utils->Creator,(void*)this));
2020                                 list.clear();
2021                                 u->second->GetExtList(list);
2022                                 for (unsigned int j = 0; j < list.size(); j++)
2023                                 {
2024                                         FOREACH_MOD_I(this->Instance,I_OnSyncUserMetaData,OnSyncUserMetaData(u->second,(Module*)Utils->Creator,(void*)this,list[j]));
2025                                 }
2026                         }
2027                 }
2028         }
2029
2030         /** This function is called when we want to send a netburst to a local
2031          * server. There is a set order we must do this, because for example
2032          * users require their servers to exist, and channels require their
2033          * users to exist. You get the idea.
2034          */
2035         void DoBurst(TreeServer* s)
2036         {
2037                 std::string burst = "BURST "+ConvToStr(time(NULL));
2038                 std::string endburst = "ENDBURST";
2039                 // Because by the end of the netburst, it  could be gone!
2040                 std::string name = s->GetName();
2041                 this->Instance->SNO->WriteToSnoMask('l',"Bursting to \2"+name+"\2.");
2042                 this->WriteLine(burst);
2043                 /* send our version string */
2044                 this->WriteLine(std::string(":")+this->Instance->Config->ServerName+" VERSION :"+this->Instance->GetVersionString());
2045                 /* Send server tree */
2046                 this->SendServers(Utils->TreeRoot,s,1);
2047                 /* Send users and their oper status */
2048                 this->SendUsers(s);
2049                 /* Send everything else (channel modes, xlines etc) */
2050                 this->SendChannelModes(s);
2051                 this->SendXLines(s);
2052                 FOREACH_MOD_I(this->Instance,I_OnSyncOtherMetaData,OnSyncOtherMetaData((Module*)Utils->Creator,(void*)this));
2053                 this->WriteLine(endburst);
2054                 this->Instance->SNO->WriteToSnoMask('l',"Finished bursting to \2"+name+"\2.");
2055         }
2056
2057         /** This function is called when we receive data from a remote
2058          * server. We buffer the data in a std::string (it doesnt stay
2059          * there for long), reading using InspSocket::Read() which can
2060          * read up to 16 kilobytes in one operation.
2061          *
2062          * IF THIS FUNCTION RETURNS FALSE, THE CORE CLOSES AND DELETES
2063          * THE SOCKET OBJECT FOR US.
2064          */
2065         virtual bool OnDataReady()
2066         {
2067                 char* data = this->Read();
2068                 /* Check that the data read is a valid pointer and it has some content */
2069                 if (data && *data)
2070                 {
2071                         this->in_buffer.append(data);
2072                         /* While there is at least one new line in the buffer,
2073                          * do something useful (we hope!) with it.
2074                          */
2075                         while (in_buffer.find("\n") != std::string::npos)
2076                         {
2077                                 std::string ret = in_buffer.substr(0,in_buffer.find("\n")-1);
2078                                 in_buffer = in_buffer.substr(in_buffer.find("\n")+1,in_buffer.length()-in_buffer.find("\n"));
2079                                 if (ret.find("\r") != std::string::npos)
2080                                         ret = in_buffer.substr(0,in_buffer.find("\r")-1);
2081                                 /* Process this one, abort if it
2082                                  * didnt return true.
2083                                  */
2084                                 if (this->ctx_in)
2085                                 {
2086                                         char out[1024];
2087                                         char result[1024];
2088                                         memset(result,0,1024);
2089                                         memset(out,0,1024);
2090                                         /* ERROR + CAPAB is still allowed unencryped */
2091                                         if ((ret.substr(0,7) != "ERROR :") && (ret.substr(0,6) != "CAPAB "))
2092                                         {
2093                                                 int nbytes = from64tobits(out, ret.c_str(), 1024);
2094                                                 if ((nbytes > 0) && (nbytes < 1024))
2095                                                 {
2096                                                         ctx_in->Decrypt(out, result, nbytes, 0);
2097                                                         for (int t = 0; t < nbytes; t++)
2098                                                                 if (result[t] == '\7') result[t] = 0;
2099                                                         ret = result;
2100                                                 }
2101                                         }
2102                                 }
2103                                 if (!this->ProcessLine(ret))
2104                                 {
2105                                         Instance->Log(DEBUG,"ProcessLine says no!");
2106                                         return false;
2107                                 }
2108                         }
2109                         return true;
2110                 }
2111                 /* EAGAIN returns an empty but non-NULL string, so this
2112                  * evaluates to TRUE for EAGAIN but to FALSE for EOF.
2113                  */
2114                 return (data && !*data);
2115         }
2116
2117         int WriteLine(std::string line)
2118         {
2119                 Instance->Log(DEBUG,"OUT: %s",line.c_str());
2120                 if (this->ctx_out)
2121                 {
2122                         char result[10240];
2123                         char result64[10240];
2124                         if (this->keylength)
2125                         {
2126                                 // pad it to the key length
2127                                 int n = this->keylength - (line.length() % this->keylength);
2128                                 if (n)
2129                                         line.append(n,'\7');
2130                         }
2131                         unsigned int ll = line.length();
2132                         ctx_out->Encrypt(line.c_str(), result, ll, 0);
2133                         to64frombits((unsigned char*)result64,(unsigned char*)result,ll);
2134                         line = result64;
2135                 }
2136                 return this->Write(line + "\r\n");
2137         }
2138
2139         /* Handle ERROR command */
2140         bool Error(std::deque<std::string> &params)
2141         {
2142                 if (params.size() < 1)
2143                         return false;
2144                 this->Instance->SNO->WriteToSnoMask('l',"ERROR from %s: %s",(InboundServerName != "" ? InboundServerName.c_str() : myhost.c_str()),params[0].c_str());
2145                 /* we will return false to cause the socket to close. */
2146                 return false;
2147         }
2148
2149         /** remote MOTD. leet, huh? */
2150         bool Motd(std::string prefix, std::deque<std::string> &params)
2151         {
2152                 if (params.size() > 0)
2153                 {
2154                         if (this->Instance->MatchText(this->Instance->Config->ServerName, params[0]))
2155                         {
2156                                 /* It's for our server */
2157                                 string_list results;
2158                                 userrec* source = this->Instance->FindNick(prefix);
2159
2160                                 if (source)
2161                                 {
2162                                         std::deque<std::string> par;
2163                                         par.push_back(prefix);
2164                                         par.push_back("");
2165
2166                                         if (!Instance->Config->MOTD.size())
2167                                         {
2168                                                 par[1] = std::string("::")+Instance->Config->ServerName+" 422 "+source->nick+" :Message of the day file is missing.";
2169                                                 Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
2170                                                 return true;
2171                                         }
2172
2173                                         par[1] = std::string("::")+Instance->Config->ServerName+" 375 "+source->nick+" :"+Instance->Config->ServerName+" message of the day";
2174                                         Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
2175
2176                                         for (unsigned int i = 0; i < Instance->Config->MOTD.size(); i++)
2177                                         {
2178                                                 par[1] = std::string("::")+Instance->Config->ServerName+" 372 "+source->nick+" :- "+Instance->Config->MOTD[i];
2179                                                 Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
2180                                         }
2181
2182                                         par[1] = std::string("::")+Instance->Config->ServerName+" 376 "+source->nick+" End of message of the day.";
2183                                         Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
2184                                 }
2185                         }
2186                         else
2187                         {
2188                                 /* Pass it on */
2189                                 userrec* source = this->Instance->FindNick(prefix);
2190                                 if (source)
2191                                         Utils->DoOneToOne(prefix, "MOTD", params, params[0]);
2192                         }
2193                 }
2194                 return true;
2195         }
2196
2197         /** remote ADMIN. leet, huh? */
2198         bool Admin(std::string prefix, std::deque<std::string> &params)
2199         {
2200                 if (params.size() > 0)
2201                 {
2202                         if (this->Instance->MatchText(this->Instance->Config->ServerName, params[0]))
2203                         {
2204                                 /* It's for our server */
2205                                 string_list results;
2206                                 userrec* source = this->Instance->FindNick(prefix);
2207
2208                                 if (source)
2209                                 {
2210                                         std::deque<std::string> par;
2211                                         par.push_back(prefix);
2212                                         par.push_back("");
2213
2214                                         par[1] = std::string("::")+Instance->Config->ServerName+" 256 "+source->nick+" :Administrative info for "+Instance->Config->ServerName;
2215                                         Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
2216
2217                                         par[1] = std::string("::")+Instance->Config->ServerName+" 257 "+source->nick+" :Name     - "+Instance->Config->AdminName;
2218                                         Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
2219
2220                                         par[1] = std::string("::")+Instance->Config->ServerName+" 258 "+source->nick+" :Nickname - "+Instance->Config->AdminNick;
2221                                         Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
2222
2223                                         par[1] = std::string("::")+Instance->Config->ServerName+" 258 "+source->nick+" :E-Mail   - "+Instance->Config->AdminEmail;
2224                                         Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
2225                                 }
2226                         }
2227                         else
2228                         {
2229                                 /* Pass it on */
2230                                 userrec* source = this->Instance->FindNick(prefix);
2231                                 if (source)
2232                                         Utils->DoOneToOne(prefix, "ADMIN", params, params[0]);
2233                         }
2234                 }
2235                 return true;
2236         }
2237
2238         bool Stats(std::string prefix, std::deque<std::string> &params)
2239         {
2240                 /* Get the reply to a STATS query if it matches this servername,
2241                  * and send it back as a load of PUSH queries
2242                  */
2243                 if (params.size() > 1)
2244                 {
2245                         if (this->Instance->MatchText(this->Instance->Config->ServerName, params[1]))
2246                         {
2247                                 /* It's for our server */
2248                                 string_list results;
2249                                 userrec* source = this->Instance->FindNick(prefix);
2250                                 if (source)
2251                                 {
2252                                         std::deque<std::string> par;
2253                                         par.push_back(prefix);
2254                                         par.push_back("");
2255                                         DoStats(this->Instance, *(params[0].c_str()), source, results);
2256                                         for (size_t i = 0; i < results.size(); i++)
2257                                         {
2258                                                 par[1] = "::" + results[i];
2259                                                 Utils->DoOneToOne(this->Instance->Config->ServerName, "PUSH",par, source->server);
2260                                         }
2261                                 }
2262                         }
2263                         else
2264                         {
2265                                 /* Pass it on */
2266                                 userrec* source = this->Instance->FindNick(prefix);
2267                                 if (source)
2268                                         Utils->DoOneToOne(prefix, "STATS", params, params[1]);
2269                         }
2270                 }
2271                 return true;
2272         }
2273
2274
2275         /** Because the core won't let users or even SERVERS set +o,
2276          * we use the OPERTYPE command to do this.
2277          */
2278         bool OperType(std::string prefix, std::deque<std::string> &params)
2279         {
2280                 if (params.size() != 1)
2281                 {
2282                         Instance->Log(DEBUG,"Received invalid oper type from %s",prefix.c_str());
2283                         return true;
2284                 }
2285                 std::string opertype = params[0];
2286                 userrec* u = this->Instance->FindNick(prefix);
2287                 if (u)
2288                 {
2289                         u->modes[UM_OPERATOR] = 1;
2290                         strlcpy(u->oper,opertype.c_str(),NICKMAX-1);
2291                         Utils->DoOneToAllButSender(u->nick,"OPERTYPE",params,u->server);
2292                 }
2293                 return true;
2294         }
2295
2296         /** Because Andy insists that services-compatible servers must
2297          * implement SVSNICK and SVSJOIN, that's exactly what we do :p
2298          */
2299         bool ForceNick(std::string prefix, std::deque<std::string> &params)
2300         {
2301                 if (params.size() < 3)
2302                         return true;
2303
2304                 userrec* u = this->Instance->FindNick(params[0]);
2305
2306                 if (u)
2307                 {
2308                         Utils->DoOneToAllButSender(prefix,"SVSNICK",params,prefix);
2309                         if (IS_LOCAL(u))
2310                         {
2311                                 std::deque<std::string> par;
2312                                 par.push_back(params[1]);
2313                                 /* This is not required as one is sent in OnUserPostNick below
2314                                  */
2315                                 //Utils->DoOneToMany(u->nick,"NICK",par);
2316                                 if (!u->ForceNickChange(params[1].c_str()))
2317                                 {
2318                                         userrec::QuitUser(this->Instance, u, "Nickname collision");
2319                                         return true;
2320                                 }
2321                                 u->age = atoi(params[2].c_str());
2322                         }
2323                 }
2324                 return true;
2325         }
2326
2327         bool ServiceJoin(std::string prefix, std::deque<std::string> &params)
2328         {
2329                 if (params.size() < 2)
2330                         return true;
2331
2332                 userrec* u = this->Instance->FindNick(params[0]);
2333
2334                 if (u)
2335                 {
2336                         chanrec::JoinUser(this->Instance, u, params[1].c_str(), false);
2337                         Utils->DoOneToAllButSender(prefix,"SVSJOIN",params,prefix);
2338                 }
2339                 return true;
2340         }
2341
2342         bool RemoteRehash(std::string prefix, std::deque<std::string> &params)
2343         {
2344                 if (params.size() < 1)
2345                         return false;
2346
2347                 std::string servermask = params[0];
2348
2349                 if (this->Instance->MatchText(this->Instance->Config->ServerName,servermask))
2350                 {
2351                         this->Instance->SNO->WriteToSnoMask('l',"Remote rehash initiated from server \002"+prefix+"\002.");
2352                         this->Instance->RehashServer();
2353                         Utils->ReadConfiguration(false);
2354                         InitializeDisabledCommands(Instance->Config->DisabledCommands, Instance);
2355                 }
2356                 Utils->DoOneToAllButSender(prefix,"REHASH",params,prefix);
2357                 return true;
2358         }
2359
2360         bool RemoteKill(std::string prefix, std::deque<std::string> &params)
2361         {
2362                 if (params.size() != 2)
2363                         return true;
2364
2365                 std::string nick = params[0];
2366                 userrec* u = this->Instance->FindNick(prefix);
2367                 userrec* who = this->Instance->FindNick(nick);
2368
2369                 if (who)
2370                 {
2371                         /* Prepend kill source, if we don't have one */
2372                         std::string sourceserv = prefix;
2373                         if (u)
2374                         {
2375                                 sourceserv = u->server;
2376                         }
2377                         if (*(params[1].c_str()) != '[')
2378                         {
2379                                 params[1] = "[" + sourceserv + "] Killed (" + params[1] +")";
2380                         }
2381                         std::string reason = params[1];
2382                         params[1] = ":" + params[1];
2383                         Utils->DoOneToAllButSender(prefix,"KILL",params,sourceserv);
2384                         who->Write(":%s KILL %s :%s (%s)", sourceserv.c_str(), who->nick, sourceserv.c_str(), reason.c_str());
2385                         userrec::QuitUser(this->Instance,who,reason);
2386                 }
2387                 return true;
2388         }
2389
2390         bool LocalPong(std::string prefix, std::deque<std::string> &params)
2391         {
2392                 if (params.size() < 1)
2393                         return true;
2394
2395                 if (params.size() == 1)
2396                 {
2397                         TreeServer* ServerSource = Utils->FindServer(prefix);
2398                         if (ServerSource)
2399                         {
2400                                 ServerSource->SetPingFlag();
2401                         }
2402                 }
2403                 else
2404                 {
2405                         std::string forwardto = params[1];
2406                         if (forwardto == this->Instance->Config->ServerName)
2407                         {
2408                                 /*
2409                                  * this is a PONG for us
2410                                  * if the prefix is a user, check theyre local, and if they are,
2411                                  * dump the PONG reply back to their fd. If its a server, do nowt.
2412                                  * Services might want to send these s->s, but we dont need to yet.
2413                                  */
2414                                 userrec* u = this->Instance->FindNick(prefix);
2415
2416                                 if (u)
2417                                 {
2418                                         u->WriteServ("PONG %s %s",params[0].c_str(),params[1].c_str());
2419                                 }
2420                         }
2421                         else
2422                         {
2423                                 // not for us, pass it on :)
2424                                 Utils->DoOneToOne(prefix,"PONG",params,forwardto);
2425                         }
2426                 }
2427
2428                 return true;
2429         }
2430
2431         bool MetaData(std::string prefix, std::deque<std::string> &params)
2432         {
2433                 if (params.size() < 3)
2434                         return true;
2435
2436                 TreeServer* ServerSource = Utils->FindServer(prefix);
2437
2438                 if (ServerSource)
2439                 {
2440                         if (params[0] == "*")
2441                         {
2442                                 FOREACH_MOD_I(this->Instance,I_OnDecodeMetaData,OnDecodeMetaData(TYPE_OTHER,NULL,params[1],params[2]));
2443                         }
2444                         else if (*(params[0].c_str()) == '#')
2445                         {
2446                                 chanrec* c = this->Instance->FindChan(params[0]);
2447                                 if (c)
2448                                 {
2449                                         FOREACH_MOD_I(this->Instance,I_OnDecodeMetaData,OnDecodeMetaData(TYPE_CHANNEL,c,params[1],params[2]));
2450                                 }
2451                         }
2452                         else if (*(params[0].c_str()) != '#')
2453                         {
2454                                 userrec* u = this->Instance->FindNick(params[0]);
2455                                 if (u)
2456                                 {
2457                                         FOREACH_MOD_I(this->Instance,I_OnDecodeMetaData,OnDecodeMetaData(TYPE_USER,u,params[1],params[2]));
2458                                 }
2459                         }
2460                 }
2461
2462                 params[2] = ":" + params[2];
2463                 Utils->DoOneToAllButSender(prefix,"METADATA",params,prefix);
2464                 return true;
2465         }
2466
2467         bool ServerVersion(std::string prefix, std::deque<std::string> &params)
2468         {
2469                 if (params.size() < 1)
2470                         return true;
2471
2472                 TreeServer* ServerSource = Utils->FindServer(prefix);
2473
2474                 if (ServerSource)
2475                 {
2476                         ServerSource->SetVersion(params[0]);
2477                 }
2478                 params[0] = ":" + params[0];
2479                 Utils->DoOneToAllButSender(prefix,"VERSION",params,prefix);
2480                 return true;
2481         }
2482
2483         bool ChangeHost(std::string prefix, std::deque<std::string> &params)
2484         {
2485                 if (params.size() < 1)
2486                         return true;
2487
2488                 userrec* u = this->Instance->FindNick(prefix);
2489
2490                 if (u)
2491                 {
2492                         u->ChangeDisplayedHost(params[0].c_str());
2493                         Utils->DoOneToAllButSender(prefix,"FHOST",params,u->server);
2494                 }
2495                 return true;
2496         }
2497
2498         bool AddLine(std::string prefix, std::deque<std::string> &params)
2499         {
2500                 if (params.size() < 6)
2501                         return true;
2502
2503                 bool propogate = false;
2504
2505                 switch (*(params[0].c_str()))
2506                 {
2507                         case 'Z':
2508                                 propogate = Instance->XLines->add_zline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
2509                                 Instance->XLines->zline_set_creation_time(params[1].c_str(), atoi(params[3].c_str()));
2510                         break;
2511                         case 'Q':
2512                                 propogate = Instance->XLines->add_qline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
2513                                 Instance->XLines->qline_set_creation_time(params[1].c_str(), atoi(params[3].c_str()));
2514                         break;
2515                         case 'E':
2516                                 propogate = Instance->XLines->add_eline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
2517                                 Instance->XLines->eline_set_creation_time(params[1].c_str(), atoi(params[3].c_str()));
2518                         break;
2519                         case 'G':
2520                                 propogate = Instance->XLines->add_gline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
2521                                 Instance->XLines->gline_set_creation_time(params[1].c_str(), atoi(params[3].c_str()));
2522                         break;
2523                         case 'K':
2524                                 propogate = Instance->XLines->add_kline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
2525                         break;
2526                         default:
2527                                 /* Just in case... */
2528                                 this->Instance->SNO->WriteToSnoMask('x',"\2WARNING\2: Invalid xline type '"+params[0]+"' sent by server "+prefix+", ignored!");
2529                                 propogate = false;
2530                         break;
2531                 }
2532
2533                 /* Send it on its way */
2534                 if (propogate)
2535                 {
2536                         if (atoi(params[4].c_str()))
2537                         {
2538                                 this->Instance->SNO->WriteToSnoMask('x',"%s Added %cLINE on %s to expire in %lu seconds (%s).",prefix.c_str(),*(params[0].c_str()),params[1].c_str(),atoi(params[4].c_str()),params[5].c_str());
2539                         }
2540                         else
2541                         {
2542                                 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());
2543                         }
2544                         params[5] = ":" + params[5];
2545                         Utils->DoOneToAllButSender(prefix,"ADDLINE",params,prefix);
2546                 }
2547                 if (!this->bursting)
2548                 {
2549                         Instance->Log(DEBUG,"Applying lines...");
2550                         Instance->XLines->apply_lines(APPLY_ZLINES|APPLY_GLINES|APPLY_QLINES);
2551                 }
2552                 return true;
2553         }
2554
2555         bool ChangeName(std::string prefix, std::deque<std::string> &params)
2556         {
2557                 if (params.size() < 1)
2558                         return true;
2559
2560                 userrec* u = this->Instance->FindNick(prefix);
2561
2562                 if (u)
2563                 {
2564                         u->ChangeName(params[0].c_str());
2565                         params[0] = ":" + params[0];
2566                         Utils->DoOneToAllButSender(prefix,"FNAME",params,u->server);
2567                 }
2568                 return true;
2569         }
2570
2571         bool Whois(std::string prefix, std::deque<std::string> &params)
2572         {
2573                 if (params.size() < 1)
2574                         return true;
2575
2576                 Instance->Log(DEBUG,"In IDLE command");
2577                 userrec* u = this->Instance->FindNick(prefix);
2578
2579                 if (u)
2580                 {
2581                         Instance->Log(DEBUG,"USER EXISTS: %s",u->nick);
2582                         // an incoming request
2583                         if (params.size() == 1)
2584                         {
2585                                 userrec* x = this->Instance->FindNick(params[0]);
2586                                 if ((x) && (IS_LOCAL(x)))
2587                                 {
2588                                         userrec* x = this->Instance->FindNick(params[0]);
2589                                         char signon[MAXBUF];
2590                                         char idle[MAXBUF];
2591
2592                                         snprintf(signon,MAXBUF,"%lu",(unsigned long)x->signon);
2593                                         snprintf(idle,MAXBUF,"%lu",(unsigned long)abs((x->idle_lastmsg)-time(NULL)));
2594                                         std::deque<std::string> par;
2595                                         par.push_back(prefix);
2596                                         par.push_back(signon);
2597                                         par.push_back(idle);
2598                                         // ours, we're done, pass it BACK
2599                                         Utils->DoOneToOne(params[0],"IDLE",par,u->server);
2600                                 }
2601                                 else
2602                                 {
2603                                         // not ours pass it on
2604                                         Utils->DoOneToOne(prefix,"IDLE",params,x->server);
2605                                 }
2606                         }
2607                         else if (params.size() == 3)
2608                         {
2609                                 std::string who_did_the_whois = params[0];
2610                                 userrec* who_to_send_to = this->Instance->FindNick(who_did_the_whois);
2611                                 if ((who_to_send_to) && (IS_LOCAL(who_to_send_to)))
2612                                 {
2613                                         // an incoming reply to a whois we sent out
2614                                         std::string nick_whoised = prefix;
2615                                         unsigned long signon = atoi(params[1].c_str());
2616                                         unsigned long idle = atoi(params[2].c_str());
2617                                         if ((who_to_send_to) && (IS_LOCAL(who_to_send_to)))
2618                                                 do_whois(this->Instance,who_to_send_to,u,signon,idle,nick_whoised.c_str());
2619                                 }
2620                                 else
2621                                 {
2622                                         // not ours, pass it on
2623                                         Utils->DoOneToOne(prefix,"IDLE",params,who_to_send_to->server);
2624                                 }
2625                         }
2626                 }
2627                 return true;
2628         }
2629
2630         bool Push(std::string prefix, std::deque<std::string> &params)
2631         {
2632                 if (params.size() < 2)
2633                         return true;
2634
2635                 userrec* u = this->Instance->FindNick(params[0]);
2636
2637                 if (!u)
2638                         return true;
2639
2640                 if (IS_LOCAL(u))
2641                 {
2642                         u->Write(params[1]);
2643                 }
2644                 else
2645                 {
2646                         // continue the raw onwards
2647                         params[1] = ":" + params[1];
2648                         Utils->DoOneToOne(prefix,"PUSH",params,u->server);
2649                 }
2650                 return true;
2651         }
2652
2653         bool Time(std::string prefix, std::deque<std::string> &params)
2654         {
2655                 // :source.server TIME remote.server sendernick
2656                 // :remote.server TIME source.server sendernick TS
2657                 if (params.size() == 2)
2658                 {
2659                         // someone querying our time?
2660                         if (this->Instance->Config->ServerName == params[0])
2661                         {
2662                                 userrec* u = this->Instance->FindNick(params[1]);
2663                                 if (u)
2664                                 {
2665                                         char curtime[256];
2666                                         snprintf(curtime,256,"%lu",(unsigned long)time(NULL));
2667                                         params.push_back(curtime);
2668                                         params[0] = prefix;
2669                                         Utils->DoOneToOne(this->Instance->Config->ServerName,"TIME",params,params[0]);
2670                                 }
2671                         }
2672                         else
2673                         {
2674                                 // not us, pass it on
2675                                 userrec* u = this->Instance->FindNick(params[1]);
2676                                 if (u)
2677                                         Utils->DoOneToOne(prefix,"TIME",params,params[0]);
2678                         }
2679                 }
2680                 else if (params.size() == 3)
2681                 {
2682                         // a response to a previous TIME
2683                         userrec* u = this->Instance->FindNick(params[1]);
2684                         if ((u) && (IS_LOCAL(u)))
2685                         {
2686                         time_t rawtime = atol(params[2].c_str());
2687                         struct tm * timeinfo;
2688                         timeinfo = localtime(&rawtime);
2689                                 char tms[26];
2690                                 snprintf(tms,26,"%s",asctime(timeinfo));
2691                                 tms[24] = 0;
2692                         u->WriteServ("391 %s %s :%s",u->nick,prefix.c_str(),tms);
2693                         }
2694                         else
2695                         {
2696                                 if (u)
2697                                         Utils->DoOneToOne(prefix,"TIME",params,u->server);
2698                         }
2699                 }
2700                 return true;
2701         }
2702
2703         bool LocalPing(std::string prefix, std::deque<std::string> &params)
2704         {
2705                 if (params.size() < 1)
2706                         return true;
2707
2708                 if (params.size() == 1)
2709                 {
2710                         std::string stufftobounce = params[0];
2711                         this->WriteLine(std::string(":")+this->Instance->Config->ServerName+" PONG "+stufftobounce);
2712                         return true;
2713                 }
2714                 else
2715                 {
2716                         std::string forwardto = params[1];
2717                         if (forwardto == this->Instance->Config->ServerName)
2718                         {
2719                                 // this is a ping for us, send back PONG to the requesting server
2720                                 params[1] = params[0];
2721                                 params[0] = forwardto;
2722                                 Utils->DoOneToOne(forwardto,"PONG",params,params[1]);
2723                         }
2724                         else
2725                         {
2726                                 // not for us, pass it on :)
2727                                 Utils->DoOneToOne(prefix,"PING",params,forwardto);
2728                         }
2729                         return true;
2730                 }
2731         }
2732
2733         bool RemoveStatus(std::string prefix, std::deque<std::string> &params)
2734         {
2735                 if (params.size() < 1)
2736                         return true;
2737
2738                 chanrec* c = Instance->FindChan(params[0]);
2739
2740                 if (c)
2741                 {
2742                         irc::modestacker modestack(false);
2743                         CUList *ulist = c->GetUsers();
2744                         const char* y[127];
2745                         std::deque<std::string> stackresult;
2746                         std::string x;
2747
2748                         for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
2749                         {
2750                                 std::string modesequence = Instance->Modes->ModeString(i->second, c);
2751                                 if (modesequence.length())
2752                                 {
2753                                         Instance->Log(DEBUG,"Mode sequence = '%s'",modesequence.c_str());
2754                                         irc::spacesepstream sep(modesequence);
2755                                         std::string modeletters = sep.GetToken();
2756                                         Instance->Log(DEBUG,"Mode letters = '%s'",modeletters.c_str());
2757
2758                                         while (!modeletters.empty())
2759                                         {
2760                                                 char mletter = *(modeletters.begin());
2761                                                 modestack.Push(mletter,sep.GetToken());
2762                                                 Instance->Log(DEBUG,"Push letter = '%c'",mletter);
2763                                                 modeletters.erase(modeletters.begin());
2764                                                 Instance->Log(DEBUG,"Mode letters = '%s'",modeletters.c_str());
2765                                         }
2766                                 }
2767                         }
2768
2769                         while (modestack.GetStackedLine(stackresult))
2770                         {
2771                                 Instance->Log(DEBUG,"Stacked line size %d",stackresult.size());
2772                                 stackresult.push_front(ConvToStr(c->age));
2773                                 stackresult.push_front(c->name);
2774                                 Utils->DoOneToMany(Instance->Config->ServerName, "FMODE", stackresult);
2775                                 stackresult.erase(stackresult.begin() + 1);
2776                                 Instance->Log(DEBUG,"Stacked items:");
2777                                 for (size_t z = 0; z < stackresult.size(); z++)
2778                                 {
2779                                         y[z] = stackresult[z].c_str();
2780                                         Instance->Log(DEBUG,"\tstackresult[%d]='%s'",z,stackresult[z].c_str());
2781                                 }
2782                                 userrec* n = new userrec(Instance);
2783                                 n->SetFd(FD_MAGIC_NUMBER);
2784                                 Instance->SendMode(y, stackresult.size(), n);
2785                                 delete n;
2786                         }
2787                 }
2788                 return true;
2789         }
2790
2791         bool RemoteServer(std::string prefix, std::deque<std::string> &params)
2792         {
2793                 if (params.size() < 4)
2794                         return false;
2795
2796                 std::string servername = params[0];
2797                 std::string password = params[1];
2798                 // hopcount is not used for a remote server, we calculate this ourselves
2799                 std::string description = params[3];
2800                 TreeServer* ParentOfThis = Utils->FindServer(prefix);
2801
2802                 if (!ParentOfThis)
2803                 {
2804                         this->WriteLine("ERROR :Protocol error - Introduced remote server from unknown server "+prefix);
2805                         return false;
2806                 }
2807                 TreeServer* CheckDupe = Utils->FindServer(servername);
2808                 if (CheckDupe)
2809                 {
2810                         this->WriteLine("ERROR :Server "+servername+" already exists!");
2811                         this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+servername+"\2 denied, already exists");
2812                         return false;
2813                 }
2814                 TreeServer* Node = new TreeServer(this->Utils,this->Instance,servername,description,ParentOfThis,NULL);
2815                 ParentOfThis->AddChild(Node);
2816                 params[3] = ":" + params[3];
2817                 Utils->DoOneToAllButSender(prefix,"SERVER",params,prefix);
2818                 this->Instance->SNO->WriteToSnoMask('l',"Server \002"+prefix+"\002 introduced server \002"+servername+"\002 ("+description+")");
2819                 return true;
2820         }
2821
2822         bool Outbound_Reply_Server(std::deque<std::string> &params)
2823         {
2824                 if (params.size() < 4)
2825                         return false;
2826
2827                 irc::string servername = params[0].c_str();
2828                 std::string sname = params[0];
2829                 std::string password = params[1];
2830                 int hops = atoi(params[2].c_str());
2831
2832                 if (hops)
2833                 {
2834                         this->WriteLine("ERROR :Server too far away for authentication");
2835                         this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, server is too far away for authentication");
2836                         return false;
2837                 }
2838                 std::string description = params[3];
2839                 for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
2840                 {
2841                         if ((x->Name == servername) && (x->RecvPass == password))
2842                         {
2843                                 TreeServer* CheckDupe = Utils->FindServer(sname);
2844                                 if (CheckDupe)
2845                                 {
2846                                         this->WriteLine("ERROR :Server "+sname+" already exists on server "+CheckDupe->GetParent()->GetName()+"!");
2847                                         this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName());
2848                                         return false;
2849                                 }
2850                                 // Begin the sync here. this kickstarts the
2851                                 // other side, waiting in WAIT_AUTH_2 state,
2852                                 // into starting their burst, as it shows
2853                                 // that we're happy.
2854                                 this->LinkState = CONNECTED;
2855                                 // we should add the details of this server now
2856                                 // to the servers tree, as a child of the root
2857                                 // node.
2858                                 TreeServer* Node = new TreeServer(this->Utils,this->Instance,sname,description,Utils->TreeRoot,this);
2859                                 Utils->TreeRoot->AddChild(Node);
2860                                 params[3] = ":" + params[3];
2861                                 Utils->DoOneToAllButSender(Utils->TreeRoot->GetName(),"SERVER",params,sname);
2862                                 this->bursting = true;
2863                                 this->DoBurst(Node);
2864                                 return true;
2865                         }
2866                 }
2867                 this->WriteLine("ERROR :Invalid credentials");
2868                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, invalid link credentials");
2869                 return false;
2870         }
2871
2872         bool Inbound_Server(std::deque<std::string> &params)
2873         {
2874                 if (params.size() < 4)
2875                         return false;
2876
2877                 irc::string servername = params[0].c_str();
2878                 std::string sname = params[0];
2879                 std::string password = params[1];
2880                 int hops = atoi(params[2].c_str());
2881
2882                 if (hops)
2883                 {
2884                         this->WriteLine("ERROR :Server too far away for authentication");
2885                         this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, server is too far away for authentication");
2886                         return false;
2887                 }
2888                 std::string description = params[3];
2889                 for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
2890                 {
2891                         if ((x->Name == servername) && (x->RecvPass == password))
2892                         {
2893                                 TreeServer* CheckDupe = Utils->FindServer(sname);
2894                                 if (CheckDupe)
2895                                 {
2896                                         this->WriteLine("ERROR :Server "+sname+" already exists on server "+CheckDupe->GetParent()->GetName()+"!");
2897                                         this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName());
2898                                         return false;
2899                                 }
2900                                 /* If the config says this link is encrypted, but the remote side
2901                                  * hasnt bothered to send the AES command before SERVER, then we
2902                                  * boot them off as we MUST have this connection encrypted.
2903                                  */
2904                                 if ((x->EncryptionKey != "") && (!this->ctx_in))
2905                                 {
2906                                         this->WriteLine("ERROR :This link requires AES encryption to be enabled. Plaintext connection refused.");
2907                                         this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, remote server did not enable AES.");
2908                                         return false;
2909                                 }
2910                                 this->Instance->SNO->WriteToSnoMask('l',"Verified incoming server connection from \002"+sname+"\002["+(x->HiddenFromStats ? "<hidden>" : this->GetIP())+"] ("+description+")");
2911                                 this->InboundServerName = sname;
2912                                 this->InboundDescription = description;
2913                                 // this is good. Send our details: Our server name and description and hopcount of 0,
2914                                 // along with the sendpass from this block.
2915                                 this->WriteLine(std::string("SERVER ")+this->Instance->Config->ServerName+" "+x->SendPass+" 0 :"+this->Instance->Config->ServerDesc);
2916                                 // move to the next state, we are now waiting for THEM.
2917                                 this->LinkState = WAIT_AUTH_2;
2918                                 return true;
2919                         }
2920                 }
2921                 this->WriteLine("ERROR :Invalid credentials");
2922                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, invalid link credentials");
2923                 return false;
2924         }
2925
2926         void Split(std::string line, std::deque<std::string> &n)
2927         {
2928                 n.clear();
2929                 irc::tokenstream tokens(line);
2930                 std::string param;
2931                 while ((param = tokens.GetToken()) != "")
2932                         n.push_back(param);
2933                 return;
2934         }
2935
2936         bool ProcessLine(std::string line)
2937         {
2938                 std::deque<std::string> params;
2939                 irc::string command;
2940                 std::string prefix;
2941
2942                 if (line.empty())
2943                         return true;
2944
2945                 line = line.substr(0, line.find_first_of("\r\n"));
2946
2947                 Instance->Log(DEBUG,"IN: %s", line.c_str());
2948
2949                 this->Split(line.c_str(),params);
2950
2951                 if ((params[0][0] == ':') && (params.size() > 1))
2952                 {
2953                         prefix = params[0].substr(1);
2954                         params.pop_front();
2955                 }
2956
2957                 command = params[0].c_str();
2958                 params.pop_front();
2959
2960                 if ((!this->ctx_in) && (command == "AES"))
2961                 {
2962                         std::string sserv = params[0];
2963                         for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
2964                         {
2965                                 if ((x->EncryptionKey != "") && (x->Name == sserv))
2966                                 {
2967                                         this->InitAES(x->EncryptionKey,sserv);
2968                                 }
2969                         }
2970
2971                         return true;
2972                 }
2973                 else if ((this->ctx_in) && (command == "AES"))
2974                 {
2975                         this->Instance->SNO->WriteToSnoMask('l',"\2AES\2: Encryption already enabled on this connection yet %s is trying to enable it twice!",params[0].c_str());
2976                 }
2977
2978                 switch (this->LinkState)
2979                 {
2980                         TreeServer* Node;
2981
2982                         case WAIT_AUTH_1:
2983                                 // Waiting for SERVER command from remote server. Server initiating
2984                                 // the connection sends the first SERVER command, listening server
2985                                 // replies with theirs if its happy, then if the initiator is happy,
2986                                 // it starts to send its net sync, which starts the merge, otherwise
2987                                 // it sends an ERROR.
2988                                 if (command == "PASS")
2989                                 {
2990                                         /* Silently ignored */
2991                                 }
2992                                 else if (command == "SERVER")
2993                                 {
2994                                         return this->Inbound_Server(params);
2995                                 }
2996                                 else if (command == "ERROR")
2997                                 {
2998                                         return this->Error(params);
2999                                 }
3000                                 else if (command == "USER")
3001                                 {
3002                                         this->WriteLine("ERROR :Client connections to this port are prohibited.");
3003                                         return false;
3004                                 }
3005                                 else if (command == "CAPAB")
3006                                 {
3007                                         return this->Capab(params);
3008                                 }
3009                                 else if ((command == "U") || (command == "S"))
3010                                 {
3011                                         this->WriteLine("ERROR :Cannot use the old-style mesh linking protocol with m_spanningtree.so!");
3012                                         return false;
3013                                 }
3014                                 else
3015                                 {
3016                                         std::string error("ERROR :Invalid command in negotiation phase: ");
3017                                         error.append(command.c_str());
3018                                         this->WriteLine(error);
3019                                         return false;
3020                                 }
3021                         break;
3022                         case WAIT_AUTH_2:
3023                                 // Waiting for start of other side's netmerge to say they liked our
3024                                 // password.
3025                                 if (command == "SERVER")
3026                                 {
3027                                         // cant do this, they sent it to us in the WAIT_AUTH_1 state!
3028                                         // silently ignore.
3029                                         return true;
3030                                 }
3031                                 else if ((command == "U") || (command == "S"))
3032                                 {
3033                                         this->WriteLine("ERROR :Cannot use the old-style mesh linking protocol with m_spanningtree.so!");
3034                                         return false;
3035                                 }
3036                                 else if (command == "BURST")
3037                                 {
3038                                         if (params.size())
3039                                         {
3040                                                 /* If a time stamp is provided, try and check syncronization */
3041                                                 time_t THEM = atoi(params[0].c_str());
3042                                                 long delta = THEM-time(NULL);
3043                                                 if ((delta < -600) || (delta > 600))
3044                                                 {
3045                                                         this->Instance->SNO->WriteToSnoMask('l',"\2ERROR\2: Your clocks are out by %d seconds (this is more than ten minutes). Link aborted, \2PLEASE SYNC YOUR CLOCKS!\2",abs(delta));
3046                                                         this->WriteLine("ERROR :Your clocks are out by "+ConvToStr(abs(delta))+" seconds (this is more than ten minutes). Link aborted, PLEASE SYNC YOUR CLOCKS!");
3047                                                         return false;
3048                                                 }
3049                                                 else if ((delta < -60) || (delta > 60))
3050                                                 {
3051                                                         this->Instance->SNO->WriteToSnoMask('l',"\2WARNING\2: Your clocks are out by %d seconds, please consider synching your clocks.",abs(delta));
3052                                                 }
3053                                         }
3054                                         this->LinkState = CONNECTED;
3055                                         Node = new TreeServer(this->Utils,this->Instance,InboundServerName,InboundDescription,Utils->TreeRoot,this);
3056                                         Utils->TreeRoot->AddChild(Node);
3057                                         params.clear();
3058                                         params.push_back(InboundServerName);
3059                                         params.push_back("*");
3060                                         params.push_back("1");
3061                                         params.push_back(":"+InboundDescription);
3062                                         Utils->DoOneToAllButSender(Utils->TreeRoot->GetName(),"SERVER",params,InboundServerName);
3063                                         this->bursting = true;
3064                                         this->DoBurst(Node);
3065                                 }
3066                                 else if (command == "ERROR")
3067                                 {
3068                                         return this->Error(params);
3069                                 }
3070                                 else if (command == "CAPAB")
3071                                 {
3072                                         return this->Capab(params);
3073                                 }
3074
3075                         break;
3076                         case LISTENER:
3077                                 this->WriteLine("ERROR :Internal error -- listening socket accepted its own descriptor!!!");
3078                                 return false;
3079                         break;
3080                         case CONNECTING:
3081                                 if (command == "SERVER")
3082                                 {
3083                                         // another server we connected to, which was in WAIT_AUTH_1 state,
3084                                         // has just sent us their credentials. If we get this far, theyre
3085                                         // happy with OUR credentials, and they are now in WAIT_AUTH_2 state.
3086                                         // if we're happy with this, we should send our netburst which
3087                                         // kickstarts the merge.
3088                                         return this->Outbound_Reply_Server(params);
3089                                 }
3090                                 else if (command == "ERROR")
3091                                 {
3092                                         return this->Error(params);
3093                                 }
3094                         break;
3095                         case CONNECTED:
3096                                 // This is the 'authenticated' state, when all passwords
3097                                 // have been exchanged and anything past this point is taken
3098                                 // as gospel.
3099
3100                                 if (prefix != "")
3101                                 {
3102                                         std::string direction = prefix;
3103                                         userrec* t = this->Instance->FindNick(prefix);
3104                                         if (t)
3105                                         {
3106                                                 direction = t->server;
3107                                         }
3108                                         TreeServer* route_back_again = Utils->BestRouteTo(direction);
3109                                         if ((!route_back_again) || (route_back_again->GetSocket() != this))
3110                                         {
3111                                                 if (route_back_again)
3112                                                         Instance->Log(DEBUG,"Protocol violation: Fake direction in command '%s' from connection '%s'",line.c_str(),this->GetName().c_str());
3113                                                 return true;
3114                                         }
3115
3116                                         /* Fix by brain:
3117                                          * When there is activity on the socket, reset the ping counter so
3118                                          * that we're not wasting bandwidth pinging an active server.
3119                                          */
3120                                         route_back_again->SetNextPingTime(time(NULL) + 60);
3121                                         route_back_again->SetPingFlag();
3122                                 }
3123
3124                                 if (command == "SVSMODE")
3125                                 {
3126                                         /* Services expects us to implement
3127                                          * SVSMODE. In inspircd its the same as
3128                                          * MODE anyway.
3129                                          */
3130                                         command = "MODE";
3131                                 }
3132                                 std::string target = "";
3133                                 /* Yes, know, this is a mess. Its reasonably fast though as we're
3134                                  * working with std::string here.
3135                                  */
3136                                 if ((command == "NICK") && (params.size() > 1))
3137                                 {
3138                                         return this->IntroduceClient(prefix,params);
3139                                 }
3140                                 else if (command == "FJOIN")
3141                                 {
3142                                         return this->ForceJoin(prefix,params);
3143                                 }
3144                                 else if (command == "STATS")
3145                                 {
3146                                         return this->Stats(prefix, params);
3147                                 }
3148                                 else if (command == "MOTD")
3149                                 {
3150                                         return this->Motd(prefix, params);
3151                                 }
3152                                 else if (command == "ADMIN")
3153                                 {
3154                                         return this->Admin(prefix, params);
3155                                 }
3156                                 else if (command == "SERVER")
3157                                 {
3158                                         return this->RemoteServer(prefix,params);
3159                                 }
3160                                 else if (command == "ERROR")
3161                                 {
3162                                         return this->Error(params);
3163                                 }
3164                                 else if (command == "OPERTYPE")
3165                                 {
3166                                         return this->OperType(prefix,params);
3167                                 }
3168                                 else if (command == "FMODE")
3169                                 {
3170                                         return this->ForceMode(prefix,params);
3171                                 }
3172                                 else if (command == "KILL")
3173                                 {
3174                                         return this->RemoteKill(prefix,params);
3175                                 }
3176                                 else if (command == "FTOPIC")
3177                                 {
3178                                         return this->ForceTopic(prefix,params);
3179                                 }
3180                                 else if (command == "REHASH")
3181                                 {
3182                                         return this->RemoteRehash(prefix,params);
3183                                 }
3184                                 else if (command == "METADATA")
3185                                 {
3186                                         return this->MetaData(prefix,params);
3187                                 }
3188                                 else if (command == "REMSTATUS")
3189                                 {
3190                                         return this->RemoveStatus(prefix,params);
3191                                 }
3192                                 else if (command == "PING")
3193                                 {
3194                                         /*
3195                                          * We just got a ping from a server that's bursting.
3196                                          * This can't be right, so set them to not bursting, and
3197                                          * apply their lines.
3198                                          */
3199                                         if (this->bursting)
3200                                         {
3201                                                 this->bursting = false;
3202                                                 Instance->XLines->apply_lines(APPLY_ZLINES|APPLY_GLINES|APPLY_QLINES);
3203                                         }
3204                                         if (prefix == "")
3205                                         {
3206                                                 prefix = this->GetName();
3207                                         }
3208                                         return this->LocalPing(prefix,params);
3209                                 }
3210                                 else if (command == "PONG")
3211                                 {
3212                                         /*
3213                                          * We just got a pong from a server that's bursting.
3214                                          * This can't be right, so set them to not bursting, and
3215                                          * apply their lines.
3216                                          */
3217                                         if (this->bursting)
3218                                         {
3219                                                 this->bursting = false;
3220                                                 Instance->XLines->apply_lines(APPLY_ZLINES|APPLY_GLINES|APPLY_QLINES);
3221                                         }
3222                                         if (prefix == "")
3223                                         {
3224                                                 prefix = this->GetName();
3225                                         }
3226                                         return this->LocalPong(prefix,params);
3227                                 }
3228                                 else if (command == "VERSION")
3229                                 {
3230                                         return this->ServerVersion(prefix,params);
3231                                 }
3232                                 else if (command == "FHOST")
3233                                 {
3234                                         return this->ChangeHost(prefix,params);
3235                                 }
3236                                 else if (command == "FNAME")
3237                                 {
3238                                         return this->ChangeName(prefix,params);
3239                                 }
3240                                 else if (command == "ADDLINE")
3241                                 {
3242                                         return this->AddLine(prefix,params);
3243                                 }
3244                                 else if (command == "SVSNICK")
3245                                 {
3246                                         if (prefix == "")
3247                                         {
3248                                                 prefix = this->GetName();
3249                                         }
3250                                         return this->ForceNick(prefix,params);
3251                                 }
3252                                 else if (command == "IDLE")
3253                                 {
3254                                         return this->Whois(prefix,params);
3255                                 }
3256                                 else if (command == "PUSH")
3257                                 {
3258                                         return this->Push(prefix,params);
3259                                 }
3260                                 else if (command == "TIME")
3261                                 {
3262                                         return this->Time(prefix,params);
3263                                 }
3264                                 else if ((command == "KICK") && (Utils->IsServer(prefix)))
3265                                 {
3266                                         std::string sourceserv = this->myhost;
3267                                         if (params.size() == 3)
3268                                         {
3269                                                 userrec* user = this->Instance->FindNick(params[1]);
3270                                                 chanrec* chan = this->Instance->FindChan(params[0]);
3271                                                 if (user && chan)
3272                                                 {
3273                                                         if (!chan->ServerKickUser(user, params[2].c_str(), false))
3274                                                                 /* Yikes, the channels gone! */
3275                                                                 delete chan;
3276                                                 }
3277                                         }
3278                                         if (this->InboundServerName != "")
3279                                         {
3280                                                 sourceserv = this->InboundServerName;
3281                                         }
3282                                         return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
3283                                 }
3284                                 else if (command == "SVSJOIN")
3285                                 {
3286                                         if (prefix == "")
3287                                         {
3288                                                 prefix = this->GetName();
3289                                         }
3290                                         return this->ServiceJoin(prefix,params);
3291                                 }
3292                                 else if (command == "SQUIT")
3293                                 {
3294                                         if (params.size() == 2)
3295                                         {
3296                                                 this->Squit(Utils->FindServer(params[0]),params[1]);
3297                                         }
3298                                         return true;
3299                                 }
3300                                 else if (command == "ENDBURST")
3301                                 {
3302                                         this->bursting = false;
3303                                         Instance->XLines->apply_lines(APPLY_ZLINES|APPLY_GLINES|APPLY_QLINES);
3304                                         std::string sourceserv = this->myhost;
3305                                         if (this->InboundServerName != "")
3306                                         {
3307                                                 sourceserv = this->InboundServerName;
3308                                         }
3309                                         this->Instance->SNO->WriteToSnoMask('l',"Received end of netburst from \2%s\2",sourceserv.c_str());
3310                                         return true;
3311                                 }
3312                                 else
3313                                 {
3314                                         // not a special inter-server command.
3315                                         // Emulate the actual user doing the command,
3316                                         // this saves us having a huge ugly parser.
3317                                         userrec* who = this->Instance->FindNick(prefix);
3318                                         std::string sourceserv = this->myhost;
3319                                         if (this->InboundServerName != "")
3320                                         {
3321                                                 sourceserv = this->InboundServerName;
3322                                         }
3323                                         if (who)
3324                                         {
3325                                                 if ((command == "NICK") && (params.size() > 0))
3326                                                 {
3327                                                         /* On nick messages, check that the nick doesnt
3328                                                          * already exist here. If it does, kill their copy,
3329                                                          * and our copy.
3330                                                          */
3331                                                         userrec* x = this->Instance->FindNick(params[0]);
3332                                                         if ((x) && (x != who))
3333                                                         {
3334                                                                 std::deque<std::string> p;
3335                                                                 p.push_back(params[0]);
3336                                                                 p.push_back("Nickname collision ("+prefix+" -> "+params[0]+")");
3337                                                                 Utils->DoOneToMany(this->Instance->Config->ServerName,"KILL",p);
3338                                                                 p.clear();
3339                                                                 p.push_back(prefix);
3340                                                                 p.push_back("Nickname collision");
3341                                                                 Utils->DoOneToMany(this->Instance->Config->ServerName,"KILL",p);
3342                                                                 userrec::QuitUser(this->Instance,x,"Nickname collision ("+prefix+" -> "+params[0]+")");
3343                                                                 userrec* y = this->Instance->FindNick(prefix);
3344                                                                 if (y)
3345                                                                 {
3346                                                                         userrec::QuitUser(this->Instance,y,"Nickname collision");
3347                                                                 }
3348                                                                 return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
3349                                                         }
3350                                                 }
3351                                                 // its a user
3352                                                 target = who->server;
3353                                                 const char* strparams[127];
3354                                                 for (unsigned int q = 0; q < params.size(); q++)
3355                                                 {
3356                                                         strparams[q] = params[q].c_str();
3357                                                 }
3358                                                 switch (this->Instance->CallCommandHandler(command.c_str(), strparams, params.size(), who))
3359                                                 {
3360                                                         case CMD_INVALID:
3361                                                                 this->WriteLine("ERROR :Unrecognised command '"+std::string(command.c_str())+"' -- possibly loaded mismatched modules");
3362                                                                 return false;
3363                                                         break;
3364                                                         case CMD_FAILURE:
3365                                                                 return true;
3366                                                         break;
3367                                                         default:
3368                                                                 /* CMD_SUCCESS and CMD_USER_DELETED fall through here */
3369                                                         break;
3370                                                 }
3371                                         }
3372                                         else
3373                                         {
3374                                                 // its not a user. Its either a server, or somethings screwed up.
3375                                                 if (Utils->IsServer(prefix))
3376                                                 {
3377                                                         target = this->Instance->Config->ServerName;
3378                                                 }
3379                                                 else
3380                                                 {
3381                                                         Instance->Log(DEBUG,"Command with unknown origin '%s'",prefix.c_str());
3382                                                         return true;
3383                                                 }
3384                                         }
3385                                         return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
3386
3387                                 }
3388                                 return true;
3389                         break;
3390                 }
3391                 return true;
3392         }
3393
3394         virtual std::string GetName()
3395         {
3396                 std::string sourceserv = this->myhost;
3397                 if (this->InboundServerName != "")
3398                 {
3399                         sourceserv = this->InboundServerName;
3400                 }
3401                 return sourceserv;
3402         }
3403
3404         virtual void OnTimeout()
3405         {
3406                 if (this->LinkState == CONNECTING)
3407                 {
3408                         this->Instance->SNO->WriteToSnoMask('l',"CONNECT: Connection to \002"+myhost+"\002 timed out.");
3409                         Link* MyLink = Utils->FindLink(myhost);
3410                         if (MyLink)
3411                                 Utils->DoFailOver(MyLink);
3412                 }
3413         }
3414
3415         virtual void OnClose()
3416         {
3417                 // Connection closed.
3418                 // If the connection is fully up (state CONNECTED)
3419                 // then propogate a netsplit to all peers.
3420                 std::string quitserver = this->myhost;
3421                 if (this->InboundServerName != "")
3422                 {
3423                         quitserver = this->InboundServerName;
3424                 }
3425                 TreeServer* s = Utils->FindServer(quitserver);
3426                 if (s)
3427                 {
3428                         Squit(s,"Remote host closed the connection");
3429                 }
3430                 this->Instance->SNO->WriteToSnoMask('l',"Connection to '\2%s\2' failed.",quitserver.c_str());
3431         }
3432
3433         virtual int OnIncomingConnection(int newsock, char* ip)
3434         {
3435                 /* To prevent anyone from attempting to flood opers/DDoS by connecting to the server port,
3436                  * or discovering if this port is the server port, we don't allow connections from any
3437                  * IPs for which we don't have a link block.
3438                  */
3439                 bool found = false;
3440
3441                 found = (std::find(Utils->ValidIPs.begin(), Utils->ValidIPs.end(), ip) != Utils->ValidIPs.end());
3442                 if (!found)
3443                 {
3444                         for (vector<std::string>::iterator i = Utils->ValidIPs.begin(); i != Utils->ValidIPs.end(); i++)
3445                                 if (irc::sockets::MatchCIDR(ip, (*i).c_str()))
3446                                         found = true;
3447
3448                         if (!found)
3449                         {
3450                                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from %s denied (no link blocks with that IP address)", ip);
3451                                 close(newsock);
3452                                 return false;
3453                         }
3454                 }
3455                 TreeSocket* s = new TreeSocket(this->Utils, this->Instance, newsock, ip);
3456                 s = s; /* Whinge whinge whinge, thats all GCC ever does. */
3457                 return true;
3458         }
3459 };
3460
3461 /** This class is used to resolve server hostnames during /connect and autoconnect.
3462  * As of 1.1, the resolver system is seperated out from InspSocket, so we must do this
3463  * resolver step first ourselves if we need it. This is totally nonblocking, and will
3464  * callback to OnLookupComplete or OnError when completed. Once it has completed we
3465  * will have an IP address which we can then use to continue our connection.
3466  */
3467 class ServernameResolver : public Resolver
3468 {
3469  private:
3470         /** A copy of the Link tag info for what we're connecting to.
3471          * We take a copy, rather than using a pointer, just in case the
3472          * admin takes the tag away and rehashes while the domain is resolving.
3473          */
3474         Link MyLink;
3475         SpanningTreeUtilities* Utils;
3476  public:
3477         ServernameResolver(SpanningTreeUtilities* Util, InspIRCd* Instance, const std::string &hostname, Link x) : Resolver(Instance, hostname, DNS_QUERY_FORWARD), MyLink(x), Utils(Util)
3478         {
3479                 /* Nothing in here, folks */
3480         }
3481
3482         void OnLookupComplete(const std::string &result)
3483         {
3484                 /* Initiate the connection, now that we have an IP to use.
3485                  * Passing a hostname directly to InspSocket causes it to
3486                  * just bail and set its FD to -1.
3487                  */
3488                 TreeServer* CheckDupe = Utils->FindServer(MyLink.Name.c_str());
3489                 if (!CheckDupe) /* Check that nobody tried to connect it successfully while we were resolving */
3490                 {
3491                         TreeSocket* newsocket = new TreeSocket(this->Utils, ServerInstance, result,MyLink.Port,false,10,MyLink.Name.c_str());
3492                         if (newsocket->GetFd() > -1)
3493                         {
3494                                 /* We're all OK */
3495                         }
3496                         else
3497                         {
3498                                 /* Something barfed, show the opers */
3499                                 ServerInstance->SNO->WriteToSnoMask('l',"CONNECT: Error connecting \002%s\002: %s.",MyLink.Name.c_str(),strerror(errno));
3500                                 delete newsocket;
3501                                 Utils->DoFailOver(&MyLink);
3502                         }
3503                 }
3504         }
3505
3506         void OnError(ResolverError e, const std::string &errormessage)
3507         {
3508                 /* Ooops! */
3509                 ServerInstance->SNO->WriteToSnoMask('l',"CONNECT: Error connecting \002%s\002: Unable to resolve hostname - %s",MyLink.Name.c_str(),errormessage.c_str());
3510                 Utils->DoFailOver(&MyLink);
3511         }
3512 };
3513
3514 /** Handle resolving of server IPs for the cache
3515  */
3516 class SecurityIPResolver : public Resolver
3517 {
3518  private:
3519         Link MyLink;
3520         SpanningTreeUtilities* Utils;
3521  public:
3522         SecurityIPResolver(SpanningTreeUtilities* U, InspIRCd* Instance, const std::string &hostname, Link x) : Resolver(Instance, hostname, DNS_QUERY_FORWARD), MyLink(x), Utils(U)
3523         {
3524         }
3525
3526         void OnLookupComplete(const std::string &result)
3527         {
3528                 ServerInstance->Log(DEBUG,"Security IP cache: Adding IP address '%s' for Link '%s'",result.c_str(),MyLink.Name.c_str());
3529                 Utils->ValidIPs.push_back(result);
3530         }
3531
3532         void OnError(ResolverError e, const std::string &errormessage)
3533         {
3534                 ServerInstance->Log(DEBUG,"Could not resolve IP associated with Link '%s': %s",MyLink.Name.c_str(),errormessage.c_str());
3535         }
3536 };
3537
3538 SpanningTreeUtilities::SpanningTreeUtilities(InspIRCd* Instance, ModuleSpanningTree* C) : ServerInstance(Instance), Creator(C)
3539 {
3540         Bindings.clear();
3541         this->TreeRoot = new TreeServer(this, ServerInstance, ServerInstance->Config->ServerName, ServerInstance->Config->ServerDesc);
3542         this->ReadConfiguration(true);
3543 }
3544
3545 SpanningTreeUtilities::~SpanningTreeUtilities()
3546 {
3547         for (unsigned int i = 0; i < Bindings.size(); i++)
3548         {
3549                 ServerInstance->Log(DEBUG,"Freeing binding %d of %d",i, Bindings.size());
3550                 ServerInstance->SE->DelFd(Bindings[i]);
3551                 Bindings[i]->Close();
3552                 DELETE(Bindings[i]);
3553         }
3554         ServerInstance->Log(DEBUG,"Freeing connected servers...");
3555         while (TreeRoot->ChildCount())
3556         {
3557                 TreeServer* child_server = TreeRoot->GetChild(0);
3558                 ServerInstance->Log(DEBUG,"Freeing connected server %s", child_server->GetName().c_str());
3559                 if (child_server)
3560                 {
3561                         TreeSocket* sock = child_server->GetSocket();
3562                         ServerInstance->SE->DelFd(sock);
3563                         sock->Close();
3564                         DELETE(sock);
3565                 }
3566         }
3567 }
3568
3569 void SpanningTreeUtilities::AddThisServer(TreeServer* server, std::deque<TreeServer*> &list)
3570 {
3571         for (unsigned int c = 0; c < list.size(); c++)
3572         {
3573                 if (list[c] == server)
3574                 {
3575                         return;
3576                 }
3577         }
3578         list.push_back(server);
3579 }
3580
3581 /** returns a list of DIRECT servernames for a specific channel */
3582 void SpanningTreeUtilities::GetListOfServersForChannel(chanrec* c, std::deque<TreeServer*> &list)
3583 {
3584         CUList *ulist = c->GetUsers();
3585         for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
3586         {
3587                 if (i->second->GetFd() < 0)
3588                 {
3589                         TreeServer* best = this->BestRouteTo(i->second->server);
3590                         if (best)
3591                                 AddThisServer(best,list);
3592                 }
3593         }
3594         return;
3595 }
3596
3597 bool SpanningTreeUtilities::DoOneToAllButSenderRaw(std::string data, std::string omit, std::string prefix, irc::string command, std::deque<std::string> &params)
3598 {
3599         TreeServer* omitroute = this->BestRouteTo(omit);
3600         if ((command == "NOTICE") || (command == "PRIVMSG"))
3601         {
3602                 if (params.size() >= 2)
3603                 {
3604                         /* Prefixes */
3605                         if ((*(params[0].c_str()) == '@') || (*(params[0].c_str()) == '%') || (*(params[0].c_str()) == '+'))
3606                         {
3607                                 params[0] = params[0].substr(1, params[0].length()-1);
3608                         }
3609                         if ((*(params[0].c_str()) != '#') && (*(params[0].c_str()) != '$'))
3610                         {
3611                                 // special routing for private messages/notices
3612                                 userrec* d = ServerInstance->FindNick(params[0]);
3613                                 if (d)
3614                                 {
3615                                         std::deque<std::string> par;
3616                                         par.push_back(params[0]);
3617                                         par.push_back(":"+params[1]);
3618                                         this->DoOneToOne(prefix,command.c_str(),par,d->server);
3619                                         return true;
3620                                 }
3621                         }
3622                         else if (*(params[0].c_str()) == '$')
3623                         {
3624                                 std::deque<std::string> par;
3625                                 par.push_back(params[0]);
3626                                 par.push_back(":"+params[1]);
3627                                 this->DoOneToAllButSender(prefix,command.c_str(),par,omitroute->GetName());
3628                                 return true;
3629                         }
3630                         else
3631                         {
3632                                 chanrec* c = ServerInstance->FindChan(params[0]);
3633                                 if (c)
3634                                 {
3635                                         std::deque<TreeServer*> list;
3636                                         GetListOfServersForChannel(c,list);
3637                                         unsigned int lsize = list.size();
3638                                         for (unsigned int i = 0; i < lsize; i++)
3639                                         {
3640                                                 TreeSocket* Sock = list[i]->GetSocket();
3641                                                 if ((Sock) && (list[i]->GetName() != omit) && (omitroute != list[i]))
3642                                                 {
3643                                                         Sock->WriteLine(data);
3644                                                 }
3645                                         }
3646                                         return true;
3647                                 }
3648                         }
3649                 }
3650         }
3651         unsigned int items =this->TreeRoot->ChildCount();
3652         for (unsigned int x = 0; x < items; x++)
3653         {
3654                 TreeServer* Route = this->TreeRoot->GetChild(x);
3655                 if ((Route) && (Route->GetSocket()) && (Route->GetName() != omit) && (omitroute != Route))
3656                 {
3657                         TreeSocket* Sock = Route->GetSocket();
3658                         if (Sock)
3659                                 Sock->WriteLine(data);
3660                 }
3661         }
3662         return true;
3663 }
3664
3665 bool SpanningTreeUtilities::DoOneToAllButSender(std::string prefix, std::string command, std::deque<std::string> &params, std::string omit)
3666 {
3667         TreeServer* omitroute = this->BestRouteTo(omit);
3668         std::string FullLine = ":" + prefix + " " + command;
3669         unsigned int words = params.size();
3670         for (unsigned int x = 0; x < words; x++)
3671         {
3672                 FullLine = FullLine + " " + params[x];
3673         }
3674         unsigned int items = this->TreeRoot->ChildCount();
3675         for (unsigned int x = 0; x < items; x++)
3676         {
3677                 TreeServer* Route = this->TreeRoot->GetChild(x);
3678                 // Send the line IF:
3679                 // The route has a socket (its a direct connection)
3680                 // The route isnt the one to be omitted
3681                 // The route isnt the path to the one to be omitted
3682                 if ((Route) && (Route->GetSocket()) && (Route->GetName() != omit) && (omitroute != Route))
3683                 {
3684                         TreeSocket* Sock = Route->GetSocket();
3685                         if (Sock)
3686                                 Sock->WriteLine(FullLine);
3687                 }
3688         }
3689         return true;
3690 }
3691
3692 bool SpanningTreeUtilities::DoOneToMany(std::string prefix, std::string command, std::deque<std::string> &params)
3693 {
3694         std::string FullLine = ":" + prefix + " " + command;
3695         unsigned int words = params.size();
3696         for (unsigned int x = 0; x < words; x++)
3697         {
3698                 FullLine = FullLine + " " + params[x];
3699         }
3700         unsigned int items = this->TreeRoot->ChildCount();
3701         for (unsigned int x = 0; x < items; x++)
3702         {
3703                 TreeServer* Route = this->TreeRoot->GetChild(x);
3704                 if (Route && Route->GetSocket())
3705                 {
3706                         TreeSocket* Sock = Route->GetSocket();
3707                         if (Sock)
3708                                 Sock->WriteLine(FullLine);
3709                 }
3710         }
3711         return true;
3712 }
3713
3714 bool SpanningTreeUtilities::DoOneToMany(const char* prefix, const char* command, std::deque<std::string> &params)
3715 {
3716         std::string spfx = prefix;
3717         std::string scmd = command;
3718         return this->DoOneToMany(spfx, scmd, params);
3719 }
3720
3721 bool SpanningTreeUtilities::DoOneToAllButSender(const char* prefix, const char* command, std::deque<std::string> &params, std::string omit)
3722 {
3723         std::string spfx = prefix;
3724         std::string scmd = command;
3725         return this->DoOneToAllButSender(spfx, scmd, params, omit);
3726 }
3727
3728 bool SpanningTreeUtilities::DoOneToOne(std::string prefix, std::string command, std::deque<std::string> &params, std::string target)
3729 {
3730         TreeServer* Route = this->BestRouteTo(target);
3731         if (Route)
3732         {
3733                 std::string FullLine = ":" + prefix + " " + command;
3734                 unsigned int words = params.size();
3735                 for (unsigned int x = 0; x < words; x++)
3736                 {
3737                         FullLine = FullLine + " " + params[x];
3738                 }
3739                 if (Route && Route->GetSocket())
3740                 {
3741                         TreeSocket* Sock = Route->GetSocket();
3742                         if (Sock)
3743                                 Sock->WriteLine(FullLine);
3744                 }
3745                 return true;
3746         }
3747         else
3748         {
3749                 return false;
3750         }
3751 }
3752
3753 void SpanningTreeUtilities::ReadConfiguration(bool rebind)
3754 {
3755         ConfigReader* Conf = new ConfigReader(ServerInstance);
3756         if (rebind)
3757         {
3758                 for (int j =0; j < Conf->Enumerate("bind"); j++)
3759                 {
3760                         std::string Type = Conf->ReadValue("bind","type",j);
3761                         std::string IP = Conf->ReadValue("bind","address",j);
3762                         int Port = Conf->ReadInteger("bind","port",j,true);
3763                         if (Type == "servers")
3764                         {
3765                                 ServerInstance->Log(DEBUG,"m_spanningtree: Binding server port %s:%d", IP.c_str(), Port);
3766                                 if (IP == "*")
3767                                 {
3768                                         IP = "";
3769                                 }
3770                                 TreeSocket* listener = new TreeSocket(this, ServerInstance, IP.c_str(),Port,true,10);
3771                                 if (listener->GetState() == I_LISTENING)
3772                                 {
3773                                         ServerInstance->Log(DEFAULT,"m_spanningtree: Binding server port %s:%d successful!", IP.c_str(), Port);
3774                                         Bindings.push_back(listener);
3775                                 }
3776                                 else
3777                                 {
3778                                         ServerInstance->Log(DEFAULT,"m_spanningtree: Warning: Failed to bind server port %d",Port);
3779                                         listener->Close();
3780                                         DELETE(listener);
3781                                 }
3782                                 ServerInstance->Log(DEBUG,"Done with this binding");
3783                         }
3784                 }
3785         }
3786         FlatLinks = Conf->ReadFlag("options","flatlinks",0);
3787         HideULines = Conf->ReadFlag("options","hideulines",0);
3788         AnnounceTSChange = Conf->ReadFlag("options","announcets",0);
3789         LinkBlocks.clear();
3790         ValidIPs.clear();
3791         for (int j =0; j < Conf->Enumerate("link"); j++)
3792         {
3793                 Link L;
3794                 std::string Allow = Conf->ReadValue("link","allowmask",j);
3795                 L.Name = (Conf->ReadValue("link","name",j)).c_str();
3796                 L.IPAddr = Conf->ReadValue("link","ipaddr",j);
3797                 L.FailOver = Conf->ReadValue("link","failover",j).c_str();
3798                 L.Port = Conf->ReadInteger("link","port",j,true);
3799                 L.SendPass = Conf->ReadValue("link","sendpass",j);
3800                 L.RecvPass = Conf->ReadValue("link","recvpass",j);
3801                 L.AutoConnect = Conf->ReadInteger("link","autoconnect",j,true);
3802                 L.EncryptionKey =  Conf->ReadValue("link","encryptionkey",j);
3803                 L.HiddenFromStats = Conf->ReadFlag("link","hidden",j);
3804                 L.NextConnectTime = time(NULL) + L.AutoConnect;
3805                 /* Bugfix by brain, do not allow people to enter bad configurations */
3806                 if (L.Name != ServerInstance->Config->ServerName)
3807                 {
3808                         if ((L.IPAddr != "") && (L.RecvPass != "") && (L.SendPass != "") && (L.Name != "") && (L.Port))
3809                         {
3810                                 ValidIPs.push_back(L.IPAddr);
3811
3812                                 if (Allow.length())
3813                                         ValidIPs.push_back(Allow);
3814
3815                                 /* Needs resolving */
3816                                 insp_inaddr binip;
3817                                 if (insp_aton(L.IPAddr.c_str(), &binip) < 1)
3818                                 {
3819                                         try
3820                                         {
3821                                                 SecurityIPResolver* sr = new SecurityIPResolver(this, ServerInstance, L.IPAddr, L);
3822                                                 ServerInstance->AddResolver(sr);
3823                                         }
3824                                         catch (ModuleException& e)
3825                                         {
3826                                                 ServerInstance->Log(DEBUG,"Error in resolver: %s",e.GetReason());
3827                                         }
3828                                 }
3829
3830                                 LinkBlocks.push_back(L);
3831                                 ServerInstance->Log(DEBUG,"m_spanningtree: Read server %s with host %s:%d",L.Name.c_str(),L.IPAddr.c_str(),L.Port);
3832                         }
3833                         else
3834                         {
3835                                 if (L.IPAddr == "")
3836                                 {
3837                                         ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', IP address not defined!",L.Name.c_str());
3838                                 }
3839                                 else if (L.RecvPass == "")
3840                                 {
3841                                         ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', recvpass not defined!",L.Name.c_str());
3842                                 }
3843                                 else if (L.SendPass == "")
3844                                 {
3845                                         ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', sendpass not defined!",L.Name.c_str());
3846                                 }
3847                                 else if (L.Name == "")
3848                                 {
3849                                         ServerInstance->Log(DEFAULT,"Invalid configuration, link tag without a name!");
3850                                 }
3851                                 else if (!L.Port)
3852                                 {
3853                                         ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', no port specified!",L.Name.c_str());
3854                                 }
3855                         }
3856                 }
3857                 else
3858                 {
3859                         ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', link tag has the same server name as the local server!",L.Name.c_str());
3860                 }
3861         }
3862         DELETE(Conf);
3863 }
3864
3865
3866
3867 class ModuleSpanningTree : public Module
3868 {
3869         int line;
3870         int NumServers;
3871         unsigned int max_local;
3872         unsigned int max_global;
3873         cmd_rconnect* command_rconnect;
3874         SpanningTreeUtilities* Utils;
3875
3876  public:
3877
3878         ModuleSpanningTree(InspIRCd* Me)
3879                 : Module::Module(Me), max_local(0), max_global(0)
3880         {
3881                 Utils = new SpanningTreeUtilities(Me, this);
3882
3883                 command_rconnect = new cmd_rconnect(ServerInstance, this, Utils);
3884                 ServerInstance->AddCommand(command_rconnect);
3885         }
3886
3887         void ShowLinks(TreeServer* Current, userrec* user, int hops)
3888         {
3889                 std::string Parent = Utils->TreeRoot->GetName();
3890                 if (Current->GetParent())
3891                 {
3892                         Parent = Current->GetParent()->GetName();
3893                 }
3894                 for (unsigned int q = 0; q < Current->ChildCount(); q++)
3895                 {
3896                         if ((Utils->HideULines) && (ServerInstance->ULine(Current->GetChild(q)->GetName().c_str())))
3897                         {
3898                                 if (*user->oper)
3899                                 {
3900                                          ShowLinks(Current->GetChild(q),user,hops+1);
3901                                 }
3902                         }
3903                         else
3904                         {
3905                                 ShowLinks(Current->GetChild(q),user,hops+1);
3906                         }
3907                 }
3908                 /* Don't display the line if its a uline, hide ulines is on, and the user isnt an oper */
3909                 if ((Utils->HideULines) && (ServerInstance->ULine(Current->GetName().c_str())) && (!*user->oper))
3910                         return;
3911                 user->WriteServ("364 %s %s %s :%d %s",user->nick,Current->GetName().c_str(),(Utils->FlatLinks && (!*user->oper)) ? ServerInstance->Config->ServerName : Parent.c_str(),(Utils->FlatLinks && (!*user->oper)) ? 0 : hops,Current->GetDesc().c_str());
3912         }
3913
3914         int CountLocalServs()
3915         {
3916                 return Utils->TreeRoot->ChildCount();
3917         }
3918
3919         int CountServs()
3920         {
3921                 return Utils->serverlist.size();
3922         }
3923
3924         void HandleLinks(const char** parameters, int pcnt, userrec* user)
3925         {
3926                 ShowLinks(Utils->TreeRoot,user,0);
3927                 user->WriteServ("365 %s * :End of /LINKS list.",user->nick);
3928                 return;
3929         }
3930
3931         void HandleLusers(const char** parameters, int pcnt, userrec* user)
3932         {
3933                 unsigned int n_users = ServerInstance->UserCount();
3934
3935                 /* Only update these when someone wants to see them, more efficient */
3936                 if ((unsigned int)ServerInstance->LocalUserCount() > max_local)
3937                         max_local = ServerInstance->LocalUserCount();
3938                 if (n_users > max_global)
3939                         max_global = n_users;
3940
3941                 user->WriteServ("251 %s :There are %d users and %d invisible on %d servers",user->nick,n_users-ServerInstance->InvisibleUserCount(),ServerInstance->InvisibleUserCount(),this->CountServs());
3942                 if (ServerInstance->OperCount())
3943                         user->WriteServ("252 %s %d :operator(s) online",user->nick,ServerInstance->OperCount());
3944                 if (ServerInstance->UnregisteredUserCount())
3945                         user->WriteServ("253 %s %d :unknown connections",user->nick,ServerInstance->UnregisteredUserCount());
3946                 if (ServerInstance->ChannelCount())
3947                         user->WriteServ("254 %s %d :channels formed",user->nick,ServerInstance->ChannelCount());
3948                 user->WriteServ("254 %s :I have %d clients and %d servers",user->nick,ServerInstance->LocalUserCount(),this->CountLocalServs());
3949                 user->WriteServ("265 %s :Current Local Users: %d  Max: %d",user->nick,ServerInstance->LocalUserCount(),max_local);
3950                 user->WriteServ("266 %s :Current Global Users: %d  Max: %d",user->nick,n_users,max_global);
3951                 return;
3952         }
3953
3954         // WARNING: NOT THREAD SAFE - DONT GET ANY SMART IDEAS.
3955
3956         void ShowMap(TreeServer* Current, userrec* user, int depth, char matrix[128][80], float &totusers, float &totservers)
3957         {
3958                 if (line < 128)
3959                 {
3960                         for (int t = 0; t < depth; t++)
3961                         {
3962                                 matrix[line][t] = ' ';
3963                         }
3964
3965                         // For Aligning, we need to work out exactly how deep this thing is, and produce
3966                         // a 'Spacer' String to compensate.
3967                         char spacer[40];
3968
3969                         memset(spacer,' ',40);
3970                         if ((40 - Current->GetName().length() - depth) > 1) {
3971                                 spacer[40 - Current->GetName().length() - depth] = '\0';
3972                         }
3973                         else
3974                         {
3975                                 spacer[5] = '\0';
3976                         }
3977
3978                         float percent;
3979                         char text[80];
3980                         if (ServerInstance->clientlist.size() == 0) {
3981                                 // If there are no users, WHO THE HELL DID THE /MAP?!?!?!
3982                                 percent = 0;
3983                         }
3984                         else
3985                         {
3986                                 percent = ((float)Current->GetUserCount() / (float)ServerInstance->clientlist.size()) * 100;
3987                         }
3988                         snprintf(text, 80, "%s %s%5d [%5.2f%%]", Current->GetName().c_str(), spacer, Current->GetUserCount(), percent);
3989                         totusers += Current->GetUserCount();
3990                         totservers++;
3991                         strlcpy(&matrix[line][depth],text,80);
3992                         line++;
3993                         for (unsigned int q = 0; q < Current->ChildCount(); q++)
3994                         {
3995                                 if ((Utils->HideULines) && (ServerInstance->ULine(Current->GetChild(q)->GetName().c_str())))
3996                                 {
3997                                         if (*user->oper)
3998                                         {
3999                                                 ShowMap(Current->GetChild(q),user,(Utils->FlatLinks && (!*user->oper)) ? depth : depth+2,matrix,totusers,totservers);
4000                                         }
4001                                 }
4002                                 else
4003                                 {
4004                                         ShowMap(Current->GetChild(q),user,(Utils->FlatLinks && (!*user->oper)) ? depth : depth+2,matrix,totusers,totservers);
4005                                 }
4006                         }
4007                 }
4008         }
4009
4010         int HandleMotd(const char** parameters, int pcnt, userrec* user)
4011         {
4012                 if (pcnt > 0)
4013                 {
4014                         /* Remote MOTD, the server is within the 1st parameter */
4015                         std::deque<std::string> params;
4016                         params.push_back(parameters[0]);
4017
4018                         /* Send it out remotely, generate no reply yet */
4019                         TreeServer* s = Utils->FindServerMask(parameters[0]);
4020                         if (s)
4021                         {
4022                                 Utils->DoOneToOne(user->nick, "MOTD", params, s->GetName());
4023                         }
4024                         else
4025                         {
4026                                 user->WriteServ( "402 %s %s :No such server", user->nick, parameters[0]);
4027                         }
4028                         return 1;
4029                 }
4030                 return 0;
4031         }
4032
4033         int HandleAdmin(const char** parameters, int pcnt, userrec* user)
4034         {
4035                 if (pcnt > 0)
4036                 {
4037                         /* Remote ADMIN, the server is within the 1st parameter */
4038                         std::deque<std::string> params;
4039                         params.push_back(parameters[0]);
4040
4041                         /* Send it out remotely, generate no reply yet */
4042                         TreeServer* s = Utils->FindServerMask(parameters[0]);
4043                         if (s)
4044                         {
4045                                 Utils->DoOneToOne(user->nick, "ADMIN", params, s->GetName());
4046                         }
4047                         else
4048                         {
4049                                 user->WriteServ( "402 %s %s :No such server", user->nick, parameters[0]);
4050                         }
4051                         return 1;
4052                 }
4053                 return 0;
4054         }
4055
4056         int HandleStats(const char** parameters, int pcnt, userrec* user)
4057         {
4058                 if (pcnt > 1)
4059                 {
4060                         /* Remote STATS, the server is within the 2nd parameter */
4061                         std::deque<std::string> params;
4062                         params.push_back(parameters[0]);
4063                         params.push_back(parameters[1]);
4064                         /* Send it out remotely, generate no reply yet */
4065                         TreeServer* s = Utils->FindServerMask(parameters[1]);
4066                         if (s)
4067                         {
4068                                 params[1] = s->GetName();
4069                                 Utils->DoOneToOne(user->nick, "STATS", params, s->GetName());
4070                         }
4071                         else
4072                         {
4073                                 user->WriteServ( "402 %s %s :No such server", user->nick, parameters[0]);
4074                         }
4075                         return 1;
4076                 }
4077                 return 0;
4078         }
4079
4080         // Ok, prepare to be confused.
4081         // After much mulling over how to approach this, it struck me that
4082         // the 'usual' way of doing a /MAP isnt the best way. Instead of
4083         // keeping track of a ton of ascii characters, and line by line
4084         // under recursion working out where to place them using multiplications
4085         // and divisons, we instead render the map onto a backplane of characters
4086         // (a character matrix), then draw the branches as a series of "L" shapes
4087         // from the nodes. This is not only friendlier on CPU it uses less stack.
4088
4089         void HandleMap(const char** parameters, int pcnt, userrec* user)
4090         {
4091                 // This array represents a virtual screen which we will
4092                 // "scratch" draw to, as the console device of an irc
4093                 // client does not provide for a proper terminal.
4094                 float totusers = 0;
4095                 float totservers = 0;
4096                 char matrix[128][80];
4097                 for (unsigned int t = 0; t < 128; t++)
4098                 {
4099                         matrix[t][0] = '\0';
4100                 }
4101                 line = 0;
4102                 // The only recursive bit is called here.
4103                 ShowMap(Utils->TreeRoot,user,0,matrix,totusers,totservers);
4104                 // Process each line one by one. The algorithm has a limit of
4105                 // 128 servers (which is far more than a spanning tree should have
4106                 // anyway, so we're ok). This limit can be raised simply by making
4107                 // the character matrix deeper, 128 rows taking 10k of memory.
4108                 for (int l = 1; l < line; l++)
4109                 {
4110                         // scan across the line looking for the start of the
4111                         // servername (the recursive part of the algorithm has placed
4112                         // the servers at indented positions depending on what they
4113                         // are related to)
4114                         int first_nonspace = 0;
4115                         while (matrix[l][first_nonspace] == ' ')
4116                         {
4117                                 first_nonspace++;
4118                         }
4119                         first_nonspace--;
4120                         // Draw the `- (corner) section: this may be overwritten by
4121                         // another L shape passing along the same vertical pane, becoming
4122                         // a |- (branch) section instead.
4123                         matrix[l][first_nonspace] = '-';
4124                         matrix[l][first_nonspace-1] = '`';
4125                         int l2 = l - 1;
4126                         // Draw upwards until we hit the parent server, causing possibly
4127                         // other corners (`-) to become branches (|-)
4128                         while ((matrix[l2][first_nonspace-1] == ' ') || (matrix[l2][first_nonspace-1] == '`'))
4129                         {
4130                                 matrix[l2][first_nonspace-1] = '|';
4131                                 l2--;
4132                         }
4133                 }
4134                 // dump the whole lot to the user. This is the easy bit, honest.
4135                 for (int t = 0; t < line; t++)
4136                 {
4137                         user->WriteServ("006 %s :%s",user->nick,&matrix[t][0]);
4138                 }
4139                 float avg_users = totusers / totservers;
4140                 user->WriteServ("270 %s :%.0f server%s and %.0f user%s, average %.2f users per server",user->nick,totservers,(totservers > 1 ? "s" : ""),totusers,(totusers > 1 ? "s" : ""),avg_users);
4141         user->WriteServ("007 %s :End of /MAP",user->nick);
4142                 return;
4143         }
4144
4145         int HandleSquit(const char** parameters, int pcnt, userrec* user)
4146         {
4147                 TreeServer* s = Utils->FindServerMask(parameters[0]);
4148                 if (s)
4149                 {
4150                         if (s == Utils->TreeRoot)
4151                         {
4152                                  user->WriteServ("NOTICE %s :*** SQUIT: Foolish mortal, you cannot make a server SQUIT itself! (%s matches local server name)",user->nick,parameters[0]);
4153                                 return 1;
4154                         }
4155                         TreeSocket* sock = s->GetSocket();
4156                         if (sock)
4157                         {
4158                                 ServerInstance->Log(DEBUG,"Splitting server %s",s->GetName().c_str());
4159                                 ServerInstance->SNO->WriteToSnoMask('l',"SQUIT: Server \002%s\002 removed from network by %s",parameters[0],user->nick);
4160                                 sock->Squit(s,"Server quit by "+std::string(user->nick)+"!"+std::string(user->ident)+"@"+std::string(user->host));
4161                                 ServerInstance->SE->DelFd(sock);
4162                                 sock->Close();
4163                                 delete sock;
4164                         }
4165                         else
4166                         {
4167                                 user->WriteServ("NOTICE %s :*** SQUIT: The server \002%s\002 is not directly connected.",user->nick,parameters[0]);
4168                         }
4169                 }
4170                 else
4171                 {
4172                          user->WriteServ("NOTICE %s :*** SQUIT: The server \002%s\002 does not exist on the network.",user->nick,parameters[0]);
4173                 }
4174                 return 1;
4175         }
4176
4177         int HandleTime(const char** parameters, int pcnt, userrec* user)
4178         {
4179                 if ((IS_LOCAL(user)) && (pcnt))
4180                 {
4181                         TreeServer* found = Utils->FindServerMask(parameters[0]);
4182                         if (found)
4183                         {
4184                                 // we dont' override for local server
4185                                 if (found == Utils->TreeRoot)
4186                                         return 0;
4187
4188                                 std::deque<std::string> params;
4189                                 params.push_back(found->GetName());
4190                                 params.push_back(user->nick);
4191                                 Utils->DoOneToOne(ServerInstance->Config->ServerName,"TIME",params,found->GetName());
4192                         }
4193                         else
4194                         {
4195                                 user->WriteServ("402 %s %s :No such server",user->nick,parameters[0]);
4196                         }
4197                 }
4198                 return 1;
4199         }
4200
4201         int HandleRemoteWhois(const char** parameters, int pcnt, userrec* user)
4202         {
4203                 if ((IS_LOCAL(user)) && (pcnt > 1))
4204                 {
4205                         userrec* remote = ServerInstance->FindNick(parameters[1]);
4206                         if ((remote) && (remote->GetFd() < 0))
4207                         {
4208                                 std::deque<std::string> params;
4209                                 params.push_back(parameters[1]);
4210                                 Utils->DoOneToOne(user->nick,"IDLE",params,remote->server);
4211                                 return 1;
4212                         }
4213                         else if (!remote)
4214                         {
4215                                 user->WriteServ("401 %s %s :No such nick/channel",user->nick, parameters[1]);
4216                                 user->WriteServ("318 %s %s :End of /WHOIS list.",user->nick, parameters[1]);
4217                                 return 1;
4218                         }
4219                 }
4220                 return 0;
4221         }
4222
4223         void DoPingChecks(time_t curtime)
4224         {
4225                 for (unsigned int j = 0; j < Utils->TreeRoot->ChildCount(); j++)
4226                 {
4227                         TreeServer* serv = Utils->TreeRoot->GetChild(j);
4228                         TreeSocket* sock = serv->GetSocket();
4229                         if (sock)
4230                         {
4231                                 if (curtime >= serv->NextPingTime())
4232                                 {
4233                                         if (serv->AnsweredLastPing())
4234                                         {
4235                                                 sock->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" PING "+serv->GetName());
4236                                                 serv->SetNextPingTime(curtime + 60);
4237                                         }
4238                                         else
4239                                         {
4240                                                 // they didnt answer, boot them
4241                                                 ServerInstance->SNO->WriteToSnoMask('l',"Server \002%s\002 pinged out",serv->GetName().c_str());
4242                                                 sock->Squit(serv,"Ping timeout");
4243                                                 ServerInstance->SE->DelFd(sock);
4244                                                 sock->Close();
4245                                                 delete sock;
4246                                                 return;
4247                                         }
4248                                 }
4249                         }
4250                 }
4251         }
4252
4253         void ConnectServer(Link* x)
4254         {
4255                 insp_inaddr binip;
4256
4257                 /* Do we already have an IP? If so, no need to resolve it. */
4258                 if (insp_aton(x->IPAddr.c_str(), &binip) > 0)
4259                 {
4260                         TreeSocket* newsocket = new TreeSocket(Utils, ServerInstance, x->IPAddr,x->Port,false,10,x->Name.c_str());
4261                         if (newsocket->GetFd() > -1)
4262                         {
4263                                 /* Handled automatically on success */
4264                         }
4265                         else
4266                         {
4267                                 ServerInstance->SNO->WriteToSnoMask('l',"CONNECT: Error connecting \002%s\002: %s.",x->Name.c_str(),strerror(errno));
4268                                 delete newsocket;
4269                                 Utils->DoFailOver(x);
4270                         }
4271                 }
4272                 else
4273                 {
4274                         try
4275                         {
4276                                 ServernameResolver* snr = new ServernameResolver(Utils, ServerInstance,x->IPAddr, *x);
4277                                 ServerInstance->AddResolver(snr);
4278                         }
4279                         catch (ModuleException& e)
4280                         {
4281                                 ServerInstance->Log(DEBUG,"Error in resolver: %s",e.GetReason());
4282                                 Utils->DoFailOver(x);
4283                         }
4284                 }
4285         }
4286
4287         void AutoConnectServers(time_t curtime)
4288         {
4289                 for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
4290                 {
4291                         if ((x->AutoConnect) && (curtime >= x->NextConnectTime))
4292                         {
4293                                 ServerInstance->Log(DEBUG,"Auto-Connecting %s",x->Name.c_str());
4294                                 x->NextConnectTime = curtime + x->AutoConnect;
4295                                 TreeServer* CheckDupe = Utils->FindServer(x->Name.c_str());
4296                                 if (x->FailOver.length())
4297                                 {
4298                                         TreeServer* CheckFailOver = Utils->FindServer(x->FailOver.c_str());
4299                                         if (CheckFailOver)
4300                                         {
4301                                                 /* The failover for this server is currently a member of the network.
4302                                                  * The failover probably succeeded, where the main link did not.
4303                                                  * Don't try the main link until the failover is gone again.
4304                                                  */
4305                                                 continue;
4306                                         }
4307                                 }
4308                                 if (!CheckDupe)
4309                                 {
4310                                         // an autoconnected server is not connected. Check if its time to connect it
4311                                         ServerInstance->SNO->WriteToSnoMask('l',"AUTOCONNECT: Auto-connecting server \002%s\002 (%lu seconds until next attempt)",x->Name.c_str(),x->AutoConnect);
4312                                         this->ConnectServer(&(*x));
4313                                 }
4314                         }
4315                 }
4316         }
4317
4318         int HandleVersion(const char** parameters, int pcnt, userrec* user)
4319         {
4320                 // we've already checked if pcnt > 0, so this is safe
4321                 TreeServer* found = Utils->FindServerMask(parameters[0]);
4322                 if (found)
4323                 {
4324                         std::string Version = found->GetVersion();
4325                         user->WriteServ("351 %s :%s",user->nick,Version.c_str());
4326                         if (found == Utils->TreeRoot)
4327                         {
4328                                 std::stringstream out(ServerInstance->Config->data005);
4329                                 std::string token = "";
4330                                 std::string line5 = "";
4331                                 int token_counter = 0;
4332
4333                                 while (!out.eof())
4334                                 {
4335                                         out >> token;
4336                                         line5 = line5 + token + " ";
4337                                         token_counter++;
4338
4339                                         if ((token_counter >= 13) || (out.eof() == true))
4340                                         {
4341                                                 user->WriteServ("005 %s %s:are supported by this server",user->nick,line5.c_str());
4342                                                 line5 = "";
4343                                                 token_counter = 0;
4344                                         }
4345                                 }
4346                         }
4347                 }
4348                 else
4349                 {
4350                         user->WriteServ("402 %s %s :No such server",user->nick,parameters[0]);
4351                 }
4352                 return 1;
4353         }
4354
4355         int HandleConnect(const char** parameters, int pcnt, userrec* user)
4356         {
4357                 for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
4358                 {
4359                         if (ServerInstance->MatchText(x->Name.c_str(),parameters[0]))
4360                         {
4361                                 TreeServer* CheckDupe = Utils->FindServer(x->Name.c_str());
4362                                 if (!CheckDupe)
4363                                 {
4364                                         user->WriteServ("NOTICE %s :*** CONNECT: Connecting to server: \002%s\002 (%s:%d)",user->nick,x->Name.c_str(),(x->HiddenFromStats ? "<hidden>" : x->IPAddr.c_str()),x->Port);
4365                                         ConnectServer(&(*x));
4366                                         return 1;
4367                                 }
4368                                 else
4369                                 {
4370                                         user->WriteServ("NOTICE %s :*** CONNECT: Server \002%s\002 already exists on the network and is connected via \002%s\002",user->nick,x->Name.c_str(),CheckDupe->GetParent()->GetName().c_str());
4371                                         return 1;
4372                                 }
4373                         }
4374                 }
4375                 user->WriteServ("NOTICE %s :*** CONNECT: No server matching \002%s\002 could be found in the config file.",user->nick,parameters[0]);
4376                 return 1;
4377         }
4378
4379         virtual int OnStats(char statschar, userrec* user, string_list &results)
4380         {
4381                 if (statschar == 'c')
4382                 {
4383                         for (unsigned int i = 0; i < Utils->LinkBlocks.size(); i++)
4384                         {
4385                                 results.push_back(std::string(ServerInstance->Config->ServerName)+" 213 "+user->nick+" C *@"+(Utils->LinkBlocks[i].HiddenFromStats ? "<hidden>" : Utils->LinkBlocks[i].IPAddr)+" * "+Utils->LinkBlocks[i].Name.c_str()+" "+ConvToStr(Utils->LinkBlocks[i].Port)+" "+(Utils->LinkBlocks[i].EncryptionKey != "" ? 'e' : '-')+(Utils->LinkBlocks[i].AutoConnect ? 'a' : '-')+'s');
4386                                 results.push_back(std::string(ServerInstance->Config->ServerName)+" 244 "+user->nick+" H * * "+Utils->LinkBlocks[i].Name.c_str());
4387                         }
4388                         results.push_back(std::string(ServerInstance->Config->ServerName)+" 219 "+user->nick+" "+statschar+" :End of /STATS report");
4389                         ServerInstance->SNO->WriteToSnoMask('t',"Notice: %s '%c' requested by %s (%s@%s)",(!strcmp(user->server,ServerInstance->Config->ServerName) ? "Stats" : "Remote stats"),statschar,user->nick,user->ident,user->host);
4390                         return 1;
4391                 }
4392                 return 0;
4393         }
4394
4395         virtual int OnPreCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, bool validated, const std::string &original_line)
4396         {
4397                 /* If the command doesnt appear to be valid, we dont want to mess with it. */
4398                 if (!validated)
4399                         return 0;
4400
4401                 if (command == "CONNECT")
4402                 {
4403                         return this->HandleConnect(parameters,pcnt,user);
4404                 }
4405                 else if (command == "STATS")
4406                 {
4407                         return this->HandleStats(parameters,pcnt,user);
4408                 }
4409                 else if (command == "MOTD")
4410                 {
4411                         return this->HandleMotd(parameters,pcnt,user);
4412                 }
4413                 else if (command == "ADMIN")
4414                 {
4415                         return this->HandleAdmin(parameters,pcnt,user);
4416                 }
4417                 else if (command == "SQUIT")
4418                 {
4419                         return this->HandleSquit(parameters,pcnt,user);
4420                 }
4421                 else if (command == "MAP")
4422                 {
4423                         this->HandleMap(parameters,pcnt,user);
4424                         return 1;
4425                 }
4426                 else if ((command == "TIME") && (pcnt > 0))
4427                 {
4428                         return this->HandleTime(parameters,pcnt,user);
4429                 }
4430                 else if (command == "LUSERS")
4431                 {
4432                         this->HandleLusers(parameters,pcnt,user);
4433                         return 1;
4434                 }
4435                 else if (command == "LINKS")
4436                 {
4437                         this->HandleLinks(parameters,pcnt,user);
4438                         return 1;
4439                 }
4440                 else if (command == "WHOIS")
4441                 {
4442                         if (pcnt > 1)
4443                         {
4444                                 // remote whois
4445                                 return this->HandleRemoteWhois(parameters,pcnt,user);
4446                         }
4447                 }
4448                 else if ((command == "VERSION") && (pcnt > 0))
4449                 {
4450                         this->HandleVersion(parameters,pcnt,user);
4451                         return 1;
4452                 }
4453
4454                 return 0;
4455         }
4456
4457         virtual void OnPostCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, CmdResult result, const std::string &original_line)
4458         {
4459                 if ((result == CMD_SUCCESS) && (ServerInstance->IsValidModuleCommand(command, pcnt, user)))
4460                 {
4461                         // this bit of code cleverly routes all module commands
4462                         // to all remote severs *automatically* so that modules
4463                         // can just handle commands locally, without having
4464                         // to have any special provision in place for remote
4465                         // commands and linking protocols.
4466                         std::deque<std::string> params;
4467                         params.clear();
4468                         for (int j = 0; j < pcnt; j++)
4469                         {
4470                                 if (strchr(parameters[j],' '))
4471                                 {
4472                                         params.push_back(":" + std::string(parameters[j]));
4473                                 }
4474                                 else
4475                                 {
4476                                         params.push_back(std::string(parameters[j]));
4477                                 }
4478                         }
4479                         ServerInstance->Log(DEBUG,"Globally route '%s'",command.c_str());
4480                         Utils->DoOneToMany(user->nick,command,params);
4481                 }
4482         }
4483
4484         virtual void OnGetServerDescription(const std::string &servername,std::string &description)
4485         {
4486                 TreeServer* s = Utils->FindServer(servername);
4487                 if (s)
4488                 {
4489                         description = s->GetDesc();
4490                 }
4491         }
4492
4493         virtual void OnUserInvite(userrec* source,userrec* dest,chanrec* channel)
4494         {
4495                 if (IS_LOCAL(source))
4496                 {
4497                         std::deque<std::string> params;
4498                         params.push_back(dest->nick);
4499                         params.push_back(channel->name);
4500                         Utils->DoOneToMany(source->nick,"INVITE",params);
4501                 }
4502         }
4503
4504         virtual void OnPostLocalTopicChange(userrec* user, chanrec* chan, const std::string &topic)
4505         {
4506                 std::deque<std::string> params;
4507                 params.push_back(chan->name);
4508                 params.push_back(":"+topic);
4509                 Utils->DoOneToMany(user->nick,"TOPIC",params);
4510         }
4511
4512         virtual void OnWallops(userrec* user, const std::string &text)
4513         {
4514                 if (IS_LOCAL(user))
4515                 {
4516                         std::deque<std::string> params;
4517                         params.push_back(":"+text);
4518                         Utils->DoOneToMany(user->nick,"WALLOPS",params);
4519                 }
4520         }
4521
4522         virtual void OnUserNotice(userrec* user, void* dest, int target_type, const std::string &text, char status)
4523         {
4524                 if (target_type == TYPE_USER)
4525                 {
4526                         userrec* d = (userrec*)dest;
4527                         if ((d->GetFd() < 0) && (IS_LOCAL(user)))
4528                         {
4529                                 std::deque<std::string> params;
4530                                 params.clear();
4531                                 params.push_back(d->nick);
4532                                 params.push_back(":"+text);
4533                                 Utils->DoOneToOne(user->nick,"NOTICE",params,d->server);
4534                         }
4535                 }
4536                 else if (target_type == TYPE_CHANNEL)
4537                 {
4538                         if (IS_LOCAL(user))
4539                         {
4540                                 chanrec *c = (chanrec*)dest;
4541                                 if (c)
4542                                 {
4543                                         std::string cname = c->name;
4544                                         if (status)
4545                                                 cname = status + cname;
4546                                         std::deque<TreeServer*> list;
4547                                         Utils->GetListOfServersForChannel(c,list);
4548                                         unsigned int ucount = list.size();
4549                                         for (unsigned int i = 0; i < ucount; i++)
4550                                         {
4551                                                 TreeSocket* Sock = list[i]->GetSocket();
4552                                                 if (Sock)
4553                                                         Sock->WriteLine(":"+std::string(user->nick)+" NOTICE "+cname+" :"+text);
4554                                         }
4555                                 }
4556                         }
4557                 }
4558                 else if (target_type == TYPE_SERVER)
4559                 {
4560                         if (IS_LOCAL(user))
4561                         {
4562                                 char* target = (char*)dest;
4563                                 std::deque<std::string> par;
4564                                 par.push_back(target);
4565                                 par.push_back(":"+text);
4566                                 Utils->DoOneToMany(user->nick,"NOTICE",par);
4567                         }
4568                 }
4569         }
4570
4571         virtual void OnUserMessage(userrec* user, void* dest, int target_type, const std::string &text, char status)
4572         {
4573                 if (target_type == TYPE_USER)
4574                 {
4575                         // route private messages which are targetted at clients only to the server
4576                         // which needs to receive them
4577                         userrec* d = (userrec*)dest;
4578                         if ((d->GetFd() < 0) && (IS_LOCAL(user)))
4579                         {
4580                                 std::deque<std::string> params;
4581                                 params.clear();
4582                                 params.push_back(d->nick);
4583                                 params.push_back(":"+text);
4584                                 Utils->DoOneToOne(user->nick,"PRIVMSG",params,d->server);
4585                         }
4586                 }
4587                 else if (target_type == TYPE_CHANNEL)
4588                 {
4589                         if (IS_LOCAL(user))
4590                         {
4591                                 chanrec *c = (chanrec*)dest;
4592                                 if (c)
4593                                 {
4594                                         std::string cname = c->name;
4595                                         if (status)
4596                                                 cname = status + cname;
4597                                         std::deque<TreeServer*> list;
4598                                         Utils->GetListOfServersForChannel(c,list);
4599                                         unsigned int ucount = list.size();
4600                                         for (unsigned int i = 0; i < ucount; i++)
4601                                         {
4602                                                 TreeSocket* Sock = list[i]->GetSocket();
4603                                                 if (Sock)
4604                                                         Sock->WriteLine(":"+std::string(user->nick)+" PRIVMSG "+cname+" :"+text);
4605                                         }
4606                                 }
4607                         }
4608                 }
4609                 else if (target_type == TYPE_SERVER)
4610                 {
4611                         if (IS_LOCAL(user))
4612                         {
4613                                 char* target = (char*)dest;
4614                                 std::deque<std::string> par;
4615                                 par.push_back(target);
4616                                 par.push_back(":"+text);
4617                                 Utils->DoOneToMany(user->nick,"PRIVMSG",par);
4618                         }
4619                 }
4620         }
4621
4622         virtual void OnBackgroundTimer(time_t curtime)
4623         {
4624                 AutoConnectServers(curtime);
4625                 DoPingChecks(curtime);
4626         }
4627
4628         virtual void OnUserJoin(userrec* user, chanrec* channel)
4629         {
4630                 // Only do this for local users
4631                 if (IS_LOCAL(user))
4632                 {
4633                         std::deque<std::string> params;
4634                         params.clear();
4635                         params.push_back(channel->name);
4636                         // set up their permissions and the channel TS with FJOIN.
4637                         // All users are FJOINed now, because a module may specify
4638                         // new joining permissions for the user.
4639                         params.clear();
4640                         params.push_back(channel->name);
4641                         params.push_back(ConvToStr(channel->age));
4642                         params.push_back(std::string(channel->GetAllPrefixChars(user))+","+std::string(user->nick));
4643                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"FJOIN",params);
4644                 }
4645         }
4646
4647         virtual void OnChangeHost(userrec* user, const std::string &newhost)
4648         {
4649                 // only occurs for local clients
4650                 if (user->registered != REG_ALL)
4651                         return;
4652                 std::deque<std::string> params;
4653                 params.push_back(newhost);
4654                 Utils->DoOneToMany(user->nick,"FHOST",params);
4655         }
4656
4657         virtual void OnChangeName(userrec* user, const std::string &gecos)
4658         {
4659                 // only occurs for local clients
4660                 if (user->registered != REG_ALL)
4661                         return;
4662                 std::deque<std::string> params;
4663                 params.push_back(gecos);
4664                 Utils->DoOneToMany(user->nick,"FNAME",params);
4665         }
4666
4667         virtual void OnUserPart(userrec* user, chanrec* channel, const std::string &partmessage)
4668         {
4669                 if (IS_LOCAL(user))
4670                 {
4671                         std::deque<std::string> params;
4672                         params.push_back(channel->name);
4673                         if (partmessage != "")
4674                                 params.push_back(":"+partmessage);
4675                         Utils->DoOneToMany(user->nick,"PART",params);
4676                 }
4677         }
4678
4679         virtual void OnUserConnect(userrec* user)
4680         {
4681                 char agestr[MAXBUF];
4682                 if (IS_LOCAL(user))
4683                 {
4684                         std::deque<std::string> params;
4685                         snprintf(agestr,MAXBUF,"%lu",(unsigned long)user->age);
4686                         params.push_back(agestr);
4687                         params.push_back(user->nick);
4688                         params.push_back(user->host);
4689                         params.push_back(user->dhost);
4690                         params.push_back(user->ident);
4691                         params.push_back("+"+std::string(user->FormatModes()));
4692                         params.push_back(user->GetIPString());
4693                         params.push_back(":"+std::string(user->fullname));
4694                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"NICK",params);
4695
4696                         // User is Local, change needs to be reflected!
4697                         TreeServer* SourceServer = Utils->FindServer(user->server);
4698                         if (SourceServer)
4699                         {
4700                                 SourceServer->AddUserCount();
4701                         }
4702
4703                 }
4704         }
4705
4706         virtual void OnUserQuit(userrec* user, const std::string &reason)
4707         {
4708                 if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
4709                 {
4710                         std::deque<std::string> params;
4711                         params.push_back(":"+reason);
4712                         Utils->DoOneToMany(user->nick,"QUIT",params);
4713                 }
4714                 // Regardless, We need to modify the user Counts..
4715                 TreeServer* SourceServer = Utils->FindServer(user->server);
4716                 if (SourceServer)
4717                 {
4718                         SourceServer->DelUserCount();
4719                 }
4720
4721         }
4722
4723         virtual void OnUserPostNick(userrec* user, const std::string &oldnick)
4724         {
4725                 if (IS_LOCAL(user))
4726                 {
4727                         std::deque<std::string> params;
4728                         params.push_back(user->nick);
4729                         Utils->DoOneToMany(oldnick,"NICK",params);
4730                 }
4731         }
4732
4733         virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, const std::string &reason)
4734         {
4735                 if ((source) && (IS_LOCAL(source)))
4736                 {
4737                         std::deque<std::string> params;
4738                         params.push_back(chan->name);
4739                         params.push_back(user->nick);
4740                         params.push_back(":"+reason);
4741                         Utils->DoOneToMany(source->nick,"KICK",params);
4742                 }
4743                 else if (!source)
4744                 {
4745                         std::deque<std::string> params;
4746                         params.push_back(chan->name);
4747                         params.push_back(user->nick);
4748                         params.push_back(":"+reason);
4749                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"KICK",params);
4750                 }
4751         }
4752
4753         virtual void OnRemoteKill(userrec* source, userrec* dest, const std::string &reason)
4754         {
4755                 std::deque<std::string> params;
4756                 params.push_back(dest->nick);
4757                 params.push_back(":"+reason);
4758                 Utils->DoOneToMany(source->nick,"KILL",params);
4759         }
4760
4761         virtual void OnRehash(const std::string &parameter)
4762         {
4763                 if (parameter != "")
4764                 {
4765                         std::deque<std::string> params;
4766                         params.push_back(parameter);
4767                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"REHASH",params);
4768                         // check for self
4769                         if (ServerInstance->MatchText(ServerInstance->Config->ServerName,parameter))
4770                         {
4771                                 ServerInstance->WriteOpers("*** Remote rehash initiated from server \002%s\002",ServerInstance->Config->ServerName);
4772                                 ServerInstance->RehashServer();
4773                         }
4774                 }
4775                 Utils->ReadConfiguration(false);
4776         }
4777
4778         // note: the protocol does not allow direct umode +o except
4779         // via NICK with 8 params. sending OPERTYPE infers +o modechange
4780         // locally.
4781         virtual void OnOper(userrec* user, const std::string &opertype)
4782         {
4783                 if (IS_LOCAL(user))
4784                 {
4785                         std::deque<std::string> params;
4786                         params.push_back(opertype);
4787                         Utils->DoOneToMany(user->nick,"OPERTYPE",params);
4788                 }
4789         }
4790
4791         void OnLine(userrec* source, const std::string &host, bool adding, char linetype, long duration, const std::string &reason)
4792         {
4793                 if (IS_LOCAL(source))
4794                 {
4795                         char type[8];
4796                         snprintf(type,8,"%cLINE",linetype);
4797                         std::string stype = type;
4798                         if (adding)
4799                         {
4800                                 char sduration[MAXBUF];
4801                                 snprintf(sduration,MAXBUF,"%ld",duration);
4802                                 std::deque<std::string> params;
4803                                 params.push_back(host);
4804                                 params.push_back(sduration);
4805                                 params.push_back(":"+reason);
4806                                 Utils->DoOneToMany(source->nick,stype,params);
4807                         }
4808                         else
4809                         {
4810                                 std::deque<std::string> params;
4811                                 params.push_back(host);
4812                                 Utils->DoOneToMany(source->nick,stype,params);
4813                         }
4814                 }
4815         }
4816
4817         virtual void OnAddGLine(long duration, userrec* source, const std::string &reason, const std::string &hostmask)
4818         {
4819                 OnLine(source,hostmask,true,'G',duration,reason);
4820         }
4821
4822         virtual void OnAddZLine(long duration, userrec* source, const std::string &reason, const std::string &ipmask)
4823         {
4824                 OnLine(source,ipmask,true,'Z',duration,reason);
4825         }
4826
4827         virtual void OnAddQLine(long duration, userrec* source, const std::string &reason, const std::string &nickmask)
4828         {
4829                 OnLine(source,nickmask,true,'Q',duration,reason);
4830         }
4831
4832         virtual void OnAddELine(long duration, userrec* source, const std::string &reason, const std::string &hostmask)
4833         {
4834                 OnLine(source,hostmask,true,'E',duration,reason);
4835         }
4836
4837         virtual void OnDelGLine(userrec* source, const std::string &hostmask)
4838         {
4839                 OnLine(source,hostmask,false,'G',0,"");
4840         }
4841
4842         virtual void OnDelZLine(userrec* source, const std::string &ipmask)
4843         {
4844                 OnLine(source,ipmask,false,'Z',0,"");
4845         }
4846
4847         virtual void OnDelQLine(userrec* source, const std::string &nickmask)
4848         {
4849                 OnLine(source,nickmask,false,'Q',0,"");
4850         }
4851
4852         virtual void OnDelELine(userrec* source, const std::string &hostmask)
4853         {
4854                 OnLine(source,hostmask,false,'E',0,"");
4855         }
4856
4857         virtual void OnMode(userrec* user, void* dest, int target_type, const std::string &text)
4858         {
4859                 if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
4860                 {
4861                         if (target_type == TYPE_USER)
4862                         {
4863                                 userrec* u = (userrec*)dest;
4864                                 std::deque<std::string> params;
4865                                 params.push_back(u->nick);
4866                                 params.push_back(text);
4867                                 Utils->DoOneToMany(user->nick,"MODE",params);
4868                         }
4869                         else
4870                         {
4871                                 chanrec* c = (chanrec*)dest;
4872                                 std::deque<std::string> params;
4873                                 params.push_back(c->name);
4874                                 params.push_back(text);
4875                                 Utils->DoOneToMany(user->nick,"MODE",params);
4876                         }
4877                 }
4878         }
4879
4880         virtual void OnSetAway(userrec* user)
4881         {
4882                 if (IS_LOCAL(user))
4883                 {
4884                         std::deque<std::string> params;
4885                         params.push_back(":"+std::string(user->awaymsg));
4886                         Utils->DoOneToMany(user->nick,"AWAY",params);
4887                 }
4888         }
4889
4890         virtual void OnCancelAway(userrec* user)
4891         {
4892                 if (IS_LOCAL(user))
4893                 {
4894                         std::deque<std::string> params;
4895                         params.clear();
4896                         Utils->DoOneToMany(user->nick,"AWAY",params);
4897                 }
4898         }
4899
4900         virtual void ProtoSendMode(void* opaque, int target_type, void* target, const std::string &modeline)
4901         {
4902                 TreeSocket* s = (TreeSocket*)opaque;
4903                 if (target)
4904                 {
4905                         if (target_type == TYPE_USER)
4906                         {
4907                                 userrec* u = (userrec*)target;
4908                                 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" FMODE "+u->nick+" "+ConvToStr(u->age)+" "+modeline);
4909                         }
4910                         else
4911                         {
4912                                 chanrec* c = (chanrec*)target;
4913                                 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" FMODE "+c->name+" "+ConvToStr(c->age)+" "+modeline);
4914                         }
4915                 }
4916         }
4917
4918         virtual void ProtoSendMetaData(void* opaque, int target_type, void* target, const std::string &extname, const std::string &extdata)
4919         {
4920                 TreeSocket* s = (TreeSocket*)opaque;
4921                 if (target)
4922                 {
4923                         if (target_type == TYPE_USER)
4924                         {
4925                                 userrec* u = (userrec*)target;
4926                                 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA "+u->nick+" "+extname+" :"+extdata);
4927                         }
4928                         else if (target_type == TYPE_OTHER)
4929                         {
4930                                 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA * "+extname+" :"+extdata);
4931                         }
4932                         else if (target_type == TYPE_CHANNEL)
4933                         {
4934                                 chanrec* c = (chanrec*)target;
4935                                 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA "+c->name+" "+extname+" :"+extdata);
4936                         }
4937                 }
4938         }
4939
4940         virtual void OnEvent(Event* event)
4941         {
4942                 std::deque<std::string>* params = (std::deque<std::string>*)event->GetData();
4943
4944                 if (event->GetEventID() == "send_metadata")
4945                 {
4946                         if (params->size() < 3)
4947                                 return;
4948                         (*params)[2] = ":" + (*params)[2];
4949                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"METADATA",*params);
4950                 }
4951                 else if (event->GetEventID() == "send_topic")
4952                 {
4953                         if (params->size() < 2)
4954                                 return;
4955                         (*params)[1] = ":" + (*params)[1];
4956                         params->insert(params->begin() + 1,ServerInstance->Config->ServerName);
4957                         params->insert(params->begin() + 1,ConvToStr(ServerInstance->Time()));
4958                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"FTOPIC",*params);
4959                 }
4960                 else if (event->GetEventID() == "send_mode")
4961                 {
4962                         if (params->size() < 2)
4963                                 return;
4964                         // Insert the TS value of the object, either userrec or chanrec
4965                         time_t ourTS = 0;
4966                         userrec* a = ServerInstance->FindNick((*params)[0]);
4967                         if (a)
4968                         {
4969                                 ourTS = a->age;
4970                         }
4971                         else
4972                         {
4973                                 chanrec* a = ServerInstance->FindChan((*params)[0]);
4974                                 if (a)
4975                                 {
4976                                         ourTS = a->age;
4977                                 }
4978                         }
4979                         params->insert(params->begin() + 1,ConvToStr(ourTS));
4980                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"FMODE",*params);
4981                 }
4982         }
4983
4984         virtual ~ModuleSpanningTree()
4985         {
4986                 ServerInstance->Log(DEBUG,"Performing unload of spanningtree!");
4987                 /* This will also free the listeners */
4988                 delete Utils;
4989         }
4990
4991         virtual Version GetVersion()
4992         {
4993                 return Version(1,1,0,2,VF_VENDOR,API_VERSION);
4994         }
4995
4996         void Implements(char* List)
4997         {
4998                 List[I_OnPreCommand] = List[I_OnGetServerDescription] = List[I_OnUserInvite] = List[I_OnPostLocalTopicChange] = 1;
4999                 List[I_OnWallops] = List[I_OnUserNotice] = List[I_OnUserMessage] = List[I_OnBackgroundTimer] = 1;
5000                 List[I_OnUserJoin] = List[I_OnChangeHost] = List[I_OnChangeName] = List[I_OnUserPart] = List[I_OnUserConnect] = 1;
5001                 List[I_OnUserQuit] = List[I_OnUserPostNick] = List[I_OnUserKick] = List[I_OnRemoteKill] = List[I_OnRehash] = 1;
5002                 List[I_OnOper] = List[I_OnAddGLine] = List[I_OnAddZLine] = List[I_OnAddQLine] = List[I_OnAddELine] = 1;
5003                 List[I_OnDelGLine] = List[I_OnDelZLine] = List[I_OnDelQLine] = List[I_OnDelELine] = List[I_ProtoSendMode] = List[I_OnMode] = 1;
5004                 List[I_OnStats] = List[I_ProtoSendMetaData] = List[I_OnEvent] = List[I_OnSetAway] = List[I_OnCancelAway] = List[I_OnPostCommand] = 1;
5005         }
5006
5007         /* It is IMPORTANT that m_spanningtree is the last module in the chain
5008          * so that any activity it sees is FINAL, e.g. we arent going to send out
5009          * a NICK message before m_cloaking has finished putting the +x on the user,
5010          * etc etc.
5011          * Therefore, we return PRIORITY_LAST to make sure we end up at the END of
5012          * the module call queue.
5013          */
5014         Priority Prioritize()
5015         {
5016                 return PRIORITY_LAST;
5017         }
5018 };
5019
5020 void SpanningTreeUtilities::DoFailOver(Link* x)
5021 {
5022         if (x->FailOver.length())
5023         {
5024                 if (x->FailOver == x->Name)
5025                 {
5026                         ServerInstance->SNO->WriteToSnoMask('l',"FAILOVER: Some muppet configured the failover for server \002%s\002 to point at itself. Not following it!", x->Name.c_str());
5027                         return;
5028                 }
5029                 Link* TryThisOne = this->FindLink(x->FailOver.c_str());
5030                 if (TryThisOne)
5031                 {
5032                         ServerInstance->SNO->WriteToSnoMask('l',"FAILOVER: Trying failover link for \002%s\002: \002%s\002...", x->Name.c_str(), TryThisOne->Name.c_str());
5033                         Creator->ConnectServer(TryThisOne);
5034                 }
5035                 else
5036                 {
5037                         ServerInstance->SNO->WriteToSnoMask('l',"FAILOVER: Invalid failover server specified for server \002%s\002, will not follow!", x->Name.c_str());
5038                 }
5039         }
5040 }
5041
5042 Link* SpanningTreeUtilities::FindLink(const std::string& name)
5043 {
5044         for (std::vector<Link>::iterator x = LinkBlocks.begin(); x < LinkBlocks.end(); x++)
5045         {
5046                 if (ServerInstance->MatchText(x->Name.c_str(), name.c_str()))
5047                 {
5048                         return &(*x);
5049                 }
5050         }
5051         return NULL;
5052 }
5053
5054
5055 class ModuleSpanningTreeFactory : public ModuleFactory
5056 {
5057  public:
5058         ModuleSpanningTreeFactory()
5059         {
5060         }
5061
5062         ~ModuleSpanningTreeFactory()
5063         {
5064         }
5065
5066         virtual Module * CreateModule(InspIRCd* Me)
5067         {
5068                 return new ModuleSpanningTree(Me);
5069         }
5070
5071 };
5072
5073
5074 extern "C" void * init_module( void )
5075 {
5076         return new ModuleSpanningTreeFactory;
5077 }