From 1ccc74d768ceb74ffd1b0299fc1df4cfa307d07e Mon Sep 17 00:00:00 2001
From: dapan1121 <wpan@taosdata.com>
Date: Sat, 14 Sep 2024 17:28:24 +0800
Subject: [PATCH] fix: add memory alloc validation

---
 source/libs/catalog/src/ctgCache.c           | 3 +++
 source/libs/catalog/src/ctgDbg.c             | 4 ++++
 source/libs/executor/src/mergejoinoperator.c | 3 +++
 source/libs/scheduler/src/schRemote.c        | 6 ++++++
 4 files changed, 16 insertions(+)

diff --git a/source/libs/catalog/src/ctgCache.c b/source/libs/catalog/src/ctgCache.c
index 0e54ac77a2..9a72a6d89f 100644
--- a/source/libs/catalog/src/ctgCache.c
+++ b/source/libs/catalog/src/ctgCache.c
@@ -3710,6 +3710,9 @@ int32_t ctgGetTbHashVgroupFromCache(SCatalog *pCtg, const SName *pTableName, SVg
   }
 
   *pVgroup = taosMemoryCalloc(1, sizeof(SVgroupInfo));
+  if (NULL == *pVgroup) {
+    CTG_ERR_JRET(terrno);
+  }
   CTG_ERR_JRET(ctgGetVgInfoFromHashValue(pCtg, NULL, dbCache->vgCache.vgInfo, pTableName, *pVgroup));
 
 _return:
diff --git a/source/libs/catalog/src/ctgDbg.c b/source/libs/catalog/src/ctgDbg.c
index a95df06307..0ac606d4d6 100644
--- a/source/libs/catalog/src/ctgDbg.c
+++ b/source/libs/catalog/src/ctgDbg.c
@@ -291,6 +291,10 @@ int32_t ctgdHandleDbgCommand(char *command) {
   }
 
   char *dup = taosStrdup(command);
+  if (NULL == dup) {
+    CTG_RET(terrno);
+  }
+  
   char *option = NULL;
   char *param = NULL;
   
diff --git a/source/libs/executor/src/mergejoinoperator.c b/source/libs/executor/src/mergejoinoperator.c
index 4715aa8b96..af5e4ed235 100644
--- a/source/libs/executor/src/mergejoinoperator.c
+++ b/source/libs/executor/src/mergejoinoperator.c
@@ -149,6 +149,9 @@ int32_t mJoinTrimKeepOneRow(SSDataBlock* pBlock, int32_t totalRows, const bool*
             len = varDataTLen(p1);
           }
           char* p2 = taosMemoryMalloc(len);
+          if (NULL == p2) {
+            MJ_ERR_RET(terrno);
+          }
           TAOS_MEMCPY(p2, p1, len);
           code = colDataSetVal(pDst, numOfRows, p2, false);
           if (code) {
diff --git a/source/libs/scheduler/src/schRemote.c b/source/libs/scheduler/src/schRemote.c
index 40391cea7e..7ff3d1386c 100644
--- a/source/libs/scheduler/src/schRemote.c
+++ b/source/libs/scheduler/src/schRemote.c
@@ -267,6 +267,9 @@ int32_t schProcessResponseMsg(SSchJob *pJob, SSchTask *pTask, int32_t execId, SD
       if (pMsg->pData) {
         SDecoder    coder = {0};
         SSubmitRsp2 *rsp = taosMemoryMalloc(sizeof(*rsp));
+        if (NULL == rsp) {
+          SCH_ERR_JRET(terrno);
+        }
         tDecoderInit(&coder, pMsg->pData, msgSize);
         code = tDecodeSSubmitRsp2(&coder, rsp);
         tDecoderClear(&coder);
@@ -957,6 +960,9 @@ int32_t schUpdateSendTargetInfo(SMsgSendInfo *pMsgSendInfo, SQueryNodeAddr *addr
     pMsgSendInfo->target.type = TARGET_TYPE_VNODE;
     pMsgSendInfo->target.vgId = addr->nodeId;
     pMsgSendInfo->target.dbFName = taosStrdup(pTask->plan->dbFName);
+    if (NULL == pMsgSendInfo->target.dbFName) {
+      return terrno;
+    }
   }
 
   return TSDB_CODE_SUCCESS;