Mirrors/rustdesk-server
1
0
Fork 0
mirror of https://github.com/rustdesk/rustdesk-server.git synced 2026-03-06 20:13:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
open-trade
2020-09-27 13:53:29 +08:00
parent 36c248fdb9
commit b34049b1bb
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/rendezvous_server.rs
View File

@@ -176,18 +176,16 @@ impl RendezvousServer {
if let Ok(msg_in) = RendezvousMessage::parse_from_bytes(&bytes) { if let Ok(msg_in) = RendezvousMessage::parse_from_bytes(&bytes) {
match msg_in.union { match msg_in.union {
Some(rendezvous_message::Union::punch_hole_request(ph)) => { Some(rendezvous_message::Union::punch_hole_request(ph)) => {
// there maybe several attempt, so sender can be none
if let Some(sender) = sender.take() { if let Some(sender) = sender.take() {
tcp_punch.lock().unwrap().insert(addr, sender); tcp_punch.lock().unwrap().insert(addr, sender);
} else {
break;
} }
allow_err!(rs.handle_tcp_punch_hole_request(addr, ph).await); allow_err!(rs.handle_tcp_punch_hole_request(addr, ph).await);
} }
Some(rendezvous_message::Union::request_relay(mut rf)) => { Some(rendezvous_message::Union::request_relay(mut rf)) => {
// there maybe several attempt, so sender can be none
if let Some(sender) = sender.take() { if let Some(sender) = sender.take() {
tcp_punch.lock().unwrap().insert(addr, sender); tcp_punch.lock().unwrap().insert(addr, sender);
} else {
break;
} }
if let Some(peer) = rs.pm.map.read().unwrap().get(&rf.id).map(|x| x.clone()) { if let Some(peer) = rs.pm.map.read().unwrap().get(&rf.id).map(|x| x.clone()) {
let mut msg_out = RendezvousMessage::new(); let mut msg_out = RendezvousMessage::new();