aboutsummaryrefslogtreecommitdiff
path: root/src/supplemental/websocket/websocket.c
diff options
context:
space:
mode:
authorGarrett D'Amore <garrett@damore.org>2019-05-19 12:34:26 -0700
committerGarrett D'Amore <garrett@damore.org>2019-05-19 12:34:26 -0700
commitc40cc5d16dbb22c46e47a1028265b8ee9fb5df27 (patch)
tree9a10a68cefe5c016db8625d7d158bca868ec20a8 /src/supplemental/websocket/websocket.c
parentd83d86a140c44cca5f1867479ea7e77a0550e299 (diff)
downloadnng-c40cc5d16dbb22c46e47a1028265b8ee9fb5df27.tar.gz
nng-c40cc5d16dbb22c46e47a1028265b8ee9fb5df27.tar.bz2
nng-c40cc5d16dbb22c46e47a1028265b8ee9fb5df27.zip
fixes #937 Memory Leak in wsstream
Diffstat (limited to 'src/supplemental/websocket/websocket.c')
-rw-r--r--src/supplemental/websocket/websocket.c3
1 files changed, 3 insertions, 0 deletions
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;