Start work on challenge 12.

src/set2.rs

@@ -37,17 +37,10 @@ pub fn challenge11() {
         CBC,
         ECB,
     }
-    fn random_bytes(count: usize) -> Vec<u8> {
-        (0..count)
-            .map(|_| rand::thread_rng().gen_range(0..255))
-            .collect()
-    }
 
     fn pad_data(mut v: Vec<u8>) -> Bytes {
-        let pre_count: usize = rand::thread_rng().gen_range(5..10);
-        let post_count: usize = rand::thread_rng().gen_range(5..10);
-        let mut pre_pad = random_bytes(pre_count);
-        let mut post_pad = random_bytes(post_count);
+        let mut pre_pad = Bytes::random_range(5, 10).0;
+        let mut post_pad = Bytes::random_range(5, 10).0;
         pre_pad.append(&mut v);
         pre_pad.append(&mut post_pad);
         Bytes(pre_pad)
@@ -56,7 +49,7 @@ pub fn challenge11() {
     fn encryption_oracle(Bytes(data): &Bytes) -> (Bytes, EncryptionType) {
         // Write a function that encrypts data under an unknown key --- that is, a
         // function that generates a random key and encrypts under it.
-        let key = Bytes(random_bytes(16));
+        let key = Bytes::random(16);
         // Under the hood, have the function append 5-10 bytes (count chosen randomly)
         // before the plaintext and 5-10 bytes after the plaintext.
         let padded_data = pad_data(data.to_vec());
@@ -67,7 +60,7 @@ pub fn challenge11() {
         let (data, encryption_type) = if zero_or_one == 1 {
             (ecb::encrypt(&key, &padded_data), EncryptionType::ECB)
         } else {
-            let iv = Bytes(random_bytes(16));
+            let iv = Bytes::random(16);
             (cbc::encrypt(&key, &iv, &padded_data), EncryptionType::CBC)
         };
         (data, encryption_type)
@@ -107,3 +100,17 @@ pub fn challenge11() {
     }
     run_oracle(10);
 }
+
+pub fn challenge12() {
+    let key = Bytes::random(16);
+    fn encrypt(Bytes(data): &Bytes, key: &Bytes) -> Bytes {
+        let mut data = data.to_vec();
+        let mut b = Bytes::from_utf8("Um9sbGluJyBpbiBteSA1LjAKV2l0aCBteSByYWctdG9wIGRvd24gc28gbXkgaGFpciBjYW4gYmxvdwpUaGUgZ2lybGllcyBvbiBzdGFuZGJ5IHdhdmluZyBqdXN0IHRvIHNheSBoaQpEaWQgeW91IHN0b3A/IE5vLCBJIGp1c3QgZHJvdmUgYnkK");
+        data.append(&mut b.0);
+        let cipher = ecb::encrypt(&key, &Bytes(data));
+        cipher
+    }
+    let text = Bytes::from_utf8("aaaabbbbccccddddaaaabbbbccccddddaaaabbbbccccdddd");
+    let cipher = encrypt(&text, &key);
+    println!("[xxxx] Challenge 12: {}", cipher.0.len());
+}