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  }