extmod/modwebsocket: Another case to propagate EOF.

pfalcon · pfalcon · commit 59a4fee5161c · 2016-04-13T22:17:09.000+03:00
diff --git 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;
                 }
 

Original file line number	Diff line number	Diff line change
`@@ -155,7 +155,7 @@ STATIC mp_uint_t websocket_read(mp_obj_t self_in, void *buf, mp_uint_t size, int`
`155`	`155`	`case PAYLOAD: {`
`156`	`156`	`size_t sz = MIN(size, self->msg_sz);`
`157`	`157`	`mp_uint_t out_sz = stream_p->read(self->sock, buf, sz, errcode);`
`158`		`- if (out_sz == MP_STREAM_ERROR) {`
	`158`	`+ if (out_sz == 0 \|\| out_sz == MP_STREAM_ERROR) {`
`159`	`159`	`return out_sz;`
`160`	`160`	`}`
`161`	`161`