github.com/DxChainNetwork/dxc@v0.8.1-0.20220824085222-1162e304b6e7/core/state/statedb.go (about) 1 // Copyright 2014 The go-ethereum Authors 2 // This file is part of the go-ethereum library. 3 // 4 // The go-ethereum library is free software: you can redistribute it and/or modify 5 // it under the terms of the GNU Lesser General Public License as published by 6 // the Free Software Foundation, either version 3 of the License, or 7 // (at your option) any later version. 8 // 9 // The go-ethereum library is distributed in the hope that it will be useful, 10 // but WITHOUT ANY WARRANTY; without even the implied warranty of 11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 // GNU Lesser General Public License for more details. 13 // 14 // You should have received a copy of the GNU Lesser General Public License 15 // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>. 16 17 // Package state provides a caching layer atop the Ethereum state trie. 18 package state 19 20 import ( 21 "errors" 22 "fmt" 23 "math/big" 24 "sort" 25 "time" 26 27 "github.com/DxChainNetwork/dxc/common" 28 "github.com/DxChainNetwork/dxc/core/rawdb" 29 "github.com/DxChainNetwork/dxc/core/state/snapshot" 30 "github.com/DxChainNetwork/dxc/core/types" 31 "github.com/DxChainNetwork/dxc/crypto" 32 "github.com/DxChainNetwork/dxc/log" 33 "github.com/DxChainNetwork/dxc/metrics" 34 "github.com/DxChainNetwork/dxc/rlp" 35 "github.com/DxChainNetwork/dxc/trie" 36 ) 37 38 type revision struct { 39 id int 40 journalIndex int 41 } 42 43 var ( 44 // emptyRoot is the known root hash of an empty trie. 45 emptyRoot = common.HexToHash("56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421") 46 ) 47 48 type proofList [][]byte 49 50 func (n *proofList) Put(key []byte, value []byte) error { 51 *n = append(*n, value) 52 return nil 53 } 54 55 func (n *proofList) Delete(key []byte) error { 56 panic("not supported") 57 } 58 59 // StateDB structs within the ethereum protocol are used to store anything 60 // within the merkle trie. StateDBs take care of caching and storing 61 // nested states. It's the general query interface to retrieve: 62 // * Contracts 63 // * Accounts 64 type StateDB struct { 65 db Database 66 prefetcher *triePrefetcher 67 originalRoot common.Hash // The pre-state root, before any changes were made 68 trie Trie 69 hasher crypto.KeccakState 70 71 snaps *snapshot.Tree 72 snap snapshot.Snapshot 73 snapDestructs map[common.Hash]struct{} 74 snapAccounts map[common.Hash][]byte 75 snapStorage map[common.Hash]map[common.Hash][]byte 76 77 // This map holds 'live' objects, which will get modified while processing a state transition. 78 stateObjects map[common.Address]*stateObject 79 stateObjectsPending map[common.Address]struct{} // State objects finalized but not yet written to the trie 80 stateObjectsDirty map[common.Address]struct{} // State objects modified in the current execution 81 82 // DB error. 83 // State objects are used by the consensus core and VM which are 84 // unable to deal with database-level errors. Any error that occurs 85 // during a database read is memoized here and will eventually be returned 86 // by StateDB.Commit. 87 dbErr error 88 89 // The refund counter, also used by state transitioning. 90 refund uint64 91 92 thash common.Hash 93 txIndex int 94 logs map[common.Hash][]*types.Log 95 logSize uint 96 97 preimages map[common.Hash][]byte 98 99 // Per-transaction access list 100 accessList *accessList 101 102 // Journal of state modifications. This is the backbone of 103 // Snapshot and RevertToSnapshot. 104 journal *journal 105 validRevisions []revision 106 nextRevisionId int 107 108 // Measurements gathered during execution for debugging purposes 109 AccountReads time.Duration 110 AccountHashes time.Duration 111 AccountUpdates time.Duration 112 AccountCommits time.Duration 113 StorageReads time.Duration 114 StorageHashes time.Duration 115 StorageUpdates time.Duration 116 StorageCommits time.Duration 117 SnapshotAccountReads time.Duration 118 SnapshotStorageReads time.Duration 119 SnapshotCommits time.Duration 120 } 121 122 // New creates a new state from a given trie. 123 func New(root common.Hash, db Database, snaps *snapshot.Tree) (*StateDB, error) { 124 tr, err := db.OpenTrie(root) 125 if err != nil { 126 return nil, err 127 } 128 sdb := &StateDB{ 129 db: db, 130 trie: tr, 131 originalRoot: root, 132 snaps: snaps, 133 stateObjects: make(map[common.Address]*stateObject), 134 stateObjectsPending: make(map[common.Address]struct{}), 135 stateObjectsDirty: make(map[common.Address]struct{}), 136 logs: make(map[common.Hash][]*types.Log), 137 preimages: make(map[common.Hash][]byte), 138 journal: newJournal(), 139 accessList: newAccessList(), 140 hasher: crypto.NewKeccakState(), 141 } 142 if sdb.snaps != nil { 143 if sdb.snap = sdb.snaps.Snapshot(root); sdb.snap != nil { 144 sdb.snapDestructs = make(map[common.Hash]struct{}) 145 sdb.snapAccounts = make(map[common.Hash][]byte) 146 sdb.snapStorage = make(map[common.Hash]map[common.Hash][]byte) 147 } 148 } 149 return sdb, nil 150 } 151 152 // StartPrefetcher initializes a new trie prefetcher to pull in nodes from the 153 // state trie concurrently while the state is mutated so that when we reach the 154 // commit phase, most of the needed data is already hot. 155 func (s *StateDB) StartPrefetcher(namespace string) { 156 if s.prefetcher != nil { 157 s.prefetcher.close() 158 s.prefetcher = nil 159 } 160 if s.snap != nil { 161 s.prefetcher = newTriePrefetcher(s.db, s.originalRoot, namespace) 162 } 163 } 164 165 // StopPrefetcher terminates a running prefetcher and reports any leftover stats 166 // from the gathered metrics. 167 func (s *StateDB) StopPrefetcher() { 168 if s.prefetcher != nil { 169 s.prefetcher.close() 170 s.prefetcher = nil 171 } 172 } 173 174 // setError remembers the first non-nil error it is called with. 175 func (s *StateDB) setError(err error) { 176 if s.dbErr == nil { 177 s.dbErr = err 178 } 179 } 180 181 func (s *StateDB) Error() error { 182 return s.dbErr 183 } 184 185 func (s *StateDB) AddLog(log *types.Log) { 186 s.journal.append(addLogChange{txhash: s.thash}) 187 188 log.TxHash = s.thash 189 log.TxIndex = uint(s.txIndex) 190 log.Index = s.logSize 191 s.logs[s.thash] = append(s.logs[s.thash], log) 192 s.logSize++ 193 } 194 195 func (s *StateDB) GetLogs(hash common.Hash, blockHash common.Hash) []*types.Log { 196 logs := s.logs[hash] 197 for _, l := range logs { 198 l.BlockHash = blockHash 199 } 200 return logs 201 } 202 203 func (s *StateDB) Logs() []*types.Log { 204 var logs []*types.Log 205 for _, lgs := range s.logs { 206 logs = append(logs, lgs...) 207 } 208 return logs 209 } 210 211 // AddPreimage records a SHA3 preimage seen by the VM. 212 func (s *StateDB) AddPreimage(hash common.Hash, preimage []byte) { 213 if _, ok := s.preimages[hash]; !ok { 214 s.journal.append(addPreimageChange{hash: hash}) 215 pi := make([]byte, len(preimage)) 216 copy(pi, preimage) 217 s.preimages[hash] = pi 218 } 219 } 220 221 // Preimages returns a list of SHA3 preimages that have been submitted. 222 func (s *StateDB) Preimages() map[common.Hash][]byte { 223 return s.preimages 224 } 225 226 // AddRefund adds gas to the refund counter 227 func (s *StateDB) AddRefund(gas uint64) { 228 s.journal.append(refundChange{prev: s.refund}) 229 s.refund += gas 230 } 231 232 // SubRefund removes gas from the refund counter. 233 // This method will panic if the refund counter goes below zero 234 func (s *StateDB) SubRefund(gas uint64) { 235 s.journal.append(refundChange{prev: s.refund}) 236 if gas > s.refund { 237 panic(fmt.Sprintf("Refund counter below zero (gas: %d > refund: %d)", gas, s.refund)) 238 } 239 s.refund -= gas 240 } 241 242 // Exist reports whether the given account address exists in the state. 243 // Notably this also returns true for suicided accounts. 244 func (s *StateDB) Exist(addr common.Address) bool { 245 return s.getStateObject(addr) != nil 246 } 247 248 // Empty returns whether the state object is either non-existent 249 // or empty according to the EIP161 specification (balance = nonce = code = 0) 250 func (s *StateDB) Empty(addr common.Address) bool { 251 so := s.getStateObject(addr) 252 return so == nil || so.empty() 253 } 254 255 // GetBalance retrieves the balance from the given address or 0 if object not found 256 func (s *StateDB) GetBalance(addr common.Address) *big.Int { 257 stateObject := s.getStateObject(addr) 258 if stateObject != nil { 259 return stateObject.Balance() 260 } 261 return common.Big0 262 } 263 264 func (s *StateDB) GetNonce(addr common.Address) uint64 { 265 stateObject := s.getStateObject(addr) 266 if stateObject != nil { 267 return stateObject.Nonce() 268 } 269 270 return 0 271 } 272 273 // TxIndex returns the current transaction index set by Prepare. 274 func (s *StateDB) TxIndex() int { 275 return s.txIndex 276 } 277 278 func (s *StateDB) GetCode(addr common.Address) []byte { 279 stateObject := s.getStateObject(addr) 280 if stateObject != nil { 281 return stateObject.Code(s.db) 282 } 283 return nil 284 } 285 286 func (s *StateDB) GetCodeSize(addr common.Address) int { 287 stateObject := s.getStateObject(addr) 288 if stateObject != nil { 289 return stateObject.CodeSize(s.db) 290 } 291 return 0 292 } 293 294 func (s *StateDB) GetCodeHash(addr common.Address) common.Hash { 295 stateObject := s.getStateObject(addr) 296 if stateObject == nil { 297 return common.Hash{} 298 } 299 return common.BytesToHash(stateObject.CodeHash()) 300 } 301 302 // GetState retrieves a value from the given account's storage trie. 303 func (s *StateDB) GetState(addr common.Address, hash common.Hash) common.Hash { 304 stateObject := s.getStateObject(addr) 305 if stateObject != nil { 306 return stateObject.GetState(s.db, hash) 307 } 308 return common.Hash{} 309 } 310 311 // GetProof returns the Merkle proof for a given account. 312 func (s *StateDB) GetProof(addr common.Address) ([][]byte, error) { 313 return s.GetProofByHash(crypto.Keccak256Hash(addr.Bytes())) 314 } 315 316 // GetProofByHash returns the Merkle proof for a given account. 317 func (s *StateDB) GetProofByHash(addrHash common.Hash) ([][]byte, error) { 318 var proof proofList 319 err := s.trie.Prove(addrHash[:], 0, &proof) 320 return proof, err 321 } 322 323 // GetStorageProof returns the Merkle proof for given storage slot. 324 func (s *StateDB) GetStorageProof(a common.Address, key common.Hash) ([][]byte, error) { 325 var proof proofList 326 trie := s.StorageTrie(a) 327 if trie == nil { 328 return proof, errors.New("storage trie for requested address does not exist") 329 } 330 err := trie.Prove(crypto.Keccak256(key.Bytes()), 0, &proof) 331 return proof, err 332 } 333 334 // GetCommittedState retrieves a value from the given account's committed storage trie. 335 func (s *StateDB) GetCommittedState(addr common.Address, hash common.Hash) common.Hash { 336 stateObject := s.getStateObject(addr) 337 if stateObject != nil { 338 return stateObject.GetCommittedState(s.db, hash) 339 } 340 return common.Hash{} 341 } 342 343 // Database retrieves the low level database supporting the lower level trie ops. 344 func (s *StateDB) Database() Database { 345 return s.db 346 } 347 348 // StorageTrie returns the storage trie of an account. 349 // The return value is a copy and is nil for non-existent accounts. 350 func (s *StateDB) StorageTrie(addr common.Address) Trie { 351 stateObject := s.getStateObject(addr) 352 if stateObject == nil { 353 return nil 354 } 355 cpy := stateObject.deepCopy(s) 356 cpy.updateTrie(s.db) 357 return cpy.getTrie(s.db) 358 } 359 360 func (s *StateDB) HasSuicided(addr common.Address) bool { 361 stateObject := s.getStateObject(addr) 362 if stateObject != nil { 363 return stateObject.suicided 364 } 365 return false 366 } 367 368 /* 369 * SETTERS 370 */ 371 372 // AddBalance adds amount to the account associated with addr. 373 func (s *StateDB) AddBalance(addr common.Address, amount *big.Int) { 374 stateObject := s.GetOrNewStateObject(addr) 375 if stateObject != nil { 376 stateObject.AddBalance(amount) 377 } 378 } 379 380 // SubBalance subtracts amount from the account associated with addr. 381 func (s *StateDB) SubBalance(addr common.Address, amount *big.Int) { 382 stateObject := s.GetOrNewStateObject(addr) 383 if stateObject != nil { 384 stateObject.SubBalance(amount) 385 } 386 } 387 388 func (s *StateDB) SetBalance(addr common.Address, amount *big.Int) { 389 stateObject := s.GetOrNewStateObject(addr) 390 if stateObject != nil { 391 stateObject.SetBalance(amount) 392 } 393 } 394 395 func (s *StateDB) SetNonce(addr common.Address, nonce uint64) { 396 stateObject := s.GetOrNewStateObject(addr) 397 if stateObject != nil { 398 stateObject.SetNonce(nonce) 399 } 400 } 401 402 func (s *StateDB) SetCode(addr common.Address, code []byte) { 403 stateObject := s.GetOrNewStateObject(addr) 404 if stateObject != nil { 405 stateObject.SetCode(crypto.Keccak256Hash(code), code) 406 } 407 } 408 409 func (s *StateDB) SetState(addr common.Address, key, value common.Hash) { 410 stateObject := s.GetOrNewStateObject(addr) 411 if stateObject != nil { 412 stateObject.SetState(s.db, key, value) 413 } 414 } 415 416 // SetStorage replaces the entire storage for the specified account with given 417 // storage. This function should only be used for debugging. 418 func (s *StateDB) SetStorage(addr common.Address, storage map[common.Hash]common.Hash) { 419 stateObject := s.GetOrNewStateObject(addr) 420 if stateObject != nil { 421 stateObject.SetStorage(storage) 422 } 423 } 424 425 // Suicide marks the given account as suicided. 426 // This clears the account balance. 427 // 428 // The account's state object is still available until the state is committed, 429 // getStateObject will return a non-nil account after Suicide. 430 func (s *StateDB) Suicide(addr common.Address) bool { 431 stateObject := s.getStateObject(addr) 432 if stateObject == nil { 433 return false 434 } 435 s.journal.append(suicideChange{ 436 account: &addr, 437 prev: stateObject.suicided, 438 prevbalance: new(big.Int).Set(stateObject.Balance()), 439 }) 440 stateObject.markSuicided() 441 stateObject.data.Balance = new(big.Int) 442 443 return true 444 } 445 446 // Erase sets the code/storage-root to empty for the given account. 447 // This's a governance action. 448 // 449 // The account is still available, and with it's balance unchanged. 450 func (s *StateDB) Erase(addr common.Address) bool { 451 stateObject := s.getStateObject(addr) 452 if stateObject == nil { 453 return false 454 } 455 stateObject.erase() 456 457 return true 458 } 459 460 // 461 // Setting, updating & deleting state object methods. 462 // 463 464 // updateStateObject writes the given object to the trie. 465 func (s *StateDB) updateStateObject(obj *stateObject) { 466 // Track the amount of time wasted on updating the account from the trie 467 if metrics.EnabledExpensive { 468 defer func(start time.Time) { s.AccountUpdates += time.Since(start) }(time.Now()) 469 } 470 // Encode the account and update the account trie 471 addr := obj.Address() 472 473 data, err := rlp.EncodeToBytes(obj) 474 if err != nil { 475 panic(fmt.Errorf("can't encode object at %x: %v", addr[:], err)) 476 } 477 if err = s.trie.TryUpdate(addr[:], data); err != nil { 478 s.setError(fmt.Errorf("updateStateObject (%x) error: %v", addr[:], err)) 479 } 480 481 // If state snapshotting is active, cache the data til commit. Note, this 482 // update mechanism is not symmetric to the deletion, because whereas it is 483 // enough to track account updates at commit time, deletions need tracking 484 // at transaction boundary level to ensure we capture state clearing. 485 if s.snap != nil { 486 s.snapAccounts[obj.addrHash] = snapshot.SlimAccountRLP(obj.data.Nonce, obj.data.Balance, obj.data.Root, obj.data.CodeHash) 487 } 488 } 489 490 // deleteStateObject removes the given object from the state trie. 491 func (s *StateDB) deleteStateObject(obj *stateObject) { 492 // Track the amount of time wasted on deleting the account from the trie 493 if metrics.EnabledExpensive { 494 defer func(start time.Time) { s.AccountUpdates += time.Since(start) }(time.Now()) 495 } 496 // Delete the account from the trie 497 addr := obj.Address() 498 if err := s.trie.TryDelete(addr[:]); err != nil { 499 s.setError(fmt.Errorf("deleteStateObject (%x) error: %v", addr[:], err)) 500 } 501 } 502 503 // getStateObject retrieves a state object given by the address, returning nil if 504 // the object is not found or was deleted in this execution context. If you need 505 // to differentiate between non-existent/just-deleted, use getDeletedStateObject. 506 func (s *StateDB) getStateObject(addr common.Address) *stateObject { 507 if obj := s.getDeletedStateObject(addr); obj != nil && !obj.deleted { 508 return obj 509 } 510 return nil 511 } 512 513 // getDeletedStateObject is similar to getStateObject, but instead of returning 514 // nil for a deleted state object, it returns the actual object with the deleted 515 // flag set. This is needed by the state journal to revert to the correct s- 516 // destructed object instead of wiping all knowledge about the state object. 517 func (s *StateDB) getDeletedStateObject(addr common.Address) *stateObject { 518 // Prefer live objects if any is available 519 if obj := s.stateObjects[addr]; obj != nil { 520 return obj 521 } 522 // If no live objects are available, attempt to use snapshots 523 var ( 524 data *Account 525 err error 526 ) 527 if s.snap != nil { 528 if metrics.EnabledExpensive { 529 defer func(start time.Time) { s.SnapshotAccountReads += time.Since(start) }(time.Now()) 530 } 531 var acc *snapshot.Account 532 if acc, err = s.snap.Account(crypto.HashData(s.hasher, addr.Bytes())); err == nil { 533 if acc == nil { 534 return nil 535 } 536 data = &Account{ 537 Nonce: acc.Nonce, 538 Balance: acc.Balance, 539 CodeHash: acc.CodeHash, 540 Root: common.BytesToHash(acc.Root), 541 } 542 if len(data.CodeHash) == 0 { 543 data.CodeHash = emptyCodeHash 544 } 545 if data.Root == (common.Hash{}) { 546 data.Root = emptyRoot 547 } 548 } 549 } 550 // If snapshot unavailable or reading from it failed, load from the database 551 if s.snap == nil || err != nil { 552 if metrics.EnabledExpensive { 553 defer func(start time.Time) { s.AccountReads += time.Since(start) }(time.Now()) 554 } 555 enc, err := s.trie.TryGet(addr.Bytes()) 556 if err != nil { 557 s.setError(fmt.Errorf("getDeleteStateObject (%x) error: %v", addr.Bytes(), err)) 558 return nil 559 } 560 if len(enc) == 0 { 561 return nil 562 } 563 data = new(Account) 564 if err := rlp.DecodeBytes(enc, data); err != nil { 565 log.Error("Failed to decode state object", "addr", addr, "err", err) 566 return nil 567 } 568 } 569 // Insert into the live set 570 obj := newObject(s, addr, *data) 571 s.setStateObject(obj) 572 return obj 573 } 574 575 func (s *StateDB) setStateObject(object *stateObject) { 576 s.stateObjects[object.Address()] = object 577 } 578 579 // GetOrNewStateObject retrieves a state object or create a new state object if nil. 580 func (s *StateDB) GetOrNewStateObject(addr common.Address) *stateObject { 581 stateObject := s.getStateObject(addr) 582 if stateObject == nil { 583 stateObject, _ = s.createObject(addr) 584 } 585 return stateObject 586 } 587 588 // createObject creates a new state object. If there is an existing account with 589 // the given address, it is overwritten and returned as the second return value. 590 func (s *StateDB) createObject(addr common.Address) (newobj, prev *stateObject) { 591 prev = s.getDeletedStateObject(addr) // Note, prev might have been deleted, we need that! 592 593 var prevdestruct bool 594 if s.snap != nil && prev != nil { 595 _, prevdestruct = s.snapDestructs[prev.addrHash] 596 if !prevdestruct { 597 s.snapDestructs[prev.addrHash] = struct{}{} 598 } 599 } 600 newobj = newObject(s, addr, Account{}) 601 if prev == nil { 602 s.journal.append(createObjectChange{account: &addr}) 603 } else { 604 s.journal.append(resetObjectChange{prev: prev, prevdestruct: prevdestruct}) 605 } 606 s.setStateObject(newobj) 607 if prev != nil && !prev.deleted { 608 return newobj, prev 609 } 610 return newobj, nil 611 } 612 613 // CreateAccount explicitly creates a state object. If a state object with the address 614 // already exists the balance is carried over to the new account. 615 // 616 // CreateAccount is called during the EVM CREATE operation. The situation might arise that 617 // a contract does the following: 618 // 619 // 1. sends funds to sha(account ++ (nonce + 1)) 620 // 2. tx_create(sha(account ++ nonce)) (note that this gets the address of 1) 621 // 622 // Carrying over the balance ensures that Ether doesn't disappear. 623 func (s *StateDB) CreateAccount(addr common.Address) { 624 newObj, prev := s.createObject(addr) 625 if prev != nil { 626 newObj.setBalance(prev.data.Balance) 627 } 628 } 629 630 func (db *StateDB) ForEachStorage(addr common.Address, cb func(key, value common.Hash) bool) error { 631 so := db.getStateObject(addr) 632 if so == nil { 633 return nil 634 } 635 it := trie.NewIterator(so.getTrie(db.db).NodeIterator(nil)) 636 637 for it.Next() { 638 key := common.BytesToHash(db.trie.GetKey(it.Key)) 639 if value, dirty := so.dirtyStorage[key]; dirty { 640 if !cb(key, value) { 641 return nil 642 } 643 continue 644 } 645 646 if len(it.Value) > 0 { 647 _, content, _, err := rlp.Split(it.Value) 648 if err != nil { 649 return err 650 } 651 if !cb(key, common.BytesToHash(content)) { 652 return nil 653 } 654 } 655 } 656 return nil 657 } 658 659 // Copy creates a deep, independent copy of the state. 660 // Snapshots of the copied state cannot be applied to the copy. 661 func (s *StateDB) Copy() *StateDB { 662 // Copy all the basic fields, initialize the memory ones 663 state := &StateDB{ 664 db: s.db, 665 trie: s.db.CopyTrie(s.trie), 666 stateObjects: make(map[common.Address]*stateObject, len(s.journal.dirties)), 667 stateObjectsPending: make(map[common.Address]struct{}, len(s.stateObjectsPending)), 668 stateObjectsDirty: make(map[common.Address]struct{}, len(s.journal.dirties)), 669 refund: s.refund, 670 logs: make(map[common.Hash][]*types.Log, len(s.logs)), 671 logSize: s.logSize, 672 preimages: make(map[common.Hash][]byte, len(s.preimages)), 673 journal: newJournal(), 674 hasher: crypto.NewKeccakState(), 675 } 676 // Copy the dirty states, logs, and preimages 677 for addr := range s.journal.dirties { 678 // As documented [here](https://github.com/DxChainNetwork/dxc/pull/16485#issuecomment-380438527), 679 // and in the Finalise-method, there is a case where an object is in the journal but not 680 // in the stateObjects: OOG after touch on ripeMD prior to Byzantium. Thus, we need to check for 681 // nil 682 if object, exist := s.stateObjects[addr]; exist { 683 // Even though the original object is dirty, we are not copying the journal, 684 // so we need to make sure that anyside effect the journal would have caused 685 // during a commit (or similar op) is already applied to the copy. 686 state.stateObjects[addr] = object.deepCopy(state) 687 688 state.stateObjectsDirty[addr] = struct{}{} // Mark the copy dirty to force internal (code/state) commits 689 state.stateObjectsPending[addr] = struct{}{} // Mark the copy pending to force external (account) commits 690 } 691 } 692 // Above, we don't copy the actual journal. This means that if the copy is copied, the 693 // loop above will be a no-op, since the copy's journal is empty. 694 // Thus, here we iterate over stateObjects, to enable copies of copies 695 for addr := range s.stateObjectsPending { 696 if _, exist := state.stateObjects[addr]; !exist { 697 state.stateObjects[addr] = s.stateObjects[addr].deepCopy(state) 698 } 699 state.stateObjectsPending[addr] = struct{}{} 700 } 701 for addr := range s.stateObjectsDirty { 702 if _, exist := state.stateObjects[addr]; !exist { 703 state.stateObjects[addr] = s.stateObjects[addr].deepCopy(state) 704 } 705 state.stateObjectsDirty[addr] = struct{}{} 706 } 707 for hash, logs := range s.logs { 708 cpy := make([]*types.Log, len(logs)) 709 for i, l := range logs { 710 cpy[i] = new(types.Log) 711 *cpy[i] = *l 712 } 713 state.logs[hash] = cpy 714 } 715 for hash, preimage := range s.preimages { 716 state.preimages[hash] = preimage 717 } 718 // Do we need to copy the access list? In practice: No. At the start of a 719 // transaction, the access list is empty. In practice, we only ever copy state 720 // _between_ transactions/blocks, never in the middle of a transaction. 721 // However, it doesn't cost us much to copy an empty list, so we do it anyway 722 // to not blow up if we ever decide copy it in the middle of a transaction 723 state.accessList = s.accessList.Copy() 724 725 // If there's a prefetcher running, make an inactive copy of it that can 726 // only access data but does not actively preload (since the user will not 727 // know that they need to explicitly terminate an active copy). 728 if s.prefetcher != nil { 729 state.prefetcher = s.prefetcher.copy() 730 } 731 if s.snaps != nil { 732 // In order for the miner to be able to use and make additions 733 // to the snapshot tree, we need to copy that aswell. 734 // Otherwise, any block mined by ourselves will cause gaps in the tree, 735 // and force the miner to operate trie-backed only 736 state.snaps = s.snaps 737 state.snap = s.snap 738 // deep copy needed 739 state.snapDestructs = make(map[common.Hash]struct{}) 740 for k, v := range s.snapDestructs { 741 state.snapDestructs[k] = v 742 } 743 state.snapAccounts = make(map[common.Hash][]byte) 744 for k, v := range s.snapAccounts { 745 state.snapAccounts[k] = v 746 } 747 state.snapStorage = make(map[common.Hash]map[common.Hash][]byte) 748 for k, v := range s.snapStorage { 749 temp := make(map[common.Hash][]byte) 750 for kk, vv := range v { 751 temp[kk] = vv 752 } 753 state.snapStorage[k] = temp 754 } 755 } 756 return state 757 } 758 759 // Snapshot returns an identifier for the current revision of the state. 760 func (s *StateDB) Snapshot() int { 761 id := s.nextRevisionId 762 s.nextRevisionId++ 763 s.validRevisions = append(s.validRevisions, revision{id, s.journal.length()}) 764 return id 765 } 766 767 // RevertToSnapshot reverts all state changes made since the given revision. 768 func (s *StateDB) RevertToSnapshot(revid int) { 769 // Find the snapshot in the stack of valid snapshots. 770 idx := sort.Search(len(s.validRevisions), func(i int) bool { 771 return s.validRevisions[i].id >= revid 772 }) 773 if idx == len(s.validRevisions) || s.validRevisions[idx].id != revid { 774 panic(fmt.Errorf("revision id %v cannot be reverted", revid)) 775 } 776 snapshot := s.validRevisions[idx].journalIndex 777 778 // Replay the journal to undo changes and remove invalidated snapshots 779 s.journal.revert(s, snapshot) 780 s.validRevisions = s.validRevisions[:idx] 781 } 782 783 // GetRefund returns the current value of the refund counter. 784 func (s *StateDB) GetRefund() uint64 { 785 return s.refund 786 } 787 788 // Finalise finalises the state by removing the s destructed objects and clears 789 // the journal as well as the refunds. Finalise, however, will not push any updates 790 // into the tries just yet. Only IntermediateRoot or Commit will do that. 791 func (s *StateDB) Finalise(deleteEmptyObjects bool) { 792 addressesToPrefetch := make([][]byte, 0, len(s.journal.dirties)) 793 for addr := range s.journal.dirties { 794 obj, exist := s.stateObjects[addr] 795 if !exist { 796 // ripeMD is 'touched' at block 1714175, in tx 0x1237f737031e40bcde4a8b7e717b2d15e3ecadfe49bb1bbc71ee9deb09c6fcf2 797 // That tx goes out of gas, and although the notion of 'touched' does not exist there, the 798 // touch-event will still be recorded in the journal. Since ripeMD is a special snowflake, 799 // it will persist in the journal even though the journal is reverted. In this special circumstance, 800 // it may exist in `s.journal.dirties` but not in `s.stateObjects`. 801 // Thus, we can safely ignore it here 802 continue 803 } 804 if obj.suicided || (deleteEmptyObjects && obj.empty()) { 805 obj.deleted = true 806 807 // If state snapshotting is active, also mark the destruction there. 808 // Note, we can't do this only at the end of a block because multiple 809 // transactions within the same block might self destruct and then 810 // ressurrect an account; but the snapshotter needs both events. 811 if s.snap != nil { 812 s.snapDestructs[obj.addrHash] = struct{}{} // We need to maintain account deletions explicitly (will remain set indefinitely) 813 delete(s.snapAccounts, obj.addrHash) // Clear out any previously updated account data (may be recreated via a ressurrect) 814 delete(s.snapStorage, obj.addrHash) // Clear out any previously updated storage data (may be recreated via a ressurrect) 815 } 816 } else { 817 obj.finalise(true) // Prefetch slots in the background 818 } 819 s.stateObjectsPending[addr] = struct{}{} 820 s.stateObjectsDirty[addr] = struct{}{} 821 822 // At this point, also ship the address off to the precacher. The precacher 823 // will start loading tries, and when the change is eventually committed, 824 // the commit-phase will be a lot faster 825 addressesToPrefetch = append(addressesToPrefetch, common.CopyBytes(addr[:])) // Copy needed for closure 826 } 827 if s.prefetcher != nil && len(addressesToPrefetch) > 0 { 828 s.prefetcher.prefetch(s.originalRoot, addressesToPrefetch) 829 } 830 // Invalidate journal because reverting across transactions is not allowed. 831 s.clearJournalAndRefund() 832 } 833 834 // IntermediateRoot computes the current root hash of the state trie. 835 // It is called in between transactions to get the root hash that 836 // goes into transaction receipts. 837 func (s *StateDB) IntermediateRoot(deleteEmptyObjects bool) common.Hash { 838 // Finalise all the dirty storage states and write them into the tries 839 s.Finalise(deleteEmptyObjects) 840 841 // If there was a trie prefetcher operating, it gets aborted and irrevocably 842 // modified after we start retrieving tries. Remove it from the statedb after 843 // this round of use. 844 // 845 // This is weird pre-byzantium since the first tx runs with a prefetcher and 846 // the remainder without, but pre-byzantium even the initial prefetcher is 847 // useless, so no sleep lost. 848 prefetcher := s.prefetcher 849 if s.prefetcher != nil { 850 defer func() { 851 s.prefetcher.close() 852 s.prefetcher = nil 853 }() 854 } 855 // Although naively it makes sense to retrieve the account trie and then do 856 // the contract storage and account updates sequentially, that short circuits 857 // the account prefetcher. Instead, let's process all the storage updates 858 // first, giving the account prefeches just a few more milliseconds of time 859 // to pull useful data from disk. 860 for addr := range s.stateObjectsPending { 861 if obj := s.stateObjects[addr]; !obj.deleted { 862 obj.updateRoot(s.db) 863 } 864 } 865 // Now we're about to start to write changes to the trie. The trie is so far 866 // _untouched_. We can check with the prefetcher, if it can give us a trie 867 // which has the same root, but also has some content loaded into it. 868 if prefetcher != nil { 869 if trie := prefetcher.trie(s.originalRoot); trie != nil { 870 s.trie = trie 871 } 872 } 873 usedAddrs := make([][]byte, 0, len(s.stateObjectsPending)) 874 for addr := range s.stateObjectsPending { 875 if obj := s.stateObjects[addr]; obj.deleted { 876 s.deleteStateObject(obj) 877 } else { 878 s.updateStateObject(obj) 879 } 880 usedAddrs = append(usedAddrs, common.CopyBytes(addr[:])) // Copy needed for closure 881 } 882 if prefetcher != nil { 883 prefetcher.used(s.originalRoot, usedAddrs) 884 } 885 if len(s.stateObjectsPending) > 0 { 886 s.stateObjectsPending = make(map[common.Address]struct{}) 887 } 888 // Track the amount of time wasted on hashing the account trie 889 if metrics.EnabledExpensive { 890 defer func(start time.Time) { s.AccountHashes += time.Since(start) }(time.Now()) 891 } 892 return s.trie.Hash() 893 } 894 895 // Prepare sets the current transaction hash and index which are 896 // used when the EVM emits new state logs. 897 func (s *StateDB) Prepare(thash common.Hash, ti int) { 898 s.thash = thash 899 s.txIndex = ti 900 s.accessList = newAccessList() 901 } 902 903 func (s *StateDB) clearJournalAndRefund() { 904 if len(s.journal.entries) > 0 { 905 s.journal = newJournal() 906 s.refund = 0 907 } 908 s.validRevisions = s.validRevisions[:0] // Snapshots can be created without journal entires 909 } 910 911 // Commit writes the state to the underlying in-memory trie database. 912 func (s *StateDB) Commit(deleteEmptyObjects bool) (common.Hash, error) { 913 if s.dbErr != nil { 914 return common.Hash{}, fmt.Errorf("commit aborted due to earlier error: %v", s.dbErr) 915 } 916 // Finalize any pending changes and merge everything into the tries 917 s.IntermediateRoot(deleteEmptyObjects) 918 919 // Commit objects to the trie, measuring the elapsed time 920 codeWriter := s.db.TrieDB().DiskDB().NewBatch() 921 for addr := range s.stateObjectsDirty { 922 if obj := s.stateObjects[addr]; !obj.deleted { 923 // Write any contract code associated with the state object 924 if obj.code != nil && obj.dirtyCode { 925 rawdb.WriteCode(codeWriter, common.BytesToHash(obj.CodeHash()), obj.code) 926 obj.dirtyCode = false 927 } 928 // Write any storage changes in the state object to its storage trie 929 if err := obj.CommitTrie(s.db); err != nil { 930 return common.Hash{}, err 931 } 932 } 933 } 934 if len(s.stateObjectsDirty) > 0 { 935 s.stateObjectsDirty = make(map[common.Address]struct{}) 936 } 937 if codeWriter.ValueSize() > 0 { 938 if err := codeWriter.Write(); err != nil { 939 log.Crit("Failed to commit dirty codes", "error", err) 940 } 941 } 942 // Write the account trie changes, measuing the amount of wasted time 943 var start time.Time 944 if metrics.EnabledExpensive { 945 start = time.Now() 946 } 947 // The onleaf func is called _serially_, so we can reuse the same account 948 // for unmarshalling every time. 949 var account Account 950 root, err := s.trie.Commit(func(_ [][]byte, _ []byte, leaf []byte, parent common.Hash) error { 951 if err := rlp.DecodeBytes(leaf, &account); err != nil { 952 return nil 953 } 954 if account.Root != emptyRoot { 955 s.db.TrieDB().Reference(account.Root, parent) 956 } 957 return nil 958 }) 959 if metrics.EnabledExpensive { 960 s.AccountCommits += time.Since(start) 961 } 962 // If snapshotting is enabled, update the snapshot tree with this new version 963 if s.snap != nil { 964 if metrics.EnabledExpensive { 965 defer func(start time.Time) { s.SnapshotCommits += time.Since(start) }(time.Now()) 966 } 967 // Only update if there's a state transition (skip empty Clique blocks) 968 if parent := s.snap.Root(); parent != root { 969 if err := s.snaps.Update(root, parent, s.snapDestructs, s.snapAccounts, s.snapStorage); err != nil { 970 log.Warn("Failed to update snapshot tree", "from", parent, "to", root, "err", err) 971 } 972 // Keep 128 diff layers in the memory, persistent layer is 129th. 973 // - head layer is paired with HEAD state 974 // - head-1 layer is paired with HEAD-1 state 975 // - head-127 layer(bottom-most diff layer) is paired with HEAD-127 state 976 if err := s.snaps.Cap(root, 128); err != nil { 977 log.Warn("Failed to cap snapshot tree", "root", root, "layers", 128, "err", err) 978 } 979 } 980 s.snap, s.snapDestructs, s.snapAccounts, s.snapStorage = nil, nil, nil, nil 981 } 982 return root, err 983 } 984 985 // PrepareAccessList handles the preparatory steps for executing a state transition with 986 // regards to both EIP-2929 and EIP-2930: 987 // 988 // - Add sender to access list (2929) 989 // - Add destination to access list (2929) 990 // - Add precompiles to access list (2929) 991 // - Add the contents of the optional tx access list (2930) 992 // 993 // This method should only be called if Berlin/2929+2930 is applicable at the current number. 994 func (s *StateDB) PrepareAccessList(sender common.Address, dst *common.Address, precompiles []common.Address, list types.AccessList) { 995 s.AddAddressToAccessList(sender) 996 if dst != nil { 997 s.AddAddressToAccessList(*dst) 998 // If it's a create-tx, the destination will be added inside evm.create 999 } 1000 for _, addr := range precompiles { 1001 s.AddAddressToAccessList(addr) 1002 } 1003 for _, el := range list { 1004 s.AddAddressToAccessList(el.Address) 1005 for _, key := range el.StorageKeys { 1006 s.AddSlotToAccessList(el.Address, key) 1007 } 1008 } 1009 } 1010 1011 // AddAddressToAccessList adds the given address to the access list 1012 func (s *StateDB) AddAddressToAccessList(addr common.Address) { 1013 if s.accessList.AddAddress(addr) { 1014 s.journal.append(accessListAddAccountChange{&addr}) 1015 } 1016 } 1017 1018 // AddSlotToAccessList adds the given (address, slot)-tuple to the access list 1019 func (s *StateDB) AddSlotToAccessList(addr common.Address, slot common.Hash) { 1020 addrMod, slotMod := s.accessList.AddSlot(addr, slot) 1021 if addrMod { 1022 // In practice, this should not happen, since there is no way to enter the 1023 // scope of 'address' without having the 'address' become already added 1024 // to the access list (via call-variant, create, etc). 1025 // Better safe than sorry, though 1026 s.journal.append(accessListAddAccountChange{&addr}) 1027 } 1028 if slotMod { 1029 s.journal.append(accessListAddSlotChange{ 1030 address: &addr, 1031 slot: &slot, 1032 }) 1033 } 1034 } 1035 1036 // AddressInAccessList returns true if the given address is in the access list. 1037 func (s *StateDB) AddressInAccessList(addr common.Address) bool { 1038 return s.accessList.ContainsAddress(addr) 1039 } 1040 1041 // SlotInAccessList returns true if the given (address, slot)-tuple is in the access list. 1042 func (s *StateDB) SlotInAccessList(addr common.Address, slot common.Hash) (addressPresent bool, slotPresent bool) { 1043 return s.accessList.Contains(addr, slot) 1044 }