97 lines
2.5 KiB
Go
97 lines
2.5 KiB
Go
package evm
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
|
|
ethermint "github.com/evmos/ethermint/types"
|
|
"github.com/evmos/ethermint/x/evm/keeper"
|
|
"github.com/evmos/ethermint/x/evm/types"
|
|
)
|
|
|
|
// InitGenesis initializes genesis state based on exported genesis
|
|
func InitGenesis(
|
|
ctx sdk.Context,
|
|
k *keeper.Keeper,
|
|
accountKeeper types.AccountKeeper,
|
|
data types.GenesisState,
|
|
) []abci.ValidatorUpdate {
|
|
k.WithChainID(ctx)
|
|
|
|
k.SetParams(ctx, data.Params)
|
|
|
|
// ensure evm module account is set
|
|
if addr := accountKeeper.GetModuleAddress(types.ModuleName); addr == nil {
|
|
panic("the EVM module account has not been set")
|
|
}
|
|
|
|
for _, account := range data.Accounts {
|
|
address := common.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))
|
|
}
|
|
|
|
ethAcct, ok := acc.(ethermint.EthAccountI)
|
|
if !ok {
|
|
panic(
|
|
fmt.Errorf("account %s must be an EthAccount interface, got %T",
|
|
account.Address, acc,
|
|
),
|
|
)
|
|
}
|
|
|
|
code := common.Hex2Bytes(account.Code)
|
|
codeHash := crypto.Keccak256Hash(code)
|
|
if !bytes.Equal(ethAcct.GetCodeHash().Bytes(), codeHash.Bytes()) {
|
|
panic("code don't match codeHash")
|
|
}
|
|
|
|
k.SetCode(ctx, codeHash.Bytes(), code)
|
|
|
|
for _, storage := range account.Storage {
|
|
k.SetState(ctx, address, common.HexToHash(storage.Key), common.HexToHash(storage.Value).Bytes())
|
|
}
|
|
}
|
|
|
|
return []abci.ValidatorUpdate{}
|
|
}
|
|
|
|
// ExportGenesis exports genesis state of the EVM module
|
|
func ExportGenesis(ctx sdk.Context, k *keeper.Keeper, ak types.AccountKeeper) *types.GenesisState {
|
|
var ethGenAccounts []types.GenesisAccount
|
|
ak.IterateAccounts(ctx, func(account authtypes.AccountI) bool {
|
|
ethAccount, ok := account.(ethermint.EthAccountI)
|
|
if !ok {
|
|
// ignore non EthAccounts
|
|
return false
|
|
}
|
|
|
|
addr := ethAccount.EthAddress()
|
|
|
|
storage := k.GetAccountStorage(ctx, addr)
|
|
|
|
genAccount := types.GenesisAccount{
|
|
Address: addr.String(),
|
|
Code: common.Bytes2Hex(k.GetCode(ctx, ethAccount.GetCodeHash())),
|
|
Storage: storage,
|
|
}
|
|
|
|
ethGenAccounts = append(ethGenAccounts, genAccount)
|
|
return false
|
|
})
|
|
|
|
return &types.GenesisState{
|
|
Accounts: ethGenAccounts,
|
|
Params: k.GetParams(ctx),
|
|
}
|
|
}
|