| hacker013 wrote: |
| Could you post it here? |
| Code: |
|
#ifdef _DEBUG #define GUARD_VALUE 0xDEEDBEEF #define NEXT4ALIGN(x) ( (x + 3) & ~0x3) static u32 total = 0; static u32 objects = 0; static u32 oid = 0; void *operator new(size_t sz) { u8 *mem = reinterpret_cast<u8*>(malloc(sz+12+3)); if (mem) { reinterpret_cast<u32*>(mem)[0] = sz; // insert size before data area reinterpret_cast<u32*>(mem)[1] = ++oid; // insert allocation id before data area mem += 8; *reinterpret_cast<u32*>(mem+NEXT4ALIGN(sz)) = GUARD_VALUE; // insert guard value at end of memory total += sz; // increment total size of allocations ++objects; PrintStatusMessage("Alloc: %p,%u,%u,%u,%u,%p\n",mem,oid,sz,objects,total,__builtin_return_address(0)); } else { PrintStatusMessage("Failed allocate %u\n",sz); } return mem; } void operator delete(void *p) { u32 sz = reinterpret_cast<u32*>(p)[-2]; // retrieve size from memory u32 id = reinterpret_cast<u32*>(p)[-1]; // retrieve id if (*reinterpret_cast<u32*>(reinterpret_cast<u8*>(p)+NEXT4ALIGN(sz)) != GUARD_VALUE) // check against guard value { PrintStatusMessage("MALLOC OVERFLOW!\n"); } total -= sz; // update counter --objects; PrintStatusMessage("Free: %p,%u,%u,%u,%u\n",p,id,sz,objects,total); free(reinterpret_cast<u8*>(p)-8); } void PrintMemoryStatistics() { PrintStatusMessage("Objects: %u, sz: %u\n",objects,total); } #endif |