fix(client): handle sequence wrap in reply ordering

crates/irtt-client/src/client/tests/mod.rs

@@ -17,4 +17,5 @@ mod negotiation;
 mod no_test;
 mod open;
 mod probes_replies;
+mod sequence;
 mod ttl;

crates/irtt-client/src/client/tests/probes_replies.rs

@@ -1584,6 +1584,66 @@ fn send_probe_wraps_wire_sequence_at_u32_max() {
     server.join();
 }
 
+#[test]
+fn wrapped_reply_after_u32_max_is_not_late() {
+    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, &params, None);
+        socket.send_to(&reply, peer).unwrap();
+
+        socket
+            .set_read_timeout(Some(Duration::from_secs(2)))
+            .unwrap();
+        let mut seqs = Vec::new();
+        for _ in 0..2 {
+            let mut buf = [0_u8; 2048];
+            let (size, _) = socket.recv_from(&mut buf).unwrap();
+            tx.send(buf[..size].to_vec()).unwrap();
+            seqs.push(u32::from_le_bytes(buf[12..16].try_into().unwrap()));
+        }
+        assert_eq!(seqs, vec![u32::MAX, 0]);
+
+        let ts = TimestampFields::default();
+        let reply_max = echo_reply_packet(TOKEN, u32::MAX, &params, &ts, None);
+        socket.send_to(&reply_max, peer).unwrap();
+        thread::sleep(Duration::from_millis(10));
+        let reply_zero = echo_reply_packet(TOKEN, 0, &params, &ts, None);
+        socket.send_to(&reply_zero, peer).unwrap();
+    });
+    let config = ClientConfig {
+        socket_config: crate::SocketConfig {
+            recv_timeout: Some(Duration::from_millis(200)),
+            ..Default::default()
+        },
+        ..default_test_config(server.addr)
+    };
+    let mut client = Client::connect(config).unwrap();
+    assert_open_started(client.open(ClientTimestamp::now()).unwrap());
+    let session = client.session.as_mut().unwrap();
+    session.next_wire_seq = u32::MAX;
+    session.next_logical_seq = 41;
+
+    client.send_probe().unwrap();
+    client.send_probe().unwrap();
+    thread::sleep(Duration::from_millis(50));
+
+    let first = client.recv_once().unwrap();
+    assert!(matches!(
+        &first[0],
+        ClientEvent::EchoReply { seq: u32::MAX, .. }
+    ));
+
+    thread::sleep(Duration::from_millis(30));
+    let second = client.recv_once().unwrap();
+    assert!(
+        matches!(&second[0], ClientEvent::EchoReply { seq: 0, .. }),
+        "freshly wrapped reply should not be late, got {:?}",
+        second[0]
+    );
+    server.join();
+}
+
 #[test]
 fn send_probe_after_sending_done_is_noop() {
     let params = default_params();

crates/irtt-client/src/client/tests/sequence.rs

@@ -0,0 +1,45 @@
+use super::*;
+use std::time::Instant;
+
+use crate::{
+    probe::{CompletedSet, PendingMap, TimedOutMap},
+    session::ActiveSession,
+};
+
+#[test]
+fn sequence_ordering_handles_normal_and_wrapped_values() {
+    assert!(sequence_is_after(11, 10));
+    assert!(sequence_is_before(9, 10));
+    assert!(!sequence_is_after(10, 10));
+    assert!(!sequence_is_before(10, 10));
+
+    assert!(sequence_is_after(0, u32::MAX - 1));
+    assert!(!sequence_is_before(0, u32::MAX - 1));
+    assert!(sequence_is_after(1, u32::MAX));
+    assert!(!sequence_is_before(1, u32::MAX));
+
+    assert!(sequence_is_before(u32::MAX, 1));
+    assert!(!sequence_is_after(u32::MAX, 1));
+}
+
+#[test]
+fn highest_received_seq_updates_across_wrap() {
+    let mut session = ActiveSession {
+        next_wire_seq: 0,
+        next_logical_seq: 0,
+        highest_received_seq: Some(u32::MAX),
+        packets_sent: 0,
+        start_mono: Instant::now(),
+        end_mono: None,
+        next_send_at: Instant::now(),
+        pending: PendingMap::new(8),
+        timed_out: TimedOutMap::new(8),
+        completed: CompletedSet::new(8),
+        sending_done: false,
+    };
+
+    update_highest_received(&mut session, 0);
+    assert_eq!(session.highest_received_seq, Some(0));
+    update_highest_received(&mut session, u32::MAX);
+    assert_eq!(session.highest_received_seq, Some(0));
+}