github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/fvm/evm/emulator/state/stateDB.go

package state

import (
	"bytes"
	stdErrors "errors"
	"fmt"
	"math/big"
	"sort"

	"github.com/onflow/atree"
	gethCommon "github.com/onflow/go-ethereum/common"
	gethTypes "github.com/onflow/go-ethereum/core/types"
	gethParams "github.com/onflow/go-ethereum/params"

	"github.com/onflow/flow-go/fvm/evm/types"
	"github.com/onflow/flow-go/model/flow"
)

// StateDB implements a types.StateDB interface
//
// stateDB interface defined by the Geth doesn't support returning errors
// when state calls are happening, and requires stateDB to cache the error
// and return it at a later time (when commit is called). Only the first error
// is expected to be returned.
// Warning: current implementation of the StateDB is considered
// to be used for a single EVM transaction execution and is not
// thread safe. yet the current design supports addition of concurrency in the
// future if needed
type StateDB struct {
	ledger      atree.Ledger
	root        flow.Address
	baseView    types.BaseView
	views       []*DeltaView
	cachedError error
}

var _ types.StateDB = &StateDB{}

// NewStateDB constructs a new StateDB
func NewStateDB(ledger atree.Ledger, root flow.Address) (*StateDB, error) {
	bv, err := NewBaseView(ledger, root)
	if err != nil {
		return nil, err
	}
	return &StateDB{
		ledger:      ledger,
		root:        root,
		baseView:    bv,
		views:       []*DeltaView{NewDeltaView(bv)},
		cachedError: nil,
	}, nil
}

// Exist returns true if the given address exists in state.
//
// this should also return true for self destructed accounts during the transaction execution.
func (db *StateDB) Exist(addr gethCommon.Address) bool {
	exist, err := db.lastestView().Exist(addr)
	db.handleError(err)
	return exist
}

// Empty returns whether the given account is empty.
//
// Empty is defined according to EIP161 (balance = nonce = code = 0).
func (db *StateDB) Empty(addr gethCommon.Address) bool {
	if !db.Exist(addr) {
		return true
	}
	return db.GetNonce(addr) == 0 &&
		db.GetBalance(addr).Sign() == 0 &&
		bytes.Equal(db.GetCodeHash(addr).Bytes(), gethTypes.EmptyCodeHash.Bytes())
}

// CreateAccount creates a new account for the given address
// it sets the nonce to zero
func (db *StateDB) CreateAccount(addr gethCommon.Address) {
	err := db.lastestView().CreateAccount(addr)
	db.handleError(err)
}

// IsCreated returns true if address is recently created (context of a transaction)
func (db *StateDB) IsCreated(addr gethCommon.Address) bool {
	return db.lastestView().IsCreated(addr)
}

// SelfDestruct flags the address for deletion.
//
// while this address exists for the rest of transaction,
// the balance of this account is return zero after the SelfDestruct call.
func (db *StateDB) SelfDestruct(addr gethCommon.Address) {
	err := db.lastestView().SelfDestruct(addr)
	db.handleError(err)
}

// Selfdestruct6780 would only follow the self destruct steps if account is created
func (db *StateDB) Selfdestruct6780(addr gethCommon.Address) {
	if db.IsCreated(addr) {
		db.SelfDestruct(addr)
	}
}

// HasSelfDestructed returns true if address is flaged with self destruct.
func (db *StateDB) HasSelfDestructed(addr gethCommon.Address) bool {
	destructed, _ := db.lastestView().HasSelfDestructed(addr)
	return destructed
}

// SubBalance substitutes the amount from the balance of the given address
func (db *StateDB) SubBalance(addr gethCommon.Address, amount *big.Int) {
	err := db.lastestView().SubBalance(addr, amount)
	db.handleError(err)
}

// AddBalance adds the amount from the balance of the given address
func (db *StateDB) AddBalance(addr gethCommon.Address, amount *big.Int) {
	err := db.lastestView().AddBalance(addr, amount)
	db.handleError(err)
}

// GetBalance returns the balance of the given address
func (db *StateDB) GetBalance(addr gethCommon.Address) *big.Int {
	bal, err := db.lastestView().GetBalance(addr)
	db.handleError(err)
	return bal
}

// GetNonce returns the nonce of the given address
func (db *StateDB) GetNonce(addr gethCommon.Address) uint64 {
	nonce, err := db.lastestView().GetNonce(addr)
	db.handleError(err)
	return nonce
}

// SetNonce sets the nonce value for the given address
func (db *StateDB) SetNonce(addr gethCommon.Address, nonce uint64) {
	err := db.lastestView().SetNonce(addr, nonce)
	db.handleError(err)
}

// GetCodeHash returns the code hash of the given address
func (db *StateDB) GetCodeHash(addr gethCommon.Address) gethCommon.Hash {
	hash, err := db.lastestView().GetCodeHash(addr)
	db.handleError(err)
	return hash
}

// GetCode returns the code for the given address
func (db *StateDB) GetCode(addr gethCommon.Address) []byte {
	code, err := db.lastestView().GetCode(addr)
	db.handleError(err)
	return code
}

// GetCodeSize returns the size of the code for the given address
func (db *StateDB) GetCodeSize(addr gethCommon.Address) int {
	codeSize, err := db.lastestView().GetCodeSize(addr)
	db.handleError(err)
	return codeSize
}

// SetCode sets the code for the given address
func (db *StateDB) SetCode(addr gethCommon.Address, code []byte) {
	err := db.lastestView().SetCode(addr, code)
	db.handleError(err)
}

// AddRefund adds the amount to the total (gas) refund
func (db *StateDB) AddRefund(amount uint64) {
	err := db.lastestView().AddRefund(amount)
	db.handleError(err)
}

// SubRefund subtracts the amount from the total (gas) refund
func (db *StateDB) SubRefund(amount uint64) {
	err := db.lastestView().SubRefund(amount)
	db.handleError(err)
}

// GetRefund returns the total (gas) refund
func (db *StateDB) GetRefund() uint64 {
	return db.lastestView().GetRefund()
}

// GetCommittedState returns the value for the given storage slot considering only the commited state and not
// changes in the scope of current transaction.
func (db *StateDB) GetCommittedState(addr gethCommon.Address, key gethCommon.Hash) gethCommon.Hash {
	value, err := db.baseView.GetState(types.SlotAddress{Address: addr, Key: key})
	db.handleError(err)
	return value
}

// GetState returns the value for the given storage slot
func (db *StateDB) GetState(addr gethCommon.Address, key gethCommon.Hash) gethCommon.Hash {
	state, err := db.lastestView().GetState(types.SlotAddress{Address: addr, Key: key})
	db.handleError(err)
	return state
}

// SetState sets a value for the given storage slot
func (db *StateDB) SetState(addr gethCommon.Address, key gethCommon.Hash, value gethCommon.Hash) {
	err := db.lastestView().SetState(types.SlotAddress{Address: addr, Key: key}, value)
	db.handleError(err)
}

// GetTransientState returns the value for the given key of the transient storage
func (db *StateDB) GetTransientState(addr gethCommon.Address, key gethCommon.Hash) gethCommon.Hash {
	return db.lastestView().GetTransientState(types.SlotAddress{Address: addr, Key: key})
}

// SetTransientState sets a value for the given key of the transient storage
func (db *StateDB) SetTransientState(addr gethCommon.Address, key, value gethCommon.Hash) {
	db.lastestView().SetTransientState(types.SlotAddress{Address: addr, Key: key}, value)
}

// AddressInAccessList checks if an address is in the access list
func (db *StateDB) AddressInAccessList(addr gethCommon.Address) bool {
	return db.lastestView().AddressInAccessList(addr)
}

// SlotInAccessList checks if the given (address,slot) is in the access list
func (db *StateDB) SlotInAccessList(addr gethCommon.Address, key gethCommon.Hash) (addressOk bool, slotOk bool) {
	return db.lastestView().SlotInAccessList(types.SlotAddress{Address: addr, Key: key})
}

// AddAddressToAccessList adds the given address to the access list.
func (db *StateDB) AddAddressToAccessList(addr gethCommon.Address) {
	db.lastestView().AddAddressToAccessList(addr)
}

// AddSlotToAccessList adds the given (address,slot) to the access list.
func (db *StateDB) AddSlotToAccessList(addr gethCommon.Address, key gethCommon.Hash) {
	db.lastestView().AddSlotToAccessList(types.SlotAddress{Address: addr, Key: key})
}

// AddLog appends a lot to the collection of logs
func (db *StateDB) AddLog(log *gethTypes.Log) {
	db.lastestView().AddLog(log)
}

// AddPreimage adds a preimage to the collection of preimages
func (db *StateDB) AddPreimage(hash gethCommon.Hash, data []byte) {
	db.lastestView().AddPreimage(hash, data)
}

// RevertToSnapshot reverts the changes until we reach the given snaptshot
func (db *StateDB) RevertToSnapshot(index int) {
	if index > len(db.views) {
		db.cachedError = fmt.Errorf("invalid revert")
		return
	}
	db.views = db.views[:index]
}

// Snapshot takes an snapshot of the state and returns an int
// that can be used later for revert calls.
func (db *StateDB) Snapshot() int {
	newView := db.lastestView().NewChildView()
	db.views = append(db.views, newView)
	return len(db.views) - 1
}

// Logs returns the list of logs
// it also update each log with the block and tx info
func (db *StateDB) Logs(
	blockNumber uint64,
	txHash gethCommon.Hash,
	txIndex uint,
) []*gethTypes.Log {
	allLogs := make([]*gethTypes.Log, 0)
	for _, view := range db.views {
		for _, log := range view.Logs() {
			log.BlockNumber = blockNumber
			log.TxHash = txHash
			log.TxIndex = txIndex
			allLogs = append(allLogs, log)
		}
	}
	return allLogs
}

// Preimages returns a set of preimages
func (db *StateDB) Preimages() map[gethCommon.Hash][]byte {
	preImages := make(map[gethCommon.Hash][]byte, 0)
	for _, view := range db.views {
		for k, v := range view.Preimages() {
			preImages[k] = v
		}
	}
	return preImages
}

// Commit commits state changes back to the underlying
func (db *StateDB) Commit(finalize bool) error {
	// return error if any has been acumulated
	if db.cachedError != nil {
		return wrapError(db.cachedError)
	}

	var err error

	// iterate views and collect dirty addresses and slots
	addresses := make(map[gethCommon.Address]struct{})
	slots := make(map[types.SlotAddress]struct{})
	for _, view := range db.views {
		for key := range view.DirtyAddresses() {
			addresses[key] = struct{}{}
		}
		for key := range view.DirtySlots() {
			slots[key] = struct{}{}
		}
	}

	// sort addresses
	sortedAddresses := make([]gethCommon.Address, 0, len(addresses))
	for addr := range addresses {
		sortedAddresses = append(sortedAddresses, addr)
	}

	sort.Slice(sortedAddresses,
		func(i, j int) bool {
			return bytes.Compare(sortedAddresses[i][:], sortedAddresses[j][:]) < 0
		})

	// update accounts
	for _, addr := range sortedAddresses {
		deleted := false
		// first we need to delete accounts
		if db.HasSelfDestructed(addr) {
			err = db.baseView.DeleteAccount(addr)
			if err != nil {
				return wrapError(err)
			}
			deleted = true
		}
		// then create new ones
		// an account might be in a single transaction be deleted and recreated
		if db.IsCreated(addr) {
			err = db.baseView.CreateAccount(
				addr,
				db.GetBalance(addr),
				db.GetNonce(addr),
				db.GetCode(addr),
				db.GetCodeHash(addr),
			)
			if err != nil {
				return wrapError(err)
			}
			continue
		}
		if deleted {
			continue
		}
		err = db.baseView.UpdateAccount(
			addr,
			db.GetBalance(addr),
			db.GetNonce(addr),
			db.GetCode(addr),
			db.GetCodeHash(addr),
		)
		if err != nil {
			return wrapError(err)
		}
	}

	// sort slots
	sortedSlots := make([]types.SlotAddress, 0, len(slots))
	for slot := range slots {
		sortedSlots = append(sortedSlots, slot)
	}
	sort.Slice(sortedSlots, func(i, j int) bool {
		comp := bytes.Compare(sortedSlots[i].Address[:], sortedSlots[j].Address[:])
		if comp == 0 {
			return bytes.Compare(sortedSlots[i].Key[:], sortedSlots[j].Key[:]) < 0
		}
		return comp < 0
	})

	// update slots
	for _, sk := range sortedSlots {
		err = db.baseView.UpdateSlot(
			sk,
			db.GetState(sk.Address, sk.Key),
		)
		if err != nil {
			return wrapError(err)
		}
	}

	// don't purge views yet, people might call the logs etc
	if finalize {
		return db.Finalize()
	}
	return nil
}

// Finalize flushes all the changes
// to the permanent storage
func (db *StateDB) Finalize() error {
	err := db.baseView.Commit()
	return wrapError(err)
}

// Prepare is a highlevel logic that sadly is considered to be part of the
// stateDB interface and not on the layers above.
// based on parameters that are passed it updates accesslists
func (db *StateDB) Prepare(rules gethParams.Rules, sender, coinbase gethCommon.Address, dest *gethCommon.Address, precompiles []gethCommon.Address, txAccesses gethTypes.AccessList) {
	if rules.IsBerlin {
		db.AddAddressToAccessList(sender)

		if dest != nil {
			db.AddAddressToAccessList(*dest)
			// If it's a create-tx, the destination will be added inside egethVM.create
		}
		for _, addr := range precompiles {
			db.AddAddressToAccessList(addr)
		}
		for _, el := range txAccesses {
			db.AddAddressToAccessList(el.Address)
			for _, key := range el.StorageKeys {
				db.AddSlotToAccessList(el.Address, key)
			}
		}
		if rules.IsShanghai { // EIP-3651: warm coinbase
			db.AddAddressToAccessList(coinbase)
		}
	}
}

// Reset resets uncommitted changes and transient artifacts such as error, logs,
// preimages, access lists, ...
// The method is often called between execution of different transactions
func (db *StateDB) Reset() {
	db.views = []*DeltaView{NewDeltaView(db.baseView)}
	db.cachedError = nil
}

// Error returns the memorized database failure occurred earlier.
func (s *StateDB) Error() error {
	return wrapError(s.cachedError)
}

func (db *StateDB) lastestView() *DeltaView {
	return db.views[len(db.views)-1]
}

// set error captures the first non-nil error it is called with.
func (db *StateDB) handleError(err error) {
	if err == nil {
		return
	}
	if db.cachedError == nil {
		db.cachedError = err
	}
}

func wrapError(err error) error {
	if err == nil {
		return nil
	}

	var atreeUserError *atree.UserError
	// if is an atree user error
	if stdErrors.As(err, &atreeUserError) {
		return types.NewStateError(err)
	}

	var atreeFatalError *atree.FatalError
	// if is a atree fatal error or
	if stdErrors.As(err, &atreeFatalError) {
		return types.NewFatalError(err)
	}

	// if is a fatal error
	if types.IsAFatalError(err) {
		return err
	}

	return types.NewStateError(err)
}