Allow shutting down of TcpServers

Sming/SmingCore/Network/HttpServer.cpp

@@ -15,18 +15,19 @@
 #include "TcpClient.h"
 #include "../Wiring/WString.h"
 
-HttpServer::HttpServer()
+HttpServer::HttpServer(): active(true)
 {
 	settings.keepAliveSeconds = 0;
 	configure(settings);
 }
 
-HttpServer::HttpServer(HttpServerSettings settings)
+HttpServer::HttpServer(HttpServerSettings settings): active(true)
 {
 	configure(settings);
 }
 
-void HttpServer::configure(HttpServerSettings settings) {
+void HttpServer::configure(HttpServerSettings settings)
+{
 	this->settings = settings;
 	if(settings.minHeapSize != -1 && settings.minHeapSize > -1) {
 		minHeapSize = settings.minHeapSize;
@@ -44,6 +45,7 @@ void HttpServer::configure(HttpServerSettings settings) {
 
 HttpServer::~HttpServer()
 {
+	active = true;
 	for(int i=0; i< resourceTree.count(); i++) {
 		if(resourceTree.valueAt(i) != NULL) {
 			delete resourceTree.valueAt(i);
@@ -58,12 +60,18 @@ void HttpServer::setBodyParser(const String& contentType, HttpBodyParserDelegate
 
 TcpConnection* HttpServer::createClient(tcp_pcb *clientTcp)
 {
+	if(!active) {
+		debugf("Refusing new connections. The server is shutting down");
+		return NULL;
+	}
+
 	HttpServerConnection* con = new HttpServerConnection(clientTcp);
 	con->setResourceTree(&resourceTree);
 	con->setBodyParsers(&bodyParsers);
 	con->setCompleteDelegate(TcpClientCompleteDelegate(&HttpServer::onConnectionClose, this));
 
-	totalConnections++;
+	connections.add(con);
+	totalConnections = connections.count();
 	debugf("Opening connection. Total connections: %d", totalConnections);
 
 	return con;
@@ -85,21 +93,43 @@ void HttpServer::setDefaultHandler(const HttpPathDelegate& callback)
 	addPath("*", callback);
 }
 
-void HttpServer::addPath(const String& path, const HttpResourceDelegate& onRequestComplete) {
+void HttpServer::addPath(const String& path, const HttpResourceDelegate& onRequestComplete)
+{
 	HttpResource* resource = new HttpResource;
 	resource->onRequestComplete = onRequestComplete;
 	resourceTree[path] = resource;
 }
 
-void HttpServer::addPath(const String& path, HttpResource* resource) {
+void HttpServer::addPath(const String& path, HttpResource* resource)
+{
 	resourceTree[path] = resource;
 }
 
-void HttpServer::setDefaultResource(HttpResource* resource) {
+void HttpServer::setDefaultResource(HttpResource* resource)
+{
 	addPath("*", resource);
 }
 
-void HttpServer::onConnectionClose(TcpClient& connection, bool success) {
-	totalConnections--;
+void HttpServer::shutdown()
+{
+	active = false;
+	for(int i=0; i < connections.count(); i++) {
+		HttpServerConnection* connection = connections[i];
+		if(connection == NULL) {
+			continue;
+		}
+
+		connection->close();
+	}
+}
+
+void HttpServer::onConnectionClose(TcpClient& connection, bool success)
+{
+	connections.removeElement((HttpServerConnection*)&connection);
+	totalConnections = connections.count();
+	if(totalConnections == 0 && !active){
+		debugf("Shutting down the Http Server");
+		delete this;
+	}
 	debugf("Closing connection. Total connections: %d", totalConnections);
 }

Sming/SmingCore/Network/HttpServer.h

@@ -77,6 +77,7 @@ class HttpServer: public TcpServer
 	void setDefaultHandler(const HttpPathDelegate& callback);
 	void setDefaultResource(HttpResource* resource);
 
+	void shutdown();
 
 protected:
 	virtual TcpConnection* createClient(tcp_pcb *clientTcp);
@@ -91,6 +92,9 @@ class HttpServer: public TcpServer
 	HttpServerSettings settings;
 	ResourceTree resourceTree;
 	BodyParsers bodyParsers;
+	bool active = true;
+
+	Vector<HttpServerConnection*> connections;
 };
 
 /** @} */

Sming/SmingCore/Network/TcpConnection.cpp

@@ -309,7 +309,9 @@ void TcpConnection::close()
 	axl_free(tcp);
 #endif
 
-	tcp_poll(tcp, staticOnPoll, 1);
+	if(tcp->state == LISTEN) {
+		tcp_poll(tcp, staticOnPoll, 1);
+	}
 	tcp_arg(tcp, NULL); // reset pointer to close connection on next callback
 	tcp = NULL;