Add support for a CodeBlock holding a child.

This is required to make sure two code spaces are relatively close to one another.
In this case I need the AArch64 JIT codespace and its farcode space to be within 128MB of one another for branches.
This commit is contained in:
Ryan Houdek 2015-08-07 22:08:57 -05:00
parent 90e05f7bea
commit d5c99a5b48
1 changed files with 30 additions and 3 deletions

View File

@ -22,9 +22,19 @@ private:
protected: protected:
u8 *region; u8 *region;
size_t region_size; size_t region_size;
size_t parent_region_size;
bool m_has_child;
bool m_is_child;
CodeBlock* m_child;
public: public:
CodeBlock() : region(nullptr), region_size(0) {} CodeBlock()
: region(nullptr), region_size(0), parent_region_size(0),
m_has_child(false), m_is_child(false), m_child(nullptr)
{
}
virtual ~CodeBlock() { if (region) FreeCodeSpace(); } virtual ~CodeBlock() { if (region) FreeCodeSpace(); }
// Call this before you generate any code. // Call this before you generate any code.
@ -49,6 +59,12 @@ public:
FreeMemoryPages(region, region_size); FreeMemoryPages(region, region_size);
region = nullptr; region = nullptr;
region_size = 0; region_size = 0;
parent_region_size = 0;
if (m_has_child)
{
m_child->region = nullptr;
m_child->region_size = 0;
}
} }
bool IsInSpace(u8 *ptr) const bool IsInSpace(u8 *ptr) const
@ -70,7 +86,7 @@ public:
size_t GetSpaceLeft() const size_t GetSpaceLeft() const
{ {
return region_size - (T::GetCodePtr() - region); return (m_has_child ? parent_region_size : region_size) - (T::GetCodePtr() - region);
} }
bool IsAlmostFull() const bool IsAlmostFull() const
@ -78,5 +94,16 @@ public:
// This should be bigger than the biggest block ever. // This should be bigger than the biggest block ever.
return GetSpaceLeft() < 0x10000; return GetSpaceLeft() < 0x10000;
} }
void AddChildCodeSpace(CodeBlock* child, size_t size)
{
_assert_msg_(DYNA_REC, !m_has_child, "Already have a child! Can't have another!");
m_child = child;
m_has_child = true;
m_child->m_is_child = true;
u8* child_region = region + region_size - size;
m_child->region = child_region;
m_child->region_size = size;
m_child->ResetCodePtr();
parent_region_size = region_size - size;
}
}; };