学步园

在帖子如何判断一个C++对象是否在堆栈上 中， 又有人提出如何判断一个C++对象是否在堆上。

其实我们可以参照那个帖子的方法类似实现，我们知道堆就是Heap，在windows上我们可以通过GetProcessHeaps来得到所有的堆句柄，而我们这里只要知道Windows上的Heap Handle，其实就是堆的起始地址，就可以写如下代码了。

#include <iostream>
#include <windows.h>

using namespace std;

BOOL IsObjectOnHeap(LPVOID pObject)
{
    BOOL bRet(FALSE);
    DWORD dwHeaps = GetProcessHeaps(0, NULL);
    LPHANDLE pHeaps = new HANDLE[dwHeaps];
    if(pHeaps != NULL)
    {
        MEMORY_BASIC_INFORMATION mi = {0};
        GetProcessHeaps(dwHeaps, pHeaps);

        for(INT i=0; i<dwHeaps; ++i)
        {
            VirtualQuery((LPVOID)pHeaps[i], &mi, sizeof(mi));
            
            if(pObject >= mi.BaseAddress 
                && (DWORD)pObject < (DWORD)mi.BaseAddress + mi.RegionSize)
            {
                bRet = TRUE;
                break;
            }
        }
    }
    delete []pHeaps;

    return bRet;
}

int g_value = 10;

int main(int argc, char* argv[])
{
    int nStackValue = 1;
    int* pNew = new int(10);
    int* pNewArray = new int[100];

    static int static_value = 0;
    
    cout << "g_value:" << IsObjectOnHeap(&g_value) << endl;  //false
    cout << "nStackValue:" << IsObjectOnHeap(&nStackValue) << endl;   //false
    cout << "static_value:" << IsObjectOnHeap(&static_value) << endl;    //false
    cout << "pNew:" << IsObjectOnHeap(pNew) << endl;        //true
    cout << "pNewArray:" << IsObjectOnHeap(pNewArray) << endl;    //true

    system("pause");
    return 0;
}

以上代码在Windows下测试通过（也只能在Windows上跑），如果有不正确的地方，欢迎指正。

原文地址:  http://www.cppblog.com/weiym/archive/2012/05/12/174657.html