rpki/tests/test_cache.rs
2026-03-25 10:08:40 +08:00

1359 lines
40 KiB
Rust

mod common;
use std::collections::VecDeque;
use std::net::{Ipv4Addr, Ipv6Addr};
use std::sync::Arc;
use common::test_helper::{
as_route_origin, as_v4_route_origin, indent_block, payloads_to_string,
serial_result_detail_to_string, snapshot_hashes_to_string, test_report, v4_origin, v6_origin,
};
use rpki::data_model::resources::as_resources::Asn;
use rpki::rtr::cache::{
CacheAvailability, Delta, RtrCacheBuilder, SerialResult, SessionIds, Snapshot,
validate_payload_updates_for_rtr,
validate_payloads_for_rtr,
};
use rpki::rtr::payload::{Aspa, Payload, Timing};
use rpki::rtr::store::RtrStore;
fn delta_to_string(delta: &Delta) -> String {
format!(
"serial: {}\nannounced:\n{}withdrawn:\n{}",
delta.serial(),
indent_block(&payloads_to_string(delta.announced()), 2),
indent_block(&payloads_to_string(delta.withdrawn()), 2),
)
}
fn deltas_window_to_string(deltas: &VecDeque<Arc<Delta>>) -> String {
if deltas.is_empty() {
return " <empty>\n".to_string();
}
let mut out = String::new();
for (idx, delta) in deltas.iter().enumerate() {
out.push_str(&format!("delta[{}]:\n", idx));
out.push_str(&indent_block(&delta_to_string(delta), 2));
}
out
}
fn get_deltas_since_input_to_string(cache_session_id: u16, cache_serial: u32, client_serial: u32) -> String {
format!(
"cache.session_id: {}\ncache.serial: {}\nclient_serial: {}\n",
cache_session_id, cache_serial, client_serial
)
}
fn snapshot_hashes_and_sorted_view_to_string(snapshot: &Snapshot) -> String {
let payloads = snapshot.payloads_for_rtr();
format!(
"hashes:\n{}sorted payloads_for_rtr:\n{}",
indent_block(&snapshot_hashes_to_string(snapshot), 2),
indent_block(&payloads_to_string(&payloads), 2),
)
}
/// Snapshot ?hash ?
/// payload snapshot_hash / origins_hash ?
#[test]
fn snapshot_hash_is_stable_for_same_content_with_different_input_order() {
let a = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let b = v4_origin(198, 51, 100, 0, 24, 24, 64497);
let c = v6_origin(
Ipv6Addr::new(0x2001, 0xdb8, 0, 0, 0, 0, 0, 0),
32,
48,
64498,
);
let s1_input = vec![
Payload::RouteOrigin(a.clone()),
Payload::RouteOrigin(b.clone()),
Payload::RouteOrigin(c.clone()),
];
let s2_input = vec![
Payload::RouteOrigin(c),
Payload::RouteOrigin(a),
Payload::RouteOrigin(b),
];
let s1 = Snapshot::from_payloads(s1_input.clone());
let s2 = Snapshot::from_payloads(s2_input.clone());
let input = format!(
"s1 payloads:\n{}\ns2 payloads:\n{}",
indent_block(&payloads_to_string(&s1_input), 2),
indent_block(&payloads_to_string(&s2_input), 2),
);
let output = format!(
"s1:\n{}\ns2:\n{}\n:\n same_content: {}\n same_origins: {}\n snapshot_hash : {}\n origins_hash : {}\n",
indent_block(&snapshot_hashes_and_sorted_view_to_string(&s1), 2),
indent_block(&snapshot_hashes_and_sorted_view_to_string(&s2), 2),
s1.same_content(&s2),
s1.same_origins(&s2),
s1.snapshot_hash() == s2.snapshot_hash(),
s1.origins_hash() == s2.origins_hash(),
);
test_report(
"snapshot_hash_is_stable_for_same_content_with_different_input_order",
"test purpose",
&input,
&output,
);
assert!(s1.same_content(&s2));
assert!(s1.same_origins(&s2));
assert_eq!(s1.snapshot_hash(), s2.snapshot_hash());
assert_eq!(s1.origins_hash(), s2.origins_hash());
}
#[tokio::test]
async fn init_keeps_cache_running_when_file_loader_returns_no_data() {
let dir = tempfile::tempdir().unwrap();
let store = RtrStore::open(dir.path()).unwrap();
let cache = rpki::rtr::cache::RtrCache::default()
.init(&store, 16, Timing::new(600, 600, 7200), || Ok(vec![]))
.unwrap();
assert!(!cache.is_data_available());
assert_eq!(cache.serial(), 0);
assert!(cache.snapshot().payloads_for_rtr().is_empty());
}
#[tokio::test]
async fn init_restores_wraparound_delta_window_from_store() {
let dir = tempfile::tempdir().unwrap();
let store = RtrStore::open(dir.path()).unwrap();
let session_ids = SessionIds::from_array([42, 43, 44]);
let snapshot = Snapshot::from_payloads(vec![
Payload::RouteOrigin(v4_origin(192, 0, 2, 0, 24, 24, 64496)),
Payload::RouteOrigin(v4_origin(198, 51, 100, 0, 24, 24, 64497)),
Payload::RouteOrigin(v4_origin(203, 0, 113, 0, 24, 24, 64498)),
]);
let d_max = Delta::new(
u32::MAX,
vec![Payload::RouteOrigin(v4_origin(192, 0, 2, 0, 24, 24, 64496))],
vec![],
);
let d_zero = Delta::new(
0,
vec![Payload::RouteOrigin(v4_origin(198, 51, 100, 0, 24, 24, 64497))],
vec![],
);
let d_one = Delta::new(
1,
vec![Payload::RouteOrigin(v4_origin(203, 0, 113, 0, 24, 24, 64498))],
vec![],
);
store
.save_cache_state(
CacheAvailability::Ready,
&snapshot,
&session_ids,
u32::MAX,
Some(&d_max),
Some((u32::MAX, u32::MAX)),
false,
)
.unwrap();
store
.save_cache_state(
CacheAvailability::Ready,
&snapshot,
&session_ids,
0,
Some(&d_zero),
Some((u32::MAX, 0)),
false,
)
.unwrap();
store
.save_cache_state(
CacheAvailability::Ready,
&snapshot,
&session_ids,
1,
Some(&d_one),
Some((u32::MAX, 1)),
false,
)
.unwrap();
let cache = rpki::rtr::cache::RtrCache::default()
.init(&store, 16, Timing::new(600, 600, 7200), || Ok(Vec::new()))
.unwrap();
match cache.get_deltas_since(u32::MAX.wrapping_sub(1)) {
SerialResult::Delta(delta) => {
assert_eq!(delta.serial(), 1);
assert_eq!(delta.announced().len(), 3);
}
_ => panic!("expected wraparound delta to be restored from store"),
}
}
/// Snapshot::diff() ?
/// old_snapshot ?new_snapshot announced?withdrawn?
#[test]
fn snapshot_diff_reports_announced_and_withdrawn_correctly() {
let old_a = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let old_b = v4_origin(198, 51, 100, 0, 24, 24, 64497);
let new_c = v6_origin(
Ipv6Addr::new(0x2001, 0xdb8, 0, 1, 0, 0, 0, 0),
48,
48,
64499,
);
let old_input = vec![
Payload::RouteOrigin(old_a.clone()),
Payload::RouteOrigin(old_b.clone()),
];
let new_input = vec![
Payload::RouteOrigin(old_b),
Payload::RouteOrigin(new_c.clone()),
];
let old_snapshot = Snapshot::from_payloads(old_input.clone());
let new_snapshot = Snapshot::from_payloads(new_input.clone());
let (announced, withdrawn) = old_snapshot.diff(&new_snapshot);
let input = format!(
"old_snapshot :\n{}\nnew_snapshot :\n{}",
indent_block(&payloads_to_string(&old_input), 2),
indent_block(&payloads_to_string(&new_input), 2),
);
let output = format!(
"announced:\n{}withdrawn:\n{}",
indent_block(&payloads_to_string(&announced), 2),
indent_block(&payloads_to_string(&withdrawn), 2),
);
test_report(
"snapshot_diff_reports_announced_and_withdrawn_correctly",
"test purpose",
&input,
&output,
);
assert_eq!(announced.len(), 1);
assert_eq!(withdrawn.len(), 1);
match &announced[0] {
Payload::RouteOrigin(ro) => assert_eq!(ro, &new_c),
_ => panic!("expected announced RouteOrigin"),
}
match &withdrawn[0] {
Payload::RouteOrigin(ro) => assert_eq!(ro, &old_a),
_ => panic!("expected withdrawn RouteOrigin"),
}
}
/// Snapshot::payloads_for_rtr() ?
/// IPv4 Prefix IPv6 Prefix ?IPv4 announcement ?RTR ?
#[test]
fn snapshot_payloads_for_rtr_sorts_ipv4_before_ipv6_and_ipv4_announcements_descending() {
let v4_low = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let v4_high = v4_origin(198, 51, 100, 0, 24, 24, 64497);
let v6 = v6_origin(
Ipv6Addr::new(0x2001, 0xdb8, 0, 0, 0, 0, 0, 0),
32,
48,
64498,
);
let input_payloads = vec![
Payload::RouteOrigin(v6.clone()),
Payload::RouteOrigin(v4_low.clone()),
Payload::RouteOrigin(v4_high.clone()),
];
let snapshot = Snapshot::from_payloads(input_payloads.clone());
let output_payloads = snapshot.payloads_for_rtr();
let input = format!(
" payloads?Snapshot :\n{}",
indent_block(&payloads_to_string(&input_payloads), 2),
);
let output = format!(
"?payloads_for_rtr:\n{}",
indent_block(&payloads_to_string(&output_payloads), 2),
);
test_report(
"snapshot_payloads_for_rtr_sorts_ipv4_before_ipv6_and_ipv4_announcements_descending",
"test purpose",
&input,
&output,
);
assert_eq!(output_payloads.len(), 3);
let first = as_v4_route_origin(&output_payloads[0]);
let second = as_v4_route_origin(&output_payloads[1]);
assert_eq!(
first.prefix().address.to_ipv4(),
Some(Ipv4Addr::new(198, 51, 100, 0))
);
assert_eq!(
second.prefix().address.to_ipv4(),
Some(Ipv4Addr::new(192, 0, 2, 0))
);
let third = as_route_origin(&output_payloads[2]);
assert!(third.prefix().address.is_ipv6());
assert_eq!(
third.prefix().address.to_ipv6(),
Some(Ipv6Addr::new(0x2001, 0xdb8, 0, 0, 0, 0, 0, 0))
);
}
/// Delta::new() ?announced ?withdrawn?
/// announced announcement ithdrawn withdrawal ?
#[test]
fn delta_new_sorts_announced_descending_and_withdrawn_ascending() {
let announced_low = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let announced_high = v4_origin(198, 51, 100, 0, 24, 24, 64497);
let withdrawn_high = v4_origin(203, 0, 113, 0, 24, 24, 64501);
let withdrawn_low = v4_origin(10, 0, 0, 0, 24, 24, 64500);
let input_announced = vec![
Payload::RouteOrigin(announced_low),
Payload::RouteOrigin(announced_high),
];
let input_withdrawn = vec![
Payload::RouteOrigin(withdrawn_high),
Payload::RouteOrigin(withdrawn_low),
];
let delta = Delta::new(101, input_announced.clone(), input_withdrawn.clone());
let input = format!(
"announced?\n{}withdrawn?\n{}",
indent_block(&payloads_to_string(&input_announced), 2),
indent_block(&payloads_to_string(&input_withdrawn), 2),
);
let output = indent_block(&delta_to_string(&delta), 2);
test_report(
"delta_new_sorts_announced_descending_and_withdrawn_ascending",
"test purpose",
&input,
&output,
);
assert_eq!(delta.serial(), 101);
assert_eq!(delta.announced().len(), 2);
assert_eq!(delta.withdrawn().len(), 2);
let a0 = as_v4_route_origin(&delta.announced()[0]);
let a1 = as_v4_route_origin(&delta.announced()[1]);
assert_eq!(
a0.prefix().address.to_ipv4(),
Some(Ipv4Addr::new(198, 51, 100, 0))
);
assert_eq!(
a1.prefix().address.to_ipv4(),
Some(Ipv4Addr::new(192, 0, 2, 0))
);
let w0 = as_v4_route_origin(&delta.withdrawn()[0]);
let w1 = as_v4_route_origin(&delta.withdrawn()[1]);
assert_eq!(w0.prefix().address.to_ipv4(), Some(Ipv4Addr::new(10, 0, 0, 0)));
assert_eq!(
w1.prefix().address.to_ipv4(),
Some(Ipv4Addr::new(203, 0, 113, 0))
);
}
/// serial ?serial
/// get_deltas_since() ?UpToDate?Delta ?ResetRequired?
#[test]
fn get_deltas_since_returns_up_to_date_when_client_serial_matches_current() {
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(100)
.timing(Timing::default())
.build();
let result = cache.get_deltas_since(100);
let input = get_deltas_since_input_to_string(cache.session_id_for_version(1), cache.serial(), 100);
let output = serial_result_detail_to_string(&result);
test_report(
"get_deltas_since_returns_up_to_date_when_client_serial_matches_current",
"test purpose",
&input,
&output,
);
match result {
SerialResult::UpToDate => {}
_ => panic!("expected UpToDate"),
}
}
/// serial delta window ?/// get_deltas_since() ?ResetRequired?#[test]
fn get_deltas_since_returns_reset_required_when_client_serial_is_too_old() {
let d1 = Arc::new(Delta::new(
101,
vec![Payload::RouteOrigin(v4_origin(192, 0, 2, 0, 24, 24, 64496))],
vec![],
));
let d2 = Arc::new(Delta::new(
102,
vec![Payload::RouteOrigin(v4_origin(198, 51, 100, 0, 24, 24, 64497))],
vec![],
));
let mut deltas = VecDeque::new();
deltas.push_back(d1);
deltas.push_back(d2);
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(102)
.timing(Timing::default())
.deltas(deltas.clone())
.build();
let result = cache.get_deltas_since(99);
let input = format!(
"{}delta_window:\n{}",
get_deltas_since_input_to_string(cache.session_id_for_version(1), cache.serial(), 99),
indent_block(&deltas_window_to_string(&deltas), 2),
);
let output = serial_result_detail_to_string(&result);
test_report(
"get_deltas_since_returns_reset_required_when_client_serial_is_too_old",
"test purpose",
&input,
&output,
);
match result {
SerialResult::ResetRequired => {}
_ => panic!("expected ResetRequired"),
}
}
/// serial delta window ?
/// get_deltas_since() ?delta ?
#[test]
fn get_deltas_since_returns_minimal_merged_delta() {
let d1 = Arc::new(Delta::new(
101,
vec![Payload::RouteOrigin(v4_origin(192, 0, 2, 0, 24, 24, 64496))],
vec![],
));
let d2 = Arc::new(Delta::new(
102,
vec![Payload::RouteOrigin(v4_origin(198, 51, 100, 0, 24, 24, 64497))],
vec![],
));
let d3 = Arc::new(Delta::new(
103,
vec![Payload::RouteOrigin(v4_origin(203, 0, 113, 0, 24, 24, 64498))],
vec![],
));
let final_snapshot = Snapshot::from_payloads(vec![
Payload::RouteOrigin(v4_origin(192, 0, 2, 0, 24, 24, 64496)),
Payload::RouteOrigin(v4_origin(198, 51, 100, 0, 24, 24, 64497)),
Payload::RouteOrigin(v4_origin(203, 0, 113, 0, 24, 24, 64498)),
]);
let mut deltas = VecDeque::new();
deltas.push_back(d1);
deltas.push_back(d2);
deltas.push_back(d3);
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(103)
.timing(Timing::default())
.snapshot(final_snapshot)
.deltas(deltas.clone())
.build();
let result = cache.get_deltas_since(101);
let input = format!(
"{}delta_window:\n{}",
get_deltas_since_input_to_string(cache.session_id_for_version(1), cache.serial(), 101),
indent_block(&deltas_window_to_string(&deltas), 2),
);
let output = serial_result_detail_to_string(&result);
test_report(
"get_deltas_since_returns_minimal_merged_delta",
"test purpose",
&input,
&output,
);
match result {
SerialResult::Delta(delta) => {
assert_eq!(delta.serial(), 103);
assert_eq!(delta.announced().len(), 2);
assert_eq!(delta.withdrawn().len(), 0);
let a0 = as_v4_route_origin(&delta.announced()[0]);
let a1 = as_v4_route_origin(&delta.announced()[1]);
assert_eq!(
a0.prefix().address.to_ipv4(),
Some(Ipv4Addr::new(203, 0, 113, 0))
);
assert_eq!(
a1.prefix().address.to_ipv4(),
Some(Ipv4Addr::new(198, 51, 100, 0))
);
}
_ => panic!("expected Delta"),
}
}
/// serial serial
/// ResetRequired?
#[test]
fn get_deltas_since_returns_reset_required_when_client_serial_is_in_future() {
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(100)
.timing(Timing::default())
.build();
let result = cache.get_deltas_since(101);
let input = get_deltas_since_input_to_string(cache.session_id_for_version(1), cache.serial(), 101);
let output = serial_result_detail_to_string(&result);
test_report(
"get_deltas_since_returns_reset_required_when_client_serial_is_in_future",
"test purpose",
&input,
&output,
);
match result {
SerialResult::ResetRequired => {}
_ => panic!("expected ResetRequired"),
}
}
#[test]
fn get_deltas_since_supports_incremental_updates_across_serial_wraparound() {
let a = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let b = v4_origin(198, 51, 100, 0, 24, 24, 64497);
let d_max = Arc::new(Delta::new(
u32::MAX,
vec![Payload::RouteOrigin(a.clone())],
vec![],
));
let d_zero = Arc::new(Delta::new(
0,
vec![Payload::RouteOrigin(b.clone())],
vec![],
));
let mut deltas = VecDeque::new();
deltas.push_back(d_max);
deltas.push_back(d_zero);
let final_snapshot = Snapshot::from_payloads(vec![
Payload::RouteOrigin(a.clone()),
Payload::RouteOrigin(b.clone()),
]);
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(0)
.timing(Timing::default())
.snapshot(final_snapshot)
.deltas(deltas.clone())
.build();
let result = cache.get_deltas_since(u32::MAX.wrapping_sub(1));
let input = format!(
"{}delta_window:\n{}",
get_deltas_since_input_to_string(cache.session_id_for_version(1), cache.serial(), u32::MAX.wrapping_sub(1)),
indent_block(&deltas_window_to_string(&deltas), 2),
);
let output = serial_result_detail_to_string(&result);
test_report(
"get_deltas_since_supports_incremental_updates_across_serial_wraparound",
"test purpose",
&input,
&output,
);
match result {
SerialResult::Delta(delta) => {
assert_eq!(delta.serial(), 0);
assert_eq!(delta.announced().len(), 2);
assert_eq!(delta.withdrawn().len(), 0);
match &delta.announced()[0] {
Payload::RouteOrigin(ro) => assert_eq!(ro, &b),
_ => panic!("expected announced RouteOrigin"),
}
match &delta.announced()[1] {
Payload::RouteOrigin(ro) => assert_eq!(ro, &a),
_ => panic!("expected announced RouteOrigin"),
}
}
_ => panic!("expected Delta"),
}
}
#[test]
fn get_deltas_since_returns_reset_required_when_client_serial_is_too_old_across_wraparound() {
let d_max = Arc::new(Delta::new(
u32::MAX,
vec![Payload::RouteOrigin(v4_origin(192, 0, 2, 0, 24, 24, 64496))],
vec![],
));
let d_zero = Arc::new(Delta::new(
0,
vec![Payload::RouteOrigin(v4_origin(198, 51, 100, 0, 24, 24, 64497))],
vec![],
));
let d_one = Arc::new(Delta::new(
1,
vec![Payload::RouteOrigin(v4_origin(203, 0, 113, 0, 24, 24, 64498))],
vec![],
));
let mut deltas = VecDeque::new();
deltas.push_back(d_max);
deltas.push_back(d_zero);
deltas.push_back(d_one);
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(1)
.timing(Timing::default())
.snapshot(Snapshot::from_payloads(vec![
Payload::RouteOrigin(v4_origin(192, 0, 2, 0, 24, 24, 64496)),
Payload::RouteOrigin(v4_origin(198, 51, 100, 0, 24, 24, 64497)),
Payload::RouteOrigin(v4_origin(203, 0, 113, 0, 24, 24, 64498)),
]))
.deltas(deltas.clone())
.build();
let client_serial = u32::MAX.wrapping_sub(2);
let result = cache.get_deltas_since(client_serial);
let input = format!(
"{}delta_window:\n{}",
get_deltas_since_input_to_string(cache.session_id_for_version(1), cache.serial(), client_serial),
indent_block(&deltas_window_to_string(&deltas), 2),
);
let output = serial_result_detail_to_string(&result);
test_report(
"get_deltas_since_returns_reset_required_when_client_serial_is_too_old_across_wraparound",
"test purpose",
&input,
&output,
);
match result {
SerialResult::ResetRequired => {}
_ => panic!("expected ResetRequired"),
}
}
#[test]
fn get_deltas_since_returns_reset_required_when_client_serial_is_in_future_across_wraparound() {
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(u32::MAX)
.timing(Timing::default())
.build();
let result = cache.get_deltas_since(0);
let input = get_deltas_since_input_to_string(cache.session_id_for_version(1), cache.serial(), 0);
let output = serial_result_detail_to_string(&result);
test_report(
"get_deltas_since_returns_reset_required_when_client_serial_is_in_future_across_wraparound",
"test purpose",
&input,
&output,
);
match result {
SerialResult::ResetRequired => {}
_ => panic!("expected ResetRequired"),
}
}
/// update() ?payload ?
/// serial delta?
#[tokio::test]
async fn update_no_change_keeps_serial_and_produces_no_delta() {
let old_a = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let old_b = v4_origin(198, 51, 100, 0, 24, 24, 64497);
let old_input = vec![
Payload::RouteOrigin(old_a.clone()),
Payload::RouteOrigin(old_b.clone()),
];
let snapshot = Snapshot::from_payloads(old_input.clone());
let mut cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(100)
.timing(Timing::default())
.snapshot(snapshot.clone())
.build();
let dir = tempfile::tempdir().unwrap();
let store = RtrStore::open(dir.path()).unwrap();
let new_payloads = vec![
Payload::RouteOrigin(old_b),
Payload::RouteOrigin(old_a),
];
cache.update(new_payloads.clone(), &store).unwrap();
let current_snapshot = cache.snapshot();
let result = cache.get_deltas_since(100);
let input = format!(
"old_snapshot :\n{}new_payloads :\n{}",
indent_block(&payloads_to_string(&old_input), 2),
indent_block(&payloads_to_string(&new_payloads), 2),
);
let output = format!(
"cache.serial_after_update: {}\ncurrent_snapshot:\n{}get_deltas_since(100):\n{}",
cache.serial(),
indent_block(&snapshot_hashes_and_sorted_view_to_string(&current_snapshot), 2),
indent_block(&serial_result_detail_to_string(&result), 2),
);
test_report(
"update_no_change_keeps_serial_and_produces_no_delta",
"test purpose",
&input,
&output,
);
assert_eq!(cache.serial(), 100);
assert!(cache.snapshot().same_content(&snapshot));
match result {
SerialResult::UpToDate => {}
_ => panic!("expected UpToDate"),
}
}
/// update() payload ?
/// ?serial?announced ?
#[tokio::test]
async fn update_add_only_increments_serial_and_generates_announced_delta() {
let old_a = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let new_b = v4_origin(198, 51, 100, 0, 24, 24, 64497);
let old_input = vec![Payload::RouteOrigin(old_a.clone())];
let old_snapshot = Snapshot::from_payloads(old_input.clone());
let mut cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(100)
.timing(Timing::default())
.snapshot(old_snapshot.clone())
.build();
let dir = tempfile::tempdir().unwrap();
let store = RtrStore::open(dir.path()).unwrap();
let new_payloads = vec![
Payload::RouteOrigin(old_a.clone()),
Payload::RouteOrigin(new_b.clone()),
];
cache.update(new_payloads.clone(), &store).unwrap();
let current_snapshot = cache.snapshot();
let result = cache.get_deltas_since(100);
let input = format!(
"old_snapshot :\n{}new_payloads :\n{}",
indent_block(&payloads_to_string(&old_input), 2),
indent_block(&payloads_to_string(&new_payloads), 2),
);
let output = format!(
"cache.serial_after_update: {}\ncurrent_snapshot:\n{}get_deltas_since(100):\n{}",
cache.serial(),
indent_block(&snapshot_hashes_and_sorted_view_to_string(&current_snapshot), 2),
indent_block(&serial_result_detail_to_string(&result), 2),
);
test_report(
"update_add_only_increments_serial_and_generates_announced_delta",
"test purpose",
&input,
&output,
);
assert_eq!(cache.serial(), 101);
match result {
SerialResult::Delta(delta) => {
assert_eq!(delta.serial(), 101);
assert_eq!(delta.announced().len(), 1);
assert_eq!(delta.withdrawn().len(), 0);
match &delta.announced()[0] {
Payload::RouteOrigin(ro) => assert_eq!(ro, &new_b),
_ => panic!("expected announced RouteOrigin"),
}
}
_ => panic!("expected Delta"),
}
}
/// update() payload ?
/// ?serial?withdrawn ?
#[tokio::test]
async fn update_remove_only_increments_serial_and_generates_withdrawn_delta() {
let old_a = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let old_b = v4_origin(198, 51, 100, 0, 24, 24, 64497);
let old_input = vec![
Payload::RouteOrigin(old_a.clone()),
Payload::RouteOrigin(old_b.clone()),
];
let old_snapshot = Snapshot::from_payloads(old_input.clone());
let mut cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(100)
.timing(Timing::default())
.snapshot(old_snapshot.clone())
.build();
let dir = tempfile::tempdir().unwrap();
let store = RtrStore::open(dir.path()).unwrap();
let new_payloads = vec![Payload::RouteOrigin(old_b.clone())];
cache.update(new_payloads.clone(), &store).unwrap();
let current_snapshot = cache.snapshot();
let result = cache.get_deltas_since(100);
let input = format!(
"old_snapshot :\n{}new_payloads :\n{}",
indent_block(&payloads_to_string(&old_input), 2),
indent_block(&payloads_to_string(&new_payloads), 2),
);
let output = format!(
"cache.serial_after_update: {}\ncurrent_snapshot:\n{}get_deltas_since(100):\n{}",
cache.serial(),
indent_block(&snapshot_hashes_and_sorted_view_to_string(&current_snapshot), 2),
indent_block(&serial_result_detail_to_string(&result), 2),
);
test_report(
"update_remove_only_increments_serial_and_generates_withdrawn_delta",
"test purpose",
&input,
&output,
);
assert_eq!(cache.serial(), 101);
match result {
SerialResult::Delta(delta) => {
assert_eq!(delta.serial(), 101);
assert_eq!(delta.announced().len(), 0);
assert_eq!(delta.withdrawn().len(), 1);
match &delta.withdrawn()[0] {
Payload::RouteOrigin(ro) => assert_eq!(ro, &old_a),
_ => panic!("expected withdrawn RouteOrigin"),
}
}
_ => panic!("expected Delta"),
}
}
/// update() payload
/// ?serial announced ?withdrawn?
#[tokio::test]
async fn update_add_and_remove_increments_serial_and_generates_both_sides() {
let old_a = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let old_b = v4_origin(198, 51, 100, 0, 24, 24, 64497);
let new_c = v6_origin(
Ipv6Addr::new(0x2001, 0xdb8, 0, 1, 0, 0, 0, 0),
48,
48,
64499,
);
let old_input = vec![
Payload::RouteOrigin(old_a.clone()),
Payload::RouteOrigin(old_b.clone()),
];
let old_snapshot = Snapshot::from_payloads(old_input.clone());
let mut cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(100)
.timing(Timing::default())
.snapshot(old_snapshot.clone())
.build();
let dir = tempfile::tempdir().unwrap();
let store = RtrStore::open(dir.path()).unwrap();
let new_payloads = vec![
Payload::RouteOrigin(old_b.clone()),
Payload::RouteOrigin(new_c.clone()),
];
cache.update(new_payloads.clone(), &store).unwrap();
let current_snapshot = cache.snapshot();
let result = cache.get_deltas_since(100);
let input = format!(
"old_snapshot :\n{}new_payloads :\n{}",
indent_block(&payloads_to_string(&old_input), 2),
indent_block(&payloads_to_string(&new_payloads), 2),
);
let output = format!(
"cache.serial_after_update: {}\ncurrent_snapshot:\n{}get_deltas_since(100):\n{}",
cache.serial(),
indent_block(&snapshot_hashes_and_sorted_view_to_string(&current_snapshot), 2),
indent_block(&serial_result_detail_to_string(&result), 2),
);
test_report(
"update_add_and_remove_increments_serial_and_generates_both_sides",
"test purpose",
&input,
&output,
);
assert_eq!(cache.serial(), 101);
match result {
SerialResult::Delta(delta) => {
assert_eq!(delta.serial(), 101);
assert_eq!(delta.announced().len(), 1);
assert_eq!(delta.withdrawn().len(), 1);
match &delta.announced()[0] {
Payload::RouteOrigin(ro) => assert_eq!(ro, &new_c),
_ => panic!("expected announced RouteOrigin"),
}
match &delta.withdrawn()[0] {
Payload::RouteOrigin(ro) => assert_eq!(ro, &old_a),
_ => panic!("expected withdrawn RouteOrigin"),
}
}
_ => panic!("expected Delta"),
}
}
/// ?prefix announce withdraw
/// ?UpToDate?
#[test]
fn get_deltas_since_cancels_announce_then_withdraw_for_same_prefix() {
let a = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let d1 = Arc::new(Delta::new(
101,
vec![Payload::RouteOrigin(a.clone())],
vec![],
));
let d2 = Arc::new(Delta::new(
102,
vec![],
vec![Payload::RouteOrigin(a.clone())],
));
let mut deltas = VecDeque::new();
deltas.push_back(d1);
deltas.push_back(d2);
// A ?
let final_snapshot = Snapshot::empty();
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(102)
.timing(Timing::default())
.snapshot(final_snapshot)
.deltas(deltas.clone())
.build();
let result = cache.get_deltas_since(100);
let input = format!(
"{}delta_window:\n{}",
get_deltas_since_input_to_string(cache.session_id_for_version(1), cache.serial(), 100),
indent_block(&deltas_window_to_string(&deltas), 2),
);
let output = serial_result_detail_to_string(&result);
test_report(
"get_deltas_since_cancels_announce_then_withdraw_for_same_prefix",
"test purpose",
&input,
&output,
);
match result {
SerialResult::UpToDate => {}
_ => panic!("expected UpToDate"),
}
}
/// ?prefix withdraw announce ?
/// ?UpToDate?
#[test]
fn get_deltas_since_cancels_withdraw_then_announce_for_same_prefix() {
let a = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let d1 = Arc::new(Delta::new(
101,
vec![],
vec![Payload::RouteOrigin(a.clone())],
));
let d2 = Arc::new(Delta::new(
102,
vec![Payload::RouteOrigin(a.clone())],
vec![],
));
let mut deltas = VecDeque::new();
deltas.push_back(d1);
deltas.push_back(d2);
// ?A?
let final_snapshot = Snapshot::from_payloads(vec![Payload::RouteOrigin(a.clone())]);
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(102)
.timing(Timing::default())
.snapshot(final_snapshot)
.deltas(deltas.clone())
.build();
let result = cache.get_deltas_since(100);
let input = format!(
"{}delta_window:\n{}",
get_deltas_since_input_to_string(cache.session_id_for_version(1), cache.serial(), 100),
indent_block(&deltas_window_to_string(&deltas), 2),
);
let output = serial_result_detail_to_string(&result);
test_report(
"get_deltas_since_cancels_withdraw_then_announce_for_same_prefix",
"test purpose",
&input,
&output,
);
match result {
SerialResult::UpToDate => {}
_ => panic!("expected UpToDate"),
}
}
/// ?A ?B
/// ?delta?withdraw A + announce B?
#[test]
fn get_deltas_since_merges_replacement_into_withdraw_and_announce() {
let a = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let b = v4_origin(192, 0, 2, 0, 24, 25, 64496);
let d1 = Arc::new(Delta::new(
101,
vec![],
vec![Payload::RouteOrigin(a.clone())],
));
let d2 = Arc::new(Delta::new(
102,
vec![Payload::RouteOrigin(b.clone())],
vec![],
));
let mut deltas = VecDeque::new();
deltas.push_back(d1);
deltas.push_back(d2);
let final_snapshot = Snapshot::from_payloads(vec![Payload::RouteOrigin(b.clone())]);
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(102)
.timing(Timing::default())
.snapshot(final_snapshot)
.deltas(deltas.clone())
.build();
let result = cache.get_deltas_since(100);
let input = format!(
"{}delta_window:\n{}",
get_deltas_since_input_to_string(cache.session_id_for_version(1), cache.serial(), 100),
indent_block(&deltas_window_to_string(&deltas), 2),
);
let output = serial_result_detail_to_string(&result);
test_report(
"get_deltas_since_merges_replacement_into_withdraw_and_announce",
"test purpose",
&input,
&output,
);
match result {
SerialResult::Delta(delta) => {
assert_eq!(delta.serial(), 102);
assert_eq!(delta.announced().len(), 1);
assert_eq!(delta.withdrawn().len(), 1);
match &delta.withdrawn()[0] {
Payload::RouteOrigin(ro) => assert_eq!(ro, &a),
_ => panic!("expected withdrawn RouteOrigin"),
}
match &delta.announced()[0] {
Payload::RouteOrigin(ro) => assert_eq!(ro, &b),
_ => panic!("expected announced RouteOrigin"),
}
}
_ => panic!("expected Delta"),
}
}
/// delta ?
/// ?
#[test]
fn get_deltas_since_merges_multiple_deltas_to_final_minimal_view() {
let a = v4_origin(192, 0, 2, 0, 24, 24, 64496);
let b = v4_origin(198, 51, 100, 0, 24, 24, 64497);
let c = v4_origin(203, 0, 113, 0, 24, 24, 64498);
// 100 -> 101 : +A
let d1 = Arc::new(Delta::new(
101,
vec![Payload::RouteOrigin(a.clone())],
vec![],
));
// 101 -> 102 : -A +B
let d2 = Arc::new(Delta::new(
102,
vec![Payload::RouteOrigin(b.clone())],
vec![Payload::RouteOrigin(a.clone())],
));
// 102 -> 103 : -B +C
let d3 = Arc::new(Delta::new(
103,
vec![Payload::RouteOrigin(c.clone())],
vec![Payload::RouteOrigin(b.clone())],
));
let mut deltas = VecDeque::new();
deltas.push_back(d1);
deltas.push_back(d2);
deltas.push_back(d3);
let final_snapshot = Snapshot::from_payloads(vec![Payload::RouteOrigin(c.clone())]);
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(103)
.timing(Timing::default())
.snapshot(final_snapshot)
.deltas(deltas.clone())
.build();
// ?serial=100 A/B ?+C
let result = cache.get_deltas_since(100);
let input = format!(
"{}delta_window:\n{}",
get_deltas_since_input_to_string(cache.session_id_for_version(1), cache.serial(), 100),
indent_block(&deltas_window_to_string(&deltas), 2),
);
let output = serial_result_detail_to_string(&result);
test_report(
"get_deltas_since_merges_multiple_deltas_to_final_minimal_view",
"test purpose",
&input,
&output,
);
match result {
SerialResult::Delta(delta) => {
assert_eq!(delta.serial(), 103);
assert_eq!(delta.announced().len(), 1);
assert_eq!(delta.withdrawn().len(), 0);
match &delta.announced()[0] {
Payload::RouteOrigin(ro) => assert_eq!(ro, &c),
_ => panic!("expected announced RouteOrigin"),
}
}
_ => panic!("expected Delta"),
}
}
#[test]
fn snapshot_from_payloads_unions_aspas_by_customer() {
let first = Payload::Aspa(Aspa::new(
Asn::from(64496u32),
vec![Asn::from(64497u32)],
));
let second = Payload::Aspa(Aspa::new(
Asn::from(64496u32),
vec![Asn::from(64498u32), Asn::from(64497u32)],
));
let snapshot = Snapshot::from_payloads(vec![first, second]);
let aspas = snapshot.aspas().iter().collect::<Vec<_>>();
assert_eq!(aspas.len(), 1);
assert_eq!(aspas[0].customer_asn(), Asn::from(64496u32));
assert_eq!(
aspas[0].provider_asns(),
&[Asn::from(64497u32), Asn::from(64498u32)]
);
}
#[test]
fn snapshot_diff_replaces_aspa_with_single_announcement() {
let old_snapshot = Snapshot::from_payloads(vec![Payload::Aspa(Aspa::new(
Asn::from(64496u32),
vec![Asn::from(64497u32)],
))]);
let new_snapshot = Snapshot::from_payloads(vec![Payload::Aspa(Aspa::new(
Asn::from(64496u32),
vec![Asn::from(64498u32)],
))]);
let (announced, withdrawn) = old_snapshot.diff(&new_snapshot);
assert_eq!(announced.len(), 1);
assert!(withdrawn.is_empty());
match &announced[0] {
Payload::Aspa(aspa) => {
assert_eq!(aspa.customer_asn(), Asn::from(64496u32));
assert_eq!(aspa.provider_asns(), &[Asn::from(64498u32)]);
}
_ => panic!("expected announced ASPA"),
}
}
#[test]
fn get_deltas_since_merges_aspa_replacement_into_single_announcement() {
let old = Aspa::new(Asn::from(64496u32), vec![Asn::from(64497u32)]);
let new = Aspa::new(Asn::from(64496u32), vec![Asn::from(64498u32)]);
let d1 = Arc::new(Delta::new(101, vec![], vec![Payload::Aspa(old.clone())]));
let d2 = Arc::new(Delta::new(102, vec![Payload::Aspa(new.clone())], vec![]));
let mut deltas = VecDeque::new();
deltas.push_back(d1);
deltas.push_back(d2);
let cache = RtrCacheBuilder::new()
.session_ids(SessionIds::from_array([42, 42, 42]))
.serial(102)
.timing(Timing::default())
.snapshot(Snapshot::from_payloads(vec![Payload::Aspa(new.clone())]))
.deltas(deltas)
.build();
let result = cache.get_deltas_since(100);
match result {
SerialResult::Delta(delta) => {
assert_eq!(delta.announced().len(), 1);
assert!(delta.withdrawn().is_empty());
match &delta.announced()[0] {
Payload::Aspa(aspa) => assert_eq!(aspa, &new),
_ => panic!("expected announced ASPA"),
}
}
_ => panic!("expected Delta"),
}
}
#[test]
fn validate_payloads_for_rtr_rejects_unsorted_snapshot_payloads() {
let low = Payload::RouteOrigin(v4_origin(192, 0, 2, 0, 24, 24, 64496));
let high = Payload::RouteOrigin(v4_origin(198, 51, 100, 0, 24, 24, 64497));
let err = validate_payloads_for_rtr(&[low, high], true).unwrap_err();
assert!(err
.to_string()
.contains("RTR payload ordering violation"));
}
#[test]
fn validate_payload_updates_for_rtr_rejects_unsorted_aspa_updates() {
let withdraw = (
false,
Payload::Aspa(Aspa::new(
Asn::from(64497u32),
vec![Asn::from(64500u32)],
)),
);
let announce = (
true,
Payload::Aspa(Aspa::new(
Asn::from(64496u32),
vec![Asn::from(64499u32)],
)),
);
let err = validate_payload_updates_for_rtr(&[withdraw, announce]).unwrap_err();
assert!(err.to_string().contains("withdraw ASPA"));
assert!(err.to_string().contains("announce ASPA"));
}