redis0.1源码解析之内存分配

这是redis源码分析的第一篇，选取早期版本，一睹redis的基础原理。这篇分析内存分配。
redis内存管理是对malloc系列函数做了封装。额外记录了申请的内存大小信息。

#include <stdlib.h>
#include <string.h>

static size_t used_memory = 0;
/*
    分配sizeof(size_t)+size大小的内存，前面sizeof(size_t)个字节记录本次分配的大小，
    记录分配的总内存大小，返回用于存储数据的内存首地址，即跨过sizeof(size_t)大小个字节
*/
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);
}
// 重新分配内存，ptr是旧数据的内存首地址，size是本次需要分片的内存大小
void *zrealloc(void *ptr, size_t size) {
    void *realptr;
    size_t oldsize;
    void *newptr;
    // ptr为空即没有旧数据，新申请一块内存即可，不涉及数据迁移
    if (ptr == NULL) return zmalloc(size);
    // 旧数据占据的内存大小
    realptr = (char*)ptr-sizeof(size_t);
    // 得到数据部分的内存大小
    oldsize = *((size_t*)realptr);
    // 以旧数据的内存地址为基地址，重新分配size+sizeof(size_t)大小的内存
    newptr = realloc(realptr,size+sizeof(size_t));
    if (!newptr) return NULL;
    // 记录数据部分的内存大小
    *((size_t*)newptr) = size;
    // 重新计算已分配内存的总大小，sizeof(size_t)这块内存仍然在使用，不需要计算
    used_memory -= oldsize;
    used_memory += size;
    // 返回存储数据的内存首地址
    return (char*)newptr+sizeof(size_t);
}

void zfree(void *ptr) {
    void *realptr;
    size_t oldsize;

    if (ptr == NULL) return;
    // 算出真正的内存首地址
    realptr = (char*)ptr-sizeof(size_t);
    oldsize = *((size_t*)realptr);
    // 减去释放的内存大小
    used_memory -= oldsize+sizeof(size_t);
    free(realptr);
}
// 复制字符串
char *zstrdup(const char *s) {
    size_t l = strlen(s)+1;
    char *p = zmalloc(l);

    memcpy(p,s,l);
    return p;
}

size_t zmalloc_used_memory(void) {
    return used_memory;
}