From f7a2f1f24315cef91abd6b3ea34bb07e91662999 Mon Sep 17 00:00:00 2001
From: rustaceanrob <rob.netzke@gmail.com>
Date: Mon, 15 Sep 2025 14:45:19 +0100
Subject: [PATCH 1/2] Update flags on a record

---
 src/lib.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index d8e2098..9eb59cb 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -150,6 +150,11 @@ impl Record {
         self.services
     }
 
+    /// Update the most recent service flag information.
+    pub fn update_service_flags(&mut self, flags: ServiceFlags) {
+        self.services = flags;
+    }
+
     /// Serialize a record into bytes.
     pub fn serialize(self) -> Vec<u8> {
         let len = self.compute_size();
@@ -397,6 +402,7 @@ impl<const S: usize> Bucket<S> {
 
     fn successful_connection(&mut self, record: &Record) {
         let slot = Self::derive_slot(record);
+        let new_flags = record.services;
         if let Some(record) = &mut self.records[slot] {
             record.last_attempt = Some(
                 SystemTime::now()
@@ -409,6 +415,7 @@ impl<const S: usize> Bucket<S> {
                     .expect("time went backwards"),
             );
             record.failed_attempts = 0;
+            record.services = new_flags;
         }
     }
 

From 7139dd570e14b7bd74aa85d370b971a725afb873 Mon Sep 17 00:00:00 2001
From: rustaceanrob <rob.netzke@gmail.com>
Date: Mon, 15 Sep 2025 14:49:57 +0100
Subject: [PATCH 2/2] Create a record from an `AddrV2` source

---
 src/lib.rs | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 9eb59cb..9613ca9 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -72,6 +72,29 @@ impl Record {
         }
     }
 
+    pub fn new_from_addrv2_source(
+        addr: AddrV2,
+        port: u16,
+        services: ServiceFlags,
+        source: &AddrV2,
+    ) -> Self {
+        let source = match source {
+            AddrV2::Ipv4(ip) => IpAddr::V4(*ip).source_id(),
+            AddrV2::Ipv6(ip) => IpAddr::V6(*ip).source_id(),
+            // All the mixnets go in the same source,
+            _ => SourceId([1u8; 8]),
+        };
+        Self {
+            addr,
+            port,
+            source,
+            services,
+            failed_attempts: 0,
+            last_connection: None,
+            last_attempt: None,
+        }
+    }
+
     /// Build a new record from deserialization
     pub fn deserialize<R: Read>(reader: &mut R) -> Result<Self, std::io::Error> {
         let mut size_buf = [0u8; 1];