From f7b298ae1c4f46bba07ddf9f9813c8fa85187d12 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Jos=C3=A9=20Bollo?= <jose.bollo@iot.bzh>
Date: Thu, 1 Jun 2017 22:14:34 +0200
Subject: [PATCH] Fix bug in websocket on empty packets
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

Change-Id: I21d6e36df183b798539cf2a378976cc595a77029
Signed-off-by: José Bollo <jose.bollo@iot.bzh>
---
 src/afb-ws.c | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/afb-ws.c b/src/afb-ws.c
index b157a49f..6d488878 100644
--- 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;
 }
-- 
2.16.6