Merge export genesis

go.mod

@@ -9,7 +9,7 @@ require (
 	github.com/btcsuite/btcd v0.21.0-beta
 	github.com/btcsuite/btcutil v1.0.2
 	github.com/cespare/cp v1.1.1 // indirect
-	github.com/cosmos/cosmos-sdk v0.39.1
+	github.com/cosmos/cosmos-sdk v0.39.2
 	github.com/deckarep/golang-set v1.7.1 // indirect
 	github.com/ethereum/go-ethereum v1.9.24
 	github.com/fjl/memsize v0.0.0-20190710130421-bcb5799ab5e5 // indirect

x/evm/genesis.go

@@ -1,29 +1,61 @@
 package evm
 
 import (
+	"fmt"
+
 	sdk "github.com/cosmos/cosmos-sdk/types"
 
-	emint "github.com/okex/okexchain/app/types"
+	ethcmn "github.com/ethereum/go-ethereum/common"
+
+	ethermint "github.com/okex/okexchain/app/types"
 	"github.com/okex/okexchain/x/evm/types"
 
 	abci "github.com/tendermint/tendermint/abci/types"
 )
 
 // InitGenesis initializes genesis state based on exported genesis
-func InitGenesis(ctx sdk.Context, k Keeper, data GenesisState) []abci.ValidatorUpdate {
+func InitGenesis(ctx sdk.Context, k Keeper, accountKeeper types.AccountKeeper, data GenesisState) []abci.ValidatorUpdate { // nolint: interfacer
+	evmDenom := data.Params.EvmDenom
+
 	for _, account := range data.Accounts {
-		// FIXME: this will override bank InitGenesis balance!
-		k.SetBalance(ctx, account.Address, account.Balance)
-		k.SetCode(ctx, account.Address, account.Code)
+		address := ethcmn.HexToAddress(account.Address)
+		accAddress := sdk.AccAddress(address.Bytes())
+
+		// check that the EVM balance the matches the account balance
+		acc := accountKeeper.GetAccount(ctx, accAddress)
+		if acc == nil {
+			panic(fmt.Errorf("account not found for address %s", account.Address))
+		}
+
+		_, ok := acc.(*ethermint.EthAccount)
+		if !ok {
+			panic(
+				fmt.Errorf("account %s must be an %T type, got %T",
+					account.Address, &ethermint.EthAccount{}, acc,
+				),
+			)
+		}
+
+		evmBalance := acc.GetCoins().AmountOf(evmDenom)
+		if !evmBalance.Equal(account.Balance) {
+			panic(
+				fmt.Errorf(
+					"balance mismatch for account %s, expected %s%s, got %s%s",
+					account.Address, evmBalance, evmDenom, account.Balance, evmDenom,
+				),
+			)
+		}
+
+		k.SetBalance(ctx, address, account.Balance.BigInt())
+		k.SetCode(ctx, address, account.Code)
 		for _, storage := range account.Storage {
-			k.SetState(ctx, account.Address, storage.Key, storage.Value)
+			k.SetState(ctx, address, storage.Key, storage.Value)
 		}
 	}
 
 	var err error
 	for _, txLog := range data.TxsLogs {
-		err = k.SetLogs(ctx, txLog.Hash, txLog.Logs)
-		if err != nil {
+		if err = k.SetLogs(ctx, txLog.Hash, txLog.Logs); err != nil {
 			panic(err)
 		}
 	}
@@ -54,8 +86,7 @@ func ExportGenesis(ctx sdk.Context, k Keeper, ak types.AccountKeeper) GenesisSta
 	accounts := ak.GetAllAccounts(ctx)
 
 	for _, account := range accounts {
-
-		ethAccount, ok := account.(*emint.EthAccount)
+		ethAccount, ok := account.(*ethermint.EthAccount)
 		if !ok {
 			continue
 		}
@@ -67,9 +98,12 @@ func ExportGenesis(ctx sdk.Context, k Keeper, ak types.AccountKeeper) GenesisSta
 			panic(err)
 		}
 
+		balanceInt := k.GetBalance(ctx, addr)
+		balance := sdk.NewDecFromBigIntWithPrec(balanceInt, sdk.Precision)
+
 		genAccount := types.GenesisAccount{
-			Address: addr,
-			Balance: k.GetBalance(ctx, addr),
+			Address: addr.String(),
+			Balance: balance,
 			Code:    k.GetCode(ctx, addr),
 			Storage: storage,
 		}

x/evm/keeper/querier.go

@@ -200,7 +200,8 @@ func queryAccount(ctx sdk.Context, path []string, keeper Keeper) ([]byte, error)
 }
 
 func queryExportAccount(ctx sdk.Context, path []string, keeper Keeper) ([]byte, error) {
-	addr := ethcmn.HexToAddress(path[1])
+	hexAddress := path[1]
+	addr := ethcmn.HexToAddress(hexAddress)
 
 	var storage types.Storage
 	err := keeper.ForEachStorage(ctx, addr, func(key, value ethcmn.Hash) bool {
@@ -211,9 +212,12 @@ func queryExportAccount(ctx sdk.Context, path []string, keeper Keeper) ([]byte,
 		return nil, err
 	}
 
+	balanceInt := keeper.GetBalance(ctx, addr)
+	balance := sdk.NewDecFromBigIntWithPrec(balanceInt, sdk.Precision)
+
 	res := types.GenesisAccount{
-		Address: addr,
-		Balance: keeper.GetBalance(ctx, addr),
+		Address: hexAddress,
+		Balance: balance,
 		Code:    keeper.GetCode(ctx, addr),
 		Storage: storage,
 	}

x/evm/module.go

@@ -124,7 +124,7 @@ func (am AppModule) EndBlock(ctx sdk.Context, req abci.RequestEndBlock) []abci.V
 func (am AppModule) InitGenesis(ctx sdk.Context, data json.RawMessage) []abci.ValidatorUpdate {
 	var genesisState types.GenesisState
 	types.ModuleCdc.MustUnmarshalJSON(data, &genesisState)
-	return InitGenesis(ctx, am.keeper, genesisState)
+	return InitGenesis(ctx, am.keeper, am.ak, genesisState)
 }
 
 // ExportGenesis exports the genesis state to be used by daemon

x/evm/types/genesis.go

@@ -1,10 +1,10 @@
 package types
 
 import (
-	"bytes"
 	"errors"
 	"fmt"
-	"math/big"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
 
 	ethcmn "github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/common/hexutil"
@@ -23,22 +23,22 @@ type (
 	// Its main difference between with Geth's GenesisAccount is that it uses a custom
 	// storage type and that it doesn't contain the private key field.
 	GenesisAccount struct {
-		Address ethcmn.Address `json:"address"`
-		Balance *big.Int       `json:"balance"`
-		Code    hexutil.Bytes  `json:"code,omitempty"`
-		Storage Storage        `json:"storage,omitempty"`
+		Address string        `json:"address"`
+		Balance sdk.Dec       `json:"balance"`
+		Code    hexutil.Bytes `json:"code,omitempty"`
+		Storage Storage       `json:"storage,omitempty"`
 	}
 )
 
 // Validate performs a basic validation of a GenesisAccount fields.
 func (ga GenesisAccount) Validate() error {
-	if bytes.Equal(ga.Address.Bytes(), ethcmn.Address{}.Bytes()) {
-		return fmt.Errorf("address cannot be the zero address %s", ga.Address.String())
+	if ga.Address == (ethcmn.Address{}.String()) {
+		return fmt.Errorf("address cannot be the zero address %s", ga.Address)
 	}
-	if ga.Balance == nil {
+	if ga.Balance.IsNil() {
 		return errors.New("balance cannot be nil")
 	}
-	if ga.Balance.Sign() == -1 {
+	if ga.Balance.IsNegative() {
 		return errors.New("balance cannot be negative")
 	}
 	if ga.Code != nil && len(ga.Code) == 0 {
@@ -65,14 +65,15 @@ func (gs GenesisState) Validate() error {
 	seenAccounts := make(map[string]bool)
 	seenTxs := make(map[string]bool)
 	for _, acc := range gs.Accounts {
-		if seenAccounts[acc.Address.String()] {
-			return fmt.Errorf("duplicated genesis account %s", acc.Address.String())
+		if seenAccounts[acc.Address] {
+			return fmt.Errorf("duplicated genesis account %s", acc.Address)
 		}
 		if err := acc.Validate(); err != nil {
-			return fmt.Errorf("invalid genesis account %s: %w", acc.Address.String(), err)
+			return fmt.Errorf("invalid genesis account %s: %w", acc.Address, err)
 		}
-		seenAccounts[acc.Address.String()] = true
+		seenAccounts[acc.Address] = true
 	}
+
 	for _, tx := range gs.TxsLogs {
 		if seenTxs[tx.Hash.String()] {
 			return fmt.Errorf("duplicated logs from transaction %s", tx.Hash.String())

x/evm/types/genesis_test.go

@@ -1,7 +1,6 @@
 package types
 
 import (
-	"math/big"
 	"testing"
 
 	"github.com/stretchr/testify/require"
@@ -10,9 +9,13 @@ import (
 	ethtypes "github.com/ethereum/go-ethereum/core/types"
 	ethcrypto "github.com/ethereum/go-ethereum/crypto"
 
+	sdk "github.com/cosmos/cosmos-sdk/types"
+
 	"github.com/okex/okexchain/app/crypto/ethsecp256k1"
 )
 
+var address = ethcmn.BytesToAddress([]byte{1, 2, 3, 4, 5})
+
 func TestValidateGenesisAccount(t *testing.T) {
 	testCases := []struct {
 		name           string
@@ -22,8 +25,8 @@ func TestValidateGenesisAccount(t *testing.T) {
 		{
 			"valid genesis account",
 			GenesisAccount{
-				Address: ethcmn.BytesToAddress([]byte{1, 2, 3, 4, 5}),
-				Balance: big.NewInt(1),
+				Address: address.String(),
+				Balance: sdk.NewDec(1),
 				Code:    []byte{1, 2, 3},
 				Storage: Storage{
 					NewState(ethcmn.BytesToHash([]byte{1, 2, 3}), ethcmn.BytesToHash([]byte{1, 2, 3})),
@@ -34,32 +37,32 @@ func TestValidateGenesisAccount(t *testing.T) {
 		{
 			"empty account address bytes",
 			GenesisAccount{
-				Address: ethcmn.Address{},
-				Balance: big.NewInt(1),
+				Address: ethcmn.Address{}.String(),
+				Balance: sdk.NewDec(1),
 			},
 			false,
 		},
 		{
 			"nil account balance",
 			GenesisAccount{
-				Address: ethcmn.BytesToAddress([]byte{1, 2, 3, 4, 5}),
-				Balance: nil,
+				Address: address.String(),
+				Balance: sdk.Dec{},
 			},
 			false,
 		},
 		{
 			"nil account balance",
 			GenesisAccount{
-				Address: ethcmn.BytesToAddress([]byte{1, 2, 3, 4, 5}),
-				Balance: big.NewInt(-1),
+				Address: address.String(),
+				Balance: sdk.NewDec(-1),
 			},
 			false,
 		},
 		{
 			"empty code bytes",
 			GenesisAccount{
-				Address: ethcmn.BytesToAddress([]byte{1, 2, 3, 4, 5}),
-				Balance: big.NewInt(1),
+				Address: address.String(),
+				Balance: sdk.NewDec(1),
 				Code:    []byte{},
 			},
 			false,
@@ -97,8 +100,8 @@ func TestValidateGenesis(t *testing.T) {
 			genState: GenesisState{
 				Accounts: []GenesisAccount{
 					{
-						Address: ethcmn.BytesToAddress([]byte{1, 2, 3, 4, 5}),
-						Balance: big.NewInt(1),
+						Address: address.String(),
+						Balance: sdk.NewDec(1),
 						Code:    []byte{1, 2, 3},
 						Storage: Storage{
 							{Key: ethcmn.BytesToHash([]byte{1, 2, 3})},
@@ -138,7 +141,7 @@ func TestValidateGenesis(t *testing.T) {
 			genState: GenesisState{
 				Accounts: []GenesisAccount{
 					{
-						Address: ethcmn.Address{},
+						Address: ethcmn.Address{}.String(),
 					},
 				},
 			},
@@ -149,16 +152,16 @@ func TestValidateGenesis(t *testing.T) {
 			genState: GenesisState{
 				Accounts: []GenesisAccount{
 					{
-						Address: ethcmn.BytesToAddress([]byte{1, 2, 3, 4, 5}),
-						Balance: big.NewInt(1),
+						Address: address.String(),
+						Balance: sdk.NewDec(1),
 						Code:    []byte{1, 2, 3},
 						Storage: Storage{
 							NewState(ethcmn.BytesToHash([]byte{1, 2, 3}), ethcmn.BytesToHash([]byte{1, 2, 3})),
 						},
 					},
 					{
-						Address: ethcmn.BytesToAddress([]byte{1, 2, 3, 4, 5}),
-						Balance: big.NewInt(1),
+						Address: address.String(),
+						Balance: sdk.NewDec(1),
 						Code:    []byte{1, 2, 3},
 						Storage: Storage{
 							NewState(ethcmn.BytesToHash([]byte{1, 2, 3}), ethcmn.BytesToHash([]byte{1, 2, 3})),
@@ -173,8 +176,8 @@ func TestValidateGenesis(t *testing.T) {
 			genState: GenesisState{
 				Accounts: []GenesisAccount{
 					{
-						Address: ethcmn.BytesToAddress([]byte{1, 2, 3, 4, 5}),
-						Balance: big.NewInt(1),
+						Address: address.String(),
+						Balance: sdk.NewDec(1),
 						Code:    []byte{1, 2, 3},
 						Storage: Storage{
 							{Key: ethcmn.BytesToHash([]byte{1, 2, 3})},
@@ -223,8 +226,8 @@ func TestValidateGenesis(t *testing.T) {
 			genState: GenesisState{
 				Accounts: []GenesisAccount{
 					{
-						Address: ethcmn.BytesToAddress([]byte{1, 2, 3, 4, 5}),
-						Balance: big.NewInt(1),
+						Address: address.String(),
+						Balance: sdk.NewDec(1),
 						Code:    []byte{1, 2, 3},
 						Storage: Storage{
 							{Key: ethcmn.BytesToHash([]byte{1, 2, 3})},