1#![allow(non_snake_case)]
  2#![allow(non_camel_case_types)]
  3#![allow(clippy::upper_case_acronyms)]
  4///
  5/// This example performs the full flow:
  6/// - define the circuit to be folded
  7/// - fold the circuit with Nova+CycleFold's IVC
  8/// - generate a DeciderEthCircuit final proof
  9/// - generate the Solidity contract that verifies the proof
 10/// - verify the proof in the EVM
 11///
 12use ark_bn254::{constraints::GVar, Bn254, Fr, G1Projective as G1};
 13
 14use ark_groth16::Groth16;
 15use ark_grumpkin::{constraints::GVar as GVar2, Projective as G2};
 16
 17use folding_schemes::{
 18    commitment::{kzg::KZG, pedersen::Pedersen},
 19    folding::{
 20        nova::{
 21            decider_eth::{prepare_calldata, Decider as DeciderEth},
 22            Nova, PreprocessorParam,
 23        },
 24        traits::CommittedInstanceOps,
 25    },
 26    frontend::FCircuit,
 27    transcript::poseidon::poseidon_canonical_config,
 28    Decider, FoldingScheme,
 29};
 30use frontends::noir::{load_noir_circuit, NoirFCircuit};
 31use std::{env, time::Instant};
 32
 33use solidity_verifiers::{
 34    evm::{compile_solidity, Evm},
 35    utils::get_function_selector_for_nova_cyclefold_verifier,
 36    verifiers::nova_cyclefold::get_decider_template_for_cyclefold_decider,
 37    NovaCycleFoldVerifierKey,
 38};
 39
 40fn main() {
 41    // set the initial state
 42    let z_0 = vec![Fr::from(1)];
 43
 44    // initialize the noir fcircuit
 45    let cur_path = env::current_dir().unwrap();
 46
 47    let circuit_path = format!(
 48        "{}/frontends/src/noir/test_folder/test_mimc/target/test_mimc.json",
 49        cur_path.to_str().unwrap()
 50    );
 51
 52    let circuit = load_noir_circuit(circuit_path).unwrap();
 53    let f_circuit = NoirFCircuit {
 54        circuit,
 55        state_len: 1,
 56        external_inputs_len: 0,
 57    };
 58
 59    pub type N = Nova<G1, GVar, G2, GVar2, NoirFCircuit<Fr>, KZG<'static, Bn254>, Pedersen<G2>>;
 60    pub type D = DeciderEth<
 61        G1,
 62        GVar,
 63        G2,
 64        GVar2,
 65        NoirFCircuit<Fr>,
 66        KZG<'static, Bn254>,
 67        Pedersen<G2>,
 68        Groth16<Bn254>,
 69        N,
 70    >;
 71
 72    let poseidon_config = poseidon_canonical_config::<Fr>();
 73    let mut rng = rand::rngs::OsRng;
 74
 75    // prepare the Nova prover & verifier params
 76    let nova_preprocess_params = PreprocessorParam::new(poseidon_config, f_circuit.clone());
 77    let nova_params = N::preprocess(&mut rng, &nova_preprocess_params).unwrap();
 78
 79    // initialize the folding scheme engine, in our case we use Nova
 80    let mut nova = N::init(&nova_params, f_circuit.clone(), z_0).unwrap();
 81
 82    // prepare the Decider prover & verifier params
 83    let (decider_pp, decider_vp) =
 84        D::preprocess(&mut rng, nova_params.clone(), nova.clone()).unwrap();
 85
 86    // run n steps of the folding iteration
 87    for i in 0..5 {
 88        let start = Instant::now();
 89        nova.prove_step(rng, vec![], None).unwrap();
 90        println!("Nova::prove_step {}: {:?}", i, start.elapsed());
 91    }
 92    // verify the last IVC proof
 93    let ivc_proof = nova.ivc_proof();
 94    N::verify(
 95        nova_params.1, // Nova's verifier params
 96        ivc_proof,
 97    )
 98    .unwrap();
 99
100    let start = Instant::now();
101    let proof = D::prove(rng, decider_pp, nova.clone()).unwrap();
102    println!("generated Decider proof: {:?}", start.elapsed());
103
104    let verified = D::verify(
105        decider_vp.clone(),
106        nova.i,
107        nova.z_0.clone(),
108        nova.z_i.clone(),
109        &nova.U_i.get_commitments(),
110        &nova.u_i.get_commitments(),
111        &proof,
112    )
113    .unwrap();
114    assert!(verified);
115    println!("Decider proof verification: {}", verified);
116
117    // Now, let's generate the Solidity code that verifies this Decider final proof
118    let function_selector =
119        get_function_selector_for_nova_cyclefold_verifier(nova.z_0.len() * 2 + 1);
120
121    let calldata: Vec<u8> = prepare_calldata(
122        function_selector,
123        nova.i,
124        nova.z_0,
125        nova.z_i,
126        &nova.U_i,
127        &nova.u_i,
128        proof,
129    )
130    .unwrap();
131
132    // prepare the setup params for the solidity verifier
133    let nova_cyclefold_vk = NovaCycleFoldVerifierKey::from((decider_vp, f_circuit.state_len()));
134
135    // generate the solidity code
136    let decider_solidity_code = get_decider_template_for_cyclefold_decider(nova_cyclefold_vk);
137
138    // verify the proof against the solidity code in the EVM
139    let nova_cyclefold_verifier_bytecode = compile_solidity(&decider_solidity_code, "NovaDecider");
140    let mut evm = Evm::default();
141    let verifier_address = evm.create(nova_cyclefold_verifier_bytecode);
142    let (_, output) = evm.call(verifier_address, calldata.clone());
143    assert_eq!(*output.last().unwrap(), 1);
144
145    // save smart contract and the calldata
146    println!("storing nova-verifier.sol and the calldata into files");
147    use std::fs;
148    fs::write(
149        "./examples/nova-verifier.sol",
150        decider_solidity_code.clone(),
151    )
152    .unwrap();
153    fs::write("./examples/solidity-calldata.calldata", calldata.clone()).unwrap();
154    let s = solidity_verifiers::utils::get_formatted_calldata(calldata.clone());
155    fs::write("./examples/solidity-calldata.inputs", s.join(",\n")).expect("");
156}