From f3e11bfc29692a8abec7ec112ac5bcb66fd1ae55 Mon Sep 17 00:00:00 2001
From: Simon Rasmussen <zetlon@gmail.com>
Date: Fri, 7 Jul 2017 19:15:17 +0200
Subject: [PATCH] minor improvements

---
 include/ur_modern_driver/ur/server.h | 1 +
 src/ur/server.cpp                    | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/include/ur_modern_driver/ur/server.h b/include/ur_modern_driver/ur/server.h
index daff4d3..027e278 100644
--- a/include/ur_modern_driver/ur/server.h
+++ b/include/ur_modern_driver/ur/server.h
@@ -24,6 +24,7 @@ protected:
 
 public:
   URServer(int port);
+  ~URServer();
   std::string getIP();
   bool bind();
   bool accept();
diff --git a/src/ur/server.cpp b/src/ur/server.cpp
index 1d53ed4..5530be9 100644
--- a/src/ur/server.cpp
+++ b/src/ur/server.cpp
@@ -10,6 +10,11 @@ URServer::URServer(int port)
 {
 }
 
+URServer::~URServer()
+{
+  TCPSocket::close();
+}
+
 void URServer::setOptions(int socket_fd)
 {
   TCPSocket::setOptions(socket_fd);
@@ -61,6 +66,8 @@ bool URServer::accept()
   if(client_fd <= 0)
     return false;
 
+  setOptions(client_fd);
+
   return client_.setSocketFD(client_fd);
 }