TheLanc3/rustdesk
1
0
Fork 0
mirror of https://github.com/rustdesk/rustdesk.git synced 2026-03-13 16:21:06 +03:00
Code Issues Packages Projects Releases Wiki Activity

lan discovery will be done soon

Browse Source 
Signed-off-by: fufesou <shuanglongchen@yeah.net>
This commit is contained in:
fufesou
2021-12-19 19:58:08 +08:00
parent bcbe9ccbe5
commit 5682b088de
11 changed files with 254 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
libs/socket_cs/examples/discovery.rs
View File

@@ -1,13 +1,11 @@
use hbb_common::{
base_proto::PeerInfo,
discovery_proto::Discovery as DiscoveryProto,
discovery_proto::{Discovery as DiscoveryProto, DiscoveryBack as DiscoveryBackProto},
env_logger::*,
log, protobuf,
tokio::{self, sync::Notify},
log, protobuf, tokio,
};
use socket_cs::{discovery::*, udp::*};
use std::env;
use std::sync::Arc;
async fn lan_discover(port: u16, port_back: u16) {
let peer = PeerInfo {
@@ -27,21 +25,21 @@ async fn lan_discover(port: u16, port_back: u16) {
}
async fn listen_discovery_back(port: u16) {
fn proc_peer(peer: PeerInfo) {
fn proc_peer(info: DiscoveryBackProto) {
log::info!(
"peer recived, username: {}, hostname: {}",
peer.username,
peer.hostname
"peer recived, id: {}, username: {}, hostname: {}",
info.id,
info.peer.as_ref().unwrap().username,
info.peer.as_ref().unwrap().hostname
);
}
let exit_notify = Notify::new();
let handlers = UdpHandlers::new().handle(
CMD_DISCOVERY_BACK.as_bytes().to_vec(),
Box::new(HandlerDiscoveryBack::new(proc_peer)),
);
let server = Server::new(port, Arc::new(exit_notify));
let mut server = Server::create(port).unwrap();
server.start(handlers).await.unwrap();
loop {
@@ -50,19 +48,22 @@ async fn listen_discovery_back(port: u16) {
}
async fn listen_discovery(port: u16) {
let peer = PeerInfo {
username: "server username".to_owned(),
hostname: "server hostname".to_owned(),
let info = DiscoveryBackProto {
id: "server id".to_owned(),
peer: protobuf::MessageField::from_option(Some(PeerInfo {
username: "server username".to_owned(),
hostname: "server hostname".to_owned(),
..Default::default()
})),
..Default::default()
};
let exit_notify = Notify::new();
let handlers = UdpHandlers::new().handle(
CMD_DISCOVERY.as_bytes().to_vec(),
Box::new(HandlerDiscovery::new(peer)),
Box::new(HandlerDiscovery::new(Some(|| true), info)),
);
let server = Server::new(port, Arc::new(exit_notify));
let mut server = Server::create(port).unwrap();
server.start(handlers).await.unwrap();
loop {
std::thread::sleep(std::time::Duration::from_millis(1000));