From 0d630ecdf9a9265b9e29b400719fc804d26abb40 Mon Sep 17 00:00:00 2001
From: Yaming Pei <sanwenyuv@sina.com>
Date: Tue, 4 Mar 2025 14:19:57 +0800
Subject: [PATCH] fix: fix csv file handle leaks

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

diff --git a/tools/taos-tools/src/benchCsv.c b/tools/taos-tools/src/benchCsv.c
index cf8527d375..848cd9a6ef 100644
--- a/tools/taos-tools/src/benchCsv.c
+++ b/tools/taos-tools/src/benchCsv.c
@@ -448,7 +448,7 @@ static int csvExportCreateSql(CsvWriteMeta* write_meta) {
         goto end;
     }
 
-    succPrint("Export create sql to file: %s successfully..\n", fullname);
+    succPrint("Export create sql to file: %s successfully.\n", fullname);
 
 end:
     if (fp) {
@@ -827,6 +827,10 @@ static CsvIoError csvWrite(CsvFileHandle* fhdl, const char* buf, size_t size) {
 
 
 static void csvClose(CsvFileHandle* fhdl) {
+    if (!fhdl) {
+        return;
+    }
+
     if (fhdl->compress_level == CSV_COMPRESS_NONE) {
         if (fhdl->handle.fp) {
             fclose(fhdl->handle.fp);
@@ -838,6 +842,7 @@ static void csvClose(CsvFileHandle* fhdl) {
             fhdl->handle.gf = NULL;
         }
     }
+    tmfree(fhdl);
 }