From c40cc5d16dbb22c46e47a1028265b8ee9fb5df27 Mon Sep 17 00:00:00 2001
From: Garrett D'Amore <garrett@damore.org>
Date: Sun, 19 May 2019 12:34:26 -0700
Subject: fixes #937 Memory Leak in wsstream

---
 src/supplemental/websocket/websocket.c | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'src/supplemental/websocket/websocket.c')

diff --git a/src/supplemental/websocket/websocket.c b/src/supplemental/websocket/websocket.c
index 91affc62..d8b7f7d5 100644
--- a/src/supplemental/websocket/websocket.c
+++ b/src/supplemental/websocket/websocket.c
@@ -447,10 +447,13 @@ ws_frame_prep_tx(nni_ws *ws, ws_frame *frame)
 	// Potentially allocate space for the data if we need to.
 	// Note that an empty message is legal.
 	if ((frame->bufsz < frame->len) && (frame->len > 0)) {
+		nni_free(frame->buf, frame->bufsz);
 		frame->buf = nni_alloc(frame->len);
 		if (frame->buf == NULL) {
+			frame->bufsz = 0;
 			return (NNG_ENOMEM);
 		}
+		frame->bufsz = frame->len;
 	}
 	buf = frame->buf;
 
-- 
cgit v1.2.3-70-g09d2