pre-derive authorities

This commit is contained in:
alan 2025-02-18 10:24:49 +08:00 committed by lightclient
parent 135d9be3d8
commit 95c732cfac
No known key found for this signature in database
GPG key ID: 657913021EF45A6A
6 changed files with 93 additions and 36 deletions

View file

@ -156,6 +156,7 @@ type Message struct {
BlobGasFeeCap *big.Int
BlobHashes []common.Hash
SetCodeAuthorizations []types.SetCodeAuthorization
AuthorityCache *types.AuthorityCache
// When SkipNonceChecks is true, the message nonce is not checked against the
// account nonce in state.
@ -185,6 +186,7 @@ func TransactionToMessage(tx *types.Transaction, s types.Signer, baseFee *big.In
Data: tx.Data(),
AccessList: tx.AccessList(),
SetCodeAuthorizations: tx.SetCodeAuthorizations(),
AuthorityCache: tx.AuthorityCache(),
SkipNonceChecks: false,
SkipTransactionChecks: false,
BlobHashes: tx.BlobHashes(),
@ -504,9 +506,13 @@ func (st *stateTransition) execute() (*ExecutionResult, error) {
// Apply EIP-7702 authorizations.
if msg.SetCodeAuthorizations != nil {
for _, auth := range msg.SetCodeAuthorizations {
for i, auth := range msg.SetCodeAuthorizations {
// Note errors are ignored, we simply skip invalid authorizations here.
st.applyAuthorization(&auth)
authority, err := msg.AuthorityCache.Authority(i)
if err != nil {
continue
}
st.applyAuthorization(&auth, authority)
}
}
@ -574,7 +580,7 @@ func (st *stateTransition) execute() (*ExecutionResult, error) {
}
// validateAuthorization validates an EIP-7702 authorization against the state.
func (st *stateTransition) validateAuthorization(auth *types.SetCodeAuthorization) (authority common.Address, err error) {
func (st *stateTransition) validateAuthorization(auth *types.SetCodeAuthorization, preDerivedAuthority common.Address) (authority common.Address, err error) {
// Verify chain ID is null or equal to current chain ID.
if !auth.ChainID.IsZero() && auth.ChainID.CmpBig(st.evm.ChainConfig().ChainID) != 0 {
return authority, ErrAuthorizationWrongChainID
@ -583,11 +589,9 @@ func (st *stateTransition) validateAuthorization(auth *types.SetCodeAuthorizatio
if auth.Nonce+1 < auth.Nonce {
return authority, ErrAuthorizationNonceOverflow
}
// Validate signature values and recover authority.
authority, err = auth.Authority()
if err != nil {
return authority, fmt.Errorf("%w: %v", ErrAuthorizationInvalidSignature, err)
}
authority = preDerivedAuthority
// Check the authority account
// 1) doesn't have code or has exisiting delegation
// 2) matches the auth's nonce
@ -605,8 +609,8 @@ func (st *stateTransition) validateAuthorization(auth *types.SetCodeAuthorizatio
}
// applyAuthorization applies an EIP-7702 code delegation to the state.
func (st *stateTransition) applyAuthorization(auth *types.SetCodeAuthorization) error {
authority, err := st.validateAuthorization(auth)
func (st *stateTransition) applyAuthorization(auth *types.SetCodeAuthorization, preDerivedAuthority common.Address) error {
authority, err := st.validateAuthorization(auth, preDerivedAuthority)
if err != nil {
return err
}

View file

@ -519,20 +519,25 @@ func (tx *Transaction) SetCodeAuthorities() []common.Address {
if !ok {
return nil
}
var (
marks = make(map[common.Address]bool)
auths = make([]common.Address, 0, len(setcodetx.AuthList))
)
for _, auth := range setcodetx.AuthList {
if addr, err := auth.Authority(); err == nil {
if marks[addr] {
continue
}
marks[addr] = true
auths = append(auths, addr)
}
if authorityCache := setcodetx.authorityCache.Load(); authorityCache != nil {
return authorityCache.authorities
}
return auths
cache := ToAuthorityCache(setcodetx.AuthList)
setcodetx.authorityCache.Store(cache)
return cache.authorities
}
func (tx *Transaction) AuthorityCache() *AuthorityCache {
setcodetx, ok := tx.inner.(*SetCodeTx)
if !ok {
return nil
}
if authorityCache := setcodetx.authorityCache.Load(); authorityCache != nil {
return authorityCache
}
cache := ToAuthorityCache(setcodetx.AuthList)
setcodetx.authorityCache.Store(cache)
return cache
}
// SetTime sets the decoding time of a transaction. This is used by tests to set

View file

@ -65,19 +65,70 @@ type SetCodeTx struct {
V *uint256.Int
R *uint256.Int
S *uint256.Int
// caches
authorityCache atomic.Pointer[AuthorityCache]
}
func ToAuthorityCache(authList []SetCodeAuthorization) *AuthorityCache {
var (
marks = make(map[common.Address]bool)
auths = make([]common.Address, 0, len(authList))
invalid []invalidAuth // empty since it's expected to be empty most of the time
)
for i, auth := range authList {
if addr, err := auth.Authority(); err == nil {
if marks[addr] {
continue
}
marks[addr] = true
auths = append(auths, addr)
} else {
invalid = append(invalid, invalidAuth{index: i, error: err})
}
}
return &AuthorityCache{authorities: auths, invalidAuths: invalid}
}
type AuthorityCache struct {
authorities []common.Address
invalidAuths []invalidAuth
}
type invalidAuth struct {
index int
error error
}
func (ac *AuthorityCache) Authority(i int) (common.Address, error) {
indexToSub := 0
for _, invalid := range ac.invalidAuths {
if i == invalid.index {
return common.Address{}, invalid.error
}
if i > invalid.index {
indexToSub++
continue
}
break
}
targetIndex := i - indexToSub
if targetIndex < 0 || targetIndex >= len(ac.authorities) {
return common.Address{}, errors.New("index out of range")
}
return ac.authorities[i-indexToSub], nil
}
//go:generate go run github.com/fjl/gencodec -type SetCodeAuthorization -field-override authorizationMarshaling -out gen_authorization.go
// SetCodeAuthorization is an authorization from an account to deploy code at its address.
type SetCodeAuthorization struct {
ChainID uint256.Int `json:"chainId" gencodec:"required"`
Address common.Address `json:"address" gencodec:"required"`
Nonce uint64 `json:"nonce" gencodec:"required"`
V uint8 `json:"yParity" gencodec:"required"`
R uint256.Int `json:"r" gencodec:"required"`
S uint256.Int `json:"s" gencodec:"required"`
authority atomic.Pointer[common.Address] `json:"-"`
ChainID uint256.Int `json:"chainId" gencodec:"required"`
Address common.Address `json:"address" gencodec:"required"`
Nonce uint64 `json:"nonce" gencodec:"required"`
V uint8 `json:"yParity" gencodec:"required"`
R uint256.Int `json:"r" gencodec:"required"`
S uint256.Int `json:"s" gencodec:"required"`
}
// field type overrides for gencodec
@ -137,7 +188,6 @@ func (a *SetCodeAuthorization) Authority() (common.Address, error) {
}
var addr common.Address
copy(addr[:], crypto.Keccak256(pub[1:])[12:])
a.authority.Store(&addr)
return addr, nil
}

View file

@ -185,11 +185,7 @@ func (t *prestateTracer) OnTxStart(env *tracing.VMContext, tx *types.Transaction
t.lookupAccount(env.Coinbase)
// Add accounts with authorizations to the prestate before they get applied.
for _, auth := range tx.SetCodeAuthorizations() {
addr, err := auth.Authority()
if err != nil {
continue
}
for _, addr := range tx.SetCodeAuthorities() {
t.lookupAccount(addr)
}
}

View file

@ -490,6 +490,7 @@ func (args *TransactionArgs) ToMessage(baseFee *big.Int, skipNonceCheck bool) *c
BlobGasFeeCap: (*big.Int)(args.BlobFeeCap),
BlobHashes: args.BlobHashes,
SetCodeAuthorizations: args.AuthorizationList,
AuthorityCache: types.ToAuthorityCache(args.AuthorizationList),
SkipNonceChecks: skipNonceCheck,
SkipTransactionChecks: true,
}

View file

@ -477,6 +477,7 @@ func (tx *stTransaction) toMessage(ps stPostState, baseFee *big.Int) (*core.Mess
BlobHashes: tx.BlobVersionedHashes,
BlobGasFeeCap: tx.BlobGasFeeCap,
SetCodeAuthorizations: authList,
AuthorityCache: types.ToAuthorityCache(authList),
}
return msg, nil
}