From 22f9f4b0fd16babce7ccdb82bb2e1b0072fe2110 Mon Sep 17 00:00:00 2001
From: universeroc <universeroc@gmail.com>
Date: Wed, 7 Aug 2019 16:06:00 +0800
Subject: [PATCH] Check memory allocation after malloc

---
 src/util/src/ihash.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/util/src/ihash.c b/src/util/src/ihash.c
index 5773d10de2..a10a2d5bf7 100644
--- a/src/util/src/ihash.c
+++ b/src/util/src/ihash.c
@@ -54,9 +54,12 @@ char *taosAddIntHash(void *handle, int32_t key, char *pData) {
 
   hash = (*pObj->hashFp)(pObj, key);
 
+  pNode = (IHashNode *)malloc(sizeof(IHashNode) + (size_t)pObj->dataSize);
+  if (pNode == NULL)
+    return NULL;
+  
   pthread_mutex_lock(&pObj->mutex);
 
-  pNode = (IHashNode *)malloc(sizeof(IHashNode) + (size_t)pObj->dataSize);
   pNode->key = key;
   if (pData != NULL) {
     memcpy(pNode->data, pData, (size_t)pObj->dataSize);