Merge pull request #26 from lattice-based-cryptography/add_criterion_benchmarking

add three criterion benchmarks for `keygen`, `encrypt`, `decrypt` using the `criterion` crate.

Author: jacksonwalters

Cargo.toml

@@ -14,4 +14,22 @@ num-traits = "=0.2.19"
 rand = "0.8.5"
 rand_distr = "0.4.3"
 ntt = "0.1.9"
-ring-lwe = "0.1.5"
+ring-lwe = "0.1.5"
+
+[dev-dependencies]
+criterion = "0.5.1"
+
+[[bench]]
+name = "benchmark_keygen"
+path = "benches/benchmark_keygen.rs"
+harness = false
+
+[[bench]]
+name = "benchmark_encrypt"
+path = "benches/benchmark_encrypt.rs"
+harness = false
+
+[[bench]]
+name = "benchmark_decrypt"
+path = "benches/benchmark_decrypt.rs"
+harness = false

benches/benchmark_decrypt.rs

@@ -0,0 +1,32 @@
+use criterion::{criterion_group, criterion_main, Criterion};
+use module_lwe::decrypt::{decrypt,decrypt_string};
+use module_lwe::keygen::{keygen,keygen_string};
+use module_lwe::encrypt::{encrypt,encrypt_string};
+use module_lwe::utils::Parameters;
+
+fn bench_decrypt(c: &mut Criterion) {
+    let params = Parameters::default();
+    let (pk, sk) = keygen(&params, None);
+    let m_b = vec![0, 1, 0, 1, 1, 0, 1, 0]; // Example binary message
+    let (u, v) = encrypt(&pk.0, &pk.1, &m_b, &params, None);
+
+    c.bench_function("decrypt", |b| {
+        b.iter(|| decrypt(&sk, &u, &v, &params))
+    });
+}
+
+fn bench_decrypt_string(c: &mut Criterion) {
+    let params = Parameters::default();
+    let keypair = keygen_string(&params, None);
+    let sk_string = keypair.get("secret").unwrap();
+    let pk_string = keypair.get("public").unwrap();
+    let message = String::from("hello");
+    let ciphertext_string = encrypt_string(&pk_string, &message, &params, None);
+    
+    c.bench_function("decrypt_string", |b| {
+        b.iter(|| decrypt_string(&sk_string, &ciphertext_string, &params))
+    });
+}
+
+criterion_group!(benches, bench_decrypt, bench_decrypt_string);
+criterion_main!(benches);

benches/benchmark_encrypt.rs

@@ -0,0 +1,28 @@
+use criterion::{criterion_group, criterion_main, Criterion};
+use module_lwe::encrypt::{encrypt,encrypt_string};
+use module_lwe::keygen::{keygen,keygen_string};
+use module_lwe::utils::Parameters;
+
+fn bench_encrypt(c: &mut Criterion) {
+    let params = Parameters::default();
+    let (pk, _) = keygen(&params, None);
+    let m_b = vec![0, 1, 0, 1, 1, 0, 1, 0]; // Example binary message
+
+    c.bench_function("encrypt", |b| {
+        b.iter(|| encrypt(&pk.0, &pk.1, &m_b, &params, None))
+    });
+}
+
+fn bench_encrypt_string(c: &mut Criterion) {
+    let params = Parameters::default();
+    let keypair = keygen_string(&params, None);
+    let pk_string = keypair.get("public").unwrap();
+    let message = String::from("hello");
+
+    c.bench_function("encrypt_string", |b| {
+        b.iter(|| encrypt_string(&pk_string, &message, &params, None))
+    });
+}
+
+criterion_group!(benches, bench_encrypt, bench_encrypt_string);
+criterion_main!(benches);

benches/benchmark_keygen.rs

@@ -0,0 +1,21 @@
+use criterion::{criterion_group, criterion_main, Criterion};
+use module_lwe::keygen::{keygen,keygen_string};
+use module_lwe::utils::Parameters;
+
+fn bench_keygen(c: &mut Criterion) {
+    let params = Parameters::default();
+    c.bench_function("keygen", |b| {
+        b.iter(|| keygen(&params, None))
+    });
+}
+
+fn bench_keygen_string(c: &mut Criterion) {
+    let params = Parameters::default();
+    
+    c.bench_function("keygen_string", |b| {
+        b.iter(|| keygen_string(&params, None))
+    });
+}
+
+criterion_group!(benches, bench_keygen, bench_keygen_string);
+criterion_main!(benches);