1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
#ifndef UNITY_DEFAULT_ALLOCATOR_H_
#define UNITY_DEFAULT_ALLOCATOR_H_
#if ENABLE_MEMORY_MANAGER
#include "BaseAllocator.h"
#include "Runtime/Threads/Mutex.h"
#include "Runtime/Allocator/LowLevelDefaultAllocator.h"
#include "Runtime/Utilities/BitUtility.h"
enum RequestType
{
kRegister,
kUnregister,
kTest
};
template<class LLAlloctor>
class UnityDefaultAllocator : public BaseAllocator
{
public:
UnityDefaultAllocator(const char* name);
virtual void* Allocate (size_t size, int align);
virtual void* Reallocate (void* p, size_t size, int align);
virtual void Deallocate (void* p);
virtual bool Contains (const void* p);
virtual size_t GetPtrSize(const void* ptr) const;
virtual ProfilerAllocationHeader* GetProfilerHeader(const void* ptr) const;
static int GetOverheadSize(void* ptr);
private:
// needs 30 bit (4byte aligned allocs and packed as bitarray) ( 1 byte -> 32Bytes, 4bytes rep 128Bytes(7bit))
enum
{
kTargetBitsRepresentedPerBit = StaticLog2<kDefaultMemoryAlignment>::value,
kTargetBitsRepresentedPerByte = 5 + kTargetBitsRepresentedPerBit,
kPage1Bits = 7, // 128*4Bytes = 512Bytes (page: 4GB/128 = 32MB per pointer)
kPage2Bits = 7, // 128*4Bytes = 512Bytes (page: 32MB/128 = 256KB per pointer)
kPage3Bits = 5, // 32*4Bytes = 128Bytes (page: 256K/32 = 8K per pointer)
kPage4Bits = 32-kPage1Bits-kPage2Bits-kPage3Bits-kTargetBitsRepresentedPerByte
};
struct PageAllocationElement
{
PageAllocationElement() : m_HighBits(0), m_PageAllocations(NULL){}
UInt32 m_HighBits;
int**** m_PageAllocations;
};
static const int kNumPageAllocationBlocks = 5; //each block represents 4gb of memory;
PageAllocationElement m_PageAllocationList[kNumPageAllocationBlocks];
template<RequestType requestType>
bool AllocationPage(const void* p);
Mutex m_AllocLock;
void RegisterAllocation(const void* p);
void RegisterDeallocation(const void* p);
void* AddHeaderAndFooter( void* ptr, size_t size, int align ) const;
};
#endif
#endif
|
