rustdesk/rustdesk-server
1
0
Fork 0
mirror of https://github.com/rustdesk/rustdesk-server.git synced 2026-06-18 14:51:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

relay for local addr in case in different sub net

Browse Source
This commit is contained in:
open-trade
2020-09-21 15:13:14 +08:00
parent a5b4d8027f
commit f3425044a6
2 changed files with 26 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/hbb_common
+1 -1
Submodule libs/hbb_common updated: afce320583...e4313f5f0f
src/rendezvous_server.rs
+10 -10
View File
@@ -196,7 +196,7 @@ impl RendezvousServer {
break; break;
} }
Some(rendezvous_message::Union::local_addr(la)) => { Some(rendezvous_message::Union::local_addr(la)) => {
allow_err!(rs.handle_local_addr(&la, addr, None).await); allow_err!(rs.handle_local_addr(la, addr, None).await);
break; break;
} }
_ => { _ => {
@@ -231,13 +231,11 @@ impl RendezvousServer {
self.update_addr(rp.id, addr, socket).await?; self.update_addr(rp.id, addr, socket).await?;
if self.serial > rp.serial { if self.serial > rp.serial {
let mut msg_out = RendezvousMessage::new(); let mut msg_out = RendezvousMessage::new();
let mut mi = MiscInfo::new(); msg_out.set_configure_update(ConfigUpdate {
mi.set_configure_update(ConfigUpdate {
serial: self.serial, serial: self.serial,
rendezvous_servers: self.rendezvous_servers.clone(), rendezvous_servers: self.rendezvous_servers.clone(),
..Default::default() ..Default::default()
}); });
msg_out.set_misc_info(mi);
socket.send(&msg_out, addr).await?; socket.send(&msg_out, addr).await?;
} }
} }
@@ -286,10 +284,9 @@ impl RendezvousServer {
self.handle_hole_sent(phs, addr, Some(socket)).await?; self.handle_hole_sent(phs, addr, Some(socket)).await?;
} }
Some(rendezvous_message::Union::local_addr(la)) => { Some(rendezvous_message::Union::local_addr(la)) => {
self.handle_local_addr(&la, addr, Some(socket)).await?; self.handle_local_addr(la, addr, Some(socket)).await?;
} }
Some(rendezvous_message::Union::misc_info(mi)) => match mi.union { Some(rendezvous_message::Union::configure_update(mut cu)) => {
Some(misc_info::Union::configure_update(mut cu)) => {
if addr.ip() == std::net::IpAddr::V4(std::net::Ipv4Addr::new(127, 0, 0, 1)) if addr.ip() == std::net::IpAddr::V4(std::net::Ipv4Addr::new(127, 0, 0, 1))
&& cu.serial > self.serial && cu.serial > self.serial
{ {
@@ -307,8 +304,6 @@ impl RendezvousServer {
} }
} }
_ => {} _ => {}
},
_ => {}
} }
} }
Ok(()) Ok(())
@@ -408,7 +403,7 @@ impl RendezvousServer {
#[inline] #[inline]
async fn handle_local_addr<'a>( async fn handle_local_addr<'a>(
&mut self, &mut self,
la: &LocalAddr, la: LocalAddr,
addr: SocketAddr, addr: SocketAddr,
socket: Option<&'a mut FramedSocket>, socket: Option<&'a mut FramedSocket>,
) -> ResultType<()> { ) -> ResultType<()> {
@@ -421,8 +416,13 @@ impl RendezvousServer {
&addr &addr
); );
let mut msg_out = RendezvousMessage::new(); let mut msg_out = RendezvousMessage::new();
let mut relay_server = la.relay_server;
if relay_server.is_empty() {
relay_server = self.relay_server.clone();
}
msg_out.set_punch_hole_response(PunchHoleResponse { msg_out.set_punch_hole_response(PunchHoleResponse {
socket_addr: la.local_addr.clone(), socket_addr: la.local_addr.clone(),
relay_server,
..Default::default() ..Default::default()
}); });
if let Some(socket) = socket { if let Some(socket) = socket {