lihao/skynet
1
0
Fork 0
mirror of https://github.com/cloudwu/skynet.git synced 2026-07-24 12:20:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

rewrite get_allocated_field

Browse Source
This commit is contained in:
Cloud Wu
2014-04-18 23:40:50 +08:00
parent 2680425b4f
commit 23764545bb
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
skynet-src/malloc_hook.c
View File

@@ -12,7 +12,7 @@ static size_t _used_memory = 0;
static size_t _memory_block = 0;
typedef struct _mem_data {
uint32_t handle;
size_t allocated;
ssize_t allocated;
} mem_data;
#define SLOT_SIZE 0x10000
@@ -24,11 +24,16 @@ static size_t*
get_allocated_field(uint32_t handle) {
int h = (int)(handle & (SLOT_SIZE - 1));
mem_data *data = &mem_stats[h];
if(data->handle == 0 || data->allocated == 0) {
__sync_synchronize();
if(!__sync_bool_compare_and_swap(&data->handle, 0, handle)) {
return 0;
}
uint32_t old_handle = data->handle;
ssize_t old_alloc = data->allocated;
if(old_handle == 0 || old_alloc <= 0) {
// data->allocated may less than zero, because it may not count at start.
if(!__sync_bool_compare_and_swap(&data->handle, old_handle, handle)) {
return 0;
}
if (old_alloc < 0) {
__sync_bool_compare_and_swap(&data->allocated, old_alloc, 0);
}
}
if(data->handle != handle) {
return 0;