redis 内存管理zmalloc.c

zmalloc.h

对c的函数进行了封装

void *zmalloc(size_t size); //调用zmalloc申请size个大小的空间 
void *zcalloc(size_t size); // 调用系统函数calloc函数申请空间
void *zrealloc(void *ptr, size_t size); // 原内存重新调整空间为size的大小
void zfree(void *ptr); //释放内存 free
char *zstrdup(const char *s); // 字符串复制方法 
size_t zmalloc_used_memory(void); //获取当前已经占用的内存大小 
void zmalloc_enable_thread_safeness(void); // 是否设置线程安全模式 
void zmalloc_set_oom_handler(void (*oom_handler)(size_t)); // 可自定义设置内存溢出的处理方法 
float zmalloc_get_fragmentation_ratio(size_t rss); // 所给大小与已使用内存大小之比 
size_t zmalloc_get_rss(void); // 获取rss的大小
size_t zmalloc_get_private_dirty(void); //获取私有的脏数据大小 
void zlibc_free(void *ptr);  // 原始系统free释放方法 

zmalloc.c

介绍几个函数

#if defined(USE_TCMALLOC)  //使用tcmalloc
#define malloc(size) tc_malloc(size)
#define calloc(count,size) tc_calloc(count,size)
#define realloc(ptr,size) tc_realloc(ptr,size)
#define free(ptr) tc_free(ptr)
#elif defined(USE_JEMALLOC)  //使用jemalloc
#define malloc(size) je_malloc(size)
#define calloc(count,size) je_calloc(count,size)
#define realloc(ptr,size) je_realloc(ptr,size)
#define free(ptr) je_free(ptr)
#endif

#ifdef HAVE_ATOMIC
#define update_zmalloc_stat_add(__n) __sync_add_and_fetch(&used_memory, (__n)) 
#define update_zmalloc_stat_sub(__n) __sync_sub_and_fetch(&used_memory, (__n))
#else
#define update_zmalloc_stat_add(__n) do { \       //增加__n的大小到used_memory
    pthread_mutex_lock(&used_memory_mutex); \
    used_memory += (__n); \
    pthread_mutex_unlock(&used_memory_mutex); \
} while(0)

#define update_zmalloc_stat_sub(__n) do { \   //减小__n的大小到used_memory
    pthread_mutex_lock(&used_memory_mutex); \
    used_memory -= (__n); \
    pthread_mutex_unlock(&used_memory_mutex); \
} while(0)
#endif
变量解释：
  static size_t used_memory = 0;  // redis服务使用的总的内存大小
  static int zmalloc_thread_safe = 0; //线程安全  加锁的方式增加或减小used_memory
  pthread_mutex_t used_memory_mutex = PTHREAD_MUTEX_INITIALIZER;

#define update_zmalloc_stat_alloc(__n) do { \  //底层调用的上面谈到的两个宏
    size_t _n = (__n); \
    if (_n&(sizeof(long)-1)) _n += sizeof(long)-(_n&(sizeof(long)-1)); \  // 字长对齐对申请n字节大小的内存进行内存大小的对齐 表示实际申请的大小  如果n=5,sizeof(long) = 8,那么真正的n值大小等于n=5+8-5=8
    if (zmalloc_thread_safe) { \
        update_zmalloc_stat_add(_n); \
    } else { \
        used_memory += _n; \
    } \
} while(0)

#define update_zmalloc_stat_free(__n) do { \ //同上
    size_t _n = (__n); \
    if (_n&(sizeof(long)-1)) _n += sizeof(long)-(_n&(sizeof(long)-1)); \
    if (zmalloc_thread_safe) { \
        update_zmalloc_stat_sub(_n); \
    } else { \
        used_memory -= _n; \
    } \
} while(0)

示例函数

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; //PREFIX_SIZE 等于size_t的大小 用于保存实际分配的内存大小
    oldsize = *((size_t*)realptr);
    newptr = realloc(realptr,size+PREFIX_SIZE);
    if (!newptr) zmalloc_oom_handler(size);

    *((size_t*)newptr) = size;  //分配内存的前PREFIX_SIZE个字节用于记录真实的大小 方便内存管理
    update_zmalloc_stat_free(oldsize);
    update_zmalloc_stat_alloc(size);
    return (char*)newptr+PREFIX_SIZE;
#endif
}


#ifndef HAVE_MALLOC_SIZE
size_t zmalloc_size(void *ptr) {
    void *realptr = (char*)ptr-PREFIX_SIZE;
    size_t size = *((size_t*)realptr);
    /* Assume at least that all the allocations are padded at sizeof(long) by
     * the underlying allocator. */
    if (size&(sizeof(long)-1)) size += sizeof(long)-(size&(sizeof(long)-1));
    return size+PREFIX_SIZE;
}
#endif