src/modules/m_spanningtree/treesocket.h

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #ifndef __TREESOCKET_H__
  15 #define __TREESOCKET_H__
  16
  17 #include "socket.h"
  18 #include "inspircd.h"
  19 #include "xline.h"
  20 #include "../transport.h"
  21
  22 #include "utils.h"
  23 #include "handshaketimer.h"
  24
  25 /*
  26  * The server list in InspIRCd is maintained as two structures
  27  * which hold the data in different ways. Most of the time, we
  28  * want to very quicky obtain three pieces of information:
  29  *
  30  * (1) The information on a server
  31  * (2) The information on the server we must send data through
  32  *     to actually REACH the server we're after
  33  * (3) Potentially, the child/parent objects of this server
  34  *
  35  * The InspIRCd spanning protocol provides easy access to these
  36  * by storing the data firstly in a recursive structure, where
  37  * each item references its parent item, and a dynamic list
  38  * of child items, and another structure which stores the items
  39  * hashed, linearly. This means that if we want to find a server
  40  * by name quickly, we can look it up in the hash, avoiding
  41  * any O(n) lookups. If however, during a split or sync, we want
  42  * to apply an operation to a server, and any of its child objects
  43  * we can resort to recursion to walk the tree structure.
  44  * Any socket can have one of five states at any one time.
  45  *
  46  * CONNECTING:  indicates an outbound socket which is
  47  *                                                      waiting to be writeable.
  48  * WAIT_AUTH_1: indicates the socket is outbound and
  49  *                                                      has successfully connected, but has not
  50  *                                                      yet sent and received SERVER strings.
  51  * WAIT_AUTH_2: indicates that the socket is inbound
  52  *                                                      but has not yet sent and received
  53  *                                                      SERVER strings.
  54  * CONNECTED:           represents a fully authorized, fully
  55  *                                                      connected server.
  56  */
  57 enum ServerState { CONNECTING, WAIT_AUTH_1, WAIT_AUTH_2, CONNECTED };
  58
  59 /** Every SERVER connection inbound or outbound is represented by
  60  * an object of type TreeSocket.
  61  * TreeSockets, being inherited from BufferedSocket, can be tied into
  62  * the core socket engine, and we cn therefore receive activity events
  63  * for them, just like activex objects on speed. (yes really, that
  64  * is a technical term!) Each of these which relates to a locally
  65  * connected server is assocated with it, by hooking it onto a
  66  * TreeSocket class using its constructor. In this way, we can
  67  * maintain a list of servers, some of which are directly connected,
  68  * some of which are not.
  69  */
  70 class TreeSocket : public BufferedSocket
  71 {
  72         SpanningTreeUtilities* Utils;           /* Utility class */
  73         std::string myhost;                     /* Canonical hostname */
  74         ServerState LinkState;                  /* Link state */
  75         std::string InboundServerName;          /* Server name sent to us by other side */
  76         std::string InboundDescription;         /* Server description (GECOS) sent to us by the other side */
  77         std::string InboundSID;                 /* Server ID sent to us by the other side */
  78         int num_lost_users;                     /* Users lost in split */
  79         int num_lost_servers;                   /* Servers lost in split */
  80         time_t NextPing;                        /* Time when we are due to ping this server */
  81         bool LastPingWasGood;                   /* Responded to last ping we sent? */
  82         std::string IP;
  83         std::string ModuleList;                 /* Required module list of other server from CAPAB */
  84         std::string OptModuleList;              /* Optional module list of other server from CAPAB */
  85         std::map<std::string,std::string> CapKeys;      /* CAPAB keys from other server */
  86         std::string ourchallenge;               /* Challenge sent for challenge/response */
  87         std::string theirchallenge;             /* Challenge recv for challenge/response */
  88         std::string OutboundPass;               /* Outbound password */
  89         int capab_phase;                        /* Have sent CAPAB already */
  90         bool auth_fingerprint;                  /* Did we auth using SSL fingerprint */
  91         bool auth_challenge;                    /* Did we auth using challenge/response */
  92         int proto_version;                      /* Remote protocol version */
  93  public:
  94         HandshakeTimer* hstimer;                /* Handshake timer, needed to work around I/O hook buffering */
  95         reference<Autoconnect> myautoconnect;           /* Autoconnect used to cause this connection, if any */
  96         time_t age;
  97
  98         /** Because most of the I/O gubbins are encapsulated within
  99          * BufferedSocket, we just call the superclass constructor for
 100          * most of the action, and append a few of our own values
 101          * to it.
 102          */
 103         TreeSocket(SpanningTreeUtilities* Util, std::string host, int port, unsigned long maxtime, const std::string &ServerName, const std::string &bindto, Autoconnect* myac, Module* HookMod = NULL);
 104
 105         /** When a listening socket gives us a new file descriptor,
 106          * we must associate it with a socket without creating a new
 107          * connection. This constructor is used for this purpose.
 108          */
 109         TreeSocket(SpanningTreeUtilities* Util, int newfd, char* ip, Autoconnect* myac, Module* HookMod = NULL);
 110
 111         /** Get link state
 112          */
 113         ServerState GetLinkState();
 114
 115         /** Get challenge set in our CAPAB for challenge/response
 116          */
 117         const std::string& GetOurChallenge();
 118
 119         /** Get challenge set in our CAPAB for challenge/response
 120          */
 121         void SetOurChallenge(const std::string &c);
 122
 123         /** Get challenge set in their CAPAB for challenge/response
 124          */
 125         const std::string& GetTheirChallenge();
 126
 127         /** Get challenge set in their CAPAB for challenge/response
 128          */
 129         void SetTheirChallenge(const std::string &c);
 130
 131         /** Compare two passwords based on authentication scheme
 132          */
 133         bool ComparePass(const Link& link, const std::string &theirs);
 134
 135         /** Clean up information used only during server negotiation
 136          */
 137         void CleanNegotiationInfo();
 138
 139         /** Destructor
 140          */
 141         ~TreeSocket();
 142
 143         /** Generate random string used for challenge-response auth
 144          */
 145         std::string RandString(unsigned int length);
 146
 147         /** Construct a password, optionally hashed with the other side's
 148          * challenge string
 149          */
 150         std::string MakePass(const std::string &password, const std::string &challenge);
 151
 152         /** When an outbound connection finishes connecting, we receive
 153          * this event, and must send our SERVER string to the other
 154          * side. If the other side is happy, as outlined in the server
 155          * to server docs on the inspircd.org site, the other side
 156          * will then send back its own server string.
 157          */
 158         virtual void OnConnected();
 159
 160         /** Handle socket error event
 161          */
 162         virtual void OnError(BufferedSocketError e);
 163
 164         /** Sends an error to the remote server, and displays it locally to show
 165          * that it was sent.
 166          */
 167         void SendError(const std::string &errormessage);
 168
 169         /** Recursively send the server tree with distances as hops.
 170          * This is used during network burst to inform the other server
 171          * (and any of ITS servers too) of what servers we know about.
 172          * If at any point any of these servers already exist on the other
 173          * end, our connection may be terminated. The hopcounts given
 174          * by this function are relative, this doesn't matter so long as
 175          * they are all >1, as all the remote servers re-calculate them
 176          * to be relative too, with themselves as hop 0.
 177          */
 178         void SendServers(TreeServer* Current, TreeServer* s, int hops);
 179
 180         /** Returns module list as a string, filtered by filter
 181          * @param filter a module version bitmask, such as VF_COMMON or VF_OPTCOMMON
 182          */
 183         std::string MyModules(int filter);
 184
 185         /** Send my capabilities to the remote side
 186          */
 187         void SendCapabilities(int phase);
 188
 189         /** Add modules to VF_COMMON list for backwards compatability */
 190         void CompatAddModules(std::vector<std::string>& modlist);
 191
 192         /* Check a comma seperated list for an item */
 193         bool HasItem(const std::string &list, const std::string &item);
 194
 195         /* Isolate and return the elements that are different between two comma seperated lists */
 196         std::string ListDifference(const std::string &one, const std::string &two);
 197
 198         bool Capab(const parameterlist &params);
 199
 200         /** This function forces this server to quit, removing this server
 201          * and any users on it (and servers and users below that, etc etc).
 202          * It's very slow and pretty clunky, but luckily unless your network
 203          * is having a REAL bad hair day, this function shouldnt be called
 204          * too many times a month ;-)
 205          */
 206         void SquitServer(std::string &from, TreeServer* Current);
 207
 208         /** This is a wrapper function for SquitServer above, which
 209          * does some validation first and passes on the SQUIT to all
 210          * other remaining servers.
 211          */
 212         void Squit(TreeServer* Current, const std::string &reason);
 213
 214         /** FMODE command - server mode with timestamp checks */
 215         bool ForceMode(const std::string &source, parameterlist &params);
 216
 217         /** FTOPIC command */
 218         bool ForceTopic(const std::string &source, parameterlist &params);
 219
 220         /** FJOIN, similar to TS6 SJOIN, but not quite. */
 221         bool ForceJoin(const std::string &source, parameterlist &params);
 222
 223         /* Used on nick collision ... XXX ugly function HACK */
 224         int DoCollision(User *u, time_t remotets, const std::string &remoteident, const std::string &remoteip, const std::string &remoteuid);
 225
 226         /** UID command */
 227         bool ParseUID(const std::string &source, parameterlist &params);
 228
 229         /** Send one or more FJOINs for a channel of users.
 230          * If the length of a single line is more than 480-NICKMAX
 231          * in length, it is split over multiple lines.
 232          */
 233         void SendFJoins(TreeServer* Current, Channel* c);
 234
 235         /** Send G, Q, Z and E lines */
 236         void SendXLines(TreeServer* Current);
 237
 238         /** Send channel modes and topics */
 239         void SendChannelModes(TreeServer* Current);
 240
 241         /** send all users and their oper state/modes */
 242         void SendUsers(TreeServer* Current);
 243
 244         /** This function is called when we want to send a netburst to a local
 245          * server. There is a set order we must do this, because for example
 246          * users require their servers to exist, and channels require their
 247          * users to exist. You get the idea.
 248          */
 249         void DoBurst(TreeServer* s);
 250
 251         /** This function is called when we receive data from a remote
 252          * server.
 253          */
 254         void OnDataReady();
 255
 256         /** Send one or more complete lines down the socket
 257          */
 258         void WriteLine(std::string line);
 259
 260         /** Handle ERROR command */
 261         bool Error(parameterlist &params);
 262
 263         /** remote MOTD. */
 264         bool Motd(const std::string &prefix, parameterlist &params);
 265
 266         /** remote ADMIN. */
 267         bool Admin(const std::string &prefix, parameterlist &params);
 268
 269         bool Stats(const std::string &prefix, parameterlist &params);
 270
 271         /** Because the core won't let users or even SERVERS set +o,
 272          * we use the OPERTYPE command to do this.
 273          */
 274         bool OperType(const std::string &prefix, parameterlist &params);
 275
 276         /** Remote AWAY */
 277         bool Away(const std::string &prefix, parameterlist &params);
 278
 279         /** Because Andy insists that services-compatible servers must
 280          * implement SVSNICK and SVSJOIN, that's exactly what we do :p
 281          */
 282         bool SVSNick(const std::string &prefix, parameterlist &params);
 283
 284         /** SAVE to resolve nick collisions without killing */
 285         bool ForceNick(const std::string &prefix, parameterlist &params);
 286
 287         /** ENCAP command
 288          */
 289         bool Encap(const std::string &prefix, parameterlist &params);
 290
 291         /** OPERQUIT command
 292          */
 293         bool OperQuit(const std::string &prefix, parameterlist &params);
 294
 295         /** SVSJOIN
 296          */
 297         bool ServiceJoin(const std::string &prefix, parameterlist &params);
 298
 299         /** SVSPART
 300          */
 301         bool ServicePart(const std::string &prefix, parameterlist &params);
 302
 303         /** KILL
 304          */
 305         bool RemoteKill(const std::string &prefix, parameterlist &params);
 306
 307         /** PONG
 308          */
 309         bool LocalPong(const std::string &prefix, parameterlist &params);
 310
 311         /** METADATA
 312          */
 313         bool MetaData(const std::string &prefix, parameterlist &params);
 314
 315         /** VERSION
 316          */
 317         bool ServerVersion(const std::string &prefix, parameterlist &params);
 318
 319         /** CHGHOST
 320          */
 321         bool ChangeHost(const std::string &prefix, parameterlist &params);
 322
 323         /** ADDLINE
 324          */
 325         bool AddLine(const std::string &prefix, parameterlist &params);
 326
 327         /** DELLINE
 328          */
 329         bool DelLine(const std::string &prefix, parameterlist &params);
 330
 331         /** CHGNAME
 332          */
 333         bool ChangeName(const std::string &prefix, parameterlist &params);
 334
 335         /** FIDENT */
 336         bool ChangeIdent(const std::string &prefix, parameterlist &params);
 337
 338         /** WHOIS
 339          */
 340         bool Whois(const std::string &prefix, parameterlist &params);
 341
 342         /** PUSH
 343          */
 344         bool Push(const std::string &prefix, parameterlist &params);
 345
 346         /** TIME
 347          */
 348         bool Time(const std::string &prefix, parameterlist &params);
 349
 350         /** PING
 351          */
 352         bool LocalPing(const std::string &prefix, parameterlist &params);
 353
 354         /** Remove all modes from a channel, including statusmodes (+qaovh etc), simplemodes, parameter modes.
 355          * This does not update the timestamp of the target channel, this must be done seperately.
 356          */
 357         bool RemoveStatus(const std::string &prefix, parameterlist &params);
 358
 359         /** <- (remote) <- SERVER
 360          */
 361         bool RemoteServer(const std::string &prefix, parameterlist &params);
 362
 363         /** (local) -> SERVER
 364          */
 365         bool Outbound_Reply_Server(parameterlist &params);
 366
 367         /** (local) <- SERVER
 368          */
 369         bool Inbound_Server(parameterlist &params);
 370
 371         /** Handle IRC line split
 372          */
 373         void Split(const std::string &line, std::string& prefix, std::string& command, parameterlist &params);
 374
 375         /** Process complete line from buffer
 376          */
 377         void ProcessLine(std::string &line);
 378
 379         void ProcessConnectedLine(std::string& prefix, std::string& command, parameterlist& params);
 380
 381         /** Get this server's name
 382          */
 383         virtual std::string GetName();
 384
 385         /** Handle socket timeout from connect()
 386          */
 387         virtual void OnTimeout();
 388         /** Handle server quit on close
 389          */
 390         virtual void Close();
 391 };
 392
 393 /* Used to validate the value lengths of multiple parameters for a command */
 394 struct cmd_validation
 395 {
 396         const char* item;
 397         size_t param;
 398         size_t length;
 399 };
 400
 401 /* Used to validate the length values in CAPAB CAPABILITIES */
 402 struct cap_validation
 403 {
 404         const char* reason;
 405         const char* key;
 406         size_t size;
 407 };
 408
 409 #endif
 410