防御性编程之三

1,一个常用的宏,用于输出正在执行的语句.

#define TEACE(ARG) cout<<#ARG<<endl; ARG

2,一个跟踪new动态分配内存的程序

#include <cstdio>
#include <cstddef>
#include <cstdlib>
#include <cassert>

using namespace std;

bool traceFlag=true; //标识是否跟踪动态信息

namespace{

struct Info //保存跟踪信息
{
    void* ptr; //地址
    const char* file;//文件名
    long line; //行号
};

const size_t MAXPTRS=10000u; //最多可以分配的地址数
Info memMap[MAXPTRS];
size_t nptrs=0; //已经分配的地址数

int findPtr(void* p) //返回地址的位置
{
    for(size_t i=0;i<nptrs;i++)
        if(memMap[i].ptr==p)
            return i;
    return -1;
}

void deletePtr(void* p) //回收这个地址
{
    int pos=findPtr(p);
    assert(pos>=0);
    for(size_t i=pos;i<nptrs-1;i++)
        memMap[i]=memMap[i+1];
    nptrs--;
}

struct Report //利用全局对象的析构函数,输出最后的测试结果
{
    ~Report()
    {
        if(nptrs>0)
        {
            printf("Leaked memory at:\n");
            for(size_t i=0;i<nptrs;i++)
                printf("%p file:%s,line:%ld\n",memMap[i].ptr,memMap[i].file,memMap[i].line);
        }
        else
            printf("No memory Leaks.\n");
    }
};
Report r; //全局对象

} //这里没有分号

void* operator new(std::size_t sz, const char* file, long line)
{
    void* p=malloc(sz);
    if(nptrs==MAXPTRS)
    {
        printf("No enougth space!\n");
        exit(1);
    }
    memMap[nptrs].ptr=p;
    memMap[nptrs].file=file;
    memMap[nptrs].line=line;
    nptrs++;
    if(traceFlag) //输出跟踪信息
        printf("Allocated %u bytes at address %p,file: %s,line: %ld\n",sz,p,file,line);
    return p;
}
void* operator new[](std::size_t sz, const char* file, long line)
{
    return operator new(sz, file, line);
}


void operator delete(void* p)
{
    if(findPtr(p)>=0)
    {
        free(p);
        assert(nptrs>0);
        deletePtr(p);
        if(traceFlag) //输出跟踪信息
            printf("Delete memoey at address %p\n",p);
    }
    else if(p)
        printf("Attempt to delete unknown pointer:%p\n",p);
}
void operator delete[](void* p)
{
    operator delete(p);
}

#define TRACE_ON() traceFlag=true;
#define TRACE_OFF() traceFlag=false;
int main()
{
    //TRACE_OFF();
    int* p=(int*)operator new(3,__FILE__,__LINE__);
    operator delete(p);
    int* p2;
    operator delete(p2);
    return 0;
}