diff options
Diffstat (limited to 'src/wsocket.c')
| -rw-r--r-- | src/wsocket.c | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/wsocket.c b/src/wsocket.c index 2993c35..af3f8d8 100644 --- a/src/wsocket.c +++ b/src/wsocket.c | |||
| @@ -269,7 +269,7 @@ int sock_recv(p_sock ps, char *data, size_t count, size_t *got, int timeout) | |||
| 269 | fd_set fds; | 269 | fd_set fds; |
| 270 | int ret; | 270 | int ret; |
| 271 | *got = 0; | 271 | *got = 0; |
| 272 | if (taken == 0) return IO_CLOSED; | 272 | if (taken == 0 || WSAGetLastError() != WSAEWOULDBLOCK) return IO_CLOSED; |
| 273 | FD_ZERO(&fds); | 273 | FD_ZERO(&fds); |
| 274 | FD_SET(sock, &fds); | 274 | FD_SET(sock, &fds); |
| 275 | ret = sock_select(0, &fds, NULL, NULL, timeout); | 275 | ret = sock_select(0, &fds, NULL, NULL, timeout); |
| @@ -295,7 +295,7 @@ int sock_recvfrom(p_sock ps, char *data, size_t count, size_t *got, | |||
| 295 | fd_set fds; | 295 | fd_set fds; |
| 296 | int ret; | 296 | int ret; |
| 297 | *got = 0; | 297 | *got = 0; |
| 298 | if (taken == 0) return IO_CLOSED; | 298 | if (taken == 0 || WSAGetLastError() != WSAEWOULDBLOCK) return IO_CLOSED; |
| 299 | FD_ZERO(&fds); | 299 | FD_ZERO(&fds); |
| 300 | FD_SET(sock, &fds); | 300 | FD_SET(sock, &fds); |
| 301 | ret = sock_select(0, &fds, NULL, NULL, timeout); | 301 | ret = sock_select(0, &fds, NULL, NULL, timeout); |