tests for sload, sstore, sha3

This commit is contained in:
debris 2015-12-30 12:03:40 +01:00
parent e3f59d82ef
commit 2e7f0e29de
3 changed files with 78 additions and 41 deletions

View File

@ -318,7 +318,7 @@ pub mod ffi {
let input = slice::from_raw_parts(begin, size as usize); let input = slice::from_raw_parts(begin, size as usize);
let outlen = out_hash.words.len() * 8; let outlen = out_hash.words.len() * 8;
let output = slice::from_raw_parts_mut(out_hash.words.as_mut_ptr() as *mut u8, outlen); let output = slice::from_raw_parts_mut(out_hash.words.as_mut_ptr() as *mut u8, outlen);
let mut sha3 = Keccak::new_sha3_256(); let mut sha3 = Keccak::new_keccak256();
sha3.update(input); sha3.update(input);
sha3.finalize(output); sha3.finalize(output);
} }

View File

@ -1,23 +1,27 @@
use util::hash::*; use util::hash::*;
use util::uint::*; use util::uint::*;
use state::*;
pub struct Env; pub struct Env {
state: State,
address: Address
}
impl Env { impl Env {
pub fn new() -> Env { pub fn new(state: State, address: Address) -> Env {
Env Env {
state: state,
address: address
}
} }
pub fn sload(&self, _index: &H256) -> H256 { pub fn sload(&self, index: &H256) -> H256 {
println!("sload!: {:?}", _index); self.state.storage_at(&self.address, index)
//unimplemented!();
H256::new()
} }
pub fn sstore(&self, _index: &H256, _value: &H256) { pub fn sstore(&mut self, index: H256, value: H256) {
println!("sstore!: {:?} , {:?}", _index, _value); println!("index: {:?}, value: {:?}", index, value);
//unimplemented!(); self.state.set_storage(&self.address, index, value)
} }
pub fn balance(&self, _address: &Address) -> U256 { pub fn balance(&self, _address: &Address) -> U256 {
@ -48,6 +52,12 @@ impl Env {
pub fn log(&self, _topics: &[H256], _data: &[u8]) { pub fn log(&self, _topics: &[H256], _data: &[u8]) {
unimplemented!(); unimplemented!();
} }
/// Drain state
// not sure if this is the best solution, but seems to be the easiest one, mk
pub fn state(self) -> State {
self.state
}
} }

View File

@ -1,4 +1,5 @@
use std::mem; use std::mem;
use std::ptr;
use evmjit; use evmjit;
use util::hash::*; use util::hash::*;
use util::uint::*; use util::uint::*;
@ -18,12 +19,11 @@ trait IntoJit<T> {
impl<'a> FromJit<&'a evmjit::I256> for U256 { impl<'a> FromJit<&'a evmjit::I256> for U256 {
fn from_jit(input: &'a evmjit::I256) -> Self { fn from_jit(input: &'a evmjit::I256) -> Self {
let mut res: U256 = unsafe { mem::uninitialized() }; unsafe {
res.0[0] = input.words[3]; let mut res: U256 = mem::uninitialized();
res.0[1] = input.words[2]; ptr::copy(input.words.as_ptr(), res.0.as_mut_ptr(), 4);
res.0[2] = input.words[1]; res
res.0[3] = input.words[0]; }
res
} }
} }
@ -42,12 +42,11 @@ impl<'a> FromJit<&'a evmjit::I256> for Address {
impl IntoJit<evmjit::I256> for U256 { impl IntoJit<evmjit::I256> for U256 {
fn into_jit(self) -> evmjit::I256 { fn into_jit(self) -> evmjit::I256 {
let mut res: evmjit::I256 = unsafe { mem::uninitialized() }; unsafe {
res.words[0] = self.0[3]; let mut res: evmjit::I256 = mem::uninitialized();
res.words[1] = self.0[2]; ptr::copy(self.0.as_ptr(), res.words.as_mut_ptr(), 4);
res.words[2] = self.0[1]; res
res.words[3] = self.0[0]; }
res
} }
} }
@ -56,7 +55,7 @@ impl IntoJit<evmjit::I256> for H256 {
let mut ret = [0; 4]; let mut ret = [0; 4];
for i in 0..self.bytes().len() { for i in 0..self.bytes().len() {
let rev = self.bytes().len() - 1 - i; let rev = self.bytes().len() - 1 - i;
let pos = i / 8; let pos = rev / 8;
ret[pos] += (self.bytes()[i] as u64) << (rev % 8) * 8; ret[pos] += (self.bytes()[i] as u64) << (rev % 8) * 8;
} }
evmjit::I256 { words: ret } evmjit::I256 { words: ret }
@ -117,7 +116,7 @@ impl<'a> evmjit::Env for EnvAdapter<'a> {
fn sstore(&mut self, index: *const evmjit::I256, value: *const evmjit::I256) { fn sstore(&mut self, index: *const evmjit::I256, value: *const evmjit::I256) {
unsafe { unsafe {
self.env.sstore(&H256::from_jit(&*index), &H256::from_jit(&*value)); self.env.sstore(H256::from_jit(&*index), H256::from_jit(&*value));
} }
} }
@ -200,12 +199,14 @@ impl evm::Evm for JitEvm {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use rustc_serialize::hex::FromHex;
use std::str::FromStr; use std::str::FromStr;
use util::hash::*; use util::hash::*;
use util::uint::*; use util::uint::*;
use evm::*; use evm::*;
use evm::jit::{FromJit, IntoJit}; use evm::jit::{FromJit, IntoJit};
use super::*; use super::*;
use state::*;
#[test] #[test]
fn test_to_and_from_u256() { fn test_to_and_from_u256() {
@ -238,25 +239,51 @@ mod tests {
} }
#[test] #[test]
fn test_env_adapter() { fn test_env_add() {
let address = Address::from_str("0f572e5295c57f15886f9b263e2f6d2d6c7b5ec6").unwrap();
let mut data = RuntimeData::new(); let mut data = RuntimeData::new();
data.coinbase = Address::from_str("2adc25665018aa1fe0e6bc666dac8fc2697ff9ba").unwrap(); data.address = address.clone();
data.difficulty = U256::from(0x0100);
data.gas_limit = U256::from(0x0f4240);
data.number = 0;
data.timestamp = 1;
data.address = Address::from_str("0f572e5295c57f15886f9b263e2f6d2d6c7b5ec6").unwrap();
data.caller = Address::from_str("cd1722f3947def4cf144679da39c4c32bdc35681").unwrap();
data.code = vec![0x60, 0x00, 0x60, 0x00, 0x55];
data.gas = 0x174876e800; data.gas = 0x174876e800;
data.gas_price = 0x3b9aca00; data.code = "7fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff7fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff01600055".from_hex().unwrap();
data.origin = Address::from_str("cd1722f3947def4cf144679da39c4c32bdc35681").unwrap();
data.call_value = U256::from_str("0de0b6b3a7640000").unwrap(); let mut env = Env::new(State::new_temp(), address.clone());
let mut env = Env::new();
let evm = JitEvm; let evm = JitEvm;
assert_eq!(evm.exec(data, &mut env), ReturnCode::Stop); assert_eq!(evm.exec(data, &mut env), ReturnCode::Stop);
let state = env.state();
assert_eq!(state.storage_at(&address, &H256::new()),
H256::from_str("fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffe").unwrap());
}
#[test]
fn test_env_sha3_0() {
let address = Address::from_str("0f572e5295c57f15886f9b263e2f6d2d6c7b5ec6").unwrap();
let mut data = RuntimeData::new();
data.address = address.clone();
data.gas = 0x174876e800;
data.code = "6000600020600055".from_hex().unwrap();
let mut env = Env::new(State::new_temp(), address.clone());
let evm = JitEvm;
assert_eq!(evm.exec(data, &mut env), ReturnCode::Stop);
let state = env.state();
assert_eq!(state.storage_at(&address, &H256::new()),
H256::from_str("c5d2460186f7233c927e7db2dcc703c0e500b653ca82273b7bfad8045d85a470").unwrap());
}
#[test]
fn test_env_sha3_1() {
let address = Address::from_str("0f572e5295c57f15886f9b263e2f6d2d6c7b5ec6").unwrap();
let mut data = RuntimeData::new();
data.address = address.clone();
data.gas = 0x174876e800;
data.code = "6005600420600055".from_hex().unwrap();
let mut env = Env::new(State::new_temp(), address.clone());
let evm = JitEvm;
assert_eq!(evm.exec(data, &mut env), ReturnCode::Stop);
let state = env.state();
assert_eq!(state.storage_at(&address, &H256::new()),
H256::from_str("c41589e7559804ea4a2080dad19d876a024ccb05117835447d72ce08c1d020ec").unwrap());
} }
} }