eoa forking

primitivefinance · Oct 16, 2023 · cb296a6 · cb296a6
1 parent 9af3560
commit cb296a6
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/bin/fork/mod.rs b/bin/fork/mod.rs
@@ -30,6 +30,7 @@ pub(crate) struct ForkConfig {
     block_number: u64,
     #[serde(rename = "contracts")]
     contracts_meta: HashMap<String, ContractMetadata>,
+    externally_owned_accounts: Vec<Address>,
 }
 
 impl ForkConfig {
@@ -42,7 +43,9 @@ impl ForkConfig {
                     .ok_or(ConfigError::NotFound("File not found!".to_owned()))?,
             ))
             .build()?;
+        println!("before");
         let mut fork_config: ForkConfig = config.try_deserialize()?;
+        println!("after");
 
         if fork_config.output_directory.is_none() {
             println!("No output path specified. Defaulting to current directory.");
@@ -84,6 +87,22 @@ impl ForkConfig {
             let storage_layout = artifacts.storage_layout;
 
             digest::create_storage_layout(contract_data, storage_layout, &mut db, ethers_db)?;
+
+            for eoa in &self.externally_owned_accounts {
+                let info = ethers_db
+                    .basic(eoa.to_fixed_bytes().into())
+                    .map_err(|_| {
+                        ArbiterError::DBError(
+                            "Failed to fetch account info with
+                EthersDB."
+                                .to_string(),
+                        )
+                    })?
+                    .ok_or(ArbiterError::DBError(
+                        "Failed to fetch account info with EthersDB.".to_string(),
+                    ))?;
+                db.insert_account_info(eoa.to_fixed_bytes().into(), info);
+            }
         }
         Ok(db)
     }

diff --git a/example_fork/weth_config.toml b/example_fork/weth_config.toml
@@ -4,6 +4,9 @@ output_filename = "test.json"
 provider = "https://eth.llamarpc.com"
 block_number = 18228556
 
+# EOAs
+externally_owned_accounts = ["0xd8dA6BF26964aF9D7eEd9e03E53415D37aA96045"] # vitalik.eth as of 10/2/2023
+
 # Contracts stored in a mapping
 # Try this out with the weth contract
 [contracts.weth]
@@ -13,4 +16,4 @@ artifacts_path = "example_fork/WETH.json"
 [contracts.weth.mappings]
 balanceOf = [
   "0xd8dA6BF26964aF9D7eEd9e03E53415D37aA96045", # vitalik.eth as of 10/2/2023
-]
+]