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

clap

Browse Source
This commit is contained in:
open-trade
2020-09-18 13:04:39 +08:00
parent 0a1fbadb4f
commit ba9862cd7f
4 changed files with 82 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/rendezvous_server.rs
View File

@@ -125,16 +125,18 @@ pub struct RendezvousServer {
tcp_punch: Arc<Mutex<HashMap<SocketAddr, Sink>>>,
pm: PeerMap,
tx: Sender,
relay_server: String,
}
impl RendezvousServer {
pub async fn start(addr: &str) -> ResultType<()> {
pub async fn start(addr: &str, relay_server: String) -> ResultType<()> {
let mut socket = FramedSocket::new(addr).await?;
let (tx, mut rx) = mpsc::unbounded_channel::<(RendezvousMessage, SocketAddr)>();
let mut rs = Self {
tcp_punch: Arc::new(Mutex::new(HashMap::new())),
pm: PeerMap::new()?,
tx: tx.clone(),
relay_server,
};
let mut listener = new_listener(addr, true).await?;
loop {
@@ -347,7 +349,10 @@ impl RendezvousServer {
Some(peer) => peer.pk,
_ => Vec::new(),
};
let relay_server = phs.relay_server;
let mut relay_server = phs.relay_server;
if relay_server.is_empty() {
relay_server = self.relay_server.clone();
}
msg_out.set_punch_hole_response(PunchHoleResponse {
socket_addr: AddrMangle::encode(addr),
pk,