mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-06-20 21:54:30 +00:00
380aa319f9
## Why this should be merged There are 3 places at which we perform the same, sensitive logic to access registered payloads and as we modify more types this is likely to expand. (e.g. `types.Header`). ## How this works Introduces `pseudo.Accessor` to abstract the reusable code. ## How this was tested Existing unit tests. Note that the `types.StateAccount` tests needed a minor refactor to provide the assertions with access to the `ExtraPayloads[T]` without introducing generic types anywhere.
64 lines
2.2 KiB
Go
64 lines
2.2 KiB
Go
// Copyright 2024 the libevm authors.
|
|
//
|
|
// The libevm additions to go-ethereum are free software: you can redistribute
|
|
// them and/or modify them under the terms of the GNU Lesser General Public License
|
|
// as published by the Free Software Foundation, either version 3 of the License,
|
|
// or (at your option) any later version.
|
|
//
|
|
// The libevm additions are distributed in the hope that they will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
|
|
// General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with the go-ethereum library. If not, see
|
|
// <http://www.gnu.org/licenses/>.
|
|
|
|
package state
|
|
|
|
import (
|
|
"github.com/ava-labs/libevm/common"
|
|
"github.com/ava-labs/libevm/core/types"
|
|
)
|
|
|
|
// GetExtra returns the extra payload from the [types.StateAccount] associated
|
|
// with the address, or a zero-value `SA` if not found. The
|
|
// [types.ExtraPayloads] MUST be sourced from [types.RegisterExtras].
|
|
func GetExtra[SA any](s *StateDB, p types.ExtraPayloads[SA], addr common.Address) SA {
|
|
stateObject := s.getStateObject(addr)
|
|
if stateObject != nil {
|
|
return p.StateAccount.Get(&stateObject.data)
|
|
}
|
|
var zero SA
|
|
return zero
|
|
}
|
|
|
|
// SetExtra sets the extra payload for the address. See [GetExtra] for details.
|
|
func SetExtra[SA any](s *StateDB, p types.ExtraPayloads[SA], addr common.Address, extra SA) {
|
|
stateObject := s.getOrNewStateObject(addr)
|
|
if stateObject != nil {
|
|
setExtraOnObject(stateObject, p, addr, extra)
|
|
}
|
|
}
|
|
|
|
func setExtraOnObject[SA any](s *stateObject, p types.ExtraPayloads[SA], addr common.Address, extra SA) {
|
|
s.db.journal.append(extraChange[SA]{
|
|
payloads: p,
|
|
account: &addr,
|
|
prev: p.StateAccount.Get(&s.data),
|
|
})
|
|
p.StateAccount.Set(&s.data, extra)
|
|
}
|
|
|
|
// extraChange is a [journalEntry] for [SetExtra] / [setExtraOnObject].
|
|
type extraChange[SA any] struct {
|
|
payloads types.ExtraPayloads[SA]
|
|
account *common.Address
|
|
prev SA
|
|
}
|
|
|
|
func (e extraChange[SA]) dirtied() *common.Address { return e.account }
|
|
|
|
func (e extraChange[SA]) revert(s *StateDB) {
|
|
e.payloads.StateAccount.Set(&s.getStateObject(*e.account).data, e.prev)
|
|
}
|