From bd3fd91a50c591d6583dba7eac7a86ddb1ac7c0f Mon Sep 17 00:00:00 2001
From: Alex Duan <417921451@qq.com>
Date: Thu, 7 Dec 2023 19:24:19 +0800
Subject: [PATCH] fix: too long multiple command line

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

diff --git a/tools/shell/src/shellEngine.c b/tools/shell/src/shellEngine.c
index 115abdcd36..e8a5b04178 100644
--- a/tools/shell/src/shellEngine.c
+++ b/tools/shell/src/shellEngine.c
@@ -1091,7 +1091,12 @@ void shellSourceFile(const char *file) {
 
   char *line = taosMemoryMalloc(TSDB_MAX_ALLOWED_SQL_LEN + 1);
   while ((read_len = taosGetsFile(pFile, TSDB_MAX_ALLOWED_SQL_LEN, line)) != -1) {
-    if (read_len >= TSDB_MAX_ALLOWED_SQL_LEN) continue;
+    if ( cmd_len + read_len >= TSDB_MAX_ALLOWED_SQL_LEN) {
+      printf("read command line too long over 1M, ignore this line. cmd_len = %d read_len=%d \n", (int32_t)cmd_len, read_len);
+      cmd_len = 0;
+      memset(line, 0, TSDB_MAX_ALLOWED_SQL_LEN + 1);
+      continue;
+    }
     line[--read_len] = '\0';
 
     if (read_len == 0 || shellIsCommentLine(line)) {  // line starts with #