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;
}