From e53c0a48775da1696bae14a9111534f904aa33c6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Bylica?= <pawel.bylica@imapp.pl>
Date: Tue, 30 Sep 2014 10:05:43 +0200
Subject: [PATCH] SWAP* instructions

---
 evmcc/Compiler.cpp            | 25 ++++++++++++++++++++
 evmcc/Stack.cpp               | 43 ++++++++++++++++++++++++++---------
 evmcc/Stack.h                 |  2 ++
 evmcc/bytecode/stack_test.evm |  2 +-
 evmcc/lll/stack_test.lll      |  4 +++-
 5 files changed, 63 insertions(+), 13 deletions(-)

diff --git a/evmcc/Compiler.cpp b/evmcc/Compiler.cpp
index 454c61450..0b1793244 100644
--- a/evmcc/Compiler.cpp
+++ b/evmcc/Compiler.cpp
@@ -134,6 +134,31 @@ std::unique_ptr<llvm::Module> Compiler::compile(const dev::bytes& bytecode)
 			break;
 		}
 
+		case Instruction::SWAP1:
+		case Instruction::SWAP2:
+		case Instruction::SWAP3:
+		case Instruction::SWAP4:
+		case Instruction::SWAP5:
+		case Instruction::SWAP6:
+		case Instruction::SWAP7:
+		case Instruction::SWAP8:
+		case Instruction::SWAP9:
+		case Instruction::SWAP10:
+		case Instruction::SWAP11:
+		case Instruction::SWAP12:
+		case Instruction::SWAP13:
+		case Instruction::SWAP14:
+		case Instruction::SWAP15:
+		case Instruction::SWAP16:
+		{
+			auto index = static_cast<uint32_t>(inst) - static_cast<uint32_t>(Instruction::SWAP1) + 1;
+			auto loValue = stack.get(index);
+			auto hiValue = stack.get(0);
+			stack.set(index, hiValue);
+			stack.set(0, loValue);
+			break;
+		}
+
 		}
 	}
 
diff --git a/evmcc/Stack.cpp b/evmcc/Stack.cpp
index 4c5a19e8a..f0af2087a 100644
--- a/evmcc/Stack.cpp
+++ b/evmcc/Stack.cpp
@@ -55,6 +55,9 @@ Stack::Stack(llvm::IRBuilder<>& _builder, llvm::Module* _module)
 	m_stackGet = llvm::Function::Create(getFuncType,
 		llvm::GlobalValue::LinkageTypes::ExternalLinkage, "evmccrt_stack_get", _module);
 
+	m_stackSet = llvm::Function::Create(getFuncType,
+		llvm::GlobalValue::LinkageTypes::ExternalLinkage, "evmccrt_stack_set", _module);
+
 	m_args[0] = m_builder.CreateCall(stackCreate, "stack.ptr");
 	m_args[1] = m_builder.CreateAlloca(i256Ty, nullptr, "stack.val");
 }
@@ -82,24 +85,33 @@ llvm::Value* Stack::get(uint32_t _index)
 }
 
 
+void Stack::set(uint32_t _index, llvm::Value* _value)
+{
+	m_builder.CreateStore(_value, m_args[1]);	  // copy value to memory
+	llvm::Value* args[] = {m_args[0], m_builder.getInt32(_index), m_args[1]};
+	m_builder.CreateCall(m_stackSet, args);
+}
+
+
 llvm::Value* Stack::top()
 {
 	return get(0);
 }
 
 
-void debugStack(const char* op, const i256& word)
+void debugStack(const char* _op, const i256& _word, uint32_t _index = 0)
 {
-	std::cerr << "STACK " << std::setw(4) << std::setfill(' ') << op << ": "
-		<< std::dec << word.a
-		<< " HEX: " << std::hex << std::setfill('0');
-	if (word.b || word.c || word.d)
+	std::cerr << "STACK " << std::setw(4) << std::setfill(' ') << _op
+			  << " [" << std::setw(2) << std::setfill('0') << _index << "] "
+			  << std::dec << _word.a
+			  << " HEX: " << std::hex;
+	if (_word.b || _word.c || _word.d)
 	{
-		std::cerr << std::setw(16) << word.d << " "
-			<< std::setw(16) << word.c << " "
-			<< std::setw(16) << word.b << " ";
+		std::cerr << std::setw(16) << _word.d << " "
+				  << std::setw(16) << _word.c << " "
+				  << std::setw(16) << _word.b << " ";
 	}
-	std::cerr << std::setw(16) << word.a << "\n";
+	std::cerr << std::setw(16) << _word.a << "\n";
 }
 
 }
@@ -138,10 +150,19 @@ EXPORT void evmccrt_stack_get(void* _stack, uint32_t _index, void* _pWord)
 {
 	auto stack = static_cast<StackImpl*>(_stack);
 	assert(_index < stack->size());
-	auto word = &(*stack)[stack->size() - _index - 1];
-	debugStack("get", *word);
+	auto word = stack->rbegin() + _index;
+	debugStack("get", *word, _index);
 	auto outWord = static_cast<i256*>(_pWord);
 	*outWord = *word;
 }
 
+EXPORT void evmccrt_stack_set(void* _stack, uint32_t _index, void* _pWord)
+{
+	auto stack = static_cast<StackImpl*>(_stack);
+	auto word = static_cast<i256*>(_pWord);
+	assert(_index < stack->size());
+	*(stack->rbegin() + _index) = *word;
+	debugStack("set", *word, _index);
+}
+
 }	// extern "C"
diff --git a/evmcc/Stack.h b/evmcc/Stack.h
index e4d93bd8c..a25de56e6 100644
--- a/evmcc/Stack.h
+++ b/evmcc/Stack.h
@@ -15,6 +15,7 @@ public:
 	llvm::Value* pop();
 	llvm::Value* top();
 	llvm::Value* get(uint32_t _index);
+	void set(uint32_t _index, llvm::Value* _value);
 
 private:
 	llvm::IRBuilder<>& m_builder;
@@ -22,6 +23,7 @@ private:
 	llvm::Function* m_stackPush;
 	llvm::Function* m_stackPop;
 	llvm::Function* m_stackGet;
+	llvm::Function* m_stackSet;
 };
 
 }
\ No newline at end of file
diff --git a/evmcc/bytecode/stack_test.evm b/evmcc/bytecode/stack_test.evm
index 6eba9fdca..02417c967 100644
--- a/evmcc/bytecode/stack_test.evm
+++ b/evmcc/bytecode/stack_test.evm
@@ -1 +1 @@
-65372d0f1dca6e661925338e3e5c2b808280848184505050505050506104576108ae8182
+65372d0f1dca6e661925338e3e5c2b808280848184505050505050506104576108ae81819290
diff --git a/evmcc/lll/stack_test.lll b/evmcc/lll/stack_test.lll
index 0edf06624..e6581563c 100644
--- a/evmcc/lll/stack_test.lll
+++ b/evmcc/lll/stack_test.lll
@@ -18,5 +18,7 @@ POP
 1111
 2222
 DUP2
-DUP3
+DUP2
+SWAP3
+SWAP1
 )
\ No newline at end of file