fix: add query rsp checking as retrieve rsp processing

863156c2 · slzhou · ee768756 · 863156c2
隐藏空白更改
内联并排

Showing with 6 addition and 0 deletion

src/client/src/tscServer.c src/client/src/tscServer.c +6 -0

未找到文件。
--- a/src/client/src/tscServer.c
+++ b/src/client/src/tscServer.c
@@ -2953,6 +2953,12 @@ int tscProcessQueryRsp(SSqlObj *pSql) {
  SSqlRes *pRes = &pSql->res;

  SQueryTableRsp *pQueryAttr = (SQueryTableRsp *)pRes->pRsp;
+
+  if (pQueryAttr == NULL) {
+    pRes->code = TSDB_CODE_TSC_OUT_OF_MEMORY;
+    return pRes->code;
+  }
+  
  pQueryAttr->qId = htobe64(pQueryAttr->qId);

  pRes->qId  = pQueryAttr->qId;