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