-
Notifications
You must be signed in to change notification settings - Fork 109
/
snapshot.go
326 lines (277 loc) · 7.7 KB
/
snapshot.go
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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
// Copyright Fuzamei Corp. 2018 All Rights Reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package state
import (
"sort"
"github.com/33cn/chain33/common/log/log15"
"github.com/33cn/chain33/types"
"github.com/33cn/plugin/plugin/dapp/evm/executor/vm/common"
evmtypes "github.com/33cn/plugin/plugin/dapp/evm/types"
)
// DataChange 数据状态变更接口
// 所有的数据状态变更事件实现此接口,并且封装各自的变更数据以及回滚动作
// 在调用合约时(具体的Tx执行时),会根据操作生成对应的变更对象并缓存下来
// 如果合约执行出错,会按生成顺序的倒序,依次调用变更对象的回滚接口进行数据回滚,并同步删除变更对象缓存
// 如果合约执行成功,会按生成顺序的郑旭,依次调用变更对象的数据和日志变更记录,回传给区块链
type DataChange interface {
revert(mdb *MemoryStateDB)
getData(mdb *MemoryStateDB) []*types.KeyValue
getLog(mdb *MemoryStateDB) []*types.ReceiptLog
}
// Snapshot 版本结构,包含版本号以及当前版本包含的变更对象在变更序列中的开始序号
type Snapshot struct {
id int
entries []DataChange
statedb *MemoryStateDB
}
// GetID 获取ID
func (ver *Snapshot) GetID() int {
return ver.id
}
// 回滚当前版本
func (ver *Snapshot) revert() bool {
if ver.entries == nil {
return true
}
for _, entry := range ver.entries {
entry.revert(ver.statedb)
}
return true
}
// 添加变更数据
func (ver *Snapshot) append(entry DataChange) {
ver.entries = append(ver.entries, entry)
}
// 获取当前版本变更数据
func (ver *Snapshot) getData() (kvSet []*types.KeyValue, logs []*types.ReceiptLog) {
// 获取中间的数据变更
dataMap := make(map[string]*types.KeyValue)
for _, entry := range ver.entries {
items := entry.getData(ver.statedb)
logEntry := entry.getLog(ver.statedb)
if logEntry != nil {
logs = append(logs, entry.getLog(ver.statedb)...)
}
// 执行去重操作
for _, kv := range items {
dataMap[string(kv.Key)] = kv
}
}
// 这里也可能会引起数据顺序不一致的问题,需要修改(目前看KV的顺序不会影响哈希计算,但代码最好保证顺序一致)
names := make([]string, 0, len(dataMap))
for name := range dataMap {
names = append(names, name)
}
sort.Strings(names)
for _, name := range names {
kvSet = append(kvSet, dataMap[name])
}
return kvSet, logs
}
type (
// 基础变更对象,用于封装默认操作
baseChange struct {
}
// 创建合约对象变更事件
createAccountChange struct {
baseChange
account string
}
// 自杀事件
suicideChange struct {
baseChange
account string
prev bool // whether account had already suicided
}
// nonce变更事件
nonceChange struct {
baseChange
account string
prev uint64
}
// 存储状态变更事件
storageChange struct {
baseChange
account string
key, prevalue common.Hash
}
// 合约代码状态变更事件
codeChange struct {
baseChange
account string
prevcode, prevhash []byte
}
// 合约ABI变更事件
abiChange struct {
baseChange
account string
prevabi string
}
// 返还金额变更事件
refundChange struct {
baseChange
prev uint64
}
// 转账事件
// 合约转账动作不执行回滚,失败后数据不会写入区块
transferChange struct {
baseChange
amount int64
data []*types.KeyValue
logs []*types.ReceiptLog
}
// 合约生成日志事件
addLogChange struct {
baseChange
txhash common.Hash
logs []*types.ReceiptLog
}
// 合约生成sha3事件
addPreimageChange struct {
baseChange
hash common.Hash
}
)
// 在baseChang中定义三个基本操作,子对象中只需要实现必要的操作
func (ch baseChange) revert(s *MemoryStateDB) {
}
func (ch baseChange) getData(s *MemoryStateDB) (kvset []*types.KeyValue) {
return nil
}
func (ch baseChange) getLog(s *MemoryStateDB) (logs []*types.ReceiptLog) {
return nil
}
// 创建账户对象的回滚,需要删除缓存中的账户和变更标记
func (ch createAccountChange) revert(s *MemoryStateDB) {
delete(s.accounts, ch.account)
}
// 创建账户对象的数据集
func (ch createAccountChange) getData(s *MemoryStateDB) (kvset []*types.KeyValue) {
acc := s.accounts[ch.account]
if acc != nil {
kvset = append(kvset, acc.GetDataKV()...)
kvset = append(kvset, acc.GetStateKV()...)
return kvset
}
return nil
}
func (ch suicideChange) revert(mdb *MemoryStateDB) {
// 如果已经自杀过了,不处理
if ch.prev {
return
}
acc := mdb.accounts[ch.account]
if acc != nil {
acc.State.Suicided = ch.prev
}
}
func (ch suicideChange) getData(mdb *MemoryStateDB) []*types.KeyValue {
// 如果已经自杀过了,不处理
if ch.prev {
return nil
}
acc := mdb.accounts[ch.account]
if acc != nil {
return acc.GetStateKV()
}
return nil
}
func (ch nonceChange) revert(mdb *MemoryStateDB) {
acc := mdb.accounts[ch.account]
if acc != nil {
acc.State.Nonce = ch.prev
}
}
func (ch nonceChange) getData(mdb *MemoryStateDB) []*types.KeyValue {
// nonce目前没有应用场景,而且每次调用都会变更,暂时先不写到状态数据库中
//acc := mdb.accounts[ch.account]
//if acc != nil {
// return acc.GetStateKV()
//}
return nil
}
func (ch codeChange) revert(mdb *MemoryStateDB) {
acc := mdb.accounts[ch.account]
if acc != nil {
acc.Data.Code = ch.prevcode
acc.Data.CodeHash = ch.prevhash
}
}
func (ch codeChange) getData(mdb *MemoryStateDB) (kvset []*types.KeyValue) {
acc := mdb.accounts[ch.account]
if acc != nil {
kvset = append(kvset, acc.GetDataKV()...)
kvset = append(kvset, acc.GetStateKV()...)
return kvset
}
return nil
}
func (ch abiChange) revert(mdb *MemoryStateDB) {
acc := mdb.accounts[ch.account]
if acc != nil {
acc.Data.Abi = ch.prevabi
}
}
func (ch abiChange) getData(mdb *MemoryStateDB) (kvset []*types.KeyValue) {
acc := mdb.accounts[ch.account]
if acc != nil {
kvset = append(kvset, acc.GetDataKV()...)
return kvset
}
return nil
}
func (ch storageChange) revert(mdb *MemoryStateDB) {
acc := mdb.accounts[ch.account]
if acc != nil {
acc.SetState(ch.key, ch.prevalue)
}
}
func (ch storageChange) getData(mdb *MemoryStateDB) []*types.KeyValue {
acc := mdb.accounts[ch.account]
if _, ok := mdb.stateDirty[ch.account]; ok && acc != nil {
return acc.GetStateKV()
}
return nil
}
func (ch storageChange) getLog(mdb *MemoryStateDB) []*types.ReceiptLog {
cfg := mdb.api.GetConfig()
if cfg.IsDappFork(mdb.blockHeight, "evm", evmtypes.ForkEVMState) {
acc := mdb.accounts[ch.account]
if acc != nil {
currentVal := acc.GetState(ch.key)
receipt := &evmtypes.EVMStateChangeItem{Key: getStateItemKey(ch.account, ch.key.Hex()), PreValue: ch.prevalue.Bytes(), CurrentValue: currentVal.Bytes()}
return []*types.ReceiptLog{{Ty: evmtypes.TyLogEVMStateChangeItem, Log: types.Encode(receipt)}}
}
}
return nil
}
func (ch refundChange) revert(mdb *MemoryStateDB) {
mdb.refund = ch.prev
}
func (ch addLogChange) revert(mdb *MemoryStateDB) {
logs := mdb.logs[ch.txhash]
if len(logs) == 1 {
delete(mdb.logs, ch.txhash)
} else {
mdb.logs[ch.txhash] = logs[:len(logs)-1]
}
mdb.logSize--
log15.Info("addLogChange::revert", "mdb.logSize", mdb.logSize)
}
func (ch addLogChange) getLog(mdb *MemoryStateDB) []*types.ReceiptLog {
cfg := mdb.api.GetConfig()
if !cfg.IsDappFork(mdb.blockHeight, "evm", evmtypes.ForkEVMState) {
return nil
}
return ch.logs
}
func (ch addPreimageChange) revert(mdb *MemoryStateDB) {
delete(mdb.preimages, ch.hash)
}
func (ch transferChange) getData(mdb *MemoryStateDB) []*types.KeyValue {
return ch.data
}
func (ch transferChange) getLog(mdb *MemoryStateDB) []*types.ReceiptLog {
return ch.logs
}