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