1 | // -*- C++ -*- |
---|
2 | |
---|
3 | #ifndef _ADDHEADERHEAP_H_ |
---|
4 | #define _ADDHEADERHEAP_H_ |
---|
5 | |
---|
6 | #include "sassert.h" |
---|
7 | #include "hldefines.h" |
---|
8 | |
---|
9 | namespace Hoard { |
---|
10 | |
---|
11 | /** |
---|
12 | * @class AddHeaderHeap |
---|
13 | */ |
---|
14 | |
---|
15 | template <class SuperblockType, |
---|
16 | size_t SuperblockSize, |
---|
17 | class SuperHeap> |
---|
18 | class AddHeaderHeap { |
---|
19 | private: |
---|
20 | |
---|
21 | HL::sassert<(((int) SuperHeap::Alignment) % SuperblockSize == 0)> verifySize1; |
---|
22 | HL::sassert<(((int) SuperHeap::Alignment) >= SuperblockSize)> verifySize2; |
---|
23 | |
---|
24 | SuperHeap theHeap; |
---|
25 | |
---|
26 | public: |
---|
27 | |
---|
28 | enum { Alignment = 0 }; |
---|
29 | |
---|
30 | MALLOC_FUNCTION INLINE void * malloc (size_t sz) { |
---|
31 | // Allocate extra space for the header, |
---|
32 | // put it at the front of the object, |
---|
33 | // and return a pointer to just past it. |
---|
34 | const size_t headerSize = sizeof(typename SuperblockType::Header); |
---|
35 | void * ptr = theHeap.malloc (sz + headerSize); |
---|
36 | if (ptr == NULL) { |
---|
37 | return NULL; |
---|
38 | } |
---|
39 | typename SuperblockType::Header * p |
---|
40 | = new (ptr) typename SuperblockType::Header (sz, sz); |
---|
41 | assert ((size_t) (p + 1) == (size_t) ptr + headerSize); |
---|
42 | return reinterpret_cast<void *>(p + 1); |
---|
43 | } |
---|
44 | |
---|
45 | INLINE static size_t getSize (void * ptr) { |
---|
46 | // Find the header (just before the pointer) and return the size |
---|
47 | // value stored there. |
---|
48 | typename SuperblockType::Header * p; |
---|
49 | p = reinterpret_cast<typename SuperblockType::Header *>(ptr); |
---|
50 | return (p - 1)->getSize (ptr); |
---|
51 | } |
---|
52 | |
---|
53 | INLINE void free (void * ptr) { |
---|
54 | // Find the header (just before the pointer) and free the whole object. |
---|
55 | typename SuperblockType::Header * p; |
---|
56 | p = reinterpret_cast<typename SuperblockType::Header *>(ptr); |
---|
57 | theHeap.free (reinterpret_cast<void *>(p - 1)); |
---|
58 | } |
---|
59 | }; |
---|
60 | |
---|
61 | }; |
---|
62 | |
---|
63 | #endif |
---|