Paweł Bylica
10 years ago
11 changed files with 265 additions and 208 deletions
@ -0,0 +1,51 @@ |
|||||
|
|
||||
|
#pragma once |
||||
|
|
||||
|
#include <csetjmp> |
||||
|
|
||||
|
#include "Utils.h" |
||||
|
|
||||
|
|
||||
|
namespace dev |
||||
|
{ |
||||
|
namespace eth |
||||
|
{ |
||||
|
namespace jit |
||||
|
{ |
||||
|
|
||||
|
using jmpBufRef = decltype(&jmp_buf{}[0]); |
||||
|
|
||||
|
struct RuntimeData |
||||
|
{ |
||||
|
enum Index |
||||
|
{ |
||||
|
Gas, |
||||
|
Address, |
||||
|
Caller, |
||||
|
Origin, |
||||
|
CallValue, |
||||
|
CallDataSize, |
||||
|
GasPrice, |
||||
|
PrevHash, |
||||
|
CoinBase, |
||||
|
TimeStamp, |
||||
|
Number, |
||||
|
Difficulty, |
||||
|
GasLimit, |
||||
|
CodeSize, |
||||
|
|
||||
|
_size, |
||||
|
|
||||
|
ReturnDataOffset = CallValue, // Reuse 2 fields for return data reference
|
||||
|
ReturnDataSize = CallDataSize |
||||
|
}; |
||||
|
|
||||
|
i256 elems[_size]; |
||||
|
byte const* callData; |
||||
|
byte const* code; |
||||
|
jmpBufRef jmpBuf; |
||||
|
}; |
||||
|
|
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,159 @@ |
|||||
|
|
||||
|
#include "RuntimeManager.h" |
||||
|
|
||||
|
#include <llvm/IR/GlobalVariable.h> |
||||
|
#include <llvm/IR/Function.h> |
||||
|
#include <llvm/IR/IntrinsicInst.h> |
||||
|
|
||||
|
#include "RuntimeData.h" |
||||
|
#include "Instruction.h" |
||||
|
|
||||
|
namespace dev |
||||
|
{ |
||||
|
namespace eth |
||||
|
{ |
||||
|
namespace jit |
||||
|
{ |
||||
|
|
||||
|
llvm::StructType* RuntimeManager::getRuntimeDataType() |
||||
|
{ |
||||
|
static llvm::StructType* type = nullptr; |
||||
|
if (!type) |
||||
|
{ |
||||
|
llvm::Type* elems[] = |
||||
|
{ |
||||
|
llvm::ArrayType::get(Type::Word, RuntimeData::_size), |
||||
|
Type::BytePtr, |
||||
|
Type::BytePtr, |
||||
|
Type::BytePtr |
||||
|
}; |
||||
|
type = llvm::StructType::create(elems, "Runtime"); |
||||
|
} |
||||
|
return type; |
||||
|
} |
||||
|
|
||||
|
namespace |
||||
|
{ |
||||
|
llvm::Twine getName(RuntimeData::Index _index) |
||||
|
{ |
||||
|
switch (_index) |
||||
|
{ |
||||
|
default: return "data"; |
||||
|
case RuntimeData::Gas: return "gas"; |
||||
|
case RuntimeData::Address: return "address"; |
||||
|
case RuntimeData::Caller: return "caller"; |
||||
|
case RuntimeData::Origin: return "origin"; |
||||
|
case RuntimeData::CallValue: return "callvalue"; |
||||
|
case RuntimeData::CallDataSize: return "calldatasize"; |
||||
|
case RuntimeData::GasPrice: return "gasprice"; |
||||
|
case RuntimeData::PrevHash: return "prevhash"; |
||||
|
case RuntimeData::CoinBase: return "coinbase"; |
||||
|
case RuntimeData::TimeStamp: return "timestamp"; |
||||
|
case RuntimeData::Number: return "number"; |
||||
|
case RuntimeData::Difficulty: return "difficulty"; |
||||
|
case RuntimeData::GasLimit: return "gaslimit"; |
||||
|
case RuntimeData::CodeSize: return "codesize"; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
RuntimeManager::RuntimeManager(llvm::IRBuilder<>& _builder): CompilerHelper(_builder) |
||||
|
{ |
||||
|
m_dataPtr = new llvm::GlobalVariable(*getModule(), Type::RuntimePtr, false, llvm::GlobalVariable::PrivateLinkage, llvm::UndefValue::get(Type::RuntimePtr), "rt"); |
||||
|
m_longjmp = llvm::Intrinsic::getDeclaration(getModule(), llvm::Intrinsic::longjmp); |
||||
|
|
||||
|
// Export data
|
||||
|
auto mainFunc = getMainFunction(); |
||||
|
llvm::Value* dataPtr = &mainFunc->getArgumentList().back(); |
||||
|
m_builder.CreateStore(dataPtr, m_dataPtr); |
||||
|
} |
||||
|
|
||||
|
llvm::Value* RuntimeManager::getRuntimePtr() |
||||
|
{ |
||||
|
if (auto mainFunc = getMainFunction()) |
||||
|
return mainFunc->arg_begin()->getNextNode(); // Runtime is the second parameter of main function
|
||||
|
return m_builder.CreateLoad(m_dataPtr, "rt"); |
||||
|
} |
||||
|
|
||||
|
llvm::Value* RuntimeManager::getPtr(RuntimeData::Index _index) |
||||
|
{ |
||||
|
llvm::Value* idxList[] = {m_builder.getInt32(0), m_builder.getInt32(0), m_builder.getInt32(_index)}; |
||||
|
return m_builder.CreateInBoundsGEP(getRuntimePtr(), idxList, getName(_index) + "Ptr"); |
||||
|
} |
||||
|
|
||||
|
llvm::Value* RuntimeManager::get(RuntimeData::Index _index) |
||||
|
{ |
||||
|
return m_builder.CreateLoad(getPtr(_index), getName(_index)); |
||||
|
} |
||||
|
|
||||
|
void RuntimeManager::set(RuntimeData::Index _index, llvm::Value* _value) |
||||
|
{ |
||||
|
m_builder.CreateStore(_value, getPtr(_index)); |
||||
|
} |
||||
|
|
||||
|
void RuntimeManager::registerReturnData(llvm::Value* _offset, llvm::Value* _size) |
||||
|
{ |
||||
|
set(RuntimeData::ReturnDataOffset, _offset); |
||||
|
set(RuntimeData::ReturnDataSize, _size); |
||||
|
} |
||||
|
|
||||
|
void RuntimeManager::raiseException(ReturnCode _returnCode) |
||||
|
{ |
||||
|
m_builder.CreateCall2(m_longjmp, getJmpBuf(), Constant::get(_returnCode)); |
||||
|
} |
||||
|
|
||||
|
llvm::Value* RuntimeManager::get(Instruction _inst) |
||||
|
{ |
||||
|
switch (_inst) |
||||
|
{ |
||||
|
default: assert(false); return nullptr; |
||||
|
case Instruction::GAS: return get(RuntimeData::Gas); |
||||
|
case Instruction::ADDRESS: return get(RuntimeData::Address); |
||||
|
case Instruction::CALLER: return get(RuntimeData::Caller); |
||||
|
case Instruction::ORIGIN: return get(RuntimeData::Origin); |
||||
|
case Instruction::CALLVALUE: return get(RuntimeData::CallValue); |
||||
|
case Instruction::CALLDATASIZE: return get(RuntimeData::CallDataSize); |
||||
|
case Instruction::GASPRICE: return get(RuntimeData::GasPrice); |
||||
|
case Instruction::PREVHASH: return get(RuntimeData::PrevHash); |
||||
|
case Instruction::COINBASE: return get(RuntimeData::CoinBase); |
||||
|
case Instruction::TIMESTAMP: return get(RuntimeData::TimeStamp); |
||||
|
case Instruction::NUMBER: return get(RuntimeData::Number); |
||||
|
case Instruction::DIFFICULTY: return get(RuntimeData::Difficulty); |
||||
|
case Instruction::GASLIMIT: return get(RuntimeData::GasLimit); |
||||
|
case Instruction::CODESIZE: return get(RuntimeData::CodeSize); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
llvm::Value* RuntimeManager::getCallData() |
||||
|
{ |
||||
|
auto ptr = getBuilder().CreateStructGEP(getRuntimePtr(), 1, "calldataPtr"); |
||||
|
return getBuilder().CreateLoad(ptr, "calldata"); |
||||
|
} |
||||
|
|
||||
|
llvm::Value* RuntimeManager::getCode() |
||||
|
{ |
||||
|
auto ptr = getBuilder().CreateStructGEP(getRuntimePtr(), 2, "codePtr"); |
||||
|
return getBuilder().CreateLoad(ptr, "code"); |
||||
|
} |
||||
|
|
||||
|
llvm::Value* RuntimeManager::getJmpBuf() |
||||
|
{ |
||||
|
auto ptr = getBuilder().CreateStructGEP(getRuntimePtr(), 3, "jmpbufPtr"); |
||||
|
return getBuilder().CreateLoad(ptr, "jmpbuf"); |
||||
|
} |
||||
|
|
||||
|
llvm::Value* RuntimeManager::getGas() |
||||
|
{ |
||||
|
return get(RuntimeData::Gas); |
||||
|
} |
||||
|
|
||||
|
void RuntimeManager::setGas(llvm::Value* _gas) |
||||
|
{ |
||||
|
llvm::Value* idxList[] = {m_builder.getInt32(0), m_builder.getInt32(0), m_builder.getInt32(RuntimeData::Gas)}; |
||||
|
auto ptr = m_builder.CreateInBoundsGEP(getRuntimePtr(), idxList, "gasPtr"); |
||||
|
m_builder.CreateStore(_gas, ptr); |
||||
|
} |
||||
|
|
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,46 @@ |
|||||
|
#pragma once |
||||
|
|
||||
|
#include "CompilerHelper.h" |
||||
|
#include "Type.h" |
||||
|
#include "RuntimeData.h" |
||||
|
#include "Instruction.h" |
||||
|
|
||||
|
namespace dev |
||||
|
{ |
||||
|
namespace eth |
||||
|
{ |
||||
|
namespace jit |
||||
|
{ |
||||
|
|
||||
|
class RuntimeManager: public CompilerHelper |
||||
|
{ |
||||
|
public: |
||||
|
RuntimeManager(llvm::IRBuilder<>& _builder); |
||||
|
|
||||
|
llvm::Value* getRuntimePtr(); |
||||
|
|
||||
|
llvm::Value* get(RuntimeData::Index _index); |
||||
|
llvm::Value* get(Instruction _inst); |
||||
|
llvm::Value* getGas(); // TODO: Remove
|
||||
|
llvm::Value* getCallData(); |
||||
|
llvm::Value* getCode(); |
||||
|
void setGas(llvm::Value* _gas); |
||||
|
|
||||
|
void registerReturnData(llvm::Value* _index, llvm::Value* _size); |
||||
|
|
||||
|
void raiseException(ReturnCode _returnCode); |
||||
|
|
||||
|
static llvm::StructType* getRuntimeDataType(); |
||||
|
|
||||
|
private: |
||||
|
llvm::Value* getPtr(RuntimeData::Index _index); |
||||
|
void set(RuntimeData::Index _index, llvm::Value* _value); |
||||
|
llvm::Value* getJmpBuf(); |
||||
|
|
||||
|
llvm::GlobalVariable* m_dataPtr = nullptr; |
||||
|
llvm::Function* m_longjmp = nullptr; |
||||
|
}; |
||||
|
|
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue