TheLanc3/rustdesk
1
0
Fork 0
mirror of https://github.com/rustdesk/rustdesk.git synced 2026-03-31 00:51:04 +03:00
Code Issues Packages Projects Releases Wiki Activity

opt update of direct/direct_failures (#12627)

Browse Source 
Signed-off-by: 21pages <sunboeasy@gmail.com>
This commit is contained in:
21pages
2025-08-12 17:48:20 +08:00
committed by GitHub
parent d6d44be1b7
commit e7909a0dbd
1 changed files with 23 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/client.rs
View File

@@ -204,7 +204,7 @@ impl Client {
debug_assert!(peer == interface.get_id()); debug_assert!(peer == interface.get_id());
interface.update_direct(None); interface.update_direct(None);
interface.update_received(false); interface.update_received(false);
match Self::_start(peer, key, token, conn_type, interface).await { match Self::_start(peer, key, token, conn_type, interface.clone()).await {
Err(err) => { Err(err) => {
let err_str = err.to_string(); let err_str = err.to_string();
if err_str.starts_with("Failed") { if err_str.starts_with("Failed") {
@@ -213,7 +213,16 @@ impl Client {
return Err(err); return Err(err);
} }
} }
Ok(x) => Ok(x), Ok(x) => {
let direct_failures = interface.get_lch().read().unwrap().direct_failures;
let direct = x.0 .1;
if !interface.is_force_relay() && (direct_failures == 0) != direct {
let n = if direct { 0 } else { 1 };
log::info!("direct_failures updated to {}", n);
interface.get_lch().write().unwrap().set_direct_failure(n);
}
Ok(x)
}
} }
} }
@@ -688,7 +697,6 @@ impl Client {
}; };
let mut direct = !conn.is_err(); let mut direct = !conn.is_err();
interface.update_direct(Some(direct));
if interface.is_force_relay() || conn.is_err() { if interface.is_force_relay() || conn.is_err() {
if !relay_server.is_empty() { if !relay_server.is_empty() {
conn = Self::request_relay( conn = Self::request_relay(
@@ -701,8 +709,9 @@ impl Client {
conn_type, conn_type,
) )
.await; .await;
interface.update_direct(Some(false));
if let Err(e) = conn { if let Err(e) = conn {
// this direct is mainly used by on_establish_connection_error, so we update it here before bail
interface.update_direct(Some(false));
bail!("Failed to connect via relay server: {}", e); bail!("Failed to connect via relay server: {}", e);
} }
typ = "Relay"; typ = "Relay";
@@ -711,19 +720,22 @@ impl Client {
bail!("Failed to make direct connection to remote desktop"); bail!("Failed to make direct connection to remote desktop");
} }
} }
if !relay_server.is_empty() && (direct_failures == 0) != direct {
let n = if direct { 0 } else { 1 };
log::info!("direct_failures updated to {}", n);
interface.get_lch().write().unwrap().set_direct_failure(n);
}
let mut conn = conn?; let mut conn = conn?;
log::info!( log::info!(
"{:?} used to establish {typ} connection with {} punch", "{:?} used to establish {typ} connection with {} punch",
start.elapsed(), start.elapsed(),
punch_type punch_type
); );
let pk = Self::secure_connection(peer_id, signed_id_pk, key, &mut conn).await?; let res = Self::secure_connection(peer_id, signed_id_pk, key, &mut conn).await;
log::info!("{} punch secure_connection ok", punch_type); let pk: Option<Vec<u8>> = match res {
Ok(pk) => pk,
Err(e) => {
// this direct is mainly used by on_establish_connection_error, so we update it here before bail
interface.update_direct(Some(direct));
bail!(e);
}
};
log::debug!("{} punch secure_connection ok", punch_type);
Ok((conn, direct, pk, kcp, typ)) Ok((conn, direct, pk, kcp, typ))
} }