parasol_cpu 0.10.0

use std::sync::Arc;

use rand::{RngCore, rng};

use parasol_runtime::test_utils::get_secret_keys_128;

use crate::{
    ArgsBuilder, Memory,
    proc::IsaOp,
    register_names::*,
    test_utils::{MaybeEncryptedUInt, make_computer_128},
};

fn can_not(val: u32, encrypted_computation: bool) {
    let (mut proc, enc) = make_computer_128();
    let sk = get_secret_keys_128();

    let expected = !val;

    let memory = Arc::new(Memory::new_default_stack());

    let args = ArgsBuilder::new()
        .arg(MaybeEncryptedUInt::<32>::new(
            val as u128,
            &enc,
            &sk,
            encrypted_computation,
        ))
        .return_value::<MaybeEncryptedUInt<32>>();

    let program = memory.allocate_program(&[
        IsaOp::Load(T0, SP, 32, 0),
        IsaOp::Not(T0, T0),
        IsaOp::Store(RP, T0, 32, 0),
        IsaOp::Ret(),
    ]);

    let ans = proc.run_program(program, &memory, args).unwrap();
    let ans = ans.get(&enc, &sk);

    assert_eq!(expected, ans);
}

#[test]
fn can_not_plaintext_inputs() {
    for _ in 0..10 {
        let val = rng().next_u32();
        can_not(val, false);
    }
}

#[test]
fn can_not_ciphertext_inputs() {
    for _ in 0..10 {
        let val = rng().next_u32();
        can_not(val, true);
    }
}