aboutsummaryrefslogblamecommitdiffstats
path: root/ethchain/vm_test.go
blob: f4a0197ea0b7411e18c9d469112f30d8c60f5187 (plain) (tree)
1
2
3
4
5
6
7
8
9

                
                 
             
                                            
                                   
                  
                 

                 
  






                                                  
                            
                         



                             

                            
                                                                                  
                              
                                           
                                      
 
                                         
                                                  
                         






                                         
                       
                           
                         
         
                                                                                              
 
                                        
                                                                                                                        
 
                                       
                                               
                               


                                                                                                                 
                                                                                
          




                                                                                                                                                                                                                       
   





                                                  
                                                        
                            
                            
                                               
                 
                      
                  
                                                                                                           
                                           
                                         
                                
                                                              
                                                                 
                                                
                 
 
 
                                                                            
 
                                                
                 
 
                             
                                                                                  









                                             


                                
 
                                                                                                                       
                                        
                                  
                                                                 
                                                       
                                          




                                                                                              
 
                                            




                                                                                                                 
          
                                               

                                       
                                                       
 
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, ""))

    script, err := mutan.Compile(strings.NewReader(`
        int32 a = 10
        int32 b = 20
        if a > b {
            int32 c = this.caller()
        }
        Exit()
    `), false)
    tx := NewContractCreationTx(ethutil.Big("0"), ethutil.Big("1000"), ethutil.Big("100"), script, nil)
    addr := tx.Hash()[12:]
    contract := MakeContract(tx, state)
    state.UpdateStateObject(contract)
    fmt.Printf("%x\n", addr)

    callerScript, err := mutan.Compile(strings.NewReader(`
        // Check if there's any cash in the initial store
        if this.store[1000] == 0 {
            this.store[1000] = 10^20
        }


        this.store[1001] = this.value() * 20
        this.store[this.origin()] = this.store[this.origin()] + 1000

        if this.store[1001] > 20 {
            this.store[1001] = 10^50
        }

        int8 ret = 0
        int8 arg = 10
        call(0xe6a12555fad1fb6eaaaed69001a87313d1fd7b54, 0, 100, arg, ret)

        big t
        for int8 i = 0; i < 10; i++ {
            t = i
        }

        if 10 > 20 {
            int8 shouldnt = 2
        } else {
            int8 should = 1
        }
    `), false)
    if err != nil {
        fmt.Println(err)
    }

    callerTx := NewContractCreationTx(ethutil.Big("0"), ethutil.Big("1000"), ethutil.Big("100"), callerScript, nil)

    // Contract addr as test address
    gas := big.NewInt(1000)
    gasPrice := big.NewInt(10)
    account := NewAccount(ContractAddr, big.NewInt(10000000))
    fmt.Println("account.Amount =", account.Amount)
    c := MakeContract(callerTx, state)
    e := account.ConvertGas(gas, gasPrice)
    if e != nil {
        fmt.Println(err)
    }
    fmt.Println("account.Amount =", account.Amount)
    callerClosure := NewClosure(account, c, c.script, state, gas, gasPrice, big.NewInt(0))

    vm := NewVm(state, nil, RuntimeVars{
        Origin:      account.Address(),
        BlockNumber: 1,
        PrevHash:    ethutil.FromHex("5e20a0453cecd065ea59c37ac63e079ee08998b6045136a8ce6635c7912ec0b6"),
        Coinbase:    ethutil.FromHex("2adc25665018aa1fe0e6bc666dac8fc2697ff9ba"),
        Time:        1,
        Diff:        big.NewInt(256),
    })
    _, e = callerClosure.Call(vm, nil, nil)
    if e != nil {
        fmt.Println("error", e)
    }
    fmt.Println("account.Amount =", account.Amount)
}