fix missed ntpv5 logic

pendulum-project · Dec 7, 2023 · fcade28 · fcade28
1 parent a5a590f
commit fcade28
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/ntp-proto/src/nts_record.rs b/ntp-proto/src/nts_record.rs
@@ -1492,6 +1492,9 @@ pub struct ClientToPoolDecoder {
 
     records: Vec<NtsRecord>,
     denied_servers: Vec<String>,
+
+    #[cfg(feature = "ntpv5")]
+    allow_v5: bool,
 }
 
 #[derive(Debug)]
@@ -1564,6 +1567,13 @@ impl ClientToPoolDecoder {
                 state.records.push(record);
                 Continue(state)
             }
+            #[cfg(feature = "ntpv5")]
+            DraftId { data } => {
+                if data == crate::packet::v5::DRAFT_VERSION.as_bytes() {
+                    state.allow_v5 = true;
+                }
+                Continue(state)
+            }
             NextProtocol { protocol_ids } => {
                 #[cfg(feature = "ntpv5")]
                 let selected = if state.allow_v5 {
@@ -1634,6 +1644,9 @@ pub struct PoolToServerDecoder {
 
     records: Vec<NtsRecord>,
     supported_algorithms: Vec<(u16, u16)>,
+
+    #[cfg(feature = "ntpv5")]
+    allow_v5: bool,
 }
 
 #[derive(Debug)]
@@ -1695,6 +1708,13 @@ impl PoolToServerDecoder {
                 state.records.push(record);
                 Continue(state)
             }
+            #[cfg(feature = "ntpv5")]
+            DraftId { data } => {
+                if data == crate::packet::v5::DRAFT_VERSION.as_bytes() {
+                    state.allow_v5 = true;
+                }
+                Continue(state)
+            }
             NextProtocol { protocol_ids } => {
                 #[cfg(feature = "ntpv5")]
                 let selected = if state.allow_v5 {