wangzhengquan
2020-07-21 91ec036cace39fd5b5f04644f6bced1f477005e0
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#ifndef _MEM_POOL_H_
#define _MEM_POOL_H_  
#include "mm.h"
#include "sem_util.h"
#define MEM_POOL_COND_KEY 0x8801
 
static int mem_pool_cond  = SemUtil::get(MEM_POOL_COND_KEY, 0);
 
// static int mem_pool_mutex  = SemUtil::get(MEM_POOL_COND_KEY, 1);
 
static inline void mem_pool_init(size_t heap_size) {
    if(mm_init(heap_size)) {
         
    }
}
 
static inline void mem_pool_destroy(void) {
    if(mm_destroy()) {
        SemUtil::remove(mem_pool_cond);
    }
    
}
 
static inline void *mem_pool_malloc (size_t size) {
    void *ptr;
    while( (ptr = mm_malloc(size)) == NULL ) {
        err_msg(0, "There is not enough memery to allocate, waiting someone else to free.");
        SemUtil::set(mem_pool_cond, 0);
        // wait for someone else to free space
        SemUtil::dec(mem_pool_cond);
 
    }
    
    return ptr;
}
 
static inline void mem_pool_free (void *ptr) {
    mm_free(ptr);
    // notify malloc
    SemUtil::set(mem_pool_cond, 1);
 
}
 
static inline void *mem_pool_realloc (void *ptr, size_t size) {
    return mm_realloc(ptr, size);
}
 
static inline hashtable_t * mem_pool_get_hashtable() {
    return mm_get_hashtable();
 
}
// extern int mm_checkheap(int verbose);
 
 
#endif