From 69d683e8720996c571c11b6f90e008682d0f95fb Mon Sep 17 00:00:00 2001 From: "Matthias P. Braendli" Date: Wed, 28 Oct 2020 11:25:09 +0100 Subject: common 81f518d: TCPSocket: let recv throw Interrupted on EINTR --- contrib/Socket.cpp | 9 +++++++-- contrib/Socket.h | 6 +++--- 2 files changed, 10 insertions(+), 5 deletions(-) (limited to 'contrib') diff --git a/contrib/Socket.cpp b/contrib/Socket.cpp index bc1b179..d41ed1c 100644 --- a/contrib/Socket.cpp +++ b/contrib/Socket.cpp @@ -630,8 +630,13 @@ ssize_t TCPSocket::recv(void *buffer, size_t length, int flags) { ssize_t ret = ::recv(m_sock, buffer, length, flags); if (ret == -1) { - std::string errstr(strerror(errno)); - throw std::runtime_error("TCP receive error: " + errstr); + if (errno == EINTR) { + throw Interrupted(); + } + else { + std::string errstr(strerror(errno)); + throw std::runtime_error("TCP receive error: " + errstr); + } } return ret; } diff --git a/contrib/Socket.h b/contrib/Socket.h index 8c6f8a9..8881be3 100644 --- a/contrib/Socket.h +++ b/contrib/Socket.h @@ -180,12 +180,12 @@ class TCPSocket { */ ssize_t send(const void* data, size_t size, int timeout_ms=0); - /* Returns number of bytes read, 0 on disconnect. Throws a - * runtime_error on error */ + class Interrupted {}; + /* Returns number of bytes read, 0 on disconnect. + * Throws Interrupted on EINTR, runtime_error on error */ ssize_t recv(void *buffer, size_t length, int flags); class Timeout {}; - class Interrupted {}; /* Returns number of bytes read, 0 on disconnect or refused connection. * Throws a Timeout on timeout, Interrupted on EINTR, a runtime_error * on error -- cgit v1.2.3