/*
 * 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_mem"

#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 "mpp_common.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;
static pthread_mutex_t mem_list_lock;

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

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

static void get_osal_mem_flag()
{
    if (osal_mem_flag < 0) {
        RK_U32 val;

        osal_mem_flag = 0;
        mpp_env_get_u32(CONFIG_OSAL_MEM_LIST, &val, 0);
        if (val) {
            osal_mem_flag |= OSAL_MEM_LIST_EN;
        }
        mpp_env_get_u32(CONFIG_OSAL_MEM_STUFF, &val, 0);
        if (val) {
            osal_mem_flag |= OSAL_MEM_STUFF_EN;
        }

        INIT_LIST_HEAD(&mem_list);

        pthread_mutexattr_t attr;
        pthread_mutexattr_init(&attr);
        pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
        pthread_mutex_init(&mem_list_lock, &attr);
        pthread_mutexattr_destroy(&attr);
    }
}

void *mpp_osal_malloc(const 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);
            node->ptr   = ptr;
            node->size  = size;
            snprintf(node->tag, sizeof(node->tag), "%s", tag);

            pthread_mutex_lock(&mem_list_lock);
            list_add_tail(&node->list, &mem_list);
            pthread_mutex_unlock(&mem_list_lock);
        }

        return ptr;
    } else
        return NULL;
}

void *mpp_osal_calloc(const char *tag, size_t size)
{
    void *ptr = mpp_osal_malloc(tag, size);
    if (ptr)
        memset(ptr, 0, size);
    return ptr;
}

void *mpp_osal_realloc(const 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;
        pthread_mutex_lock(&mem_list_lock);
        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;
            }
        }
        pthread_mutex_unlock(&mem_list_lock);
    } else {
        os_realloc(ptr, &ret, RK_OSAL_MEM_ALIGN, 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;

        pthread_mutex_lock(&mem_list_lock);
        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;
            }
        }
        pthread_mutex_unlock(&mem_list_lock);
    }

    os_free(ptr);
}

/*
 * dump memory status
 * this function need MODULE_TAG statistic information
 */
void mpp_show_mem_status()
{
    struct mem_node *pos, *n;

    pthread_mutex_lock(&mem_list_lock);
    list_for_each_entry_safe(pos, n, &mem_list, struct mem_node, list) {
        mpp_log("unfree memory %p size %d tag %s", pos->ptr, pos->size, pos->tag);
    }
    pthread_mutex_unlock(&mem_list_lock);
}