Fix bug in websocket on empty packets

Change-Id: I21d6e36df183b798539cf2a378976cc595a77029
Signed-off-by: José Bollo <jose.bollo@iot.bzh>

diff --git a/src/afb-ws.c b/src/afb-ws.c
index b157a49..6d48887 100644 (file)
--- a/src/afb-ws.c
+++ b/src/afb-ws.c
@@ -395,12 +395,12 @@ static int aws_read(struct afb_ws *ws, size_t size)
        ssize_t sz;
        char *buffer;
 
-       if (size != 0) {
+       if (size != 0 || ws->buffer.buffer == NULL) {
                buffer = realloc(ws->buffer.buffer, ws->buffer.size + size + 1);
                if (buffer == NULL)
                        return 0;
                ws->buffer.buffer = buffer;
-               do {
+               while (size != 0) {
                        sz = websock_read(ws->ws, &buffer[ws->buffer.size], size);
                        if (sz < 0) {
                                if (errno != EAGAIN)
@@ -412,7 +412,7 @@ static int aws_read(struct afb_ws *ws, size_t size)
                                ws->buffer.size += (size_t)sz;
                                size -= (size_t)sz;
                        }
-               } while (size != 0);
+               }
        }
        return 1;
 }