lw_oopc/demo/Expr/lw_oopc.c

// Copyright (C) 2008,2009,2010 by Tom Kao & MISOO Team & Yonghua Jin. All rights reserved.
// Released under the terms of the GNU Library or Lesser General Public License (LGPL).
// Author: Tom Kao(<28><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>߻<EFBFBD><DFBB><EFBFBD>)<29><>MISOO<4F>Ŷӣ<C5B6>Yonghua Jin(<28><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>)
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//    * Redistributions of source code must retain the above copyright
//	notice, this list of conditions and the following disclaimer.
//    * Redistributions in binary form must reproduce the above
//	copyright notice, this list of conditions and the following disclaimer
//	in the documentation and/or other materials provided with the
//	distribution.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#include "stdio.h"
#include "stdlib.h"
#include "string.h"
#include "lw_oopc.h"

// <20>Ƿ<EFBFBD>֧<EFBFBD>ֵ<EFBFBD><D6B5><EFBFBD><EFBFBD><EFBFBD>Ϣ<EFBFBD><CFA2>ӡ(<28>ڴ<EFBFBD><DAB4><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ͷŵ<CDB7><C5B5><EFBFBD>ϸ<EFBFBD><CFB8>Ϣ<EFBFBD><CFA2><EFBFBD><EFBFBD>ȱʡ<C8B1>رմ<D8B1>ӡ
//#define LW_OOPC_PRINT_DEBUG_INFO

#ifdef LW_OOPC_PRINT_DEBUG_INFO
    #define lw_oopc_dbginfo printf
#else
    #define lw_oopc_dbginfo
#endif

#define LW_OOPC_MAX_PATH   260
#define LW_OOPC_MEMORY_DETECTOR_RST "memory_detector_result.txt"

typedef struct LW_OOPC_MemAllocUnit
{
    char file[LW_OOPC_MAX_PATH];    // <20>ļ<EFBFBD><C4BC><EFBFBD>
    int line;                       // <20>к<EFBFBD>
    void* addr;                     // <20>ڴ<EFBFBD><DAB4><EFBFBD>ַ
    size_t size;                    // <20>ڴ<EFBFBD><DAB4><EFBFBD><EFBFBD><EFBFBD>С
    struct LW_OOPC_MemAllocUnit* next;     // <20><>һ<EFBFBD><D2BB><EFBFBD>ڴ<EFBFBD><DAB4><EFBFBD>
} LW_OOPC_MemAllocUnit;

#ifdef LW_OOPC_SUPPORT_MEMORY_LEAK_DETECTOR

static LW_OOPC_MemAllocUnit* lw_oopc_memAllocList = 0;

void* lw_oopc_malloc(size_t size, const char* type, const char* file, int line)
{
    void* addr = malloc(size);
    if (addr != 0)
    {
        LW_OOPC_MemAllocUnit* pMemAllocUnit = malloc(sizeof(LW_OOPC_MemAllocUnit));
        if (!pMemAllocUnit)
        {
            fprintf(stderr, "lw_oopc: error! malloc alloc unit failed.\n");
            exit(1);
        }

        if (strlen(file) >= LW_OOPC_MAX_PATH)
        {
            fprintf(stderr, "lw_oopc: error! file name is more than %d character: %s\n", LW_OOPC_MAX_PATH, file);
            exit(1);
        }

        strcpy(pMemAllocUnit->file, file);

        pMemAllocUnit->line = line;
        pMemAllocUnit->addr = addr;
        pMemAllocUnit->size = size;
        pMemAllocUnit->next = lw_oopc_memAllocList;
        lw_oopc_memAllocList = pMemAllocUnit;

        lw_oopc_dbginfo("lw_oopc: alloc memory in %p, size: %d, object type: %s, file: %s, line: %d\n", addr, size, type, file, line);
    }

    return addr;
}

void lw_oopc_free(void* memblock)
{
    LW_OOPC_MemAllocUnit* prevUnit = 0;
    LW_OOPC_MemAllocUnit* currUnit = lw_oopc_memAllocList;

    while(currUnit != 0)
    {
        if (currUnit->addr == memblock)
        {
            lw_oopc_dbginfo("lw_oopc: free memory in %p, size: %d\n", currUnit->addr, currUnit->size);
            if (prevUnit == 0)
            {
                lw_oopc_memAllocList = currUnit->next;
                free(currUnit->addr);
                return;
            }
            
            prevUnit->next = currUnit->next;
            return;
        }
        else
        {
            prevUnit = currUnit;
            currUnit = currUnit->next;
        }
    }

    if (currUnit == 0)
    {
        fprintf(stderr, "lw_oopc: error! you attemp to free invalid memory.\n");
        exit(1);
    }
}

void lw_oopc_report()
{
    LW_OOPC_MemAllocUnit* currUnit = lw_oopc_memAllocList;
    FILE* fp = fopen(LW_OOPC_MEMORY_DETECTOR_RST, "w+");

    if(!fp)
    {
        fprintf(stderr, "lw_oopc: error occured, open file: %s failed.\n", LW_OOPC_MEMORY_DETECTOR_RST);
    }
    
    if (currUnit != 0)
    {
        fprintf(stderr, "lw_oopc: memory leak:\n");
        if(fp)
        {
            fprintf(fp, "lw_oopc: memory leak:\n");
        }

        while(currUnit != 0)
        {
            fprintf(stderr, "memory leak in: %p, size: %d, file: %s, line: %d\n", currUnit->addr, currUnit->size, currUnit->file, currUnit->line);
            if(fp)
            {
                fprintf(fp, "memory leak in: %p, size: %d, file: %s, line: %d\n", currUnit->addr, currUnit->size, currUnit->file, currUnit->line);
            }
            currUnit = currUnit->next;
        }
    }
    else
    {
        printf("lw_oopc: no memory leak.\n");
        if(fp)
        {
            fprintf(fp, "lw_oopc: no memory leak.\n");
        }
    }

    if(fp)
    {
        fclose(fp);
    }
}
#else

void lw_oopc_report()
{
}

#endif