From 7a66d48f815b475d98e56a99a5125ba3b8592e8d Mon Sep 17 00:00:00 2001
From: "Matthias P. Braendli" <matthias.braendli@mpb.li>
Date: Wed, 21 Sep 2016 19:37:38 +0200
Subject: Add TCP input recv timeout of 30s

---
 src/InputTcpReader.cpp | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'src')

diff --git a/src/InputTcpReader.cpp b/src/InputTcpReader.cpp
index 34ccb2e..94ec0ad 100644
--- a/src/InputTcpReader.cpp
+++ b/src/InputTcpReader.cpp
@@ -32,6 +32,9 @@
 #include "InputReader.h"
 #include "PcDebug.h"
 #include "Utils.h"
+#include <sys/socket.h>
+#include <unistd.h>
+#include <errno.h>
 
 InputTcpReader::InputTcpReader()
 {
@@ -88,6 +91,15 @@ void InputTcpReader::Open(const std::string& endpoint)
         throw std::runtime_error(ss.str());
     }
 
+    /* Set recv timeout of 30s, see socket(7) for details */
+    struct timeval tv = {0};
+    tv.tv_sec = 30;
+    if (setsockopt(m_sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(struct timeval)) == -1) {
+        std::stringstream ss;
+        ss << "Could not set socket recv timeout: " << strerror(errno);
+        throw std::runtime_error(ss.str());
+    }
+
     if (connect(m_sock, (struct sockaddr*)&addr, sizeof(addr)) == -1) {
         std::stringstream ss;
         ss << "Could not connect to " << hostname << ":" << port << " :" << strerror(errno);
-- 
cgit v1.2.3