summaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
-rw-r--r--extmod/modwebsocket.c2
1 files changed, 1 insertions, 1 deletions
diff --git a/extmod/modwebsocket.c b/extmod/modwebsocket.c
index 78bbe6d13e..bc71442e72 100644
--- a/extmod/modwebsocket.c
+++ b/extmod/modwebsocket.c
@@ -155,7 +155,7 @@ STATIC mp_uint_t websocket_read(mp_obj_t self_in, void *buf, mp_uint_t size, int
case PAYLOAD: {
size_t sz = MIN(size, self->msg_sz);
mp_uint_t out_sz = stream_p->read(self->sock, buf, sz, errcode);
- if (out_sz == MP_STREAM_ERROR) {
+ if (out_sz == 0 || out_sz == MP_STREAM_ERROR) {
return out_sz;
}