Add whether the server split with an error to OnServerSplit.

1 files changed, 6 insertions, 4 deletions

diff --git a/src/modules/m_spanningtree/treeserver.h b/src/modules/m_spanningtree/treeserver.h
index 90841d0b4..14413cd92 100644
--- a/src/modules/m_spanningtree/treeserver.h
+++ b/src/modules/m_spanningtree/treeserver.h
@@ -74,7 +74,7 @@ class TreeServer : public Server
 
 	/** Used by SQuit logic to recursively remove servers
 	 */
-	void SQuitInternal(unsigned int& num_lost_servers);
+	void SQuitInternal(unsigned int& num_lost_servers, bool error);
 
 	/** Remove the reference to this server from the hash maps
 	 */
@@ -103,15 +103,17 @@ class TreeServer : public Server
 	/** SQuit a server connected to this server, removing the given server and all servers behind it
 	 * @param server Server to squit, must be directly below this server
 	 * @param reason Reason for quitting the server, sent to opers and other servers
+	 * @param error Whether the server is being squit because of an error.
 	 */
-	void SQuitChild(TreeServer* server, const std::string& reason);
+	void SQuitChild(TreeServer* server, const std::string& reason, bool error = false);
 
 	/** SQuit this server, removing this server and all servers behind it
 	 * @param reason Reason for quitting the server, sent to opers and other servers
+	 * @param error Whether the server is being squit because of an error.
 	 */
-	void SQuit(const std::string& reason)
+	void SQuit(const std::string& reason, bool error = false)
 	{
-		GetParent()->SQuitChild(this, reason);
+		GetParent()->SQuitChild(this, reason, error);
 	}
 
 	static unsigned int QuitUsers(const std::string& reason);