Add benchmarks

benches/bench_babyjubjub.rs

@@ -0,0 +1,59 @@
+use criterion::{criterion_group, criterion_main, Criterion};
+
+extern crate num;
+extern crate num_bigint;
+extern crate num_traits;
+use num_bigint::{BigInt, ToBigInt};
+
+use babyjubjub_rs::{utils, Point};
+
+fn criterion_benchmark(c: &mut Criterion) {
+    let x: BigInt = BigInt::parse_bytes(
+        b"17777552123799933955779906779655732241715742912184938656739573121738514868268",
+        10,
+    )
+    .unwrap();
+    c.bench_function("modulus", |b| {
+        b.iter(|| utils::modulus(&x, &babyjubjub_rs::Q))
+    });
+
+    let p: Point = Point {
+        x: BigInt::parse_bytes(
+            b"17777552123799933955779906779655732241715742912184938656739573121738514868268",
+            10,
+        )
+        .unwrap(),
+        y: BigInt::parse_bytes(
+            b"2626589144620713026669568689430873010625803728049924121243784502389097019475",
+            10,
+        )
+        .unwrap(),
+    };
+    let q = p.clone();
+
+    c.bench_function("add", |b| b.iter(|| p.add(&q)));
+
+    c.bench_function("mul_scalar_small", |b| {
+        b.iter(|| p.mul_scalar(&3.to_bigint().unwrap()))
+    });
+    let r: BigInt = BigInt::parse_bytes(
+        b"2626589144620713026669568689430873010625803728049924121243784502389097019475",
+        10,
+    )
+    .unwrap();
+    c.bench_function("mul_scalar", |b| b.iter(|| p.mul_scalar(&r)));
+
+    let sk = babyjubjub_rs::new_key();
+    let pk = sk.public().unwrap();
+    let msg = 5.to_bigint().unwrap();
+    c.bench_function("sign_poseidon", |b| {
+        b.iter(|| sk.sign_poseidon(msg.clone()))
+    });
+    let sig = sk.sign_poseidon(msg.clone()).unwrap();
+    c.bench_function("verify_poseidon", |b| {
+        b.iter(|| babyjubjub_rs::verify_poseidon(pk.clone(), sig.clone(), msg.clone()))
+    });
+}
+
+criterion_group!(benches, criterion_benchmark);
+criterion_main!(benches);