From 420482720fe105004df2d3bb60abdee73e24beb5 Mon Sep 17 00:00:00 2001
From: wangjiaming0909 <604227650@qq.com>
Date: Fri, 27 Sep 2024 09:09:45 +0800
Subject: [PATCH] ret check for taosMemoryAlloc

---
 source/libs/parser/src/parAstCreater.c | 12 ++++++------
 source/libs/parser/src/parAstParser.c  |  1 +
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/source/libs/parser/src/parAstCreater.c b/source/libs/parser/src/parAstCreater.c
index e52c8865c7..8904679072 100644
--- a/source/libs/parser/src/parAstCreater.c
+++ b/source/libs/parser/src/parAstCreater.c
@@ -28,12 +28,12 @@
     }                      \
   } while (0)
 
-#define CHECK_OUT_OF_MEM(p)                    \
-  do {                                         \
-    if (NULL == (p)) {                         \
-      pCxt->errCode = TSDB_CODE_OUT_OF_MEMORY; \
-      goto _err;                               \
-    }                                          \
+#define CHECK_OUT_OF_MEM(p)   \
+  do {                        \
+    if (NULL == (p)) {        \
+      pCxt->errCode = terrno; \
+      goto _err;              \
+    }                         \
   } while (0)
 
 #define CHECK_PARSER_STATUS(pCxt)             \
diff --git a/source/libs/parser/src/parAstParser.c b/source/libs/parser/src/parAstParser.c
index f4dd91f392..10d9b19e7f 100644
--- a/source/libs/parser/src/parAstParser.c
+++ b/source/libs/parser/src/parAstParser.c
@@ -47,6 +47,7 @@ int32_t parse(SParseContext* pParseCxt, SQuery** pQuery) {
   SAstCreateContext cxt;
   initAstCreateContext(pParseCxt, &cxt);
   void*   pParser = ParseAlloc((FMalloc)taosMemoryMalloc);
+  if (!pParser) return terrno;
   int32_t i = 0;
   while (1) {
     SToken t0 = {0};