2 files changed, 16 insertions, 0 deletions
diff --git a/src/crack.rs b/src/crack.rs
index 92950c4..b351249 100644
--- a/src/crack.rs
+++ b/src/crack.rs
@@ -386,6 +386,21 @@ pub fn clone_mersenne_twister_from_output (outputs: &[u32]) -> MersenneTwister {
     return MersenneTwister::from_seed((state, 0));
 }
 
+pub fn recover_16_bit_mt19937_key (ciphertext: &[u8], suffix: &[u8]) -> Option<u16> {
+    for _key in 0..65536u32 {
+        let key = _key as u16;
+        let plaintext = ::random::mt19937_stream_cipher(
+            ciphertext,
+            key as u32
+        );
+        if &plaintext[(ciphertext.len() - suffix.len())..] == suffix {
+            return Some(key);
+        }
+    }
+
+    return None;
+}
+
 fn crack_single_byte_xor_with_confidence (input: &[u8]) -> (u8, f64) {
     let mut min_diff = 100.0;
     let mut best_key = 0;
diff --git a/src/lib.rs b/src/lib.rs
index 089250c..bbaf93f 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -37,3 +37,4 @@ pub use crack::crack_single_byte_xor;
 pub use crack::crack_repeating_key_xor;
 pub use crack::crack_fixed_nonce_ctr_statistically;
 pub use crack::clone_mersenne_twister_from_output;
+pub use crack::recover_16_bit_mt19937_key;