/************************************************************************
 *
 *	Copyright (C) 2003-2004
 *	Shenzhen SCADA Control Technology Co., Ltd.
 *	All rights reserved.
 *	
 *	用于调试内存泄漏的对内存分配函数的实现
 *
 *	创建日期:	2005/04/12
 *
 ***********************************************************************/
/*#ifdef OS_LINUX
#include <sys/types.h>
#include <sys/time.h>
#endif*/

#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
#include <string.h>
#include <errno.h>

#include "time.h"
#include "list_entry.h"
#include "os_heap.h"

#if defined(_WIN32)
  #include <winsock2.h>
  #include <direct.h>
  #include <process.h>
#endif

#ifdef _fclose
#undef _fclose
#endif
#define _fclose(_fp) do {if (_fp) {while (EOF == fclose((_fp)) && EINTR == errno);}} while(0)

/*!
 *  堆内存块的类型
 */
typedef enum {
    eMalloc = 0, /* 使用malloc分配的内存块 */
    eCalloc = 1  /* 使用calloc分配的内存块 */
} heap_type_t;

/*!
 *  堆内存块附加头
 *
 *  \note
 *      必须保证本结构的尺寸被16整除。
 */
typedef struct tag_heap_head_t {

    /* 堆内存块附加头所在的链表的占位 */
    list_entry_t    anchor;

    /*! 堆内存块分配发生的时间 */
    struct timeval     tv;

    /*! 堆内存块类型 */
    heap_type_t     type;

    /*! 堆内存块分配发生的源文件名，NULL表示匿名源文件。 */
    char * fl;

    /*! 堆内存块分配发生的源代码行 */
    int  line;

    /*! 堆内存块的尺寸(字节) */
    size_t size;

} heap_list_t;


/* 边界对齐量 */
#define os_heap_alignment   16
#define os_heap_pad ((sizeof(heap_list_t) % os_heap_alignment) ? (os_heap_alignment - (sizeof(heap_list_t) % os_heap_alignment)) : 0)

/* 根据数据地址求os_heap块的起始地址 */
#define os_heap_start_addr(_data) ((void *)(((unsigned char *)(_data)) - sizeof(heap_list_t) - os_heap_pad))

/* 根据os_heap块的起始地址求数据地址 */
#define os_heap_data_addr(_start) ((void *)(((unsigned char *)(_start)) + sizeof(heap_list_t) + os_heap_pad))

/* 根据数据块的尺寸求os_heap块的尺寸 */
#define os_heap_mb_size(_size) (sizeof(heap_list_t) + os_heap_pad + (_size))

/*
 *  内存管理模块：
 *
 *  1. 初始化过程
 *  2. 具有一个私有线程，专门侦听外部的信号，并打印报告。
 *  3. 堆内存块链表头、尾指针。
 *  4. 互斥访问保护
 */

/* 堆内存块设施是否已经初始化 */
static int          g_heap_initialized = 0;

/* 堆内存块双向链表头 */
static list_entry_t g_heap_head;

/*
 *  进程中使用heap_malloc/heap_calloc分配的内存
 *  且现在还在使用的内存块的总数。
 */
static size_t       g_heap_item_counter = 0;

/* 
 *  进程中使用heap_malloc/heap_calloc分配的内存
 *  且现在还在使用的总量(字节) 
 */
static size_t       g_heap_byte_counter = 0;


/* 初始化堆内存块设施 */
static void heap_intialize();

/*!
 *  \brief 堆内存分配
 *
 *  \param size 希望分配的内存的尺寸
 *  \param file 指向发生堆内存分配的源代码文件名称，以null字符结尾。
 *              如果等于NULL， 表示匿名源代码文件名(anony)
 *  \param line 发生堆内存分配的源代码行号
 *
 *  \retutrn 
 *      对内存分配成功的情况下，返回指向分配的内存的指针。如果内存
 *      分配失败，返回NULL。
 *
 *  \note
 *      当分配的内存不在使用时，必须调用heap_free是否分配的内存。
 *
 *      参见 ANSI C 函数 malloc。
 */
void * heap_malloc (
        size_t size, 
        const char * file, 
        int line )
{
 // #ifdef DISABLE_OS_HEAP
    return malloc(size);
  /*#else /* #ifdef DISABLE_OS_HEAP */
/*
    struct timeval tv;
    char * fl = NULL;
    void * mb = NULL;
    heap_list_t * hl = NULL;

    if (0 == size) 
        return NULL;

    if (file) {
        fl = (char *)malloc(strlen(file) + 4);
        if (fl) strcpy (fl, file);
    }

    gettimeofday(&tv, NULL);

    mb = malloc(os_heap_mb_size(size));
    if (NULL == mb) {
        if (fl) free (fl);
        return mb;
    } else {
        memset (mb, 0, os_heap_mb_size(size));
    }

    hl = (heap_list_t *)mb;

    hl->type = eMalloc;
    hl->tv = tv;
    hl->fl = fl;
    hl->line = line;
    hl->size = size;

    if (!g_heap_initialized) {
        heap_intialize();
        g_heap_initialized = 1;
    }

    insert_tail_list (&g_heap_head, &(hl->anchor));
    g_heap_item_counter++;
    g_heap_byte_counter += hl->size;
    
    return os_heap_data_addr(hl);
  #endif *//* #ifdef DISABLE_OS_HEAP */
}

/*!
 *  \brief 堆内存分配, 且内存块被初始化0.
 *
 *  \param num      内存块个数
 *  \param elm_size 每一个内存块的尺寸(字节)
 *  \param file 指向发生堆内存分配的源代码文件名称，以null字符结尾。
 *              如果等于NULL， 表示匿名源代码文件名(anony)
 *  \param line 发生堆内存分配的源代码行号
 *
 *  \retutrn 
 *      对内存分配成功的情况下，返回指向分配的内存的指针。如果内存
 *      分配失败，返回NULL。
 *
 *  \note
 *      当分配的内存不在使用时，必须调用heap_free释放分配的内存。
 *      
 *      参见 ANSI C 函数 calloc。本函数与 calloc 不同之处在于不能
 *      象calloc那样保证返回的堆内存块的其实地址对齐在elm_size边界
 *      上。
 */
void * heap_calloc (
        size_t num, 
        size_t elm_size, 
        const char * file, 
        int line )
{
  //#ifdef DISABLE_OS_HEAP
    return calloc(num, elm_size);
  //#else /* #ifdef DISABLE_OS_HEAP */

  /*  struct timeval tv;
    char * fl = NULL;
    void * mb = NULL;
    heap_list_t * hl = NULL;

    if (0 == num * elm_size)
        return NULL;

    if (file) {
        fl = (char *)malloc(strlen(file) + 4);
        if (fl) strcpy (fl, file);
    }

    gettimeofday(&tv, NULL);

    mb = malloc(os_heap_mb_size(elm_size * num));
    if (NULL == mb) {
        if (fl) free (fl);
        return mb;
    } else {
        memset (mb, 0, os_heap_mb_size(elm_size * num));
    }

    hl = (heap_list_t *)mb;

    hl->type = eCalloc;
    hl->tv = tv;
    hl->fl = fl;
    hl->line = line;
    hl->size = elm_size * num;

    if (!g_heap_initialized) {
        heap_intialize();
        g_heap_initialized = 1;
    }

    insert_tail_list (&g_heap_head, &(hl->anchor));
    g_heap_item_counter++;
    g_heap_byte_counter += hl->size;
    
    return os_heap_data_addr(hl);
  #endif *//* #ifdef DISABLE_OS_HEAP */
}

/*!
 *  \brief  释放堆内存
 *
 *  \param  memblock  指向先前调用heap_alloc/heap_calloc分配的内存。
 *
 *  \return 
 *      无。
 */
void heap_free (void * mb)
{
  #ifdef DISABLE_OS_HEAP
    free (mb);
  #else /* #ifdef DISABLE_OS_HEAP */
    size_t sz = 0;
    heap_list_t * hl = NULL;

    if (NULL == mb)
        return;

    if (!g_heap_initialized) {
        heap_intialize();
        g_heap_initialized = 1;
        return;
    }

    hl = (heap_list_t *)os_heap_start_addr(mb);

    #if defined(WIN32) && (defined(DEBUG) || defined(_DEBUG))
    if (IsBadReadPtr((void *)hl, sizeof(heap_list_t))) {
        return;
    }
    
    if (IsBadReadPtr((void *)hl, os_heap_mb_size(hl->size))) {
        return;
    }
    #endif

    remove_entry_list (&(hl->anchor));
    sz = hl->size;
    if (hl->fl) 
        free ((void *)(hl->fl));
    free ((void *)hl);
    
    g_heap_item_counter--;
    g_heap_byte_counter -= sz;
    
  #endif /* #ifdef DISABLE_OS_HEAP */
}

/*!
 *  \brief  报告堆内存分配
 *
 *  调用本函数将堆内存分配情况以追加的方式输出到指定文件中。
 *
 *  \param fname 对内存分配报告将输出到该文件中。如果fname等于NULL，
 *              则报告将输出到标准输出上。
 *
 *  \retval -1 函数调用失败。
 *  \retval 0  函数调用成功。
 *
 *  \note
 *      报告的格式如下
 *      时间: 
 *      内存使用总量: 
 *      ......
 *      内存块列表i: 
 *          时间: tv
 *          类型: malloc/calloc, 
 *          源码位置: fl(line)
 *          尺寸: size
 *      ......
 */
int heap_report(const char * fname)
{
 /* #ifndef DISABLE_OS_HEAP
    int i = 0;
    FILE * fp = NULL;

    heap_list_t * hl = NULL;

    list_entry_t * item = NULL;
	list_entry_t * first = NULL;
	list_entry_t * next = NULL;

    char szdt[32];

    if (fname) {
        fp = fopen (fname, "a+t");
        if (NULL == fp)
            return -1;
    } else
        fp = stdout;

    if (!g_heap_initialized) {
        heap_intialize();
        g_heap_initialized = 1;
        if (fname) _fclose(fp);
        return 0;
    }

    fprintf (fp, "\n"
                 "heap allocation summary...\n"
                 "time: %s\n"
                 "heap_item_counter: %u\n"
                 "heap_byte_counter: %u(bytes)\n",
                 cur_tm_string (szdt), 
                 g_heap_item_counter,
                 g_heap_byte_counter);
    
    first = &g_heap_head;
	for ( next = first->flink; next != first; next = next->flink )
	{
		item = next;
		hl = CONTAINING_RECORD ( item, heap_list_t, anchor );

        fprintf (fp, "  heap block %d:"
                     " times: %s"
                     " type: %s"
                     " size: %08u (bytes)"
                     " source: %s [%d]\n",
                     i++, timeval_string(&(hl->tv), szdt),
                     (eMalloc == hl->type ? "malloc" : "calloc"),
                     hl->size, (hl->fl ? hl->fl : "nony"), hl->line);
	}

    if (fname) _fclose(fp);
  #endif*/

    return 0;
}

static void heap_intialize()
{
    Initialize_list_head (&g_heap_head);
}