94 lines
1.6 KiB
C
94 lines
1.6 KiB
C
/*
|
|
* Copyright (C) 2020-2022 Artinchip Technology Co. Ltd
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*
|
|
* author: qi.xu@artinchip.com
|
|
* Desc: virtual memory allocator
|
|
*/
|
|
|
|
#include <unistd.h>
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "mpp_mem.h"
|
|
|
|
#define DEBUG_MEM (0)
|
|
|
|
#if DEBUG_MEM
|
|
#include <pthread.h>
|
|
#define MAX_NUM 20
|
|
struct memory_node {
|
|
void* addr;
|
|
int size;
|
|
int used;
|
|
};
|
|
|
|
static pthread_mutex_t g_mem_mutex = PTHREAD_MUTEX_INITIALIZER;
|
|
int g_mem_count = 0;
|
|
int g_mem_total_size = 0;
|
|
struct memory_node g_mem_info[MAX_NUM] = {{0,0,0}};
|
|
|
|
#endif
|
|
|
|
void *mpp_alloc(size_t size)
|
|
{
|
|
void *ptr = NULL;
|
|
|
|
ptr = malloc(size);
|
|
|
|
#if DEBUG_MEM
|
|
int i = 0;
|
|
pthread_mutex_lock(&g_mem_mutex);
|
|
g_mem_count ++;
|
|
g_mem_total_size += size;
|
|
|
|
for(i=0; i<MAX_NUM; i++) {
|
|
if(g_mem_info[i].used == 0) {
|
|
g_mem_info[i].addr = ptr;
|
|
g_mem_info[i].size = size;
|
|
g_mem_info[i].used = 1;
|
|
logi("alloc memory, addr: %x, %d", ptr, size);
|
|
break;
|
|
}
|
|
}
|
|
if(i == MAX_NUM) {
|
|
loge("exceed max number");
|
|
}
|
|
pthread_mutex_unlock(&g_mem_mutex);
|
|
#endif
|
|
return ptr;
|
|
}
|
|
|
|
void mpp_free(void *ptr)
|
|
{
|
|
if(ptr == NULL)
|
|
return;
|
|
|
|
free(ptr);
|
|
|
|
#if DEBUG_MEM
|
|
int i;
|
|
pthread_mutex_lock(&g_mem_mutex);
|
|
for(i=0; i<MAX_NUM; i++) {
|
|
if(g_mem_info[i].used && g_mem_info[i].addr == ptr) {
|
|
g_mem_count--;
|
|
g_mem_total_size -= g_mem_info[i].size;
|
|
logi("free memory, addr: %x, %d", ptr, g_mem_info[i].size);
|
|
break;
|
|
}
|
|
}
|
|
if(i == MAX_NUM) {
|
|
loge("cannot find this memory");
|
|
}
|
|
logi("total memory size: %d", g_mem_total_size);
|
|
pthread_mutex_unlock(&g_mem_mutex);
|
|
#endif
|
|
}
|
|
|
|
void *mpp_realloc(void *ptr,size_t size)
|
|
{
|
|
return realloc(ptr,size);
|
|
}
|