From 3cba47385adae87b7c095c15c070948e8a7a753f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Bylica?= <pawel.bylica@imapp.pl>
Date: Mon, 27 Oct 2014 16:13:38 +0100
Subject: [PATCH] Moving CODE data from Ext to Runtime [#81470252]

---
 libevmjit/Compiler.cpp |  2 +-
 libevmjit/Ext.cpp      | 19 -------------------
 libevmjit/Ext.h        |  5 -----
 libevmjit/Runtime.cpp  |  8 ++++++++
 libevmjit/Runtime.h    |  2 ++
 5 files changed, 11 insertions(+), 25 deletions(-)

diff --git a/libevmjit/Compiler.cpp b/libevmjit/Compiler.cpp
index a872a0097..600551c29 100644
--- a/libevmjit/Compiler.cpp
+++ b/libevmjit/Compiler.cpp
@@ -733,7 +733,7 @@ void Compiler::compileBasicBlock(BasicBlock& basicBlock, bytesConstRef bytecode,
 			auto srcIdx = stack.pop();
 			auto reqBytes = stack.pop();
 
-			auto srcPtr = ext.code();	// TODO: Code & its size are constants, feature #80814234
+			auto srcPtr = _runtimeManager.getCode();	// TODO: Code & its size are constants, feature #80814234
 			auto srcSize = _runtimeManager.get(RuntimeData::CodeSize);
 
 			memory.copyBytes(srcPtr, srcSize, srcIdx, destMemIdx, reqBytes);
diff --git a/libevmjit/Ext.cpp b/libevmjit/Ext.cpp
index b75cd3195..6659425ea 100644
--- a/libevmjit/Ext.cpp
+++ b/libevmjit/Ext.cpp
@@ -51,17 +51,8 @@ Ext::Ext(RuntimeManager& _runtimeManager):
 	m_arg7 = m_builder.CreateAlloca(i256Ty, nullptr, "ext.arg7");
 	m_arg8 = m_builder.CreateAlloca(i256Ty, nullptr, "ext.arg8");
 
-	Type* elements[] = {
-		i8PtrTy, // byte* code
-
-	};
-	auto extDataTy = StructType::create(elements, "ext.Data");
-
-	m_data = m_builder.CreateAlloca(extDataTy, nullptr, "ext.data");
-
 	using llvm::types::i;
 	using Linkage = llvm::GlobalValue::LinkageTypes;
-	m_init = Function::Create(FunctionType::get(m_builder.getVoidTy(), extDataTy->getPointerTo(), false), Linkage::ExternalLinkage, "ext_init", module);
 	m_store = Function::Create(TypeBuilder<void(i<256>*, i<256>*), true>::get(ctx), Linkage::ExternalLinkage, "ext_store", module);
 	m_setStore = Function::Create(TypeBuilder<void(i<256>*, i<256>*), true>::get(ctx), Linkage::ExternalLinkage, "ext_setStore", module);
 	m_calldataload = Function::Create(TypeBuilder<void(i<256>*, i<256>*), true>::get(ctx), Linkage::ExternalLinkage, "ext_calldataload", module);
@@ -75,8 +66,6 @@ Ext::Ext(RuntimeManager& _runtimeManager):
 	m_exp = Function::Create(TypeBuilder<void(i<256>*, i<256>*, i<256>*), true>::get(ctx), Linkage::ExternalLinkage, "ext_exp", module);
 	m_codeAt = Function::Create(TypeBuilder<i<8>*(i<256>*), true>::get(ctx), Linkage::ExternalLinkage, "ext_codeAt", module);
 	m_codesizeAt = Function::Create(TypeBuilder<void(i<256>*, i<256>*), true>::get(ctx), Linkage::ExternalLinkage, "ext_codesizeAt", module);
-
-	m_builder.CreateCall(m_init, m_data);
 }
 
 llvm::Value* Ext::store(llvm::Value* _index)
@@ -93,14 +82,6 @@ void Ext::setStore(llvm::Value* _index, llvm::Value* _value)
 	m_builder.CreateCall(m_setStore, m_args); // Uses native endianness
 }
 
-Value* Ext::getDataElem(unsigned _index, const Twine& _name)
-{
-	auto valuePtr = m_builder.CreateStructGEP(m_data, _index, _name);
-	return m_builder.CreateLoad(valuePtr);
-}
-
-Value* Ext::code() { return getDataElem(0, "code"); }
-
 Value* Ext::calldataload(Value* _index)
 {
 	m_builder.CreateStore(_index, m_args[0]);
diff --git a/libevmjit/Ext.h b/libevmjit/Ext.h
index f798ec549..e7200e95a 100644
--- a/libevmjit/Ext.h
+++ b/libevmjit/Ext.h
@@ -20,8 +20,6 @@ public:
 	llvm::Value* store(llvm::Value* _index);
 	void setStore(llvm::Value* _index, llvm::Value* _value);
 
-	llvm::Value* code();
-
 	llvm::Value* balance(llvm::Value* _address);
 	void suicide(llvm::Value* _address);
 	llvm::Value* calldataload(llvm::Value* _index);
@@ -33,8 +31,6 @@ public:
 	llvm::Value* codeAt(llvm::Value* _addr);
 	llvm::Value* codesizeAt(llvm::Value* _addr);
 
-private:
-	llvm::Value* getDataElem(unsigned _index, const llvm::Twine& _name = "");
 
 private:
 	llvm::Value* m_args[2];
@@ -46,7 +42,6 @@ private:
 	llvm::Value* m_arg7;
 	llvm::Value* m_arg8;
 	llvm::Value* m_data;
-	llvm::Function* m_init;
 	llvm::Function* m_store;
 	llvm::Function* m_setStore;
 	llvm::Function* m_calldataload;
diff --git a/libevmjit/Runtime.cpp b/libevmjit/Runtime.cpp
index 523d5ea90..489034a17 100644
--- a/libevmjit/Runtime.cpp
+++ b/libevmjit/Runtime.cpp
@@ -23,6 +23,7 @@ llvm::StructType* RuntimeData::getType()
 		llvm::Type* elems[] =
 		{
 			llvm::ArrayType::get(Type::i256, _size),
+			Type::BytePtr,
 			Type::BytePtr
 		};
 		type = llvm::StructType::create(elems, "RuntimeData");
@@ -77,6 +78,7 @@ Runtime::Runtime(u256 _gas, ExtVMFace& _ext):
 	set(RuntimeData::GasLimit, _ext.currentBlock.gasLimit);
 	set(RuntimeData::CodeSize, _ext.code.size());	// TODO: Use constant
 	m_data.callData = _ext.data.data();
+	m_data.code = _ext.code.data();
 }
 
 Runtime::~Runtime()
@@ -165,6 +167,12 @@ llvm::Value* RuntimeManager::getCallData()
 	return getBuilder().CreateLoad(ptr, "calldata");
 }
 
+llvm::Value* RuntimeManager::getCode()
+{
+	auto ptr = getBuilder().CreateStructGEP(getRuntimePtr(), 2, "codePtr");
+	return getBuilder().CreateLoad(ptr, "code");
+}
+
 llvm::Value* RuntimeManager::getGas()
 {
 	return get(RuntimeData::Gas);
diff --git a/libevmjit/Runtime.h b/libevmjit/Runtime.h
index d8f1a2624..7eb7c2d38 100644
--- a/libevmjit/Runtime.h
+++ b/libevmjit/Runtime.h
@@ -46,6 +46,7 @@ struct RuntimeData
 
 	i256 elems[_size];
 	byte const* callData;
+	byte const* code;
 
 	static llvm::StructType* getType();
 };
@@ -92,6 +93,7 @@ public:
 	llvm::Value* get(Instruction _inst);
 	llvm::Value* getGas();	// TODO: Remove
 	llvm::Value* getCallData();
+	llvm::Value* getCode();
 	void setGas(llvm::Value* _gas);
 
 private: