From 5df4afd509303deb0f26e33628e68a6e1ac3a920 Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Wed, 15 Apr 2020 13:01:38 +0200
Subject: [PATCH] Finalize UDP Recv implementation

---
 core-writer/include/UdpReceiver.hpp      |  2 +-
 core-writer/src/receiver/UdpReceiver.cpp | 16 ++++++++++++++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/core-writer/include/UdpReceiver.hpp b/core-writer/include/UdpReceiver.hpp
index d5c6cf8..b077184 100644
--- a/core-writer/include/UdpReceiver.hpp
+++ b/core-writer/include/UdpReceiver.hpp
@@ -6,7 +6,7 @@
 
 class UdpReceiver {
 
-    int socket_fd_ = -1;
+    int socket_fd_;
 
 public:
     UdpReceiver();
diff --git a/core-writer/src/receiver/UdpReceiver.cpp b/core-writer/src/receiver/UdpReceiver.cpp
index 1d80e96..a1c10f7 100644
--- a/core-writer/src/receiver/UdpReceiver.cpp
+++ b/core-writer/src/receiver/UdpReceiver.cpp
@@ -5,6 +5,16 @@
 
 using namespace std;
 
+UdpReceiver::UdpReceiver() :
+    socket_fd_(-1)
+{
+}
+
+UdpReceiver::~UdpReceiver()
+{
+    close();
+}
+
 void UdpReceiver::bind(const uint16_t port, const size_t usec_timeout)
 {
     if (socket_fd_ > -1) {
@@ -56,4 +66,10 @@ bool UdpReceiver::receive(void* buffer, size_t buffer_n_bytes)
     }
 
     return true;
+}
+
+void UdpReceiver::close()
+{
+    ::close(socket_fd_);
+    socket_fd_ = -1;
 }
\ No newline at end of file