feat(precompiles): add BalanceHandler to handle native balance change

evmd/app.go

@@ -448,15 +448,6 @@ func NewExampleApp(
 		runtime.ProvideCometInfoService(),
 	)
 	// If evidence needs to be handled for the app, set routes in router here and seal
-	// Note: The evidence precompile allows evidence to be submitted through an EVM transaction.
-	// If you implement a custom evidence handler in the router that changes token balances (e.g. penalizing
-	// addresses, deducting fees, etc.), be aware that the precompile logic (e.g. SetBalanceChangeEntries)
-	// must be properly integrated to reflect these balance changes in the EVM state. Otherwise, there is a risk
-	// of desynchronization between the Cosmos SDK state and the EVM state when evidence is submitted via the EVM.
-	//
-	// For example, if your custom evidence handler deducts tokens from a user’s account, ensure that the evidence
-	// precompile also applies these deductions through the EVM’s balance tracking. Failing to do so may cause
-	// inconsistencies in reported balances and break state synchronization.
 	app.EvidenceKeeper = *evidenceKeeper
 
 	// Cosmos EVM keepers

precompiles/bank/bank.go

@@ -104,7 +104,7 @@ func (p Precompile) RequiredGas(input []byte) uint64 {
 
 // Run executes the precompiled contract bank query methods defined in the ABI.
 func (p Precompile) Run(evm *vm.EVM, contract *vm.Contract, readOnly bool) (bz []byte, err error) {
-	ctx, stateDB, method, initialGas, args, err := p.RunSetup(evm, contract, readOnly, p.IsTransaction)
+	ctx, _, method, initialGas, args, err := p.RunSetup(evm, contract, readOnly, p.IsTransaction)
 	if err != nil {
 		return nil, err
 	}
@@ -134,9 +134,6 @@ func (p Precompile) Run(evm *vm.EVM, contract *vm.Contract, readOnly bool) (bz [
 	if !contract.UseGas(cost, nil, tracing.GasChangeCallPrecompiledContract) {
 		return nil, vm.ErrOutOfGas
 	}
-	if err = p.AddJournalEntries(stateDB); err != nil {
-		return nil, err
-	}
 
 	return bz, nil
 }

precompiles/common/balance_handler.go

@@ -0,0 +1,107 @@
+package common
+
+import (
+	"fmt"
+
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/core/tracing"
+	"github.com/holiman/uint256"
+
+	"github.com/cosmos/evm/utils"
+	"github.com/cosmos/evm/x/vm/statedb"
+	evmtypes "github.com/cosmos/evm/x/vm/types"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
+)
+
+// BalanceHandler is a struct that handles balance changes in the Cosmos SDK context.
+type BalanceHandler struct {
+	prevEventsLen int
+}
+
+// NewBalanceHandler creates a new BalanceHandler instance.
+func NewBalanceHandler() *BalanceHandler {
+	return &BalanceHandler{
+		prevEventsLen: 0,
+	}
+}
+
+// BeforeBalanceChange is called before any balance changes by precompile methods.
+// It records the current number of events in the context to later process balance changes
+// using the recorded events.
+func (bh *BalanceHandler) BeforeBalanceChange(ctx sdk.Context) {
+	bh.prevEventsLen = len(ctx.EventManager().Events())
+}
+
+// AfterBalanceChange processes the recorded events and updates the stateDB accordingly.
+// It handles the bank events for coin spent and coin received, updating the balances
+// of the spender and receiver addresses respectively.
+func (bh *BalanceHandler) AfterBalanceChange(ctx sdk.Context, stateDB *statedb.StateDB) error {
+	events := ctx.EventManager().Events()
+
+	for _, event := range events[bh.prevEventsLen:] {
+		switch event.Type {
+		case banktypes.EventTypeCoinSpent:
+			spenderHexAddr, err := parseHexAddress(event, banktypes.AttributeKeySpender)
+			if err != nil {
+				return fmt.Errorf("failed to parse spender address from event %q: %w", banktypes.EventTypeCoinSpent, err)
+			}
+
+			amount, err := parseAmount(event)
+			if err != nil {
+				return fmt.Errorf("failed to parse amount from event %q: %w", banktypes.EventTypeCoinSpent, err)
+			}
+
+			stateDB.SubBalance(spenderHexAddr, amount, tracing.BalanceChangeUnspecified)
+
+		case banktypes.EventTypeCoinReceived:
+			receiverHexAddr, err := parseHexAddress(event, banktypes.AttributeKeyReceiver)
+			if err != nil {
+				return fmt.Errorf("failed to parse receiver address from event %q: %w", banktypes.EventTypeCoinReceived, err)
+			}
+
+			amount, err := parseAmount(event)
+			if err != nil {
+				return fmt.Errorf("failed to parse amount from event %q: %w", banktypes.EventTypeCoinReceived, err)
+			}
+
+			stateDB.AddBalance(receiverHexAddr, amount, tracing.BalanceChangeUnspecified)
+		}
+	}
+
+	return nil
+}
+
+func parseHexAddress(event sdk.Event, key string) (common.Address, error) {
+	attr, ok := event.GetAttribute(key)
+	if !ok {
+		return common.Address{}, fmt.Errorf("event %q missing attribute %q", event.Type, key)
+	}
+
+	accAddr, err := sdk.AccAddressFromBech32(attr.Value)
+	if err != nil {
+		return common.Address{}, fmt.Errorf("invalid address %q: %w", attr.Value, err)
+	}
+
+	return common.Address(accAddr.Bytes()), nil
+}
+
+func parseAmount(event sdk.Event) (*uint256.Int, error) {
+	amountAttr, ok := event.GetAttribute(sdk.AttributeKeyAmount)
+	if !ok {
+		return nil, fmt.Errorf("event %q missing attribute %q", banktypes.EventTypeCoinSpent, sdk.AttributeKeyAmount)
+	}
+
+	amountCoins, err := sdk.ParseCoinsNormalized(amountAttr.Value)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse coins from %q: %w", amountAttr.Value, err)
+	}
+
+	amountBigInt := amountCoins.AmountOf(evmtypes.GetEVMCoinDenom()).BigInt()
+	amount, err := utils.Uint256FromBigInt(evmtypes.ConvertAmountTo18DecimalsBigInt(amountBigInt))
+	if err != nil {
+		return nil, fmt.Errorf("failed to convert coin amount to Uint256: %w", err)
+	}
+	return amount, nil
+}

precompiles/common/balance_handler_test.go

@@ -0,0 +1,206 @@
+package common
+
+import (
+	"testing"
+
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/core/tracing"
+	"github.com/holiman/uint256"
+	"github.com/stretchr/testify/require"
+
+	testutil "github.com/cosmos/evm/testutil"
+	testconstants "github.com/cosmos/evm/testutil/constants"
+	"github.com/cosmos/evm/x/vm/statedb"
+	evmtypes "github.com/cosmos/evm/x/vm/types"
+	"github.com/cosmos/evm/x/vm/types/mocks"
+
+	storetypes "cosmossdk.io/store/types"
+
+	sdktestutil "github.com/cosmos/cosmos-sdk/testutil"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
+)
+
+func setupBalanceHandlerTest(t *testing.T) {
+	t.Helper()
+
+	sdk.GetConfig().SetBech32PrefixForAccount(testconstants.ExampleBech32Prefix, "")
+	configurator := evmtypes.NewEVMConfigurator()
+	configurator.ResetTestConfig()
+	require.NoError(t, configurator.WithEVMCoinInfo(testconstants.ExampleChainCoinInfo[testconstants.ExampleChainID]).Configure())
+}
+
+func TestParseHexAddress(t *testing.T) {
+	var accAddr sdk.AccAddress
+
+	testCases := []struct {
+		name     string
+		maleate  func() sdk.Event
+		key      string
+		expAddr  common.Address
+		expError bool
+	}{
+		{
+			name: "valid address",
+			maleate: func() sdk.Event {
+				return sdk.NewEvent("bank", sdk.NewAttribute(banktypes.AttributeKeySpender, accAddr.String()))
+			},
+			key:      banktypes.AttributeKeySpender,
+			expError: false,
+		},
+		{
+			name: "missing attribute",
+			maleate: func() sdk.Event {
+				return sdk.NewEvent("bank")
+			},
+			key:      banktypes.AttributeKeySpender,
+			expError: true,
+		},
+		{
+			name: "invalid address",
+			maleate: func() sdk.Event {
+				return sdk.NewEvent("bank", sdk.NewAttribute(banktypes.AttributeKeySpender, "invalid"))
+			},
+			key:      banktypes.AttributeKeySpender,
+			expError: true,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			setupBalanceHandlerTest(t)
+
+			_, addrs, err := testutil.GeneratePrivKeyAddressPairs(1)
+			require.NoError(t, err)
+			accAddr = addrs[0]
+
+			event := tc.maleate()
+
+			addr, err := parseHexAddress(event, tc.key)
+			if tc.expError {
+				require.Error(t, err)
+				return
+			}
+
+			require.NoError(t, err)
+			require.Equal(t, common.Address(accAddr.Bytes()), addr)
+		})
+	}
+}
+
+func TestParseAmount(t *testing.T) {
+	testCases := []struct {
+		name     string
+		maleate  func() sdk.Event
+		expAmt   *uint256.Int
+		expError bool
+	}{
+		{
+			name: "valid amount",
+			maleate: func() sdk.Event {
+				coinStr := sdk.NewCoins(sdk.NewInt64Coin(evmtypes.GetEVMCoinDenom(), 5)).String()
+				return sdk.NewEvent("bank", sdk.NewAttribute(sdk.AttributeKeyAmount, coinStr))
+			},
+			expAmt: uint256.NewInt(5),
+		},
+		{
+			name: "missing amount",
+			maleate: func() sdk.Event {
+				return sdk.NewEvent("bank")
+			},
+			expError: true,
+		},
+		{
+			name: "invalid coins",
+			maleate: func() sdk.Event {
+				return sdk.NewEvent("bank", sdk.NewAttribute(sdk.AttributeKeyAmount, "invalid"))
+			},
+			expError: true,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			setupBalanceHandlerTest(t)
+
+			amt, err := parseAmount(tc.maleate())
+			if tc.expError {
+				require.Error(t, err)
+				return
+			}
+
+			require.NoError(t, err)
+			require.True(t, amt.Eq(tc.expAmt))
+		})
+	}
+}
+
+func TestAfterBalanceChange(t *testing.T) {
+	setupBalanceHandlerTest(t)
+
+	storeKey := storetypes.NewKVStoreKey("test")
+	tKey := storetypes.NewTransientStoreKey("test_t")
+	ctx := sdktestutil.DefaultContext(storeKey, tKey)
+
+	stateDB := statedb.New(ctx, mocks.NewEVMKeeper(), statedb.NewEmptyTxConfig(common.BytesToHash(ctx.HeaderHash())))
+
+	_, addrs, err := testutil.GeneratePrivKeyAddressPairs(2)
+	require.NoError(t, err)
+	spenderAcc := addrs[0]
+	receiverAcc := addrs[1]
+	spender := common.Address(spenderAcc.Bytes())
+	receiver := common.Address(receiverAcc.Bytes())
+
+	// initial balance for spender
+	stateDB.AddBalance(spender, uint256.NewInt(5), tracing.BalanceChangeUnspecified)
+
+	bh := NewBalanceHandler()
+	bh.BeforeBalanceChange(ctx)
+
+	coins := sdk.NewCoins(sdk.NewInt64Coin(evmtypes.GetEVMCoinDenom(), 3))
+	ctx.EventManager().EmitEvents(sdk.Events{
+		banktypes.NewCoinSpentEvent(spenderAcc, coins),
+		banktypes.NewCoinReceivedEvent(receiverAcc, coins),
+	})
+
+	err = bh.AfterBalanceChange(ctx, stateDB)
+	require.NoError(t, err)
+
+	require.Equal(t, "2", stateDB.GetBalance(spender).String())
+	require.Equal(t, "3", stateDB.GetBalance(receiver).String())
+}
+
+func TestAfterBalanceChangeErrors(t *testing.T) {
+	setupBalanceHandlerTest(t)
+
+	storeKey := storetypes.NewKVStoreKey("test")
+	tKey := storetypes.NewTransientStoreKey("test_t")
+	ctx := sdktestutil.DefaultContext(storeKey, tKey)
+	stateDB := statedb.New(ctx, mocks.NewEVMKeeper(), statedb.NewEmptyTxConfig(common.BytesToHash(ctx.HeaderHash())))
+
+	_, addrs, err := testutil.GeneratePrivKeyAddressPairs(1)
+	require.NoError(t, err)
+	addr := addrs[0]
+
+	bh := NewBalanceHandler()
+	bh.BeforeBalanceChange(ctx)
+
+	// invalid address in event
+	coins := sdk.NewCoins(sdk.NewInt64Coin(evmtypes.GetEVMCoinDenom(), 1))
+	ctx.EventManager().EmitEvent(banktypes.NewCoinSpentEvent(addr, coins))
+	ctx.EventManager().Events()[len(ctx.EventManager().Events())-1].Attributes[0].Value = "invalid"
+	err = bh.AfterBalanceChange(ctx, stateDB)
+	require.Error(t, err)
+
+	// reset events
+	ctx = ctx.WithEventManager(sdk.NewEventManager())
+	bh.BeforeBalanceChange(ctx)
+
+	// invalid amount
+	ev := sdk.NewEvent(banktypes.EventTypeCoinSpent,
+		sdk.NewAttribute(banktypes.AttributeKeySpender, addr.String()),
+		sdk.NewAttribute(sdk.AttributeKeyAmount, "invalid"))
+	ctx.EventManager().EmitEvent(ev)
+	err = bh.AfterBalanceChange(ctx, stateDB)
+	require.Error(t, err)
+}