2011-07-12 21:27:37 +02:00
|
|
|
#ifndef __ALLOCATOR_H__
|
|
|
|
#define __ALLOCATOR_H__
|
|
|
|
|
2011-07-12 23:59:45 +02:00
|
|
|
#ifdef _DEBUG_MEMORY
|
2011-07-12 21:27:37 +02:00
|
|
|
|
2011-07-12 23:59:45 +02:00
|
|
|
#include <unordered_set>
|
2011-07-12 21:27:37 +02:00
|
|
|
|
2011-07-12 23:59:45 +02:00
|
|
|
#ifdef _REENTRANT
|
|
|
|
#include <boost/thread.hpp>
|
|
|
|
#endif
|
2011-07-12 21:27:37 +02:00
|
|
|
|
2011-07-12 23:59:45 +02:00
|
|
|
struct AllocationBlock
|
|
|
|
{
|
|
|
|
unsigned int bytes;
|
2011-07-12 21:27:37 +02:00
|
|
|
void** backtraceBuffer;
|
2011-07-12 23:59:45 +02:00
|
|
|
unsigned char backtraceSize;
|
|
|
|
unsigned int records;
|
|
|
|
};
|
2011-07-12 21:27:37 +02:00
|
|
|
|
2011-07-12 23:59:45 +02:00
|
|
|
struct block_hash : std::unary_function<AllocationBlock *, std::size_t> {
|
|
|
|
std::size_t operator()(const AllocationBlock *block) const {
|
|
|
|
struct HashKey {
|
|
|
|
unsigned int bytes;
|
|
|
|
void *backtraceTop[3];
|
|
|
|
unsigned char backtraceSize;
|
|
|
|
} hashKey;
|
|
|
|
|
|
|
|
hashKey.bytes = block->bytes;
|
|
|
|
for(int i=0;i<3;++i) {
|
|
|
|
if(i < block->backtraceSize)
|
|
|
|
hashKey.backtraceTop[i] = block->backtraceBuffer[i];
|
|
|
|
else
|
|
|
|
hashKey.backtraceTop[i] = NULL;
|
|
|
|
}
|
|
|
|
hashKey.backtraceSize = block->backtraceSize;
|
|
|
|
|
|
|
|
//std::hash<HashKey> hasher;
|
|
|
|
return 1;//hasher(hashKey);
|
|
|
|
}
|
2011-07-12 21:27:37 +02:00
|
|
|
};
|
|
|
|
|
2011-07-12 23:59:45 +02:00
|
|
|
struct block_equal_to : std::binary_function<AllocationBlock *, AllocationBlock *, bool> {
|
|
|
|
bool operator()(const AllocationBlock* a, const AllocationBlock *b) const
|
|
|
|
{
|
|
|
|
if(a->bytes != b->bytes || a->backtraceSize != b->backtraceSize)
|
|
|
|
return false;
|
|
|
|
for(int i=0;i<a->backtraceSize;++i)
|
|
|
|
if(a->backtraceBuffer[i] != b->backtraceBuffer[i])
|
|
|
|
return false;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
//TODO: use mem tags
|
2011-07-12 21:27:37 +02:00
|
|
|
|
|
|
|
class Allocator
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
Allocator();
|
|
|
|
~Allocator();
|
|
|
|
|
|
|
|
void *allocate(size_t bytes);
|
2011-07-12 23:59:45 +02:00
|
|
|
void deallocate(void *p);
|
2011-07-12 21:27:37 +02:00
|
|
|
|
|
|
|
void dumpLeaks();
|
|
|
|
|
|
|
|
private:
|
2011-07-12 23:59:45 +02:00
|
|
|
AllocationBlock* findBlock(void** backtraceBuffer, int backtraceSize, unsigned int bytes);
|
2011-07-12 21:27:37 +02:00
|
|
|
|
2011-07-12 23:59:45 +02:00
|
|
|
typedef std::unordered_set<AllocationBlock*, block_hash, block_equal_to> AllocationBlocksList;
|
2011-07-12 21:27:37 +02:00
|
|
|
AllocationBlocksList m_allocationsBlocks;
|
|
|
|
|
2011-07-12 23:59:45 +02:00
|
|
|
#ifdef _REENTRANT
|
2011-07-12 21:27:37 +02:00
|
|
|
boost::recursive_mutex m_allocatorLock;
|
2011-07-12 23:59:45 +02:00
|
|
|
#endif
|
2011-07-12 21:27:37 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
void* operator new(size_t bytes, int dummy);
|
|
|
|
void* operator new(size_t bytes);
|
|
|
|
void* operator new[](size_t bytes);
|
|
|
|
void operator delete(void *p);
|
2011-07-12 23:59:45 +02:00
|
|
|
void operator delete[](void *p);
|
|
|
|
|
|
|
|
#endif // _DEBUG_MEMORY
|
2011-07-12 21:27:37 +02:00
|
|
|
|
|
|
|
#endif
|
|
|
|
|