Return expected version in the RTR server error PDU on mismatch.

NLnetLabs · Jan 12, 2024 · ea535b9 · ea535b9
1 parent 89b700b
commit ea535b9
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/rtr/server.rs b/src/rtr/server.rs
@@ -20,6 +20,13 @@ use super::payload::{Action, PayloadRef, Timing};
 use super::state::State;
 
 
+//============ Constants =====================================================
+
+/// The maximum protocol version we support.
+///
+/// We support all protcol versions from 0 up to and including this value.
+const MAX_VERSION: u8 = 2;
+
 //============ Traits ========================================================
 
 //------------ PayloadSource et al. ------------------------------------------
@@ -323,7 +330,7 @@ where Sock: AsyncRead + Unpin {
             if current != header.version() {
                 Err(Query::Error(
                     pdu::Error::new(
-                        header.version(),
+                        current, // respond with the version expected
                         8,
                         header,
                         "version switched during connection"
@@ -334,10 +341,10 @@ where Sock: AsyncRead + Unpin {
                 Ok(())
             }
         }
-        else if header.version() > 2 {
+        else if header.version() > MAX_VERSION {
             Err(Query::Error(
                 pdu::Error::new(
-                    header.version(),
+                    MAX_VERSION, // respond with what they should try
                     4,
                     header,
                     "only versions 0 to 2 supported"