Do not test reassembled buffer for WS_OP_CLOSE

PUBLISHED_FROM=0e8de208b553b63c8367f66893389587bd17359f

Do not test reassembled buffer for WS_OP_CLOSE
PUBLISHED_FROM=0e8de208b553b63c8367f66893389587bd17359f
aed6d05d · Alexander Alashkin · Cesanta Bot · d85d95a4 · aed6d05d
Commit aed6d05d authored Mar 01, 2017 by Alexander Alashkin Committed by Cesanta Bot Mar 01, 2017
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

mongoose.c mongoose.c +2 -2

No files found.
--- a/mongoose.c
+++ b/mongoose.c
@@ -8888,8 +8888,8 @@ static int mg_deliver_websocket_data(struct mg_connection *nc) {
      mbuf_remove(&nc->recv_mbuf, (size_t) frame_len); /* Cleanup frame */
    }

-    /* If client closes, close too */
-    if ((buf[0] & 0x0f) == WEBSOCKET_OP_CLOSE) {
+    /* If the frame is not reassembled - client closes and close too */
+    if (!reass && (buf[0] & 0x0f) == WEBSOCKET_OP_CLOSE) {
      nc->flags |= MG_F_SEND_AND_CLOSE;
    }
  }