Allow immediate server deletion if there are no active connections. (#…

…1285)
SmingHub · Nov 14, 2017 · 8fe3888 · 8fe3888
1 parent 286ef0e
commit 8fe3888
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Sming/SmingCore/Network/TcpServer.cpp b/Sming/SmingCore/Network/TcpServer.cpp
@@ -50,6 +50,7 @@ TcpServer::TcpServer(TcpClientDataDelegate clientReceiveDataHandler)
 
 TcpServer::~TcpServer()
 {
+	debugf("Server is destroyed.");
 }
 
 TcpConnection* TcpServer::createClient(tcp_pcb *clientTcp)
@@ -258,6 +259,11 @@ void TcpServer::shutdown()
 		tcp = NULL;
 	}
 
+	if(!connections.count()) {
+		delete this;
+		return;
+	}
+
 	for(int i=0; i < connections.count(); i++) {
 		TcpConnection* connection = connections[i];
 		if(connection == NULL) {
@@ -278,7 +284,6 @@ void TcpServer::onClientDestroy(TcpConnection& connection)
 	}
 
 	if(connections.count() == 0) {
-		debugf("Server is destroyed.");
 		delete this;
 	}
 }