From a16eee14a18d4c345d6485593d3543e342c86af3 Mon Sep 17 00:00:00 2001 From: Jesse Luehrs Date: Tue, 7 Apr 2015 01:18:04 -0400 Subject: problem 26 --- tests/set4.rs | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) (limited to 'tests') diff --git a/tests/set4.rs b/tests/set4.rs index 3967f85..35c3edf 100644 --- a/tests/set4.rs +++ b/tests/set4.rs @@ -48,3 +48,41 @@ fn problem_25 () { let got = matasano::crack_aes_128_ctr_random_access(&ciphertext[..], edit); assert_eq!(&got[..], &plaintext[..]); } + +#[test] +fn problem_26 () { + let key = util::random_aes_128_key(); + let nonce = rand::thread_rng().gen(); + let prefix = "comment1=cooking%20MCs;userdata="; + let suffix = ";comment2=%20like%20a%20pound%20of%20bacon"; + let admin = ";admin=true;"; + + let escape = |input: &str| { + input.replace("%", "%25").replace(";", "%3B").replace("=", "%3D") + }; + + let encode = |input: &str| -> Vec { + let plaintext: Vec = prefix + .as_bytes() + .iter() + .chain(escape(input).as_bytes().iter()) + .chain(suffix.as_bytes().iter()) + .map(|x| *x) + .collect(); + return matasano::aes_128_ctr(&plaintext[..], &key[..], nonce); + }; + + let verify = |ciphertext: &[u8]| -> bool { + let plaintext = matasano::aes_128_ctr(ciphertext, &key[..], nonce); + return (0..(plaintext.len() - admin.len())).any(|i| { + plaintext + .iter() + .skip(i) + .zip(admin.as_bytes().iter()) + .all(|(&c1, &c2)| c1 == c2) + }); + }; + + let ciphertext = matasano::crack_ctr_bitflipping(&encode); + assert!(verify(&ciphertext[..])); +} -- cgit v1.2.3-54-g00ecf