add immutable

Author: pgherveou

crates/anvil-polkadot/src/api_server/server.rs

@@ -359,6 +359,9 @@ impl ApiServer {
             EthRequest::SetStorageAt(address, key, value) => {
                 self.set_storage_at(address, key, value).to_rpc_result()
             }
+            EthRequest::SetImmutableStorageAt(address, data) => {
+                self.set_immutable_storage_at(address, data).to_rpc_result()
+            }
             EthRequest::SetChainId(chain_id) => self.set_chain_id(chain_id).to_rpc_result(),
             // --- Revert ---
             EthRequest::EvmSnapshot(()) => self.snapshot().await.to_rpc_result(),
@@ -1316,6 +1319,20 @@ impl ApiServer {
         Ok(())
     }
 
+    fn set_immutable_storage_at(
+        &self,
+        address: Address,
+        data: alloy_primitives::Bytes,
+    ) -> Result<()> {
+        node_info!("anvil_setImmutableStorageAt");
+
+        let latest_block = self.latest_block();
+
+        self.backend.inject_immutable_data(latest_block, address, data.to_vec());
+
+        Ok(())
+    }
+
     // ----- Wallet RPCs
     async fn sign(&self, address: Address, content: impl AsRef<[u8]>) -> Result<String> {
         node_info!("eth_sign");

crates/anvil-polkadot/src/substrate_node/genesis.rs

@@ -9,20 +9,20 @@ use alloy_primitives::{Address, U256};
 use codec::Encode;
 use polkadot_sdk::{
     pallet_revive::{evm::Account, genesis::ContractData},
-    sc_chain_spec::{BuildGenesisBlock, resolve_state_version_from_wasm},
-    sc_client_api::{BlockImportOperation, backend::Backend},
+    sc_chain_spec::{resolve_state_version_from_wasm, BuildGenesisBlock},
+    sc_client_api::{backend::Backend, BlockImportOperation},
     sc_executor::RuntimeVersionOf,
     sp_blockchain,
-    sp_core::{self, H160, storage::Storage},
+    sp_core::{self, storage::Storage, H160},
     sp_runtime::{
-        BuildStorage, FixedU128,
         traits::{Block as BlockT, Hash as HashT, HashingFor, Header as HeaderT},
+        BuildStorage, FixedU128,
     },
 };
 use serde::{Deserialize, Serialize};
-use serde_json::{Value, json};
+use serde_json::{json, Value};
 use std::{collections::BTreeMap, marker::PhantomData, sync::Arc};
-use substrate_runtime::{WASM_BINARY, constants::NATIVE_TO_ETH_RATIO};
+use substrate_runtime::{constants::NATIVE_TO_ETH_RATIO, WASM_BINARY};
 use subxt_signer::eth::Keypair;
 
 /// Genesis settings
@@ -116,7 +116,13 @@ impl GenesisConfig {
                 let genesis_nonce: u64 = account.nonce.unwrap_or_default();
                 let contract_data: Option<ContractData> = if account.code.is_some() {
                     Some(ContractData {
-                        code: account.code.clone().map(|code| code.to_vec()).unwrap_or_default(),
+                        code: account
+                            .code
+                            .clone()
+                            .map(|code| code.to_vec())
+                            .unwrap_or_default()
+                            .into(),
+                        immutable_data: Default::default(),
                         storage: account
                             .storage
                             .clone()

crates/anvil-polkadot/src/substrate_node/service/backend.rs

@@ -207,6 +207,11 @@ impl BackendWithOverlay {
         overrides.set_code_info(at, code_hash, code_info);
     }
 
+    pub fn inject_immutable_data(&self, at: Hash, address: Address, data: Vec<u8>) {
+        let mut overrides = self.overrides.lock();
+        overrides.set_immutable_data(at, address, data);
+    }
+
     pub fn inject_child_storage(
         &self,
         at: Hash,
@@ -353,6 +358,16 @@ impl StorageOverrides {
         self.add(latest_block, changeset);
     }
 
+    fn set_immutable_data(&mut self, latest_block: Hash, address: Address, data: Vec<u8>) {
+        let mut changeset = BlockOverrides::default();
+        changeset.top.insert(
+            well_known_keys::immutable_data_of(H160::from_slice(address.as_slice())),
+            Some(data.encode()),
+        );
+
+        self.add(latest_block, changeset);
+    }
+
     fn set_child_storage(
         &mut self,
         latest_block: Hash,

crates/anvil-polkadot/src/substrate_node/service/storage.rs

@@ -90,4 +90,13 @@ pub mod well_known_keys {
 
         key
     }
+
+    pub fn immutable_data_of(address: H160) -> Vec<u8> {
+        let mut key = Vec::new();
+        key.extend_from_slice(&twox_128("Revive".as_bytes()));
+        key.extend_from_slice(&twox_128("ImmutableDataOf".as_bytes()));
+        key.extend_from_slice(&address.encode());
+
+        key
+    }
 }

crates/anvil/core/src/eth/mod.rs

@@ -428,6 +428,14 @@ pub enum EthRequest {
         B256,
     ),
 
+    /// Sets the immutable data associated with a contract address
+    #[serde(rename = "anvil_setImmutableStorageAt")]
+    SetImmutableStorageAt(
+        Address,
+        /// immutable data
+        Bytes,
+    ),
+
     /// Sets the coinbase address
     #[serde(rename = "anvil_setCoinbase", alias = "hardhat_setCoinbase", with = "sequence")]
     SetCoinbase(Address),

crates/anvil/src/eth/api.rs

@@ -392,6 +392,11 @@ impl EthApi {
             EthRequest::SetStorageAt(addr, slot, val) => {
                 self.anvil_set_storage_at(addr, slot, val).await.to_rpc_result()
             }
+            EthRequest::SetImmutableStorageAt(_, _) => {
+                return ResponseResult::Error(RpcError::invalid_params(
+                    "anvil_setImmutableStorageAt is not supported on EVM anvil",
+                ));
+            }
             EthRequest::SetCoinbase(addr) => self.anvil_set_coinbase(addr).await.to_rpc_result(),
             EthRequest::EthCoinbase(()) => self.author().to_rpc_result(),
             EthRequest::SetChainId(id) => self.anvil_set_chain_id(id).await.to_rpc_result(),