add three criterion benchmarks

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,19 @@
+use criterion::{criterion_group, criterion_main, Criterion};
+use module_lwe::decrypt::decrypt;
+use module_lwe::keygen::keygen;
+use module_lwe::encrypt::encrypt;
+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))
+    });
+}
+
+criterion_group!(benches, bench_decrypt);
+criterion_main!(benches);

benches/benchmark_encrypt.rs

@@ -0,0 +1,17 @@
+use criterion::{criterion_group, criterion_main, Criterion};
+use module_lwe::encrypt::encrypt;
+use module_lwe::keygen::keygen;
+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))
+    });
+}
+
+criterion_group!(benches, bench_encrypt);
+criterion_main!(benches);

benches/benchmark_keygen.rs

@@ -0,0 +1,13 @@
+use criterion::{criterion_group, criterion_main, Criterion};
+use module_lwe::keygen::keygen;
+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))
+    });
+}
+
+criterion_group!(benches, bench_keygen);
+criterion_main!(benches);