56 lines
1.9 KiB
Rust
56 lines
1.9 KiB
Rust
|
//! SM4 国标测试向量（GB/T 32907-2016 附录 A）
|
|||
|
|
//!
|
|||
|
|
//! A.1 示例1：单次 ECB 加密
|
|||
|
|
//! A.2 示例2：1,000,000 次迭代 ECB 加密（验证算法迭代正确性）
|
|||
|
|
|
|||
|
|
use libsmx::sm4::{sm4_decrypt_ecb, sm4_encrypt_ecb};
|
|||
|
|
|
|||
|
|
/// GB/T 32907-2016 附录 A.1
|
|||
|
|
/// 密钥：0123456789abcdeffedcba9876543210
|
|||
|
|
/// 明文：0123456789abcdeffedcba9876543210
|
|||
|
|
/// 密文：681edf34d206965e86b3e94f536e4246
|
|||
|
|
#[test]
|
|||
|
|
fn test_sm4_ecb_vector_a1_single() {
|
|||
|
|
let key = hex::decode("0123456789abcdeffedcba9876543210").unwrap();
|
|||
|
|
let plaintext = hex::decode("0123456789abcdeffedcba9876543210").unwrap();
|
|||
|
|
let expected_ct = hex::decode("681edf34d206965e86b3e94f536e4246").unwrap();
|
|||
|
|
|
|||
|
|
let key_arr: [u8; 16] = key.try_into().unwrap();
|
|||
|
|
let ct = sm4_encrypt_ecb(&key_arr, &plaintext);
|
|||
|
|
assert_eq!(ct, expected_ct, "GB/T 32907 附录 A.1 加密失败");
|
|||
|
|
|
|||
|
|
let pt = sm4_decrypt_ecb(&key_arr, &ct);
|
|||
|
|
assert_eq!(pt, plaintext, "GB/T 32907 附录 A.1 解密失败");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// GB/T 32907-2016 附录 A.2
|
|||
|
|
/// 密钥：0123456789abcdeffedcba9876543210
|
|||
|
|
/// 明文：0123456789abcdeffedcba9876543210（反复迭代 1,000,000 次）
|
|||
|
|
/// 密文：595298c7c6fd271f0402f804c33d3f66
|
|||
|
|
#[test]
|
|||
|
|
fn test_sm4_ecb_vector_a2_million_iterations() {
|
|||
|
|
let key: [u8; 16] = hex::decode("0123456789abcdeffedcba9876543210")
|
|||
|
|
.unwrap()
|
|||
|
|
.try_into()
|
|||
|
|
.unwrap();
|
|||
|
|
|
|||
|
|
let mut data: Vec<u8> = hex::decode("0123456789abcdeffedcba9876543210").unwrap();
|
|||
|
|
|
|||
|
|
for _ in 0..1_000_000 {
|
|||
|
|
data = sm4_encrypt_ecb(&key, &data);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let expected = hex::decode("595298c7c6fd271f0402f804c33d3f66").unwrap();
|
|||
|
|
assert_eq!(data, expected, "GB/T 32907 附录 A.2 百万次迭代失败");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// ECB 解密是加密的逆操作（往返测试）
|
|||
|
|
#[test]
|
|||
|
|
fn test_sm4_ecb_roundtrip() {
|
|||
|
|
let key = [0x01u8; 16];
|
|||
|
|
let plaintext = b"SM4 ECB test!!!\x00";
|
|||
|
|
let ct = sm4_encrypt_ecb(&key, plaintext);
|
|||
|
|
let pt = sm4_decrypt_ecb(&key, &ct);
|
|||
|
|
assert_eq!(pt.as_slice(), plaintext.as_slice());
|
|||
|
|
}
|