alterwain/Frida-android-native
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modified: src/client.rs
Some checks failed
gitea/Frida-android-native/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Michael Wain 2024-10-13 21:02:36 +03:00
parent 22c918b782
commit 055e127475
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/client.rs
View File

@ -46,12 +46,13 @@ pub async fn client_mode(client_config: ClientConfiguration, fd: i32, close_toke
let cipher_shared: Arc<Mutex<Option<x25519_dalek::SharedSecret>>> = Arc::new(Mutex::new(None)); let cipher_shared: Arc<Mutex<Option<x25519_dalek::SharedSecret>>> = Arc::new(Mutex::new(None));
let cancel_dr = close_token.clone(); let cancel_dr = subtasks_quit.clone();
let dev_read_task = tokio::spawn(async move { let dev_read_task = tokio::spawn(async move {
let mut buf = vec![0; 1400]; // mtu let mut buf = vec![0; 1400]; // mtu
loop { loop {
let sf = cancel_dr.lock().unwrap();
tokio::select! { tokio::select! {
_ = subtasks_quit.lock().unwrap().cancelled() => { _ = sf.cancelled() => {
info!("Cancellation token has been thrown dev_read_task"); info!("Cancellation token has been thrown dev_read_task");
return; return;
} }
@ -62,19 +63,21 @@ pub async fn client_mode(client_config: ClientConfiguration, fd: i32, close_toke
} }
} }
}; };
drop(sf);
} }
}); });
let priv_key = BASE64_STANDARD.decode(client_config.client.private_key).unwrap(); let priv_key = BASE64_STANDARD.decode(client_config.client.private_key).unwrap();
let cipher_shared_clone = cipher_shared.clone(); let cipher_shared_clone = cipher_shared.clone();
let cancel_sr = close_token.clone(); let cancel_sr = subtasks_quit.clone();
let sock_read_task = tokio::spawn(async move { let sock_read_task = tokio::spawn(async move {
let mut buf = vec![0; 4096]; let mut buf = vec![0; 4096];
loop { loop {
let sf = cancel_sr.lock().unwrap();
tokio::select! { tokio::select! {
_ = subtasks_quit.lock().unwrap().cancelled() => { _ = sf.cancelled() => {
info!("Cancellation token has been thrown sock_read_task"); info!("Cancellation token has been thrown sock_read_task");
return; return;
} }
@ -121,6 +124,7 @@ pub async fn client_mode(client_config: ClientConfiguration, fd: i32, close_toke
} }
} }
}; };
drop(sf);
} }
}); });