From 22ea676a5a288839842f8fed1bc53ff21b379ff1 Mon Sep 17 00:00:00 2001
From: afwerar <1296468573@qq.com>
Date: Mon, 18 Jul 2022 11:29:15 +0800
Subject: [PATCH] shell: limit taos_history file size

---
 tools/shell/src/shellEngine.c | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/tools/shell/src/shellEngine.c b/tools/shell/src/shellEngine.c
index d6b7f18fb9..d661667412 100644
--- a/tools/shell/src/shellEngine.c
+++ b/tools/shell/src/shellEngine.c
@@ -748,6 +748,13 @@ int32_t shellDumpResult(TAOS_RES *tres, char *fname, int32_t *error_no, bool ver
 
 void shellReadHistory() {
   SShellHistory *pHistory = &shell.history;
+  int64_t file_size;
+  if (taosStatFile(pHistory->file, &file_size, NULL) != 0) {
+    return;
+  } else if (file_size > SHELL_MAX_COMMAND_SIZE) {
+    taosRemoveFile(pHistory->file);
+    return;
+  }
   TdFilePtr      pFile = taosOpenFile(pHistory->file, TD_FILE_READ | TD_FILE_STREAM);
   if (pFile == NULL) return;
 
@@ -771,6 +778,12 @@ void shellReadHistory() {
 
 void shellWriteHistory() {
   SShellHistory *pHistory = &shell.history;
+  int64_t file_size;
+  if (taosStatFile(pHistory->file, &file_size, NULL) != 0) {
+    return;
+  } else if (file_size > SHELL_MAX_COMMAND_SIZE) {
+    taosRemoveFile(pHistory->file);
+  }
   TdFilePtr      pFile = taosOpenFile(pHistory->file, TD_FILE_CREATE | TD_FILE_WRITE | TD_FILE_STREAM | TD_FILE_APPEND);
   if (pFile == NULL) return;