|
| 1 | +use super::{ProofVerifier, TaskType, VKDump}; |
| 2 | + |
| 3 | +use anyhow::Result; |
| 4 | + |
| 5 | +use crate::utils::panic_catch; |
| 6 | +use euclidv2_prover::{BatchProof, BundleProof, ChunkProof}; |
| 7 | +use euclidv2_verifier::verifier::{BatchVerifier, BundleVerifier, ChunkVerifier}; |
| 8 | +use std::{fs::File, path::Path}; |
| 9 | + |
| 10 | +pub struct EuclidV2Verifier { |
| 11 | + chunk_verifier: ChunkVerifier, |
| 12 | + batch_verifier: BatchVerifier, |
| 13 | + bundle_verifier: BundleVerifier, |
| 14 | +} |
| 15 | + |
| 16 | +impl EuclidV2Verifier { |
| 17 | + pub fn new(assets_dir: &str) -> Self { |
| 18 | + let verifier_bin = Path::new(assets_dir).join("verifier.bin"); |
| 19 | + let config = Path::new(assets_dir).join("root-verifier-vm-config"); |
| 20 | + let exe = Path::new(assets_dir).join("root-verifier-committed-exe"); |
| 21 | + |
| 22 | + Self { |
| 23 | + chunk_verifier: ChunkVerifier::setup(&config, &exe, &verifier_bin) |
| 24 | + .expect("Setting up chunk verifier"), |
| 25 | + batch_verifier: BatchVerifier::setup(&config, &exe, &verifier_bin) |
| 26 | + .expect("Setting up batch verifier"), |
| 27 | + bundle_verifier: BundleVerifier::setup(&config, &exe, &verifier_bin) |
| 28 | + .expect("Setting up bundle verifier"), |
| 29 | + } |
| 30 | + } |
| 31 | +} |
| 32 | + |
| 33 | +impl ProofVerifier for EuclidV2Verifier { |
| 34 | + fn verify(&self, task_type: super::TaskType, proof: Vec<u8>) -> Result<bool> { |
| 35 | + panic_catch(|| match task_type { |
| 36 | + TaskType::Chunk => { |
| 37 | + let proof = serde_json::from_slice::<ChunkProof>(proof.as_slice()).unwrap(); |
| 38 | + self.chunk_verifier |
| 39 | + .verify_proof(proof.proof.as_root_proof().unwrap()) |
| 40 | + } |
| 41 | + TaskType::Batch => { |
| 42 | + let proof = serde_json::from_slice::<BatchProof>(proof.as_slice()).unwrap(); |
| 43 | + self.batch_verifier |
| 44 | + .verify_proof(proof.proof.as_root_proof().unwrap()) |
| 45 | + } |
| 46 | + TaskType::Bundle => { |
| 47 | + let proof = serde_json::from_slice::<BundleProof>(proof.as_slice()).unwrap(); |
| 48 | + self.bundle_verifier |
| 49 | + .verify_proof_evm(&proof.proof.as_evm_proof().unwrap()) |
| 50 | + } |
| 51 | + }) |
| 52 | + .map_err(|err_str: String| anyhow::anyhow!(err_str)) |
| 53 | + } |
| 54 | + |
| 55 | + fn dump_vk(&self, file: &Path) { |
| 56 | + let f = File::create(file).expect("Failed to open file to dump VK"); |
| 57 | + |
| 58 | + let dump = VKDump { |
| 59 | + chunk_vk: base64::encode(self.chunk_verifier.get_app_vk()), |
| 60 | + batch_vk: base64::encode(self.batch_verifier.get_app_vk()), |
| 61 | + bundle_vk: base64::encode(self.bundle_verifier.get_app_vk()), |
| 62 | + }; |
| 63 | + serde_json::to_writer(f, &dump).expect("Failed to dump VK"); |
| 64 | + } |
| 65 | +} |
0 commit comments