zmalloc fix, return NULL or real malloc failure

8d196eba · antirez · 71aee3e9 · 8d196eba
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

zmalloc.c zmalloc.c +1 -0

未找到文件。
--- a/zmalloc.c
+++ b/zmalloc.c
@@ -36,6 +36,7 @@ static size_t used_memory = 0;
 void *zmalloc(size_t size) {
    void *ptr = malloc(size+sizeof(size_t));

+    if (!ptr) return NULL;
    *((size_t*)ptr) = size;
    used_memory += size+sizeof(size_t);
    return (char*)ptr+sizeof(size_t);