factor this out

1 files changed, 12 insertions, 7 deletions

diff --git a/src/aes.rs b/src/aes.rs
index eb281e2..9baf595 100644
--- a/src/aes.rs
+++ b/src/aes.rs
@@ -56,13 +56,7 @@ 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;
-            }
-        }
+        let dups = count_duplicate_blocks(input);
         if dups > max_dups {
             max_dups = dups;
             found = input.clone();
@@ -71,6 +65,17 @@ pub fn find_aes_128_ecb_encrypted_string (inputs: &[Vec<u8>]) -> Vec<u8> {
     return found;
 }
 
+fn count_duplicate_blocks (input: &[u8]) -> usize {
+    let mut set = HashSet::new();
+    let mut dups = 0;
+    for block in input.chunks(16) {
+        if !set.insert(block) {
+            dups += 1;
+        }
+    }
+    return dups;
+}
+
 #[test]
 fn test_encrypt_decrypt () {
     let plaintext = b"Summertime and the wind is blowing outside in lower \