初始提交：SM2/SM3/SM4/SM9 密码算法库

- SM3 哈希函数 (GB/T 32905-2013) - SM4 分组密码，支持 ECB/CBC/OFB/CFB/CTR/GCM/CCM/XTS 模式 (GB/T 32907-2016) - SM2 椭圆曲线密码 (GB/T 32918.1-5-2016) - SM9 标识密码 (GB/T 38635.1-2-2020) - 全程常量时间运算 - 支持 no_std，带 alloc 特性 - 完整的国标测试向量 - Criterion 性能基准测试
2026-03-07 13:03:10 +08:00
commit e929e6a103
34 changed files with 7530 additions and 0 deletions
@@ -0,0 +1,63 @@
+use criterion::{criterion_group, criterion_main, Criterion};
+use libsmx::sm2::{decrypt, encrypt, generate_keypair, get_e, get_z, sign, verify};
+use rand::rngs::OsRng;
+
+fn bench_sm2_keygen(c: &mut Criterion) {
+    c.bench_function("SM2/keygen", |b| {
+        b.iter(|| generate_keypair(&mut OsRng))
+    });
+}
+
+fn bench_sm2_sign(c: &mut Criterion) {
+    let (pri_key, pub_key) = generate_keypair(&mut OsRng);
+    let id = b"benchuser";
+    let msg = b"benchmark message for SM2 sign";
+    let z = get_z(id, &pub_key);
+    let e = get_e(&z, msg);
+
+    c.bench_function("SM2/sign", |b| {
+        b.iter(|| sign(&e, &pri_key, &mut OsRng))
+    });
+}
+
+fn bench_sm2_verify(c: &mut Criterion) {
+    let (pri_key, pub_key) = generate_keypair(&mut OsRng);
+    let id = b"benchuser";
+    let msg = b"benchmark message for SM2 verify";
+    let z = get_z(id, &pub_key);
+    let e = get_e(&z, msg);
+    let sig = sign(&e, &pri_key, &mut OsRng);
+
+    c.bench_function("SM2/verify", |b| {
+        b.iter(|| verify(&e, &pub_key, &sig))
+    });
+}
+
+fn bench_sm2_encrypt(c: &mut Criterion) {
+    let (_pri_key, pub_key) = generate_keypair(&mut OsRng);
+    let msg = b"SM2 encryption benchmark plaintext";
+
+    c.bench_function("SM2/encrypt", |b| {
+        b.iter(|| encrypt(&pub_key, msg, &mut OsRng))
+    });
+}
+
+fn bench_sm2_decrypt(c: &mut Criterion) {
+    let (pri_key, pub_key) = generate_keypair(&mut OsRng);
+    let msg = b"SM2 decryption benchmark plaintext";
+    let ct = encrypt(&pub_key, msg, &mut OsRng).unwrap();
+
+    c.bench_function("SM2/decrypt", |b| {
+        b.iter(|| decrypt(&pri_key, &ct))
+    });
+}
+
+criterion_group!(
+    benches,
+    bench_sm2_keygen,
+    bench_sm2_sign,
+    bench_sm2_verify,
+    bench_sm2_encrypt,
+    bench_sm2_decrypt
+);
+criterion_main!(benches);
@@ -0,0 +1,16 @@
+use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
+use libsmx::sm3::Sm3Hasher;
+
+fn bench_sm3_digest(c: &mut Criterion) {
+    let mut group = c.benchmark_group("SM3");
+    for size in [64usize, 1024, 65536] {
+        let data = vec![0x42u8; size];
+        group.bench_with_input(BenchmarkId::new("digest", size), &data, |b, d| {
+            b.iter(|| Sm3Hasher::digest(d));
+        });
+    }
+    group.finish();
+}
+
+criterion_group!(benches, bench_sm3_digest);
+criterion_main!(benches);
@@ -0,0 +1,24 @@
+use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
+use libsmx::sm4::{sm4_encrypt_ecb, Sm4Key};
+
+fn bench_sm4_ecb(c: &mut Criterion) {
+    let mut group = c.benchmark_group("SM4-ECB");
+    let key = [0u8; 16];
+    for size in [16usize, 1024, 65536] {
+        let data = vec![0u8; size];
+        group.bench_with_input(BenchmarkId::new("encrypt", size), &data, |b, d| {
+            b.iter(|| sm4_encrypt_ecb(&key, d));
+        });
+    }
+    group.finish();
+}
+
+fn bench_sm4_key_new(c: &mut Criterion) {
+    let key = [0u8; 16];
+    c.bench_function("SM4/key_expand", |b| {
+        b.iter(|| Sm4Key::new(&key));
+    });
+}
+
+criterion_group!(benches, bench_sm4_ecb, bench_sm4_key_new);
+criterion_main!(benches);
@@ -0,0 +1,81 @@
+use criterion::{criterion_group, criterion_main, Criterion};
+use libsmx::sm9::{
+    generate_enc_master_keypair, generate_enc_user_key, generate_sign_master_keypair,
+    generate_sign_user_key, sm9_decrypt, sm9_encrypt, sm9_sign, sm9_verify, Sm9EncPubKey,
+    Sm9SignPubKey,
+};
+use rand::rngs::OsRng;
+
+fn bench_sm9_sign_keygen(c: &mut Criterion) {
+    c.bench_function("SM9/sign_master_keygen", |b| {
+        b.iter(|| generate_sign_master_keypair(&mut OsRng))
+    });
+}
+
+fn bench_sm9_user_sign_keygen(c: &mut Criterion) {
+    let (master_priv, _) = generate_sign_master_keypair(&mut OsRng);
+    let id = b"benchuser";
+    c.bench_function("SM9/sign_user_keygen", |b| {
+        b.iter(|| generate_sign_user_key(&master_priv, id))
+    });
+}
+
+fn bench_sm9_sign(c: &mut Criterion) {
+    let (master_priv, sign_pub) = generate_sign_master_keypair(&mut OsRng);
+    let pub_key = Sm9SignPubKey::from_bytes(sign_pub.as_bytes()).unwrap();
+    let id = b"benchuser";
+    let da = generate_sign_user_key(&master_priv, id).unwrap();
+    let msg = b"SM9 signature benchmark message";
+
+    c.bench_function("SM9/sign", |b| {
+        b.iter(|| sm9_sign(msg, &da, &pub_key, &mut OsRng))
+    });
+}
+
+fn bench_sm9_verify(c: &mut Criterion) {
+    let (master_priv, sign_pub) = generate_sign_master_keypair(&mut OsRng);
+    let pub_key = Sm9SignPubKey::from_bytes(sign_pub.as_bytes()).unwrap();
+    let id = b"benchuser";
+    let da = generate_sign_user_key(&master_priv, id).unwrap();
+    let msg = b"SM9 verify benchmark message";
+    let (h, s) = sm9_sign(msg, &da, &pub_key, &mut OsRng).unwrap();
+
+    c.bench_function("SM9/verify", |b| {
+        b.iter(|| sm9_verify(msg, &h, &s, id, &pub_key))
+    });
+}
+
+fn bench_sm9_encrypt(c: &mut Criterion) {
+    let (_master_priv, enc_pub) = generate_enc_master_keypair(&mut OsRng);
+    let pub_key = Sm9EncPubKey::from_bytes(enc_pub.as_bytes()).unwrap();
+    let id = b"benchuser";
+    let msg = b"SM9 encryption benchmark plaintext";
+
+    c.bench_function("SM9/encrypt", |b| {
+        b.iter(|| sm9_encrypt(id, msg, &pub_key, &mut OsRng))
+    });
+}
+
+fn bench_sm9_decrypt(c: &mut Criterion) {
+    let (master_priv, enc_pub) = generate_enc_master_keypair(&mut OsRng);
+    let pub_key = Sm9EncPubKey::from_bytes(enc_pub.as_bytes()).unwrap();
+    let id = b"benchuser";
+    let de = generate_enc_user_key(&master_priv, id).unwrap();
+    let msg = b"SM9 decryption benchmark plaintext";
+    let ct = sm9_encrypt(id, msg, &pub_key, &mut OsRng).unwrap();
+
+    c.bench_function("SM9/decrypt", |b| {
+        b.iter(|| sm9_decrypt(id, &ct, &de))
+    });
+}
+
+criterion_group!(
+    benches,
+    bench_sm9_sign_keygen,
+    bench_sm9_user_sign_keygen,
+    bench_sm9_sign,
+    bench_sm9_verify,
+    bench_sm9_encrypt,
+    bench_sm9_decrypt,
+);
+criterion_main!(benches);