light_tool/
hmac.rs

1use std::fmt::Write;
2
3
4/// SHA-256 常量
5const K: [u32; 64] = [
6    0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5, 0x3956c25b, 0x59f111f1, 0x923f82a4,
7    0xab1c5ed5, 0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3, 0x72be5d74, 0x80deb1fe,
8    0x9bdc06a7, 0xc19bf174, 0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc, 0x2de92c6f,
9    0x4a7484aa, 0x5cb0a9dc, 0x76f988da, 0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7,
10    0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967, 0x27b70a85, 0x2e1b2138, 0x4d2c6dfc,
11    0x53380d13, 0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85, 0xa2bfe8a1, 0xa81a664b,
12    0xc24b8b70, 0xc76c51a3, 0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070, 0x19a4c116,
13    0x1e376c08, 0x2748774c, 0x34b0bcb5, 0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
14    0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208, 0x90befffa, 0xa4506ceb, 0xbef9a3f7,
15    0xc67178f2,
16];
17
18/// 初始哈希值
19const INIT_HASH: [u32; 8] = [
20    0x6a09e667, 0xbb67ae85, 0x3c6ef372, 0xa54ff53a, 0x510e527f, 0x9b05688c, 0x1f83d9ab, 0x5be0cd19,
21];
22
23/// SHA-256 哈希函数
24///
25/// # Example
26/// ```rust
27/// use light_tool::hmac;
28/// println!("sha256: {}", hmac::sha256("Hello, world!"))
29/// ```
30pub fn sha256(input: &str) -> String {
31    let bytes = input.as_bytes();
32    let bit_len = bytes.len() * 8;
33    let mut data = bytes.to_vec();
34
35    // 填充数据：附加 1 位 + 填充 0 + 附加原始长度
36    data.push(0x80);
37    while (data.len() * 8) % 512 != 448 {
38        data.push(0x00);
39    }
40
41    for &byte in &(bit_len as u64).to_be_bytes() {
42        data.push(byte);
43    }
44
45    // 分块处理，每块 512 位（64 字节）
46    let mut hash = INIT_HASH;
47    for chunk in data.chunks(64) {
48        let mut w = [0u32; 64];
49        for i in 0..16 {
50            w[i] = u32::from_be_bytes([
51                chunk[i * 4],
52                chunk[i * 4 + 1],
53                chunk[i * 4 + 2],
54                chunk[i * 4 + 3],
55            ]);
56        }
57        for i in 16..64 {
58            let s0 = w[i - 15].rotate_right(7) ^ w[i - 15].rotate_right(18) ^ (w[i - 15] >> 3);
59            let s1 = w[i - 2].rotate_right(17) ^ w[i - 2].rotate_right(19) ^ (w[i - 2] >> 10);
60            w[i] = w[i - 16]
61                .wrapping_add(s0)
62                .wrapping_add(w[i - 7])
63                .wrapping_add(s1);
64        }
65
66        let mut a = hash[0];
67        let mut b = hash[1];
68        let mut c = hash[2];
69        let mut d = hash[3];
70        let mut e = hash[4];
71        let mut f = hash[5];
72        let mut g = hash[6];
73        let mut h = hash[7];
74
75        for i in 0..64 {
76            let s1 = e.rotate_right(6) ^ e.rotate_right(11) ^ e.rotate_right(25);
77            let ch = (e & f) ^ (!e & g);
78            let temp1 = h
79                .wrapping_add(s1)
80                .wrapping_add(ch)
81                .wrapping_add(K[i])
82                .wrapping_add(w[i]);
83            let s0 = a.rotate_right(2) ^ a.rotate_right(13) ^ a.rotate_right(22);
84            let maj = (a & b) ^ (a & c) ^ (b & c);
85            let temp2 = s0.wrapping_add(maj);
86
87            h = g;
88            g = f;
89            f = e;
90            e = d.wrapping_add(temp1);
91            d = c;
92            c = b;
93            b = a;
94            a = temp1.wrapping_add(temp2);
95        }
96
97        hash[0] = hash[0].wrapping_add(a);
98        hash[1] = hash[1].wrapping_add(b);
99        hash[2] = hash[2].wrapping_add(c);
100        hash[3] = hash[3].wrapping_add(d);
101        hash[4] = hash[4].wrapping_add(e);
102        hash[5] = hash[5].wrapping_add(f);
103        hash[6] = hash[6].wrapping_add(g);
104        hash[7] = hash[7].wrapping_add(h);
105    }
106
107    // 格式化为十六进制字符串
108    let mut result = String::new();
109    for &value in &hash {
110        write!(&mut result, "{:08x}", value).expect("Failed to write hash");
111    }
112
113    result
114}
115
116#[cfg(test)]
117mod tests {
118    use super::*;
119
120    #[test]
121    fn test_sha256() {
122        let input = "Hello, world!";
123        let expected_output = "315f5bdb76d078c43b8ac0064e4a0164612b1fce77c869345bfc94c75894edd3";
124        let actual_output = sha256(input);
125        assert_eq!(expected_output, actual_output)
126    }
127}
light_tool/hmac.rs

light_tool/
hmac.rs
