use super::{
FIPS_DISCOVERY_BACKOFF_BASE_SECS, FIPS_DISCOVERY_BACKOFF_MAX_SECS,
FIPS_DISCOVERY_FORWARD_MIN_INTERVAL_SECS, FIPS_DYNAMIC_PEER_ENDPOINT_PRIORITY,
FIPS_ENDPOINT_FAST_LINK_DEAD_TIMEOUT_SECS, FIPS_ENDPOINT_HEARTBEAT_INTERVAL_SECS,
FIPS_ENDPOINT_LINK_DEAD_TIMEOUT_SECS, FIPS_ENDPOINT_PENDING_PACKETS_PER_DEST,
FIPS_ENDPOINT_REKEY_AFTER_SECS, FIPS_ENDPOINT_SESSION_IDLE_TIMEOUT_SECS,
FIPS_ENDPOINT_DIRECT_PACKET_RUN_MAX_PACKETS, FIPS_LAN_DISCOVERY_SCOPE_PREFIX,
FIPS_MESH_EVENT_DRAIN_LIMIT,
FIPS_NOSTR_EXTENDED_COOLDOWN_SECS, FIPS_NOSTR_FAILURE_STREAK_THRESHOLD,
FIPS_NOSTR_EXIT_OPEN_DISCOVERY_MAX_PENDING, FIPS_NOSTR_OPEN_DISCOVERY_MAX_PENDING,
FIPS_NOSTR_STARTUP_SWEEP_MAX_AGE_SECS, FIPS_RECENT_NON_ROSTER_TRANSIT_MAX_SEEDS,
FIPS_RECONNECT_BACKOFF_BASE_SECS, FIPS_RECONNECT_BACKOFF_MAX_SECS,
FIPS_STATIC_NON_ROSTER_TRANSIT_MAX_SEEDS,
FIPS_CONFIGURED_PEER_ENDPOINT_PRIORITY, FIPS_PRIVATE_DYNAMIC_PEER_ENDPOINT_PRIORITY,
FipsEndpointPeerTransportConfig, FipsEndpointTransportConfig, FipsPeerActivity,
FipsPeerActivitySnapshot, FipsPeerAddressHint, FipsPeerIdentityMap, FipsPeerRxKind,
FipsPrivateMeshEvent,
FipsPrivateMeshRuntime, FipsPrivateTunnelConfig, Ipv4Subnet,
cap_recent_non_roster_transit_endpoints, control_frame_destination_peer,
control_frame_source_pubkey, decode_endpoint_control_frame, drain_event_batch,
endpoint_identity_for_send,
filter_stamped_tunnel_endpoints, filter_static_tunnel_endpoints_with_policy,
filter_static_tunnel_endpoints_with_policy_and_route_check,
fips_endpoint_config_with_open_discovery_limit, fips_endpoint_peers_from_mesh,
fips_lan_discovery_scope, fips_peer_address_from_hint,
fips_tunnel_requires_endpoint_restart, linux_cap_eff_has_net_admin,
linux_private_ipv4_route_subnets_from_ip_route,
linux_route_get_has_direct_private_endpoint_route, linux_tun_setup_error,
macos_private_ipv4_route_subnets_from_netstat,
macos_route_get_has_direct_private_endpoint_route, mesh_status_from_endpoint_peer,
other_endpoint_peer_statuses, parse_fips_nostr_discovery_policy,
parse_linux_tun_tx_queue_len, participant_pubkey_bytes, peer_activity_map, peer_identity_map,
prioritize_join_roster_peer,
static_endpoint_allowed_on_current_underlay_with_route_check, strip_cidr,
tag_authenticated_transport_addr, unix_timestamp,
};
#[cfg(any(target_os = "linux", target_os = "macos"))]
use super::{BorrowedTunFd, TunPipelinePacket, raw_write_packet_to_tun};
#[cfg(target_os = "linux")]
use super::LINUX_VIRTIO_NET_HDR_LEN;
use fips_endpoint::{
Config, ConnectPolicy, FipsEndpointData, FipsEndpointDirectPacketRun, FipsEndpointMessage,
FipsEndpointPeer, NodeAddr, NostrDiscoveryPolicy, PeerConfig as FipsPeerConfig,
PeerIdentity, RoutingMode, TransportInstances, UdpConfig,
};
use nostr_sdk::prelude::{Keys, ToBech32};
use nostr_vpn_core::config::{
AppConfig, InternetSource, PendingOutboundJoinRequest, derive_mesh_tunnel_ip,
};
use nostr_vpn_core::fips_control::{
FipsControlFrame, JoinRosterControl, NetworkRoster, PeerEndpointHint, SignedRoster,
encode_fips_control_frame,
};
use nostr_vpn_core::fips_mesh::{FipsMeshPeerConfig, FipsMeshRuntime};
use nostr_vpn_core::join_requests::MeshJoinRequest;
use std::collections::{HashMap, HashSet};
use std::net::{IpAddr, Ipv4Addr, UdpSocket};
use std::time::Duration;
const FIPS_NOSTR_DISCOVERY_APP: &str = "fips-overlay-v1";
fn send_tunnel_packet_batch_owned_with_capacity(
runtime: &FipsPrivateMeshRuntime,
packets: Vec<Vec<u8>>,
turn_capacity: usize,
) -> anyhow::Result<usize> {
if packets.is_empty() {
return Ok(0);
}
let input_packets = packets.len();
let mesh = runtime.mesh.load();
let peer_identities = runtime.peer_identities.load();
let mut runs = Vec::new();
let mut routed_packets = 0usize;
{
let _t = crate::pipeline_profile::Timer::start(crate::pipeline_profile::Stage::MeshRoute);
for packet in packets {
let Some(outgoing) = mesh.route_outbound_packet_owned_with_peer(packet) else {
continue;
};
routed_packets += 1;
let participant_key = outgoing.participant_pubkey_bytes.copied();
#[cfg(feature = "paid-exit")]
runtime.note_paid_route_outbound_packet(
Some(outgoing.participant_pubkey),
outgoing.participant_pubkey_bytes,
&outgoing.bytes,
)?;
FipsPrivateMeshRuntime::push_endpoint_send_run(
&mut runs,
&peer_identities,
outgoing.participant_pubkey,
participant_key,
outgoing.endpoint_node_addr,
outgoing.bytes,
);
}
}
drop(peer_identities);
drop(mesh);
crate::pipeline_profile::record_mesh_send_batch(
input_packets,
routed_packets,
runs.len(),
turn_capacity,
);
let _t =
crate::pipeline_profile::Timer::start(crate::pipeline_profile::Stage::MeshEndpointSend);
runtime.blocking_send_endpoint_send_runs(runs)
}
async fn recv_mesh_event_batch_into(
runtime: &FipsPrivateMeshRuntime,
messages: &mut Vec<FipsEndpointMessage>,
events: &mut Vec<FipsPrivateMeshEvent>,
limit: usize,
) -> anyhow::Result<Option<usize>> {
let limit = limit.clamp(1, FIPS_MESH_EVENT_DRAIN_LIMIT);
events.clear();
loop {
if drain_direct_endpoint_mesh_events_into(runtime, events).await? > 0 {
return Ok(Some(events.len()));
}
let Some(_) = (match tokio::time::timeout(
Duration::from_millis(10),
runtime.endpoint.recv_batch_into(messages, limit),
)
.await
{
Ok(result) => result,
Err(_) => continue,
}) else {
return Ok(None);
};
let now = Some(unix_timestamp());
events.reserve(messages.len());
for message in messages.drain(..) {
if let Some(event) = endpoint_message_to_mesh_event(runtime, message, now).await? {
events.push(event);
}
}
if !events.is_empty() {
return Ok(Some(events.len()));
}
}
}
async fn endpoint_message_to_mesh_event(
runtime: &FipsPrivateMeshRuntime,
message: FipsEndpointMessage,
now: Option<u64>,
) -> anyhow::Result<Option<FipsPrivateMeshEvent>> {
let outcome = runtime.endpoint_message_to_mesh_event_outcome(message, now)?;
if let Some(reply) = outcome.reply
&& let Err(error) = runtime
.endpoint
.send_batch_to_peer(reply.peer, vec![reply.data])
.await
{
eprintln!("fips: failed to reply to peer ping: {error}");
}
Ok(outcome.event)
}
async fn drain_direct_endpoint_mesh_events_into(
runtime: &FipsPrivateMeshRuntime,
events: &mut Vec<FipsPrivateMeshEvent>,
) -> anyhow::Result<usize> {
let initial_len = events.len();
let runs = match runtime
.direct_endpoint_rx
.try_recv(FIPS_ENDPOINT_DIRECT_PACKET_RUN_MAX_PACKETS)
{
Ok(runs) => runs,
Err(std::sync::mpsc::TryRecvError::Empty) => return Ok(0),
Err(std::sync::mpsc::TryRecvError::Disconnected) => return Ok(0),
};
direct_endpoint_packet_runs_to_mesh_events(
runtime,
runs,
Some(unix_timestamp()),
events,
)
.await?;
Ok(events.len().saturating_sub(initial_len))
}
async fn direct_endpoint_packet_runs_to_mesh_events(
runtime: &FipsPrivateMeshRuntime,
runs: Vec<FipsEndpointDirectPacketRun>,
now: Option<u64>,
events: &mut Vec<FipsPrivateMeshEvent>,
) -> anyhow::Result<()> {
for run in runs {
let source_peer = *run.source_peer();
let enqueued_at_ms = run.enqueued_at_ms();
for packet in run.packet_slices() {
let message = FipsEndpointMessage {
source_peer,
data: FipsEndpointData::new(packet.to_vec()),
enqueued_at_ms,
};
if let Some(event) = endpoint_message_to_mesh_event(runtime, message, now).await? {
events.push(event);
}
}
}
Ok(())
}
#[test]
fn macos_udp_send_buffer_derives_release_defaults() {
assert_eq!(super::macos_default_udp_send_buf_size(), 256 * 1024);
#[cfg(target_os = "macos")]
{
assert_eq!(super::DEFAULT_FIPS_UDP_SEND_BUF_SIZE, Some(256 * 1024));
}
}
#[test]
fn linux_tun_tx_queue_len_env_keeps_bounded_default() {
assert_eq!(parse_linux_tun_tx_queue_len(None, 4096), Some(4096));
assert_eq!(parse_linux_tun_tx_queue_len(Some(""), 4096), Some(4096));
assert_eq!(parse_linux_tun_tx_queue_len(Some("500"), 4096), Some(500));
assert_eq!(parse_linux_tun_tx_queue_len(Some("1"), 4096), Some(64));
assert_eq!(
parse_linux_tun_tx_queue_len(Some("999999"), 4096),
Some(65_536)
);
assert_eq!(parse_linux_tun_tx_queue_len(Some("0"), 4096), None);
assert_eq!(parse_linux_tun_tx_queue_len(Some("off"), 4096), None);
assert_eq!(parse_linux_tun_tx_queue_len(Some("no"), 4096), None);
}
#[test]
fn parses_fips_nostr_discovery_policy_override() {
for (raw, expected) in [
("configured-only", NostrDiscoveryPolicy::ConfiguredOnly),
("configured_only", NostrDiscoveryPolicy::ConfiguredOnly),
("open", NostrDiscoveryPolicy::Open),
("disabled", NostrDiscoveryPolicy::Disabled),
] {
assert_eq!(parse_fips_nostr_discovery_policy(raw), Some(expected));
}
assert_eq!(parse_fips_nostr_discovery_policy("wat"), None);
}
#[test]
fn authenticated_transport_addr_preserves_tcp_type_and_legacy_udp() {
assert_eq!(
tag_authenticated_transport_addr(
Some("203.0.113.20:51820".to_string()),
Some("udp".to_string())
),
Some("203.0.113.20:51820".to_string())
);
assert_eq!(
tag_authenticated_transport_addr(
Some("203.0.113.20:443".to_string()),
Some("tcp".to_string())
),
Some("tcp:203.0.113.20:443".to_string())
);
assert_eq!(
tag_authenticated_transport_addr(Some("tcp:203.0.113.20:443".to_string()), None),
Some("tcp:203.0.113.20:443".to_string())
);
}
#[cfg(any(target_os = "linux", target_os = "macos"))]
#[test]
fn raw_tun_write_keeps_fd_open_and_writes_platform_frame() {
let mut pipe_fds = [0; 2];
let pipe_result = unsafe { libc::pipe(pipe_fds.as_mut_ptr()) };
assert_eq!(pipe_result, 0, "pipe should open");
let read_fd = pipe_fds[0];
let write_fd = pipe_fds[1];
let packet = [0x45, 0, 0, 20, 1, 2, 3, 4];
let tun_fd = BorrowedTunFd::new(write_fd);
raw_write_packet_to_tun(&tun_fd, &packet, 2).expect("write packet frame");
raw_write_packet_to_tun(&tun_fd, &packet, 2).expect("fd should remain writable");
let expected_frame: Vec<u8> = {
#[cfg(target_os = "macos")]
{
let mut frame = vec![0, 0, 0, 2];
frame.extend_from_slice(&packet);
frame
}
#[cfg(target_os = "linux")]
{
let mut frame = vec![0; LINUX_VIRTIO_NET_HDR_LEN];
frame.extend_from_slice(&packet);
frame
}
};
let mut expected = expected_frame.clone();
expected.extend_from_slice(&expected_frame);
let mut read_buf = vec![0_u8; expected.len()];
let read = unsafe {
libc::read(
read_fd,
read_buf.as_mut_ptr().cast::<libc::c_void>(),
read_buf.len(),
)
};
unsafe {
libc::close(read_fd);
libc::close(write_fd);
}
assert_eq!(read as usize, expected.len());
assert_eq!(read_buf, expected);
}
#[cfg(any(target_os = "linux", target_os = "macos"))]
#[test]
fn blocking_tun_write_keeps_fd_open_and_writes_platform_frame() {
let mut pipe_fds = [0; 2];
let pipe_result = unsafe { libc::pipe(pipe_fds.as_mut_ptr()) };
assert_eq!(pipe_result, 0, "pipe should open");
let read_fd = pipe_fds[0];
let write_fd = pipe_fds[1];
let stop = std::sync::atomic::AtomicBool::new(false);
let packet = [0x45, 0, 0, 20, 1, 2, 3, 4];
let tun_fd = BorrowedTunFd::new(write_fd);
super::write_packet_to_tun_blocking(tun_fd, &packet, &stop);
super::write_packet_to_tun_blocking(tun_fd, &packet, &stop);
let expected_frame: Vec<u8> = {
#[cfg(target_os = "macos")]
{
let mut frame = vec![0, 0, 0, 2];
frame.extend_from_slice(&packet);
frame
}
#[cfg(target_os = "linux")]
{
let mut frame = vec![0; LINUX_VIRTIO_NET_HDR_LEN];
frame.extend_from_slice(&packet);
frame
}
};
let mut expected = expected_frame.clone();
expected.extend_from_slice(&expected_frame);
let mut read_buf = vec![0_u8; expected.len()];
let read = unsafe {
libc::read(
read_fd,
read_buf.as_mut_ptr().cast::<libc::c_void>(),
read_buf.len(),
)
};
unsafe {
libc::close(read_fd);
libc::close(write_fd);
}
assert_eq!(read as usize, expected.len());
assert_eq!(read_buf, expected);
}
#[test]
fn fips_peer_address_hint_splits_transport_tags_for_live_updates() {
let tcp = fips_peer_address_from_hint(&FipsPeerAddressHint {
addr: "tcp:203.0.113.20:443".to_string(),
seen_at_ms: Some(123_000),
priority: FIPS_DYNAMIC_PEER_ENDPOINT_PRIORITY,
});
assert_eq!(tcp.transport, "tcp");
assert_eq!(tcp.addr, "203.0.113.20:443");
assert_eq!(tcp.seen_at_ms, Some(123_000));
assert_eq!(tcp.priority, FIPS_DYNAMIC_PEER_ENDPOINT_PRIORITY);
let udp = fips_peer_address_from_hint(&FipsPeerAddressHint {
addr: "udp:203.0.113.21:2121".to_string(),
seen_at_ms: None,
priority: FIPS_CONFIGURED_PEER_ENDPOINT_PRIORITY,
});
assert_eq!(udp.transport, "udp");
assert_eq!(udp.addr, "203.0.113.21:2121");
assert_eq!(udp.priority, FIPS_CONFIGURED_PEER_ENDPOINT_PRIORITY);
}
#[test]
fn fips_private_tunnel_config_uses_non_roster_peer_setting_for_discovery_policy() {
if std::env::var("NVPN_FIPS_NOSTR_DISCOVERY_POLICY").is_ok() {
return;
}
let mut app = AppConfig::generated();
app.fips_host_tunnel_enabled = false;
app.connect_to_non_roster_fips_peers = false;
let network_id = app.effective_network_id();
let config = FipsPrivateTunnelConfig::from_app(
&app,
&network_id,
"utun100",
app.own_nostr_pubkey_hex().ok().as_deref(),
None,
&[],
)
.expect("configured-only tunnel config");
assert_eq!(
config.nostr_discovery_policy,
NostrDiscoveryPolicy::ConfiguredOnly
);
app.node.advertise_exit_node = true;
let config = FipsPrivateTunnelConfig::from_app(
&app,
&network_id,
"utun100",
app.own_nostr_pubkey_hex().ok().as_deref(),
None,
&[],
)
.expect("advertised exit tunnel config");
assert_eq!(
config.nostr_discovery_policy,
NostrDiscoveryPolicy::Open
);
assert_eq!(
config.open_discovery_max_pending,
FIPS_NOSTR_EXIT_OPEN_DISCOVERY_MAX_PENDING
);
app.node.advertise_exit_node = false;
app.connect_to_non_roster_fips_peers = true;
let config = FipsPrivateTunnelConfig::from_app(
&app,
&network_id,
"utun100",
app.own_nostr_pubkey_hex().ok().as_deref(),
None,
&[],
)
.expect("open tunnel config");
assert_eq!(config.nostr_discovery_policy, NostrDiscoveryPolicy::Open);
}
#[cfg(feature = "paid-exit")]
#[test]
fn fips_private_tunnel_config_opens_discovery_for_paid_exit_sellers() {
if std::env::var("NVPN_FIPS_NOSTR_DISCOVERY_POLICY").is_ok() {
return;
}
let mut app = AppConfig::generated();
app.fips_host_tunnel_enabled = false;
app.connect_to_non_roster_fips_peers = false;
app.paid_exit.enabled = true;
let network_id = app.effective_network_id();
let config = FipsPrivateTunnelConfig::from_app(
&app,
&network_id,
"utun100",
app.own_nostr_pubkey_hex().ok().as_deref(),
None,
&[],
)
.expect("paid exit seller tunnel config");
assert_eq!(config.nostr_discovery_policy, NostrDiscoveryPolicy::Open);
}
#[test]
fn fips_restart_predicate_includes_nostr_discovery_enabled() {
let app = AppConfig::generated();
let network_id = app.effective_network_id();
let current = FipsPrivateTunnelConfig::from_app(
&app,
&network_id,
"utun100",
app.own_nostr_pubkey_hex().ok().as_deref(),
None,
&[],
)
.expect("fips tunnel config");
let mut next = current.clone();
next.nostr_discovery_enabled = !current.nostr_discovery_enabled;
assert!(
fips_tunnel_requires_endpoint_restart(¤t, &next),
"toggling Nostr discovery must tear down old relay subscriptions"
);
}
#[test]
fn linux_cap_eff_parsing_detects_net_admin() {
assert_eq!(
linux_cap_eff_has_net_admin("CapEff:\t0000000000000000\n"),
Some(false)
);
assert_eq!(
linux_cap_eff_has_net_admin("CapEff:\t0000000000001000\n"),
Some(true)
);
assert_eq!(linux_cap_eff_has_net_admin("Name:\tnvpn\n"), None);
}
#[test]
fn linux_tun_setup_error_points_to_root_service_or_docker_flags() {
let message = linux_tun_setup_error("utun100", "current process lacks CAP_NET_ADMIN");
assert!(message.contains("CAP_NET_ADMIN"));
assert!(message.contains("/dev/net/tun"));
assert!(message.contains("utun100"));
assert!(message.contains("sudo nvpn start --connect"));
assert!(message.contains("system service"));
assert!(message.contains("--cap-add NET_ADMIN --device /dev/net/tun"));
}
fn ipv4_packet(source: Ipv4Addr, destination: Ipv4Addr) -> Vec<u8> {
let payload = [0xde, 0xad, 0xbe, 0xef];
let total_len = 20 + payload.len();
let mut packet = vec![0_u8; total_len];
packet[0] = 0x45;
packet[2..4].copy_from_slice(&(total_len as u16).to_be_bytes());
packet[8] = 64;
packet[9] = 17;
packet[12..16].copy_from_slice(&source.octets());
packet[16..20].copy_from_slice(&destination.octets());
packet[20..].copy_from_slice(&payload);
packet
}
fn assert_peer_data_activity(
runtime: &FipsPrivateMeshRuntime,
participant_pubkey: &str,
expected_endpoint_data_bytes: u64,
) {
let status = runtime
.peer_statuses()
.into_iter()
.find(|status| status.pubkey == participant_pubkey)
.expect("peer status");
assert_eq!(status.last_seen_at, status.last_data_seen_at);
assert!(
status.last_data_seen_at.is_some(),
"admitted endpoint data should stamp data freshness"
);
assert_eq!(status.last_control_seen_at, None);
assert_eq!(status.tx_bytes, expected_endpoint_data_bytes);
assert_eq!(status.rx_bytes, expected_endpoint_data_bytes);
}
#[test]
fn drain_event_batch_respects_limit() {
let (tx, mut rx) =
tokio::sync::mpsc::channel::<FipsPrivateMeshEvent>(FIPS_MESH_EVENT_DRAIN_LIMIT + 8);
for index in 0..(FIPS_MESH_EVENT_DRAIN_LIMIT + 5) {
tx.try_send(FipsPrivateMeshEvent::Presence {
participant_pubkey: format!("peer-{index}"),
last_seen_at: index as u64,
})
.expect("queue test event");
}
let drained = drain_event_batch(&mut rx, FIPS_MESH_EVENT_DRAIN_LIMIT);
assert_eq!(drained.len(), FIPS_MESH_EVENT_DRAIN_LIMIT);
assert_eq!(rx.len(), 5);
}
#[test]
fn stateful_control_datagram_is_recognized_for_drop_before_tunnel_data() {
let source = Keys::generate();
let source_npub = source.public_key().to_bech32().expect("source npub");
let frame = FipsControlFrame::Capabilities {
network_id: "network".to_string(),
capabilities: Default::default(),
};
let message = FipsEndpointMessage {
source_peer: PeerIdentity::from_npub(&source_npub).expect("source identity"),
data: FipsEndpointData::new(
encode_fips_control_frame(&frame).expect("encode stateful frame"),
),
enqueued_at_ms: 0,
};
assert_eq!(
decode_endpoint_control_frame(&message).expect("decode control datagram"),
Some(frame)
);
}
#[test]
fn peer_activity_map_preserves_existing_configured_peer_activity() {
use std::sync::Arc;
let alice = Keys::generate().public_key().to_hex();
let bob = Keys::generate().public_key().to_hex();
let removed = Keys::generate().public_key().to_hex();
let alice_key = participant_pubkey_bytes(&alice).expect("alice key");
let bob_key = participant_pubkey_bytes(&bob).expect("bob key");
let removed_key = participant_pubkey_bytes(&removed).expect("removed key");
let alice_activity = Arc::new(FipsPeerActivity::default());
alice_activity.note_tx(42);
alice_activity.note_rx(7, 123, FipsPeerRxKind::Control);
alice_activity.note_rx(11, 130, FipsPeerRxKind::Data);
let mut previous = HashMap::new();
previous.insert(alice_key, Arc::clone(&alice_activity));
previous.insert(removed_key, Arc::new(FipsPeerActivity::default()));
let next = peer_activity_map(&[alice.clone(), bob.clone()], Some(&previous));
assert!(Arc::ptr_eq(next.get(&alice_key).unwrap(), &alice_activity));
assert_eq!(
next.get(&alice_key).unwrap().snapshot(),
FipsPeerActivitySnapshot {
last_seen_at: Some(130),
last_control_seen_at: Some(123),
last_data_seen_at: Some(130),
tx_bytes: 42,
rx_bytes: 18,
}
);
assert_eq!(
next.get(&bob_key).unwrap().snapshot(),
FipsPeerActivitySnapshot::default()
);
assert!(!next.contains_key(&removed_key));
}
#[cfg(feature = "paid-exit")]
#[test]
fn paid_route_accounting_uses_pubkey_bytes_and_ignores_invalid_identity() {
use super::{
FipsPaidRouteAccounting, FipsPaidRouteAccountingPeer, FipsPaidRouteAccountingRole,
};
let participant = Keys::generate().public_key().to_hex();
let participant_key = participant_pubkey_bytes(&participant).expect("participant key");
let packet = paid_route_test_ipv4_udp_packet(64);
let mut accounting = FipsPaidRouteAccounting::default();
accounting.replace_peers([FipsPaidRouteAccountingPeer::parse(
&participant,
FipsPaidRouteAccountingRole::LocalBuyer,
)
.expect("accounting peer")]);
accounting.record_outbound(None, Some(&participant_key), &packet);
let usage = accounting.drain(&participant);
assert_eq!(usage.tx_bytes, 64);
assert_eq!(usage.tx_packets, 1);
assert_eq!(usage.billable_bytes, 64);
assert_eq!(usage.billable_packets, 1);
assert!(
FipsPaidRouteAccountingPeer::parse(
"not-a-pubkey",
FipsPaidRouteAccountingRole::LocalBuyer,
)
.is_none()
);
accounting.record_outbound(Some("not-a-pubkey"), None, &packet);
let invalid_usage = accounting.drain("not-a-pubkey");
assert_eq!(invalid_usage.tx_bytes, 0);
assert_eq!(invalid_usage.billable_bytes, 0);
}
#[test]
fn peer_identity_map_resolves_endpoint_identities_and_skips_invalid_npubs() {
let participant = Keys::generate().public_key().to_hex();
let endpoint_keys = Keys::generate();
let endpoint_hex = endpoint_keys.public_key().to_hex();
let endpoint_npub = endpoint_keys.public_key().to_bech32().expect("npub");
let invalid_participant = "invalid-participant".to_string();
let identities = peer_identity_map(&[
FipsMeshPeerConfig {
participant_pubkey: participant.clone(),
endpoint_npub: format!(" {endpoint_hex} "),
allowed_ips: Vec::new(),
},
FipsMeshPeerConfig {
participant_pubkey: invalid_participant.clone(),
endpoint_npub: "not-an-npub".to_string(),
allowed_ips: Vec::new(),
},
]);
let endpoint_node_addr = *PeerIdentity::from_npub(&endpoint_npub)
.expect("endpoint identity")
.node_addr()
.as_bytes();
let participant_key = participant_pubkey_bytes(&participant).expect("participant key");
assert_eq!(identities.by_participant.len(), 1);
assert!(identities.by_participant.contains_key(&participant_key));
assert_eq!(identities.by_endpoint_node_addr.len(), 1);
assert_eq!(
identities
.identity_for_participant(&participant)
.expect("resolved endpoint identity")
.npub(),
endpoint_npub
);
assert_eq!(
identities
.identity_for_send(Some(&participant_key), &endpoint_node_addr)
.expect("resolved endpoint identity by node addr")
.npub(),
endpoint_npub
);
assert_eq!(
identities
.identity_for_send(None, &endpoint_node_addr)
.expect("resolved endpoint identity by node addr without participant")
.npub(),
endpoint_npub
);
assert_eq!(
endpoint_identity_for_send(&identities, Some(&participant_key), &endpoint_node_addr)
.expect("send identity")
.npub(),
endpoint_npub
);
assert!(
identities
.identity_for_participant(&invalid_participant)
.is_none()
);
}
#[cfg(feature = "paid-exit")]
fn paid_route_test_ipv4_udp_packet(total_len: usize) -> Vec<u8> {
assert!(total_len >= 28);
assert!(total_len <= u16::MAX as usize);
let udp_len = total_len - 20;
let mut packet = vec![0u8; total_len];
packet[0] = 0x45;
packet[2..4].copy_from_slice(&(total_len as u16).to_be_bytes());
packet[9] = 17;
packet[12..16].copy_from_slice(&[10, 8, 0, 2]);
packet[16..20].copy_from_slice(&[198, 51, 100, 1]);
packet[20..22].copy_from_slice(&12345u16.to_be_bytes());
packet[22..24].copy_from_slice(&53u16.to_be_bytes());
packet[24..26].copy_from_slice(&(udp_len as u16).to_be_bytes());
packet
}