openethereum/src/evm/interpreter.rs

823 lines
23 KiB
Rust
Raw Normal View History

///! Rust VM implementation
use common::*;
use evm;
use super::instructions as instructions;
use super::instructions::Instruction;
type CodePosition = usize;
type Gas = U256;
type ProgramCounter = usize;
/// Stack trait with VM-friendly API
trait Stack<T> {
/// Returns `Stack[len(Stack) - no_from_top]`
fn peek(&self, no_from_top: usize) -> &T;
/// Swaps Stack[len(Stack)] and Stack[len(Stack) - no_from_top]
fn swap_with_top(&mut self, no_from_top: usize);
/// Returns true if Stack has at least `no_of_elems` elements
fn has(&self, no_of_elems: usize) -> bool;
/// Get element from top and remove it from Stack. Panics if stack is empty.
fn pop_back(&mut self) -> T;
/// Get elements from top and remove them from Stack. Panics if stack is empty.
fn pop_n(&mut self, no_of_elems: usize) -> Vec<T>;
/// Add element on top of the Stack
fn push(&mut self, elem: T);
/// Get number of elements on Stack
fn size(&self) -> usize;
}
2016-01-13 12:02:49 +01:00
impl<S : fmt::Display> Stack<S> for Vec<S> {
fn peek(&self, no_from_top: usize) -> &S {
return &self[self.len() - no_from_top - 1];
}
fn swap_with_top(&mut self, no_from_top: usize) {
let len = self.len();
self.swap(len - no_from_top - 1, len - 1);
}
fn has(&self, no_of_elems: usize) -> bool {
self.len() >= no_of_elems
}
fn pop_back(&mut self) -> S {
let val = self.pop();
println!("Popping from stack.");
match val {
Some(x) => x,
None => panic!("Tried to pop from empty stack.")
}
}
fn pop_n(&mut self, no_of_elems: usize) -> Vec<S> {
let mut vec = Vec::new();
for _i in 1..no_of_elems+1 {
vec.push(self.pop_back());
}
vec
}
fn push(&mut self, elem: S) {
2016-01-13 12:02:49 +01:00
println!("Pushing to stack: {}", elem);
self.push(elem);
}
fn size(&self) -> usize {
self.len()
}
}
2016-01-13 22:30:41 +01:00
trait Memory {
/// Retrieve current size of the memory
fn size(&self) -> usize;
/// Resize (shrink or expand) the memory to specified size (fills 0)
fn resize(&mut self, new_size: usize);
/// Resize the memory only if its smaller
fn expand(&mut self, new_size: U256);
2016-01-13 22:30:41 +01:00
/// Write single byte to memory
fn write_byte(&mut self, offset: U256, value: U256);
/// Write a word to memory. Does not resize memory!
2016-01-13 22:30:41 +01:00
fn write(&mut self, offset: U256, value: U256);
/// Read a word from memory
fn read(&self, offset: U256) -> U256;
/// Write slice of bytes to memory. Does not resize memory!
2016-01-13 22:30:41 +01:00
fn write_slice(&mut self, offset: U256, &[u8]);
/// Retrieve part of the memory between offset and offset + size
fn read_slice(&self, offset: U256, size: U256) -> &[u8];
}
impl Memory for Vec<u8> {
fn size(&self) -> usize {
return self.len()
}
fn read_slice(&self, init_off_u: U256, init_size_u: U256) -> &[u8] {
let init_off = init_off_u.low_u64() as usize;
let init_size = init_size_u.low_u64() as usize;
&self[init_off..init_off + init_size]
}
fn read(&self, offset: U256) -> U256 {
2016-01-13 22:30:41 +01:00
let off = offset.low_u64() as usize;
let mut val = U256::zero();
for pos in off..off+32 {
2016-01-13 22:30:41 +01:00
val = val << 8;
val = val | U256::from(self[pos] as u8);
2016-01-13 22:30:41 +01:00
}
val
}
fn write_slice(&mut self, offset: U256, slice: &[u8]) {
let off = offset.low_u64() as usize;
2016-01-13 22:30:41 +01:00
// TODO [todr] Optimize?
for pos in off..off+slice.len() {
println!("Writing {:x}", slice[pos - off]);
2016-01-13 22:30:41 +01:00
self[pos] = slice[pos - off];
}
}
fn write(&mut self, offset: U256, value: U256) {
let off = offset.low_u64() as usize;
let mut val = value;
2016-01-13 22:30:41 +01:00
let end = off + 32;
for pos in off..end {
self[end - pos - 1] = (val & U256::from(0xff)).low_u64() as u8;
val = val >> 8;
}
2016-01-13 22:30:41 +01:00
}
fn write_byte(&mut self, offset: U256, value: U256) {
let off = offset.low_u64() as usize;
let val = value.low_u64() as u64;
self[off] = (val & 0xff) as u8;
}
fn resize(&mut self, new_size: usize) {
self.resize(new_size, 0);
}
fn expand(&mut self, new_size: U256) {
let size = new_size.low_u64() as usize;
if size > self.len() {
Memory::resize(self, size)
}
}
2016-01-13 22:30:41 +01:00
}
/// Abstraction over raw vector of Bytes. Easier state management of PC.
struct CodeReader<'a> {
position: ProgramCounter,
code: &'a Bytes
}
impl<'a> CodeReader<'a> {
/// Get `no_of_bytes` from code and convert to U256. Move PC
fn read(&mut self, no_of_bytes: usize) -> U256 {
2016-01-13 12:02:49 +01:00
let mut value = U256::from(self.code[self.position]);
for off in 1..no_of_bytes {
let pos = self.position + off;
value = value << 8;
// TODO [todr] Directly bitor with u8?
value = value | U256::from(self.code[pos]);
}
// Move PC
self.position += no_of_bytes;
2016-01-13 12:02:49 +01:00
value
}
2016-01-13 01:53:33 +01:00
fn len (&self) -> usize {
self.code.len()
}
}
2016-01-13 17:11:00 +01:00
enum InstructionCost {
Gas(U256),
GasMem(U256, U256),
GasMemCopy(U256, U256, U256)
}
enum InstructionResult {
AdditionalGasCost(U256),
JumpToPosition(U256),
StopExecutionWithGasCost(U256),
StopExecution
}
pub struct Interpreter;
impl evm::Evm for Interpreter {
fn exec(&self, params: &ActionParams, ext: &mut evm::Ext) -> evm::Result {
let code = &params.code;
let valid_jump_destinations = self.find_jump_destinations(&code);
// TODO reserve stack
let mut current_gas = params.gas.clone();
let mut stack = vec![];
2016-01-13 22:30:41 +01:00
let mut mem = vec![];
let mut reader = CodeReader {
position: 0,
code: &code
};
while reader.position < code.len() {
let instruction = code[reader.position];
2016-01-13 12:02:49 +01:00
reader.position += 1;
// Calculate gas cost
let gas_cost = try!(self.get_gas_cost_and_expand_mem(ext, instruction, &mut mem, &stack));
try!(self.verify_gas(&current_gas, &gas_cost));
current_gas = current_gas - gas_cost;
2016-01-13 17:11:00 +01:00
println!("Gas cost: {} (left: {})", gas_cost, current_gas);
println!("Executing: {} ", instructions::get_info(instruction).name);
// Execute instruction
let result = try!(self.exec_instruction(
2016-01-13 22:30:41 +01:00
current_gas, params, ext, instruction, &mut reader, &mut mem, &mut stack
));
// Advance
match result {
InstructionResult::JumpToPosition(position) => {
let pos = try!(self.verify_jump(position, &valid_jump_destinations));
reader.position = pos;
},
InstructionResult::AdditionalGasCost(gas_cost) => {
current_gas = current_gas - gas_cost;
},
InstructionResult::StopExecutionWithGasCost(gas_cost) => {
current_gas = current_gas - gas_cost;
reader.position = code.len();
},
InstructionResult::StopExecution => {
reader.position = code.len();
}
}
}
Ok(current_gas)
}
}
impl Interpreter {
fn get_gas_cost_and_expand_mem(&self,
ext: &evm::Ext,
instruction: Instruction,
mem: &mut Memory,
stack: &Stack<U256>
) -> evm::Result {
let schedule = ext.schedule();
let info = instructions::get_info(instruction);
if !schedule.have_delegate_call && instruction == instructions::DELEGATECALL {
return Err(evm::Error::BadInstruction);
}
if info.tier == instructions::GasPriceTier::InvalidTier {
return Err(evm::Error::BadInstruction);
}
try!(self.verify_instructions_requirements(&info, schedule.stack_limit, stack));
let tier = instructions::get_tier_idx(info.tier);
2016-01-13 17:11:00 +01:00
let default_gas = U256::from(schedule.tier_step_gas[tier]);
let cost = match instruction {
instructions::SSTORE => {
let address = H256::from(stack.peek(0));
let val = U256::from(ext.sload(&address).as_slice());
let gas = if self.is_zero(&val) && !self.is_zero(stack.peek(1)) {
schedule.sstore_set_gas
} else {
schedule.sstore_reset_gas
};
InstructionCost::Gas(U256::from(gas))
},
instructions::SLOAD => {
InstructionCost::Gas(U256::from(schedule.sload_gas))
},
instructions::MSTORE => {
InstructionCost::GasMem(default_gas, add_u256_usize(stack.peek(0), 32))
},
instructions::MLOAD => {
InstructionCost::GasMem(default_gas, add_u256_usize(stack.peek(0), 32))
},
instructions::MSTORE8 => {
InstructionCost::GasMem(default_gas, add_u256_usize(stack.peek(0), 1))
},
instructions::RETURN => {
InstructionCost::GasMem(default_gas, self.mem_needed(stack.peek(0), stack.peek(1)))
},
instructions::SHA3 => {
let words = add_u256_usize(stack.peek(1), 31) / U256::from(32);
let gas = U256::from(schedule.sha3_gas) + (U256::from(schedule.sha3_word_gas) * words);
InstructionCost::GasMem(gas, self.mem_needed(stack.peek(0), stack.peek(1)))
},
instructions::CALLDATACOPY => {
InstructionCost::GasMemCopy(default_gas, self.mem_needed(stack.peek(0), stack.peek(2)), stack.peek(2).clone())
},
instructions::CODECOPY => {
InstructionCost::GasMemCopy(default_gas, self.mem_needed(stack.peek(0), stack.peek(2)), stack.peek(2).clone())
},
instructions::EXTCODECOPY => {
InstructionCost::GasMemCopy(default_gas, self.mem_needed(stack.peek(1), stack.peek(3)), stack.peek(3).clone())
},
instructions::JUMPDEST => {
InstructionCost::Gas(U256::one())
},
instructions::LOG0...instructions::LOG4 => {
let no_of_topics = instructions::get_log_topics(instruction);
let log_gas = schedule.log_gas + schedule.log_topic_gas * no_of_topics;
let data_gas = stack.peek(1).clone() * U256::from(schedule.log_data_gas);
let gas = data_gas + U256::from(log_gas);
InstructionCost::GasMem(gas, self.mem_needed(stack.peek(0), stack.peek(1)))
},
instructions::CALL | instructions::CALLCODE => {
let gas = add_u256_usize(stack.peek(0), schedule.call_gas + schedule.call_value_transfer_gas);
let mem = cmp::max(
self.mem_needed(stack.peek(5), stack.peek(6)),
self.mem_needed(stack.peek(3), stack.peek(4))
);
InstructionCost::GasMem(gas, mem)
},
instructions::DELEGATECALL => {
let gas = add_u256_usize(stack.peek(0), schedule.call_gas);
let mem = cmp::max(
self.mem_needed(stack.peek(4), stack.peek(5)),
self.mem_needed(stack.peek(2), stack.peek(3))
);
InstructionCost::GasMem(gas, mem)
},
instructions::CREATE => {
let gas = U256::from(schedule.create_gas);
let mem = self.mem_needed(stack.peek(1), stack.peek(2));
InstructionCost::GasMem(gas, mem)
},
instructions::EXP => {
let expon = stack.peek(1);
// TODO [todr] not sure how to calculate that
let gas = U256::from(schedule.exp_gas);
InstructionCost::Gas(gas)
},
_ => InstructionCost::Gas(default_gas)
};
2016-01-13 17:11:00 +01:00
match cost {
InstructionCost::Gas(gas) => {
Ok(gas)
},
InstructionCost::GasMem(gas, mem_size) => {
mem.expand(mem_size);
let mem_gas = self.mem_gas_cost(schedule, mem.size(), &mem_size);
2016-01-13 22:30:41 +01:00
Ok(gas + mem_gas)
2016-01-13 17:11:00 +01:00
},
InstructionCost::GasMemCopy(gas, mem_size, copy) => {
mem.expand(mem_size);
let mem_gas = self.mem_gas_cost(schedule, mem.size(), &mem_size);
2016-01-13 17:11:00 +01:00
let copy_gas = U256::from(schedule.copy_gas) * (add_u256_usize(&copy, 31) / U256::from(32));
2016-01-13 22:30:41 +01:00
Ok(gas + copy_gas + mem_gas)
2016-01-13 17:11:00 +01:00
}
}
}
2016-01-13 22:30:41 +01:00
fn mem_gas_cost(&self, schedule: &evm::Schedule, current_mem_size: usize, mem_size: &U256) -> U256 {
let gas_for_mem = |mem_size: usize| {
let s = mem_size / 32;
schedule.memory_gas * s + s * s / schedule.quad_coeff_div
};
let mem_size_rounded = add_u256_usize(mem_size, 31).low_u64() as usize / 32 * 32;
let mem_gas = gas_for_mem(mem_size_rounded);
let current_mem_gas = gas_for_mem(current_mem_size);
U256::from(if mem_gas > current_mem_gas {
mem_gas - current_mem_gas
} else {
0
})
}
2016-01-13 17:11:00 +01:00
fn mem_needed(&self, offset: &U256, size: &U256) -> U256 {
if self.is_zero(size) {
U256::zero()
} else {
2016-01-13 17:11:00 +01:00
offset.clone() + size.clone()
}
}
fn exec_instruction(&self,
gas: Gas,
params: &ActionParams,
ext: &mut evm::Ext,
instruction: Instruction,
code: &mut CodeReader,
2016-01-13 22:30:41 +01:00
mem: &mut Memory,
stack: &mut Stack<U256>
) -> Result<InstructionResult, evm::Error> {
match instruction {
instructions::JUMP => {
let jump = stack.pop_back();
return Ok(InstructionResult::JumpToPosition(
jump
));
},
instructions::JUMPI => {
let condition = stack.pop_back();
let jump = stack.pop_back();
2016-01-13 17:11:00 +01:00
if !self.is_zero(&condition) {
return Ok(InstructionResult::JumpToPosition(
jump
));
}
},
instructions::JUMPDEST => {
// ignore
},
instructions::CREATE => {
let endowment = stack.pop_back();
let init_off = stack.pop_back();
let init_size = stack.pop_back();
// TODO [todr] Fix u64 for gas
2016-01-13 22:30:41 +01:00
let contract_code = mem.read_slice(init_off, init_size);
// TODO [todr] Fix u64 for gasLeft
let (gas_left, maybe_address) = try!(
ext.create(gas.low_u64(), &endowment, &contract_code)
);
match maybe_address {
2016-01-13 01:53:33 +01:00
Some(address) => stack.push(address_to_u256(address)),
None => stack.push(U256::zero())
}
return Ok(InstructionResult::AdditionalGasCost(
gas - Gas::from(gas_left)
));
},
// CALL, CALLCODE, DELEGATECALL
instructions::RETURN => {
let init_off = stack.pop_back();
let init_size = stack.pop_back();
2016-01-13 22:30:41 +01:00
let return_code = mem.read_slice(init_off, init_size);
// TODO [todr] Fix u64 for gas
let gas_left = try!(ext.ret(gas.low_u64(), &return_code));
// TODO [todr] Fix u64 for gasLeft
return Ok(InstructionResult::StopExecutionWithGasCost(
gas - Gas::from(gas_left)
));
},
instructions::STOP => {
return Ok(InstructionResult::StopExecution);
},
instructions::SUICIDE => {
// TODO [todr] Suicide should have argument with address of contract that funds should be transfered to
let address = stack.pop_back();
// ext.suicide(Address::from(address));
ext.suicide();
return Ok(InstructionResult::StopExecution);
},
instructions::LOG0...instructions::LOG4 => {
let no_of_topics = instructions::get_log_topics(instruction);
let topics_data = stack.pop_n(no_of_topics + 2);
let offset = topics_data[0];
let size = topics_data[1];
let topics = topics_data
.iter()
.skip(2)
.map(H256::from)
.collect();
2016-01-13 22:30:41 +01:00
ext.log(topics, mem.read_slice(offset, size));
},
instructions::PUSH1...instructions::PUSH32 => {
let bytes = instructions::get_push_bytes(instruction);
let val = code.read(bytes);
stack.push(val);
},
instructions::MLOAD => {
2016-01-13 22:30:41 +01:00
let word = mem.read(stack.pop_back());
stack.push(U256::from(word));
},
instructions::MSTORE => {
2016-01-13 22:30:41 +01:00
let offset = stack.pop_back();
let word = stack.pop_back();
mem.write(offset, word);
},
instructions::MSTORE8 => {
2016-01-13 22:30:41 +01:00
let offset = stack.pop_back();
let byte = stack.pop_back();
mem.write_byte(offset, byte);
},
instructions::MSIZE => {
2016-01-13 22:30:41 +01:00
stack.push(U256::from(mem.size()));
},
2016-01-13 01:53:33 +01:00
instructions::SHA3 => {
let offset = stack.pop_back();
let size = stack.pop_back();
2016-01-13 22:30:41 +01:00
let sha3 = mem.read_slice(offset, size).sha3();
2016-01-13 01:53:33 +01:00
stack.push(U256::from(sha3.as_slice()));
},
instructions::SLOAD => {
let key = H256::from(&stack.pop_back());
let word = U256::from(ext.sload(&key).as_slice());
stack.push(word);
},
instructions::SSTORE => {
let key = H256::from(&stack.pop_back());
let word = H256::from(&stack.pop_back());
ext.sstore(key, word);
},
instructions::PC => {
stack.push(U256::from(code.position));
},
instructions::GAS => {
stack.push(gas.clone());
},
2016-01-13 01:53:33 +01:00
instructions::ADDRESS => {
stack.push(address_to_u256(params.address.clone()));
},
instructions::ORIGIN => {
stack.push(address_to_u256(params.origin.clone()));
},
instructions::BALANCE => {
let address = u256_to_address(&stack.pop_back());
let balance = ext.balance(&address);
stack.push(balance);
},
instructions::CALLER => {
stack.push(address_to_u256(params.sender.clone()));
},
instructions::CALLVALUE => {
stack.push(params.value.clone());
},
instructions::CALLDATALOAD => {
let id = stack.pop_back().low_u64() as usize;
let mut v = params.data[id..id+32].to_vec();
v.resize(32, 0);
stack.push(U256::from(&v[..]))
},
2016-01-13 01:53:33 +01:00
instructions::CALLDATASIZE => {
stack.push(U256::from(params.data.len()));
},
instructions::CODESIZE => {
stack.push(U256::from(code.len()));
},
instructions::EXTCODESIZE => {
let address = u256_to_address(&stack.pop_back());
let len = ext.extcode(&address).len();
stack.push(U256::from(len));
},
2016-01-13 22:30:41 +01:00
instructions::CALLDATACOPY => {
self.copy_data_to_memory(mem, stack, &params.data);
},
instructions::CODECOPY => {
self.copy_data_to_memory(mem, stack, &params.code);
},
instructions::EXTCODECOPY => {
let address = u256_to_address(&stack.pop_back());
let code = ext.extcode(&address);
for b in &code {
println!("Code: {:x}", b);
}
2016-01-13 22:30:41 +01:00
self.copy_data_to_memory(mem, stack, &code);
},
2016-01-13 01:53:33 +01:00
instructions::GASPRICE => {
stack.push(params.gas_price.clone());
},
instructions::BLOCKHASH => {
let block_number = stack.pop_back();
let block_hash = ext.blockhash(&block_number);
stack.push(U256::from(block_hash.as_slice()));
},
instructions::COINBASE => {
stack.push(address_to_u256(ext.env_info().author.clone()));
},
instructions::TIMESTAMP => {
stack.push(U256::from(ext.env_info().timestamp));
},
instructions::NUMBER => {
stack.push(U256::from(ext.env_info().number));
},
instructions::DIFFICULTY => {
stack.push(ext.env_info().difficulty.clone());
},
instructions::GASLIMIT => {
stack.push(ext.env_info().gas_limit.clone());
},
_ => {
self.exec_stack_instruction(instruction, stack);
}
};
Ok(InstructionResult::AdditionalGasCost(U256::zero()))
}
2016-01-13 22:30:41 +01:00
fn copy_data_to_memory(&self,
mem: &mut Memory,
stack: &mut Stack<U256>,
data: &Bytes) {
let offset = stack.pop_back();
let index = stack.pop_back().low_u64() as usize;
let size = stack.pop_back().low_u64() as usize;
mem.write_slice(offset, &data[index..index+size]);
2016-01-13 22:30:41 +01:00
}
2016-01-13 17:11:00 +01:00
fn verify_instructions_requirements(&self,
info: &instructions::InstructionInfo,
stack_limit: usize,
stack: &Stack<U256>) -> Result<(), evm::Error> {
if !stack.has(info.args) {
Err(evm::Error::StackUnderflow(info.args, stack.size()))
} else if stack.size() - info.args + info.ret > stack_limit {
Err(evm::Error::OutOfStack(info.ret - info.args, stack_limit))
} else {
Ok(())
}
}
fn verify_gas(&self, current_gas: &U256, gas_cost: &U256) -> Result<(), evm::Error> {
if current_gas < gas_cost {
Err(evm::Error::OutOfGas)
} else {
Ok(())
}
}
fn verify_jump(&self, jump_u: U256, valid_jump_destinations: &HashSet<usize>) -> Result<usize, evm::Error> {
let jump = jump_u.low_u64() as usize;
if valid_jump_destinations.contains(&jump) {
Ok(jump)
} else {
Err(evm::Error::BadJumpDestination)
}
}
2016-01-13 17:11:00 +01:00
fn is_zero(&self, val: &U256) -> bool {
&U256::zero() == val
}
2016-01-13 01:53:33 +01:00
fn bool_to_u256(&self, val: bool) -> U256 {
if val {
U256::one()
} else {
U256::zero()
}
}
fn exec_stack_instruction(&self, instruction: Instruction, stack : &mut Stack<U256>) {
match instruction {
instructions::DUP1...instructions::DUP16 => {
let position = instructions::get_dup_position(instruction);
let val = stack.peek(position).clone();
stack.push(val);
},
instructions::SWAP1...instructions::SWAP16 => {
let position = instructions::get_swap_position(instruction);
stack.swap_with_top(position)
},
2016-01-13 01:53:33 +01:00
instructions::POP => {
stack.pop_back();
},
instructions::ADD => {
let a = stack.pop_back();
let b = stack.pop_back();
stack.push(a + b);
},
2016-01-13 01:53:33 +01:00
instructions::MUL => {
let a = stack.pop_back();
let b = stack.pop_back();
stack.push(a * b);
},
instructions::SUB => {
let a = stack.pop_back();
let b = stack.pop_back();
stack.push(a - b);
},
instructions::DIV => {
let a = stack.pop_back();
let b = stack.pop_back();
2016-01-13 17:11:00 +01:00
stack.push(if self.is_zero(&b) {
2016-01-13 01:53:33 +01:00
a / b
} else {
U256::zero()
});
},
instructions::MOD => {
let a = stack.pop_back();
let b = stack.pop_back();
2016-01-13 17:11:00 +01:00
stack.push(if self.is_zero(&b) {
2016-01-13 01:53:33 +01:00
a % b
} else {
U256::zero()
});
},
// instructions::SDIV => {},
// instructions::SMOD => {},
// instructions::EXP => {},
instructions::NOT => {
let a = stack.pop_back();
stack.push(!a);
},
instructions::LT => {
let a = stack.pop_back();
let b = stack.pop_back();
stack.push(self.bool_to_u256(a < b));
},
// instructions::SLT => {},
instructions::GT => {
let a = stack.pop_back();
let b = stack.pop_back();
stack.push(self.bool_to_u256(a > b));
},
// instructions::SGT => {},
instructions::EQ => {
let a = stack.pop_back();
let b = stack.pop_back();
stack.push(self.bool_to_u256(a == b));
},
instructions::ISZERO => {
let a = stack.pop_back();
2016-01-13 17:11:00 +01:00
stack.push(self.bool_to_u256(self.is_zero(&a)));
2016-01-13 01:53:33 +01:00
},
instructions::AND => {
let a = stack.pop_back();
let b = stack.pop_back();
stack.push(a & b);
},
instructions::OR => {
let a = stack.pop_back();
let b = stack.pop_back();
stack.push(a | b);
},
instructions::XOR => {
let a = stack.pop_back();
let b = stack.pop_back();
stack.push(a ^ b);
},
// instructions::BYTE => {},
// instructions::ADDMOD => {},
// instructions::MULMOD => {},
// instructions::SIGNEXTEND => {},
_ => panic!(format!("Unknown stack instruction: {:x}", instruction))
}
}
fn find_jump_destinations(&self, code : &Bytes) -> HashSet<CodePosition> {
let mut jump_dests = HashSet::new();
let mut position = 0;
while position < code.len() {
let instruction = code[position];
if instruction == instructions::JUMPDEST {
jump_dests.insert(position);
} else if instructions::is_push(instruction) {
position += instructions::get_push_bytes(instruction);
}
position += 1;
}
return jump_dests;
}
2016-01-13 01:53:33 +01:00
}
2016-01-13 17:11:00 +01:00
fn add_u256_usize(value: &U256, num: usize) -> U256 {
value.clone() + U256::from(num)
}
2016-01-13 01:53:33 +01:00
fn u256_to_address(value: &U256) -> Address {
Address::from(H256::from(value))
}
fn address_to_u256(value: Address) -> U256 {
U256::from(H256::from(value).as_slice())
}
#[cfg(test)]
mod tests {
use common::*;
use super::*;
#[test]
fn test_find_jump_destinations() {
// given
let interpreter = Interpreter;
let code = "7fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff7fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff5b01600055".from_hex().unwrap();
// when
let valid_jump_destinations = interpreter.find_jump_destinations(&code);
// then
assert!(valid_jump_destinations.contains(&66));
}
2016-01-13 22:30:41 +01:00
#[test]
fn test_memory_read_and_write() {
// given
let mem : &mut super::Memory = &mut vec![];
mem.resize(32);
2016-01-13 22:30:41 +01:00
// when
mem.write(U256::from(0x00), U256::from(0xabcdef));
// then
assert_eq!(mem.read(U256::from(0x00)), U256::from(0xabcdef));
2016-01-13 22:30:41 +01:00
}
#[test]
fn test_memory_read_and_write_byte() {
// given
let mem : &mut super::Memory = &mut vec![];
mem.resize(32);
2016-01-13 22:30:41 +01:00
// when
mem.write_byte(U256::from(0x1d), U256::from(0xab));
mem.write_byte(U256::from(0x1e), U256::from(0xcd));
mem.write_byte(U256::from(0x1f), U256::from(0xef));
2016-01-13 22:30:41 +01:00
// then
assert_eq!(mem.read(U256::from(0x00)), U256::from(0xabcdef));
2016-01-13 22:30:41 +01:00
}
}