From: Shuanglei Tao <tsl0922@gmail.com>
Date: Thu, 18 Aug 2022 04:14:46 +0000 (+0800)
Subject: protocol: fix potential pty buf leak on process exit
X-Git-Url: http://git.prime8.dev/?a=commitdiff_plain;h=efefd567959aecd2a060cf2de453a53692c30de4;p=ttyd.git

protocol: fix potential pty buf leak on process exit
---

diff --git a/src/protocol.c b/src/protocol.c
index 731b54f..5341c3d 100644
--- 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]);
       }