代码测试之内存泄露

　　内存泄露的危害是不言而喻的，但是查找内存泄露却是一件苦难而且复杂的工作。我们都知道，解决bug是一件非常简单的事情，但是寻找bug的出处却是一件非常吃力的事情。因此，我们有必要在自己编写代码的时候，把查找内存泄露的工作放在很重要的位置上面。那么有没有什么办法来解决这一问题呢？

　　我想要做到解决内存泄露，必须做到下面两个方面：

　　（1）必须记录内存在哪个函数申请的，具体文件的行数是多少

　　（2）内存应该什么时候被释放

　　要完成第1个条件其实并不困难。我们可以用节点的方法记录我们申请的内存：

　　a）设置节点的数据结构

typedef struct _MEMORY_NODE 
{ 
    char functionName[64]; 
    int line; 
    void* pAddress; 
    struct _MEMORY_NODE* next; 
 
}MEMORY_NODE;
 　　其中 functionName记录函数名称，line记录行数， pAddress记录分配的地址， next记录下一个内存节点。

　　b）修改内存的分配函数

　　对业务侧的malloc进行函数修改，添加下面一句宏语句

　　#define malloc(param)  MemoryMalloc(__FUNCTION__， __LINE__， param)

　　在桩函数侧书写下面的代码

void* MemoryMalloc(const char* name， int line， int size) 
{ 
    void* pData = (void*)malloc(size); 
    MEMORY_NODE* pMemNode = NULL; 
    if(NULL == pData) return NULL; 
    memset((char*)pData， 0， size); 
 
    pMemNode = (MEMORY_NODE*)malloc(sizeof(MEMORY_NODE)); 
    if(NULL == pMemNode){ 
        free(pData); 
        return NULL; 
    } 
    memset((char*)pMemNode， 0， sizeof(MEMORY_NODE)); 
    memmove(pMemNode->functionName， name， strlen(name)); 
    pMemNode->line = line; 
    pMemNode->pAddress = pData; 
    pMemNode->next = NULL; 
    add_memory_node(pMemNode); 
 
    return pData; 
}
 　　内存的分配过程中还涉及到了节点的添加，所以我们还需要添加下面的代码

static MEMORY_NODE* gMemNode = NULL;  
  
void add_memory_node(MEMORY_NODE* pMemNode)  
{  
    MEMORY_NODE* pNode = gMemNode;  
    if(NULL == pMemNode) return;  
    if(NULL == gMemNode){  
        gMemNode = pMemNode;  
        return;  
    }  
  
    while(NULL != pNode->next){  
        pNode = pNode->next;  
    }  
    pNode->next = pMemNode;  
    return;  
}
 
　　文中gMemNode表示所有内存节点的根节点，我们每增加一次malloc过程会对内存节点进行记录。在记录过程中，我们还会记录调用malloc的函数名称和具体文件行数，这主要是为了方便我们在后面进行故障定位的时候更好地查找。

　　完成了第一个条件之后，我们要对第二个条件进行完成。

　　a）内存什么时候释放，这取决于我们在函数中是怎么实现的，但是我们在编写测试用例的时候却是应该知道内存释放没有，比如说如果测试用例全部结束了，我们有理由相信assert(gMemNode == NULL)这应该是恒等于真的。

　　b）内存释放的时候，我们应该做些什么？和节点的添加一样，我们在内存释放的时候需要free指定的内存，free节点，free节点的内存，下面是在释放的时候我们需要进行的操作

　　对业务侧的free函数进行修改，添加下面一句宏代码，

　　#define free(param)      MemoryFree(param)

　　在桩函数侧输入下面的代码：

　　view plaincopy to clipboardprint?

void MemoryFree(void* pAddress)  
{  
    if(NULL == pAddress) return;  
    delete_memory_node(pAddress);  
    free(pAddress);  
}
 　　在删除内存的时候，需要删除节点，删除节点的内存

void delete_memory_node(void* pAddress)  
{  
    MEMORY_NODE* pHead = gMemNode;  
    MEMORY_NODE* pMemNode = gMemNode;  
    while(NULL != pMemNode){  
        if(pAddress == pMemNode->pAddress)  
            break;  
        pMemNode = pMemNode->next;  
    }  
    if(NULL == pMemNode) {  
        assert(1 == 0);  
        return;  
    }  
  
    while(pMemNode != pHead->next){  
        pHead = pHead->next;  
    }  
  
    if(pMemNode == gMemNode){  
        gMemNode = gMemNode->next;  
    }else{  
        pHead->next = pMemNode->next;  
    }  
    free(pMemNode);  
    return;  
}
 　　有了上面一小段代码的帮助，我们在编写测试用例的时候，可以在函数执行后，通过判断内存节点是否为空的方法判断内存是否已经释放。如果内存没有释放，我们还能通过节点的信息帮助我们是哪里发生了错误，但是这个方法还有两个缺点：

　　（1）没有考虑缓存的情况，好多内存分配了之后并不会在函数中马上释放，而是放在缓存池中等待下一次调用，这需要我们准确把握和判断了。