在自定义堆内存设置(Setting memory on a custom heap)

2019-10-19 23:28发布

是否有可能采取从HeapCreate(堆手柄),并设置所有可用内存在一定的价值?

我试着按区域枚举堆和设置这样的说法,但我得到的访问冲突。

我基本上要在我的堆内存设置为我可用来进行调试,然后再将其通过HeapDestroy最终摧毁了一个自定义值。

Answer 1:

简短的回答是“不,你不能填补特定值堆”。

你可以用调试版本包装你的堆访问功能,例如

// In some header file
#define HeapAlloc(a, b, c) MyHeapAlloc(a, b, c, __FILE__, __LINE__)
#define HeapFree(a, b, c) MyHeapFree(a, b, c, __FILE__, __LINE__)

... 

// In a separate source file:

#ifdef HeapAlloc
#undef HeapAlloc
#undef HeapFree
#endif

struct extra
{
    extra *next;
    size_t size;
    const char *file;
    int line;
};

extra* head;

LPVOID MyHeapAlloc(HANDLE heap, DWORD flags, SIZE_T size, const char *file, int line)
{
    LPVOID res = HeapAlloc(heap, flags, size + sizeof(extra));
    if (!res)
    {
        cout << "Allocation failed, called from " << file << ":" << line << endl;
        return 0;
    }

    extra *p = reinterpret_cast<extra*>(res);
    res = reinterpret_cast<void*>(&p[1]); 

    p->next = head;
    p->size = size;
    p->file = file;
    p->line = line;

    memset(res, 0xAA, size); 
    return res;
}

BOOL MyHeapFree(HANDLE heap, DWORD flags, LPVOID mem, const char *file, int line)
{
    extra *p = reinterpret_cast<extra*>(mem);
    p = reinterpret_cast<void*>(&p[-1]); 

    extra *q = head;
    extra *prev = 0;
    while(q)
    {
        if (reinterpret_cast<void*>(&q[1]) == mem)
        {
            break;
        }
        prev = q;
        q = next;
   }
   if (!q)
   {
       cout << "Attempt to free memory that wasn't allocated from " << file << ":" << line << endl;
       return false; 
   }

   /* Unlink q - if prev = NULL then it's the first one, so just move head */
   if (prev)
   {
      prev->next = q->next;
   }
   else
   {
      head = q->next;
   }

   memset(mem, 0xBB, q->size);
   return HeapFree(heap, flags, reinterpret_cast<void *>(q); 
}

我刚输入的所有的,所以可能会有轻微的错别字,但希望它显示了处理内存分配的方法。 对于某些情况下,可能必须垫的extra结构为16或32个字节的倍数,以确保下面的数据的对齐。 当然,你可以在任何时间,倾倒链表指向head ,看什么进行分配。



文章来源: Setting memory on a custom heap