problem 8

1 files changed, 20 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 9506289..28541e6 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -2,6 +2,7 @@ extern crate "rustc-serialize" as serialize;
 extern crate openssl;
 
 use std::ascii::AsciiExt;
+use std::collections::HashSet;
 use std::num::Float;
 
 use serialize::base64::{ToBase64,STANDARD};
@@ -85,6 +86,25 @@ pub fn find_single_byte_xor_encrypted_string (inputs: &[Vec<u8>]) -> Vec<u8> {
     return best_decrypted;
 }
 
+pub fn find_aes_128_ecb_encrypted_string (inputs: &[Vec<u8>]) -> Vec<u8> {
+    let mut max_dups = 0;
+    let mut found = vec![];
+    for input in inputs {
+        let mut set = HashSet::new();
+        let mut dups = 0;
+        for block in input.chunks(16) {
+            if !set.insert(block) {
+                dups += 1;
+            }
+        }
+        if dups > max_dups {
+            max_dups = dups;
+            found = input.clone();
+        }
+    }
+    return found;
+}
+
 pub fn crack_repeating_key_xor (input: &[u8]) -> Vec<u8> {
     let mut keysizes = vec![];
     for keysize in 2..40 {