dilithium, hybrid, ctap implementation, experiments

third_party/dilithium/benches/sign_bench.rs

@@ -0,0 +1,70 @@
+// Benchmarks for key generation and signing with Dilithium.
+// cargo criterion --features std
+
+extern crate core;
+extern crate criterion;
+extern crate dilithium;
+extern crate rng256;
+
+use core::time::Duration;
+use criterion::*;
+use dilithium::sign::SecKey;
+use rng256::Rng256;
+
+const SAMPLE_SIZE: usize = 1000;
+const MEASUREMENT_TIME: Duration = Duration::from_secs(10);
+
+fn bench_sk(c: &mut Criterion) {
+    let mut rng = rng256::ThreadRng256 {};
+
+    c.bench_function("gensk", |b| {
+        b.iter_batched(
+            || {},
+            |_| {
+                SecKey::gensk(&mut rng);
+            },
+            BatchSize::SmallInput,
+        )
+    });
+}
+
+fn bench_pk(c: &mut Criterion) {
+    let mut rng = rng256::ThreadRng256 {};
+
+    c.bench_function("genpk", |b| {
+        b.iter_batched(
+            || SecKey::gensk(&mut rng),
+            |sk| {
+                sk.genpk();
+            },
+            BatchSize::SmallInput,
+        )
+    });
+}
+
+fn bench_sign(c: &mut Criterion) {
+    const MESSAGE_LENGTH: usize = 64;
+    let mut rng = rng256::ThreadRng256 {};
+
+    c.bench_function("sign", |b| {
+        b.iter_batched(
+            || {
+                let sk = SecKey::gensk(&mut rng);
+                let mut message = [0; MESSAGE_LENGTH];
+                rng.fill_bytes(&mut message);
+                (sk, message)
+            },
+            |(sk, message)| {
+                sk.sign(&message);
+            },
+            BatchSize::SmallInput,
+        )
+    });
+}
+
+criterion_group! {
+    name = benches;
+    config = Criterion::default().sample_size(SAMPLE_SIZE).measurement_time(MEASUREMENT_TIME);
+    targets = bench_sk, bench_pk, bench_sign
+}
+criterion_main!(benches);