From: Lars Op den Kamp Date: Thu, 9 Feb 2012 14:36:59 +0000 (+0100) Subject: platform: fix tcp socket error handling and return value. X-Git-Tag: upstream/2.2.0~1^2~35^2~127 X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=3459c4f94455845e39fa60feb197442e0aea9abe;p=deb_libcec.git platform: fix tcp socket error handling and return value. --- diff --git a/src/lib/platform/windows/os-socket.h b/src/lib/platform/windows/os-socket.h index 60badd7..5174cba 100644 --- a/src/lib/platform/windows/os-socket.h +++ b/src/lib/platform/windows/os-socket.h @@ -147,7 +147,7 @@ namespace PLATFORM ssize_t iReturn = send(socket, (char*)data, (int)len, 0); if (iReturn < (ssize_t)len) - *iError = errno; + *iError = GetSocketError(); return iReturn; } @@ -184,19 +184,28 @@ namespace PLATFORM FD_SET(socket, &fd_read); if (select((int)socket + 1, &fd_read, NULL, NULL, &tv) == 0) + { + *iError = ETIMEDOUT; return ETIMEDOUT; + } TcpSocketSetBlocking(socket, false); } ssize_t iReadResult = (iTimeoutMs > 0) ? - recv(socket, (char*)data + iBytesRead, (int)(len - iBytesRead), MSG_WAITALL) : + recv(socket, (char*)data + iBytesRead, (int)(len - iBytesRead), 0) : recv(socket, (char*)data, (int)len, MSG_WAITALL); *iError = GetSocketError(); + + if (iTimeoutMs > 0) + { + TcpSocketSetBlocking(socket, true); + iNow = GetTimeMs(); + } + if (iReadResult < 0) { - if (errno == EAGAIN && iTimeoutMs > 0) + if (*iError == EAGAIN && iTimeoutMs > 0) continue; - *iError = errno; return -1; } else if (iReadResult == 0 || (iReadResult != (ssize_t)len && iTimeoutMs == 0)) @@ -206,14 +215,12 @@ namespace PLATFORM } iBytesRead += iReadResult; - - if (iTimeoutMs > 0) - { - TcpSocketSetBlocking(socket, true); - iNow = GetTimeMs(); - } } - return 0; + + if (iBytesRead < (ssize_t)len && *iError == 0) + *iError = ETIMEDOUT; + + return iBytesRead; } inline bool TcpResolveAddress(const char *strHost, uint16_t iPort, int *iError, struct addrinfo **info) @@ -279,7 +286,7 @@ namespace PLATFORM } else { - *iError = errno; + *iError = GetSocketError(); } }