protocol: fix potential pty buf leak on process exit

tsl0922 · Aug 18, 2022 · efefd56 · efefd56
1 parent 520ce17
commit efefd56
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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]);
       }