Move all crates to new crates dir

crates/udp_bench/src/announce.rs

@@ -0,0 +1,115 @@
+use std::net::{Ipv4Addr, SocketAddr, SocketAddrV4};
+use std::time::{Duration, Instant};
+
+use aquatic_common::CanonicalSocketAddr;
+use crossbeam_channel::{Receiver, Sender};
+use indicatif::ProgressIterator;
+use rand::Rng;
+use rand_distr::Gamma;
+
+use aquatic_udp::common::*;
+use aquatic_udp_protocol::*;
+
+use crate::common::*;
+use crate::config::BenchConfig;
+
+pub fn bench_announce_handler(
+    bench_config: &BenchConfig,
+    request_sender: &Sender<(SocketWorkerIndex, ConnectedRequest, CanonicalSocketAddr)>,
+    response_receiver: &Receiver<(ConnectedResponse, CanonicalSocketAddr)>,
+    rng: &mut impl Rng,
+    info_hashes: &[InfoHash],
+) -> (usize, Duration) {
+    let requests = create_requests(rng, info_hashes, bench_config.num_announce_requests);
+
+    let p = 10_000 * bench_config.num_threads; // FIXME: adjust to sharded workers
+    let mut num_responses = 0usize;
+
+    let mut dummy: u16 = rng.gen();
+
+    let pb = create_progress_bar("Announce", bench_config.num_rounds as u64);
+
+    // Start benchmark
+
+    let before = Instant::now();
+
+    for round in (0..bench_config.num_rounds).progress_with(pb) {
+        for request_chunk in requests.chunks(p) {
+            for (request, src) in request_chunk {
+                request_sender
+                    .send((
+                        SocketWorkerIndex(0),
+                        ConnectedRequest::Announce(request.clone()),
+                        *src,
+                    ))
+                    .unwrap();
+            }
+
+            while let Ok((ConnectedResponse::AnnounceIpv4(r), _)) = response_receiver.try_recv() {
+                num_responses += 1;
+
+                if let Some(last_peer) = r.peers.last() {
+                    dummy ^= last_peer.port.0;
+                }
+            }
+        }
+
+        let total = bench_config.num_announce_requests * (round + 1);
+
+        while num_responses < total {
+            if let Ok((ConnectedResponse::AnnounceIpv4(r), _)) = response_receiver.recv() {
+                num_responses += 1;
+
+                if let Some(last_peer) = r.peers.last() {
+                    dummy ^= last_peer.port.0;
+                }
+            }
+        }
+    }
+
+    let elapsed = before.elapsed();
+
+    if dummy == 0 {
+        println!("dummy dummy");
+    }
+
+    (num_responses, elapsed)
+}
+
+pub fn create_requests(
+    rng: &mut impl Rng,
+    info_hashes: &[InfoHash],
+    number: usize,
+) -> Vec<(AnnounceRequest, CanonicalSocketAddr)> {
+    let gamma = Gamma::new(GAMMA_SHAPE, GAMMA_SCALE).unwrap();
+
+    let max_index = info_hashes.len() - 1;
+
+    let mut requests = Vec::new();
+
+    for _ in 0..number {
+        let info_hash_index = gamma_usize(rng, gamma, max_index);
+
+        let request = AnnounceRequest {
+            connection_id: ConnectionId(0),
+            transaction_id: TransactionId(rng.gen()),
+            info_hash: info_hashes[info_hash_index],
+            peer_id: PeerId(rng.gen()),
+            bytes_downloaded: NumberOfBytes(rng.gen()),
+            bytes_uploaded: NumberOfBytes(rng.gen()),
+            bytes_left: NumberOfBytes(rng.gen()),
+            event: AnnounceEvent::Started,
+            ip_address: None,
+            key: PeerKey(rng.gen()),
+            peers_wanted: NumberOfPeers(rng.gen()),
+            port: Port(rng.gen()),
+        };
+
+        requests.push((
+            request,
+            CanonicalSocketAddr::new(SocketAddr::V4(SocketAddrV4::new(Ipv4Addr::LOCALHOST, 1))),
+        ));
+    }
+
+    requests
+}

crates/udp_bench/src/common.rs

@@ -0,0 +1,24 @@
+use indicatif::{ProgressBar, ProgressStyle};
+use rand::Rng;
+use rand_distr::Gamma;
+
+pub const GAMMA_SHAPE: f64 = 0.2;
+pub const GAMMA_SCALE: f64 = 100.0;
+
+pub const NUM_INFO_HASHES: usize = 10_000;
+
+pub fn create_progress_bar(name: &str, iterations: u64) -> ProgressBar {
+    let t = format!("{:<8} {}", name, "{wide_bar} {pos:>2}/{len:>2}");
+    let style = ProgressStyle::default_bar()
+        .template(&t)
+        .expect("setup indicatif template");
+
+    ProgressBar::new(iterations).with_style(style)
+}
+
+pub fn gamma_usize(rng: &mut impl Rng, gamma: Gamma<f64>, max: usize) -> usize {
+    let p: f64 = rng.sample(gamma);
+    let p = (p.min(101.0f64) - 1.0) / 100.0;
+
+    (p * max as f64) as usize
+}

crates/udp_bench/src/config.rs

@@ -0,0 +1,35 @@
+use aquatic_toml_config::TomlConfig;
+use serde::Deserialize;
+
+#[derive(Clone, Debug, PartialEq, TomlConfig, Deserialize)]
+#[serde(deny_unknown_fields)]
+pub struct BenchConfig {
+    pub num_rounds: usize,
+    pub num_threads: usize,
+    pub num_connect_requests: usize,
+    pub num_announce_requests: usize,
+    pub num_scrape_requests: usize,
+    pub num_hashes_per_scrape_request: usize,
+}
+
+impl Default for BenchConfig {
+    fn default() -> Self {
+        Self {
+            num_rounds: 10,
+            num_threads: 2,
+            num_connect_requests: 5_000_000,
+            num_announce_requests: 2_000_000,
+            num_scrape_requests: 2_000_000,
+            num_hashes_per_scrape_request: 20,
+        }
+    }
+}
+
+impl aquatic_common::cli::Config for BenchConfig {}
+
+#[cfg(test)]
+mod tests {
+    use super::BenchConfig;
+
+    ::aquatic_toml_config::gen_serialize_deserialize_test!(BenchConfig);
+}

crates/udp_bench/src/main.rs

@@ -0,0 +1,127 @@
+//! Benchmark announce and scrape handlers
+//!
+//! Example outputs:
+//! ```
+//! # Results over 10 rounds with 2 threads
+//! Announce:    429 540 requests/second,  2328.07 ns/request
+//! Scrape:    1 873 545 requests/second,   533.75 ns/request
+//! ```
+
+use aquatic_common::{PanicSentinelWatcher, ServerStartInstant};
+use aquatic_udp::workers::swarm::run_swarm_worker;
+use crossbeam_channel::unbounded;
+use num_format::{Locale, ToFormattedString};
+use rand::{rngs::SmallRng, thread_rng, Rng, SeedableRng};
+use std::time::Duration;
+
+use aquatic_common::cli::run_app_with_cli_and_config;
+use aquatic_udp::common::*;
+use aquatic_udp::config::Config;
+use aquatic_udp_protocol::*;
+
+use config::BenchConfig;
+
+mod announce;
+mod common;
+mod config;
+mod scrape;
+
+#[global_allocator]
+static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
+
+fn main() {
+    run_app_with_cli_and_config::<BenchConfig>(
+        "aquatic_udp_bench: Run aquatic_udp benchmarks",
+        env!("CARGO_PKG_VERSION"),
+        run,
+        None,
+    )
+}
+
+pub fn run(bench_config: BenchConfig) -> ::anyhow::Result<()> {
+    // Setup common state, spawn request handlers
+
+    let mut aquatic_config = Config::default();
+    let (_, sentinel) = PanicSentinelWatcher::create_with_sentinel();
+
+    aquatic_config.cleaning.torrent_cleaning_interval = 60 * 60 * 24;
+
+    let (request_sender, request_receiver) = unbounded();
+    let (response_sender, response_receiver) = unbounded();
+
+    let response_sender = ConnectedResponseSender::new(vec![response_sender]);
+    let (statistics_sender, _) = unbounded();
+
+    let server_start_instant = ServerStartInstant::new();
+
+    {
+        let config = aquatic_config.clone();
+        let state = State::new(config.swarm_workers);
+
+        ::std::thread::spawn(move || {
+            run_swarm_worker(
+                sentinel,
+                config,
+                state,
+                server_start_instant,
+                request_receiver,
+                response_sender,
+                statistics_sender,
+                SwarmWorkerIndex(0),
+            )
+        });
+    }
+
+    // Run benchmarks
+
+    let mut rng = SmallRng::from_rng(thread_rng()).unwrap();
+    let info_hashes = create_info_hashes(&mut rng);
+
+    let a = announce::bench_announce_handler(
+        &bench_config,
+        &request_sender,
+        &response_receiver,
+        &mut rng,
+        &info_hashes,
+    );
+
+    let s = scrape::bench_scrape_handler(
+        &bench_config,
+        &request_sender,
+        &response_receiver,
+        &mut rng,
+        &info_hashes,
+    );
+
+    println!(
+        "\n# Results over {} rounds with {} threads",
+        bench_config.num_rounds, bench_config.num_threads,
+    );
+
+    print_results("Announce:", a.0, a.1);
+    print_results("Scrape:  ", s.0, s.1);
+
+    Ok(())
+}
+
+pub fn print_results(request_type: &str, num_responses: usize, duration: Duration) {
+    let per_second = ((num_responses as f64 / (duration.as_micros() as f64 / 1000000.0)) as usize)
+        .to_formatted_string(&Locale::se);
+
+    let time_per_request = duration.as_nanos() as f64 / (num_responses as f64);
+
+    println!(
+        "{} {:>10} requests/second, {:>8.2} ns/request",
+        request_type, per_second, time_per_request,
+    );
+}
+
+fn create_info_hashes(rng: &mut impl Rng) -> Vec<InfoHash> {
+    let mut info_hashes = Vec::new();
+
+    for _ in 0..common::NUM_INFO_HASHES {
+        info_hashes.push(InfoHash(rng.gen()));
+    }
+
+    info_hashes
+}

crates/udp_bench/src/scrape.rs

@@ -0,0 +1,123 @@
+use std::net::{Ipv4Addr, SocketAddr, SocketAddrV4};
+use std::time::{Duration, Instant};
+
+use aquatic_common::CanonicalSocketAddr;
+use crossbeam_channel::{Receiver, Sender};
+use indicatif::ProgressIterator;
+use rand::Rng;
+use rand_distr::Gamma;
+
+use aquatic_udp::common::*;
+use aquatic_udp_protocol::*;
+
+use crate::common::*;
+use crate::config::BenchConfig;
+
+pub fn bench_scrape_handler(
+    bench_config: &BenchConfig,
+    request_sender: &Sender<(SocketWorkerIndex, ConnectedRequest, CanonicalSocketAddr)>,
+    response_receiver: &Receiver<(ConnectedResponse, CanonicalSocketAddr)>,
+    rng: &mut impl Rng,
+    info_hashes: &[InfoHash],
+) -> (usize, Duration) {
+    let requests = create_requests(
+        rng,
+        info_hashes,
+        bench_config.num_scrape_requests,
+        bench_config.num_hashes_per_scrape_request,
+    );
+
+    let p = 10_000 * bench_config.num_threads; // FIXME: adjust to sharded workers
+    let mut num_responses = 0usize;
+
+    let mut dummy: i32 = rng.gen();
+
+    let pb = create_progress_bar("Scrape", bench_config.num_rounds as u64);
+
+    // Start benchmark
+
+    let before = Instant::now();
+
+    for round in (0..bench_config.num_rounds).progress_with(pb) {
+        for request_chunk in requests.chunks(p) {
+            for (request, src) in request_chunk {
+                let request = ConnectedRequest::Scrape(PendingScrapeRequest {
+                    slab_key: 0,
+                    info_hashes: request
+                        .info_hashes
+                        .clone()
+                        .into_iter()
+                        .enumerate()
+                        .collect(),
+                });
+
+                request_sender
+                    .send((SocketWorkerIndex(0), request, *src))
+                    .unwrap();
+            }
+
+            while let Ok((ConnectedResponse::Scrape(response), _)) = response_receiver.try_recv() {
+                num_responses += 1;
+
+                if let Some(stat) = response.torrent_stats.values().last() {
+                    dummy ^= stat.leechers.0;
+                }
+            }
+        }
+
+        let total = bench_config.num_scrape_requests * (round + 1);
+
+        while num_responses < total {
+            if let Ok((ConnectedResponse::Scrape(response), _)) = response_receiver.recv() {
+                num_responses += 1;
+
+                if let Some(stat) = response.torrent_stats.values().last() {
+                    dummy ^= stat.leechers.0;
+                }
+            }
+        }
+    }
+
+    let elapsed = before.elapsed();
+
+    if dummy == 0 {
+        println!("dummy dummy");
+    }
+
+    (num_responses, elapsed)
+}
+
+pub fn create_requests(
+    rng: &mut impl Rng,
+    info_hashes: &[InfoHash],
+    number: usize,
+    hashes_per_request: usize,
+) -> Vec<(ScrapeRequest, CanonicalSocketAddr)> {
+    let gamma = Gamma::new(GAMMA_SHAPE, GAMMA_SCALE).unwrap();
+
+    let max_index = info_hashes.len() - 1;
+
+    let mut requests = Vec::new();
+
+    for _ in 0..number {
+        let mut request_info_hashes = Vec::new();
+
+        for _ in 0..hashes_per_request {
+            let info_hash_index = gamma_usize(rng, gamma, max_index);
+            request_info_hashes.push(info_hashes[info_hash_index])
+        }
+
+        let request = ScrapeRequest {
+            connection_id: ConnectionId(0),
+            transaction_id: TransactionId(rng.gen()),
+            info_hashes: request_info_hashes,
+        };
+
+        requests.push((
+            request,
+            CanonicalSocketAddr::new(SocketAddr::V4(SocketAddrV4::new(Ipv4Addr::LOCALHOST, 1))),
+        ));
+    }
+
+    requests
+}