Function test_tap_hash_var_p

bitvm/src/chunk/taps_msm.rs:341–399

()

    #[test]
    fn test_tap_hash_var_p() {
        let mut prng = ChaCha20Rng::seed_from_u64(1);
        let q = ark_bn254::G1Affine::rand(&mut prng);
        let t = ark_bn254::G1Affine::rand(&mut prng);
        let r = (t + q).into_affine();

        for should_corrupt_output_hash in [true, false] {
            let (hint_out, input_is_valid, op_scr, mut hint_script) = chunk_hash_p(t, q);
            assert!(input_is_valid);
            assert_eq!(r, hint_out);
            let t = DataType::G1Data(t);
            let hint_out = DataType::G1Data(hint_out);
            hint_script.extend_from_slice(&t.to_witness(ElementType::G1));

            let mut output_hash = hint_out.to_hash();

            if should_corrupt_output_hash {
                if let CompressedStateObject::Hash(r) = output_hash {
                    let random_hash = extern_hash_nibbles(vec![r, r]);
                    output_hash = CompressedStateObject::Hash(random_hash);
                }
            }

            let bitcom_scr = script! {
                {output_hash.as_hint_type().push()}
                {Fq::toaltstack()}
                {t.to_hash().as_hint_type().push()}
                {Fq::toaltstack()}
            };
            let hash_script = script! {
                {hash_messages(vec![ElementType::G1, ElementType::G1])}
                OP_TRUE
            };

            let tap_len = op_scr.len();
            let script = script! {
                for h in hint_script {
                    { h.push() }
                }
                {bitcom_scr}
                {op_scr}
                {hash_script}
            };

            let res = execute_script(script);
            if res.final_stack.len() > 1 {
                for i in 0..res.final_stack.len() {
                    println!("{i:} {:?}", res.final_stack.get(i));
                }
            }
            assert_eq!(res.success, should_corrupt_output_hash);
            assert!(res.final_stack.len() == 1);

            println!(
                "chunk_hash_p disprovable({}) script {} stack {}",
                should_corrupt_output_hash, tap_len, res.stats.max_nb_stack_items
            );
        }