src/modules/m_spanningtree/treesocket.h

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
   6  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
   7  *   Copyright (C) 2007 Craig Edwards <craigedwards@brainbox.cc>
   8  *
   9  * This file is part of InspIRCd.  InspIRCd is free software: you can
  10  * redistribute it and/or modify it under the terms of the GNU General Public
  11  * License as published by the Free Software Foundation, version 2.
  12  *
  13  * This program is distributed in the hope that it will be useful, but WITHOUT
  14  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  15  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  16  * details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20  */
  21
  22
  23 #ifndef M_SPANNINGTREE_TREESOCKET_H
  24 #define M_SPANNINGTREE_TREESOCKET_H
  25
  26 #include "socket.h"
  27 #include "inspircd.h"
  28 #include "xline.h"
  29
  30 #include "utils.h"
  31
  32 /*
  33  * The server list in InspIRCd is maintained as two structures
  34  * which hold the data in different ways. Most of the time, we
  35  * want to very quicky obtain three pieces of information:
  36  *
  37  * (1) The information on a server
  38  * (2) The information on the server we must send data through
  39  *     to actually REACH the server we're after
  40  * (3) Potentially, the child/parent objects of this server
  41  *
  42  * The InspIRCd spanning protocol provides easy access to these
  43  * by storing the data firstly in a recursive structure, where
  44  * each item references its parent item, and a dynamic list
  45  * of child items, and another structure which stores the items
  46  * hashed, linearly. This means that if we want to find a server
  47  * by name quickly, we can look it up in the hash, avoiding
  48  * any O(n) lookups. If however, during a split or sync, we want
  49  * to apply an operation to a server, and any of its child objects
  50  * we can resort to recursion to walk the tree structure.
  51  * Any socket can have one of five states at any one time.
  52  *
  53  * CONNECTING:  indicates an outbound socket which is
  54  *                                                      waiting to be writeable.
  55  * WAIT_AUTH_1: indicates the socket is outbound and
  56  *                                                      has successfully connected, but has not
  57  *                                                      yet sent and received SERVER strings.
  58  * WAIT_AUTH_2: indicates that the socket is inbound
  59  *                                                      but has not yet sent and received
  60  *                                                      SERVER strings.
  61  * CONNECTED:   represents a fully authorized, fully
  62  *                                                      connected server.
  63  * DYING:       represents a server that has had an error.
  64  */
  65 enum ServerState { CONNECTING, WAIT_AUTH_1, WAIT_AUTH_2, CONNECTED, DYING };
  66
  67 struct CapabData
  68 {
  69         reference<Link> link;                   /* Link block used for this connection */
  70         reference<Autoconnect> ac;              /* Autoconnect used to cause this connection, if any */
  71         std::string ModuleList;                 /* Required module list of other server from CAPAB */
  72         std::string OptModuleList;              /* Optional module list of other server from CAPAB */
  73         std::string ChanModes;
  74         std::string UserModes;
  75         std::map<std::string,std::string> CapKeys;      /* CAPAB keys from other server */
  76         std::string ourchallenge;               /* Challenge sent for challenge/response */
  77         std::string theirchallenge;             /* Challenge recv for challenge/response */
  78         int capab_phase;                        /* Have sent CAPAB already */
  79         bool auth_fingerprint;                  /* Did we auth using SSL fingerprint */
  80         bool auth_challenge;                    /* Did we auth using challenge/response */
  81 };
  82
  83 /** Every SERVER connection inbound or outbound is represented by an object of
  84  * type TreeSocket. During setup, the object can be found in Utils->timeoutlist;
  85  * after setup, MyRoot will have been created as a child of Utils->TreeRoot
  86  */
  87 class TreeSocket : public BufferedSocket
  88 {
  89         SpanningTreeUtilities* Utils;           /* Utility class */
  90         std::string linkID;                     /* Description for this link */
  91         ServerState LinkState;                  /* Link state */
  92         CapabData* capab;                       /* Link setup data (held until burst is sent) */
  93         TreeServer* MyRoot;                     /* The server we are talking to */
  94         time_t NextPing;                        /* Time when we are due to ping this server */
  95         bool LastPingWasGood;                   /* Responded to last ping we sent? */
  96         int proto_version;                      /* Remote protocol version */
  97         bool ConnectionFailureShown; /* Set to true if a connection failure message was shown */
  98  public:
  99         time_t age;
 100
 101         /** Because most of the I/O gubbins are encapsulated within
 102          * BufferedSocket, we just call the superclass constructor for
 103          * most of the action, and append a few of our own values
 104          * to it.
 105          */
 106         TreeSocket(SpanningTreeUtilities* Util, Link* link, Autoconnect* myac, const std::string& ipaddr);
 107
 108         /** When a listening socket gives us a new file descriptor,
 109          * we must associate it with a socket without creating a new
 110          * connection. This constructor is used for this purpose.
 111          */
 112         TreeSocket(SpanningTreeUtilities* Util, int newfd, ListenSocket* via, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server);
 113
 114         /** Get link state
 115          */
 116         ServerState GetLinkState();
 117
 118         /** Get challenge set in our CAPAB for challenge/response
 119          */
 120         const std::string& GetOurChallenge();
 121
 122         /** Get challenge set in our CAPAB for challenge/response
 123          */
 124         void SetOurChallenge(const std::string &c);
 125
 126         /** Get challenge set in their CAPAB for challenge/response
 127          */
 128         const std::string& GetTheirChallenge();
 129
 130         /** Get challenge set in their CAPAB for challenge/response
 131          */
 132         void SetTheirChallenge(const std::string &c);
 133
 134         /** Compare two passwords based on authentication scheme
 135          */
 136         bool ComparePass(const Link& link, const std::string &theirs);
 137
 138         /** Clean up information used only during server negotiation
 139          */
 140         void CleanNegotiationInfo();
 141
 142         CullResult cull();
 143         /** Destructor
 144          */
 145         ~TreeSocket();
 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         /* Isolate and return the elements that are different between two lists */
 193         void ListDifference(const std::string &one, const std::string &two, char sep,
 194                 std::string& mleft, std::string& mright);
 195
 196         bool Capab(const parameterlist &params);
 197
 198         /** This function forces this server to quit, removing this server
 199          * and any users on it (and servers and users below that, etc etc).
 200          * It's very slow and pretty clunky, but luckily unless your network
 201          * is having a REAL bad hair day, this function shouldnt be called
 202          * too many times a month ;-)
 203          */
 204         void SquitServer(std::string &from, TreeServer* Current, int& num_lost_servers, int& num_lost_users);
 205
 206         /** This is a wrapper function for SquitServer above, which
 207          * does some validation first and passes on the SQUIT to all
 208          * other remaining servers.
 209          */
 210         void Squit(TreeServer* Current, const std::string &reason);
 211
 212         /* Used on nick collision ... XXX ugly function HACK */
 213         int DoCollision(User *u, time_t remotets, const std::string &remoteident, const std::string &remoteip, const std::string &remoteuid);
 214
 215         /** Send one or more FJOINs for a channel of users.
 216          * If the length of a single line is more than 480-NICKMAX
 217          * in length, it is split over multiple lines.
 218          */
 219         void SendFJoins(Channel* c);
 220
 221         /** Send G, Q, Z and E lines */
 222         void SendXLines();
 223
 224         /** Send channel modes and topics */
 225         void SendChannelModes();
 226
 227         /** send all users and their oper state/modes */
 228         void SendUsers();
 229
 230         /** This function is called when we want to send a netburst to a local
 231          * server. There is a set order we must do this, because for example
 232          * users require their servers to exist, and channels require their
 233          * users to exist. You get the idea.
 234          */
 235         void DoBurst(TreeServer* s);
 236
 237         /** This function is called when we receive data from a remote
 238          * server.
 239          */
 240         void OnDataReady();
 241
 242         /** Send one or more complete lines down the socket
 243          */
 244         void WriteLine(std::string line);
 245
 246         /** Handle ERROR command */
 247         void Error(parameterlist &params);
 248
 249         /** Remote AWAY */
 250         bool Away(const std::string &prefix, parameterlist &params);
 251
 252         /** SAVE to resolve nick collisions without killing */
 253         bool ForceNick(const std::string &prefix, parameterlist &params);
 254
 255         /** ENCAP command
 256          */
 257         void Encap(User* who, parameterlist &params);
 258
 259         /** OPERQUIT command
 260          */
 261         bool OperQuit(const std::string &prefix, parameterlist &params);
 262
 263         /** PONG
 264          */
 265         bool LocalPong(const std::string &prefix, parameterlist &params);
 266
 267         /** VERSION
 268          */
 269         bool ServerVersion(const std::string &prefix, parameterlist &params);
 270
 271         /** ADDLINE
 272          */
 273         bool AddLine(const std::string &prefix, parameterlist &params);
 274
 275         /** DELLINE
 276          */
 277         bool DelLine(const std::string &prefix, parameterlist &params);
 278
 279         /** WHOIS
 280          */
 281         bool Whois(const std::string &prefix, parameterlist &params);
 282
 283         /** PUSH
 284          */
 285         bool Push(const std::string &prefix, parameterlist &params);
 286
 287         /** PING
 288          */
 289         bool LocalPing(const std::string &prefix, parameterlist &params);
 290
 291         /** <- (remote) <- SERVER
 292          */
 293         bool RemoteServer(const std::string &prefix, parameterlist &params);
 294
 295         /** (local) -> SERVER
 296          */
 297         bool Outbound_Reply_Server(parameterlist &params);
 298
 299         /** (local) <- SERVER
 300          */
 301         bool Inbound_Server(parameterlist &params);
 302
 303         /** Handle IRC line split
 304          */
 305         void Split(const std::string &line, std::string& prefix, std::string& command, parameterlist &params);
 306
 307         /** Process complete line from buffer
 308          */
 309         void ProcessLine(std::string &line);
 310
 311         void ProcessConnectedLine(std::string& prefix, std::string& command, parameterlist& params);
 312
 313         /** Handle socket timeout from connect()
 314          */
 315         virtual void OnTimeout();
 316         /** Handle server quit on close
 317          */
 318         virtual void Close();
 319
 320         /** Returns true if this server was introduced to the rest of the network
 321          */
 322         bool Introduced();
 323 };
 324
 325 #endif
 326