Merge pull request #2769 from subspace/bundle-weight-host-func

Introduce the `bundle_weight` host function

crates/sp-domains-fraud-proof/Cargo.toml

@@ -34,6 +34,7 @@ sp-state-machine = { optional = true, git = "https://github.com/subspace/polkado
 sp-subspace-mmr = { version = "0.1.0", default-features = false, path = "../sp-subspace-mmr" }
 sp-std = { default-features = false, git = "https://github.com/subspace/polkadot-sdk", rev = "9b8cdb87de8f1c0e6b48c468b6196d1d99eeabee" }
 sp-trie = { default-features = false, git = "https://github.com/subspace/polkadot-sdk", rev = "9b8cdb87de8f1c0e6b48c468b6196d1d99eeabee" }
+sp-weights = { default-features = false, git = "https://github.com/subspace/polkadot-sdk", rev = "9b8cdb87de8f1c0e6b48c468b6196d1d99eeabee" }
 subspace-core-primitives = { version = "0.1.0", default-features = false, path = "../subspace-core-primitives" }
 subspace-runtime-primitives = { version = "0.1.0", default-features = false, path = "../subspace-runtime-primitives" }
 trie-db = { version = "0.28.0", default-features = false }
@@ -93,6 +94,7 @@ std = [
     "sp-state-machine/std",
     "sp-subspace-mmr/std",
     "sp-trie/std",
+    "sp-weights/std",
     "subspace-core-primitives/std",
     "subspace-runtime-primitives/std",
     "trie-db/std",

crates/sp-domains-fraud-proof/src/host_functions.rs

@@ -30,6 +30,7 @@ use sp_runtime::traits::{Block as BlockT, Hash as HashT, Header as HeaderT, Numb
 use sp_runtime::OpaqueExtrinsic;
 use sp_state_machine::{OverlayedChanges, StateMachine, TrieBackend, TrieBackendBuilder};
 use sp_trie::{MemoryDB, StorageProof};
+use sp_weights::Weight;
 use std::borrow::Cow;
 use std::marker::PhantomData;
 use std::sync::Arc;
@@ -108,6 +109,12 @@ pub trait FraudProofHostFunctions: Send + Sync {
         domain_runtime_code: Vec<u8>,
         call: StatelessDomainRuntimeCall,
     ) -> Option<bool>;
+
+    fn bundle_weight(
+        &self,
+        domain_runtime_code: Vec<u8>,
+        bundle_body: Vec<OpaqueExtrinsic>,
+    ) -> Option<Weight>;
 }
 
 sp_externalities::decl_extension! {
@@ -826,6 +833,27 @@ where
             )),
         }
     }
+
+    fn bundle_weight(
+        &self,
+        domain_runtime_code: Vec<u8>,
+        bundle_body: Vec<OpaqueExtrinsic>,
+    ) -> Option<Weight> {
+        let domain_stateless_runtime = StatelessRuntime::<DomainBlock, _>::new(
+            self.executor.clone(),
+            domain_runtime_code.into(),
+        );
+        let mut estimated_bundle_weight = Weight::default();
+        for opaque_extrinsic in bundle_body {
+            let encoded_extrinsic = opaque_extrinsic.encode();
+            let extrinsic =
+                <DomainBlock as BlockT>::Extrinsic::decode(&mut encoded_extrinsic.as_slice())
+                    .ok()?;
+            let tx_weight = domain_stateless_runtime.extrinsic_weight(&extrinsic).ok()?;
+            estimated_bundle_weight = estimated_bundle_weight.saturating_add(tx_weight);
+        }
+        Some(estimated_bundle_weight)
+    }
 }
 
 type CreateProofCheckBackedResult<H> = Result<

crates/sp-domains-fraud-proof/src/runtime_interface.rs

@@ -17,6 +17,7 @@ use sp_domains::DomainId;
 use sp_externalities::ExternalitiesExt;
 use sp_runtime::OpaqueExtrinsic;
 use sp_runtime_interface::runtime_interface;
+use sp_weights::Weight;
 
 /// Domain fraud proof related runtime interface
 #[runtime_interface]
@@ -158,4 +159,15 @@ pub trait FraudProofRuntimeInterface {
             .expect("No `FraudProofExtension` associated for the current context!")
             .domain_runtime_call(domain_runtime_code, call)
     }
+
+    #[version(1, register_only)]
+    fn bundle_weight(
+        &mut self,
+        domain_runtime_code: Vec<u8>,
+        bundle_body: Vec<OpaqueExtrinsic>,
+    ) -> Option<Weight> {
+        self.extension::<FraudProofExtension>()
+            .expect("No `FraudProofExtension` associated for the current context!")
+            .bundle_weight(domain_runtime_code, bundle_body)
+    }
 }

domains/client/block-preprocessor/Cargo.toml

@@ -30,6 +30,7 @@ sp-messenger = { version = "0.1.0", path = "../../primitives/messenger" }
 sp-runtime = { git = "https://github.com/subspace/polkadot-sdk", rev = "9b8cdb87de8f1c0e6b48c468b6196d1d99eeabee" }
 sp-state-machine = { git = "https://github.com/subspace/polkadot-sdk", rev = "9b8cdb87de8f1c0e6b48c468b6196d1d99eeabee" }
 sp-timestamp = { git = "https://github.com/subspace/polkadot-sdk", rev = "9b8cdb87de8f1c0e6b48c468b6196d1d99eeabee" }
+sp-weights = { git = "https://github.com/subspace/polkadot-sdk", rev = "9b8cdb87de8f1c0e6b48c468b6196d1d99eeabee" }
 sp-version = { git = "https://github.com/subspace/polkadot-sdk", rev = "9b8cdb87de8f1c0e6b48c468b6196d1d99eeabee" }
 subspace-core-primitives = { version = "0.1.0", path = "../../../crates/subspace-core-primitives" }
 subspace-runtime-primitives = { version = "0.1.0", path = "../../../crates/subspace-runtime-primitives" }

domains/client/block-preprocessor/src/stateless_runtime.rs

@@ -14,6 +14,7 @@ use sp_runtime::traits::{Block as BlockT, NumberFor};
 use sp_runtime::Storage;
 use sp_state_machine::BasicExternalities;
 use sp_version::RuntimeVersion;
+use sp_weights::Weight;
 use std::borrow::Cow;
 use std::marker::PhantomData;
 use std::sync::Arc;
@@ -310,4 +311,11 @@ where
             Ok(sp_domains::operator_block_fees_final_key())
         }
     }
+
+    pub fn extrinsic_weight(
+        &self,
+        extrinsic: &<Block as BlockT>::Extrinsic,
+    ) -> Result<Weight, ApiError> {
+        <Self as DomainCoreApi<Block>>::extrinsic_weight(self, Default::default(), extrinsic)
+    }
 }