problem 28

3 files changed, 114 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 40a9773..8452240 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -10,6 +10,7 @@ mod data;
 mod http;
 mod primitives;
 mod random;
+mod sha1;
 
 pub use aes::decrypt_aes_128_ecb;
 pub use aes::decrypt_aes_128_cbc;
@@ -26,6 +27,8 @@ pub use primitives::unpad_pkcs7;
 pub use primitives::repeating_key_xor;
 pub use random::MersenneTwister;
 pub use random::mt19937_stream_cipher;
+pub use sha1::sha1;
+pub use sha1::sha1_mac;
 pub use crack::BlockCipherMode;
 pub use crack::find_aes_128_ecb_encrypted_string;
 pub use crack::detect_ecb_cbc;
diff --git a/src/sha1.rs b/src/sha1.rs
new file mode 100644
index 0000000..728c4db
--- /dev/null
+++ b/src/sha1.rs
@@ -0,0 +1,109 @@
+#[cfg(test)] use serialize::hex::ToHex;
+
+pub fn sha1 (bytes: &[u8]) -> [u8; 20] {
+    let mut h: [u32; 5] = [
+        0x67452301,
+        0xEFCDAB89,
+        0x98BADCFE,
+        0x10325476,
+        0xC3D2E1F0,
+    ];
+
+    let ml: u64 = bytes.len() as u64 * 8;
+    let ml_bytes: [u8; 8] = unsafe {
+        ::std::mem::transmute(ml.to_be())
+    };
+
+    let message: Vec<u8> = bytes
+        .iter()
+        .map(|x| *x)
+        .chain(::std::iter::repeat(0x80).take(1))
+        .chain(::std::iter::repeat(0x00).take(55 - (bytes.len() % 64)))
+        .chain(ml_bytes.iter().map(|x| *x))
+        .collect();
+    assert!(message.len() % 64 == 0);
+
+    for chunk in message.chunks(64) {
+        let chunk_words: &[u32; 16] = unsafe {
+            ::std::mem::transmute(chunk.as_ptr())
+        };
+        let mut w: [u32; 80] = unsafe { ::std::mem::uninitialized() };
+        for i in 0..16 {
+            w[i] = ::std::num::Int::from_be(chunk_words[i]);
+        }
+        for i in 16..80 {
+            w[i] = (w[i - 3] ^ w[i - 8] ^ w[i - 14] ^ w[i - 16]).rotate_left(1);
+        }
+
+        let mut a = h[0];
+        let mut b = h[1];
+        let mut c = h[2];
+        let mut d = h[3];
+        let mut e = h[4];
+
+        for i in 0..80 {
+            let (f, k) = match i {
+                0...19  => ((b & c) | (!b & d),          0x5A827999),
+                20...39 => (b ^ c ^ d,                   0x6ED9EBA1),
+                40...59 => ((b & c) | (b & d) | (c & d), 0x8F1BBCDC),
+                60...79 => (b ^ c ^ d,                   0xCA62C1D6),
+                _ => panic!(),
+            };
+
+            let temp = a.rotate_left(5)
+                .wrapping_add(f)
+                .wrapping_add(e)
+                .wrapping_add(k)
+                .wrapping_add(w[i]);
+
+            e = d;
+            d = c;
+            c = b.rotate_left(30);
+            b = a;
+            a = temp;
+        }
+
+        h[0] = h[0].wrapping_add(a);
+        h[1] = h[1].wrapping_add(b);
+        h[2] = h[2].wrapping_add(c);
+        h[3] = h[3].wrapping_add(d);
+        h[4] = h[4].wrapping_add(e);
+    }
+
+    for word in h.iter_mut() {
+        *word = word.to_be();
+    }
+
+    return unsafe { ::std::mem::transmute(h) };
+}
+
+pub fn sha1_mac (bytes: &[u8], key: &[u8]) -> [u8; 20] {
+    let full_bytes: Vec<u8> = key
+        .iter()
+        .map(|x| *x)
+        .chain(bytes.iter().map(|x| *x))
+        .collect();
+    return sha1(&full_bytes[..]);
+}
+
+#[test]
+fn test_sha1 () {
+    let tests = [
+        (
+            &b""[..],
+            "da39a3ee5e6b4b0d3255bfef95601890afd80709"
+        ),
+        (
+            &b"The quick brown fox jumps over the lazy dog"[..],
+            "2fd4e1c67a2d28fced849ee1bb76e7391b93eb12"
+        ),
+        (
+            &b"The quick brown fox jumps over the lazy cog"[..],
+            "de9f2c7fd25e1b3afad3e85a0bd17d9b100db4b3"
+        ),
+    ];
+    for &(input, expected) in tests.iter() {
+        let got = &sha1(input)[..].to_hex();
+        assert_eq!(got, expected);
+    }
+}
diff --git a/tests/set4.rs b/tests/set4.rs
index 5aab8a1..495f548 100644
--- a/tests/set4.rs
+++ b/tests/set4.rs
@@ -131,3 +131,5 @@ fn problem_27 () {
     let ciphertext = matasano::crack_cbc_iv_key(&encode, &verify);
     assert!(verify(&ciphertext[..]).unwrap());
 }
+
+// problem 28 is just matasano::sha1_mac