首先Zmalloc的接口定義在頭文件Zmalloc.h裡:
[cpp]
- void *zmalloc(size_t size);
- void *zcalloc(size_t size);
- void *zrealloc(void *ptr, size_t size);
- void zfree(void *ptr);
- char *zstrdup(const char *s);
- size_t zmalloc_used_memory(void);
- void zmalloc_enable_thread_safeness(void);
- float zmalloc_get_fragmentation_ratio(void);
- size_t zmalloc_get_rss(void);
前五個函數對應於C標准庫函數。zmalloc_used_memory用來返回當前已用內存。
舉例分析zmalloc函數:
[cpp]
- void *zmalloc(size_t size) {
- void *ptr = malloc(size+PREFIX_SIZE);
-
- if (!ptr) zmalloc_oom(size);
- #ifdef HAVE_MALLOC_SIZE
- update_zmalloc_stat_alloc(zmalloc_size(ptr),size);
- return ptr;
- #else
- *((size_t*)ptr) = size;
- update_zmalloc_stat_alloc(size+PREFIX_SIZE,size);
- return (char*)ptr+PREFIX_SIZE;
- #endif
- }
和標准庫的malloc函數的區別在於,zmalloc還在內存塊頭部保存了內存塊的大小。
PREFIX_SIZE定義為系統中一個標准的size_t的大小:
update_zmalloc_stat_alloc是一個宏,用來更新內存占用量的統計,定義為:
[cpp]
- #define update_zmalloc_stat_alloc(__n,__size) do { \
- size_t _n = (__n); \
- if (_n&(sizeof(long)-1)) _n += sizeof(long)-(_n&(sizeof(long)-1)); \
- if (zmalloc_thread_safe) { \
- pthread_mutex_lock(&used_memory_mutex); \
- used_memory += _n; \
- pthread_mutex_unlock(&used_memory_mutex); \
- } else { \
- used_memory += _n; \
- } \
- } while(0)
如果啟用了線程安全選項,將在加鎖之後才對臨界變量used_memory_mutex進行修改。相關變量定義在:
[cpp]
- static size_t used_memory = 0;
- static int zmalloc_thread_safe = 0;
- pthread_mutex_t used_memory_mutex = PTHREAD_MUTEX_INITIALIZER;
在函數size_t zmalloc_used_memory(void)中,返回已占用內存量的大小,即靜態變量used_memory。
函數zmalloc_get_rss()獲取RSS
(Resident Set Size) 的方式有三種:如果定義了PROC_FS,那麼將從"/proc/[getpid()]/stat"中讀取;如果定義了TASK_INFO,將從該進程id對應的task_info_t結構中讀取resident_size變量;最後,如果前兩者都沒有定義,那麼簡單地返回used_memory。
zmalloc_get_fragmentation_ratio(void)計算碎片率的公式為:(float)zmalloc_get_rss()/zmalloc_used_memory()。