闲着无事,写了一个检测内存越界访问的工具(windows平台)

// 文件名:memcheck.h
// 作者 : Moonwell
// msn : [email protected]
// 在需要检测内存越界的地方加上#include "memcheck.h"
// 程序运行过程中,出现内存越界访问的代码将会出现中断

#ifndef _MEM_CHECK_H
#define _MEM_CHECK_H

void * operator new (size_t size)
{
//创建的页面个数
int page_num = (int) ( size / 4096 + 1 );

//偏移量
size_t offset = page_num * 4096 - size;

//在内存块的后面创建一个额外的保护页面,并且将页面的属性设置为不可读写
void *p = VirtualAlloc(NULL,page_num*4096+1,MEM_COMMIT,PAGE_EXECUTE_READWRITE);

//定位最后一个保护页面的地址
void *pchecker = (char *)p + page_num*4096;

//设置最后一页为不可读写
DWORD old_value;
VirtualProtect(pchecker,4096,PAGE_NOACCESS,&old_value);

return (char *)p + offset;
}
 
void operator delete (void *p)
{
//找到VirtualAlloc时候的起始地址
p = (char *)p - (size_t)p % 4096;

VirtualFree(p,0,MEM_RELEASE);
}

#endif

//总共没有几行代码,使用举例如下:

#include <windows.h>
#include "memcheck.h"

int main(int argc,char *argv[])
{
       char * a = new char[12340];
       printf("%c",a[12340]); //程序运行到这里将出现debug中断

       delete [] a;
       return 0;
}

注意:每次new操作,至少会创建比原来多4k的内存,比较耗内存.

本文地址：http://com.8s8s.com/it/it22312.htm