00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032 #ifndef __EMULATORUTILITY_HEAPALLOCATOR_H__
00033 #define __EMULATORUTILITY_HEAPALLOCATOR_H__
00034
00035 namespace Onikiri {
00036 namespace EmulatorUtility {
00037
00038
00039
00040
00041
00042 class HeapAllocator
00043 {
00044 private:
00045 HeapAllocator() {}
00046 public:
00047
00048 explicit HeapAllocator(u64 pageSize);
00049
00050
00051 bool AddMemoryBlock(u64 start, u64 length);
00052
00053
00054
00055
00056
00057
00058 u64 Alloc(u64 addr, u64 length);
00059
00060
00061
00062
00063
00064 u64 ReAlloc(u64 old_addr, u64 old_size, u64 new_size);
00065
00066
00067
00068 bool Free(u64 addr);
00069
00070 bool Free(u64 addr, u64 size);
00071
00072
00073 u64 GetBlockSize(u64 addr) const;
00074
00075
00076 u64 IsIntersected(u64 addr, u64 length) const;
00077
00078 u64 GetPageSize() const { return m_pageSize; }
00079
00080 private:
00081 struct MemoryBlock
00082 {
00083 u64 Addr;
00084 u64 Bytes;
00085
00086 explicit MemoryBlock(u64 addr_arg = 0, u64 bytes_arg = 0) : Addr(addr_arg), Bytes(bytes_arg) { }
00087 bool operator<(const MemoryBlock& mb) const{
00088 return Addr < mb.Addr;
00089 }
00090 bool operator==(const MemoryBlock& mb) const {
00091 return Addr == mb.Addr;
00092 }
00093 bool Intersects(const MemoryBlock& mb) const {
00094 return (Addr <= mb.Addr && mb.Addr < Addr+Bytes)
00095 || (mb.Addr <= Addr && Addr < mb.Addr+mb.Bytes);
00096 }
00097 bool Contains(const MemoryBlock& mb) const {
00098 return (Addr <= mb.Addr && mb.Addr+mb.Bytes < Addr+Bytes);
00099 }
00100 };
00101 typedef std::list<MemoryBlock> BlockList;
00102
00103
00104 void IntegrateFreeBlocks();
00105 BlockList::iterator FindMemoryBlock(BlockList& blockList, u64 addr);
00106 BlockList::const_iterator FindMemoryBlock(const BlockList& blockList, u64 addr) const;
00107
00108
00109 u64 in_pages(u64 bytes) const {
00110 return (bytes + m_pageSize - 1)/m_pageSize;
00111 }
00112
00113
00114
00115 BlockList m_freeList;
00116 BlockList m_allocList;
00117 u64 m_pageSize;
00118 };
00119
00120 }
00121 }
00122
00123 #endif