extmod/modlwip: lwip_tcp_receive(): Properly handle EOF for non-blocking sock.

1 files changed, 3 insertions, 0 deletions

diff --git a/extmod/modlwip.c b/extmod/modlwip.c
index 1ebcd8923..441fd8cec 100644
--- a/extmod/modlwip.c
+++ b/extmod/modlwip.c
@@ -451,6 +451,9 @@ STATIC mp_uint_t lwip_tcp_receive(lwip_socket_obj_t *socket, byte *buf, mp_uint_
 
         // Non-blocking socket
         if (socket->timeout == 0) {
+            if (socket->state == STATE_PEER_CLOSED) {
+                return 0;
+            }
             *_errno = EAGAIN;
             return -1;
         }