Add security_bits function to PcsConfig

crates/prover/src/core/fri.rs

@@ -72,6 +72,10 @@ impl FriConfig {
     const fn last_layer_domain_size(&self) -> usize {
         1 << (self.log_last_layer_degree_bound + self.log_blowup_factor)
     }
+
+    pub const fn security_bits(&self) -> u32 {
+        self.log_blowup_factor * self.n_queries as u32
+    }
 }
 
 pub trait FriOps: ColumnOps<BaseField> + PolyOps + Sized + ColumnOps<SecureField> {

crates/prover/src/core/pcs/mod.rs

@@ -33,6 +33,12 @@ pub struct PcsConfig {
     pub pow_bits: u32,
     pub fri_config: FriConfig,
 }
+impl PcsConfig {
+    pub const fn security_bits(&self) -> u32 {
+        self.pow_bits + self.fri_config.security_bits()
+    }
+}
+
 impl Default for PcsConfig {
     fn default() -> Self {
         Self {
@@ -41,3 +47,16 @@ impl Default for PcsConfig {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    #[test]
+    fn test_security_bits() {
+        let config = super::PcsConfig {
+            pow_bits: 42,
+            fri_config: super::FriConfig::new(10, 10, 70),
+        };
+        // 10 * 70 + 42 = 742
+        assert!(config.security_bits() == 742);
+    }
+}

crates/prover/src/examples/blake/air.rs

@@ -509,8 +509,10 @@ pub fn verify_blake<MC: MerkleChannel>(
         stark_proof,
     }: BlakeProof<MC::H>,
 ) -> Result<(), VerificationError> {
-    let channel = &mut MC::C::default();
     // TODO(alonf): Consider mixing the config into the channel.
+    let channel = &mut MC::C::default();
+    const REQUIRED_SECURITY_BITS: u32 = 5;
+    assert!(stark_proof.config.security_bits() >= REQUIRED_SECURITY_BITS);
     let commitment_scheme = &mut CommitmentSchemeVerifier::<MC>::new(stark_proof.config);
 
     let log_sizes = stmt0.log_sizes();