feat: benchmark blob validation (#6355)

crates/primitives/Cargo.toml

@@ -117,6 +117,10 @@ test-utils = ["dep:plain_hasher", "dep:hash-db", "dep:ethers-core"]
 name = "recover_ecdsa_crit"
 harness = false
 
+[[bench]]
+name = "validate_blob_tx"
+harness = false
+
 [[bench]]
 name = "trie_root"
 required-features = ["arbitrary", "test-utils"]

crates/primitives/benches/validate_blob_tx.rs

@@ -0,0 +1,90 @@
+#![allow(missing_docs)]
+use std::sync::Arc;
+
+use alloy_primitives::hex;
+use c_kzg::{KzgCommitment, KzgSettings};
+use criterion::{
+    criterion_group, criterion_main, measurement::WallTime, BenchmarkGroup, Criterion,
+};
+use proptest::{
+    prelude::*,
+    strategy::ValueTree,
+    test_runner::{RngAlgorithm, TestRng, TestRunner},
+};
+use reth_primitives::{
+    constants::eip4844::MAINNET_KZG_TRUSTED_SETUP, eip4844::kzg_to_versioned_hash,
+    BlobTransactionSidecar, TxEip4844,
+};
+use revm_primitives::MAX_BLOB_NUMBER_PER_BLOCK;
+
+// constant seed to use for the rng
+const SEED: [u8; 32] = hex!("1337133713371337133713371337133713371337133713371337133713371337");
+
+/// Benchmarks EIP-48444 blob validation.
+fn blob_validation(c: &mut Criterion) {
+    let mut group = c.benchmark_group("Blob Transaction KZG validation");
+    let kzg_settings = MAINNET_KZG_TRUSTED_SETUP.clone();
+
+    for num_blobs in 1..=MAX_BLOB_NUMBER_PER_BLOCK {
+        println!("Benchmarking validation for tx with {num_blobs} blobs");
+        validate_blob_tx(&mut group, "ValidateBlob", num_blobs, kzg_settings.clone());
+    }
+}
+
+fn validate_blob_tx(
+    group: &mut BenchmarkGroup<'_, WallTime>,
+    description: &str,
+    num_blobs: u64,
+    kzg_settings: Arc<KzgSettings>,
+) {
+    let setup = || {
+        let config = ProptestConfig::default();
+        let rng = TestRng::from_seed(RngAlgorithm::ChaCha, &SEED);
+        let mut runner = TestRunner::new_with_rng(config, rng);
+
+        // generate tx and sidecar
+        let mut tx = any::<TxEip4844>().new_tree(&mut runner).unwrap().current();
+        let mut blob_sidecar =
+            any::<BlobTransactionSidecar>().new_tree(&mut runner).unwrap().current();
+
+        while blob_sidecar.blobs.len() < num_blobs as usize {
+            let blob_sidecar_ext =
+                any::<BlobTransactionSidecar>().new_tree(&mut runner).unwrap().current();
+
+            // extend the sidecar with the new blobs
+            blob_sidecar.blobs.extend(blob_sidecar_ext.blobs);
+            blob_sidecar.proofs.extend(blob_sidecar_ext.proofs);
+            blob_sidecar.commitments.extend(blob_sidecar_ext.commitments);
+
+            if blob_sidecar.blobs.len() > num_blobs as usize {
+                blob_sidecar.blobs.truncate(num_blobs as usize);
+                blob_sidecar.proofs.truncate(num_blobs as usize);
+                blob_sidecar.commitments.truncate(num_blobs as usize);
+            }
+        }
+
+        tx.blob_versioned_hashes = blob_sidecar
+            .commitments
+            .iter()
+            .map(|commitment| {
+                kzg_to_versioned_hash(KzgCommitment::from_bytes(&commitment.into_inner()).unwrap())
+            })
+            .collect();
+
+        (tx, blob_sidecar)
+    };
+
+    let group_id = format!("validate_blob | num blobs: {} | {}", num_blobs, description,);
+
+    // for now we just use the default SubPoolLimit
+    group.bench_function(group_id, |b| {
+        b.iter_with_setup(setup, |(tx, blob_sidecar)| {
+            if let Err(err) = std::hint::black_box(tx.validate_blob(&blob_sidecar, &kzg_settings)) {
+                println!("Validation failed: {:?}", err);
+            }
+        });
+    });
+}
+
+criterion_group!(validate_blob, blob_validation);
+criterion_main!(validate_blob);