diff --git a/crates/prover/src/core/air/air_ext.rs b/crates/prover/src/core/air/air_ext.rs index abc47b960..0a9c091c2 100644 --- a/crates/prover/src/core/air/air_ext.rs +++ b/crates/prover/src/core/air/air_ext.rs @@ -8,6 +8,7 @@ use crate::core::fields::qm31::SecureField; use crate::core::fields::secure_column::SECURE_EXTENSION_DEGREE; use crate::core::pcs::{CommitmentTreeProver, TreeVec}; use crate::core::poly::circle::SecureCirclePoly; +use crate::core::prover::VerificationError; use crate::core::vcs::blake2_merkle::Blake2sMerkleHasher; use crate::core::vcs::ops::MerkleOps; use crate::core::{ColumnVec, InteractionElements, LookupValues}; @@ -125,6 +126,12 @@ pub trait AirExt: Air { }) .collect_vec() } + + fn verify_lookups(&self, lookup_values: &LookupValues) -> Result<(), VerificationError> { + self.components() + .iter() + .try_for_each(|component| component.verify_lookups(lookup_values)) + } } impl AirExt for A {} diff --git a/crates/prover/src/core/air/mod.rs b/crates/prover/src/core/air/mod.rs index 403d5ce90..1bcf52145 100644 --- a/crates/prover/src/core/air/mod.rs +++ b/crates/prover/src/core/air/mod.rs @@ -6,6 +6,7 @@ use super::fields::qm31::SecureField; use super::pcs::TreeVec; use super::poly::circle::{CircleEvaluation, CirclePoly}; use super::poly::BitReversedOrder; +use super::prover::VerificationError; use super::{ColumnVec, InteractionElements, LookupValues}; use crate::trace_generation::AirTraceVerifier; @@ -62,6 +63,9 @@ pub trait Component { interaction_elements: &InteractionElements, lookup_values: &LookupValues, ); + + /// Verifies the lookups used by the component. + fn verify_lookups(&self, lookup_values: &LookupValues) -> Result<(), VerificationError>; } pub trait ComponentProver: Component { diff --git a/crates/prover/src/core/prover/mod.rs b/crates/prover/src/core/prover/mod.rs index 7f58a97fd..9e61f9e3f 100644 --- a/crates/prover/src/core/prover/mod.rs +++ b/crates/prover/src/core/prover/mod.rs @@ -223,6 +223,8 @@ pub fn verify( ); } + air.verify_lookups(&proof.lookup_values)?; + channel.mix_felts( &proof .lookup_values @@ -331,6 +333,8 @@ pub enum ProvingError { pub enum VerificationError { #[error("Proof has invalid structure: {0}.")] InvalidStructure(String), + #[error("{0} lookup values do not match.")] + InvalidLookup(String), #[error(transparent)] Merkle(#[from] MerkleVerificationError), #[error( @@ -348,6 +352,7 @@ pub enum VerificationError { mod tests { use num_traits::Zero; + use super::VerificationError; use crate::core::air::accumulation::{DomainEvaluationAccumulator, PointEvaluationAccumulator}; use crate::core::air::{Air, AirProver, Component, ComponentProver, ComponentTrace}; use crate::core::backend::cpu::CpuCircleEvaluation; @@ -448,6 +453,10 @@ mod tests { ) { evaluation_accumulator.accumulate(qm31!(0, 0, 0, 1)) } + + fn verify_lookups(&self, _lookup_values: &LookupValues) -> Result<(), VerificationError> { + Ok(()) + } } impl ComponentTraceGenerator for TestComponent { diff --git a/crates/prover/src/examples/fibonacci/component.rs b/crates/prover/src/examples/fibonacci/component.rs index 2da6da918..ba14a5b80 100644 --- a/crates/prover/src/examples/fibonacci/component.rs +++ b/crates/prover/src/examples/fibonacci/component.rs @@ -14,7 +14,7 @@ use crate::core::fields::{ExtensionOf, FieldExpOps}; use crate::core::pcs::TreeVec; use crate::core::poly::circle::{CanonicCoset, CircleEvaluation}; use crate::core::poly::BitReversedOrder; -use crate::core::prover::BASE_TRACE; +use crate::core::prover::{VerificationError, BASE_TRACE}; use crate::core::utils::bit_reverse_index; use crate::core::{ColumnVec, InteractionElements, LookupValues}; use crate::trace_generation::registry::ComponentGenerationRegistry; @@ -124,6 +124,10 @@ impl Component for FibonacciComponent { &mask[0][0][..1].try_into().unwrap(), )); } + + fn verify_lookups(&self, _lookup_values: &LookupValues) -> Result<(), VerificationError> { + Ok(()) + } } impl ComponentTraceGenerator for FibonacciComponent { diff --git a/crates/prover/src/examples/poseidon/mod.rs b/crates/prover/src/examples/poseidon/mod.rs index a111338da..78a4fb387 100644 --- a/crates/prover/src/examples/poseidon/mod.rs +++ b/crates/prover/src/examples/poseidon/mod.rs @@ -23,6 +23,7 @@ use crate::core::fields::{FieldExpOps, FieldOps}; use crate::core::pcs::TreeVec; use crate::core::poly::circle::{CanonicCoset, CircleEvaluation, PolyOps}; use crate::core::poly::BitReversedOrder; +use crate::core::prover::VerificationError; use crate::core::{ColumnVec, InteractionElements, LookupValues}; use crate::trace_generation::{AirTraceGenerator, AirTraceVerifier, ComponentTraceGenerator}; @@ -138,6 +139,10 @@ impl Component for PoseidonComponent { } assert_eq!(eval.col_index, N_COLUMNS); } + + fn verify_lookups(&self, _lookup_values: &LookupValues) -> Result<(), VerificationError> { + Ok(()) + } } #[inline(always)] diff --git a/crates/prover/src/examples/wide_fibonacci/component.rs b/crates/prover/src/examples/wide_fibonacci/component.rs index 2294e38a2..2fe6ba4f9 100644 --- a/crates/prover/src/examples/wide_fibonacci/component.rs +++ b/crates/prover/src/examples/wide_fibonacci/component.rs @@ -14,6 +14,7 @@ use crate::core::fields::FieldExpOps; use crate::core::pcs::TreeVec; use crate::core::poly::circle::{CanonicCoset, CircleEvaluation, SecureCirclePoly}; use crate::core::poly::BitReversedOrder; +use crate::core::prover::VerificationError; use crate::core::utils::shifted_secure_combination; use crate::core::{ColumnVec, InteractionElements, LookupValues}; use crate::examples::wide_fibonacci::trace_gen::write_lookup_column; @@ -259,6 +260,10 @@ impl Component for WideFibComponent { constraint_zero_domain, ); } + + fn verify_lookups(&self, _lookup_values: &LookupValues) -> Result<(), VerificationError> { + Ok(()) + } } impl ComponentTraceGenerator for WideFibComponent { diff --git a/crates/prover/src/examples/wide_fibonacci/simd.rs b/crates/prover/src/examples/wide_fibonacci/simd.rs index bbc7774b9..f0aa7b671 100644 --- a/crates/prover/src/examples/wide_fibonacci/simd.rs +++ b/crates/prover/src/examples/wide_fibonacci/simd.rs @@ -20,7 +20,7 @@ use crate::core::fields::{FieldExpOps, FieldOps}; use crate::core::pcs::TreeVec; use crate::core::poly::circle::{CanonicCoset, CircleEvaluation}; use crate::core::poly::BitReversedOrder; -use crate::core::prover::BASE_TRACE; +use crate::core::prover::{VerificationError, BASE_TRACE}; use crate::core::{ColumnVec, InteractionElements, LookupValues}; use crate::examples::wide_fibonacci::component::{ALPHA_ID, N_COLUMNS, Z_ID}; use crate::trace_generation::registry::ComponentGenerationRegistry; @@ -127,6 +127,10 @@ impl Component for SimdWideFibComponent { evaluation_accumulator.accumulate(numerator * denom_inverse); } } + + fn verify_lookups(&self, _lookup_values: &LookupValues) -> Result<(), VerificationError> { + Ok(()) + } } impl AirProver for SimdWideFibAir { diff --git a/crates/prover/src/trace_generation/registry.rs b/crates/prover/src/trace_generation/registry.rs index f5ab16654..1ec7971a9 100644 --- a/crates/prover/src/trace_generation/registry.rs +++ b/crates/prover/src/trace_generation/registry.rs @@ -89,6 +89,13 @@ mod tests { ) { todo!() } + + fn verify_lookups( + &self, + _lookup_values: &LookupValues, + ) -> Result<(), crate::core::prover::VerificationError> { + todo!() + } } type ComponentACpuInputs = Vec<(M31, M31)>;