light_tool/
md5.rs

1/// MD5 结构体
2pub struct Md5 {
3    data: Vec<u8>,       // 原始数据块
4    bit_len: u64,        // 消息总位数
5    state: [u32; 4],     // MD5 状态变量（A, B, C, D）
6}
7
8impl Md5 {
9    /// 初始化 MD5 状态变量
10    pub fn new() -> Self {
11        Self {
12            data: Vec::new(),
13            bit_len: 0,
14            state: [
15                0x67452301, // A
16                0xefcdab89, // B
17                0x98badcfe, // C
18                0x10325476, // D
19            ],
20        }
21    }
22
23    /// 更新 MD5 数据块
24    pub fn update(&mut self, input: &[u8]) {
25        self.data.extend_from_slice(input);
26        self.bit_len += (input.len() as u64) * 8;
27        while self.data.len() >= 64 {
28            let block: [u8; 64] = self.data[..64].try_into().unwrap();
29            self.process_block(&block);
30            self.data.drain(..64);
31        }
32    }
33
34    /// 处理 64 字节块
35    fn process_block(&mut self, block: &[u8; 64]) {
36        const S: [[u32; 4]; 4] = [
37            [7, 12, 17, 22], [5, 9, 14, 20], [4, 11, 16, 23], [6, 10, 15, 21],
38        ];
39        const K: [u32; 64] = [
40            0xd76aa478, 0xe8c7b756, 0x242070db, 0xc1bdceee, 0xf57c0faf, 0x4787c62a, 0xa8304613,
41            0xfd469501, 0x698098d8, 0x8b44f7af, 0xffff5bb1, 0x895cd7be, 0x6b901122, 0xfd987193,
42            0xa679438e, 0x49b40821, 0xf61e2562, 0xc040b340, 0x265e5a51, 0xe9b6c7aa, 0xd62f105d,
43            0x02441453, 0xd8a1e681, 0xe7d3fbc8, 0x21e1cde6, 0xc33707d6, 0xf4d50d87, 0x455a14ed,
44            0xa9e3e905, 0xfcefa3f8, 0x676f02d9, 0x8d2a4c8a, 0xfffa3942, 0x8771f681, 0x6d9d6122,
45            0xfde5380c, 0xa4beea44, 0x4bdecfa9, 0xf6bb4b60, 0xbebfbc70, 0x289b7ec6, 0xeaa127fa,
46            0xd4ef3085, 0x04881d05, 0xd9d4d039, 0xe6db99e5, 0x1fa27cf8, 0xc4ac5665, 0xf4292244,
47            0x432aff97, 0xab9423a7, 0xfc93a039, 0x655b59c3, 0x8f0ccc92, 0xffeff47d, 0x85845dd1,
48            0x6fa87e4f, 0xfe2ce6e0, 0xa3014314, 0x4e0811a1, 0xf7537e82, 0xbd3af235, 0x2ad7d2bb,
49            0xeb86d391,
50        ];
51
52        let mut a = self.state[0];
53        let mut b = self.state[1];
54        let mut c = self.state[2];
55        let mut d = self.state[3];
56
57        let mut m = [0u32; 16];
58        for (i, chunk) in block.chunks(4).enumerate() {
59            m[i] = u32::from_le_bytes(chunk.try_into().unwrap());
60        }
61
62        for i in 0..64 {
63            let (f, g) = match i {
64                0..=15 => ((b & c) | (!b & d), i),
65                16..=31 => ((d & b) | (!d & c), (5 * i + 1) % 16),
66                32..=47 => (b ^ c ^ d, (3 * i + 5) % 16),
67                48..=63 => (c ^ (b | !d), (7 * i) % 16),
68                _ => unreachable!(),
69            };
70
71            let temp = d;
72            d = c;
73            c = b;
74            b = b.wrapping_add(
75                a.wrapping_add(f)
76                    .wrapping_add(K[i])
77                    .wrapping_add(m[g])
78                    .rotate_left(S[i / 16][i % 4]),
79            );
80            a = temp;
81        }
82
83        self.state[0] = self.state[0].wrapping_add(a);
84        self.state[1] = self.state[1].wrapping_add(b);
85        self.state[2] = self.state[2].wrapping_add(c);
86        self.state[3] = self.state[3].wrapping_add(d);
87    }
88
89    /// 计算最终的哈希值
90    pub fn finalize(mut self) -> [u8; 16] {
91        let bit_len_bytes = self.bit_len.to_le_bytes();
92        self.update(&[0x80]);
93        while self.data.len() % 64 != 56 {
94            self.update(&[0x00]);
95        }
96        self.update(&bit_len_bytes);
97
98        let mut hash = [0u8; 16];
99        for (i, &val) in self.state.iter().enumerate() {
100            hash[i * 4..(i + 1) * 4].copy_from_slice(&val.to_le_bytes());
101        }
102        hash
103    }
104}
105
106/// Generate md5 string
107///
108/// # Example
109///
110/// ```no_run
111/// use light_tool::md5;
112/// println!("md5 string: {}", md5::str("hello world"))
113/// ```
114pub fn str<T>(input: T) -> String
115where
116    T: AsRef<[u8]>,
117{
118    let mut md5 = Md5::new();
119    md5.update(input.as_ref());
120    let digest = md5.finalize();
121    digest.iter().map(|byte| format!("{:02x}", byte)).collect()
122}
123
124/// Generate md5 string with salt
125///
126/// # Example
127///
128/// ```no_run
129/// use light_tool::md5;
130/// println!("md5 string with salt: {}", md5::salt("123", "456"))
131/// ```
132pub fn salt<T>(input: T, salt: T) -> String
133where
134    T: AsRef<[u8]>,
135{
136    let mut combined = Vec::with_capacity(input.as_ref().len() + salt.as_ref().len());
137    combined.extend_from_slice(input.as_ref());
138    combined.extend_from_slice(salt.as_ref());
139    str(combined)
140}
141
142#[cfg(test)]
143mod tests {
144    use super::*;
145
146    #[test]
147    fn test_md5() {
148        assert_eq!(
149            salt("123", "456"),
150            "e10adc3949ba59abbe56e057f20f883e"
151        );
152        assert_eq!(
153            str("123456"),
154            "e10adc3949ba59abbe56e057f20f883e"
155        );
156        assert_eq!(
157            str(b"hello world"),
158            "5eb63bbbe01eeed093cb22bb8f5acdc3"
159        );
160        assert_eq!(
161            str(b""),
162            "d41d8cd98f00b204e9800998ecf8427e"
163        );
164    }
165}
light_tool/md5.rs

light_tool/
md5.rs
