src/modules/m_spanningtree/treeserver.h

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
   5  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
   6  *   Copyright (C) 2007 Craig Edwards <craigedwards@brainbox.cc>
   7  *
   8  * This file is part of InspIRCd.  InspIRCd is free software: you can
   9  * redistribute it and/or modify it under the terms of the GNU General Public
  10  * License as published by the Free Software Foundation, version 2.
  11  *
  12  * This program is distributed in the hope that it will be useful, but WITHOUT
  13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  14  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  15  * details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21
  22 #pragma once
  23
  24 #include "treesocket.h"
  25
  26 /** Each server in the tree is represented by one class of
  27  * type TreeServer. A locally connected TreeServer can
  28  * have a class of type TreeSocket associated with it, for
  29  * remote servers, the TreeSocket entry will be NULL.
  30  * Each server also maintains a pointer to its parent
  31  * (NULL if this server is ours, at the top of the tree)
  32  * and a pointer to its "Route" (see the comments in the
  33  * constructors below), and also a dynamic list of pointers
  34  * to its children which can be iterated recursively
  35  * if required. Creating or deleting objects of type
  36  i* TreeServer automatically maintains the hash_map of
  37  * TreeServer items, deleting and inserting them as they
  38  * are created and destroyed.
  39  */
  40 class TreeServer : public Server
  41 {
  42         TreeServer* Parent;                     /* Parent entry */
  43         TreeServer* Route;                      /* Route entry */
  44         std::vector<TreeServer*> Children;      /* List of child objects */
  45         std::string VersionString;              /* Version string or empty string */
  46
  47         /** Full version string including patch version and other info
  48          */
  49         std::string fullversion;
  50
  51         TreeSocket* Socket;                     /* Socket used to communicate with this server */
  52         time_t NextPing;                        /* After this time, the server should be PINGed*/
  53         bool LastPingWasGood;                   /* True if the server responded to the last PING with a PONG */
  54         std::string sid;                        /* Server ID */
  55
  56         /** Counter counting how many servers are bursting in front of this server, including
  57          * this server. Set to parents' value on construction then it is increased if the
  58          * server itself starts bursting. Decreased when a server on the path to this server
  59          * finishes burst.
  60          */
  61         unsigned int behind_bursting;
  62
  63         /** This method is used to add this TreeServer to the
  64          * hash maps. It is only called by the constructors.
  65          */
  66         void AddHashEntry();
  67
  68         /** Used by SQuit logic to recursively remove servers
  69          */
  70         void SQuitInternal(const std::string& reason, int& num_lost_servers, int& num_lost_users);
  71
  72  public:
  73         typedef std::vector<TreeServer*> ChildServers;
  74         FakeUser* const ServerUser;             /* User representing this server */
  75         const time_t age;
  76
  77         bool Warned;                            /* True if we've warned opers about high latency on this server */
  78
  79         unsigned int UserCount;                 /* How many users are on this server? [note: doesn't care about +i] */
  80         unsigned int OperCount;                 /* How many opers are on this server? */
  81
  82         /** We use this constructor only to create the 'root' item, Utils->TreeRoot, which
  83          * represents our own server. Therefore, it has no route, no parent, and
  84          * no socket associated with it. Its version string is our own local version.
  85          */
  86         TreeServer();
  87
  88         /** When we create a new server, we call this constructor to initialize it.
  89          * This constructor initializes the server's Route and Parent, and sets up
  90          * its ping counters so that it will be pinged one minute from now.
  91          */
  92         TreeServer(const std::string& Name, const std::string& Desc, const std::string& id, TreeServer* Above, TreeSocket* Sock, bool Hide);
  93
  94         /** SQuit a server connected to this server, removing the given server and all servers behind it
  95          * @param server Server to squit, must be directly below this server
  96          * @param reason Reason for quitting the server, sent to opers and other servers
  97          */
  98         void SQuitChild(TreeServer* server, const std::string& reason);
  99
 100         /** SQuit this server, removing this server and all servers behind it
 101          * @param reason Reason for quitting the server, sent to opers and other servers
 102          */
 103         void SQuit(const std::string& reason)
 104         {
 105                 GetParent()->SQuitChild(this, reason);
 106         }
 107
 108         int QuitUsers(const std::string &reason);
 109
 110         /** Get route.
 111          * The 'route' is defined as the locally-
 112          * connected server which can be used to reach this server.
 113          */
 114         TreeServer* GetRoute();
 115
 116         /** Returns true if this server is the tree root (i.e.: us)
 117          */
 118         bool IsRoot() const { return (this->Parent == NULL); }
 119
 120         /** Returns true if this server is locally connected
 121          */
 122         bool IsLocal() const { return (this->Route == this); }
 123
 124         /** Get server version string
 125          */
 126         const std::string& GetVersion();
 127
 128         /** Get the full version string of this server
 129          * @return The full version string of this server, including patch version and other info
 130          */
 131         const std::string& GetFullVersion() const { return fullversion; }
 132
 133         /** Set time we are next due to ping this server
 134          */
 135         void SetNextPingTime(time_t t);
 136
 137         /** Get the time we are next due to ping this server
 138          */
 139         time_t NextPingTime();
 140
 141         /** Last ping time in milliseconds, used to calculate round trip time
 142          */
 143         unsigned long LastPingMsec;
 144
 145         /** Round trip time of last ping
 146          */
 147         unsigned long rtt;
 148
 149         /** When we recieved BURST from this server, used to calculate total burst time at ENDBURST.
 150          */
 151         unsigned long StartBurst;
 152
 153         /** True if this server is hidden
 154          */
 155         bool Hidden;
 156
 157         /** True if the server answered their last ping
 158          */
 159         bool AnsweredLastPing();
 160
 161         /** Set the server as responding to its last ping
 162          */
 163         void SetPingFlag();
 164
 165         /** Get the TreeSocket pointer for local servers.
 166          * For remote servers, this returns NULL.
 167          */
 168         TreeSocket* GetSocket();
 169
 170         /** Get the parent server.
 171          * For the root node, this returns NULL.
 172          */
 173         TreeServer* GetParent();
 174
 175         /** Set the server version string
 176          */
 177         void SetVersion(const std::string &Version);
 178
 179         /** Set the full version string
 180          * @param verstr The version string to set
 181          */
 182         void SetFullVersion(const std::string& verstr) { fullversion = verstr; }
 183
 184         /** Sets the description of this server. Called when the description of a remote server changes
 185          * and we are notified about it.
 186          * @param descstr The description to set
 187          */
 188         void SetDesc(const std::string& descstr) { description = descstr; }
 189
 190         /** Return all child servers
 191          */
 192         const ChildServers& GetChildren() const { return Children; }
 193
 194         /** Add a child server
 195          */
 196         void AddChild(TreeServer* Child);
 197
 198         /** Delete a child server, return false if it didn't exist.
 199          */
 200         bool DelChild(TreeServer* Child);
 201
 202         /** Removes child nodes of this node, and of that node, etc etc.
 203          * This is used during netsplits to automatically tidy up the
 204          * server tree. It is slow, we don't use it for much else.
 205          */
 206         void Tidy();
 207
 208         /** Get server ID
 209          */
 210         const std::string& GetID();
 211
 212         /** Marks a server as having finished bursting and performs appropriate actions.
 213          */
 214         void FinishBurst();
 215         /** Recursive call for child servers */
 216         void FinishBurstInternal();
 217
 218         /** (Re)check the uline state of this server
 219          */
 220         void CheckULine();
 221
 222         /** Get the bursting state of this server
 223          * @return True if this server is bursting, false if it isn't
 224          */
 225         bool IsBursting() const { return (StartBurst != 0); }
 226
 227         /** Check whether this server is behind a bursting server or is itself bursting.
 228          * This can tell whether a user is on a part of the network that is still bursting.
 229          * @return True if this server is bursting or is behind a server that is bursting, false if it isn't
 230          */
 231         bool IsBehindBursting() const { return (behind_bursting != 0); }
 232
 233         /** Set the bursting state of the server
 234          * @param startms Time the server started bursting, if 0 or omitted, use current time
 235          */
 236         void BeginBurst(unsigned long startms = 0);
 237
 238         CullResult cull();
 239
 240         /** Destructor
 241          * Removes the reference to this object from the
 242          * hash maps.
 243          */
 244         ~TreeServer();
 245
 246         /** Returns the TreeServer the given user is connected to
 247          * @param user The user whose server to return
 248          * @return The TreeServer this user is connected to.
 249          */
 250         static TreeServer* Get(User* user)
 251         {
 252                 return static_cast<TreeServer*>(user->server);
 253         }
 254 };