Prevent creating closed sockets

narrowlink · Mar 4, 2024 · f6e618e · f6e618e
1 parent 8a294de
commit f6e618e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/examples/tun_wintun.rs b/examples/tun_wintun.rs
@@ -62,7 +62,6 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                     }
                 };
                 println!("==== New TCP connection ====");
-                dbg!(tcp.local_addr());
                 tokio::spawn(async move {
                     let _ = tokio::io::copy_bidirectional(&mut tcp, &mut s).await;
                     println!("====== end tcp connection ======");

diff --git a/src/lib.rs b/src/lib.rs
@@ -119,6 +119,9 @@ impl IpStack {
                                     IpStackPacketProtocol::Tcp(h) => {
                                         match IpStackTcpStream::new(packet.src_addr(),packet.dst_addr(),h, pkt_sender.clone(),config.mtu,config.tcp_timeout).await{
                                             Ok(stream) => {
+                                                if stream.is_closed(){
+                                                    continue;
+                                                }
                                                 entry.insert(stream.stream_sender());
                                                 accept_sender.send(IpStackStream::Tcp(stream))?;
                                             }

diff --git a/src/stream/tcp.rs b/src/stream/tcp.rs
@@ -196,6 +196,9 @@ impl IpStackTcpStream {
     pub fn peer_addr(&self) -> SocketAddr {
         self.dst_addr
     }
+    pub(crate) fn is_closed(&self) -> bool {
+        matches!(self.tcb.get_state(), TcpState::Closed)
+    }
 }
 
 impl AsyncRead for IpStackTcpStream {