fix(smol): stop udp futures and reclaim resources for closed sockets

XOR-op · Nov 24, 2024 · 372de00 · 372de00
1 parent bd169f4
commit 372de00
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/boltconn/src/transport/smol.rs b/boltconn/src/transport/smol.rs
@@ -694,7 +694,8 @@ impl SmolStack {
 
     pub fn purge_timeout_udp(&mut self) {
         self.udp_conn.retain(|_port, task| {
-            if task.last_active.elapsed() > self.udp_timeout {
+            let socket = self.socket_set.get_mut::<SmolUdpSocket>(task.handle);
+            if !socket.is_open() || task.last_active.elapsed() > self.udp_timeout {
                 self.socket_set.remove(task.handle);
                 task.abort_handle.cancel();
                 false