changing ascon back to Vec<u8> (#33)

Author: WingZer0o

src/sponges/ascon_aead.rs

@@ -9,7 +9,10 @@ pub struct AsconAead;
 
 impl CASAsconAead for AsconAead {
     /// Encrypts with AscondAead
-    fn encrypt(key: [u8; 16], nonce: [u8; 16], plaintext: Vec<u8>) -> Vec<u8> {
+    fn encrypt(key: Vec<u8>, nonce: Vec<u8>, plaintext: Vec<u8>) -> Vec<u8> {
+        if key.len() != 16 || nonce.len() != 16 {
+            panic!("Key and nonce must be 16 bytes long");
+        }
         let key_generic_array = GenericArray::from_slice(&key);
         let nonce_generic_array = GenericArray::from_slice(&nonce);
         let cipher = Ascon128::new(key_generic_array);
@@ -18,7 +21,10 @@ impl CASAsconAead for AsconAead {
     }
 
     /// Encrypts with AscondAead on the threadpool
-    fn encrypt_threadpool(key: [u8; 16], nonce: [u8; 16], plaintext: Vec<u8>) -> Vec<u8> {
+    fn encrypt_threadpool(key: Vec<u8>, nonce: Vec<u8>, plaintext: Vec<u8>) -> Vec<u8> {
+        if key.len() != 16 || nonce.len() != 16 {
+            panic!("Key and nonce must be 16 bytes long");
+        }
         let (sender, receiver) = mpsc::channel();
         rayon::spawn(move || {
             let ciphertext = Self::encrypt(key, nonce, plaintext);
@@ -29,7 +35,10 @@ impl CASAsconAead for AsconAead {
     }
 
     /// Decrypts with AscondAead 
-    fn decrypt(key: [u8; 16], nonce: [u8; 16], ciphertext: Vec<u8>) -> Vec<u8> {
+    fn decrypt(key: Vec<u8>, nonce: Vec<u8>, ciphertext: Vec<u8>) -> Vec<u8> {
+        if key.len() != 16 || nonce.len() != 16 {
+            panic!("Key and nonce must be 16 bytes long");
+        }
         let key_generic_array = GenericArray::from_slice(&key);
         let nonce_generic_array = GenericArray::from_slice(&nonce);
         let cipher = Ascon128::new(key_generic_array);
@@ -38,7 +47,10 @@ impl CASAsconAead for AsconAead {
     }
 
     /// Decrypts with AscondAead on the threadpool
-    fn decrypt_threadpool(key: [u8; 16], nonce: [u8; 16], ciphertext: Vec<u8>) -> Vec<u8> {
+    fn decrypt_threadpool(key: Vec<u8>, nonce: Vec<u8>, ciphertext: Vec<u8>) -> Vec<u8> {
+        if key.len() != 16 || nonce.len() != 16 {
+            panic!("Key and nonce must be 16 bytes long");
+        }
         let (sender, receiver) = mpsc::channel();
         rayon::spawn(move || {
             let plaintext = Self::decrypt(key, nonce, ciphertext);
@@ -49,12 +61,12 @@ impl CASAsconAead for AsconAead {
     }
 
     /// Generates a 16-byte key for Ascon Aead
-    fn generate_key() -> [u8; 16] {
-        return Ascon128::generate_key(&mut OsRng).into();
+    fn generate_key() -> Vec<u8> {
+        return Ascon128::generate_key(&mut OsRng).to_vec();
     }
 
     /// Generates a 16-byte key for Ascon Aead on the threadpool
-    fn generate_key_threadpool() -> [u8; 16] {
+    fn generate_key_threadpool() -> Vec<u8> {
         let (sender, receiver) = mpsc::channel();
         rayon::spawn(move || {
             let key = Self::generate_key();
@@ -65,12 +77,12 @@ impl CASAsconAead for AsconAead {
     }
 
     /// Generates a Ascon Aead nonce
-    fn generate_nonce() -> [u8; 16] {
-        return Ascon128::generate_nonce(&mut OsRng).into();
+    fn generate_nonce() -> Vec<u8> {
+        return Ascon128::generate_nonce(&mut OsRng).to_vec();
     }
 
     /// Generates a Ascon Aead nonce on the threadpool
-    fn generate_nonce_threadpool() -> [u8; 16] {
+    fn generate_nonce_threadpool() -> Vec<u8> {
         let (sender, receiver) = mpsc::channel();
         rayon::spawn(move || {
             let key = <AsconAead as CASAsconAead>::generate_nonce();

src/sponges/cas_ascon_aead.rs

@@ -1,10 +1,10 @@
 pub trait CASAsconAead {
-    fn generate_key() -> [u8; 16];
-    fn generate_key_threadpool() -> [u8; 16];
-    fn generate_nonce() -> [u8; 16];
-    fn generate_nonce_threadpool() -> [u8; 16];
-    fn encrypt(key: [u8; 16], nonce: [u8; 16], plaintext: Vec<u8>) -> Vec<u8>;
-    fn encrypt_threadpool(key: [u8; 16], nonce: [u8; 16], plaintext: Vec<u8>) -> Vec<u8>;
-    fn decrypt(key: [u8; 16], nonce: [u8; 16], ciphertext: Vec<u8>) -> Vec<u8>;
-    fn decrypt_threadpool(key: [u8; 16], nonce: [u8; 16], ciphertext: Vec<u8>) -> Vec<u8>;
+    fn generate_key() -> Vec<u8>;
+    fn generate_key_threadpool() -> Vec<u8>;
+    fn generate_nonce() -> Vec<u8>;
+    fn generate_nonce_threadpool() -> Vec<u8>;
+    fn encrypt(key: Vec<u8>, nonce: Vec<u8>, plaintext: Vec<u8>) -> Vec<u8>;
+    fn encrypt_threadpool(key: Vec<u8>, nonce: Vec<u8>, plaintext: Vec<u8>) -> Vec<u8>;
+    fn decrypt(key: Vec<u8>, nonce: Vec<u8>, ciphertext: Vec<u8>) -> Vec<u8>;
+    fn decrypt_threadpool(key: Vec<u8>, nonce: Vec<u8>, ciphertext: Vec<u8>) -> Vec<u8>;
 }

tests/sponges.rs

@@ -9,7 +9,7 @@ mod tests {
         let file_bytes: Vec<u8> = std::fs::read(path).unwrap();
         let ascon_nonce = <AsconAead as CASAsconAead>::generate_nonce();
         let ascon_key = <AsconAead as CASAsconAead>::generate_key();
-        let encrypted_bytes = <AsconAead as CASAsconAead>::encrypt(ascon_key, ascon_nonce, file_bytes.clone());
+        let encrypted_bytes = <AsconAead as CASAsconAead>::encrypt(ascon_key.clone(), ascon_nonce.clone(), file_bytes.clone());
         let mut file =  File::create("encrypted.docx").unwrap();
         file.write_all(&encrypted_bytes).unwrap();
 
@@ -25,7 +25,7 @@ mod tests {
         let file_bytes: Vec<u8> = std::fs::read(path).unwrap();
         let ascon_nonce = <AsconAead as CASAsconAead>::generate_nonce_threadpool();
         let ascon_key = <AsconAead as CASAsconAead>::generate_key_threadpool();
-        let encrypted_bytes = <AsconAead as CASAsconAead>::encrypt_threadpool(ascon_key, ascon_nonce, file_bytes.clone());
+        let encrypted_bytes = <AsconAead as CASAsconAead>::encrypt_threadpool(ascon_key.clone(), ascon_nonce.clone(), file_bytes.clone());
         let mut file =  File::create("encrypted.docx").unwrap();
         file.write_all(&encrypted_bytes).unwrap();