ot bench

pop-mpc/benches/ot.rs

@@ -1,16 +1,21 @@
+use aes::cipher::{generic_array::GenericArray, NewBlockCipher};
+use aes::Aes128;
 use criterion::{black_box, criterion_group, criterion_main, Criterion};
-use pop_mpc::block::BLOCK_ONES;
+use pop_mpc::block::{Block, BLOCK_ONES};
 use pop_mpc::ot::base::*;
+use pop_mpc::ot::extension::*;
+use pop_mpc::utils::u8vec_to_boolvec;
+use rand::RngCore;
 use rand::SeedableRng;
 use rand_chacha::ChaCha12Rng;
 
 fn criterion_benchmark(c: &mut Criterion) {
-    c.bench_function("base_ot_1000", move |bench| {
+    c.bench_function("base_ot_1024", move |bench| {
         let mut s_rng = ChaCha12Rng::from_entropy();
         let mut r_rng = ChaCha12Rng::from_entropy();
 
-        let s_inputs = [[BLOCK_ONES; 2]; 1000];
-        let choice = [false; 1000];
+        let s_inputs = [[BLOCK_ONES; 2]; 1024];
+        let choice = [false; 1024];
 
         bench.iter(|| {
             let mut sender = BaseOTSender::new(&mut s_rng);
@@ -24,6 +29,39 @@ fn criterion_benchmark(c: &mut Criterion) {
             black_box(receive);
         });
     });
+
+    c.bench_function("ot_1024", move |bench| {
+        let mut choice = vec![0u8; 128];
+        let mut rng = ChaCha12Rng::from_entropy();
+        rng.fill_bytes(&mut choice);
+        let choice = u8vec_to_boolvec(&choice);
+        let inputs: Vec<[Block; 2]> = (0..1024)
+            .map(|_| [Block::random(&mut rng), Block::random(&mut rng)])
+            .collect();
+
+        bench.iter(|| {
+            let s_rng = ChaCha12Rng::from_entropy();
+            let s_cipher = Aes128::new(GenericArray::from_slice(&[0u8; 16]));
+            let r_rng = ChaCha12Rng::from_entropy();
+            let r_cipher = Aes128::new(GenericArray::from_slice(&[0u8; 16]));
+
+            let mut receiver = OTReceiver::new(r_rng, r_cipher);
+            let base_sender_setup = receiver.base_setup();
+
+            let mut sender = OTSender::new(s_rng, s_cipher, base_sender_setup);
+            let base_receiver_setup = sender.base_setup();
+
+            let send_seeds = receiver.base_send_seeds(base_receiver_setup);
+            sender.base_receive_seeds(send_seeds);
+            let receiver_setup = receiver.extension_setup(&choice);
+            sender.extension_setup(receiver_setup);
+
+            let send = sender.send(&inputs);
+            let receive = receiver.receive(&choice, send);
+
+            black_box(receive);
+        });
+    });
 }
 
 criterion_group!(benches, criterion_benchmark);