-
Notifications
You must be signed in to change notification settings - Fork 377
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add functions to generate aggregation snark, and test cases.
- Loading branch information
1 parent
461f402
commit 5a373a7
Showing
12 changed files
with
382 additions
and
51 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,30 +1,64 @@ | ||
use crate::utils::{c_char_to_str, c_char_to_vec}; | ||
use libc::c_char; | ||
use prover::{utils::init_env_and_log, zkevm, Proof}; | ||
use prover::{aggregator, config::AGG_LAYER4_DEGREE, utils::init_env_and_log, zkevm, Proof}; | ||
use std::{fs::File, io::Read}; | ||
|
||
static mut VERIFIER: Option<&zkevm::Verifier> = None; | ||
static mut CHUNK_VERIFIER: Option<&zkevm::Verifier> = None; | ||
static mut AGG_VERIFIER: Option<&aggregator::Verifier> = None; | ||
|
||
/// # Safety | ||
#[no_mangle] | ||
pub unsafe extern "C" fn init_verifier(params_path: *const c_char, agg_vk_path: *const c_char) { | ||
init_env_and_log("ffi_verify"); | ||
pub unsafe extern "C" fn init_chunk_verifier(params_dir: *const c_char, vk_path: *const c_char) { | ||
init_env_and_log("ffi_chunk_verify"); | ||
|
||
let params_path = c_char_to_str(params_path); | ||
let agg_vk_path = c_char_to_str(agg_vk_path); | ||
let mut f = File::open(agg_vk_path).unwrap(); | ||
let mut agg_vk = vec![]; | ||
f.read_to_end(&mut agg_vk).unwrap(); | ||
let vk_path = c_char_to_str(vk_path); | ||
let mut f = File::open(vk_path).unwrap(); | ||
let mut vk = vec![]; | ||
f.read_to_end(&mut vk).unwrap(); | ||
|
||
let v = Box::new(zkevm::Verifier::from_params_dir(params_path, Some(agg_vk))); | ||
VERIFIER = Some(Box::leak(v)) | ||
let params_dir = c_char_to_str(params_dir); | ||
let verifier = Box::new(zkevm::Verifier::from_params_dir(params_dir, Some(vk))); | ||
|
||
CHUNK_VERIFIER = Some(Box::leak(verifier)); | ||
} | ||
|
||
/// # Safety | ||
#[no_mangle] | ||
pub unsafe extern "C" fn init_agg_verifier(params_dir: *const c_char, vk_path: *const c_char) { | ||
init_env_and_log("ffi_agg_verify"); | ||
|
||
let vk_path = c_char_to_str(vk_path); | ||
let mut f = File::open(vk_path).unwrap(); | ||
let mut vk = vec![]; | ||
f.read_to_end(&mut vk).unwrap(); | ||
|
||
let params_dir = c_char_to_str(params_dir); | ||
let verifier = Box::new(aggregator::Verifier::from_params_dir( | ||
params_dir, | ||
*AGG_LAYER4_DEGREE, | ||
Some(vk), | ||
)); | ||
|
||
AGG_VERIFIER = Some(Box::leak(verifier)); | ||
} | ||
|
||
/// # Safety | ||
#[no_mangle] | ||
pub unsafe extern "C" fn verify_chunk_proof(proof: *const c_char) -> c_char { | ||
let proof_vec = c_char_to_vec(proof); | ||
let chunk_proof = serde_json::from_slice::<Proof>(proof_vec.as_slice()).unwrap(); | ||
let verified = VERIFIER.unwrap().verify_chunk_proof(chunk_proof).is_ok(); | ||
let proof = serde_json::from_slice::<Proof>(proof_vec.as_slice()).unwrap(); | ||
let verified = CHUNK_VERIFIER.unwrap().verify_chunk_proof(proof).is_ok(); | ||
|
||
verified as c_char | ||
} | ||
|
||
/// # Safety | ||
#[no_mangle] | ||
pub unsafe extern "C" fn verify_agg_proof(proof: *const c_char) -> c_char { | ||
let proof_vec = c_char_to_vec(proof); | ||
let proof = serde_json::from_slice::<Proof>(proof_vec.as_slice()).unwrap(); | ||
|
||
let verified = AGG_VERIFIER.unwrap().verify_agg_proof(proof).is_ok(); | ||
|
||
verified as c_char | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.