190 lines
		
	
	
		
			4.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			190 lines
		
	
	
		
			4.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| 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))
 | |
| 
 | |
| // 	vm := New(TestEnv{}, typ)
 | |
| // 	ret, _, e := closure.Call(vm, nil)
 | |
| // 	if e != nil {
 | |
| // 		fmt.Println(e)
 | |
| // 	}
 | |
| 
 | |
| // 	return ret
 | |
| // }
 | |
| 
 | |
| // func TestBuildInSha256(t *testing.T) {
 | |
| // 	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))
 | |
| // 	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)
 | |
| // 	}
 | |
| // }
 |