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

feat: task runner adapted for sea128 task

Browse source
This commit is contained in:
0xalivecow 2024-10-22 12:28:23 +02:00
parent c21c11aed0
commit 3a777cab00
No known key found for this signature in database
3 changed files with 102 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

76
src/tasks/tasks01/sea128.rs
View file

@ -1,4 +1,4 @@
use anyhow::Result;
use anyhow::{anyhow, Result};
use base64::prelude::*;
use openssl::symm::{Cipher, Crypter, Mode};
use serde_json::Value;
@ -9,33 +9,57 @@ pub fn sea128(args: &Value) -> Result<String> {
let key_string: String = serde_json::from_value(args["key"].clone())?;
//let key: &[u8] = b64_2_num(key_string)?.to_ne_bytes();
let key = BASE64_STANDARD.decode(key_string)?;
eprintln!("{:?}", key);
let plaintexts_string: String = serde_json::from_value(args["input"].clone())?;
//eprintln!("{:?}", key);
let input_string: String = serde_json::from_value(args["input"].clone())?;
//let plaintexts: &[u8] = &b64_2_num(plaintexts_string)?.to_ne_bytes();
let plaintexts = BASE64_STANDARD.decode(plaintexts_string)?;
eprintln!("{:?}", plaintexts);
let input = BASE64_STANDARD.decode(input_string)?;
let xor_val: u128 = 0xc0ffeec0ffeec0ffeec0ffeec0ffee11;
let mut encrypter = Crypter::new(Cipher::aes_128_ecb(), Mode::Encrypt, &key, None)?;
encrypter.pad(false);
let mode: String = serde_json::from_value(args["mode"].clone())?;
match mode.as_str() {
"encrypt" => {
//eprintln!("{:?}", plaintexts);
let mut encrypter = Crypter::new(Cipher::aes_128_ecb(), Mode::Encrypt, &key, None)?;
encrypter.pad(false);
let mut ciphertext = [0; 32].to_vec();
let mut ciphertext = [0; 32].to_vec();
let mut count = encrypter.update(&plaintexts, &mut ciphertext)?;
count += encrypter.finalize(&mut ciphertext)?;
ciphertext.truncate(count);
let mut count = encrypter.update(&input, &mut ciphertext)?;
count += encrypter.finalize(&mut ciphertext)?;
ciphertext.truncate(count);
eprintln!("{:?}", &ciphertext[..]);
//eprintln!("{:?}", &ciphertext[..]);
let mut bytes: [u8; 16] = [0u8; 16];
bytes.copy_from_slice(&ciphertext);
let number: u128 = <u128>::from_be_bytes(bytes);
let mut bytes: [u8; 16] = [0u8; 16];
bytes.copy_from_slice(&ciphertext);
let number: u128 = <u128>::from_be_bytes(bytes);
let output = BASE64_STANDARD.encode((number ^ xor_val).to_be_bytes());
let output = BASE64_STANDARD.encode((number ^ xor_val).to_be_bytes());
Ok(output)
Ok(output)
}
"decrypt" => {
let mut decrypter = Crypter::new(Cipher::aes_128_ecb(), Mode::Decrypt, &key, None)?;
decrypter.pad(false);
let mut bytes: [u8; 16] = [0u8; 16];
bytes.copy_from_slice(&input);
let input_num: u128 = <u128>::from_be_bytes(bytes);
let input_afer_xor = (input_num ^ xor_val).to_be_bytes();
let mut plaintext = [0; 32].to_vec();
let mut count = decrypter.update(&input_afer_xor, &mut plaintext)?;
count += decrypter.finalize(&mut plaintext)?;
plaintext.truncate(count);
let output = BASE64_STANDARD.encode(plaintext);
Ok(output)
}
_ => Err(anyhow!("Failure. no valid mode detected")),
}
}
#[cfg(test)]
@ -48,12 +72,20 @@ mod tests {
use super::*;
#[test]
fn test_sea128() -> Result<()> {
let args =
json!({"key" : "istDASeincoolerKEYrofg==", "input" : "yv66vvrO263eyviIiDNEVQ=="});
fn test_sea128_encrypt() -> Result<()> {
let args = json!({"mode" : "encrypt", "key" : "istDASeincoolerKEYrofg==", "input" : "yv66vvrO263eyviIiDNEVQ=="});
assert_eq!(sea128(&args)?, "D5FDo3iVBoBN9gVi9/MSKQ==");
Ok(())
}
#[test]
fn test_sea128_decrypt() -> Result<()> {
let args = json!({"mode" : "decrypt", "key" : "istDASeincoolerKEYrofg==", "input" : "D5FDo3iVBoBN9gVi9/MSKQ=="});
assert_eq!(sea128(&args)?, "yv66vvrO263eyviIiDNEVQ==");
Ok(())
}
}