hardcoresushi/AIRA
1
0
Fork 0
Code Issues Pull Requests Projects Releases 3 Wiki Activity

Switching from pnet to if-addrs for Windows support

Browse Source
This commit is contained in:
Matéo Duparc 2021-05-11 14:30:53 +02:00
parent 24231ba043
commit 79a3ad1790
Signed by: hardcoresushi
GPG Key ID: 007F84120107191E
4 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Cargo.toml
View File

@ -33,7 +33,7 @@ uuid = {version = "0.8", features = ["v4"]}
webbrowser = "0.5.5" webbrowser = "0.5.5"
libmdns = "0.6" #mDNS advertiser libmdns = "0.6" #mDNS advertiser
multicast_dns = "0.5" #mDNS browser multicast_dns = "0.5" #mDNS browser
pnet_datalink = "0.27.2" if-addrs = "0.6"
base64 = "0.13.0" base64 = "0.13.0"
scrypt = "0.7.0" scrypt = "0.7.0"
zeroize = "1.2.0" zeroize = "1.2.0"

5
src/discovery.rs
View File

@ -1,12 +1,13 @@
use std::{net::IpAddr, io}; use std::{net::IpAddr, io};
use tokio::runtime::Handle;
use libmdns::{Responder, Service}; use libmdns::{Responder, Service};
use multicast_dns::discovery::{DiscoveryManager, DiscoveryListeners, ResolveListeners}; use multicast_dns::discovery::{DiscoveryManager, DiscoveryListeners, ResolveListeners};
use crate::{constants, print_error}; use crate::{constants, print_error};
const SERVICE_TYPE: &str = "_aira._tcp"; const SERVICE_TYPE: &str = "_aira._tcp";
pub fn advertise_me() -> io::Result<Service> { pub async fn advertise_me() -> io::Result<Service> {
Ok(Responder::new()?.register( Ok(Responder::spawn(&Handle::current())?.register(
SERVICE_TYPE.to_string(), SERVICE_TYPE.to_string(),
"AIRA Node".to_string(), "AIRA Node".to_string(),
constants::PORT, constants::PORT,

9
src/main.rs
View File

@ -117,12 +117,13 @@ async fn websocket_worker(mut ui_connection: UiConnection, global_vars: Arc<RwLo
ui_connection.load_msgs(&msgs.0, &msgs.1); ui_connection.load_msgs(&msgs.0, &msgs.1);
}); });
let mut ips = Vec::new(); let mut ips = Vec::new();
for interface in pnet_datalink::interfaces() { match if_addrs::get_if_addrs() {
if !interface.is_loopback() { Ok(ifaces) => for iface in ifaces {
for ip in interface.ips { if !iface.is_loopback() {
ips.push(ip.ip()); ips.push(iface.ip());
} }
} }
Err(e) => print_error!(e)
} }
ui_connection.set_local_ips(ips); ui_connection.set_local_ips(ips);
discover_peers(session_manager.clone()); discover_peers(session_manager.clone());

2
src/session_manager/mod.rs
View File

@ -528,7 +528,7 @@ impl SessionManager {
let server_v4 = TcpListener::bind(SocketAddr::new("0.0.0.0".parse().unwrap(), constants::PORT)).await?; let server_v4 = TcpListener::bind(SocketAddr::new("0.0.0.0".parse().unwrap(), constants::PORT)).await?;
let (sender, mut receiver) = mpsc::channel(1); let (sender, mut receiver) = mpsc::channel(1);
*session_manager.listener_stop_signal.lock().unwrap() = Some(sender); *session_manager.listener_stop_signal.lock().unwrap() = Some(sender);
match discovery::advertise_me() { match discovery::advertise_me().await {
Ok(service) => *session_manager.mdns_service.lock().unwrap() = Some(service), Ok(service) => *session_manager.mdns_service.lock().unwrap() = Some(service),
Err(e) => { Err(e) => {
print_error!("{}: you won't be discoverable by other peers.", e); print_error!("{}: you won't be discoverable by other peers.", e);