/*
 * Copyright 2010 Rockchip Electronics S.LSI Co. LTD
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#define MODULE_TAG "mpp_malloc"

#include <string.h>

#include "rk_type.h"
#include "mpp_err.h"

#include "mpp_log.h"
#include "mpp_env.h"
#include "mpp_mem.h"
#include "mpp_list.h"

#include "os_mem.h"

// export configure for script detection
#define CONFIG_OSAL_MEM_LIST        "osal_mem_list"
#define CONFIG_OSAL_MEM_STUFF       "osal_mem_stuff"


// default memory align size is set to 32
#define RK_OSAL_MEM_ALIGN       32

// osal_mem_flag bit mask
#define OSAL_MEM_LIST_EN        (0x00000001)
#define OSAL_MEM_STUFF_EN       (0x00000002)

static RK_S32 osal_mem_flag = -1;
static struct list_head mem_list;

struct mem_node {
    struct list_head list;
    void *ptr;
    size_t size;

    /* memory node extra information */
    char tag[32];
};

static void get_osal_mem_flag()
{
    if (osal_mem_flag < 0) {
        RK_U32 val;
        osal_mem_flag = 0;
        mpp_get_env_u32(CONFIG_OSAL_MEM_LIST, &val, 0);
        if (val) {
            osal_mem_flag |= OSAL_MEM_LIST_EN;
        }
        mpp_get_env_u32(CONFIG_OSAL_MEM_STUFF, &val, 0);
        if (val) {
            osal_mem_flag |= OSAL_MEM_STUFF_EN;
        }
        INIT_LIST_HEAD(&mem_list);
    }
}

void *mpp_osal_malloc(char *tag, size_t size)
{
    void *ptr;
    get_osal_mem_flag();

    if (MPP_OK == os_malloc(&ptr, RK_OSAL_MEM_ALIGN, size)) {
        if (osal_mem_flag & OSAL_MEM_LIST_EN) {
            struct mem_node *node = (struct mem_node *)malloc(sizeof(struct mem_node));
            INIT_LIST_HEAD(&node->list);
            list_add_tail(&node->list, &mem_list);
            node->ptr   = ptr;
            node->size  = size;
            strncpy(node->tag, tag, sizeof(node->tag));
        }

        return ptr;
    } else
        return NULL;
}

void *mpp_osal_realloc(char *tag, void *ptr, size_t size)
{
    void *ret;

    if (NULL == ptr)
        return mpp_osal_malloc(tag, size);

    if (0 == size)
        return NULL;

    get_osal_mem_flag();

    if (osal_mem_flag & OSAL_MEM_LIST_EN) {
        struct mem_node *pos, *n;
        ret = NULL;
        list_for_each_entry_safe(pos, n, &mem_list, struct mem_node, list) {
            if (ptr == pos->ptr) {
                if (MPP_OK == os_realloc(ptr, &pos->ptr, RK_OSAL_MEM_ALIGN, size)) {
                    pos->size   = size;
                    strncpy(pos->tag, tag, sizeof(pos->tag));
                    ret = pos->ptr;
                } else {
                    list_del_init(&pos->list);
                    free(pos);
                }
                break;
            }
        }
    } else {
        ret = realloc(ptr, size);
    }

    if (NULL == ret)
        mpp_err("mpp_realloc ptr 0x%p to size %d failed\n", ptr, size);

    return ret;
}

void mpp_osal_free(void *ptr)
{
    if (NULL == ptr)
        return;

    get_osal_mem_flag();

    if (osal_mem_flag & OSAL_MEM_LIST_EN) {
        struct mem_node *pos, *n;
        list_for_each_entry_safe(pos, n, &mem_list, struct mem_node, list) {
            if (ptr == pos->ptr) {
                list_del_init(&pos->list);
                free(pos);
                break;
            }
        }
    }

    os_free(ptr);
}

/*
 * dump memory status
 * this function need MODULE_TAG statistic information
 */
void mpp_show_mem_status()
{
    // TODO: add memory dump implement
}