79 lines
2.1 KiB
C++
79 lines
2.1 KiB
C++
// This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
|
|
#pragma once
|
|
|
|
#include "Luau/RegisterX64.h"
|
|
#include "UnwindBuilder.h"
|
|
|
|
#include <vector>
|
|
|
|
namespace Luau
|
|
{
|
|
namespace CodeGen
|
|
{
|
|
|
|
// This struct matches the layout of x64 RUNTIME_FUNCTION from winnt.h
|
|
struct UnwindFunctionWin
|
|
{
|
|
uint32_t beginOffset;
|
|
uint32_t endOffset;
|
|
uint32_t unwindInfoOffset;
|
|
};
|
|
|
|
// This struct matches the layout of x64 UNWIND_INFO from ehdata.h
|
|
struct UnwindInfoWin
|
|
{
|
|
uint8_t version : 3;
|
|
uint8_t flags : 5;
|
|
uint8_t prologsize;
|
|
uint8_t unwindcodecount;
|
|
uint8_t framereg : 4;
|
|
uint8_t frameregoff : 4;
|
|
};
|
|
|
|
// This struct matches the layout of UNWIND_CODE from ehdata.h
|
|
struct UnwindCodeWin
|
|
{
|
|
uint8_t offset;
|
|
uint8_t opcode : 4;
|
|
uint8_t opinfo : 4;
|
|
};
|
|
|
|
class UnwindBuilderWin : public UnwindBuilder
|
|
{
|
|
public:
|
|
void setBeginOffset(size_t beginOffset) override;
|
|
size_t getBeginOffset() const override;
|
|
|
|
void startInfo(Arch arch) override;
|
|
void startFunction() override;
|
|
void finishFunction(uint32_t beginOffset, uint32_t endOffset) override;
|
|
void finishInfo() override;
|
|
|
|
void prologueA64(uint32_t prologueSize, uint32_t stackSize, std::initializer_list<A64::RegisterA64> regs) override;
|
|
void prologueX64(uint32_t prologueSize, uint32_t stackSize, bool setupFrame, std::initializer_list<X64::RegisterX64> regs) override;
|
|
|
|
size_t getSize() const override;
|
|
size_t getFunctionCount() const override;
|
|
|
|
void finalize(char* target, size_t offset, void* funcAddress, size_t funcSize) const override;
|
|
|
|
private:
|
|
size_t beginOffset = 0;
|
|
|
|
static const unsigned kRawDataLimit = 1024;
|
|
uint8_t rawData[kRawDataLimit];
|
|
uint8_t* rawDataPos = rawData;
|
|
|
|
std::vector<UnwindFunctionWin> unwindFunctions;
|
|
|
|
// Windows unwind codes are written in reverse, so we have to collect them all first
|
|
std::vector<UnwindCodeWin> unwindCodes;
|
|
|
|
uint8_t prologSize = 0;
|
|
X64::RegisterX64 frameReg = X64::noreg;
|
|
uint8_t frameRegOffset = 0;
|
|
};
|
|
|
|
} // namespace CodeGen
|
|
} // namespace Luau
|