aboutsummaryrefslogtreecommitdiffstats
path: root/core/state/statedb_test.go
blob: 7930b620d4e601203cd0b4c5fdf016a083bcd71d (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
121
// Copyright 2016 The go-ethereum Authors
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.

package state

import (
    "math/big"
    "testing"

    "github.com/ethereum/go-ethereum/common"
    "github.com/ethereum/go-ethereum/crypto"
    "github.com/ethereum/go-ethereum/ethdb"
)

// Tests that updating a state trie does not leak any database writes prior to
// actually committing the state.
func TestUpdateLeaks(t *testing.T) {
    // Create an empty state database
    db, _ := ethdb.NewMemDatabase()
    state, _ := New(common.Hash{}, db)

    // Update it with some accounts
    for i := byte(0); i < 255; i++ {
        obj := state.GetOrNewStateObject(common.BytesToAddress([]byte{i}))
        obj.AddBalance(big.NewInt(int64(11 * i)))
        obj.SetNonce(uint64(42 * i))
        if i%2 == 0 {
            obj.SetState(common.BytesToHash([]byte{i, i, i}), common.BytesToHash([]byte{i, i, i, i}))
        }
        if i%3 == 0 {
            obj.SetCode(crypto.Keccak256Hash([]byte{i, i, i, i, i}), []byte{i, i, i, i, i})
        }
        state.UpdateStateObject(obj)
    }
    // Ensure that no data was leaked into the database
    for _, key := range db.Keys() {
        value, _ := db.Get(key)
        t.Errorf("State leaked into database: %x -> %x", key, value)
    }
}

// Tests that no intermediate state of an object is stored into the database,
// only the one right before the commit.
func TestIntermediateLeaks(t *testing.T) {
    // Create two state databases, one transitioning to the final state, the other final from the beginning
    transDb, _ := ethdb.NewMemDatabase()
    finalDb, _ := ethdb.NewMemDatabase()
    transState, _ := New(common.Hash{}, transDb)
    finalState, _ := New(common.Hash{}, finalDb)

    // Update the states with some objects
    for i := byte(0); i < 255; i++ {
        // Create a new state object with some data into the transition database
        obj := transState.GetOrNewStateObject(common.BytesToAddress([]byte{i}))
        obj.SetBalance(big.NewInt(int64(11 * i)))
        obj.SetNonce(uint64(42 * i))
        if i%2 == 0 {
            obj.SetState(common.BytesToHash([]byte{i, i, i, 0}), common.BytesToHash([]byte{i, i, i, i, 0}))
        }
        if i%3 == 0 {
            obj.SetCode(crypto.Keccak256Hash([]byte{i, i, i, i, i, 0}), []byte{i, i, i, i, i, 0})
        }
        transState.UpdateStateObject(obj)

        // Overwrite all the data with new values in the transition database
        obj.SetBalance(big.NewInt(int64(11*i + 1)))
        obj.SetNonce(uint64(42*i + 1))
        if i%2 == 0 {
            obj.SetState(common.BytesToHash([]byte{i, i, i, 0}), common.Hash{})
            obj.SetState(common.BytesToHash([]byte{i, i, i, 1}), common.BytesToHash([]byte{i, i, i, i, 1}))
        }
        if i%3 == 0 {
            obj.SetCode(crypto.Keccak256Hash([]byte{i, i, i, i, i, 1}), []byte{i, i, i, i, i, 1})
        }
        transState.UpdateStateObject(obj)

        // Create the final state object directly in the final database
        obj = finalState.GetOrNewStateObject(common.BytesToAddress([]byte{i}))
        obj.SetBalance(big.NewInt(int64(11*i + 1)))
        obj.SetNonce(uint64(42*i + 1))
        if i%2 == 0 {
            obj.SetState(common.BytesToHash([]byte{i, i, i, 1}), common.BytesToHash([]byte{i, i, i, i, 1}))
        }
        if i%3 == 0 {
            obj.SetCode(crypto.Keccak256Hash([]byte{i, i, i, i, i, 1}), []byte{i, i, i, i, i, 1})
        }
        finalState.UpdateStateObject(obj)
    }
    if _, err := transState.Commit(); err != nil {
        t.Fatalf("failed to commit transition state: %v", err)
    }
    if _, err := finalState.Commit(); err != nil {
        t.Fatalf("failed to commit final state: %v", err)
    }
    // Cross check the databases to ensure they are the same
    for _, key := range finalDb.Keys() {
        if _, err := transDb.Get(key); err != nil {
            val, _ := finalDb.Get(key)
            t.Errorf("entry missing from the transition database: %x -> %x", key, val)
        }
    }
    for _, key := range transDb.Keys() {
        if _, err := finalDb.Get(key); err != nil {
            val, _ := transDb.Get(key)
            t.Errorf("extra entry in the transition database: %x -> %x", key, val)
        }
    }
}