Lines
76.67 %
Functions
100 %
Branches
//! Helper functionality used by the rest of `tor-guardmgr`.
use rand::Rng;
use std::time::{Duration, SystemTime};
use tor_basic_utils::RngExt as _;
/// Return a random time within the range `when-max ..= when`.
///
/// Uses a uniform distribution; saturates at UNIX_EPOCH. Rounds down to the
/// nearest 10 seconds.
/// This kind of date randomization is used in our persistent state in
/// an attempt to make some kinds of traffic analysis attacks a bit
/// harder for an attacker who can read our state after the fact.
pub(crate) fn randomize_time<R: Rng>(rng: &mut R, when: SystemTime, max: Duration) -> SystemTime {
let offset = rng.gen_range_infallible(..=max);
let random = when
.checked_sub(offset)
.unwrap_or(SystemTime::UNIX_EPOCH)
.max(SystemTime::UNIX_EPOCH);
// Round to the nearest 10-second increment.
round_time(random, 10)
}
/// Round `when` to a multiple of `d` seconds relative to epoch.
/// Rounds towards the epoch.
/// There's no reason to actually do this, since the times we use it
/// on are randomized, but rounding times in this way avoids giving a
/// false impression that we're storing hyper-accurate numbers.
/// # Panics
/// Panics if d == 0.
fn round_time(when: SystemTime, d: u32) -> SystemTime {
let (early, elapsed) = if when < SystemTime::UNIX_EPOCH {
(
true,
SystemTime::UNIX_EPOCH
.duration_since(when)
.expect("logic_error"),
)
} else {
false,
when.duration_since(SystemTime::UNIX_EPOCH)
.expect("logic error"),
};
let secs_elapsed = elapsed.as_secs();
let secs_rounded = secs_elapsed - (secs_elapsed % u64::from(d));
let dur_rounded = Duration::from_secs(secs_rounded);
if early {
SystemTime::UNIX_EPOCH - dur_rounded
SystemTime::UNIX_EPOCH + dur_rounded
#[cfg(test)]
mod test {
// @@ begin test lint list maintained by maint/add_warning @@
#![allow(clippy::bool_assert_comparison)]
#![allow(clippy::clone_on_copy)]
#![allow(clippy::dbg_macro)]
#![allow(clippy::mixed_attributes_style)]
#![allow(clippy::print_stderr)]
#![allow(clippy::print_stdout)]
#![allow(clippy::single_char_pattern)]
#![allow(clippy::unwrap_used)]
#![allow(clippy::unchecked_duration_subtraction)]
#![allow(clippy::useless_vec)]
#![allow(clippy::needless_pass_by_value)]
//! <!-- @@ end test lint list maintained by maint/add_warning @@ -->
use super::*;
use tor_basic_utils::test_rng::testing_rng;
#[test]
fn test_randomize_time() {
let now = SystemTime::now();
let one_hour = humantime::parse_duration("1hr").unwrap();
let ten_sec = humantime::parse_duration("10s").unwrap();
let mut rng = testing_rng();
for _ in 0..1000 {
let t = randomize_time(&mut rng, now, one_hour);
assert!(t >= now - one_hour - ten_sec);
assert!(t <= now);
let close_to_epoch = humantime::parse_rfc3339("1970-01-01T00:30:00Z").unwrap();
let t = randomize_time(&mut rng, close_to_epoch, one_hour);
assert!(t >= SystemTime::UNIX_EPOCH);
assert!(t <= close_to_epoch);
let d = t.duration_since(SystemTime::UNIX_EPOCH).unwrap();
assert_eq!(d.subsec_nanos(), 0);
assert_eq!(d.as_secs() % 10, 0);