Merge pull request #47 from Auterion/tcp-client-timeout

Add a timeout argument to TCPClient
Auterion · Jun 13, 2024 · 7d82dca · 7d82dca
2 parents e7544fb + 02e74b1
commit 7d82dca
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/include/mav/TCPClient.h b/include/mav/TCPClient.h
@@ -55,12 +55,19 @@ namespace mav {
 
     public:
 
-        TCPClient(const std::string& address, int port) {
+        TCPClient(const std::string& address, int port, int timeout = -1) {
             _socket = socket(AF_INET, SOCK_STREAM, 0);
             if (_socket < 0) {
                 throw NetworkError("Could not create socket", errno);
             }
 
+            if (timeout > 0) {
+                struct timeval send_timeout;
+                send_timeout.tv_sec = 0;
+                send_timeout.tv_usec = timeout * 1000; // timeout is in ms
+                setsockopt(_socket, SOL_SOCKET, SO_SNDTIMEO, &send_timeout, sizeof(send_timeout));
+            }
+
             struct hostent *hp;
             hp = gethostbyname(address.c_str());
             if (hp == nullptr) {