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