Check return value of malloc

Signed-off-by: Jianhui Zhao <jianhuizhao329@gmail.com>
zhaojh329 · Jun 18, 2018 · 09a7fe4 · 09a7fe4
1 parent 8365939
commit 09a7fe4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/uwsc.c b/src/uwsc.c
@@ -147,6 +147,12 @@ static bool parse_frame(struct uwsc_client *cl, uint8_t *data, uint64_t len)
         if (!fin) {
             frame->fragmented = true;
             frame->payload = malloc(payloadlen);
+            if (!frame->payload) {
+                uwsc_log_err("No mem");
+                cl->send(cl, NULL, 0, WEBSOCKET_OP_CLOSE);
+                cl->error = UWSC_ERROR_NOMEM;
+                return false;
+            }
             memcpy(frame->payload, payload, payloadlen);
         }
     }