mirror of https://github.com/ethereum/go-ethereum
parent
3308491c92
commit
1fb84d3c5f
@ -1,174 +1,3 @@ |
||||
package vm |
||||
|
||||
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" |
||||
) |
||||
|
||||
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) { |
||||
closure, vm := setup(logger.DebugLevel, DebugVmTy) |
||||
ret, _, e := closure.Call(vm, nil) |
||||
c.Assert(e, checker.NotNil) |
||||
c.Skip("Depends on mutan. Requires serpent implementation") |
||||
c.Assert(ret, checker.DeepEquals, big9) |
||||
} |
||||
|
||||
func (s *VmSuite) TestVm(c *checker.C) { |
||||
closure, vm := setup(logger.DebugLevel, StandardVmTy) |
||||
ret, _, e := closure.Call(vm, nil) |
||||
c.Assert(e, checker.NotNil) |
||||
c.Skip("Depends on mutan. Requires serpent implementation") |
||||
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. Requires serpent implementation") |
||||
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) |
||||
|
||||
return out |
||||
`, DebugVmTy) |
||||
|
||||
exp := ethutil.RightPadBytes(crypto.Ripemd160(ethutil.LeftPadBytes([]byte{42}, 32)), 32) |
||||
c.Skip("Depends on mutan. Requires serpent implementation") |
||||
c.Assert(ret, checker.DeepEquals, exp) |
||||
} |
||||
|
||||
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)) |
||||
|
||||
vm := New(TestEnv{}, DebugVmTy) |
||||
_, _, e := closure.Call(vm, nil) |
||||
if e != nil { |
||||
fmt.Println(e) |
||||
} |
||||
} |
||||
// Tests have been removed in favour of general tests. If anything implementation specific needs testing, put it here
|
||||
|
Loading…
Reference in new issue