From a831281b3d495bf7e3fa12b883f3f2aa3ea12e38 Mon Sep 17 00:00:00 2001
From: "Roger A. Light" <roger@atchoo.org>
Date: Wed, 2 Jul 2014 20:49:06 +0100
Subject: [PATCH] Delete clients after socket close.

---
 lib/net_mosq.c | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/lib/net_mosq.c b/lib/net_mosq.c
index 87eeee2d..47eed678 100644
--- a/lib/net_mosq.c
+++ b/lib/net_mosq.c
@@ -238,6 +238,14 @@ int _mosquitto_socket_close(struct mosquitto *mosq)
 #endif
 	}
 
+#ifdef WITH_WEBSOCKETS
+	if(!mosq->wsi){
+#endif
+		HASH_ADD_KEYPTR(hh_for_free, db->contexts_for_free, mosq, sizeof(void *), mosq);
+#ifdef WITH_WEBSOCKETS
+	}
+#endif
+
 #ifdef WITH_BROKER
 	if(mosq->listener){
 		mosq->listener->client_count--;