xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/util/src/ihash.c
View File

@ -54,9 +54,12 @@ char *taosAddIntHash(void *handle, int32_t key, char *pData) {
hash = (*pObj->hashFp)(pObj, key); hash = (*pObj->hashFp)(pObj, key);
pNode = (IHashNode *)malloc(sizeof(IHashNode) + (size_t)pObj->dataSize);
if (pNode == NULL)
return NULL;
pthread_mutex_lock(&pObj->mutex); pthread_mutex_lock(&pObj->mutex);
pNode = (IHashNode *)malloc(sizeof(IHashNode) + (size_t)pObj->dataSize);
pNode->key = key; pNode->key = key;
if (pData != NULL) { if (pData != NULL) {
memcpy(pNode->data, pData, (size_t)pObj->dataSize); memcpy(pNode->data, pData, (size_t)pObj->dataSize);