From c21c11aed0a8803eb8f77e405413cf89185c1db8 Mon Sep 17 00:00:00 2001
From: 0xalivecow <mail@ta-lotz.de>
Date: Tue, 22 Oct 2024 11:37:09 +0200
Subject: [PATCH] feat: initial sea128 encrypt is working

---
 Cargo.toml                      |  1 +
 src/tasks/tasks01/block2poly.rs |  8 ++---
 src/tasks/tasks01/mod.rs        |  2 +-
 src/tasks/tasks01/sea128.rs     | 59 +++++++++++++++++++++++++++++++++
 src/utils/poly.rs               | 26 +++++++++++++++
 5 files changed, 89 insertions(+), 7 deletions(-)
 create mode 100644 src/tasks/tasks01/sea128.rs

diff --git a/Cargo.toml b/Cargo.toml
index e7a9eb9..9bcc085 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -6,5 +6,6 @@ edition = "2021"
 [dependencies]
 anyhow = "1.0.90"
 base64 = "0.22.1"
+openssl = "0.10.68"
 serde = { version = "1.0.210", features = ["derive"] }
 serde_json = "1.0"
diff --git a/src/tasks/tasks01/block2poly.rs b/src/tasks/tasks01/block2poly.rs
index 5f30cf0..61a1706 100644
--- a/src/tasks/tasks01/block2poly.rs
+++ b/src/tasks/tasks01/block2poly.rs
@@ -1,6 +1,6 @@
 use std::{str::Bytes, string};
 
-use crate::utils::poly::{self, get_coefficients};
+use crate::utils::poly::{self, b64_2_num, get_coefficients};
 use anyhow::Result;
 use base64::prelude::*;
 use serde_json::Value;
@@ -9,11 +9,7 @@ pub fn block2poly(val: &Value) -> Result<Vec<u8>> {
     // Convert JSON data in to a u128
     // TODO: Transfer decoding into own function?
     let string: String = serde_json::from_value(val["block"].clone())?;
-    let decoded: Vec<u8> = BASE64_STANDARD.decode(string)?;
-
-    let mut bytes: [u8; 16] = [0u8; 16];
-    bytes.copy_from_slice(&decoded);
-    let number: u128 = <u128>::from_ne_bytes(bytes);
+    let number = b64_2_num(&string)?;
 
     let coefficients: Vec<u8> = get_coefficients(number);
 
diff --git a/src/tasks/tasks01/mod.rs b/src/tasks/tasks01/mod.rs
index 69d4bb9..e474037 100644
--- a/src/tasks/tasks01/mod.rs
+++ b/src/tasks/tasks01/mod.rs
@@ -1,3 +1,3 @@
 pub mod block2poly;
 pub mod poly2block;
-
+pub mod sea128;
diff --git a/src/tasks/tasks01/sea128.rs b/src/tasks/tasks01/sea128.rs
new file mode 100644
index 0000000..720b0dc
--- /dev/null
+++ b/src/tasks/tasks01/sea128.rs
@@ -0,0 +1,59 @@
+use anyhow::Result;
+use base64::prelude::*;
+use openssl::symm::{Cipher, Crypter, Mode};
+use serde_json::Value;
+
+use crate::utils::poly::b64_2_num;
+
+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())?;
+    //let plaintexts: &[u8] = &b64_2_num(plaintexts_string)?.to_ne_bytes();
+    let plaintexts = BASE64_STANDARD.decode(plaintexts_string)?;
+    eprintln!("{:?}", plaintexts);
+
+    let xor_val: u128 = 0xc0ffeec0ffeec0ffeec0ffeec0ffee11;
+
+    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 count = encrypter.update(&plaintexts, &mut ciphertext)?;
+    count += encrypter.finalize(&mut ciphertext)?;
+    ciphertext.truncate(count);
+
+    eprintln!("{:?}", &ciphertext[..]);
+
+    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());
+
+    Ok(output)
+}
+
+#[cfg(test)]
+mod tests {
+    use std::fs;
+
+    use anyhow::Result;
+    use serde_json::json;
+    // Note this useful idiom: importing names from outer (for mod tests) scope.
+    use super::*;
+
+    #[test]
+    fn test_sea128() -> Result<()> {
+        let args =
+            json!({"key" : "istDASeincoolerKEYrofg==", "input" : "yv66vvrO263eyviIiDNEVQ=="});
+
+        assert_eq!(sea128(&args)?, "D5FDo3iVBoBN9gVi9/MSKQ==");
+
+        Ok(())
+    }
+}
diff --git a/src/utils/poly.rs b/src/utils/poly.rs
index 771625b..528124e 100644
--- a/src/utils/poly.rs
+++ b/src/utils/poly.rs
@@ -1,3 +1,4 @@
+use anyhow::Result;
 use base64::prelude::*;
 use std::{fmt::format, str::FromStr, u128, u8};
 
@@ -19,6 +20,16 @@ pub fn get_alpha_rep(num: u128) -> String {
     alpha_rep
 }
 
+pub fn b64_2_num(string: &String) -> Result<u128> {
+    let decoded: Vec<u8> = BASE64_STANDARD.decode(string)?;
+
+    let mut bytes: [u8; 16] = [0u8; 16];
+    bytes.copy_from_slice(&decoded);
+    let number: u128 = <u128>::from_ne_bytes(bytes);
+
+    Ok(number)
+}
+
 pub fn get_coefficients(num: u128) -> Vec<u8> {
     let mut powers: Vec<u8> = vec![];
     for shift in 0..128 {
@@ -54,6 +65,8 @@ pub fn coefficient_to_binary(coefficients: Vec<u8>) -> u128 {
 
 #[cfg(test)]
 mod tests {
+    use crate::utils::poly::b64_2_num;
+    use anyhow::Result;
     // Note this useful idiom: importing names from outer (for mod tests) scope.
     use super::*;
 
@@ -88,4 +101,17 @@ mod tests {
             "ARIAAAAAAAAAAAAAAAAAgA=="
         );
     }
+
+    #[test]
+    fn test_b64_2_num() -> Result<()> {
+        let b64_payload: String = String::from_str("juMqbhnlBwAAAAAAAAAAAA==")?;
+        assert_eq!(
+            b64_2_num(&b64_payload)?,
+            2222222222222222,
+            "Error: Value was: {}",
+            b64_2_num(&b64_payload)?
+        );
+
+        Ok(())
+    }
 }