From 60d9b675637f778266614e62e43662e77938a581 Mon Sep 17 00:00:00 2001 From: nenolod Date: Sat, 8 Apr 2006 00:20:27 +0000 Subject: [PATCH] - fork off spanningtree module for UUID stuffs :o git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@3843 e03df62e-2008-0410-955e-edbf42e46eb7 --- src/modules/m_spanningtree_ng.cpp | 3907 +++++++++++++++++++++++++++++ 1 file changed, 3907 insertions(+) create mode 100644 src/modules/m_spanningtree_ng.cpp diff --git a/src/modules/m_spanningtree_ng.cpp b/src/modules/m_spanningtree_ng.cpp new file mode 100644 index 000000000..0005f0683 --- /dev/null +++ b/src/modules/m_spanningtree_ng.cpp @@ -0,0 +1,3907 @@ +/* +------------------------------------+ + * | Inspire Internet Relay Chat Daemon | + * +------------------------------------+ + * + * InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev. + * E-mail: + * + * + * + * Written by Craig Edwards, Craig McLure, and others. + * This program is free but copyrighted software; see + * the file COPYING for details. + * + * --------------------------------------------------- + */ + +/* $ModDesc: Povides a spanning tree server link protocol */ + +using namespace std; + +#include +#include +#include +#include "globals.h" +#include "inspircd_config.h" +#ifdef GCC3 +#include +#else +#include +#endif +#include "users.h" +#include "channels.h" +#include "modules.h" +#include "commands.h" +#include "cmd_whois.h" +#include "socket.h" +#include "helperfuncs.h" +#include "inspircd.h" +#include "inspstring.h" +#include "hashcomp.h" +#include "message.h" +#include "xline.h" +#include "typedefs.h" +#include "cull_list.h" +#include "aes.h" + +#ifdef GCC3 +#define nspace __gnu_cxx +#else +#define nspace std +#endif + +/* + * The server list in InspIRCd is maintained as two structures + * which hold the data in different ways. Most of the time, we + * want to very quicky obtain three pieces of information: + * + * (1) The information on a server + * (2) The information on the server we must send data through + * to actually REACH the server we're after + * (3) Potentially, the child/parent objects of this server + * + * The InspIRCd spanning protocol provides easy access to these + * by storing the data firstly in a recursive structure, where + * each item references its parent item, and a dynamic list + * of child items, and another structure which stores the items + * hashed, linearly. This means that if we want to find a server + * by name quickly, we can look it up in the hash, avoiding + * any O(n) lookups. If however, during a split or sync, we want + * to apply an operation to a server, and any of its child objects + * we can resort to recursion to walk the tree structure. + */ + +class ModuleSpanningTree; +static ModuleSpanningTree* TreeProtocolModule; + +extern ServerConfig* Config; +extern InspIRCd* ServerInstance; +extern std::vector modules; +extern std::vector factory; +extern int MODCOUNT; + +/* Any socket can have one of five states at any one time. + * The LISTENER state indicates a socket which is listening + * for connections. It cannot receive data itself, only incoming + * sockets. + * The CONNECTING state indicates an outbound socket which is + * waiting to be writeable. + * The WAIT_AUTH_1 state indicates the socket is outbound and + * has successfully connected, but has not yet sent and received + * SERVER strings. + * The WAIT_AUTH_2 state indicates that the socket is inbound + * (allocated by a LISTENER) but has not yet sent and received + * SERVER strings. + * The CONNECTED state represents a fully authorized, fully + * connected server. + */ +enum ServerState { LISTENER, CONNECTING, WAIT_AUTH_1, WAIT_AUTH_2, CONNECTED }; + +/* We need to import these from the core for use in netbursts */ +extern user_hash clientlist; +extern chan_hash chanlist; + +/* Foward declarations */ +class TreeServer; +class TreeSocket; + +/* This variable represents the root of the server tree + * (for all intents and purposes, it's us) + */ +TreeServer *TreeRoot; + +static Server* Srv; + +/* This hash_map holds the hash equivalent of the server + * tree, used for rapid linear lookups. + */ +typedef nspace::hash_map, irc::StrHashComp> server_hash; +server_hash serverlist; + +/* More forward declarations */ +bool DoOneToOne(std::string prefix, std::string command, std::deque ¶ms, std::string target); +bool DoOneToAllButSender(std::string prefix, std::string command, std::deque ¶ms, std::string omit); +bool DoOneToMany(std::string prefix, std::string command, std::deque ¶ms); +bool DoOneToAllButSenderRaw(std::string data, std::string omit, std::string prefix, irc::string command, std::deque ¶ms); +void ReadConfiguration(bool rebind); + +/* Flatten links and /MAP for non-opers */ +bool FlatLinks; +/* Hide U-Lined servers in /MAP and /LINKS */ +bool HideULines; + +/* Imported from xline.cpp for use during netburst */ +extern std::vector klines; +extern std::vector glines; +extern std::vector zlines; +extern std::vector qlines; +extern std::vector elines; +extern std::vector pklines; +extern std::vector pglines; +extern std::vector pzlines; +extern std::vector pqlines; +extern std::vector pelines; + +/* Each server in the tree is represented by one class of + * type TreeServer. A locally connected TreeServer can + * have a class of type TreeSocket associated with it, for + * remote servers, the TreeSocket entry will be NULL. + * Each server also maintains a pointer to its parent + * (NULL if this server is ours, at the top of the tree) + * and a pointer to its "Route" (see the comments in the + * constructors below), and also a dynamic list of pointers + * to its children which can be iterated recursively + * if required. Creating or deleting objects of type + i* TreeServer automatically maintains the hash_map of + * TreeServer items, deleting and inserting them as they + * are created and destroyed. + */ + +class TreeServer +{ + TreeServer* Parent; /* Parent entry */ + TreeServer* Route; /* Route entry */ + std::vector Children; /* List of child objects */ + irc::string ServerName; /* Server's name */ + std::string ServerDesc; /* Server's description */ + std::string VersionString; /* Version string or empty string */ + int UserCount; /* Not used in this version */ + int OperCount; /* Not used in this version */ + TreeSocket* Socket; /* For directly connected servers this points at the socket object */ + time_t NextPing; /* After this time, the server should be PINGed*/ + bool LastPingWasGood; /* True if the server responded to the last PING with a PONG */ + std::map Users; /* Users on this server */ + + public: + + /* We don't use this constructor. Its a dummy, and won't cause any insertion + * of the TreeServer into the hash_map. See below for the two we DO use. + */ + TreeServer() + { + Parent = NULL; + ServerName = ""; + ServerDesc = ""; + VersionString = ""; + UserCount = OperCount = 0; + VersionString = Srv->GetVersion(); + } + + /* We use this constructor only to create the 'root' item, TreeRoot, which + * represents our own server. Therefore, it has no route, no parent, and + * no socket associated with it. Its version string is our own local version. + */ + TreeServer(std::string Name, std::string Desc) : ServerName(Name.c_str()), ServerDesc(Desc) + { + Parent = NULL; + VersionString = ""; + UserCount = OperCount = 0; + VersionString = Srv->GetVersion(); + Route = NULL; + Socket = NULL; /* Fix by brain */ + AddHashEntry(); + } + + /* When we create a new server, we call this constructor to initialize it. + * This constructor initializes the server's Route and Parent, and sets up + * its ping counters so that it will be pinged one minute from now. + */ + TreeServer(std::string Name, std::string Desc, TreeServer* Above, TreeSocket* Sock) : Parent(Above), ServerName(Name.c_str()), ServerDesc(Desc), Socket(Sock) + { + VersionString = ""; + UserCount = OperCount = 0; + this->SetNextPingTime(time(NULL) + 120); + this->SetPingFlag(); + + /* find the 'route' for this server (e.g. the one directly connected + * to the local server, which we can use to reach it) + * + * In the following example, consider we have just added a TreeServer + * class for server G on our network, of which we are server A. + * To route traffic to G (marked with a *) we must send the data to + * B (marked with a +) so this algorithm initializes the 'Route' + * value to point at whichever server traffic must be routed through + * to get here. If we were to try this algorithm with server B, + * the Route pointer would point at its own object ('this'). + * + * A + * / \ + * + B C + * / \ \ + * D E F + * / \ + * * G H + * + * We only run this algorithm when a server is created, as + * the routes remain constant while ever the server exists, and + * do not need to be re-calculated. + */ + + Route = Above; + if (Route == TreeRoot) + { + Route = this; + } + else + { + while (this->Route->GetParent() != TreeRoot) + { + this->Route = Route->GetParent(); + } + } + + /* Because recursive code is slow and takes a lot of resources, + * we store two representations of the server tree. The first + * is a recursive structure where each server references its + * children and its parent, which is used for netbursts and + * netsplits to dump the whole dataset to the other server, + * and the second is used for very fast lookups when routing + * messages and is instead a hash_map, where each item can + * be referenced by its server name. The AddHashEntry() + * call below automatically inserts each TreeServer class + * into the hash_map as it is created. There is a similar + * maintainance call in the destructor to tidy up deleted + * servers. + */ + + this->AddHashEntry(); + } + + void AddUser(userrec* user) + { + log(DEBUG,"Add user %s to server %s",user->nick,this->ServerName.c_str()); + std::map::iterator iter; + iter = Users.find(user); + if (iter == Users.end()) + Users[user] = user; + } + + void DelUser(userrec* user) + { + log(DEBUG,"Remove user %s from server %s",user->nick,this->ServerName.c_str()); + std::map::iterator iter; + iter = Users.find(user); + if (iter != Users.end()) + Users.erase(iter); + } + + int QuitUsers(const std::string &reason) + { + int x = Users.size(); + log(DEBUG,"Removing all users from server %s",this->ServerName.c_str()); + const char* reason_s = reason.c_str(); + for (std::map::iterator n = Users.begin(); n != Users.end(); n++) + { + log(DEBUG,"Kill %s",n->second->nick); + kill_link(n->second,reason_s); + } + Users.clear(); + return x; + } + + /* This method is used to add the structure to the + * hash_map for linear searches. It is only called + * by the constructors. + */ + void AddHashEntry() + { + server_hash::iterator iter; + iter = serverlist.find(this->ServerName.c_str()); + if (iter == serverlist.end()) + serverlist[this->ServerName.c_str()] = this; + } + + /* This method removes the reference to this object + * from the hash_map which is used for linear searches. + * It is only called by the default destructor. + */ + void DelHashEntry() + { + server_hash::iterator iter; + iter = serverlist.find(this->ServerName.c_str()); + if (iter != serverlist.end()) + serverlist.erase(iter); + } + + /* These accessors etc should be pretty self- + * explanitory. + */ + + TreeServer* GetRoute() + { + return Route; + } + + std::string GetName() + { + return ServerName.c_str(); + } + + std::string GetDesc() + { + return ServerDesc; + } + + std::string GetVersion() + { + return VersionString; + } + + void SetNextPingTime(time_t t) + { + this->NextPing = t; + LastPingWasGood = false; + } + + time_t NextPingTime() + { + return NextPing; + } + + bool AnsweredLastPing() + { + return LastPingWasGood; + } + + void SetPingFlag() + { + LastPingWasGood = true; + } + + int GetUserCount() + { + return UserCount; + } + + void AddUserCount() + { + UserCount++; + } + + void DelUserCount() + { + UserCount--; + } + + int GetOperCount() + { + return OperCount; + } + + TreeSocket* GetSocket() + { + return Socket; + } + + TreeServer* GetParent() + { + return Parent; + } + + void SetVersion(std::string Version) + { + VersionString = Version; + } + + unsigned int ChildCount() + { + return Children.size(); + } + + TreeServer* GetChild(unsigned int n) + { + if (n < Children.size()) + { + /* Make sure they cant request + * an out-of-range object. After + * all we know what these programmer + * types are like *grin*. + */ + return Children[n]; + } + else + { + return NULL; + } + } + + void AddChild(TreeServer* Child) + { + Children.push_back(Child); + } + + bool DelChild(TreeServer* Child) + { + for (std::vector::iterator a = Children.begin(); a < Children.end(); a++) + { + if (*a == Child) + { + Children.erase(a); + return true; + } + } + return false; + } + + /* Removes child nodes of this node, and of that node, etc etc. + * This is used during netsplits to automatically tidy up the + * server tree. It is slow, we don't use it for much else. + */ + bool Tidy() + { + bool stillchildren = true; + while (stillchildren) + { + stillchildren = false; + for (std::vector::iterator a = Children.begin(); a < Children.end(); a++) + { + TreeServer* s = (TreeServer*)*a; + s->Tidy(); + Children.erase(a); + delete s; + stillchildren = true; + break; + } + } + return true; + } + + ~TreeServer() + { + /* We'd better tidy up after ourselves, eh? */ + this->DelHashEntry(); + } +}; + +/* The Link class might as well be a struct, + * but this is C++ and we don't believe in structs (!). + * It holds the entire information of one + * tag from the main config file. We maintain a list + * of them, and populate the list on rehash/load. + */ + +class Link +{ + public: + irc::string Name; + std::string IPAddr; + int Port; + std::string SendPass; + std::string RecvPass; + unsigned long AutoConnect; + time_t NextConnectTime; + std::string EncryptionKey; + bool HiddenFromStats; +}; + +/* The usual stuff for inspircd modules, + * plus the vector of Link classes which we + * use to store the tags from the config + * file. + */ +ConfigReader *Conf; +std::vector LinkBlocks; + +/* Yay for fast searches! + * This is hundreds of times faster than recursion + * or even scanning a linked list, especially when + * there are more than a few servers to deal with. + * (read as: lots). + */ +TreeServer* FindServer(std::string ServerName) +{ + server_hash::iterator iter; + iter = serverlist.find(ServerName.c_str()); + if (iter != serverlist.end()) + { + return iter->second; + } + else + { + return NULL; + } +} + +/* Returns the locally connected server we must route a + * message through to reach server 'ServerName'. This + * only applies to one-to-one and not one-to-many routing. + * See the comments for the constructor of TreeServer + * for more details. + */ +TreeServer* BestRouteTo(std::string ServerName) +{ + if (ServerName.c_str() == TreeRoot->GetName()) + return NULL; + TreeServer* Found = FindServer(ServerName); + if (Found) + { + return Found->GetRoute(); + } + else + { + return NULL; + } +} + +/* Find the first server matching a given glob mask. + * Theres no find-using-glob method of hash_map [awwww :-(] + * so instead, we iterate over the list using an iterator + * and match each one until we get a hit. Yes its slow, + * deal with it. + */ +TreeServer* FindServerMask(std::string ServerName) +{ + for (server_hash::iterator i = serverlist.begin(); i != serverlist.end(); i++) + { + if (Srv->MatchText(i->first.c_str(),ServerName)) + return i->second; + } + return NULL; +} + +/* A convenient wrapper that returns true if a server exists */ +bool IsServer(std::string ServerName) +{ + return (FindServer(ServerName) != NULL); +} + + +class cmd_rconnect : public command_t +{ + Module* Creator; + public: + cmd_rconnect (Module* Callback) : command_t("RCONNECT", 'o', 2), Creator(Callback) + { + this->source = "m_spanningtree.so"; + } + + void Handle (char **parameters, int pcnt, userrec *user) + { + WriteServ(user->fd,"NOTICE %s :*** RCONNECT: Sending remote connect to \002%s\002 to connect server \002%s\002.",user->nick,parameters[0],parameters[1]); + /* Is this aimed at our server? */ + if (Srv->MatchText(Srv->GetServerName(),parameters[0])) + { + /* Yes, initiate the given connect */ + WriteOpers("*** Remote CONNECT from %s matching \002%s\002, connecting server \002%s\002",user->nick,parameters[0],parameters[1]); + char* para[1]; + para[0] = parameters[1]; + Creator->OnPreCommand("CONNECT", para, 1, user, true); + } + } +}; + + + +/* Every SERVER connection inbound or outbound is represented by + * an object of type TreeSocket. + * TreeSockets, being inherited from InspSocket, can be tied into + * the core socket engine, and we cn therefore receive activity events + * for them, just like activex objects on speed. (yes really, that + * is a technical term!) Each of these which relates to a locally + * connected server is assocated with it, by hooking it onto a + * TreeSocket class using its constructor. In this way, we can + * maintain a list of servers, some of which are directly connected, + * some of which are not. + */ + +class TreeSocket : public InspSocket +{ + std::string myhost; + std::string in_buffer; + ServerState LinkState; + std::string InboundServerName; + std::string InboundDescription; + int num_lost_users; + int num_lost_servers; + time_t NextPing; + bool LastPingWasGood; + bool bursting; + AES* ctx_in; + AES* ctx_out; + unsigned int keylength; + + public: + + /* Because most of the I/O gubbins are encapsulated within + * InspSocket, we just call the superclass constructor for + * most of the action, and append a few of our own values + * to it. + */ + TreeSocket(std::string host, int port, bool listening, unsigned long maxtime) + : InspSocket(host, port, listening, maxtime) + { + myhost = host; + this->LinkState = LISTENER; + this->ctx_in = NULL; + this->ctx_out = NULL; + } + + TreeSocket(std::string host, int port, bool listening, unsigned long maxtime, std::string ServerName) + : InspSocket(host, port, listening, maxtime) + { + myhost = ServerName; + this->LinkState = CONNECTING; + this->ctx_in = NULL; + this->ctx_out = NULL; + } + + /* When a listening socket gives us a new file descriptor, + * we must associate it with a socket without creating a new + * connection. This constructor is used for this purpose. + */ + TreeSocket(int newfd, char* ip) + : InspSocket(newfd, ip) + { + this->LinkState = WAIT_AUTH_1; + this->ctx_in = NULL; + this->ctx_out = NULL; + this->SendCapabilities(); + } + + ~TreeSocket() + { + if (ctx_in) + delete ctx_in; + if (ctx_out) + delete ctx_out; + } + + void InitAES(std::string key,std::string SName) + { + if (key == "") + return; + + ctx_in = new AES(); + ctx_out = new AES(); + log(DEBUG,"Initialized AES key %s",key.c_str()); + // key must be 16, 24, 32 etc bytes (multiple of 8) + keylength = key.length(); + if (!(keylength == 16 || keylength == 24 || keylength == 32)) + { + WriteOpers("*** \2ERROR\2: Key length for encryptionkey is not 16, 24 or 32 bytes in length!"); + log(DEBUG,"Key length not 16, 24 or 32 characters!"); + } + else + { + WriteOpers("*** \2AES\2: Initialized %d bit encryption to server %s",keylength*8,SName.c_str()); + 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\ + \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); + 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\ + \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); + } + } + + /* When an outbound connection finishes connecting, we receive + * this event, and must send our SERVER string to the other + * side. If the other side is happy, as outlined in the server + * to server docs on the inspircd.org site, the other side + * will then send back its own server string. + */ + virtual bool OnConnected() + { + if (this->LinkState == CONNECTING) + { + /* we do not need to change state here. */ + for (std::vector::iterator x = LinkBlocks.begin(); x < LinkBlocks.end(); x++) + { + if (x->Name == this->myhost) + { + Srv->SendOpers("*** Connection to \2"+myhost+"\2["+(x->HiddenFromStats ? "" : this->GetIP())+"] established."); + this->SendCapabilities(); + if (x->EncryptionKey != "") + { + if (!(x->EncryptionKey.length() == 16 || x->EncryptionKey.length() == 24 || x->EncryptionKey.length() == 32)) + { + WriteOpers("\2WARNING\2: Your encryption key is NOT 16, 24 or 32 characters in length, encryption will \2NOT\2 be enabled."); + } + else + { + this->WriteLine("AES "+Srv->GetServerName()); + this->InitAES(x->EncryptionKey,x->Name.c_str()); + } + } + /* found who we're supposed to be connecting to, send the neccessary gubbins. */ + this->WriteLine("SERVER "+Srv->GetServerName()+" "+x->SendPass+" 0 :"+Srv->GetServerDescription()); + return true; + } + } + } + /* There is a (remote) chance that between the /CONNECT and the connection + * being accepted, some muppet has removed the block and rehashed. + * If that happens the connection hangs here until it's closed. Unlikely + * and rather harmless. + */ + Srv->SendOpers("*** Connection to \2"+myhost+"\2 lost link tag(!)"); + return true; + } + + virtual void OnError(InspSocketError e) + { + /* We don't handle this method, because all our + * dirty work is done in OnClose() (see below) + * which is still called on error conditions too. + */ + if (e == I_ERR_CONNECT) + { + Srv->SendOpers("*** Connection failed: Connection refused"); + } + } + + virtual int OnDisconnect() + { + /* For the same reason as above, we don't + * handle OnDisconnect() + */ + return true; + } + + /* Recursively send the server tree with distances as hops. + * This is used during network burst to inform the other server + * (and any of ITS servers too) of what servers we know about. + * If at any point any of these servers already exist on the other + * end, our connection may be terminated. The hopcounts given + * by this function are relative, this doesn't matter so long as + * they are all >1, as all the remote servers re-calculate them + * to be relative too, with themselves as hop 0. + */ + void SendServers(TreeServer* Current, TreeServer* s, int hops) + { + char command[1024]; + for (unsigned int q = 0; q < Current->ChildCount(); q++) + { + TreeServer* recursive_server = Current->GetChild(q); + if (recursive_server != s) + { + snprintf(command,1024,":%s SERVER %s * %d :%s",Current->GetName().c_str(),recursive_server->GetName().c_str(),hops,recursive_server->GetDesc().c_str()); + this->WriteLine(command); + this->WriteLine(":"+recursive_server->GetName()+" VERSION :"+recursive_server->GetVersion()); + /* down to next level */ + this->SendServers(recursive_server, s, hops+1); + } + } + } + + std::string MyCapabilities() + { + ServerConfig* Config = Srv->GetConfig(); + std::vector modlist; + std::string capabilities = ""; + + for (int i = 0; i <= MODCOUNT; i++) + { + if ((modules[i]->GetVersion().Flags & VF_STATIC) || (modules[i]->GetVersion().Flags & VF_COMMON)) + modlist.push_back(Config->module_names[i]); + } + sort(modlist.begin(),modlist.end()); + for (unsigned int i = 0; i < modlist.size(); i++) + { + if (i) + capabilities = capabilities + ","; + capabilities = capabilities + modlist[i]; + } + return capabilities; + } + + void SendCapabilities() + { + this->WriteLine("CAPAB "+MyCapabilities()); + } + + bool Capab(std::deque params) + { + if (params.size() != 1) + { + this->WriteLine("ERROR :Invalid number of parameters for CAPAB"); + return false; + } + + if (params[0] != this->MyCapabilities()) + { + std::string quitserver = this->myhost; + if (this->InboundServerName != "") + { + quitserver = this->InboundServerName; + } + + WriteOpers("*** \2ERROR\2: Server '%s' does not have the same set of modules loaded, cannot link!",quitserver.c_str()); + WriteOpers("*** Our networked module set is: '%s'",this->MyCapabilities().c_str()); + WriteOpers("*** Other server's networked module set is: '%s'",params[0].c_str()); + WriteOpers("*** These lists must match exactly on both servers. Please correct these errors, and try again."); + this->WriteLine("ERROR :CAPAB mismatch; My capabilities: '"+this->MyCapabilities()+"'"); + return false; + } + + return true; + } + + /* This function forces this server to quit, removing this server + * and any users on it (and servers and users below that, etc etc). + * It's very slow and pretty clunky, but luckily unless your network + * is having a REAL bad hair day, this function shouldnt be called + * too many times a month ;-) + */ + void SquitServer(std::string &from, TreeServer* Current) + { + /* recursively squit the servers attached to 'Current'. + * We're going backwards so we don't remove users + * while we still need them ;) + */ + for (unsigned int q = 0; q < Current->ChildCount(); q++) + { + TreeServer* recursive_server = Current->GetChild(q); + this->SquitServer(from,recursive_server); + } + /* Now we've whacked the kids, whack self */ + num_lost_servers++; + num_lost_users += Current->QuitUsers(from); + /*for (user_hash::iterator u = clientlist.begin(); u != clientlist.end(); u++) + { + if (!strcasecmp(u->second->server,Current->GetName().c_str())) + { + Goners->AddItem(u->second,from); + num_lost_users++; + } + }*/ + } + + /* This is a wrapper function for SquitServer above, which + * does some validation first and passes on the SQUIT to all + * other remaining servers. + */ + void Squit(TreeServer* Current,std::string reason) + { + if ((Current) && (Current != TreeRoot)) + { + std::deque params; + params.push_back(Current->GetName()); + params.push_back(":"+reason); + DoOneToAllButSender(Current->GetParent()->GetName(),"SQUIT",params,Current->GetName()); + if (Current->GetParent() == TreeRoot) + { + Srv->SendOpers("Server \002"+Current->GetName()+"\002 split: "+reason); + } + else + { + Srv->SendOpers("Server \002"+Current->GetName()+"\002 split from server \002"+Current->GetParent()->GetName()+"\002 with reason: "+reason); + } + num_lost_servers = 0; + num_lost_users = 0; + std::string from = Current->GetParent()->GetName()+" "+Current->GetName(); + SquitServer(from, Current); + Current->Tidy(); + Current->GetParent()->DelChild(Current); + delete Current; + WriteOpers("Netsplit complete, lost \002%d\002 users on \002%d\002 servers.", num_lost_users, num_lost_servers); + } + else + { + log(DEFAULT,"Squit from unknown server"); + } + } + + /* FMODE command */ + bool ForceMode(std::string source, std::deque ¶ms) + { + if (params.size() < 2) + return true; + userrec* who = new userrec(); + who->fd = FD_MAGIC_NUMBER; + char* modelist[64]; + memset(&modelist,0,sizeof(modelist)); + for (unsigned int q = 0; q < params.size(); q++) + { + modelist[q] = (char*)params[q].c_str(); + } + Srv->SendMode(modelist,params.size(),who); + DoOneToAllButSender(source,"FMODE",params,source); + delete who; + return true; + } + + /* FTOPIC command */ + bool ForceTopic(std::string source, std::deque ¶ms) + { + if (params.size() != 4) + return true; + time_t ts = atoi(params[1].c_str()); + std::string nsource = source; + + chanrec* c = Srv->FindChannel(params[0]); + if (c) + { + if ((ts >= c->topicset) || (!*c->topic)) + { + std::string oldtopic = c->topic; + strlcpy(c->topic,params[3].c_str(),MAXTOPIC); + strlcpy(c->setby,params[2].c_str(),NICKMAX-1); + c->topicset = ts; + /* if the topic text is the same as the current topic, + * dont bother to send the TOPIC command out, just silently + * update the set time and set nick. + */ + if (oldtopic != params[3]) + { + userrec* user = Srv->FindNick(source); + if (!user) + { + WriteChannelWithServ((char*)source.c_str(), c, "TOPIC %s :%s", c->name, c->topic); + } + else + { + WriteChannel(c, user, "TOPIC %s :%s", c->name, c->topic); + nsource = user->server; + } + /* all done, send it on its way */ + params[3] = ":" + params[3]; + DoOneToAllButSender(source,"FTOPIC",params,nsource); + } + } + + } + + return true; + } + + /* FJOIN, similar to unreal SJOIN */ + bool ForceJoin(std::string source, std::deque ¶ms) + { + if (params.size() < 3) + return true; + + char first[MAXBUF]; + char modestring[MAXBUF]; + char* mode_users[127]; + memset(&mode_users,0,sizeof(mode_users)); + mode_users[0] = first; + mode_users[1] = modestring; + strcpy(mode_users[1],"+"); + unsigned int modectr = 2; + + userrec* who = NULL; + std::string channel = params[0]; + time_t TS = atoi(params[1].c_str()); + char* key = ""; + + chanrec* chan = Srv->FindChannel(channel); + if (chan) + { + key = chan->key; + } + strlcpy(mode_users[0],channel.c_str(),MAXBUF); + + /* default is a high value, which if we dont have this + * channel will let the other side apply their modes. + */ + time_t ourTS = time(NULL)+600; + chanrec* us = Srv->FindChannel(channel); + if (us) + { + ourTS = us->age; + } + + log(DEBUG,"FJOIN detected, our TS=%lu, their TS=%lu",ourTS,TS); + + /* do this first, so our mode reversals are correctly received by other servers + * if there is a TS collision. + */ + DoOneToAllButSender(source,"FJOIN",params,source); + + for (unsigned int usernum = 2; usernum < params.size(); usernum++) + { + /* process one channel at a time, applying modes. */ + char* usr = (char*)params[usernum].c_str(); + /* Safety check just to make sure someones not sent us an FJOIN full of spaces + * (is this even possible?) */ + if (usr && *usr) + { + char permissions = *usr; + switch (permissions) + { + case '@': + usr++; + mode_users[modectr++] = usr; + strlcat(modestring,"o",MAXBUF); + break; + case '%': + usr++; + mode_users[modectr++] = usr; + strlcat(modestring,"h",MAXBUF); + break; + case '+': + usr++; + mode_users[modectr++] = usr; + strlcat(modestring,"v",MAXBUF); + break; + } + who = Srv->FindNick(usr); + if (who) + { + Srv->JoinUserToChannel(who,channel,key); + if (modectr >= (MAXMODES-1)) + { + /* theres a mode for this user. push them onto the mode queue, and flush it + * if there are more than MAXMODES to go. + */ + if ((ourTS >= TS) || (Srv->IsUlined(who->server))) + { + /* We also always let u-lined clients win, no matter what the TS value */ + log(DEBUG,"Our our channel newer than theirs, accepting their modes"); + Srv->SendMode(mode_users,modectr,who); + } + else + { + log(DEBUG,"Their channel newer than ours, bouncing their modes"); + /* bouncy bouncy! */ + std::deque params; + /* modes are now being UNSET... */ + *mode_users[1] = '-'; + for (unsigned int x = 0; x < modectr; x++) + { + params.push_back(mode_users[x]); + } + // tell everyone to bounce the modes. bad modes, bad! + DoOneToMany(Srv->GetServerName(),"FMODE",params); + } + strcpy(mode_users[1],"+"); + modectr = 2; + } + } + } + } + /* there werent enough modes built up to flush it during FJOIN, + * or, there are a number left over. flush them out. + */ + if ((modectr > 2) && (who)) + { + if (ourTS >= TS) + { + log(DEBUG,"Our our channel newer than theirs, accepting their modes"); + Srv->SendMode(mode_users,modectr,who); + } + else + { + log(DEBUG,"Their channel newer than ours, bouncing their modes"); + std::deque params; + *mode_users[1] = '-'; + for (unsigned int x = 0; x < modectr; x++) + { + params.push_back(mode_users[x]); + } + DoOneToMany(Srv->GetServerName(),"FMODE",params); + } + } + return true; + } + + /* NICK command */ + bool IntroduceClient(std::string source, std::deque ¶ms) + { + if (params.size() < 8) + return true; + if (params.size() > 8) + { + this->WriteLine(":"+Srv->GetServerName()+" KILL "+params[1]+" :Invalid client introduction ("+params[1]+"?)"); + return true; + } + // NICK age nick host dhost ident +modes ip :gecos + // 0 123 4 56 7 + time_t age = atoi(params[0].c_str()); + std::string modes = params[5]; + while (*(modes.c_str()) == '+') + { + char* m = (char*)modes.c_str(); + m++; + modes = m; + } + char* tempnick = (char*)params[1].c_str(); + log(DEBUG,"Introduce client %s!%s@%s",tempnick,params[4].c_str(),params[2].c_str()); + + user_hash::iterator iter; + iter = clientlist.find(tempnick); + if (iter != clientlist.end()) + { + // nick collision + 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); + this->WriteLine(":"+Srv->GetServerName()+" KILL "+tempnick+" :Nickname collision"); + return true; + } + + clientlist[tempnick] = new userrec(); + clientlist[tempnick]->fd = FD_MAGIC_NUMBER; + strlcpy(clientlist[tempnick]->nick, tempnick,NICKMAX-1); + strlcpy(clientlist[tempnick]->host, params[2].c_str(),63); + strlcpy(clientlist[tempnick]->dhost, params[3].c_str(),63); + clientlist[tempnick]->server = (char*)FindServerNamePtr(source.c_str()); + strlcpy(clientlist[tempnick]->ident, params[4].c_str(),IDENTMAX); + strlcpy(clientlist[tempnick]->fullname, params[7].c_str(),MAXGECOS); + clientlist[tempnick]->registered = 7; + clientlist[tempnick]->signon = age; + strlcpy(clientlist[tempnick]->modes, modes.c_str(),53); + for (char *v = clientlist[tempnick]->modes; *v; v++) + { + switch (*v) + { + case 'i': + clientlist[tempnick]->modebits |= UM_INVISIBLE; + break; + case 'w': + clientlist[tempnick]->modebits |= UM_WALLOPS; + break; + case 's': + clientlist[tempnick]->modebits |= UM_SERVERNOTICE; + break; + default: + break; + } + } + inet_aton(params[6].c_str(),&clientlist[tempnick]->ip4); + + WriteOpers("*** Client connecting at %s: %s!%s@%s [%s]",clientlist[tempnick]->server,clientlist[tempnick]->nick,clientlist[tempnick]->ident,clientlist[tempnick]->host,(char*)inet_ntoa(clientlist[tempnick]->ip4)); + + params[7] = ":" + params[7]; + DoOneToAllButSender(source,"NICK",params,source); + + // Increment the Source Servers User Count.. + TreeServer* SourceServer = FindServer(source); + if (SourceServer) + { + SourceServer->AddUser(clientlist[tempnick]); + SourceServer->AddUserCount(); + } + + return true; + } + + /* Send one or more FJOINs for a channel of users. + * If the length of a single line is more than 480-NICKMAX + * in length, it is split over multiple lines. + */ + void SendFJoins(TreeServer* Current, chanrec* c) + { + log(DEBUG,"Sending FJOINs to other server for %s",c->name); + char list[MAXBUF]; + std::string individual_halfops = ":"+Srv->GetServerName()+" FMODE "+c->name; + + size_t dlen, curlen; + dlen = curlen = snprintf(list,MAXBUF,":%s FJOIN %s %lu",Srv->GetServerName().c_str(),c->name,(unsigned long)c->age); + int numusers = 0; + char* ptr = list + dlen; + + CUList *ulist = c->GetUsers(); + std::vector specific_halfop; + std::vector specific_voice; + + for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++) + { + int x = cflags(i->second,c); + if ((x & UCMODE_HOP) && (x & UCMODE_OP)) + { + specific_halfop.push_back(i->second); + } + if (((x & UCMODE_HOP) || (x & UCMODE_OP)) && (x & UCMODE_VOICE)) + { + specific_voice.push_back(i->second); + } + + const char* n = ""; + if (x & UCMODE_OP) + { + n = "@"; + } + else if (x & UCMODE_HOP) + { + n = "%"; + } + else if (x & UCMODE_VOICE) + { + n = "+"; + } + + size_t ptrlen = snprintf(ptr, MAXBUF, " %s%s", n, i->second->nick); + + curlen += ptrlen; + ptr += ptrlen; + + numusers++; + + if (curlen > (480-NICKMAX)) + { + this->WriteLine(list); + dlen = curlen = snprintf(list,MAXBUF,":%s FJOIN %s %lu",Srv->GetServerName().c_str(),c->name,(unsigned long)c->age); + ptr = list + dlen; + ptrlen = 0; + numusers = 0; + for (unsigned int y = 0; y < specific_voice.size(); y++) + { + this->WriteLine(":"+Srv->GetServerName()+" FMODE "+c->name+" +v "+specific_voice[y]->nick); + } + for (unsigned int y = 0; y < specific_halfop.size(); y++) + { + this->WriteLine(":"+Srv->GetServerName()+" FMODE "+c->name+" +h "+specific_halfop[y]->nick); + } + } + } + if (numusers) + { + this->WriteLine(list); + for (unsigned int y = 0; y < specific_voice.size(); y++) + { + this->WriteLine(":"+Srv->GetServerName()+" FMODE "+c->name+" +v "+specific_voice[y]->nick); + } + for (unsigned int y = 0; y < specific_halfop.size(); y++) + { + this->WriteLine(":"+Srv->GetServerName()+" FMODE "+c->name+" +h "+specific_halfop[y]->nick); + } + } + } + + /* Send G, Q, Z and E lines */ + void SendXLines(TreeServer* Current) + { + char data[MAXBUF]; + std::string n = Srv->GetServerName(); + const char* sn = n.c_str(); + int iterations = 0; + /* Yes, these arent too nice looking, but they get the job done */ + for (std::vector::iterator i = zlines.begin(); i != zlines.end(); i++, iterations++) + { + 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); + this->WriteLine(data); + } + for (std::vector::iterator i = qlines.begin(); i != qlines.end(); i++, iterations++) + { + 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); + this->WriteLine(data); + } + for (std::vector::iterator i = glines.begin(); i != glines.end(); i++, iterations++) + { + 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); + this->WriteLine(data); + } + for (std::vector::iterator i = elines.begin(); i != elines.end(); i++, iterations++) + { + 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); + this->WriteLine(data); + } + for (std::vector::iterator i = pzlines.begin(); i != pzlines.end(); i++, iterations++) + { + 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); + this->WriteLine(data); + } + for (std::vector::iterator i = pqlines.begin(); i != pqlines.end(); i++, iterations++) + { + 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); + this->WriteLine(data); + } + for (std::vector::iterator i = pglines.begin(); i != pglines.end(); i++, iterations++) + { + 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); + this->WriteLine(data); + } + for (std::vector::iterator i = pelines.begin(); i != pelines.end(); i++, iterations++) + { + 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); + this->WriteLine(data); + } + } + + /* Send channel modes and topics */ + void SendChannelModes(TreeServer* Current) + { + char data[MAXBUF]; + std::deque list; + int iterations = 0; + std::string n = Srv->GetServerName(); + const char* sn = n.c_str(); + for (chan_hash::iterator c = chanlist.begin(); c != chanlist.end(); c++, iterations++) + { + SendFJoins(Current, c->second); + snprintf(data,MAXBUF,":%s FMODE %s +%s",sn,c->second->name,chanmodes(c->second,true)); + this->WriteLine(data); + if (*c->second->topic) + { + snprintf(data,MAXBUF,":%s FTOPIC %s %lu %s :%s",sn,c->second->name,(unsigned long)c->second->topicset,c->second->setby,c->second->topic); + this->WriteLine(data); + } + for (BanList::iterator b = c->second->bans.begin(); b != c->second->bans.end(); b++) + { + snprintf(data,MAXBUF,":%s FMODE %s +b %s",sn,c->second->name,b->data); + this->WriteLine(data); + } + FOREACH_MOD(I_OnSyncChannel,OnSyncChannel(c->second,(Module*)TreeProtocolModule,(void*)this)); + list.clear(); + c->second->GetExtList(list); + for (unsigned int j = 0; j < list.size(); j++) + { + FOREACH_MOD(I_OnSyncChannelMetaData,OnSyncChannelMetaData(c->second,(Module*)TreeProtocolModule,(void*)this,list[j])); + } + } + } + + /* send all users and their oper state/modes */ + void SendUsers(TreeServer* Current) + { + char data[MAXBUF]; + std::deque list; + int iterations = 0; + for (user_hash::iterator u = clientlist.begin(); u != clientlist.end(); u++, iterations++) + { + if (u->second->registered == 7) + { + 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->modes,(char*)inet_ntoa(u->second->ip4),u->second->fullname); + this->WriteLine(data); + if (*u->second->oper) + { + this->WriteLine(":"+std::string(u->second->nick)+" OPERTYPE "+std::string(u->second->oper)); + } + if (*u->second->awaymsg) + { + this->WriteLine(":"+std::string(u->second->nick)+" AWAY :"+std::string(u->second->awaymsg)); + } + FOREACH_MOD(I_OnSyncUser,OnSyncUser(u->second,(Module*)TreeProtocolModule,(void*)this)); + list.clear(); + u->second->GetExtList(list); + for (unsigned int j = 0; j < list.size(); j++) + { + FOREACH_MOD(I_OnSyncUserMetaData,OnSyncUserMetaData(u->second,(Module*)TreeProtocolModule,(void*)this,list[j])); + } + } + } + } + + /* This function is called when we want to send a netburst to a local + * server. There is a set order we must do this, because for example + * users require their servers to exist, and channels require their + * users to exist. You get the idea. + */ + void DoBurst(TreeServer* s) + { + /* The calls here to ServerInstance-> yield the processing + * back to the core so that a large burst is split into at least 6 sections + * (possibly more) + */ + std::string burst = "BURST "+ConvToStr(time(NULL)); + std::string endburst = "ENDBURST"; + // Because by the end of the netburst, it could be gone! + std::string name = s->GetName(); + Srv->SendOpers("*** Bursting to \2"+name+"\2."); + this->WriteLine(burst); + /* send our version string */ + this->WriteLine(":"+Srv->GetServerName()+" VERSION :"+Srv->GetVersion()); + /* Send server tree */ + this->SendServers(TreeRoot,s,1); + /* Send users and their oper status */ + this->SendUsers(s); + /* Send everything else (channel modes, xlines etc) */ + this->SendChannelModes(s); + this->SendXLines(s); + FOREACH_MOD(I_OnSyncOtherMetaData,OnSyncOtherMetaData((Module*)TreeProtocolModule,(void*)this)); + this->WriteLine(endburst); + Srv->SendOpers("*** Finished bursting to \2"+name+"\2."); + } + + /* This function is called when we receive data from a remote + * server. We buffer the data in a std::string (it doesnt stay + * there for long), reading using InspSocket::Read() which can + * read up to 16 kilobytes in one operation. + * + * IF THIS FUNCTION RETURNS FALSE, THE CORE CLOSES AND DELETES + * THE SOCKET OBJECT FOR US. + */ + virtual bool OnDataReady() + { + char* data = this->Read(); + /* Check that the data read is a valid pointer and it has some content */ + if (data && *data) + { + this->in_buffer.append(data); + /* While there is at least one new line in the buffer, + * do something useful (we hope!) with it. + */ + while (in_buffer.find("\n") != std::string::npos) + { + std::string ret = in_buffer.substr(0,in_buffer.find("\n")-1); + in_buffer = in_buffer.substr(in_buffer.find("\n")+1,in_buffer.length()-in_buffer.find("\n")); + if (ret.find("\r") != std::string::npos) + ret = in_buffer.substr(0,in_buffer.find("\r")-1); + /* Process this one, abort if it + * didnt return true. + */ + if (this->ctx_in) + { + char out[1024]; + char result[1024]; + memset(result,0,1024); + memset(out,0,1024); + log(DEBUG,"Original string '%s'",ret.c_str()); + /* ERROR + CAPAB is still allowed unencryped */ + if ((ret.substr(0,7) != "ERROR :") && (ret.substr(0,6) != "CAPAB ")) + { + int nbytes = from64tobits(out, ret.c_str(), 1024); + if ((nbytes > 0) && (nbytes < 1024)) + { + log(DEBUG,"m_spanningtree: decrypt %d bytes",nbytes); + ctx_in->Decrypt(out, result, nbytes, 0); + for (int t = 0; t < nbytes; t++) + if (result[t] == '\7') result[t] = 0; + ret = result; + } + } + } + if (!this->ProcessLine(ret)) + { + log(DEBUG,"ProcessLine says no!"); + return false; + } + } + return true; + } + /* EAGAIN returns an empty but non-NULL string, so this + * evaluates to TRUE for EAGAIN but to FALSE for EOF. + */ + return (data && !*data); + } + + int WriteLine(std::string line) + { + log(DEBUG,"OUT: %s",line.c_str()); + if (this->ctx_out) + { + char result[10240]; + char result64[10240]; + if (this->keylength) + { + // pad it to the key length + int n = this->keylength - (line.length() % this->keylength); + if (n) + { + 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); + line.append(n,'\7'); + } + } + unsigned int ll = line.length(); + ctx_out->Encrypt(line.c_str(), result, ll, 0); + to64frombits((unsigned char*)result64,(unsigned char*)result,ll); + line = result64; + //int from64tobits(char *out, const char *in, int maxlen); + } + return this->Write(line + "\r\n"); + } + + /* Handle ERROR command */ + bool Error(std::deque ¶ms) + { + if (params.size() < 1) + return false; + WriteOpers("*** ERROR from %s: %s",(InboundServerName != "" ? InboundServerName.c_str() : myhost.c_str()),params[0].c_str()); + /* we will return false to cause the socket to close. */ + return false; + } + + /* Because the core won't let users or even SERVERS set +o, + * we use the OPERTYPE command to do this. + */ + bool OperType(std::string prefix, std::deque ¶ms) + { + if (params.size() != 1) + { + log(DEBUG,"Received invalid oper type from %s",prefix.c_str()); + return true; + } + std::string opertype = params[0]; + userrec* u = Srv->FindNick(prefix); + if (u) + { + strlcpy(u->oper,opertype.c_str(),NICKMAX-1); + if (!strchr(u->modes,'o')) + { + strcat(u->modes,"o"); + } + DoOneToAllButSender(u->nick,"OPERTYPE",params,u->server); + } + return true; + } + + /* Because Andy insists that services-compatible servers must + * implement SVSNICK and SVSJOIN, that's exactly what we do :p + */ + bool ForceNick(std::string prefix, std::deque ¶ms) + { + if (params.size() < 3) + return true; + + userrec* u = Srv->FindNick(params[0]); + + if (u) + { + DoOneToAllButSender(prefix,"SVSNICK",params,prefix); + if (IS_LOCAL(u)) + { + std::deque par; + par.push_back(params[1]); + DoOneToMany(u->nick,"NICK",par); + Srv->ChangeUserNick(u,params[1]); + u->age = atoi(params[2].c_str()); + } + } + return true; + } + + bool ServiceJoin(std::string prefix, std::deque ¶ms) + { + if (params.size() < 2) + return true; + + userrec* u = Srv->FindNick(params[0]); + + if (u) + { + Srv->JoinUserToChannel(u,params[1],""); + DoOneToAllButSender(prefix,"SVSJOIN",params,prefix); + } + return true; + } + + bool RemoteRehash(std::string prefix, std::deque ¶ms) + { + if (params.size() < 1) + return false; + + std::string servermask = params[0]; + + if (Srv->MatchText(Srv->GetServerName(),servermask)) + { + Srv->SendOpers("*** Remote rehash initiated from server \002"+prefix+"\002."); + Srv->RehashServer(); + ReadConfiguration(false); + } + DoOneToAllButSender(prefix,"REHASH",params,prefix); + return true; + } + + bool RemoteKill(std::string prefix, std::deque ¶ms) + { + if (params.size() != 2) + return true; + + std::string nick = params[0]; + userrec* u = Srv->FindNick(prefix); + userrec* who = Srv->FindNick(nick); + + if (who) + { + /* Prepend kill source, if we don't have one */ + std::string sourceserv = prefix; + if (u) + { + sourceserv = u->server; + } + if (*(params[1].c_str()) != '[') + { + params[1] = "[" + sourceserv + "] Killed (" + params[1] +")"; + } + std::string reason = params[1]; + params[1] = ":" + params[1]; + DoOneToAllButSender(prefix,"KILL",params,sourceserv); + Srv->QuitUser(who,reason); + } + return true; + } + + bool LocalPong(std::string prefix, std::deque ¶ms) + { + if (params.size() < 1) + return true; + + if (params.size() == 1) + { + TreeServer* ServerSource = FindServer(prefix); + if (ServerSource) + { + ServerSource->SetPingFlag(); + } + } + else + { + std::string forwardto = params[1]; + if (forwardto == Srv->GetServerName()) + { + /* + * this is a PONG for us + * if the prefix is a user, check theyre local, and if they are, + * dump the PONG reply back to their fd. If its a server, do nowt. + * Services might want to send these s->s, but we dont need to yet. + */ + userrec* u = Srv->FindNick(prefix); + + if (u) + { + WriteServ(u->fd,"PONG %s %s",params[0].c_str(),params[1].c_str()); + } + } + else + { + // not for us, pass it on :) + DoOneToOne(prefix,"PONG",params,forwardto); + } + } + + return true; + } + + bool MetaData(std::string prefix, std::deque ¶ms) + { + if (params.size() < 3) + return true; + + TreeServer* ServerSource = FindServer(prefix); + + if (ServerSource) + { + if (params[0] == "*") + { + FOREACH_MOD(I_OnDecodeMetaData,OnDecodeMetaData(TYPE_OTHER,NULL,params[1],params[2])); + } + else if (*(params[0].c_str()) == '#') + { + chanrec* c = Srv->FindChannel(params[0]); + if (c) + { + FOREACH_MOD(I_OnDecodeMetaData,OnDecodeMetaData(TYPE_CHANNEL,c,params[1],params[2])); + } + } + else if (*(params[0].c_str()) != '#') + { + userrec* u = Srv->FindNick(params[0]); + if (u) + { + FOREACH_MOD(I_OnDecodeMetaData,OnDecodeMetaData(TYPE_USER,u,params[1],params[2])); + } + } + } + + params[2] = ":" + params[2]; + DoOneToAllButSender(prefix,"METADATA",params,prefix); + return true; + } + + bool ServerVersion(std::string prefix, std::deque ¶ms) + { + if (params.size() < 1) + return true; + + TreeServer* ServerSource = FindServer(prefix); + + if (ServerSource) + { + ServerSource->SetVersion(params[0]); + } + params[0] = ":" + params[0]; + DoOneToAllButSender(prefix,"VERSION",params,prefix); + return true; + } + + bool ChangeHost(std::string prefix, std::deque ¶ms) + { + if (params.size() < 1) + return true; + + userrec* u = Srv->FindNick(prefix); + + if (u) + { + Srv->ChangeHost(u,params[0]); + DoOneToAllButSender(prefix,"FHOST",params,u->server); + } + return true; + } + + bool AddLine(std::string prefix, std::deque ¶ms) + { + if (params.size() < 6) + return true; + + bool propogate = false; + + switch (*(params[0].c_str())) + { + case 'Z': + propogate = add_zline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str()); + zline_set_creation_time((char*)params[1].c_str(), atoi(params[3].c_str())); + break; + case 'Q': + propogate = add_qline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str()); + qline_set_creation_time((char*)params[1].c_str(), atoi(params[3].c_str())); + break; + case 'E': + propogate = add_eline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str()); + eline_set_creation_time((char*)params[1].c_str(), atoi(params[3].c_str())); + break; + case 'G': + propogate = add_gline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str()); + gline_set_creation_time((char*)params[1].c_str(), atoi(params[3].c_str())); + break; + case 'K': + propogate = add_kline(atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str()); + break; + default: + /* Just in case... */ + Srv->SendOpers("*** \2WARNING\2: Invalid xline type '"+params[0]+"' sent by server "+prefix+", ignored!"); + propogate = false; + break; + } + + /* Send it on its way */ + if (propogate) + { + if (atoi(params[4].c_str())) + { + 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()); + } + else + { + WriteOpers("*** %s Added permenant %cLINE on %s (%s).",prefix.c_str(),*(params[0].c_str()),params[1].c_str(),params[5].c_str()); + } + params[5] = ":" + params[5]; + DoOneToAllButSender(prefix,"ADDLINE",params,prefix); + } + if (!this->bursting) + { + log(DEBUG,"Applying lines..."); + apply_lines(APPLY_ZLINES|APPLY_GLINES|APPLY_QLINES); + } + return true; + } + + bool ChangeName(std::string prefix, std::deque ¶ms) + { + if (params.size() < 1) + return true; + + userrec* u = Srv->FindNick(prefix); + + if (u) + { + Srv->ChangeGECOS(u,params[0]); + params[0] = ":" + params[0]; + DoOneToAllButSender(prefix,"FNAME",params,u->server); + } + return true; + } + + bool Whois(std::string prefix, std::deque ¶ms) + { + if (params.size() < 1) + return true; + + log(DEBUG,"In IDLE command"); + userrec* u = Srv->FindNick(prefix); + + if (u) + { + log(DEBUG,"USER EXISTS: %s",u->nick); + // an incoming request + if (params.size() == 1) + { + userrec* x = Srv->FindNick(params[0]); + if ((x) && (x->fd > -1)) + { + userrec* x = Srv->FindNick(params[0]); + log(DEBUG,"Got IDLE"); + char signon[MAXBUF]; + char idle[MAXBUF]; + log(DEBUG,"Sending back IDLE 3"); + snprintf(signon,MAXBUF,"%lu",(unsigned long)x->signon); + snprintf(idle,MAXBUF,"%lu",(unsigned long)abs((x->idle_lastmsg)-time(NULL))); + std::deque par; + par.push_back(prefix); + par.push_back(signon); + par.push_back(idle); + // ours, we're done, pass it BACK + DoOneToOne(params[0],"IDLE",par,u->server); + } + else + { + // not ours pass it on + DoOneToOne(prefix,"IDLE",params,x->server); + } + } + else if (params.size() == 3) + { + std::string who_did_the_whois = params[0]; + userrec* who_to_send_to = Srv->FindNick(who_did_the_whois); + if ((who_to_send_to) && (who_to_send_to->fd > -1)) + { + log(DEBUG,"Got final IDLE"); + // an incoming reply to a whois we sent out + std::string nick_whoised = prefix; + unsigned long signon = atoi(params[1].c_str()); + unsigned long idle = atoi(params[2].c_str()); + if ((who_to_send_to) && (who_to_send_to->fd > -1)) + do_whois(who_to_send_to,u,signon,idle,(char*)nick_whoised.c_str()); + } + else + { + // not ours, pass it on + DoOneToOne(prefix,"IDLE",params,who_to_send_to->server); + } + } + } + return true; + } + + bool Push(std::string prefix, std::deque ¶ms) + { + if (params.size() < 2) + return true; + + userrec* u = Srv->FindNick(params[0]); + + if (IS_LOCAL(u)) + { + // push the raw to the user + if (Srv->IsUlined(prefix)) + { + ::Write(u->fd,"%s",params[1].c_str()); + } + else + { + log(DEBUG,"PUSH from non-ulined server dropped into the bit-bucket: :%s PUSH %s :%s",prefix.c_str(),params[0].c_str(),params[1].c_str()); + } + } + else + { + // continue the raw onwards + params[1] = ":" + params[1]; + DoOneToOne(prefix,"PUSH",params,u->server); + } + return true; + } + + bool Time(std::string prefix, std::deque ¶ms) + { + // :source.server TIME remote.server sendernick + // :remote.server TIME source.server sendernick TS + if (params.size() == 2) + { + // someone querying our time? + if (Srv->GetServerName() == params[0]) + { + userrec* u = Srv->FindNick(params[1]); + if (u) + { + char curtime[256]; + snprintf(curtime,256,"%lu",(unsigned long)time(NULL)); + params.push_back(curtime); + params[0] = prefix; + DoOneToOne(Srv->GetServerName(),"TIME",params,params[0]); + } + } + else + { + // not us, pass it on + userrec* u = Srv->FindNick(params[1]); + if (u) + DoOneToOne(prefix,"TIME",params,params[0]); + } + } + else if (params.size() == 3) + { + // a response to a previous TIME + userrec* u = Srv->FindNick(params[1]); + if ((u) && (IS_LOCAL(u))) + { + time_t rawtime = atol(params[2].c_str()); + struct tm * timeinfo; + timeinfo = localtime(&rawtime); + char tms[26]; + snprintf(tms,26,"%s",asctime(timeinfo)); + tms[24] = 0; + WriteServ(u->fd,"391 %s %s :%s",u->nick,prefix.c_str(),tms); + } + else + { + if (u) + DoOneToOne(prefix,"TIME",params,u->server); + } + } + return true; + } + + bool LocalPing(std::string prefix, std::deque ¶ms) + { + if (params.size() < 1) + return true; + + if (params.size() == 1) + { + std::string stufftobounce = params[0]; + this->WriteLine(":"+Srv->GetServerName()+" PONG "+stufftobounce); + return true; + } + else + { + std::string forwardto = params[1]; + if (forwardto == Srv->GetServerName()) + { + // this is a ping for us, send back PONG to the requesting server + params[1] = params[0]; + params[0] = forwardto; + DoOneToOne(forwardto,"PONG",params,params[1]); + } + else + { + // not for us, pass it on :) + DoOneToOne(prefix,"PING",params,forwardto); + } + return true; + } + } + + bool RemoteServer(std::string prefix, std::deque ¶ms) + { + if (params.size() < 4) + return false; + + std::string servername = params[0]; + std::string password = params[1]; + // hopcount is not used for a remote server, we calculate this ourselves + std::string description = params[3]; + TreeServer* ParentOfThis = FindServer(prefix); + + if (!ParentOfThis) + { + this->WriteLine("ERROR :Protocol error - Introduced remote server from unknown server "+prefix); + return false; + } + TreeServer* CheckDupe = FindServer(servername); + if (CheckDupe) + { + this->WriteLine("ERROR :Server "+servername+" already exists on server "+CheckDupe->GetParent()->GetName()+"!"); + Srv->SendOpers("*** Server connection from \2"+servername+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName()); + return false; + } + TreeServer* Node = new TreeServer(servername,description,ParentOfThis,NULL); + ParentOfThis->AddChild(Node); + params[3] = ":" + params[3]; + DoOneToAllButSender(prefix,"SERVER",params,prefix); + Srv->SendOpers("*** Server \002"+prefix+"\002 introduced server \002"+servername+"\002 ("+description+")"); + return true; + } + + bool Outbound_Reply_Server(std::deque ¶ms) + { + if (params.size() < 4) + return false; + + irc::string servername = params[0].c_str(); + std::string sname = params[0]; + std::string password = params[1]; + int hops = atoi(params[2].c_str()); + + if (hops) + { + this->WriteLine("ERROR :Server too far away for authentication"); + Srv->SendOpers("*** Server connection from \2"+sname+"\2 denied, server is too far away for authentication"); + return false; + } + std::string description = params[3]; + for (std::vector::iterator x = LinkBlocks.begin(); x < LinkBlocks.end(); x++) + { + if ((x->Name == servername) && (x->RecvPass == password)) + { + TreeServer* CheckDupe = FindServer(sname); + if (CheckDupe) + { + this->WriteLine("ERROR :Server "+sname+" already exists on server "+CheckDupe->GetParent()->GetName()+"!"); + Srv->SendOpers("*** Server connection from \2"+sname+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName()); + return false; + } + // Begin the sync here. this kickstarts the + // other side, waiting in WAIT_AUTH_2 state, + // into starting their burst, as it shows + // that we're happy. + this->LinkState = CONNECTED; + // we should add the details of this server now + // to the servers tree, as a child of the root + // node. + TreeServer* Node = new TreeServer(sname,description,TreeRoot,this); + TreeRoot->AddChild(Node); + params[3] = ":" + params[3]; + DoOneToAllButSender(TreeRoot->GetName(),"SERVER",params,sname); + this->bursting = true; + this->DoBurst(Node); + return true; + } + } + this->WriteLine("ERROR :Invalid credentials"); + Srv->SendOpers("*** Server connection from \2"+sname+"\2 denied, invalid link credentials"); + return false; + } + + bool Inbound_Server(std::deque ¶ms) + { + if (params.size() < 4) + return false; + + irc::string servername = params[0].c_str(); + std::string sname = params[0]; + std::string password = params[1]; + int hops = atoi(params[2].c_str()); + + if (hops) + { + this->WriteLine("ERROR :Server too far away for authentication"); + Srv->SendOpers("*** Server connection from \2"+sname+"\2 denied, server is too far away for authentication"); + return false; + } + std::string description = params[3]; + for (std::vector::iterator x = LinkBlocks.begin(); x < LinkBlocks.end(); x++) + { + if ((x->Name == servername) && (x->RecvPass == password)) + { + TreeServer* CheckDupe = FindServer(sname); + if (CheckDupe) + { + this->WriteLine("ERROR :Server "+sname+" already exists on server "+CheckDupe->GetParent()->GetName()+"!"); + Srv->SendOpers("*** Server connection from \2"+sname+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName()); + return false; + } + /* If the config says this link is encrypted, but the remote side + * hasnt bothered to send the AES command before SERVER, then we + * boot them off as we MUST have this connection encrypted. + */ + if ((x->EncryptionKey != "") && (!this->ctx_in)) + { + this->WriteLine("ERROR :This link requires AES encryption to be enabled. Plaintext connection refused."); + Srv->SendOpers("*** Server connection from \2"+sname+"\2 denied, remote server did not enable AES."); + return false; + } + Srv->SendOpers("*** Verified incoming server connection from \002"+sname+"\002["+(x->HiddenFromStats ? "" : this->GetIP())+"] ("+description+")"); + this->InboundServerName = sname; + this->InboundDescription = description; + // this is good. Send our details: Our server name and description and hopcount of 0, + // along with the sendpass from this block. + this->WriteLine("SERVER "+Srv->GetServerName()+" "+x->SendPass+" 0 :"+Srv->GetServerDescription()); + // move to the next state, we are now waiting for THEM. + this->LinkState = WAIT_AUTH_2; + return true; + } + } + this->WriteLine("ERROR :Invalid credentials"); + Srv->SendOpers("*** Server connection from \2"+sname+"\2 denied, invalid link credentials"); + return false; + } + + void Split(std::string line, bool stripcolon, std::deque &n) + { + // we don't do anything with a line > 2048 + if (line.length() > 2048) + { + log(DEBUG,"Line too long!"); + return; + } + if (!strchr(line.c_str(),' ')) + { + n.push_back(line); + return; + } + std::stringstream s(line); + int count = 0; + char param[1024]; + char* pptr = param; + + n.clear(); + int item = 0; + while (!s.eof()) + { + char c = 0; + s.get(c); + if (c == ' ') + { + *pptr = 0; + if (*param) + n.push_back(param); + *param = count = 0; + pptr = param; + item++; + } + else + { + if (!s.eof()) + { + *pptr++ = c; + count++; + } + if ((*param == ':') && (count == 1) && (item > 0)) + { + *param = count = 0; + pptr = param; + while (!s.eof()) + { + s.get(c); + if (!s.eof()) + { + *pptr++ = c; + count++; + } + } + *pptr = 0; + n.push_back(param); + *param = count = 0; + pptr = param; + } + } + } + *pptr = 0; + if (*param) + { + n.push_back(param); + } + + return; + } + + bool ProcessLine(std::string line) + { + char* l = (char*)line.c_str(); + for (char* x = l; *x; x++) + { + if ((*x == '\r') || (*x == '\n')) + *x = 0; + } + if (!*l) + return true; + + log(DEBUG,"IN: %s",l); + + std::deque params; + this->Split(l,true,params); + irc::string command = ""; + std::string prefix = ""; + if (((params[0].c_str())[0] == ':') && (params.size() > 1)) + { + prefix = params[0]; + command = params[1].c_str(); + char* pref = (char*)prefix.c_str(); + prefix = ++pref; + params.pop_front(); + params.pop_front(); + } + else + { + prefix = ""; + command = params[0].c_str(); + params.pop_front(); + } + + if ((!this->ctx_in) && (command == "AES")) + { + std::string sserv = params[0]; + for (std::vector::iterator x = LinkBlocks.begin(); x < LinkBlocks.end(); x++) + { + if ((x->EncryptionKey != "") && (x->Name == sserv)) + { + this->InitAES(x->EncryptionKey,sserv); + } + } + + return true; + } + else if ((this->ctx_in) && (command == "AES")) + { + WriteOpers("*** \2AES\2: Encryption already enabled on this connection yet %s is trying to enable it twice!",params[0].c_str()); + } + + switch (this->LinkState) + { + TreeServer* Node; + + case WAIT_AUTH_1: + // Waiting for SERVER command from remote server. Server initiating + // the connection sends the first SERVER command, listening server + // replies with theirs if its happy, then if the initiator is happy, + // it starts to send its net sync, which starts the merge, otherwise + // it sends an ERROR. + if (command == "PASS") + { + /* Silently ignored */ + } + else if (command == "SERVER") + { + return this->Inbound_Server(params); + } + else if (command == "ERROR") + { + return this->Error(params); + } + else if (command == "USER") + { + this->WriteLine("ERROR :Client connections to this port are prohibited."); + return false; + } + else if (command == "CAPAB") + { + return this->Capab(params); + } + else if ((command == "U") || (command == "S")) + { + this->WriteLine("ERROR :Cannot use the old-style mesh linking protocol with m_spanningtree.so!"); + return false; + } + else + { + this->WriteLine("ERROR :Invalid command in negotiation phase."); + return false; + } + break; + case WAIT_AUTH_2: + // Waiting for start of other side's netmerge to say they liked our + // password. + if (command == "SERVER") + { + // cant do this, they sent it to us in the WAIT_AUTH_1 state! + // silently ignore. + return true; + } + else if ((command == "U") || (command == "S")) + { + this->WriteLine("ERROR :Cannot use the old-style mesh linking protocol with m_spanningtree.so!"); + return false; + } + else if (command == "BURST") + { + if (params.size()) + { + /* If a time stamp is provided, try and check syncronization */ + time_t THEM = atoi(params[0].c_str()); + long delta = THEM-time(NULL); + if ((delta < -600) || (delta > 600)) + { + 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)); + this->WriteLine("ERROR :Your clocks are out by "+ConvToStr(abs(delta))+" seconds (this is more than ten minutes). Link aborted, PLEASE SYNC YOUR CLOCKS!"); + return false; + } + else if ((delta < -60) || (delta > 60)) + { + WriteOpers("*** \2WARNING\2: Your clocks are out by %d seconds, please consider synching your clocks.",abs(delta)); + } + } + this->LinkState = CONNECTED; + Node = new TreeServer(InboundServerName,InboundDescription,TreeRoot,this); + TreeRoot->AddChild(Node); + params.clear(); + params.push_back(InboundServerName); + params.push_back("*"); + params.push_back("1"); + params.push_back(":"+InboundDescription); + DoOneToAllButSender(TreeRoot->GetName(),"SERVER",params,InboundServerName); + this->bursting = true; + this->DoBurst(Node); + } + else if (command == "ERROR") + { + return this->Error(params); + } + else if (command == "CAPAB") + { + return this->Capab(params); + } + + break; + case LISTENER: + this->WriteLine("ERROR :Internal error -- listening socket accepted its own descriptor!!!"); + return false; + break; + case CONNECTING: + if (command == "SERVER") + { + // another server we connected to, which was in WAIT_AUTH_1 state, + // has just sent us their credentials. If we get this far, theyre + // happy with OUR credentials, and they are now in WAIT_AUTH_2 state. + // if we're happy with this, we should send our netburst which + // kickstarts the merge. + return this->Outbound_Reply_Server(params); + } + else if (command == "ERROR") + { + return this->Error(params); + } + break; + case CONNECTED: + // This is the 'authenticated' state, when all passwords + // have been exchanged and anything past this point is taken + // as gospel. + + if (prefix != "") + { + std::string direction = prefix; + userrec* t = Srv->FindNick(prefix); + if (t) + { + direction = t->server; + } + TreeServer* route_back_again = BestRouteTo(direction); + if ((!route_back_again) || (route_back_again->GetSocket() != this)) + { + if (route_back_again) + log(DEBUG,"Protocol violation: Fake direction in command '%s' from connection '%s'",line.c_str(),this->GetName().c_str()); + return true; + } + + /* Fix by brain: + * When there is activity on the socket, reset the ping counter so + * that we're not wasting bandwidth pinging an active server. + */ + route_back_again->SetNextPingTime(time(NULL) + 120); + route_back_again->SetPingFlag(); + } + + if (command == "SVSMODE") + { + /* Services expects us to implement + * SVSMODE. In inspircd its the same as + * MODE anyway. + */ + command = "MODE"; + } + std::string target = ""; + /* Yes, know, this is a mess. Its reasonably fast though as we're + * working with std::string here. + */ + if ((command == "NICK") && (params.size() > 1)) + { + return this->IntroduceClient(prefix,params); + } + else if (command == "FJOIN") + { + return this->ForceJoin(prefix,params); + } + else if (command == "SERVER") + { + return this->RemoteServer(prefix,params); + } + else if (command == "ERROR") + { + return this->Error(params); + } + else if (command == "OPERTYPE") + { + return this->OperType(prefix,params); + } + else if (command == "FMODE") + { + return this->ForceMode(prefix,params); + } + else if (command == "KILL") + { + return this->RemoteKill(prefix,params); + } + else if (command == "FTOPIC") + { + return this->ForceTopic(prefix,params); + } + else if (command == "REHASH") + { + return this->RemoteRehash(prefix,params); + } + else if (command == "METADATA") + { + return this->MetaData(prefix,params); + } + else if (command == "PING") + { + /* + * We just got a ping from a server that's bursting. + * This can't be right, so set them to not bursting, and + * apply their lines. + */ + if (this->bursting) + { + this->bursting = false; + apply_lines(APPLY_ZLINES|APPLY_GLINES|APPLY_QLINES); + } + if (prefix == "") + { + prefix = this->GetName(); + } + return this->LocalPing(prefix,params); + } + else if (command == "PONG") + { + /* + * We just got a pong from a server that's bursting. + * This can't be right, so set them to not bursting, and + * apply their lines. + */ + if (this->bursting) + { + this->bursting = false; + apply_lines(APPLY_ZLINES|APPLY_GLINES|APPLY_QLINES); + } + if (prefix == "") + { + prefix = this->GetName(); + } + return this->LocalPong(prefix,params); + } + else if (command == "VERSION") + { + return this->ServerVersion(prefix,params); + } + else if (command == "FHOST") + { + return this->ChangeHost(prefix,params); + } + else if (command == "FNAME") + { + return this->ChangeName(prefix,params); + } + else if (command == "ADDLINE") + { + return this->AddLine(prefix,params); + } + else if (command == "SVSNICK") + { + if (prefix == "") + { + prefix = this->GetName(); + } + return this->ForceNick(prefix,params); + } + else if (command == "IDLE") + { + return this->Whois(prefix,params); + } + else if (command == "PUSH") + { + return this->Push(prefix,params); + } + else if (command == "TIME") + { + return this->Time(prefix,params); + } + else if ((command == "KICK") && (IsServer(prefix))) + { + std::string sourceserv = this->myhost; + if (params.size() == 3) + { + userrec* user = Srv->FindNick(params[1]); + chanrec* chan = Srv->FindChannel(params[0]); + if (user && chan) + { + server_kick_channel(user,chan,(char*)params[2].c_str(),false); + } + } + if (this->InboundServerName != "") + { + sourceserv = this->InboundServerName; + } + return DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params); + } + else if (command == "SVSJOIN") + { + if (prefix == "") + { + prefix = this->GetName(); + } + return this->ServiceJoin(prefix,params); + } + else if (command == "SQUIT") + { + if (params.size() == 2) + { + this->Squit(FindServer(params[0]),params[1]); + } + return true; + } + else if (command == "ENDBURST") + { + this->bursting = false; + apply_lines(APPLY_ZLINES|APPLY_GLINES|APPLY_QLINES); + std::string sourceserv = this->myhost; + if (this->InboundServerName != "") + { + sourceserv = this->InboundServerName; + } + WriteOpers("*** Received end of netburst from \2%s\2",sourceserv.c_str()); + return true; + } + else + { + // not a special inter-server command. + // Emulate the actual user doing the command, + // this saves us having a huge ugly parser. + userrec* who = Srv->FindNick(prefix); + std::string sourceserv = this->myhost; + if (this->InboundServerName != "") + { + sourceserv = this->InboundServerName; + } + if (who) + { + if (command == "QUIT") + { + TreeServer* s = FindServer(who->server); + if (s) + { + s->DelUser(who); + } + } + else if ((command == "NICK") && (params.size() > 0)) + { + /* On nick messages, check that the nick doesnt + * already exist here. If it does, kill their copy, + * and our copy. + */ + userrec* x = Srv->FindNick(params[0]); + if (x) + { + std::deque p; + p.push_back(params[0]); + p.push_back("Nickname collision ("+prefix+" -> "+params[0]+")"); + DoOneToMany(Srv->GetServerName(),"KILL",p); + p.clear(); + p.push_back(prefix); + p.push_back("Nickname collision"); + DoOneToMany(Srv->GetServerName(),"KILL",p); + Srv->QuitUser(x,"Nickname collision ("+prefix+" -> "+params[0]+")"); + userrec* y = Srv->FindNick(prefix); + if (y) + { + Srv->QuitUser(y,"Nickname collision"); + } + return DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params); + } + } + // its a user + target = who->server; + char* strparams[127]; + for (unsigned int q = 0; q < params.size(); q++) + { + strparams[q] = (char*)params[q].c_str(); + } + if (!Srv->CallCommandHandler(command.c_str(), strparams, params.size(), who)) + { + this->WriteLine("ERROR :Unrecognised command '"+std::string(command.c_str())+"' -- possibly loaded mismatched modules"); + return false; + } + } + else + { + // its not a user. Its either a server, or somethings screwed up. + if (IsServer(prefix)) + { + target = Srv->GetServerName(); + } + else + { + log(DEBUG,"Command with unknown origin '%s'",prefix.c_str()); + return true; + } + } + return DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params); + + } + return true; + break; + } + return true; + } + + virtual std::string GetName() + { + std::string sourceserv = this->myhost; + if (this->InboundServerName != "") + { + sourceserv = this->InboundServerName; + } + return sourceserv; + } + + virtual void OnTimeout() + { + if (this->LinkState == CONNECTING) + { + Srv->SendOpers("*** CONNECT: Connection to \002"+myhost+"\002 timed out."); + } + } + + virtual void OnClose() + { + // Connection closed. + // If the connection is fully up (state CONNECTED) + // then propogate a netsplit to all peers. + std::string quitserver = this->myhost; + if (this->InboundServerName != "") + { + quitserver = this->InboundServerName; + } + TreeServer* s = FindServer(quitserver); + if (s) + { + Squit(s,"Remote host closed the connection"); + } + WriteOpers("Server '\2%s\2' closed the connection.",quitserver.c_str()); + } + + virtual int OnIncomingConnection(int newsock, char* ip) + { + TreeSocket* s = new TreeSocket(newsock, ip); + Srv->AddSocket(s); + return true; + } +}; + +void AddThisServer(TreeServer* server, std::deque &list) +{ + for (unsigned int c = 0; c < list.size(); c++) + { + if (list[c] == server) + { + return; + } + } + list.push_back(server); +} + +// returns a list of DIRECT servernames for a specific channel +void GetListOfServersForChannel(chanrec* c, std::deque &list) +{ + CUList *ulist = c->GetUsers(); + for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++) + { + if (i->second->fd < 0) + { + TreeServer* best = BestRouteTo(i->second->server); + if (best) + AddThisServer(best,list); + } + } + return; +} + +bool DoOneToAllButSenderRaw(std::string data, std::string omit, std::string prefix, irc::string command, std::deque ¶ms) +{ + TreeServer* omitroute = BestRouteTo(omit); + if ((command == "NOTICE") || (command == "PRIVMSG")) + { + if ((params.size() >= 2) && (*(params[0].c_str()) != '$')) + { + /* Prefixes */ + if ((*(params[0].c_str()) == '@') || (*(params[0].c_str()) == '%') || (*(params[0].c_str()) == '+')) + { + params[0] = params[0].substr(1, params[0].length()-1); + } + if (*(params[0].c_str()) != '#') + { + // special routing for private messages/notices + userrec* d = Srv->FindNick(params[0]); + if (d) + { + std::deque par; + par.push_back(params[0]); + par.push_back(":"+params[1]); + DoOneToOne(prefix,command.c_str(),par,d->server); + return true; + } + } + else + { + log(DEBUG,"Channel privmsg going to chan %s",params[0].c_str()); + chanrec* c = Srv->FindChannel(params[0]); + if (c) + { + std::deque list; + GetListOfServersForChannel(c,list); + log(DEBUG,"Got a list of %d servers",list.size()); + unsigned int lsize = list.size(); + for (unsigned int i = 0; i < lsize; i++) + { + TreeSocket* Sock = list[i]->GetSocket(); + if ((Sock) && (list[i]->GetName() != omit) && (omitroute != list[i])) + { + log(DEBUG,"Writing privmsg to server %s",list[i]->GetName().c_str()); + Sock->WriteLine(data); + } + } + return true; + } + } + } + } + unsigned int items = TreeRoot->ChildCount(); + for (unsigned int x = 0; x < items; x++) + { + TreeServer* Route = TreeRoot->GetChild(x); + if ((Route->GetSocket()) && (Route->GetName() != omit) && (omitroute != Route)) + { + TreeSocket* Sock = Route->GetSocket(); + Sock->WriteLine(data); + } + } + return true; +} + +bool DoOneToAllButSender(std::string prefix, std::string command, std::deque ¶ms, std::string omit) +{ + TreeServer* omitroute = BestRouteTo(omit); + std::string FullLine = ":" + prefix + " " + command; + unsigned int words = params.size(); + for (unsigned int x = 0; x < words; x++) + { + FullLine = FullLine + " " + params[x]; + } + unsigned int items = TreeRoot->ChildCount(); + for (unsigned int x = 0; x < items; x++) + { + TreeServer* Route = TreeRoot->GetChild(x); + // Send the line IF: + // The route has a socket (its a direct connection) + // The route isnt the one to be omitted + // The route isnt the path to the one to be omitted + if ((Route->GetSocket()) && (Route->GetName() != omit) && (omitroute != Route)) + { + TreeSocket* Sock = Route->GetSocket(); + Sock->WriteLine(FullLine); + } + } + return true; +} + +bool DoOneToMany(std::string prefix, std::string command, std::deque ¶ms) +{ + std::string FullLine = ":" + prefix + " " + command; + unsigned int words = params.size(); + for (unsigned int x = 0; x < words; x++) + { + FullLine = FullLine + " " + params[x]; + } + unsigned int items = TreeRoot->ChildCount(); + for (unsigned int x = 0; x < items; x++) + { + TreeServer* Route = TreeRoot->GetChild(x); + if (Route->GetSocket()) + { + TreeSocket* Sock = Route->GetSocket(); + Sock->WriteLine(FullLine); + } + } + return true; +} + +bool DoOneToOne(std::string prefix, std::string command, std::deque ¶ms, std::string target) +{ + TreeServer* Route = BestRouteTo(target); + if (Route) + { + std::string FullLine = ":" + prefix + " " + command; + unsigned int words = params.size(); + for (unsigned int x = 0; x < words; x++) + { + FullLine = FullLine + " " + params[x]; + } + if (Route->GetSocket()) + { + TreeSocket* Sock = Route->GetSocket(); + Sock->WriteLine(FullLine); + } + return true; + } + else + { + return true; + } +} + +std::vector Bindings; + +void ReadConfiguration(bool rebind) +{ + Conf = new ConfigReader; + if (rebind) + { + for (int j =0; j < Conf->Enumerate("bind"); j++) + { + std::string Type = Conf->ReadValue("bind","type",j); + std::string IP = Conf->ReadValue("bind","address",j); + long Port = Conf->ReadInteger("bind","port",j,true); + if (Type == "servers") + { + if (IP == "*") + { + IP = ""; + } + TreeSocket* listener = new TreeSocket(IP.c_str(),Port,true,10); + if (listener->GetState() == I_LISTENING) + { + Srv->AddSocket(listener); + Bindings.push_back(listener); + } + else + { + log(DEFAULT,"m_spanningtree: Warning: Failed to bind server port %d",Port); + listener->Close(); + delete listener; + } + } + } + } + FlatLinks = Conf->ReadFlag("options","flatlinks",0); + HideULines = Conf->ReadFlag("options","hideulines",0); + LinkBlocks.clear(); + for (int j =0; j < Conf->Enumerate("link"); j++) + { + Link L; + L.Name = (Conf->ReadValue("link","name",j)).c_str(); + L.IPAddr = Conf->ReadValue("link","ipaddr",j); + L.Port = Conf->ReadInteger("link","port",j,true); + L.SendPass = Conf->ReadValue("link","sendpass",j); + L.RecvPass = Conf->ReadValue("link","recvpass",j); + L.AutoConnect = Conf->ReadInteger("link","autoconnect",j,true); + L.EncryptionKey = Conf->ReadValue("link","encryptionkey",j); + L.HiddenFromStats = Conf->ReadFlag("link","hidden",j); + L.NextConnectTime = time(NULL) + L.AutoConnect; + /* Bugfix by brain, do not allow people to enter bad configurations */ + if ((L.IPAddr != "") && (L.RecvPass != "") && (L.SendPass != "") && (L.Name != "") && (L.Port)) + { + LinkBlocks.push_back(L); + log(DEBUG,"m_spanningtree: Read server %s with host %s:%d",L.Name.c_str(),L.IPAddr.c_str(),L.Port); + } + else + { + if (L.IPAddr == "") + { + log(DEFAULT,"Invalid configuration for server '%s', IP address not defined!",L.Name.c_str()); + } + else if (L.RecvPass == "") + { + log(DEFAULT,"Invalid configuration for server '%s', recvpass not defined!",L.Name.c_str()); + } + else if (L.SendPass == "") + { + log(DEFAULT,"Invalid configuration for server '%s', sendpass not defined!",L.Name.c_str()); + } + else if (L.Name == "") + { + log(DEFAULT,"Invalid configuration, link tag without a name!"); + } + else if (!L.Port) + { + log(DEFAULT,"Invalid configuration for server '%s', no port specified!",L.Name.c_str()); + } + } + } + delete Conf; +} + + +class ModuleSpanningTree : public Module +{ + std::vector Bindings; + int line; + int NumServers; + unsigned int max_local; + unsigned int max_global; + cmd_rconnect* command_rconnect; + + public: + + ModuleSpanningTree(Server* Me) + : Module::Module(Me), max_local(0), max_global(0) + { + Srv = Me; + Bindings.clear(); + + // Create the root of the tree + TreeRoot = new TreeServer(Srv->GetServerName(),Srv->GetServerDescription()); + + ReadConfiguration(true); + + command_rconnect = new cmd_rconnect(this); + Srv->AddCommand(command_rconnect); + } + + void ShowLinks(TreeServer* Current, userrec* user, int hops) + { + std::string Parent = TreeRoot->GetName(); + if (Current->GetParent()) + { + Parent = Current->GetParent()->GetName(); + } + for (unsigned int q = 0; q < Current->ChildCount(); q++) + { + if ((HideULines) && (Srv->IsUlined(Current->GetChild(q)->GetName()))) + { + if (*user->oper) + { + ShowLinks(Current->GetChild(q),user,hops+1); + } + } + else + { + ShowLinks(Current->GetChild(q),user,hops+1); + } + } + /* Don't display the line if its a uline, hide ulines is on, and the user isnt an oper */ + if ((HideULines) && (Srv->IsUlined(Current->GetName())) && (!*user->oper)) + return; + WriteServ(user->fd,"364 %s %s %s :%d %s",user->nick,Current->GetName().c_str(),(FlatLinks && (!*user->oper)) ? Srv->GetServerName().c_str() : Parent.c_str(),(FlatLinks && (!*user->oper)) ? 0 : hops,Current->GetDesc().c_str()); + } + + int CountLocalServs() + { + return TreeRoot->ChildCount(); + } + + int CountServs() + { + return serverlist.size(); + } + + void HandleLinks(char** parameters, int pcnt, userrec* user) + { + ShowLinks(TreeRoot,user,0); + WriteServ(user->fd,"365 %s * :End of /LINKS list.",user->nick); + return; + } + + void HandleLusers(char** parameters, int pcnt, userrec* user) + { + unsigned int n_users = usercnt(); + + /* Only update these when someone wants to see them, more efficient */ + if ((unsigned int)local_count() > max_local) + max_local = local_count(); + if (n_users > max_global) + max_global = n_users; + + WriteServ(user->fd,"251 %s :There are %d users and %d invisible on %d servers",user->nick,n_users-usercount_invisible(),usercount_invisible(),this->CountServs()); + WriteServ(user->fd,"252 %s %d :operator(s) online",user->nick,usercount_opers()); + WriteServ(user->fd,"253 %s %d :unknown connections",user->nick,usercount_unknown()); + WriteServ(user->fd,"254 %s %d :channels formed",user->nick,chancount()); + WriteServ(user->fd,"254 %s :I have %d clients and %d servers",user->nick,local_count(),this->CountLocalServs()); + WriteServ(user->fd,"265 %s :Current Local Users: %d Max: %d",user->nick,local_count(),max_local); + WriteServ(user->fd,"266 %s :Current Global Users: %d Max: %d",user->nick,n_users,max_global); + return; + } + + // WARNING: NOT THREAD SAFE - DONT GET ANY SMART IDEAS. + + void ShowMap(TreeServer* Current, userrec* user, int depth, char matrix[128][80], float &totusers, float &totservers) + { + if (line < 128) + { + for (int t = 0; t < depth; t++) + { + matrix[line][t] = ' '; + } + + // For Aligning, we need to work out exactly how deep this thing is, and produce + // a 'Spacer' String to compensate. + char spacer[40]; + + memset(spacer,' ',40); + if ((40 - Current->GetName().length() - depth) > 1) { + spacer[40 - Current->GetName().length() - depth] = '\0'; + } + else + { + spacer[5] = '\0'; + } + + float percent; + char text[80]; + if (clientlist.size() == 0) { + // If there are no users, WHO THE HELL DID THE /MAP?!?!?! + percent = 0; + } + else + { + percent = ((float)Current->GetUserCount() / (float)clientlist.size()) * 100; + } + snprintf(text, 80, "%s %s%5d [%5.2f%%]", Current->GetName().c_str(), spacer, Current->GetUserCount(), percent); + totusers += Current->GetUserCount(); + totservers++; + strlcpy(&matrix[line][depth],text,80); + line++; + for (unsigned int q = 0; q < Current->ChildCount(); q++) + { + if ((HideULines) && (Srv->IsUlined(Current->GetChild(q)->GetName()))) + { + if (*user->oper) + { + ShowMap(Current->GetChild(q),user,(FlatLinks && (!*user->oper)) ? depth : depth+2,matrix,totusers,totservers); + } + } + else + { + ShowMap(Current->GetChild(q),user,(FlatLinks && (!*user->oper)) ? depth : depth+2,matrix,totusers,totservers); + } + } + } + } + + // Ok, prepare to be confused. + // After much mulling over how to approach this, it struck me that + // the 'usual' way of doing a /MAP isnt the best way. Instead of + // keeping track of a ton of ascii characters, and line by line + // under recursion working out where to place them using multiplications + // and divisons, we instead render the map onto a backplane of characters + // (a character matrix), then draw the branches as a series of "L" shapes + // from the nodes. This is not only friendlier on CPU it uses less stack. + + void HandleMap(char** parameters, int pcnt, userrec* user) + { + // This array represents a virtual screen which we will + // "scratch" draw to, as the console device of an irc + // client does not provide for a proper terminal. + float totusers = 0; + float totservers = 0; + char matrix[128][80]; + for (unsigned int t = 0; t < 128; t++) + { + matrix[t][0] = '\0'; + } + line = 0; + // The only recursive bit is called here. + ShowMap(TreeRoot,user,0,matrix,totusers,totservers); + // Process each line one by one. The algorithm has a limit of + // 128 servers (which is far more than a spanning tree should have + // anyway, so we're ok). This limit can be raised simply by making + // the character matrix deeper, 128 rows taking 10k of memory. + for (int l = 1; l < line; l++) + { + // scan across the line looking for the start of the + // servername (the recursive part of the algorithm has placed + // the servers at indented positions depending on what they + // are related to) + int first_nonspace = 0; + while (matrix[l][first_nonspace] == ' ') + { + first_nonspace++; + } + first_nonspace--; + // Draw the `- (corner) section: this may be overwritten by + // another L shape passing along the same vertical pane, becoming + // a |- (branch) section instead. + matrix[l][first_nonspace] = '-'; + matrix[l][first_nonspace-1] = '`'; + int l2 = l - 1; + // Draw upwards until we hit the parent server, causing possibly + // other corners (`-) to become branches (|-) + while ((matrix[l2][first_nonspace-1] == ' ') || (matrix[l2][first_nonspace-1] == '`')) + { + matrix[l2][first_nonspace-1] = '|'; + l2--; + } + } + // dump the whole lot to the user. This is the easy bit, honest. + for (int t = 0; t < line; t++) + { + WriteServ(user->fd,"006 %s :%s",user->nick,&matrix[t][0]); + } + float avg_users = totusers / totservers; + WriteServ(user->fd,"270 %s :%.0f server%s and %.0f user%s, average %.2f users per server",user->nick,totservers,(totservers > 1 ? "s" : ""),totusers,(totusers > 1 ? "s" : ""),avg_users); + WriteServ(user->fd,"007 %s :End of /MAP",user->nick); + return; + } + + int HandleSquit(char** parameters, int pcnt, userrec* user) + { + TreeServer* s = FindServerMask(parameters[0]); + if (s) + { + if (s == TreeRoot) + { + WriteServ(user->fd,"NOTICE %s :*** SQUIT: Foolish mortal, you cannot make a server SQUIT itself! (%s matches local server name)",user->nick,parameters[0]); + return 1; + } + TreeSocket* sock = s->GetSocket(); + if (sock) + { + log(DEBUG,"Splitting server %s",s->GetName().c_str()); + WriteOpers("*** SQUIT: Server \002%s\002 removed from network by %s",parameters[0],user->nick); + sock->Squit(s,"Server quit by "+std::string(user->nick)+"!"+std::string(user->ident)+"@"+std::string(user->host)); + Srv->RemoveSocket(sock); + } + else + { + WriteServ(user->fd,"NOTICE %s :*** SQUIT: The server \002%s\002 is not directly connected.",user->nick,parameters[0]); + } + } + else + { + WriteServ(user->fd,"NOTICE %s :*** SQUIT: The server \002%s\002 does not exist on the network.",user->nick,parameters[0]); + } + return 1; + } + + int HandleTime(char** parameters, int pcnt, userrec* user) + { + if ((user->fd > -1) && (pcnt)) + { + TreeServer* found = FindServerMask(parameters[0]); + if (found) + { + // we dont' override for local server + if (found == TreeRoot) + return 0; + + std::deque params; + params.push_back(found->GetName()); + params.push_back(user->nick); + DoOneToOne(Srv->GetServerName(),"TIME",params,found->GetName()); + } + else + { + WriteServ(user->fd,"402 %s %s :No such server",user->nick,parameters[0]); + } + } + return 1; + } + + int HandleRemoteWhois(char** parameters, int pcnt, userrec* user) + { + if ((user->fd > -1) && (pcnt > 1)) + { + userrec* remote = Srv->FindNick(parameters[1]); + if ((remote) && (remote->fd < 0)) + { + std::deque params; + params.push_back(parameters[1]); + DoOneToOne(user->nick,"IDLE",params,remote->server); + return 1; + } + else if (!remote) + { + WriteServ(user->fd,"401 %s %s :No such nick/channel",user->nick, parameters[1]); + WriteServ(user->fd,"318 %s %s :End of /WHOIS list.",user->nick, parameters[1]); + return 1; + } + } + return 0; + } + + void DoPingChecks(time_t curtime) + { + for (unsigned int j = 0; j < TreeRoot->ChildCount(); j++) + { + TreeServer* serv = TreeRoot->GetChild(j); + TreeSocket* sock = serv->GetSocket(); + if (sock) + { + if (curtime >= serv->NextPingTime()) + { + if (serv->AnsweredLastPing()) + { + sock->WriteLine(":"+Srv->GetServerName()+" PING "+serv->GetName()); + serv->SetNextPingTime(curtime + 120); + } + else + { + // they didnt answer, boot them + WriteOpers("*** Server \002%s\002 pinged out",serv->GetName().c_str()); + sock->Squit(serv,"Ping timeout"); + Srv->RemoveSocket(sock); + return; + } + } + } + } + } + + void AutoConnectServers(time_t curtime) + { + for (std::vector::iterator x = LinkBlocks.begin(); x < LinkBlocks.end(); x++) + { + if ((x->AutoConnect) && (curtime >= x->NextConnectTime)) + { + log(DEBUG,"Auto-Connecting %s",x->Name.c_str()); + x->NextConnectTime = curtime + x->AutoConnect; + TreeServer* CheckDupe = FindServer(x->Name.c_str()); + if (!CheckDupe) + { + // an autoconnected server is not connected. Check if its time to connect it + WriteOpers("*** AUTOCONNECT: Auto-connecting server \002%s\002 (%lu seconds until next attempt)",x->Name.c_str(),x->AutoConnect); + TreeSocket* newsocket = new TreeSocket(x->IPAddr,x->Port,false,10,x->Name.c_str()); + if (newsocket->GetFd() > -1) + { + Srv->AddSocket(newsocket); + } + else + { + WriteOpers("*** AUTOCONNECT: Error autoconnecting \002%s\002: %s.",x->Name.c_str(),strerror(errno)); + delete newsocket; + } + } + } + } + } + + int HandleVersion(char** parameters, int pcnt, userrec* user) + { + // we've already checked if pcnt > 0, so this is safe + TreeServer* found = FindServerMask(parameters[0]); + if (found) + { + std::string Version = found->GetVersion(); + WriteServ(user->fd,"351 %s :%s",user->nick,Version.c_str()); + if (found == TreeRoot) + { + std::stringstream out(Config->data005); + std::string token = ""; + std::string line5 = ""; + int token_counter = 0; + + while (!out.eof()) + { + out >> token; + line5 = line5 + token + " "; + token_counter++; + + if ((token_counter >= 13) || (out.eof() == true)) + { + WriteServ(user->fd,"005 %s %s:are supported by this server",user->nick,line5.c_str()); + line5 = ""; + token_counter = 0; + } + } + } + } + else + { + WriteServ(user->fd,"402 %s %s :No such server",user->nick,parameters[0]); + } + return 1; + } + + int HandleConnect(char** parameters, int pcnt, userrec* user) + { + for (std::vector::iterator x = LinkBlocks.begin(); x < LinkBlocks.end(); x++) + { + if (Srv->MatchText(x->Name.c_str(),parameters[0])) + { + TreeServer* CheckDupe = FindServer(x->Name.c_str()); + if (!CheckDupe) + { + WriteServ(user->fd,"NOTICE %s :*** CONNECT: Connecting to server: \002%s\002 (%s:%d)",user->nick,x->Name.c_str(),(x->HiddenFromStats ? "" : x->IPAddr.c_str()),x->Port); + TreeSocket* newsocket = new TreeSocket(x->IPAddr,x->Port,false,10,x->Name.c_str()); + if (newsocket->GetFd() > -1) + { + Srv->AddSocket(newsocket); + } + else + { + WriteServ(user->fd,"NOTICE %s :*** CONNECT: Error connecting \002%s\002: %s.",user->nick,x->Name.c_str(),strerror(errno)); + delete newsocket; + } + return 1; + } + else + { + WriteServ(user->fd,"NOTICE %s :*** CONNECT: Server \002%s\002 already exists on the network and is connected via \002%s\002",user->nick,x->Name.c_str(),CheckDupe->GetParent()->GetName().c_str()); + return 1; + } + } + } + WriteServ(user->fd,"NOTICE %s :*** CONNECT: No server matching \002%s\002 could be found in the config file.",user->nick,parameters[0]); + return 1; + } + + virtual int OnStats(char statschar, userrec* user) + { + if (statschar == 'c') + { + for (unsigned int i = 0; i < LinkBlocks.size(); i++) + { + WriteServ(user->fd,"213 %s C *@%s * %s %d 0 %c%c%c",user->nick,(LinkBlocks[i].HiddenFromStats ? "" : LinkBlocks[i].IPAddr).c_str(),LinkBlocks[i].Name.c_str(),LinkBlocks[i].Port,(LinkBlocks[i].EncryptionKey != "" ? 'e' : '-'),(LinkBlocks[i].AutoConnect ? 'a' : '-'),'s'); + WriteServ(user->fd,"244 %s H * * %s",user->nick,LinkBlocks[i].Name.c_str()); + } + WriteServ(user->fd,"219 %s %c :End of /STATS report",user->nick,statschar); + WriteOpers("*** Notice: Stats '%c' requested by %s (%s@%s)",statschar,user->nick,user->ident,user->host); + return 1; + } + return 0; + } + + virtual int OnPreCommand(const std::string &command, char **parameters, int pcnt, userrec *user, bool validated) + { + /* If the command doesnt appear to be valid, we dont want to mess with it. */ + if (!validated) + return 0; + + if (command == "CONNECT") + { + return this->HandleConnect(parameters,pcnt,user); + } + else if (command == "SQUIT") + { + return this->HandleSquit(parameters,pcnt,user); + } + else if (command == "MAP") + { + this->HandleMap(parameters,pcnt,user); + return 1; + } + else if ((command == "TIME") && (pcnt > 0)) + { + return this->HandleTime(parameters,pcnt,user); + } + else if (command == "LUSERS") + { + this->HandleLusers(parameters,pcnt,user); + return 1; + } + else if (command == "LINKS") + { + this->HandleLinks(parameters,pcnt,user); + return 1; + } + else if (command == "WHOIS") + { + if (pcnt > 1) + { + // remote whois + return this->HandleRemoteWhois(parameters,pcnt,user); + } + } + else if ((command == "VERSION") && (pcnt > 0)) + { + this->HandleVersion(parameters,pcnt,user); + return 1; + } + else if (Srv->IsValidModuleCommand(command, pcnt, user)) + { + // this bit of code cleverly routes all module commands + // to all remote severs *automatically* so that modules + // can just handle commands locally, without having + // to have any special provision in place for remote + // commands and linking protocols. + std::deque params; + params.clear(); + for (int j = 0; j < pcnt; j++) + { + if (strchr(parameters[j],' ')) + { + params.push_back(":" + std::string(parameters[j])); + } + else + { + params.push_back(std::string(parameters[j])); + } + } + log(DEBUG,"Globally route '%s'",command.c_str()); + DoOneToMany(user->nick,command,params); + } + return 0; + } + + virtual void OnGetServerDescription(const std::string &servername,std::string &description) + { + TreeServer* s = FindServer(servername); + if (s) + { + description = s->GetDesc(); + } + } + + virtual void OnUserInvite(userrec* source,userrec* dest,chanrec* channel) + { + if (source->fd > -1) + { + std::deque params; + params.push_back(dest->nick); + params.push_back(channel->name); + DoOneToMany(source->nick,"INVITE",params); + } + } + + virtual void OnPostLocalTopicChange(userrec* user, chanrec* chan, const std::string &topic) + { + std::deque params; + params.push_back(chan->name); + params.push_back(":"+topic); + DoOneToMany(user->nick,"TOPIC",params); + } + + virtual void OnWallops(userrec* user, const std::string &text) + { + if (user->fd > -1) + { + std::deque params; + params.push_back(":"+text); + DoOneToMany(user->nick,"WALLOPS",params); + } + } + + virtual void OnUserNotice(userrec* user, void* dest, int target_type, const std::string &text, char status) + { + if (target_type == TYPE_USER) + { + userrec* d = (userrec*)dest; + if ((d->fd < 0) && (user->fd > -1)) + { + std::deque params; + params.clear(); + params.push_back(d->nick); + params.push_back(":"+text); + DoOneToOne(user->nick,"NOTICE",params,d->server); + } + } + else + { + if (user->fd > -1) + { + chanrec *c = (chanrec*)dest; + std::string cname = c->name; + if (status) + cname = status + cname; + std::deque list; + GetListOfServersForChannel(c,list); + unsigned int ucount = list.size(); + for (unsigned int i = 0; i < ucount; i++) + { + TreeSocket* Sock = list[i]->GetSocket(); + if (Sock) + Sock->WriteLine(":"+std::string(user->nick)+" NOTICE "+cname+" :"+text); + } + } + } + } + + virtual void OnUserMessage(userrec* user, void* dest, int target_type, const std::string &text, char status) + { + if (target_type == TYPE_USER) + { + // route private messages which are targetted at clients only to the server + // which needs to receive them + userrec* d = (userrec*)dest; + if ((d->fd < 0) && (user->fd > -1)) + { + std::deque params; + params.clear(); + params.push_back(d->nick); + params.push_back(":"+text); + DoOneToOne(user->nick,"PRIVMSG",params,d->server); + } + } + else + { + if (user->fd > -1) + { + chanrec *c = (chanrec*)dest; + std::string cname = c->name; + if (status) + cname = status + cname; + std::deque list; + GetListOfServersForChannel(c,list); + unsigned int ucount = list.size(); + for (unsigned int i = 0; i < ucount; i++) + { + TreeSocket* Sock = list[i]->GetSocket(); + if (Sock) + Sock->WriteLine(":"+std::string(user->nick)+" PRIVMSG "+cname+" :"+text); + } + } + } + } + + virtual void OnBackgroundTimer(time_t curtime) + { + AutoConnectServers(curtime); + DoPingChecks(curtime); + } + + virtual void OnUserJoin(userrec* user, chanrec* channel) + { + // Only do this for local users + if (user->fd > -1) + { + std::deque params; + params.clear(); + params.push_back(channel->name); + + if (channel->GetUserCounter() > 1) + { + // not the first in the channel + DoOneToMany(user->nick,"JOIN",params); + } + else + { + // first in the channel, set up their permissions + // and the channel TS with FJOIN. + char ts[24]; + snprintf(ts,24,"%lu",(unsigned long)channel->age); + params.clear(); + params.push_back(channel->name); + params.push_back(ts); + params.push_back("@"+std::string(user->nick)); + DoOneToMany(Srv->GetServerName(),"FJOIN",params); + } + } + } + + virtual void OnChangeHost(userrec* user, const std::string &newhost) + { + // only occurs for local clients + if (user->registered != 7) + return; + std::deque params; + params.push_back(newhost); + DoOneToMany(user->nick,"FHOST",params); + } + + virtual void OnChangeName(userrec* user, const std::string &gecos) + { + // only occurs for local clients + if (user->registered != 7) + return; + std::deque params; + params.push_back(gecos); + DoOneToMany(user->nick,"FNAME",params); + } + + virtual void OnUserPart(userrec* user, chanrec* channel, const std::string &partmessage) + { + if (user->fd > -1) + { + std::deque params; + params.push_back(channel->name); + if (partmessage != "") + params.push_back(":"+partmessage); + DoOneToMany(user->nick,"PART",params); + } + } + + virtual void OnUserConnect(userrec* user) + { + char agestr[MAXBUF]; + if (user->fd > -1) + { + std::deque params; + snprintf(agestr,MAXBUF,"%lu",(unsigned long)user->age); + params.push_back(agestr); + params.push_back(user->nick); + params.push_back(user->host); + params.push_back(user->dhost); + params.push_back(user->ident); + params.push_back("+"+std::string(user->modes)); + params.push_back((char*)inet_ntoa(user->ip4)); + params.push_back(":"+std::string(user->fullname)); + DoOneToMany(Srv->GetServerName(),"NICK",params); + + // User is Local, change needs to be reflected! + TreeServer* SourceServer = FindServer(user->server); + if (SourceServer) + { + SourceServer->AddUserCount(); + } + + } + } + + virtual void OnUserQuit(userrec* user, const std::string &reason) + { + if ((user->fd > -1) && (user->registered == 7)) + { + std::deque params; + params.push_back(":"+reason); + DoOneToMany(user->nick,"QUIT",params); + } + // Regardless, We need to modify the user Counts.. + TreeServer* SourceServer = FindServer(user->server); + if (SourceServer) + { + SourceServer->DelUserCount(); + } + + } + + virtual void OnUserPostNick(userrec* user, const std::string &oldnick) + { + if (user->fd > -1) + { + std::deque params; + params.push_back(user->nick); + DoOneToMany(oldnick,"NICK",params); + } + } + + virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, const std::string &reason) + { + if ((source) && (source->fd > -1)) + { + std::deque params; + params.push_back(chan->name); + params.push_back(user->nick); + params.push_back(":"+reason); + DoOneToMany(source->nick,"KICK",params); + } + else if (!source) + { + std::deque params; + params.push_back(chan->name); + params.push_back(user->nick); + params.push_back(":"+reason); + DoOneToMany(Srv->GetServerName(),"KICK",params); + } + } + + virtual void OnRemoteKill(userrec* source, userrec* dest, const std::string &reason) + { + std::deque params; + params.push_back(dest->nick); + params.push_back(":"+reason); + DoOneToMany(source->nick,"KILL",params); + } + + virtual void OnRehash(const std::string ¶meter) + { + if (parameter != "") + { + std::deque params; + params.push_back(parameter); + DoOneToMany(Srv->GetServerName(),"REHASH",params); + // check for self + if (Srv->MatchText(Srv->GetServerName(),parameter)) + { + Srv->SendOpers("*** Remote rehash initiated from server \002"+Srv->GetServerName()+"\002."); + Srv->RehashServer(); + } + } + ReadConfiguration(false); + } + + // note: the protocol does not allow direct umode +o except + // via NICK with 8 params. sending OPERTYPE infers +o modechange + // locally. + virtual void OnOper(userrec* user, const std::string &opertype) + { + if (user->fd > -1) + { + std::deque params; + params.push_back(opertype); + DoOneToMany(user->nick,"OPERTYPE",params); + } + } + + void OnLine(userrec* source, const std::string &host, bool adding, char linetype, long duration, const std::string &reason) + { + if (source->fd > -1) + { + char type[8]; + snprintf(type,8,"%cLINE",linetype); + std::string stype = type; + if (adding) + { + char sduration[MAXBUF]; + snprintf(sduration,MAXBUF,"%ld",duration); + std::deque params; + params.push_back(host); + params.push_back(sduration); + params.push_back(":"+reason); + DoOneToMany(source->nick,stype,params); + } + else + { + std::deque params; + params.push_back(host); + DoOneToMany(source->nick,stype,params); + } + } + } + + virtual void OnAddGLine(long duration, userrec* source, const std::string &reason, const std::string &hostmask) + { + OnLine(source,hostmask,true,'G',duration,reason); + } + + virtual void OnAddZLine(long duration, userrec* source, const std::string &reason, const std::string &ipmask) + { + OnLine(source,ipmask,true,'Z',duration,reason); + } + + virtual void OnAddQLine(long duration, userrec* source, const std::string &reason, const std::string &nickmask) + { + OnLine(source,nickmask,true,'Q',duration,reason); + } + + virtual void OnAddELine(long duration, userrec* source, const std::string &reason, const std::string &hostmask) + { + OnLine(source,hostmask,true,'E',duration,reason); + } + + virtual void OnDelGLine(userrec* source, const std::string &hostmask) + { + OnLine(source,hostmask,false,'G',0,""); + } + + virtual void OnDelZLine(userrec* source, const std::string &ipmask) + { + OnLine(source,ipmask,false,'Z',0,""); + } + + virtual void OnDelQLine(userrec* source, const std::string &nickmask) + { + OnLine(source,nickmask,false,'Q',0,""); + } + + virtual void OnDelELine(userrec* source, const std::string &hostmask) + { + OnLine(source,hostmask,false,'E',0,""); + } + + virtual void OnMode(userrec* user, void* dest, int target_type, const std::string &text) + { + if ((user->fd > -1) && (user->registered == 7)) + { + if (target_type == TYPE_USER) + { + userrec* u = (userrec*)dest; + std::deque params; + params.push_back(u->nick); + params.push_back(text); + DoOneToMany(user->nick,"MODE",params); + } + else + { + chanrec* c = (chanrec*)dest; + std::deque params; + params.push_back(c->name); + params.push_back(text); + DoOneToMany(user->nick,"MODE",params); + } + } + } + + virtual void OnSetAway(userrec* user) + { + if (IS_LOCAL(user)) + { + std::deque params; + params.push_back(":"+std::string(user->awaymsg)); + DoOneToMany(user->nick,"AWAY",params); + } + } + + virtual void OnCancelAway(userrec* user) + { + if (IS_LOCAL(user)) + { + std::deque params; + params.clear(); + DoOneToMany(user->nick,"AWAY",params); + } + } + + virtual void ProtoSendMode(void* opaque, int target_type, void* target, const std::string &modeline) + { + TreeSocket* s = (TreeSocket*)opaque; + if (target) + { + if (target_type == TYPE_USER) + { + userrec* u = (userrec*)target; + s->WriteLine(":"+Srv->GetServerName()+" FMODE "+u->nick+" "+modeline); + } + else + { + chanrec* c = (chanrec*)target; + s->WriteLine(":"+Srv->GetServerName()+" FMODE "+c->name+" "+modeline); + } + } + } + + virtual void ProtoSendMetaData(void* opaque, int target_type, void* target, const std::string &extname, const std::string &extdata) + { + TreeSocket* s = (TreeSocket*)opaque; + if (target) + { + if (target_type == TYPE_USER) + { + userrec* u = (userrec*)target; + s->WriteLine(":"+Srv->GetServerName()+" METADATA "+u->nick+" "+extname+" :"+extdata); + } + else if (target_type == TYPE_OTHER) + { + s->WriteLine(":"+Srv->GetServerName()+" METADATA * "+extname+" :"+extdata); + } + else if (target_type == TYPE_CHANNEL) + { + chanrec* c = (chanrec*)target; + s->WriteLine(":"+Srv->GetServerName()+" METADATA "+c->name+" "+extname+" :"+extdata); + } + } + } + + virtual void OnEvent(Event* event) + { + if (event->GetEventID() == "send_metadata") + { + std::deque* params = (std::deque*)event->GetData(); + if (params->size() < 3) + return; + (*params)[2] = ":" + (*params)[2]; + DoOneToMany(Srv->GetServerName(),"METADATA",*params); + } + } + + virtual ~ModuleSpanningTree() + { + } + + virtual Version GetVersion() + { + return Version(1,0,0,0,VF_STATIC|VF_VENDOR); + } + + void Implements(char* List) + { + List[I_OnPreCommand] = List[I_OnGetServerDescription] = List[I_OnUserInvite] = List[I_OnPostLocalTopicChange] = 1; + List[I_OnWallops] = List[I_OnUserNotice] = List[I_OnUserMessage] = List[I_OnBackgroundTimer] = 1; + List[I_OnUserJoin] = List[I_OnChangeHost] = List[I_OnChangeName] = List[I_OnUserPart] = List[I_OnUserConnect] = 1; + List[I_OnUserQuit] = List[I_OnUserPostNick] = List[I_OnUserKick] = List[I_OnRemoteKill] = List[I_OnRehash] = 1; + List[I_OnOper] = List[I_OnAddGLine] = List[I_OnAddZLine] = List[I_OnAddQLine] = List[I_OnAddELine] = 1; + List[I_OnDelGLine] = List[I_OnDelZLine] = List[I_OnDelQLine] = List[I_OnDelELine] = List[I_ProtoSendMode] = List[I_OnMode] = 1; + List[I_OnStats] = List[I_ProtoSendMetaData] = List[I_OnEvent] = List[I_OnSetAway] = List[I_OnCancelAway] = 1; + } + + /* It is IMPORTANT that m_spanningtree is the last module in the chain + * so that any activity it sees is FINAL, e.g. we arent going to send out + * a NICK message before m_cloaking has finished putting the +x on the user, + * etc etc. + * Therefore, we return PRIORITY_LAST to make sure we end up at the END of + * the module call queue. + */ + Priority Prioritize() + { + return PRIORITY_LAST; + } +}; + + +class ModuleSpanningTreeFactory : public ModuleFactory +{ + public: + ModuleSpanningTreeFactory() + { + } + + ~ModuleSpanningTreeFactory() + { + } + + virtual Module * CreateModule(Server* Me) + { + TreeProtocolModule = new ModuleSpanningTree(Me); + return TreeProtocolModule; + } + +}; + + +extern "C" void * init_module( void ) +{ + return new ModuleSpanningTreeFactory; +} -- 2.39.2