redis基础组件-zmalloc简介

数据结构

1
2
3
------------
len|context
------------

特点

  • 申请内存时,自带整个MEM的len,用于统计当前进程总共消耗的MEM

模糊点

  • 在申请内存的接口中怎么确定ptr+sizeof(size_t)后的地址一定就能满足内存对齐?

主要接口函数

  • 申请一片内存

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    void *zmalloc(size_t size) {
    void *ptr = malloc(size+PREFIX_SIZE);

    if (!ptr) zmalloc_oom(size);
    #ifdef HAVE_MALLOC_SIZE
    increment_used_memory(redis_malloc_size(ptr));
    return ptr;
    #else
    *((size_t*)ptr) = size;
    increment_used_memory(size+PREFIX_SIZE);
    return (char*)ptr+PREFIX_SIZE;
    #endif
    }
  • 调整内存大小

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    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 = redis_malloc_size(ptr);
    newptr = realloc(ptr,size);
    if (!newptr) zmalloc_oom(size);

    decrement_used_memory(oldsize);
    increment_used_memory(redis_malloc_size(newptr));
    return newptr;
    #else
    realptr = (char*)ptr-PREFIX_SIZE;
    oldsize = *((size_t*)realptr);
    newptr = realloc(realptr,size+PREFIX_SIZE);
    if (!newptr) zmalloc_oom(size);

    *((size_t*)newptr) = size;
    decrement_used_memory(oldsize);
    increment_used_memory(size);
    return (char*)newptr+PREFIX_SIZE;
    #endif
    }
  • 释放内存

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    void zfree(void *ptr) {
    #ifndef HAVE_MALLOC_SIZE
    void *realptr;
    size_t oldsize;
    #endif

    if (ptr == NULL) return;
    #ifdef HAVE_MALLOC_SIZE
    decrement_used_memory(redis_malloc_size(ptr));
    free(ptr);
    #else
    realptr = (char*)ptr-PREFIX_SIZE;
    oldsize = *((size_t*)realptr);
    decrement_used_memory(oldsize+PREFIX_SIZE);
    free(realptr);
    #endif
    }
-------------本文结束 感谢阅读-------------