extmod/modlwip: Check for state change during recv busy-wait loop.

For example, the peer may close the connection while recv is waiting for
incoming data.
pull/1888/head
Damien George 2015-12-30 19:03:58 +00:00 zatwierdzone przez Paul Sokolovsky
rodzic 6185dc5f3d
commit 6d2e9e70b3
1 zmienionych plików z 7 dodań i 1 usunięć

Wyświetl plik

@ -391,13 +391,19 @@ STATIC mp_uint_t lwip_tcp_receive(lwip_socket_obj_t *socket, byte *buf, mp_uint_
if (socket->incoming.pbuf == NULL) {
mp_uint_t start = mp_hal_ticks_ms();
while (socket->incoming.pbuf == NULL) {
while (socket->state == STATE_CONNECTED && socket->incoming.pbuf == NULL) {
if (socket->timeout != -1 && mp_hal_ticks_ms() - start > socket->timeout) {
*_errno = ETIMEDOUT;
return -1;
}
poll_sockets();
}
if (socket->state == STATE_PEER_CLOSED) {
return 0;
} else if (socket->state != STATE_CONNECTED) {
*_errno = -socket->state;
return -1;
}
}
struct pbuf *p = socket->incoming.pbuf;