Fix ChannelStream's channel not being closed (#397)

Co-authored-by: Eugene <inbox@null.page>
Eugeny · Dec 1, 2024 · fb8d3ea · fb8d3ea
1 parent 2a4f451
commit fb8d3ea
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/russh/src/channels/io/rx.rs b/russh/src/channels/io/rx.rs
@@ -89,3 +89,16 @@ where
         }
     }
 }
+
+impl<'i, S> Drop for ChannelRx<'i, S>
+where
+    S: From<(ChannelId, ChannelMsg)>,
+{
+    fn drop(&mut self) {
+        if let ChannelAsMut::Owned(ref mut channel) = &mut self.channel {
+            let _ = channel
+                .sender
+                .try_send((channel.id, ChannelMsg::Close).into());
+        }
+    }
+}