ehn: malloc error handle

This commit is contained in:
Hongze Cheng 2024-09-27 09:00:42 +08:00
parent c6afba6c76
commit 54410b068d
1 changed files with 8 additions and 2 deletions

View File

@ -184,10 +184,16 @@ int32_t stmtBackupQueryFields(STscStmt* pStmt) {
int32_t size = pRes->numOfCols * sizeof(TAOS_FIELD);
pRes->fields = taosMemoryMalloc(size);
pRes->userFields = taosMemoryMalloc(size);
if (NULL == pRes->fields || NULL == pRes->userFields) {
if (pRes->fields == NULL) {
STMT_ERR_RET(terrno);
}
pRes->userFields = taosMemoryMalloc(size);
if (pRes->userFields == NULL) {
taosMemoryFreeClear(pRes->fields);
STMT_ERR_RET(terrno);
}
(void)memcpy(pRes->fields, pStmt->exec.pRequest->body.resInfo.fields, size);
(void)memcpy(pRes->userFields, pStmt->exec.pRequest->body.resInfo.userFields, size);