aboutsummaryrefslogtreecommitdiffstats
path: root/ethchain/vm_test.go
blob: 589f0bf4aa83a37032497750b4e0edfe5da52e13 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package ethchain

import (
    "bytes"
    "fmt"
    "github.com/ethereum/eth-go/ethdb"
    "github.com/ethereum/eth-go/ethutil"
    "github.com/obscuren/mutan"
    "math/big"
    "strings"
    "testing"
)

func TestRun3(t *testing.T) {
    ethutil.ReadConfig("")

    db, _ := ethdb.NewMemDatabase()
    state := NewState(ethutil.NewTrie(db, ""))

    script := Compile([]string{
        "PUSH", "300",
        "PUSH", "0",
        "MSTORE",

        "PUSH", "32",
        "CALLDATA",

        "PUSH", "64",
        "PUSH", "0",
        "RETURN",
    })
    tx := NewContractCreationTx(ethutil.Big("0"), ethutil.Big("1000"), script)
    addr := tx.Hash()[12:]
    contract := MakeContract(tx, state)
    state.UpdateContract(contract)

    callerScript := ethutil.Assemble(
        "PUSH", 1337, // Argument
        "PUSH", 65, // argument mem offset
        "MSTORE",
        "PUSH", 64, // ret size
        "PUSH", 0, // ret offset

        "PUSH", 32, // arg size
        "PUSH", 65, // arg offset
        "PUSH", 1000, /// Gas
        "PUSH", 0, /// value
        "PUSH", addr, // Sender
        "CALL",
        "PUSH", 64,
        "PUSH", 0,
        "RETURN",
    )
    callerTx := NewContractCreationTx(ethutil.Big("0"), ethutil.Big("1000"), callerScript)

    // Contract addr as test address
    account := NewAccount(ContractAddr, big.NewInt(10000000))
    callerClosure := NewClosure(account, MakeContract(callerTx, state), state, big.NewInt(1000000000), new(big.Int))

    vm := NewVm(state, RuntimeVars{
        origin:      account.Address(),
        blockNumber: 1,
        prevHash:    ethutil.FromHex("5e20a0453cecd065ea59c37ac63e079ee08998b6045136a8ce6635c7912ec0b6"),
        coinbase:    ethutil.FromHex("2adc25665018aa1fe0e6bc666dac8fc2697ff9ba"),
        time:        1,
        diff:        big.NewInt(256),
        // XXX Tx data? Could be just an argument to the closure instead
        txData: nil,
    })
    ret := callerClosure.Call(vm, nil)

    exp := []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 44, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 5, 57}
    if bytes.Compare(ret, exp) != 0 {
        t.Errorf("expected return value to be %v, got %v", exp, ret)
    }
}

func TestRun4(t *testing.T) {
    ethutil.ReadConfig("")

    db, _ := ethdb.NewMemDatabase()
    state := NewState(ethutil.NewTrie(db, ""))

    mutan.Compile(strings.NewReader(`
        a = 1337
        c = 1
        store[0] = 50
        d = store[0]
    `), false)

    asm, err := mutan.Compile(strings.NewReader(`
        a = 3 + 3
        store[1000] = a
        store[1000]
    `), false)
    if err != nil {
        fmt.Println(err)
    }
    asm = append(asm, "LOG")
    fmt.Println(asm)

    callerScript := ethutil.Assemble(asm...)
    callerTx := NewContractCreationTx(ethutil.Big("0"), ethutil.Big("1000"), callerScript)

    // Contract addr as test address
    account := NewAccount(ContractAddr, big.NewInt(10000000))
    callerClosure := NewClosure(account, MakeContract(callerTx, state), state, big.NewInt(1000000000), new(big.Int))

    vm := NewVm(state, RuntimeVars{
        origin:      account.Address(),
        blockNumber: 1,
        prevHash:    ethutil.FromHex("5e20a0453cecd065ea59c37ac63e079ee08998b6045136a8ce6635c7912ec0b6"),
        coinbase:    ethutil.FromHex("2adc25665018aa1fe0e6bc666dac8fc2697ff9ba"),
        time:        1,
        diff:        big.NewInt(256),
        // XXX Tx data? Could be just an argument to the closure instead
        txData: nil,
    })
    callerClosure.Call(vm, nil)
}