release socks on shutdown

2 files changed, 20 insertions, 1 deletions

diff --git a/src/lbdata.cc b/src/lbdata.cc
index 4dc8d76..a2666e7 100644
--- a/src/lbdata.cc
+++ b/src/lbdata.cc
@@ -398,3 +398,22 @@ LBTest::init()
     syslog(LOG_ERR, "wan_lb: failed to bind recv sock");
   }
 }
+
+LBTest::~LBTest()
+{
+  if (_recv_icmp_sock != 0) {
+    close(_recv_icmp_sock);
+    _recv_icmp_sock = 0;
+  }
+
+  if (_send_raw_sock != 0) {
+    close(_send_raw_sock);
+    _send_raw_sock = 0;
+  }
+
+  if (_send_icmp_sock != 0) {
+    close(_send_icmp_sock);
+    _send_icmp_sock = 0;
+  }
+
+}
diff --git a/src/lbdata.hh b/src/lbdata.hh
index 4444009..a901653 100644
--- a/src/lbdata.hh
+++ b/src/lbdata.hh
@@ -103,7 +103,7 @@ public:
     _debug(debug),
     _state(K_NONE)
   {}
-  virtual ~LBTest() {}
+  virtual ~LBTest();
 
   virtual void
   init();