fix: fix format zkey

This commit is contained in:
Ya-wen, Jeng 2025-03-06 11:32:22 +08:00
parent 145cc0a262
commit 59f134d2e8
2 changed files with 6 additions and 5 deletions

View File

@ -116,6 +116,7 @@ pub fn groth16_prover_zkey_file_wrapper(
zkey_path: &str,
wtns_buffer: Vec<u8>,
) -> Result<ProofResult> {
let formatted_zkey_path: &str = &zkey_path.to_string();
let wtns_size = wtns_buffer.len() as u64;
let mut proof_buffer = vec![0u8; 4 * 1024 * 1024]; // Adjust size as needed
@ -131,7 +132,7 @@ pub fn groth16_prover_zkey_file_wrapper(
unsafe {
let result = groth16_prover_zkey_file(
zkey_path.as_ptr() as *const std::ffi::c_char,
formatted_zkey_path.as_ptr() as *const std::ffi::c_char,
wtns_buffer.as_ptr() as *const std::os::raw::c_void, // Witness buffer
wtns_size,
proof_ptr,

View File

@ -56,7 +56,7 @@ mod tests {
#[test]
fn test_prove_rapidsnark() -> Result<()> {
// Create a new MoproCircom instance
let zkey_path = "./test-vectors/multiplier2_final.zkey".to_string();
let zkey_path = "./test-vectors/multiplier2_final.zkey";
let mut inputs = HashMap::new();
let a = BigInt::from_str(
@ -72,7 +72,7 @@ mod tests {
// Generate Proof
let proof_result =
rust_rapidsnark::groth16_prover_zkey_file_wrapper(&zkey_path, wtns_buffer)?;
rust_rapidsnark::groth16_prover_zkey_file_wrapper(zkey_path, wtns_buffer)?;
let vkey = std::fs::read_to_string("./test-vectors/multiplier2.vkey.json")?;
let valid = rust_rapidsnark::groth16_verify_wrapper(
@ -87,7 +87,7 @@ mod tests {
#[test]
fn test_prove_rapidsnark_keccak() -> Result<()> {
// Create a new MoproCircom instance
let zkey_path = "./test-vectors/keccak256_256_test_final.zkey".to_string();
let zkey_path = "./test-vectors/keccak256_256_test_final.zkey";
// Prepare inputs
let input_vec = vec![
116, 101, 115, 116, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
@ -103,7 +103,7 @@ mod tests {
// Generate Proof
let proof_result =
rust_rapidsnark::groth16_prover_zkey_file_wrapper(&zkey_path, wtns_buffer)?;
rust_rapidsnark::groth16_prover_zkey_file_wrapper(zkey_path, wtns_buffer)?;
let vkey = std::fs::read_to_string("./test-vectors/keccak256_256_test.vkey.json")?;
let valid = rust_rapidsnark::groth16_verify_wrapper(