Read all available data during recv poll

CL: mg: Read all available data during recv poll PUBLISHED_FROM=b081025b9fb392c988924c7097b5333c38053f6f

Read all available data during recv poll
CL: mg: Read all available data during recv poll PUBLISHED_FROM=b081025b9fb392c988924c7097b5333c38053f6f
93286836 · Deomid Ryabkov · Cesanta Bot · 91b57aa9 · 93286836 · 93286836
Commit 93286836 authored Nov 02, 2018 by Deomid Ryabkov Committed by Cesanta Bot Nov 02, 2018
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

mongoose.c mongoose.c +2 -0

mg_net.c src/mg_net.c +2 -0

No files found.
--- a/mongoose.c
+++ b/mongoose.c
@@ -2882,6 +2882,7 @@ static int mg_do_recv(struct mg_connection *nc) {
      ((nc->flags & MG_F_LISTENING) && !(nc->flags & MG_F_UDP))) {
    return -1;
  }
+  do {
    len = recv_avail_size(nc, len);
    if (len == 0) return -2;
    if (nc->recv_mbuf.size < nc->recv_mbuf.len + len) {
@@ -2894,6 +2895,7 @@ static int mg_do_recv(struct mg_connection *nc) {
    } else {
      res = mg_recv_tcp(nc, buf, len);
    }
+  } while (res > 0);
  return res;
 }

--- a/src/mg_net.c
+++ b/src/mg_net.c
@@ -565,6 +565,7 @@ static int mg_do_recv(struct mg_connection *nc) {
      ((nc->flags & MG_F_LISTENING) && !(nc->flags & MG_F_UDP))) {
    return -1;
  }
+  do {
    len = recv_avail_size(nc, len);
    if (len == 0) return -2;
    if (nc->recv_mbuf.size < nc->recv_mbuf.len + len) {
@@ -577,6 +578,7 @@ static int mg_do_recv(struct mg_connection *nc) {
    } else {
      res = mg_recv_tcp(nc, buf, len);
    }
+  } while (res > 0);
  return res;
 }