protocol: fix potential pty buf leak on process exit

diff --git a/src/protocol.c b/src/protocol.c
index 731b54fa5ed803a19f016f53260cd07149788523..5341c3d38f6d391d4c486754d353a4d3878df4a2 100644 (file)
--- a/src/protocol.c
+++ b/src/protocol.c
@@ -363,9 +363,8 @@ int callback_tty(struct lws *wsi, enum lws_callback_reasons reason, void *user,
 
       server->client_count--;
       lwsl_notice("WS closed from %s, clients: %d\n", pss->address, server->client_count);
-      if (pss->buffer != NULL) {
-        free(pss->buffer);
-      }
+      if (pss->buffer != NULL) free(pss->buffer);
+      if (pss->pty_buf != NULL) pty_buf_free(pss->pty_buf);
       for (int i = 0; i < pss->argc; i++) {
         free(pss->args[i]);
       }