84 lines
3.0 KiB
Rust
84 lines
3.0 KiB
Rust
#[cfg(all(target_os = "linux", feature = "ancillary"))]
|
|
use super::*;
|
|
|
|
#[test]
|
|
#[cfg(all(target_os = "linux", feature = "ancillary"))]
|
|
fn echo_reply_metadata_propagates_observed_dscp_with_ancillary() {
|
|
let params = default_params();
|
|
let server = start_fake_server(move |socket, tx| {
|
|
let (_, peer) = recv_request(&socket, &tx);
|
|
let reply = open_reply(FLAG_OPEN | FLAG_REPLY, TOKEN, ¶ms, None);
|
|
socket.send_to(&reply, peer).unwrap();
|
|
|
|
let mut buf = [0_u8; 2048];
|
|
let (size, _) = socket.recv_from(&mut buf).unwrap();
|
|
tx.send(buf[..size].to_vec()).unwrap();
|
|
let seq = u32::from_le_bytes(buf[12..16].try_into().unwrap());
|
|
socket2::SockRef::from(&socket).set_tos_v4(184).unwrap();
|
|
let reply_packet =
|
|
echo_reply_packet(TOKEN, seq, ¶ms, &TimestampFields::default(), None);
|
|
socket.send_to(&reply_packet, peer).unwrap();
|
|
});
|
|
let config = default_test_config();
|
|
let mut client = Client::connect(server.addr.to_string(), config).unwrap();
|
|
client
|
|
.set_recv_timeout(Some(Duration::from_millis(200)))
|
|
.unwrap();
|
|
assert_open_started(client.open().unwrap());
|
|
|
|
client.send_probe().unwrap();
|
|
thread::sleep(Duration::from_millis(50));
|
|
|
|
let events = client.recv_once().unwrap();
|
|
assert_eq!(events.len(), 1);
|
|
match &events[0] {
|
|
ClientEvent::EchoReply { packet_meta, .. } => {
|
|
let Some(traffic_class) = packet_meta.traffic_class else {
|
|
metadata_unavailable_skip(
|
|
"echo_reply_metadata_propagates_observed_dscp_with_ancillary",
|
|
);
|
|
client.close().unwrap();
|
|
server.join();
|
|
return;
|
|
};
|
|
assert_eq!(traffic_class, 184);
|
|
assert_eq!(packet_meta.dscp, Some(46));
|
|
assert_eq!(packet_meta.ecn, Some(0));
|
|
}
|
|
other => panic!("expected EchoReply, got {other:?}"),
|
|
}
|
|
|
|
client.close().unwrap();
|
|
server.join();
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(all(target_os = "linux", feature = "ancillary"))]
|
|
fn echo_reply_metadata_propagates_kernel_rx_timestamp_with_ancillary() {
|
|
let params = default_params();
|
|
let (mut client, server) = open_client_with_echo_server(¶ms);
|
|
|
|
client.send_probe().unwrap();
|
|
thread::sleep(Duration::from_millis(50));
|
|
|
|
let events = client.recv_once().unwrap();
|
|
assert_eq!(events.len(), 1);
|
|
match &events[0] {
|
|
ClientEvent::EchoReply { packet_meta, .. } => {
|
|
let Some(timestamp) = packet_meta.kernel_rx_timestamp else {
|
|
kernel_rx_timestamp_unavailable_skip(
|
|
"echo_reply_metadata_propagates_kernel_rx_timestamp_with_ancillary",
|
|
);
|
|
client.close().unwrap();
|
|
server.join();
|
|
return;
|
|
};
|
|
assert!(timestamp.duration_since(SystemTime::UNIX_EPOCH).is_ok());
|
|
}
|
|
other => panic!("expected EchoReply, got {other:?}"),
|
|
}
|
|
|
|
client.close().unwrap();
|
|
server.join();
|
|
}
|