redis之zmalloc

号外号外，新建Redis交流讨论群：332160890，欢迎加入！！

zmalloc介绍

这部分是用来申请内存的，之所以单独成了一部分在于redis希望能够将不同系统提供的api统一起来，另外还有一部分系统无法获取malloc数目，因此在这里也进行了一部分的封装；这部分内容很简单，实际上就是malloc, calloc, realloc, free几个操作的变形

代码分析

使用变量

used_memory，已使用的内存数；
malloc_thread_safe，分配内存线程是否安全；
used_memory_mutex，已使用内存数变量互斥锁；

代码分解

zmalloc_default_oom,默认的内存溢出响应函数

static void zmalloc_default_oom(size_t size) {    fprintf(stderr, "zmalloc: Out of memory trying to allocate %zu bytes\n",        size);    fflush(stderr);    abort();}

zmalloc，这里面判断是否需要添加prefix_size，也就是将分配的内存大小写到分配内存的前部，根据实际需要大小执行malloc函数

void *zmalloc(size_t size) {    void *ptr = malloc(size+PREFIX_SIZE);    if (!ptr) zmalloc_oom_handler(size);#ifdef HAVE_MALLOC_SIZE    update_zmalloc_stat_alloc(zmalloc_size(ptr));    return ptr;#else    *((size_t*)ptr) = size;    update_zmalloc_stat_alloc(size+PREFIX_SIZE);    return (char*)ptr+PREFIX_SIZE;  //有些系统无法知道分配内存大小，需要自己添加#endif}

zcalloc，这个函数与zmalloc的函数区别在于申请得到的内存是否初始化，zcalloc会初始化，zmalloc则是乱数据

void *zcalloc(size_t size) {    void *ptr = calloc(1, size+PREFIX_SIZE);    if (!ptr) zmalloc_oom_handler(size);#ifdef HAVE_MALLOC_SIZE    update_zmalloc_stat_alloc(zmalloc_size(ptr));    return ptr;#else    *((size_t*)ptr) = size;    update_zmalloc_stat_alloc(size+PREFIX_SIZE);    return (char*)ptr+PREFIX_SIZE;#endif}

zrealloc，这个函数是用来重新分配内存的，如果ptr为空，则直接申请内存即可，如果当前地址不能满足size大小的内存空间，则需要重新申请内存，然后将ptr中数据拷贝过去，返回新申请内存地址；如果当前地址满足，则扩大当前地址范围

void *zrealloc(void *ptr, size_t size) {#ifndef HAVE_MALLOC_SIZE    void *realptr;#endif    size_t oldsize;    void *newptr;    if (ptr == NULL) return zmalloc(size);#ifdef HAVE_MALLOC_SIZE    oldsize = zmalloc_size(ptr);    newptr = realloc(ptr,size);  //重新分配内存    if (!newptr) zmalloc_oom_handler(size);    update_zmalloc_stat_free(oldsize);    update_zmalloc_stat_alloc(zmalloc_size(newptr));    return newptr;#else    realptr = (char*)ptr-PREFIX_SIZE;    oldsize = *((size_t*)realptr);    newptr = realloc(realptr,size+PREFIX_SIZE);    if (!newptr) zmalloc_oom_handler(size);    *((size_t*)newptr) = size;    update_zmalloc_stat_free(oldsize);    update_zmalloc_stat_alloc(size);    return (char*)newptr+PREFIX_SIZE;#endif}

zfree，释放内存；

void zfree(void *ptr) {#ifndef HAVE_MALLOC_SIZE    void *realptr;    size_t oldsize;#endif    if (ptr == NULL) return;#ifdef HAVE_MALLOC_SIZE    update_zmalloc_stat_free(zmalloc_size(ptr));    free(ptr);#else    realptr = (char*)ptr-PREFIX_SIZE;    oldsize = *((size_t*)realptr);    update_zmalloc_stat_free(oldsize+PREFIX_SIZE);    free(realptr);#endif}