lan_discovery_WOL: Win10 test done

Signed-off-by: fufesou <shuanglongchen@yeah.net>

src/rendezvous_mediator.rs

@@ -700,10 +700,9 @@ fn wait_response(
                             if mac != p.mac {
                                 allow_err!(tx.send(config::DiscoveryPeer {
                                     id: p.id.clone(),
-                                    mac_ips: HashMap::from([(
-                                        p.mac.clone(),
-                                        addr.ip().to_string()
-                                    )]),
+                                    ip_mac: HashMap::from([
+                                        (addr.ip().to_string(), p.mac.clone(),)
+                                    ]),
                                     username: p.username.clone(),
                                     hostname: p.hostname.clone(),
                                     platform: p.platform.clone(),
@@ -744,7 +743,6 @@ async fn handle_received_peers(mut rx: UnboundedReceiver<config::DiscoveryPeer>)
         peer.online = false;
     });
 
-    // handle received peers
     let mut response_set = HashSet::new();
     let mut last_write_time = Instant::now() - std::time::Duration::from_secs(4);
     loop {
@@ -753,20 +751,24 @@ async fn handle_received_peers(mut rx: UnboundedReceiver<config::DiscoveryPeer>)
                 Some(peer) => {
                     let in_response_set = !response_set.insert(peer.id.clone());
                     let mut pre_found = false;
+                    // Try find and update peer
                     for peer1 in &mut peers {
                         if peer1.is_same_peer(&peer) {
                             if in_response_set {
-                                peer1.mac_ips.extend(peer.mac_ips.clone());
+                                // Merge ip_mac and update other infos
+                                peer1.ip_mac.extend(peer.ip_mac.clone());
                                 peer1.hostname = peer.hostname.clone();
                                 peer1.platform = peer.platform.clone();
                                 peer1.online = true;
                             } else {
+                                // Update all peer infos
                                 *peer1 = peer.clone();
                             }
                             pre_found = true;
                             break
                         }
                     }
+                    // Push if not found
                     if !pre_found {
                         peers.push(peer);
                     }
@@ -796,3 +798,31 @@ pub async fn discover() -> ResultType<()> {
     log::info!("discover ping done");
     Ok(())
 }
+
+pub fn send_wol(id: String) {
+    let interfaces = default_net::get_interfaces();
+    for peer in &config::LanPeers::load().peers {
+        if peer.id == id {
+            for (ip, mac) in peer.ip_mac.iter() {
+                if let Ok(mac_addr) = mac.parse() {
+                    if let Ok(IpAddr::V4(ip)) = ip.parse() {
+                        for interface in &interfaces {
+                            for ipv4 in &interface.ipv4 {
+                                if (u32::from(ipv4.addr) & u32::from(ipv4.netmask))
+                                    == (u32::from(ip) & u32::from(ipv4.netmask))
+                                {
+                                    allow_err!(wol::send_wol(
+                                        mac_addr,
+                                        None,
+                                        Some(IpAddr::V4(ipv4.addr))
+                                    ));
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+            break;
+        }
+    }
+}

src/ui.rs

@@ -547,6 +547,10 @@ impl UI {
         config::LanPeers::store(&peers);
     }
 
+    fn send_wol(&mut self, id: String) {
+        crate::rendezvous_mediator::send_wol(id)
+    }
+
     fn new_remote(&mut self, id: String, remote_type: String) {
         let mut lock = self.0.lock().unwrap();
         let args = vec![format!("--{}", remote_type), id.clone()];
@@ -786,6 +790,7 @@ impl sciter::EventHandler for UI {
         fn closing(i32, i32, i32, i32);
         fn get_size();
         fn new_remote(String, bool);
+        fn send_wol(String);
         fn remove_peer(String);
         fn remove_discovered(String);
         fn get_connect_status();

src/ui/ab.tis

@@ -318,10 +318,11 @@ class SessionList: Reactor.Component {
                     <li #tunnel>{translate('TCP Tunneling')}</li>
                     {false && !handler.using_public_server() && <li #force-always-relay><span>{svg_checkmark}</span>{translate('Always connect via relay')}</li>}
                     <li #rdp>RDP<EditRdpPort /></li>
+                    <li #wol>{translate('WOL')}</li>
                     <div .separator />
                     {this.type != "lan" && <li #rename>{translate('Rename')}</li>}
                     {this.type != "fav" && <li #remove>{translate('Remove')}</li>}
-                    {is_win && this.type != "lan" && <li #shortcut>{translate('Create Desktop Shortcut')}</li>}
+                    {is_win && <li #shortcut>{translate('Create Desktop Shortcut')}</li>}
                     <li #forget-password>{translate('Unremember Password')}</li>
                     {(!this.type || this.type == "fav") && <li #add-fav>{translate('Add to Favorites')}</li>}
                     {(!this.type || this.type == "fav") && <li #remove-fav>{translate('Remove from Favorites')}</li>}
@@ -419,6 +420,8 @@ class SessionList: Reactor.Component {
             createNewConnect(id, "connect");
         } else if (action == "transfer") {
             createNewConnect(id, "file-transfer");
+        } else if (action == "wol") {
+            handler.send_wol(id);
         } else if (action == "remove") {
             if (this.type == "ab") {
                 for (var i = 0; i < ab.peers.length; ++i) {