hal8174/alan-side-channel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Parallelize calculate models

Browse source
This commit is contained in:
hal8174 2025-07-08 22:40:41 +02:00
parent 0dedbd16a3
commit fc9133056c
Signed by: hal8174
SSH key fingerprint: SHA256:JwuqS+eVfISfKr+DkDQ6NWAbGd1jFAHkPpCM1yCnlTs
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/main.rs
View file

@ -6,7 +6,9 @@ use aes::{
cipher::{BlockEncrypt, KeyInit, generic_array::GenericArray}, cipher::{BlockEncrypt, KeyInit, generic_array::GenericArray},
}; };
use clap::Parser; use clap::Parser;
use rayon::iter::{IntoParallelIterator, ParallelIterator}; use rayon::iter::{
IndexedParallelIterator, IntoParallelIterator, IntoParallelRefMutIterator, ParallelIterator,
};
use std::{ use std::{
io::{BufRead, Read}, io::{BufRead, Read},
path::{Path, PathBuf}, path::{Path, PathBuf},
@ -75,14 +77,13 @@ fn calculate_models(ciphertexts: &[[u8; 16]]) -> Vec<[Vec<u8>; 16]> {
}) })
.collect(); .collect();
r.par_iter_mut().enumerate().for_each(|(j, r)| {
for k in 0..16 { for k in 0..16 {
for j in 0..256 {
for (i, c) in ciphertexts.iter().enumerate() { for (i, c) in ciphertexts.iter().enumerate() {
r[j][k][i] = RSBOX[(c[k] ^ (j as u8)) as usize]; r[k][i] = RSBOX[(c[k] ^ (j as u8)) as usize];
} }
} }
} });
r r
} }