mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-07-21 20:26:41 +00:00
parent
5f1ea8f832
commit
4f871447a3
2 changed files with 34 additions and 36 deletions
|
|
@ -51,67 +51,66 @@ Memory::Memory(llvm::IRBuilder<>& _builder, llvm::Module* _module)
|
||||||
m_size->setUnnamedAddr(true); // Address is not important
|
m_size->setUnnamedAddr(true); // Address is not important
|
||||||
|
|
||||||
m_resize = llvm::Function::Create(llvm::FunctionType::get(Type::BytePtr, Type::WordPtr, false), llvm::Function::ExternalLinkage, "mem_resize", _module);
|
m_resize = llvm::Function::Create(llvm::FunctionType::get(Type::BytePtr, Type::WordPtr, false), llvm::Function::ExternalLinkage, "mem_resize", _module);
|
||||||
m_storeWord = createStoreFunc(Type::i256, _module);
|
m_loadWord = createFunc(false, Type::i256, _module);
|
||||||
m_storeByte = createStoreFunc(Type::Byte, _module);
|
m_storeWord = createFunc(true, Type::i256, _module);
|
||||||
|
m_storeByte = createFunc(true, Type::Byte, _module);
|
||||||
}
|
}
|
||||||
|
|
||||||
llvm::Function* Memory::createStoreFunc(llvm::Type* _valueType, llvm::Module* _module)
|
llvm::Function* Memory::createFunc(bool _isStore, llvm::Type* _valueType, llvm::Module* _module)
|
||||||
{
|
{
|
||||||
auto wordValue = _valueType == Type::i256;
|
auto isWord = _valueType == Type::i256;
|
||||||
|
|
||||||
llvm::Type* storeArgs[] = {Type::i256, _valueType};
|
llvm::Type* storeArgs[] = {Type::i256, _valueType};
|
||||||
auto name = wordValue ? "store" : "store8";
|
auto name = _isStore ? isWord ? "mstore" : "mstore8" : "mload";
|
||||||
auto storeFunc = llvm::Function::Create(llvm::FunctionType::get(Type::Void, storeArgs, false), llvm::Function::PrivateLinkage, name, _module);
|
auto funcType = _isStore ? llvm::FunctionType::get(Type::Void, storeArgs, false) : llvm::FunctionType::get(Type::i256, Type::i256, false);
|
||||||
|
auto func = llvm::Function::Create(funcType, llvm::Function::PrivateLinkage, name, _module);
|
||||||
|
|
||||||
auto checkBB = llvm::BasicBlock::Create(storeFunc->getContext(), "check", storeFunc);
|
auto checkBB = llvm::BasicBlock::Create(func->getContext(), "check", func);
|
||||||
auto resizeBB = llvm::BasicBlock::Create(storeFunc->getContext(), "resize", storeFunc);
|
auto resizeBB = llvm::BasicBlock::Create(func->getContext(), "resize", func);
|
||||||
auto storeBB = llvm::BasicBlock::Create(storeFunc->getContext(), "store", storeFunc);
|
auto accessBB = llvm::BasicBlock::Create(func->getContext(), "access", func);
|
||||||
|
|
||||||
llvm::IRBuilder<> builder(checkBB);
|
llvm::IRBuilder<> builder(checkBB);
|
||||||
llvm::Value* index = storeFunc->arg_begin();
|
llvm::Value* index = func->arg_begin();
|
||||||
index->setName("index");
|
index->setName("index");
|
||||||
llvm::Value* value = ++storeFunc->arg_begin();
|
|
||||||
value->setName("value");
|
|
||||||
auto valueSize = _valueType->getPrimitiveSizeInBits() / 8;
|
auto valueSize = _valueType->getPrimitiveSizeInBits() / 8;
|
||||||
auto sizeRequired = builder.CreateAdd(index, builder.getIntN(256, valueSize), "sizeRequired");
|
auto sizeRequired = builder.CreateAdd(index, builder.getIntN(256, valueSize), "sizeRequired");
|
||||||
auto size = builder.CreateLoad(m_size, "size");
|
auto size = builder.CreateLoad(m_size, "size");
|
||||||
auto resizeNeeded = builder.CreateICmpULE(sizeRequired, size, "resizeNeeded");
|
auto resizeNeeded = builder.CreateICmpULE(sizeRequired, size, "resizeNeeded");
|
||||||
builder.CreateCondBr(resizeNeeded, resizeBB, storeBB); // OPT branch weights?
|
builder.CreateCondBr(resizeNeeded, resizeBB, accessBB); // OPT branch weights?
|
||||||
|
|
||||||
builder.SetInsertPoint(resizeBB);
|
builder.SetInsertPoint(resizeBB);
|
||||||
builder.CreateStore(sizeRequired, m_size);
|
builder.CreateStore(sizeRequired, m_size);
|
||||||
auto newData = builder.CreateCall(m_resize, m_size, "newData");
|
auto newData = builder.CreateCall(m_resize, m_size, "newData");
|
||||||
builder.CreateStore(newData, m_data);
|
builder.CreateStore(newData, m_data);
|
||||||
builder.CreateBr(storeBB);
|
builder.CreateBr(accessBB);
|
||||||
|
|
||||||
builder.SetInsertPoint(storeBB);
|
builder.SetInsertPoint(accessBB);
|
||||||
auto data = builder.CreateLoad(m_data, "data");
|
auto data = builder.CreateLoad(m_data, "data");
|
||||||
auto ptr = builder.CreateGEP(data, index, "ptr");
|
auto ptr = builder.CreateGEP(data, index, "ptr");
|
||||||
if (wordValue)
|
if (isWord)
|
||||||
ptr = builder.CreateBitCast(ptr, Type::WordPtr, "wordPtr");
|
ptr = builder.CreateBitCast(ptr, Type::WordPtr, "wordPtr");
|
||||||
|
if (_isStore)
|
||||||
|
{
|
||||||
|
llvm::Value* value = ++func->arg_begin();
|
||||||
|
value->setName("value");
|
||||||
builder.CreateStore(value, ptr);
|
builder.CreateStore(value, ptr);
|
||||||
builder.CreateRetVoid();
|
builder.CreateRetVoid();
|
||||||
|
}
|
||||||
return storeFunc;
|
else
|
||||||
|
{
|
||||||
|
auto ret = builder.CreateLoad(ptr);
|
||||||
|
builder.CreateRet(ret);
|
||||||
|
}
|
||||||
|
return func;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
llvm::Value* Memory::loadWord(llvm::Value* _addr)
|
llvm::Value* Memory::loadWord(llvm::Value* _addr)
|
||||||
{
|
{
|
||||||
// trunc _addr (an i256) to i64 index and use it to index the memory
|
auto value = m_builder.CreateCall(m_loadWord, _addr);
|
||||||
auto index = m_builder.CreateTrunc(_addr, m_builder.getInt64Ty(), "mem.index");
|
|
||||||
auto index31 = m_builder.CreateAdd(index, llvm::ConstantInt::get(m_builder.getInt64Ty(), 31), "mem.index.31");
|
|
||||||
|
|
||||||
// load from evmccrt_memory_require()[index]
|
|
||||||
auto base = m_builder.CreateCall(m_memRequire, index31, "base");
|
|
||||||
auto ptr = m_builder.CreateGEP(base, index, "ptr");
|
|
||||||
|
|
||||||
auto i256ptrTy = m_builder.getIntNTy(256)->getPointerTo();
|
|
||||||
auto wordPtr = m_builder.CreateBitCast(ptr, i256ptrTy, "wordptr");
|
|
||||||
auto byte = m_builder.CreateLoad(wordPtr, "word");
|
|
||||||
|
|
||||||
dump(0);
|
dump(0);
|
||||||
return byte;
|
return value;
|
||||||
}
|
}
|
||||||
|
|
||||||
void Memory::storeWord(llvm::Value* _addr, llvm::Value* _word)
|
void Memory::storeWord(llvm::Value* _addr, llvm::Value* _word)
|
||||||
|
|
@ -131,9 +130,7 @@ void Memory::storeByte(llvm::Value* _addr, llvm::Value* _word)
|
||||||
|
|
||||||
llvm::Value* Memory::getSize()
|
llvm::Value* Memory::getSize()
|
||||||
{
|
{
|
||||||
auto size = m_builder.CreateCall(m_memSize, "mem.size");
|
return m_builder.CreateLoad(m_size);
|
||||||
auto word = m_builder.CreateZExt(size, m_builder.getIntNTy(256), "mem.wsize");
|
|
||||||
return word;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void Memory::dump(uint64_t _begin, uint64_t _end)
|
void Memory::dump(uint64_t _begin, uint64_t _end)
|
||||||
|
|
|
||||||
|
|
@ -22,7 +22,7 @@ public:
|
||||||
void dump(uint64_t _begin, uint64_t _end = 0);
|
void dump(uint64_t _begin, uint64_t _end = 0);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
llvm::Function* createStoreFunc(llvm::Type* _type, llvm::Module* _module);
|
llvm::Function* createFunc(bool _isStore, llvm::Type* _type, llvm::Module* _module);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
llvm::IRBuilder<>& m_builder;
|
llvm::IRBuilder<>& m_builder;
|
||||||
|
|
@ -30,6 +30,7 @@ private:
|
||||||
llvm::GlobalVariable* m_data;
|
llvm::GlobalVariable* m_data;
|
||||||
llvm::GlobalVariable* m_size;
|
llvm::GlobalVariable* m_size;
|
||||||
|
|
||||||
|
llvm::Function* m_loadWord;
|
||||||
llvm::Function* m_storeWord;
|
llvm::Function* m_storeWord;
|
||||||
llvm::Function* m_storeByte;
|
llvm::Function* m_storeByte;
|
||||||
llvm::Function* m_resize;
|
llvm::Function* m_resize;
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue