Jon Häggblad
4af70ef255
* Lock files * Add flag to disable cover traffic * Add flag to disable per hop delays * Add flag to enable mixed size packets * Add meta flag to set medium speed * Special case zero averge hop delay to be exactly zero * Extract out generate_hop_delays function
54 lines
1.5 KiB
Rust
54 lines
1.5 KiB
Rust
// Copyright 2023 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use std::time::Duration;
|
|
|
|
use nym_sphinx_addressing::clients::Recipient;
|
|
use nym_sphinx_types::{delays, Delay, Node};
|
|
use thiserror::Error;
|
|
|
|
pub trait SphinxRouteMaker {
|
|
type Error;
|
|
|
|
fn sphinx_route(&mut self, hops: u8, destination: &Recipient)
|
|
-> Result<Vec<Node>, Self::Error>;
|
|
}
|
|
|
|
#[derive(Debug, Error, Clone, Copy)]
|
|
#[error("the route vector contains {available} nodes while {requested} hops are required")]
|
|
pub struct InvalidNumberOfHops {
|
|
available: usize,
|
|
requested: u8,
|
|
}
|
|
|
|
// if one wants to provide a hardcoded route, they can
|
|
impl SphinxRouteMaker for Vec<Node> {
|
|
type Error = InvalidNumberOfHops;
|
|
|
|
fn sphinx_route(
|
|
&mut self,
|
|
hops: u8,
|
|
_destination: &Recipient,
|
|
) -> Result<Vec<Node>, InvalidNumberOfHops> {
|
|
// it's the responsibility of the caller to ensure the hardcoded route has correct number of hops
|
|
// and that it's final hop include the recipient's gateway.
|
|
|
|
if self.len() != hops as usize {
|
|
Err(InvalidNumberOfHops {
|
|
available: self.len(),
|
|
requested: hops,
|
|
})
|
|
} else {
|
|
Ok(self.clone())
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn generate_hop_delays(average_packet_delay: Duration, num_hops: usize) -> Vec<Delay> {
|
|
if average_packet_delay.is_zero() {
|
|
vec![nym_sphinx_types::Delay::new_from_millis(0); num_hops]
|
|
} else {
|
|
delays::generate_from_average_duration(num_hops, average_packet_delay)
|
|
}
|
|
}
|