Tweak

Author: vkill

async-ssh2-lite/tests/integration_tests/session__channel_forward_listen.rs

@@ -90,8 +90,13 @@ async fn __run__session__channel_forward_listen__with_tokio_spawn<
                         tokio::task::spawn(async move {
                             let mut buf = vec![0; 128];
                             let mut n_read = 0;
+                            let mut n_retry = 0;
                             loop {
-                                let n = channel.read(&mut buf[n_read..]).await?;
+                                let n = tokio::time::timeout(
+                                    tokio::time::Duration::from_millis(3000),
+                                    channel.read(&mut buf[n_read..]),
+                                )
+                                .await??;
                                 n_read += n;
                                 if n == 0 {
                                     break;
@@ -100,6 +105,11 @@ async fn __run__session__channel_forward_listen__with_tokio_spawn<
                                 if n_read >= 78 {
                                     break;
                                 }
+                                n_retry += 1;
+                                if n_retry > 3 {
+                                    eprintln!("Max read attempts reached.");
+                                    break;
+                                }
                             }
                             println!(
                                 "channel.read successful, data:{}",