forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-07-22 20:56:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

added support for more granular storage slot override

Browse source
This commit is contained in:
kawahwookiee 2025-10-10 14:28:41 +02:00
parent 659342a523
commit b1d03eff56
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
internal/ethapi/override/override.go
View file

@ -42,6 +42,7 @@ type OverrideAccount struct {
Balance *hexutil.Big `json:"balance"`
State map[common.Hash]common.Hash `json:"state"`
StateDiff map[common.Hash]common.Hash `json:"stateDiff"`
StateDiffMask map[common.Hash]common.Hash `json:"stateDiffMask"`
MovePrecompileTo *common.Address `json:"movePrecompileToAddress"`
}
@ -98,8 +99,8 @@ func (diff *StateOverride) Apply(statedb *state.StateDB, precompiles vm.Precompi
u256Balance, _ := uint256.FromBig((*big.Int)(account.Balance))
statedb.SetBalance(addr, u256Balance, tracing.BalanceChangeUnspecified)
}
if account.State != nil && account.StateDiff != nil {
return fmt.Errorf("account %s has both 'state' and 'stateDiff'", addr.Hex())
if account.State != nil && (account.StateDiff != nil || account.StateDiffMask != nil) {
return fmt.Errorf("account %s has both 'state' and 'stateDiff' or 'stateDiffMask'", addr.Hex())
}
// Replace entire state if caller requires.
if account.State != nil {
@ -108,7 +109,15 @@ func (diff *StateOverride) Apply(statedb *state.StateDB, precompiles vm.Precompi
// Apply state diff into specified accounts.
if account.StateDiff != nil {
for key, value := range account.StateDiff {
statedb.SetState(addr, key, value)
if mask, ok := account.StateDiffMask[key]; !ok {
statedb.SetState(addr, key, value)
} else {
curSlotVal := statedb.GetState(addr, key)
for i := range curSlotVal {
curSlotVal[i] = ^mask[i]&curSlotVal[i] | value[i]
}
statedb.SetState(addr, key, curSlotVal)
}
}
}
}