Check memory allocation after malloc

This commit is contained in:
universeroc 2019-08-07 16:06:00 +08:00 committed by GitHub
parent d0e5e817a2
commit 22f9f4b0fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions

View File

@ -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);