kryptos 0.6.3

A Rust implementation of various older, insecure, cryptographic ciphers.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

extern crate regex;
use self::regex::{Captures, Regex};
use common::binary_to_char;

/// Baconian Cipher
///
/// The struct is generated through the new() function.
///
pub struct Baconian {
    distinct: bool,
}

impl Baconian {
    /// Initializes a rail fence cipher with a supplied height.
    ///
    /// # Examples
    ///
    /// ```
    /// use kryptos::ciphers::baconian::Baconian;
    ///
    /// let b = Baconian::new(true).unwrap();
    /// ```
    ///
    pub fn new(distinct: bool) -> Result<Self, String> {
        Ok(Baconian { distinct })
    }

    /// Enciphers a message with a baconian cipher.
    ///
    /// # Examples
    ///
    /// ```
    /// use kryptos::ciphers::baconian::Baconian;
    ///
    /// let b = Baconian::new(true).unwrap();
    /// assert_eq!("ABBAABABAABAABABAABB AAAABAABAA ABABAAABAAABBBBBAABB AAAAA BAABAAABAAAAABABAAABAABAABAABB", b.encipher("Must be kept a secret").unwrap());
    /// ```
    ///
    pub fn encipher(&self, plaintext: &str) -> Result<String, String> {
        if self.distinct {
            return Ok(self.distinct_encipher(plaintext).unwrap());
        }
        Ok(String::from(plaintext))
    }

    /// Deciphers a message with a baconian cipher.
    ///
    /// # Examples
    ///
    /// ```
    /// use kryptos::ciphers::baconian::Baconian;
    ///
    /// let b = Baconian::new(true).unwrap();
    /// assert_eq!("MUST BE KEPT A SECRET", b.decipher("ABBAABABAABAABABAABB AAAABAABAA ABABAAABAAABBBBBAABB AAAAA BAABAAABAAAAABABAAABAABAABAABB").unwrap());
    /// ```
    ///
    pub fn decipher(&self, ciphertext: &str) -> Result<String, String> {
        let binary = ciphertext
            .chars()
            .map(|c| match c {
                'A' => '0',
                'B' => '1',
                _ => c,
            })
            .collect::<String>();

        let re = Regex::new(r"[01]{5}").unwrap();
        let result = re.replace_all(&binary, |caps: &Captures| {
            format!("{}", binary_to_char(&caps[0]).unwrap())
        });

        Ok(result.to_string())
    }

    // Encipher the text using the distinct method.
    fn distinct_encipher(&self, plaintext: &str) -> Result<String, String> {
        let binary = plaintext
            .chars()
            .map(|c| match c as u8 {
                65..=90 => format!("{:05b}", c as u8 - 65),
                97..=122 => format!("{:05b}", c as u8 - 97),
                _ => c.to_string(),
            })
            .collect::<String>();

        Ok(binary
            .chars()
            .map(|c| match c {
                '0' => 'A',
                '1' => 'B',
                _ => c,
            })
            .collect::<String>())
    }
}

#[cfg(test)]
mod tests {
    use super::Baconian;

    #[test]
    fn distinct() {
        assert!(Baconian::new(true).is_ok());
    }

    #[test]
    fn encipher_with_distinct() {
        let b = Baconian::new(true).unwrap();
        assert_eq!("ABBAABABAABAABABAABB AAAABAABAA ABABAAABAAABBBBBAABB AAAAA BAABAAABAAAAABABAAABAABAABAABB", b.encipher("Must be kept a secret").unwrap());
    }

    #[test]
    fn decipher_with_distinct() {
        let b = Baconian::new(true).unwrap();
        assert_eq!("MUST BE KEPT A SECRET", b.decipher("ABBAABABAABAABABAABB AAAABAABAA ABABAAABAAABBBBBAABB AAAAA BAABAAABAAAAABABAAABAABAABAABB").unwrap());
    }
}