From 2aadb8fd27c6efc2653d0068f0afbf6a62cc622c Mon Sep 17 00:00:00 2001
From: TrAyZeN <1810leo@gmail.com>
Date: Fri, 19 Apr 2024 14:29:18 +0200
Subject: [PATCH] Add dpa microbenchmark

---
 Cargo.toml      |  4 +++
 benches/dpa.rs  | 70 +++++++++++++++++++++++++++++++++++++++++++++++++
 examples/dpa.rs |  4 +--
 3 files changed, 76 insertions(+), 2 deletions(-)
 create mode 100644 benches/dpa.rs

diff --git a/Cargo.toml b/Cargo.toml
index 2865eaf..1479091 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -36,3 +36,7 @@ harness = false
 [[bench]]
 name = "snr"
 harness = false
+
+[[bench]]
+name = "dpa"
+harness = false
diff --git a/benches/dpa.rs b/benches/dpa.rs
new file mode 100644
index 0000000..55a9e07
--- /dev/null
+++ b/benches/dpa.rs
@@ -0,0 +1,70 @@
+use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
+use muscat::dpa::{dpa, Dpa};
+use muscat::leakage::sbox;
+use ndarray::{Array1, Array2};
+use ndarray_rand::rand::{rngs::StdRng, SeedableRng};
+use ndarray_rand::rand_distr::Uniform;
+use ndarray_rand::RandomExt;
+
+fn selection_function(metadata: Array1<u8>, guess: usize) -> usize {
+    sbox(metadata[1] ^ guess as u8).into()
+}
+
+fn dpa_sequential(leakages: &Array2<f32>, plaintexts: &Array2<u8>) -> Dpa<Array1<u8>> {
+    let mut dpa = Dpa::new(leakages.shape()[1], 256, selection_function);
+
+    for i in 0..leakages.shape()[0] {
+        dpa.update(leakages.row(i), plaintexts.row(i).to_owned());
+    }
+
+    dpa.finalize();
+
+    dpa
+}
+
+fn dpa_parallel(leakages: &Array2<f32>, plaintexts: &Array2<u8>) -> Dpa<Array1<u8>> {
+    dpa(
+        leakages.view(),
+        plaintexts
+            .rows()
+            .into_iter()
+            .map(|x| x.to_owned())
+            .collect::<Array1<Array1<u8>>>()
+            .view(),
+        256,
+        selection_function,
+        500,
+    )
+}
+
+fn bench_dpa(c: &mut Criterion) {
+    // Seed rng to get the same output each run
+    let mut rng = StdRng::seed_from_u64(0);
+
+    let mut group = c.benchmark_group("dpa");
+
+    group.measurement_time(std::time::Duration::from_secs(60));
+
+    for nb_traces in [1000, 2000, 5000].into_iter() {
+        let leakages = Array2::random_using((nb_traces, 5000), Uniform::new(-2., 2.), &mut rng);
+        let plaintexts =
+            Array2::random_using((nb_traces, 16), Uniform::new_inclusive(0, 255), &mut rng);
+
+        group.bench_with_input(
+            BenchmarkId::new("sequential", nb_traces),
+            &(&leakages, &plaintexts),
+            |b, (leakages, plaintexts)| b.iter(|| dpa_sequential(leakages, plaintexts)),
+        );
+
+        group.bench_with_input(
+            BenchmarkId::new("parallel", nb_traces),
+            &(&leakages, &plaintexts),
+            |b, (leakages, plaintexts)| b.iter(|| dpa_parallel(leakages, plaintexts)),
+        );
+    }
+
+    group.finish();
+}
+
+criterion_group!(benches, bench_dpa);
+criterion_main!(benches);
diff --git a/examples/dpa.rs b/examples/dpa.rs
index 449b979..d80029c 100644
--- a/examples/dpa.rs
+++ b/examples/dpa.rs
@@ -1,9 +1,9 @@
 use anyhow::Result;
 use indicatif::ProgressIterator;
-use muscat::dpa::*;
+use muscat::dpa::Dpa;
 use muscat::leakage::sbox;
 use muscat::util::read_array2_from_npy_file;
-use ndarray::*;
+use ndarray::{s, Array1, Array2};
 use rayon::iter::{ParallelBridge, ParallelIterator};
 
 // traces format