ec_operations.test.cpp

Go to the documentation of this file.

#include "ec_operations.hpp"
#include "acir_format.hpp"
#include "acir_format_mocks.hpp"
#include "barretenberg/circuit_checker/circuit_checker.hpp"
 
#include "barretenberg/stdlib/primitives/curves/secp256k1.hpp"
#include "barretenberg/stdlib/primitives/group/cycle_group.hpp"
 
#include <gtest/gtest.h>
#include <vector>
 
namespace acir_format::tests {
 
using curve_ct = bb::stdlib::secp256k1<Builder>;
 
class EcOperations : public ::testing::Test {
  protected:
    static void SetUpTestSuite() { bb::srs::init_file_crs_factory(bb::srs::bb_crs_path()); }
};
 
size_t generate_ec_add_constraint(EcAdd& ec_add_constraint, WitnessVector& witness_values)
{
    using cycle_group_ct = bb::stdlib::cycle_group<Builder>;
    witness_values.push_back(0);
    auto g1 = grumpkin::g1::affine_one;
    cycle_group_ct input_point(g1);
    // Doubling
    cycle_group_ct result = input_point.dbl();
    // add: x,y,x2,y2
    witness_values.push_back(g1.x);
    witness_values.push_back(g1.y);
    witness_values.push_back(g1.x);
    witness_values.push_back(g1.y);
    witness_values.push_back(result.x.get_value());
    witness_values.push_back(result.y.get_value());
    witness_values.push_back(fr(0));
    witness_values.push_back(fr(0));
    witness_values.push_back(fr(1)); // predicate
    ec_add_constraint = EcAdd{
        .input1_x = WitnessOrConstant<bb::fr>::from_index(1),
        .input1_y = WitnessOrConstant<bb::fr>::from_index(2),
        .input1_infinite = WitnessOrConstant<bb::fr>::from_index(7),
        .input2_x = WitnessOrConstant<bb::fr>::from_index(3),
        .input2_y = WitnessOrConstant<bb::fr>::from_index(4),
        .input2_infinite = WitnessOrConstant<bb::fr>::from_index(7),
        .predicate = WitnessOrConstant<bb::fr>::from_index(9),
        .result_x = 5,
        .result_y = 6,
        .result_infinite = 8,
    };
    return witness_values.size();
}
 
TEST_F(EcOperations, TestECOperations)
{
    EcAdd ec_add_constraint;
 
    WitnessVector witness_values;
    size_t num_variables = generate_ec_add_constraint(ec_add_constraint, witness_values);
 
    AcirFormat constraint_system{
        .varnum = static_cast<uint32_t>(num_variables + 1),
        .num_acir_opcodes = 1,
        .public_inputs = {},
        .ec_add_constraints = { ec_add_constraint },
        .original_opcode_indices = create_empty_original_opcode_indices(),
    };
    mock_opcode_indices(constraint_system);
 
    AcirProgram program{ constraint_system, witness_values };
    auto builder = create_circuit(program);
 
    EXPECT_TRUE(CircuitChecker::check(builder));
}
 
TEST_F(EcOperations, TestECPredicate)
{
    EcAdd ec_add_constraint;
 
    WitnessVector witness_values;
    size_t num_variables = generate_ec_add_constraint(ec_add_constraint, witness_values);
    witness_values[2] = fr(0); // Bad value
    witness_values[9] = fr(0); // Set the predicate to false
    AcirFormat constraint_system{
        .varnum = static_cast<uint32_t>(num_variables + 1),
        .num_acir_opcodes = 1,
        .public_inputs = {},
        .ec_add_constraints = { ec_add_constraint },
        .original_opcode_indices = create_empty_original_opcode_indices(),
    };
    mock_opcode_indices(constraint_system);
 
    AcirProgram program{ constraint_system, witness_values };
    auto builder = create_circuit(program);
 
    EXPECT_TRUE(CircuitChecker::check(builder));
}
 
size_t generate_msm_constraint(MultiScalarMul& msm_constraint, WitnessVector& witness_values)
{
    witness_values.emplace_back(fr(0));
 
    witness_values = {
        // dummy
        fr(0),
        // g1: x,y,infinite
        fr(1),
        fr("0x0000000000000002cf135e7506a45d632d270d45f1181294833fc48d823f272c"),
        fr(0),
        // low, high scalars
        fr(1),
        fr(0),
        // result
        fr("0x06ce1b0827aafa85ddeb49cdaa36306d19a74caa311e13d46d8bc688cdbffffe"),
        fr("0x1c122f81a3a14964909ede0ba2a6855fc93faf6fa1a788bf467be7e7a43f80ac"),
        fr(0),
        // predicate
        fr(1),
    };
    msm_constraint = MultiScalarMul{
        .points = { WitnessOrConstant<fr>{
                        .index = 1,
                        .value = fr(0),
                        .is_constant = false,
                    },
                    WitnessOrConstant<fr>{
                        .index = 2,
                        .value = fr(0),
                        .is_constant = false,
                    },
                    WitnessOrConstant<fr>{
                        .index = 3,
                        .value = fr(0),
                        .is_constant = false,
                    },
                    WitnessOrConstant<fr>{
                        .index = 1,
                        .value = fr(0),
                        .is_constant = false,
                    },
                    WitnessOrConstant<fr>{
                        .index = 2,
                        .value = fr(0),
                        .is_constant = false,
                    },
                    WitnessOrConstant<fr>{
                        .index = 3,
                        .value = fr(0),
                        .is_constant = false,
                    } },
        .scalars = { WitnessOrConstant<fr>{
                         .index = 4,
                         .value = fr(0),
                         .is_constant = false,
                     },
                     WitnessOrConstant<fr>{
                         .index = 5,
                         .value = fr(0),
                         .is_constant = false,
                     },
                     WitnessOrConstant<fr>{
                         .index = 4,
                         .value = fr(0),
                         .is_constant = false,
                     },
                     WitnessOrConstant<fr>{
                         .index = 5,
                         .value = fr(0),
                         .is_constant = false,
                     } },
        .predicate =
            WitnessOrConstant<fr>{
                .index = 9,
                .value = fr(1),
                .is_constant = false,
            },
        .out_point_x = 6,
        .out_point_y = 7,
        .out_point_is_infinite = 8,
    };
    return witness_values.size();
}
TEST_F(EcOperations, TestECMultiScalarMul)
{
    MultiScalarMul msm_constrain;
    WitnessVector witness_values;
    generate_msm_constraint(msm_constrain, witness_values);
    auto res_x = fr("0x06ce1b0827aafa85ddeb49cdaa36306d19a74caa311e13d46d8bc688cdbffffe");
    auto assert_equal = poly_triple{
        .a = 6,
        .b = 0,
        .c = 0,
        .q_m = 0,
        .q_l = fr::neg_one(),
        .q_r = 0,
        .q_o = 0,
        .q_c = res_x,
    };
 
    size_t num_variables = witness_values.size();
    AcirFormat constraint_system{
        .varnum = static_cast<uint32_t>(num_variables + 1),
        .num_acir_opcodes = 1,
        .public_inputs = {},
        .multi_scalar_mul_constraints = { msm_constrain },
        .poly_triple_constraints = { assert_equal },
        .original_opcode_indices = create_empty_original_opcode_indices(),
    };
    mock_opcode_indices(constraint_system);
 
    AcirProgram program{ constraint_system, witness_values };
    auto builder = create_circuit(program);
 
    EXPECT_TRUE(CircuitChecker::check(builder));
}
 
TEST_F(EcOperations, TestECMsmPredicate)
{
    MultiScalarMul msm_constrain;
    WitnessVector witness_values;
    generate_msm_constraint(msm_constrain, witness_values);
    witness_values[2] = fr(0); // bad value
    witness_values[9] = fr(0); // false predicate
    auto res_x = fr("0x06ce1b0827aafa85ddeb49cdaa36306d19a74caa311e13d46d8bc688cdbffffe");
    auto assert_equal = poly_triple{
        .a = 6,
        .b = 0,
        .c = 0,
        .q_m = 0,
        .q_l = fr::neg_one(),
        .q_r = 0,
        .q_o = 0,
        .q_c = res_x,
    };
 
    size_t num_variables = witness_values.size();
    AcirFormat constraint_system{
        .varnum = static_cast<uint32_t>(num_variables + 1),
        .num_acir_opcodes = 1,
        .public_inputs = {},
        .multi_scalar_mul_constraints = { msm_constrain },
        .poly_triple_constraints = { assert_equal },
        .original_opcode_indices = create_empty_original_opcode_indices(),
    };
    mock_opcode_indices(constraint_system);
 
    AcirProgram program{ constraint_system, witness_values };
    auto builder = create_circuit(program);
 
    EXPECT_TRUE(CircuitChecker::check(builder));
}
} // namespace acir_format::tests