From 2cc7c129fee44a9f7640b5193cdc7ca69622b144 Mon Sep 17 00:00:00 2001 From: alterdekim Date: Sun, 13 Oct 2024 23:05:14 +0300 Subject: [PATCH] modified: src/client.rs --- src/client.rs | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/client.rs b/src/client.rs index ef6be7e..fb8ec71 100644 --- a/src/client.rs +++ b/src/client.rs @@ -47,11 +47,12 @@ pub async fn client_mode(client_config: ClientConfiguration, fd: i32, close_toke let cipher_shared: Arc>> = Arc::new(Mutex::new(None)); + let dr_cc = dr_cancel.clone(); let dev_read_task = tokio::spawn(async move { let mut buf = vec![0; 1400]; // mtu loop { tokio::select! { - _ = dr_cancel.cancelled() => { + _ = dr_cc.cancelled() => { info!("Cancellation token has been thrown dev_read_task"); return; } @@ -68,12 +69,13 @@ pub async fn client_mode(client_config: ClientConfiguration, fd: i32, close_toke let priv_key = BASE64_STANDARD.decode(client_config.client.private_key).unwrap(); let cipher_shared_clone = cipher_shared.clone(); + let sr_cc = sr_cancel.clone(); let sock_read_task = tokio::spawn(async move { let mut buf = vec![0; 4096]; loop { tokio::select! { - _ = sr_cancel.cancelled() => { + _ = sr_cc.cancelled() => { info!("Cancellation token has been thrown sock_read_task"); return; }