summaryrefslogtreecommitdiffstats
path: root/third_party/rust/base64/benches
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-19 01:47:29 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-19 01:47:29 +0000
commit0ebf5bdf043a27fd3dfb7f92e0cb63d88954c44d (patch)
treea31f07c9bcca9d56ce61e9a1ffd30ef350d513aa /third_party/rust/base64/benches
parentInitial commit. (diff)
downloadfirefox-esr-0ebf5bdf043a27fd3dfb7f92e0cb63d88954c44d.tar.xz
firefox-esr-0ebf5bdf043a27fd3dfb7f92e0cb63d88954c44d.zip
Adding upstream version 115.8.0esr.upstream/115.8.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/rust/base64/benches')
-rw-r--r--third_party/rust/base64/benches/benchmarks.rs242
1 files changed, 242 insertions, 0 deletions
diff --git a/third_party/rust/base64/benches/benchmarks.rs b/third_party/rust/base64/benches/benchmarks.rs
new file mode 100644
index 0000000000..61d542f364
--- /dev/null
+++ b/third_party/rust/base64/benches/benchmarks.rs
@@ -0,0 +1,242 @@
+#[macro_use]
+extern crate criterion;
+
+use base64::{
+ display,
+ engine::{general_purpose::STANDARD, Engine},
+ write,
+};
+use criterion::{black_box, Bencher, BenchmarkId, Criterion, Throughput};
+use rand::{Rng, SeedableRng};
+use std::io::{self, Read, Write};
+
+fn do_decode_bench(b: &mut Bencher, &size: &usize) {
+ let mut v: Vec<u8> = Vec::with_capacity(size * 3 / 4);
+ fill(&mut v);
+ let encoded = STANDARD.encode(&v);
+
+ b.iter(|| {
+ let orig = STANDARD.decode(&encoded);
+ black_box(&orig);
+ });
+}
+
+fn do_decode_bench_reuse_buf(b: &mut Bencher, &size: &usize) {
+ let mut v: Vec<u8> = Vec::with_capacity(size * 3 / 4);
+ fill(&mut v);
+ let encoded = STANDARD.encode(&v);
+
+ let mut buf = Vec::new();
+ b.iter(|| {
+ STANDARD.decode_vec(&encoded, &mut buf).unwrap();
+ black_box(&buf);
+ buf.clear();
+ });
+}
+
+fn do_decode_bench_slice(b: &mut Bencher, &size: &usize) {
+ let mut v: Vec<u8> = Vec::with_capacity(size * 3 / 4);
+ fill(&mut v);
+ let encoded = STANDARD.encode(&v);
+
+ let mut buf = Vec::new();
+ buf.resize(size, 0);
+ b.iter(|| {
+ STANDARD.decode_slice(&encoded, &mut buf).unwrap();
+ black_box(&buf);
+ });
+}
+
+fn do_decode_bench_stream(b: &mut Bencher, &size: &usize) {
+ let mut v: Vec<u8> = Vec::with_capacity(size * 3 / 4);
+ fill(&mut v);
+ let encoded = STANDARD.encode(&v);
+
+ let mut buf = Vec::new();
+ buf.resize(size, 0);
+ buf.truncate(0);
+
+ b.iter(|| {
+ let mut cursor = io::Cursor::new(&encoded[..]);
+ let mut decoder = base64::read::DecoderReader::new(&mut cursor, &STANDARD);
+ decoder.read_to_end(&mut buf).unwrap();
+ buf.clear();
+ black_box(&buf);
+ });
+}
+
+fn do_encode_bench(b: &mut Bencher, &size: &usize) {
+ let mut v: Vec<u8> = Vec::with_capacity(size);
+ fill(&mut v);
+ b.iter(|| {
+ let e = STANDARD.encode(&v);
+ black_box(&e);
+ });
+}
+
+fn do_encode_bench_display(b: &mut Bencher, &size: &usize) {
+ let mut v: Vec<u8> = Vec::with_capacity(size);
+ fill(&mut v);
+ b.iter(|| {
+ let e = format!("{}", display::Base64Display::new(&v, &STANDARD));
+ black_box(&e);
+ });
+}
+
+fn do_encode_bench_reuse_buf(b: &mut Bencher, &size: &usize) {
+ let mut v: Vec<u8> = Vec::with_capacity(size);
+ fill(&mut v);
+ let mut buf = String::new();
+ b.iter(|| {
+ STANDARD.encode_string(&v, &mut buf);
+ buf.clear();
+ });
+}
+
+fn do_encode_bench_slice(b: &mut Bencher, &size: &usize) {
+ let mut v: Vec<u8> = Vec::with_capacity(size);
+ fill(&mut v);
+ let mut buf = Vec::new();
+ // conservative estimate of encoded size
+ buf.resize(v.len() * 2, 0);
+ b.iter(|| STANDARD.encode_slice(&v, &mut buf).unwrap());
+}
+
+fn do_encode_bench_stream(b: &mut Bencher, &size: &usize) {
+ let mut v: Vec<u8> = Vec::with_capacity(size);
+ fill(&mut v);
+ let mut buf = Vec::new();
+
+ buf.reserve(size * 2);
+ b.iter(|| {
+ buf.clear();
+ let mut stream_enc = write::EncoderWriter::new(&mut buf, &STANDARD);
+ stream_enc.write_all(&v).unwrap();
+ stream_enc.flush().unwrap();
+ });
+}
+
+fn do_encode_bench_string_stream(b: &mut Bencher, &size: &usize) {
+ let mut v: Vec<u8> = Vec::with_capacity(size);
+ fill(&mut v);
+
+ b.iter(|| {
+ let mut stream_enc = write::EncoderStringWriter::new(&STANDARD);
+ stream_enc.write_all(&v).unwrap();
+ stream_enc.flush().unwrap();
+ let _ = stream_enc.into_inner();
+ });
+}
+
+fn do_encode_bench_string_reuse_buf_stream(b: &mut Bencher, &size: &usize) {
+ let mut v: Vec<u8> = Vec::with_capacity(size);
+ fill(&mut v);
+
+ let mut buf = String::new();
+ b.iter(|| {
+ buf.clear();
+ let mut stream_enc = write::EncoderStringWriter::from_consumer(&mut buf, &STANDARD);
+ stream_enc.write_all(&v).unwrap();
+ stream_enc.flush().unwrap();
+ let _ = stream_enc.into_inner();
+ });
+}
+
+fn fill(v: &mut Vec<u8>) {
+ let cap = v.capacity();
+ // weak randomness is plenty; we just want to not be completely friendly to the branch predictor
+ let mut r = rand::rngs::SmallRng::from_entropy();
+ while v.len() < cap {
+ v.push(r.gen::<u8>());
+ }
+}
+
+const BYTE_SIZES: [usize; 5] = [3, 50, 100, 500, 3 * 1024];
+
+// Benchmarks over these byte sizes take longer so we will run fewer samples to
+// keep the benchmark runtime reasonable.
+const LARGE_BYTE_SIZES: [usize; 3] = [3 * 1024 * 1024, 10 * 1024 * 1024, 30 * 1024 * 1024];
+
+fn encode_benchmarks(c: &mut Criterion, label: &str, byte_sizes: &[usize]) {
+ let mut group = c.benchmark_group(label);
+ group
+ .warm_up_time(std::time::Duration::from_millis(500))
+ .measurement_time(std::time::Duration::from_secs(3));
+
+ for size in byte_sizes {
+ group
+ .throughput(Throughput::Bytes(*size as u64))
+ .bench_with_input(BenchmarkId::new("encode", size), size, do_encode_bench)
+ .bench_with_input(
+ BenchmarkId::new("encode_display", size),
+ size,
+ do_encode_bench_display,
+ )
+ .bench_with_input(
+ BenchmarkId::new("encode_reuse_buf", size),
+ size,
+ do_encode_bench_reuse_buf,
+ )
+ .bench_with_input(
+ BenchmarkId::new("encode_slice", size),
+ size,
+ do_encode_bench_slice,
+ )
+ .bench_with_input(
+ BenchmarkId::new("encode_reuse_buf_stream", size),
+ size,
+ do_encode_bench_stream,
+ )
+ .bench_with_input(
+ BenchmarkId::new("encode_string_stream", size),
+ size,
+ do_encode_bench_string_stream,
+ )
+ .bench_with_input(
+ BenchmarkId::new("encode_string_reuse_buf_stream", size),
+ size,
+ do_encode_bench_string_reuse_buf_stream,
+ );
+ }
+
+ group.finish();
+}
+
+fn decode_benchmarks(c: &mut Criterion, label: &str, byte_sizes: &[usize]) {
+ let mut group = c.benchmark_group(label);
+
+ for size in byte_sizes {
+ group
+ .warm_up_time(std::time::Duration::from_millis(500))
+ .measurement_time(std::time::Duration::from_secs(3))
+ .throughput(Throughput::Bytes(*size as u64))
+ .bench_with_input(BenchmarkId::new("decode", size), size, do_decode_bench)
+ .bench_with_input(
+ BenchmarkId::new("decode_reuse_buf", size),
+ size,
+ do_decode_bench_reuse_buf,
+ )
+ .bench_with_input(
+ BenchmarkId::new("decode_slice", size),
+ size,
+ do_decode_bench_slice,
+ )
+ .bench_with_input(
+ BenchmarkId::new("decode_stream", size),
+ size,
+ do_decode_bench_stream,
+ );
+ }
+
+ group.finish();
+}
+
+fn bench(c: &mut Criterion) {
+ encode_benchmarks(c, "encode_small_input", &BYTE_SIZES[..]);
+ encode_benchmarks(c, "encode_large_input", &LARGE_BYTE_SIZES[..]);
+ decode_benchmarks(c, "decode_small_input", &BYTE_SIZES[..]);
+ decode_benchmarks(c, "decode_large_input", &LARGE_BYTE_SIZES[..]);
+}
+
+criterion_group!(benches, bench);
+criterion_main!(benches);