cairo-vm 2.0.1

use num_integer::Integer;

use super::secp::bigint_utils::BigInt3;
use crate::stdlib::{collections::HashMap, prelude::*};
use crate::{
    hint_processor::hint_processor_definition::HintReference,
    math_utils::div_mod,
    serde::deserialize_program::ApTracking,
    types::{errors::math_errors::MathError, exec_scope::ExecutionScopes},
    vm::{errors::hint_errors::HintError, vm_core::VirtualMachine},
};
use num_bigint::BigInt;
use num_traits::Zero;

/* Implements Hint:
%{
    from starkware.cairo.common.cairo_secp.secp_utils import pack
    from starkware.python.math_utils import div_mod, safe_div

    N = pack(ids.n, PRIME)
    x = pack(ids.x, PRIME) % N
    s = pack(ids.s, PRIME) % N,
    value = res = div_mod(x, s, N)
%}
 */
pub fn ec_recover_divmod_n_packed(
    vm: &mut VirtualMachine,
    exec_scopes: &mut ExecutionScopes,
    ids_data: &HashMap<String, HintReference>,
    ap_tracking: &ApTracking,
) -> Result<(), HintError> {
    let n = BigInt3::from_var_name("n", vm, ids_data, ap_tracking)?.pack86();
    if n.is_zero() {
        return Err(MathError::DividedByZero.into());
    }
    let x = BigInt3::from_var_name("x", vm, ids_data, ap_tracking)?
        .pack86()
        .mod_floor(&n);
    let s = BigInt3::from_var_name("s", vm, ids_data, ap_tracking)?
        .pack86()
        .mod_floor(&n);

    let value = div_mod(&x, &s, &n)?;
    exec_scopes.insert_value("value", value.clone());
    exec_scopes.insert_value("res", value);
    Ok(())
}

/* Implements Hint:
%{
    from starkware.cairo.common.cairo_secp.secp_utils import pack
    from starkware.python.math_utils import div_mod, safe_div

    a = pack(ids.x, PRIME)
    b = pack(ids.s, PRIME)
    value = res = a - b
%}
 */
pub fn ec_recover_sub_a_b(
    vm: &mut VirtualMachine,
    exec_scopes: &mut ExecutionScopes,
    ids_data: &HashMap<String, HintReference>,
    ap_tracking: &ApTracking,
) -> Result<(), HintError> {
    let a = BigInt3::from_var_name("a", vm, ids_data, ap_tracking)?.pack86();
    let b = BigInt3::from_var_name("b", vm, ids_data, ap_tracking)?.pack86();

    let value = a - b;
    exec_scopes.insert_value("value", value.clone());
    exec_scopes.insert_value("res", value);
    Ok(())
}

/* Implements Hint:
%{
    from starkware.cairo.common.cairo_secp.secp_utils import pack
    from starkware.python.math_utils import div_mod, safe_div

    a = pack(ids.a, PRIME)
    b = pack(ids.b, PRIME)
    product = a * b
    m = pack(ids.m, PRIME)

    value = res = product % m
%}
 */
pub fn ec_recover_product_mod(
    vm: &mut VirtualMachine,
    exec_scopes: &mut ExecutionScopes,
    ids_data: &HashMap<String, HintReference>,
    ap_tracking: &ApTracking,
) -> Result<(), HintError> {
    let a = BigInt3::from_var_name("a", vm, ids_data, ap_tracking)?.pack86();
    let b = BigInt3::from_var_name("b", vm, ids_data, ap_tracking)?.pack86();
    let m = BigInt3::from_var_name("m", vm, ids_data, ap_tracking)?.pack86();
    if m.is_zero() {
        return Err(MathError::DividedByZero.into());
    }

    let product = a * b;
    let value = product.mod_floor(&m);
    exec_scopes.insert_value("product", product);
    exec_scopes.insert_value("m", m);
    exec_scopes.insert_value("value", value.clone());
    exec_scopes.insert_value("res", value);
    Ok(())
}

/* Implements Hint:
%{
    value = k = product // m
%}
 */
pub fn ec_recover_product_div_m(exec_scopes: &mut ExecutionScopes) -> Result<(), HintError> {
    let product: &BigInt = exec_scopes.get_ref("product")?;
    let m: &BigInt = exec_scopes.get_ref("m")?;
    if m.is_zero() {
        return Err(MathError::DividedByZero.into());
    }
    let value = product.div_floor(m);
    exec_scopes.insert_value("k", value.clone());
    exec_scopes.insert_value("value", value);
    Ok(())
}

#[cfg(test)]
mod tests {
    use num_bigint::BigInt;

    use super::*;
    use crate::hint_processor::builtin_hint_processor::hint_code;
    use crate::hint_processor::hint_processor_definition::HintReference;
    use crate::utils::test_utils::*;
    use crate::{
        any_box,
        hint_processor::{
            builtin_hint_processor::builtin_hint_processor_definition::{
                BuiltinHintProcessor, HintProcessorData,
            },
            hint_processor_definition::HintProcessorLogic,
        },
        types::exec_scope::ExecutionScopes,
    };
    use assert_matches::assert_matches;

    #[cfg(target_arch = "wasm32")]
    use wasm_bindgen_test::*;

    #[test]
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
    fn run_ec_recover_divmod_n_packed_ok() {
        let mut vm = vm!();
        let mut exec_scopes = ExecutionScopes::new();

        vm.run_context.fp = 8;
        let ids_data = non_continuous_ids_data![("n", -8), ("x", -5), ("s", -2)];

        vm.segments = segments![
            //n
            ((1, 0), 177),
            ((1, 1), 0),
            ((1, 2), 0),
            //x
            ((1, 3), 25),
            ((1, 4), 0),
            ((1, 5), 0),
            //s
            ((1, 6), 5),
            ((1, 7), 0),
            ((1, 8), 0)
        ];

        assert!(run_hint!(
            vm,
            ids_data,
            hint_code::EC_RECOVER_DIV_MOD_N_PACKED,
            &mut exec_scopes
        )
        .is_ok());

        check_scope!(
            &exec_scopes,
            [("value", BigInt::from(5)), ("res", BigInt::from(5))]
        );
    }

    #[test]
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
    fn run_ec_recover_divmod_n_is_zero() {
        let mut vm = vm!();
        let mut exec_scopes = ExecutionScopes::new();

        vm.run_context.fp = 8;
        let ids_data = non_continuous_ids_data![("n", -8), ("x", -5), ("s", -2)];

        vm.segments = segments![
            //n
            ((1, 0), 0),
            ((1, 1), 0),
            ((1, 2), 0),
            //x
            ((1, 3), 25),
            ((1, 4), 0),
            ((1, 5), 0),
            //s
            ((1, 6), 5),
            ((1, 7), 0),
            ((1, 8), 0)
        ];

        assert_matches!(
            run_hint!(
                vm,
                ids_data,
                hint_code::EC_RECOVER_DIV_MOD_N_PACKED,
                &mut exec_scopes
            ),
            Err(HintError::Math(MathError::DividedByZero))
        );
    }

    #[test]
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
    fn run_ec_recover_sub_a_b_ok() {
        let mut vm = vm!();
        let mut exec_scopes = ExecutionScopes::new();

        vm.run_context.fp = 8;
        let ids_data = non_continuous_ids_data![("a", -8), ("b", -5)];

        vm.segments = segments![
            //a
            ((1, 0), 100),
            ((1, 1), 0),
            ((1, 2), 0),
            //b
            ((1, 3), 25),
            ((1, 4), 0),
            ((1, 5), 0),
        ];

        assert!(run_hint!(
            vm,
            ids_data,
            hint_code::EC_RECOVER_SUB_A_B,
            &mut exec_scopes
        )
        .is_ok());

        check_scope!(
            &exec_scopes,
            [("value", BigInt::from(75)), ("res", BigInt::from(75))]
        );
    }

    #[test]
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
    fn run_ec_recover_product_mod_ok() {
        let mut vm = vm!();
        let mut exec_scopes = ExecutionScopes::new();

        vm.run_context.fp = 8;
        let ids_data = non_continuous_ids_data![("a", -8), ("b", -5), ("m", -2)];

        vm.segments = segments![
            //a
            ((1, 0), 60),
            ((1, 1), 0),
            ((1, 2), 0),
            //b
            ((1, 3), 2),
            ((1, 4), 0),
            ((1, 5), 0),
            //m
            ((1, 6), 100),
            ((1, 7), 0),
            ((1, 8), 0)
        ];

        assert!(run_hint!(
            vm,
            ids_data,
            hint_code::EC_RECOVER_PRODUCT_MOD,
            &mut exec_scopes
        )
        .is_ok());

        check_scope!(
            &exec_scopes,
            [
                ("value", BigInt::from(20)),
                ("res", BigInt::from(20)),
                ("product", BigInt::from(120)),
                ("m", BigInt::from(100))
            ]
        );
    }

    #[test]
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
    fn run_ec_recover_product_mod_m_zero() {
        let mut vm = vm!();
        let mut exec_scopes = ExecutionScopes::new();

        vm.run_context.fp = 8;
        let ids_data = non_continuous_ids_data![("a", -8), ("b", -5), ("m", -2)];

        vm.segments = segments![
            //a
            ((1, 0), 60),
            ((1, 1), 0),
            ((1, 2), 0),
            //b
            ((1, 3), 2),
            ((1, 4), 0),
            ((1, 5), 0),
            //m
            ((1, 6), 0),
            ((1, 7), 0),
            ((1, 8), 0)
        ];

        assert_matches!(
            run_hint!(
                vm,
                ids_data,
                hint_code::EC_RECOVER_PRODUCT_MOD,
                &mut exec_scopes
            ),
            Err(HintError::Math(MathError::DividedByZero))
        );
    }

    #[test]
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
    fn run_ec_recover_product_div_m_ok() {
        let mut vm = vm!();
        let mut exec_scopes = ExecutionScopes::new();
        exec_scopes.insert_value("product", BigInt::from(250));
        exec_scopes.insert_value("m", BigInt::from(100));

        let ids_data = ids_data!["none"];

        assert!(run_hint!(
            vm,
            ids_data,
            hint_code::EC_RECOVER_PRODUCT_DIV_M,
            &mut exec_scopes
        )
        .is_ok());

        check_scope!(
            &exec_scopes,
            [("value", BigInt::from(2)), ("k", BigInt::from(2))]
        );
    }

    #[test]
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
    fn run_ec_recover_product_div_m_zero() {
        let mut vm = vm!();
        let mut exec_scopes = ExecutionScopes::new();
        exec_scopes.insert_value("product", BigInt::from(250));
        exec_scopes.insert_value("m", BigInt::from(0));

        let ids_data = ids_data!["none"];

        assert_matches!(
            run_hint!(
                vm,
                ids_data,
                hint_code::EC_RECOVER_PRODUCT_DIV_M,
                &mut exec_scopes
            ),
            Err(HintError::Math(MathError::DividedByZero))
        );
    }
}