path: root/vm/vm_test.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)
    }
}