Generate and apply the fraud proof verification weights

Signed-off-by: linning <[email protected]>

Author: NingLin-P

crates/pallet-domains/src/extensions.rs

@@ -8,6 +8,7 @@ use frame_support::weights::Weight;
 use frame_system::pallet_prelude::RuntimeCallFor;
 use parity_scale_codec::{Decode, Encode};
 use scale_info::prelude::fmt;
+use sp_domains_fraud_proof::weights::fraud_proof_verification_weights;
 use sp_domains_fraud_proof::InvalidTransactionCode;
 use sp_runtime::impl_tx_ext_default;
 use sp_runtime::traits::{
@@ -183,8 +184,9 @@ where
             Some(DomainsCall::submit_bundle { .. }) => {
                 <Runtime as Config>::WeightInfo::validate_submit_bundle()
             }
-            Some(DomainsCall::submit_fraud_proof { .. }) => {
+            Some(DomainsCall::submit_fraud_proof { fraud_proof }) => {
                 <Runtime as Config>::WeightInfo::fraud_proof_pre_check()
+                    .saturating_add(fraud_proof_verification_weights::<_, _, _, _>(fraud_proof))
             }
             Some(DomainsCall::submit_receipt { .. }) => {
                 <Runtime as Config>::WeightInfo::validate_singleton_receipt()

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

@@ -16,6 +16,7 @@ pub mod storage_proof;
 #[cfg(test)]
 mod tests;
 pub mod verification;
+pub mod weights;
 
 #[cfg(not(feature = "std"))]
 extern crate alloc;

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

@@ -0,0 +1,132 @@
+use crate::fraud_proof::{FraudProof, FraudProofVariant};
+use sp_domains::InvalidBundleType;
+use sp_runtime::traits::Header as HeaderT;
+use sp_weights::Weight;
+
+pub fn fraud_proof_verification_weights<Number, Hash, DomainHeader: HeaderT, MmrHash>(
+    fp: &FraudProof<Number, Hash, DomainHeader, MmrHash>,
+) -> Weight {
+    let mut weight = match &fp.proof {
+        FraudProofVariant::InvalidStateTransition(_) => {
+            invalid_state_transition_proof_verification()
+        }
+        FraudProofVariant::ValidBundle(_) => valid_bundle_proof_verification(),
+        FraudProofVariant::InvalidExtrinsicsRoot(_) => invalid_domain_extrinsics_root_fraud_proof(),
+        FraudProofVariant::InvalidDomainBlockHash(_) => {
+            invalid_domain_block_hash_fraud_proof_verification()
+        }
+        FraudProofVariant::InvalidBlockFees(_) => invalid_block_fees_fraud_proof_verification(),
+        FraudProofVariant::InvalidTransfers(_) => invalid_transfers_fraud_proof_verification(),
+        FraudProofVariant::InvalidBundles(p) => match p.invalid_bundle_type {
+            InvalidBundleType::UndecodableTx(_) => {
+                invalid_bundle_undecodable_tx_fraud_proof_verification()
+            }
+            InvalidBundleType::OutOfRangeTx(_) => {
+                invalid_bundle_out_of_range_tx_fraud_proof_verification()
+            }
+            InvalidBundleType::IllegalTx(_) => invalid_bundle_illegal_tx_fraud_proof_verification(),
+            InvalidBundleType::InvalidXDM(_) => {
+                invalid_bundle_invalid_xdm_fraud_proof_verification()
+            }
+            InvalidBundleType::InherentExtrinsic(_) => {
+                invalid_bundle_inherent_extrinsic_fraud_proof_verification()
+            }
+            InvalidBundleType::InvalidBundleWeight => {
+                invalid_bundle_weight_fraud_proof_verification()
+            }
+        },
+        #[cfg(any(feature = "std", feature = "runtime-benchmarks"))]
+        FraudProofVariant::Dummy => Weight::zero(),
+    };
+    if fp.maybe_mmr_proof.is_some() {
+        weight = weight.saturating_add(consensus_state_root_mmr_proof_verification());
+    }
+    if fp.maybe_domain_runtime_code_proof.is_some() {
+        weight = weight.saturating_add(domain_runtime_code_proof_verification());
+    }
+    weight
+}
+
+fn consensus_state_root_mmr_proof_verification() -> Weight {
+    // Verification time is around 2_022_700 pico seconds
+    // Generated by the `mmr_proof_and_runtime_code_proof_verification` bench
+    Weight::from_parts(2_022_700, 0)
+}
+
+fn domain_runtime_code_proof_verification() -> Weight {
+    // Verification time is around 2_022_700 pico seconds
+    // Generated by the `mmr_proof_and_runtime_code_proof_verification` bench
+    Weight::from_parts(925_650_000, 0)
+}
+
+fn invalid_state_transition_proof_verification() -> Weight {
+    // Verification time is around 359_040_000 pico seconds
+    // Generated by the `invalid_state_transition_proof_verification` bench
+    Weight::from_parts(359_040_000, 0)
+}
+
+fn valid_bundle_proof_verification() -> Weight {
+    // Verification time is around 754_330_000 pico seconds
+    // Generated by the `valid_bundle_proof_verification` bench
+    Weight::from_parts(754_330_000, 0)
+}
+
+fn invalid_domain_extrinsics_root_fraud_proof() -> Weight {
+    // Verification time is around 1_495_800_000 pico seconds
+    // Generated by the `invalid_domain_extrinsics_root_fraud_proof` bench
+    Weight::from_parts(1_495_800_000, 0)
+}
+
+fn invalid_domain_block_hash_fraud_proof_verification() -> Weight {
+    // Verification time is around 5_929_100 pico seconds
+    // Generated by the `invalid_domain_block_hash_fraud_proof_verification` bench
+    Weight::from_parts(5_929_100, 0)
+}
+
+fn invalid_block_fees_fraud_proof_verification() -> Weight {
+    // Verification time is around 746_860_000 pico seconds
+    // Generated by the `invalid_block_fees_fraud_proof_verification` bench
+    Weight::from_parts(746_860_000, 0)
+}
+
+fn invalid_transfers_fraud_proof_verification() -> Weight {
+    // Verification time is around 744_740_000 pico seconds
+    // Generated by the `invalid_transfers_fraud_proof_verification` bench
+    Weight::from_parts(744_740_000, 0)
+}
+
+fn invalid_bundle_undecodable_tx_fraud_proof_verification() -> Weight {
+    // Verification time is around 751_320_000 pico seconds
+    // Generated by the `invalid_bundle_undecodable_tx_fraud_proof_verification` bench
+    Weight::from_parts(751_320_000, 0)
+}
+
+fn invalid_bundle_out_of_range_tx_fraud_proof_verification() -> Weight {
+    // Verification time is around 745_380_000 pico seconds
+    // Generated by the `invalid_bundle_out_of_range_tx_fraud_proof_verification` bench
+    Weight::from_parts(745_380_000, 0)
+}
+
+fn invalid_bundle_illegal_tx_fraud_proof_verification() -> Weight {
+    // Verification time is around 198_720_000 pico seconds
+    // Generated by the `invalid_bundle_out_of_range_tx_fraud_proof_verification` bench
+    Weight::from_parts(198_720_000, 0)
+}
+
+fn invalid_bundle_invalid_xdm_fraud_proof_verification() -> Weight {
+    // Verification time is around 750_950_000 pico seconds
+    // Generated by the `invalid_bundle_invalid_xdm_fraud_proof_verification` bench
+    Weight::from_parts(750_950_000, 0)
+}
+
+fn invalid_bundle_inherent_extrinsic_fraud_proof_verification() -> Weight {
+    // Verification time is around 746_340_000 pico seconds
+    // Generated by the `invalid_bundle_inherent_extrinsic_fraud_proof_verification` bench
+    Weight::from_parts(746_340_000, 0)
+}
+
+fn invalid_bundle_weight_fraud_proof_verification() -> Weight {
+    // Verification time is around 761_690_000 pico seconds
+    // Generated by the `invalid_bundle_weight_fraud_proof_verification` bench
+    Weight::from_parts(761_690_000, 0)
+}