diff options
-rw-r--r-- | vm/main_test.go | 9 | ||||
-rw-r--r-- | vm/vm_test.go | 358 |
2 files changed, 184 insertions, 183 deletions
diff --git a/vm/main_test.go b/vm/main_test.go new file mode 100644 index 000000000..0ae03bf6a --- /dev/null +++ b/vm/main_test.go @@ -0,0 +1,9 @@ +package vm + +import ( + "testing" + + checker "gopkg.in/check.v1" +) + +func Test(t *testing.T) { checker.TestingT(t) } diff --git a/vm/vm_test.go b/vm/vm_test.go index d9e2a3295..19aa171a6 100644 --- a/vm/vm_test.go +++ b/vm/vm_test.go @@ -1,189 +1,181 @@ package vm -// import ( -// "bytes" -// "fmt" -// "io/ioutil" -// "log" -// "math/big" -// "os" -// "testing" - -// "github.com/ethereum/go-ethereum/crypto" -// "github.com/ethereum/go-ethereum/ethutil" -// "github.com/ethereum/go-ethereum/logger" -// "github.com/ethereum/go-ethereum/state" -// "github.com/ethereum/go-ethereum/trie" -// // "github.com/obscuren/mutan" -// ) - -// type TestEnv struct{} - -// func (TestEnv) Origin() []byte { return nil } -// func (TestEnv) BlockNumber() *big.Int { return nil } -// func (TestEnv) BlockHash() []byte { return nil } -// func (TestEnv) PrevHash() []byte { return nil } -// func (TestEnv) Coinbase() []byte { return nil } -// func (TestEnv) Time() int64 { return 0 } -// func (TestEnv) GasLimit() *big.Int { return nil } -// func (TestEnv) Difficulty() *big.Int { return nil } -// func (TestEnv) Value() *big.Int { return nil } -// func (TestEnv) AddLog(state.Log) {} - -// func (TestEnv) Transfer(from, to Account, amount *big.Int) error { -// return nil -// } - -// // This is likely to fail if anything ever gets looked up in the state trie :-) -// func (TestEnv) State() *state.State { -// return state.New(trie.New(nil, "")) -// } - -// const mutcode = ` -// var x = 0; -// for i := 0; i < 10; i++ { -// x = i -// } - -// return x` - -// func setup(level logger.LogLevel, typ Type) (*Closure, VirtualMachine) { -// code, err := ethutil.Compile(mutcode, true) -// if err != nil { -// log.Fatal(err) -// } - -// // Pipe output to /dev/null -// logger.AddLogSystem(logger.NewStdLogSystem(ioutil.Discard, log.LstdFlags, level)) - -// ethutil.ReadConfig(".ethtest", "/tmp/ethtest", "") - -// stateObject := state.NewStateObject([]byte{'j', 'e', 'f', 'f'}) -// callerClosure := NewClosure(nil, stateObject, stateObject, code, big.NewInt(1000000), big.NewInt(0)) - -// return callerClosure, New(TestEnv{}, typ) -// } - -// var big9 = ethutil.Hex2Bytes("0000000000000000000000000000000000000000000000000000000000000009") - -// func TestDebugVm(t *testing.T) { -// // if mutan.Version < "0.6" { -// // t.Skip("skipping for mutan version", mutan.Version, " < 0.6") -// // } - -// closure, vm := setup(logger.DebugLevel, DebugVmTy) -// ret, _, e := closure.Call(vm, nil) -// if e != nil { -// t.Fatalf("Call returned error: %v", e) -// } -// if !bytes.Equal(ret, big9) { -// t.Errorf("Wrong return value '%x', want '%x'", ret, big9) -// } -// } - -// func TestVm(t *testing.T) { -// // if mutan.Version < "0.6" { -// // t.Skip("skipping for mutan version", mutan.Version, " < 0.6") -// // } - -// closure, vm := setup(logger.DebugLevel, StandardVmTy) -// ret, _, e := closure.Call(vm, nil) -// if e != nil { -// t.Fatalf("Call returned error: %v", e) -// } -// if !bytes.Equal(ret, big9) { -// t.Errorf("Wrong return value '%x', want '%x'", ret, big9) -// } -// } - -// func BenchmarkDebugVm(b *testing.B) { -// closure, vm := setup(logger.InfoLevel, DebugVmTy) - -// b.ResetTimer() - -// for i := 0; i < b.N; i++ { -// closure.Call(vm, nil) -// } -// } - -// func BenchmarkVm(b *testing.B) { -// closure, vm := setup(logger.InfoLevel, StandardVmTy) - -// b.ResetTimer() - -// for i := 0; i < b.N; i++ { -// closure.Call(vm, nil) -// } -// } - -// func RunCode(mutCode string, typ Type) []byte { -// code, err := ethutil.Compile(mutCode, true) -// if err != nil { -// log.Fatal(err) -// } - -// logger.AddLogSystem(logger.NewStdLogSystem(os.Stdout, log.LstdFlags, logger.InfoLevel)) - -// ethutil.ReadConfig(".ethtest", "/tmp/ethtest", "") - -// stateObject := state.NewStateObject([]byte{'j', 'e', 'f', 'f'}) -// closure := NewClosure(nil, stateObject, stateObject, code, big.NewInt(1000000), big.NewInt(0)) +import ( + "fmt" + "io/ioutil" + "log" + "math/big" + "os" + + "github.com/ethereum/go-ethereum/crypto" + "github.com/ethereum/go-ethereum/ethutil" + "github.com/ethereum/go-ethereum/logger" + "github.com/ethereum/go-ethereum/state" + "github.com/ethereum/go-ethereum/trie" + checker "gopkg.in/check.v1" + // "github.com/obscuren/mutan" +) + +type VmSuite struct{} + +var _ = checker.Suite(&VmSuite{}) +var big9 = ethutil.Hex2Bytes("0000000000000000000000000000000000000000000000000000000000000009") + +const mutcode = ` +var x = 0; +for i := 0; i < 10; i++ { + x = i +} + +return x` + +type TestEnv struct{} + +func (TestEnv) Origin() []byte { return nil } +func (TestEnv) BlockNumber() *big.Int { return nil } +func (TestEnv) BlockHash() []byte { return nil } +func (TestEnv) PrevHash() []byte { return nil } +func (TestEnv) Coinbase() []byte { return nil } +func (TestEnv) Time() int64 { return 0 } +func (TestEnv) GasLimit() *big.Int { return nil } +func (TestEnv) Difficulty() *big.Int { return nil } +func (TestEnv) Value() *big.Int { return nil } +func (TestEnv) AddLog(*state.Log) {} +func (TestEnv) Transfer(from, to Account, amount *big.Int) error { + return nil +} + +// This is likely to fail if anything ever gets looked up in the state trie :-) +func (TestEnv) State() *state.State { + return state.New(trie.New(nil, "")) +} + +func setup(level logger.LogLevel, typ Type) (*Closure, VirtualMachine) { + code, err := ethutil.Compile(mutcode, true) + if err != nil { + log.Fatal(err) + } + + // Pipe output to /dev/null + logger.AddLogSystem(logger.NewStdLogSystem(ioutil.Discard, log.LstdFlags, level)) + + ethutil.ReadConfig(".ethtest", "/tmp/ethtest", "") + + stateObject := state.NewStateObject([]byte{'j', 'e', 'f', 'f'}) + callerClosure := NewClosure(nil, stateObject, stateObject, code, big.NewInt(1000000), big.NewInt(0)) + + return callerClosure, New(TestEnv{}, typ) +} + +func (s *VmSuite) TestDebugVm(c *checker.C) { + // if mutan.Version < "0.6" { + // t.Skip("skipping for mutan version", mutan.Version, " < 0.6") + // } + closure, vm := setup(logger.DebugLevel, DebugVmTy) + ret, _, e := closure.Call(vm, nil) + c.Assert(e, checker.NotNil) + c.Skip("Depends on mutan") + c.Assert(ret, checker.DeepEquals, big9) +} + +func (s *VmSuite) TestVm(c *checker.C) { + // if mutan.Version < "0.6" { + // t.Skip("skipping for mutan version", mutan.Version, " < 0.6") + // } + closure, vm := setup(logger.DebugLevel, StandardVmTy) + ret, _, e := closure.Call(vm, nil) + c.Assert(e, checker.NotNil) + c.Skip("Depends on mutan") + c.Assert(ret, checker.DeepEquals, big9) +} + +func (s *VmSuite) BenchmarkDebugVm(c *checker.C) { + closure, vm := setup(logger.InfoLevel, StandardVmTy) + + c.ResetTimer() + + for i := 0; i < c.N; i++ { + closure.Call(vm, nil) + } +} + +func (s *VmSuite) BenchmarkVm(c *checker.C) { + closure, vm := setup(logger.InfoLevel, DebugVmTy) + + c.ResetTimer() + + for i := 0; i < c.N; i++ { + closure.Call(vm, nil) + } +} + +func RunCode(mutCode string, typ Type) []byte { + code, err := ethutil.Compile(mutCode, true) + if err != nil { + log.Fatal(err) + } + + logger.AddLogSystem(logger.NewStdLogSystem(os.Stdout, log.LstdFlags, logger.InfoLevel)) + + ethutil.ReadConfig(".ethtest", "/tmp/ethtest", "") + + stateObject := state.NewStateObject([]byte{'j', 'e', 'f', 'f'}) + closure := NewClosure(nil, stateObject, stateObject, code, big.NewInt(1000000), big.NewInt(0)) + + vm := New(TestEnv{}, typ) + ret, _, e := closure.Call(vm, nil) + if e != nil { + fmt.Println(e) + } + + return ret +} + +func (s *VmSuite) TestBuildInSha256(c *checker.C) { + ret := RunCode(` + var in = 42 + var out = 0 + + call(0x2, 0, 10000, in, out) + + return out + `, DebugVmTy) + + exp := crypto.Sha256(ethutil.LeftPadBytes([]byte{42}, 32)) + c.Skip("Depends on mutan") + c.Assert(ret, checker.DeepEquals, exp) +} + +func (s *VmSuite) TestBuildInRipemd(c *checker.C) { + ret := RunCode(` + var in = 42 + var out = 0 + + call(0x3, 0, 10000, in, out) -// vm := New(TestEnv{}, typ) -// ret, _, e := closure.Call(vm, nil) -// if e != nil { -// fmt.Println(e) -// } - -// return ret -// } + return out + `, DebugVmTy) -// func TestBuildInSha256(t *testing.T) { -// ret := RunCode(` -// var in = 42 -// var out = 0 + exp := ethutil.RightPadBytes(crypto.Ripemd160(ethutil.LeftPadBytes([]byte{42}, 32)), 32) + c.Skip("Depends on mutan") + c.Assert(ret, checker.DeepEquals, exp) +} -// call(0x2, 0, 10000, in, out) +func (s *VmSuite) TestOog(c *checker.C) { + // This tests takes a long time and will eventually run out of gas + // t.Skip() + c.Skip("This tests takes a long time and will eventually run out of gas") + + logger.AddLogSystem(logger.NewStdLogSystem(os.Stdout, log.LstdFlags, logger.InfoLevel)) + + ethutil.ReadConfig(".ethtest", "/tmp/ethtest", "") + + stateObject := state.NewStateObject([]byte{'j', 'e', 'f', 'f'}) + closure := NewClosure(nil, stateObject, stateObject, ethutil.Hex2Bytes("60ff60ff600057"), big.NewInt(1000000), big.NewInt(0)) -// return out -// `, DebugVmTy) - -// exp := crypto.Sha256(ethutil.LeftPadBytes([]byte{42}, 32)) -// if bytes.Compare(ret, exp) != 0 { -// t.Errorf("Expected %x, got %x", exp, ret) -// } -// } - -// func TestBuildInRipemd(t *testing.T) { -// ret := RunCode(` -// var in = 42 -// var out = 0 - -// call(0x3, 0, 10000, in, out) - -// return out -// `, DebugVmTy) - -// exp := ethutil.RightPadBytes(crypto.Ripemd160(ethutil.LeftPadBytes([]byte{42}, 32)), 32) -// if bytes.Compare(ret, exp) != 0 { -// t.Errorf("Expected %x, got %x", exp, ret) -// } -// } - -// func TestOog(t *testing.T) { -// // This tests takes a long time and will eventually run out of gas -// // t.Skip() - -// logger.AddLogSystem(logger.NewStdLogSystem(os.Stdout, log.LstdFlags, logger.InfoLevel)) - -// ethutil.ReadConfig(".ethtest", "/tmp/ethtest", "") - -// stateObject := state.NewStateObject([]byte{'j', 'e', 'f', 'f'}) -// closure := NewClosure(nil, stateObject, stateObject, ethutil.Hex2Bytes("60ff60ff600057"), big.NewInt(1000000), big.NewInt(0)) - -// vm := New(TestEnv{}, DebugVmTy) -// _, _, e := closure.Call(vm, nil) -// if e != nil { -// fmt.Println(e) -// } -// } + vm := New(TestEnv{}, DebugVmTy) + _, _, e := closure.Call(vm, nil) + if e != nil { + fmt.Println(e) + } +} |