logos-storage/circom-compat-ffi
1
0
Fork 0
mirror of https://github.com/logos-storage/circom-compat-ffi.git synced 2026-01-07 15:23:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

return proof result

Browse Source
This commit is contained in:
Dmitriy Ryajov 2024-01-26 19:29:40 -06:00
parent fbead75f42
commit fb0150130f
No known key found for this signature in database
GPG Key ID: DA8C680CE7C657A4
1 changed files with 22 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/ffi.rs
View File

@ -3,7 +3,6 @@ use std::{
ffi::{c_char, CStr}, ffi::{c_char, CStr},
fs::File, fs::File,
panic::{catch_unwind, AssertUnwindSafe}, panic::{catch_unwind, AssertUnwindSafe},
ptr::slice_from_raw_parts,
ptr::slice_from_raw_parts_mut, ptr::slice_from_raw_parts_mut,
}; };
@ -55,7 +54,7 @@ struct CircomCompatCtx {
_marker: core::marker::PhantomData<(*mut CircomCompatCtx, core::marker::PhantomPinned)>, _marker: core::marker::PhantomData<(*mut CircomCompatCtx, core::marker::PhantomPinned)>,
} }
fn to_err_code(result: Result<(), Box<dyn Any + Send>>) -> i32 { fn to_err_code(result: Result<i32, Box<dyn Any + Send>>) -> i32 {
match result { match result {
Ok(_) => ERR_OK, Ok(_) => ERR_OK,
Err(e) => match e.downcast_ref::<i32>() { Err(e) => match e.downcast_ref::<i32>() {
@ -118,6 +117,8 @@ pub unsafe extern "C" fn init_circom_config(
}; };
*cfg_ptr = Box::into_raw(Box::new(circom_bn254_cfg)); *cfg_ptr = Box::into_raw(Box::new(circom_bn254_cfg));
ERR_OK
})); }));
to_err_code(result) to_err_code(result)
@ -144,6 +145,8 @@ pub unsafe extern "C" fn init_circom_compat(
}; };
*ctx_ptr = Box::into_raw(Box::new(circom_compat_ctx)); *ctx_ptr = Box::into_raw(Box::new(circom_compat_ctx));
ERR_OK
})); }));
to_err_code(result) to_err_code(result)
@ -239,6 +242,8 @@ pub unsafe extern "C" fn prove_circuit(
.unwrap(); .unwrap();
*proof_ptr = Box::leak(Box::new((&circom_proof).into())); *proof_ptr = Box::leak(Box::new((&circom_proof).into()));
ERR_OK
})); }));
to_err_code(result) to_err_code(result)
@ -263,6 +268,8 @@ pub unsafe extern "C" fn get_pub_inputs(
.ok_or_else(|| ERR_GET_PUB_INPUTS) .ok_or_else(|| ERR_GET_PUB_INPUTS)
.unwrap(); .unwrap();
*inputs_ptr = Box::leak(Box::new(inputs.as_slice().into())); *inputs_ptr = Box::leak(Box::new(inputs.as_slice().into()));
ERR_OK
})); }));
to_err_code(result) to_err_code(result)
@ -280,6 +287,8 @@ pub unsafe extern "C" fn get_verifying_key(
let vk = prepare_verifying_key(&proving_key.vk).vk; let vk = prepare_verifying_key(&proving_key.vk).vk;
*vk_ptr = Box::leak(Box::new((&vk).into())); *vk_ptr = Box::leak(Box::new((&vk).into()));
ERR_OK
})); }));
to_err_code(result) to_err_code(result)
@ -295,9 +304,15 @@ pub unsafe extern "C" fn verify_circuit(
let result = catch_unwind(AssertUnwindSafe(|| { let result = catch_unwind(AssertUnwindSafe(|| {
let inputs_vec: Vec<Fr> = (*inputs).into(); let inputs_vec: Vec<Fr> = (*inputs).into();
let prepared_key = prepare_verifying_key(&(*pvk).into()); let prepared_key = prepare_verifying_key(&(*pvk).into());
GrothBn::verify_proof(&prepared_key, &(*proof).into(), inputs_vec.as_slice()) let passed = GrothBn::verify_proof(&prepared_key, &(*proof).into(), inputs_vec.as_slice())
.map_err(|_| ERR_FAILED_TO_VERIFY_PROOF) .map_err(|_| ERR_FAILED_TO_VERIFY_PROOF)
.unwrap(); .unwrap();
if !passed {
ERR_FAILED_TO_VERIFY_PROOF
} else {
ERR_OK
}
})); }));
to_err_code(result) to_err_code(result)
@ -329,6 +344,8 @@ pub unsafe extern "C" fn push_input_u256_array(
input input
.iter() .iter()
.for_each(|c| (*circom.builder).push_input(name, *c)); .for_each(|c| (*circom.builder).push_input(name, *c));
ERR_OK
})); }));
to_err_code(result) to_err_code(result)
@ -350,6 +367,8 @@ macro_rules! build_fn
let circom = &mut *to_circom(ctx_ptr); let circom = &mut *to_circom(ctx_ptr);
(*circom.builder).push_input(name, input); (*circom.builder).push_input(name, input);
ERR_OK
})); }));
to_err_code(result) to_err_code(result)