1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
use criterion::{criterion_group, criterion_main, Criterion};
use sequoia_openpgp::cert::Cert;
use sequoia_openpgp::parse::Parse;
use sequoia_openpgp::serialize::stream::{Encryptor, LiteralWriter, Message};
use sequoia_openpgp::policy::StandardPolicy;
use std::io::Write;
lazy_static::lazy_static! {
static ref ZEROS_1_MB: Vec<u8> = vec![0; 1 * 1024 * 1024];
static ref ZEROS_10_MB: Vec<u8> = vec![0; 10 * 1024 * 1024];
}
fn encrypt_to_testy(bytes: &[u8]) -> sequoia_openpgp::Result<()> {
let mut sink = vec![];
let testy =
Cert::from_bytes(&include_bytes!("../tests/data/keys/testy.pgp")[..])?;
let p = &StandardPolicy::new();
let recipients = testy
.keys()
.with_policy(p, None)
.supported()
.alive()
.revoked(false)
.for_transport_encryption();
let message =
Encryptor::for_recipients(Message::new(&mut sink), recipients).build()?;
let mut w = LiteralWriter::new(message).build()?;
w.write_all(bytes)?;
w.finalize()?;
Ok(())
}
fn encrypt_with_password(bytes: &[u8]) -> sequoia_openpgp::Result<()> {
let mut sink = vec![];
let message = Encryptor::with_passwords(
Message::new(&mut sink),
Some("ściśle tajne"),
)
.build()?;
let mut w = LiteralWriter::new(message).build()?;
w.write_all(bytes)?;
w.finalize()?;
Ok(())
}
fn bench_encrypt(c: &mut Criterion) {
let mut group = c.benchmark_group("encrypt message");
// Encrypt a very short message.
let bytes = b"Hello world.";
group.bench_function(format!("password {:?}", bytes.len()), |b| {
b.iter(|| encrypt_with_password(bytes))
});
group.bench_function(format!("recipient {:?}", bytes.len()), |b| {
b.iter(|| encrypt_to_testy(bytes))
});
// Encrypt a medium length message.
let bytes = &ZEROS_1_MB[..];
group.bench_function(format!("password {:?}", bytes.len()), |b| {
b.iter(|| encrypt_with_password(bytes))
});
group.bench_function(format!("recipient {:?}", bytes.len()), |b| {
b.iter(|| encrypt_to_testy(bytes))
});
// Encrypt a very long message.
let bytes = &ZEROS_10_MB[..];
group.bench_function(format!("password {:?}", bytes.len()), |b| {
b.iter(|| encrypt_with_password(bytes))
});
group.bench_function(format!("recipient {:?}", bytes.len()), |b| {
b.iter(|| encrypt_to_testy(bytes))
});
group.finish();
}
criterion_group!(benches, bench_encrypt);
criterion_main!(benches);
|