0xalivecow/kauma
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: remove unneded prints and enable mt

Browse source
This commit is contained in:
Alivecow 2024-12-05 16:37:54 +01:00
parent 7a0d1219f9
commit 6d1b735a0b
2 changed files with 1 additions and 81 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/tasks/mod.rs
View file

@ -241,7 +241,7 @@ pub fn task_distribute_st(testcases: &Testcases) -> Result<Responses> {
pub fn task_distribute(testcases: &Testcases) -> Result<Responses> { pub fn task_distribute(testcases: &Testcases) -> Result<Responses> {
let cpus = num_cpus::get(); let cpus = num_cpus::get();
if cpus > 1000000 { if cpus > 1 {
task_distribute_mt(testcases) task_distribute_mt(testcases)
} else { } else {
task_distribute_st(testcases) task_distribute_st(testcases)

80
src/tasks/tasks01/gcm_crack.rs
View file

@ -98,13 +98,8 @@ pub fn gcm_crack(args: &Value) -> Result<CrackAnswer> {
let (m3_data, _) = parse_message(&args["m3"])?; let (m3_data, _) = parse_message(&args["m3"])?;
eprintln!("m1 poly: {:?}", m1_h_poly.clone().to_c_array());
eprintln!("m2 poly: {:?}", m2_h_poly.clone().to_c_array());
let combine_poly = m1_h_poly + m2_h_poly; let combine_poly = m1_h_poly + m2_h_poly;
eprintln!("combine poly: {:?}", combine_poly.clone().to_c_array());
let combine_sff = sff(combine_poly.monic()); let combine_sff = sff(combine_poly.monic());
let mut combine_ddf: Vec<(Polynomial, u128)> = vec![]; let mut combine_ddf: Vec<(Polynomial, u128)> = vec![];
@ -112,8 +107,6 @@ pub fn gcm_crack(args: &Value) -> Result<CrackAnswer> {
combine_ddf.extend(ddf(factor)); combine_ddf.extend(ddf(factor));
} }
eprintln!("combine_ddf: {:?}", combine_ddf);
let mut combine_edf: Vec<Polynomial> = vec![]; let mut combine_edf: Vec<Polynomial> = vec![];
for (factor, degree) in combine_ddf { for (factor, degree) in combine_ddf {
if degree == 1 { if degree == 1 {
@ -121,8 +114,6 @@ pub fn gcm_crack(args: &Value) -> Result<CrackAnswer> {
} }
} }
eprintln!("combine_edf: {:?}", combine_edf);
let mut m3_auth_tag: Vec<u8> = vec![]; let mut m3_auth_tag: Vec<u8> = vec![];
let mut h_candidate: FieldElement = FieldElement::zero(); let mut h_candidate: FieldElement = FieldElement::zero();
let mut eky0: Vec<u8> = vec![]; let mut eky0: Vec<u8> = vec![];
@ -155,8 +146,6 @@ pub fn gcm_crack(args: &Value) -> Result<CrackAnswer> {
); );
if m3_auth_tag == m3_data.tag { if m3_auth_tag == m3_data.tag {
eprintln!("Candidate valid");
eprintln!("{:02X?}", m3_auth_tag);
break; break;
} else { } else {
eprintln!("H candidate not valid"); eprintln!("H candidate not valid");
@ -164,16 +153,6 @@ pub fn gcm_crack(args: &Value) -> Result<CrackAnswer> {
} }
} }
eprintln!(
"M3 Authentication TAG {:02X?}",
BASE64_STANDARD.encode(&m3_auth_tag)
);
if m3_auth_tag.is_empty() {
assert!(false);
eprintln!("No valid candidate found");
}
let (forgery_data, _) = parse_message(&args["forgery"])?; let (forgery_data, _) = parse_message(&args["forgery"])?;
let forgery_ghash = ghash( let forgery_ghash = ghash(
@ -196,62 +175,3 @@ pub fn gcm_crack(args: &Value) -> Result<CrackAnswer> {
mask: BASE64_STANDARD.encode(eky0), mask: BASE64_STANDARD.encode(eky0),
}) })
} }
#[cfg(test)]
mod tests {
use anyhow::Result;
use rand::Rng;
use serde_json::json;
use utils::ciphers::{aes_128_encrypt, gcm_encrypt_aes};
// Note this useful idiom: importing names from outer (for mod tests) scope.
use super::*;
#[test]
fn test_random() -> Result<()> {
let key = vec![1, 1, 1, 1];
let nonce = BASE64_STANDARD.decode("4gF+BtR3ku/PUQci")?;
let ad = vec![0];
let input: Vec<u8> = Vec::with_capacity(rand::thread_rng().gen_range(0..=60));
let plain1 = gcm_encrypt_aes(nonce.clone(), key.clone(), input, ad.clone())?;
let input: Vec<u8> = Vec::with_capacity(rand::thread_rng().gen_range(0..=60));
let plain2 = gcm_encrypt_aes(nonce.clone(), key.clone(), input, ad.clone())?;
let input: Vec<u8> = Vec::with_capacity(rand::thread_rng().gen_range(0..=60));
let plain3 = gcm_encrypt_aes(nonce.clone(), key.clone(), input, ad.clone())?;
let crack_input = json!({
"testcases": {
"gcm_crack46": {
"action": "gcm_crack",
"arguments": {
"nonce": "4gF+BtR3ku/PUQci",
"m1": {
"ciphertext": BASE64_STANDARD.encode(plain1.0),
"associated_data": "",
"tag": BASE64_STANDARD.encode(plain1.1)
},
"m2": {
"ciphertext": BASE64_STANDARD.encode(plain2.0),
"associated_data": "",
"tag": BASE64_STANDARD.encode(plain2.1)
},
"m3": {
"ciphertext": BASE64_STANDARD.encode(plain3.0),
"associated_data": "",
"tag": BASE64_STANDARD.encode(plain3.1)
},
"forgery": {
"ciphertext": "AXe/ZQ==",
"associated_data": ""
}
}
}
}
});
todo!();
}
}