anvil/eth/backend/
genesis.rs

1//! Genesis settings
2
3use crate::eth::backend::db::Db;
4use alloy_genesis::{Genesis, GenesisAccount};
5use alloy_primitives::{Address, U256};
6use foundry_evm::backend::DatabaseResult;
7use revm::{bytecode::Bytecode, primitives::KECCAK_EMPTY, state::AccountInfo};
8use tokio::sync::RwLockWriteGuard;
9
10/// Genesis settings
11#[derive(Clone, Debug, Default)]
12pub struct GenesisConfig {
13    /// The initial number for the genesis block
14    pub number: u64,
15    /// The initial timestamp for the genesis block
16    pub timestamp: u64,
17    /// Balance for genesis accounts
18    pub balance: U256,
19    /// All accounts that should be initialised at genesis
20    pub accounts: Vec<Address>,
21    /// The `genesis.json` if provided
22    pub genesis_init: Option<Genesis>,
23}
24
25impl GenesisConfig {
26    /// Returns fresh `AccountInfo`s for the configured `accounts`
27    pub fn account_infos(&self) -> impl Iterator<Item = (Address, AccountInfo)> + '_ {
28        self.accounts.iter().copied().map(|address| {
29            let info = AccountInfo {
30                balance: self.balance,
31                code_hash: KECCAK_EMPTY,
32                // we set this to empty so `Database::code_by_hash` doesn't get called
33                code: Some(Default::default()),
34                nonce: 0,
35                account_id: None,
36            };
37            (address, info)
38        })
39    }
40
41    /// If an initial `genesis.json` was provided, this applies the account alloc to the db
42    pub fn apply_genesis_json_alloc(
43        &self,
44        mut db: RwLockWriteGuard<'_, Box<dyn Db>>,
45    ) -> DatabaseResult<()> {
46        if let Some(ref genesis) = self.genesis_init {
47            for (addr, mut acc) in genesis.alloc.clone() {
48                let storage = std::mem::take(&mut acc.storage);
49                // insert all accounts
50                db.insert_account(addr, self.genesis_to_account_info(&acc));
51                // insert all storage values
52                for (k, v) in &storage.unwrap_or_default() {
53                    db.set_storage_at(addr, *k, *v)?;
54                }
55            }
56        }
57        Ok(())
58    }
59
60    /// Converts a [`GenesisAccount`] to an [`AccountInfo`]
61    fn genesis_to_account_info(&self, acc: &GenesisAccount) -> AccountInfo {
62        let GenesisAccount { code, balance, nonce, .. } = acc.clone();
63        let code = code.map(Bytecode::new_raw);
64        AccountInfo {
65            balance,
66            nonce: nonce.unwrap_or_default(),
67            code_hash: code.as_ref().map(|code| code.hash_slow()).unwrap_or(KECCAK_EMPTY),
68            code,
69            account_id: None,
70        }
71    }
72}
anvil/eth/backend/genesis.rs

anvil/eth/backend/
genesis.rs
