From 24d9b96a05099425a628dbe18c23b0b8565ec19b Mon Sep 17 00:00:00 2001
From: peter-murray <Bix85214>
Date: Tue, 1 Sep 2015 20:26:05 +0100
Subject: [PATCH] Fixes Issue #55, re-registering an exit listener on the
 process, all sockets are now racked and ensured that they are closed out via
 a single listener

---
 Changelog.md      |  4 ++++
 hue-api/search.js | 55 +++++++++++++++++++++++++++++++++++++++++------
 2 files changed, 53 insertions(+), 6 deletions(-)

diff --git a/Changelog.md b/Changelog.md
index aa50535..e02ae2d 100644
--- a/Changelog.md
+++ b/Changelog.md
@@ -1,5 +1,9 @@
 # Change Log
 
+## 1.1.2
+- Fixes issue #55, upnp searches re-register the exit handler repeatedly causing issues if you use upnp search multiple
+times during program execution
+
 ## 1.1.1
 - Fixes issue #52, respecting the LOCATION value from SSDP lookups
 
diff --git a/hue-api/search.js b/hue-api/search.js
index 0d01429..0ef3061 100644
--- a/hue-api/search.js
+++ b/hue-api/search.js
@@ -5,6 +5,8 @@ var Q = require("q"),
     EventEmitter = require('events').EventEmitter,
     util = require("util");
 
+var socketCleanUp = new SocketCleanUp();
+
 /**
  * Locates possible Philips Hue Bridges on the network.
  * @param timeout The maximum time to wait for responses, or if none provided will default to 5 seconds
@@ -49,12 +51,8 @@ function SSDPSearch(timeout) {
         }
     });
 
-    // Clean up if close is not called directly
-    process.on('exit', function () {
-        if (!self.closed) {
-            _close(self);
-        }
-    });
+    socketCleanUp.add(self);
+    socketCleanUp.registerExitListener();
 }
 SSDPSearch.prototype.__proto__ = EventEmitter.prototype;
 
@@ -79,6 +77,7 @@ function _close(target) {
         target.closed = true;
         target.socket.close();
     }
+    socketCleanUp.remove(target);
 }
 
 function _buildSearchPacket(vars) {
@@ -163,3 +162,47 @@ function _addUnique(array, value) {
         array.push(value);
     }
 }
+
+/**
+ * A socket register to take care of closing all sockets that are opened, if not already closed
+ *
+ * @constructor
+ */
+function SocketCleanUp() {
+    this._searches = [];
+    this._registered = false;
+}
+
+SocketCleanUp.prototype.add = function(search) {
+    this._searches.push(search);
+};
+
+SocketCleanUp.prototype.remove = function(search) {
+    var self = this
+        , searches = self._searches
+        , idx = searches.indexOf(search)
+        ;
+
+    if (idx > -1) {
+        searches.splice(idx, 1)
+    }
+};
+
+SocketCleanUp.prototype.finished = function() {
+    var self = this
+        , searches = self._searches
+        ;
+
+    searches.forEach(function(search) {
+        _close(search);
+    });
+};
+
+SocketCleanUp.prototype.registerExitListener = function() {
+    var self = this;
+
+    if (! self._registered) {
+        process.on("exit", self.finished.bind(self));
+        self._registered = true;
+    }
+};