X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_spanningtree%2Ftreeserver.h;h=ab47012b0870d7cbce55e865041ba014fc58d732;hb=c8f92d97c2cd92a07bbb76b96a67cd089c7e3407;hp=faaa47aa100d882caada2501fac152f2431a8387;hpb=5dcd385b772f2ce822cf0700912ced93bd06c914;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/src/modules/m_spanningtree/treeserver.h b/src/modules/m_spanningtree/treeserver.h
index faaa47aa1..ab47012b0 100644
--- a/src/modules/m_spanningtree/treeserver.h
+++ b/src/modules/m_spanningtree/treeserver.h
@@ -37,15 +37,13 @@
  * TreeServer items, deleting and inserting them as they
  * are created and destroyed.
  */
-class TreeServer : public classbase
+class TreeServer : public Server
 {
 	TreeServer* Parent;			/* Parent entry */
 	TreeServer* Route;			/* Route entry */
 	std::vector<TreeServer*> Children;	/* List of child objects */
-	std::string ServerName;			/* Server's name */
-	std::string ServerDesc;			/* Server's description */
 	std::string VersionString;		/* Version string or empty string */
-	TreeSocket* Socket;			/* For directly connected servers this points at the socket object */
+	TreeSocket* Socket;			/* Socket used to communicate with this server */
 	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::string sid;			/* Server ID */
@@ -56,6 +54,7 @@ class TreeServer : public classbase
 	void AddHashEntry();
 
  public:
+	typedef std::vector<TreeServer*> ChildServers;
 	FakeUser* const ServerUser;		/* User representing this server */
 	const time_t age;
 
@@ -85,13 +84,13 @@ class TreeServer : public classbase
 	 */
 	TreeServer* GetRoute();
 
-	/** Get server name
+	/** Returns true if this server is the tree root (i.e.: us)
 	 */
-	const std::string& GetName() const { return ServerName; }
+	bool IsRoot() const { return (this->Parent == NULL); }
 
-	/** Get server description (GECOS)
+	/** Returns true if this server is locally connected
 	 */
-	const std::string& GetDesc();
+	bool IsLocal() const { return (this->Route == this); }
 
 	/** Get server version string
 	 */
@@ -143,13 +142,9 @@ class TreeServer : public classbase
 	 */
 	void SetVersion(const std::string &Version);
 
-	/** Return number of child servers
+	/** Return all child servers
 	 */
-	unsigned int ChildCount();
-
-	/** Return a child server indexed 0..n
-	 */
-	TreeServer* GetChild(unsigned int n);
+	const ChildServers& GetChildren() const { return Children; }
 
 	/** Add a child server
 	 */
@@ -175,6 +170,10 @@ class TreeServer : public classbase
 	/** Recursive call for child servers */
 	void FinishBurstInternal();
 
+	/** (Re)check the uline state of this server
+	 */
+	void CheckULine();
+
 	CullResult cull();
 
 	/** Destructor
@@ -182,4 +181,13 @@ class TreeServer : public classbase
 	 * hash maps.
 	 */
 	~TreeServer();
+
+	/** Returns the TreeServer the given user is connected to
+	 * @param user The user whose server to return
+	 * @return The TreeServer this user is connected to.
+	 */
+	static TreeServer* Get(User* user)
+	{
+		return static_cast<TreeServer*>(user->server);
+	}
 };