Add the 'writable_packet_size' to ChannelTx

Eugeny · Sep 21, 2023 · 9600bcb · 9600bcb
1 parent e75de5d
commit 9600bcb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/russh/src/channels/mod.rs b/russh/src/channels/mod.rs
@@ -152,6 +152,12 @@ impl<S: From<(ChannelId, ChannelMsg)> + Send + 'static> Channel<S> {
         )
     }
 
+    /// Returns the min between the maximum packet size and the
+    /// remaining window size in the channel.
+    pub async fn writable_packet_size(&self) -> usize {
+        self.max_packet_size.min(*self.window_size.lock().await) as usize
+    }
+
     pub fn id(&self) -> ChannelId {
         self.id
     }