From 1b0636c2d274b6692c161c1199bfa17240a5f541 Mon Sep 17 00:00:00 2001
From: cpq <valenok@gmail.com>
Date: Sat, 2 Jan 2021 12:20:08 +0000
Subject: [PATCH] More checks in mg_ws_send

---
 mongoose.c | 2 +-
 src/ws.c   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/mongoose.c b/mongoose.c
index 68fb1b42..4fe39d29 100644
--- a/mongoose.c
+++ b/mongoose.c
@@ -3941,7 +3941,7 @@ size_t mg_ws_send(struct mg_connection *c, const char *buf, size_t len,
   }
   LOG(LL_VERBOSE_DEBUG, ("WS out: %d [%.*s]", (int) len, (int) len, buf));
   mg_send(c, buf, len);
-  if (c->is_client) {
+  if (c->is_client && c->send.buf != NULL) {
     uint8_t *p = c->send.buf + c->send.len - len;
     for (i = 0; i < len; i++) p[i] ^= mask[i & 3];
   }
diff --git a/src/ws.c b/src/ws.c
index 5ee6c2d9..f2e2fd2a 100644
--- a/src/ws.c
+++ b/src/ws.c
@@ -95,7 +95,7 @@ size_t mg_ws_send(struct mg_connection *c, const char *buf, size_t len,
   }
   LOG(LL_VERBOSE_DEBUG, ("WS out: %d [%.*s]", (int) len, (int) len, buf));
   mg_send(c, buf, len);
-  if (c->is_client) {
+  if (c->is_client && c->send.buf != NULL) {
     uint8_t *p = c->send.buf + c->send.len - len;
     for (i = 0; i < len; i++) p[i] ^= mask[i & 3];
   }