win,linux remove desktop wallpaper

Signed-off-by: 21pages <pages21@163.com>

src/server/connection.rs

@@ -6,6 +6,8 @@ use crate::common::update_clipboard;
 #[cfg(all(target_os = "linux", feature = "linux_headless"))]
 #[cfg(not(any(feature = "flatpak", feature = "appimage")))]
 use crate::platform::linux_desktop_manager;
+#[cfg(any(target_os = "windows", target_os = "linux"))]
+use crate::platform::WallPaperRemover;
 #[cfg(windows)]
 use crate::portable_service::client as portable_client;
 use crate::{
@@ -60,8 +62,14 @@ lazy_static::lazy_static! {
     static ref LOGIN_FAILURES: Arc::<Mutex<HashMap<String, (i32, i32, i32)>>> = Default::default();
     static ref SESSIONS: Arc::<Mutex<HashMap<String, Session>>> = Default::default();
     static ref ALIVE_CONNS: Arc::<Mutex<Vec<i32>>> = Default::default();
+    static ref AUTHED_CONNS: Arc::<Mutex<Vec<(i32, AuthConnType)>>> = Default::default();
     static ref SWITCH_SIDES_UUID: Arc::<Mutex<HashMap<String, (Instant, uuid::Uuid)>>> = Default::default();
 }
+
+#[cfg(any(target_os = "windows", target_os = "linux"))]
+lazy_static::lazy_static! {
+    static ref WALLPAPER_REMOVER: Arc<Mutex<Option<WallPaperRemover>>> = Default::default();
+}
 pub static CLICK_TIME: AtomicI64 = AtomicI64::new(0);
 #[cfg(not(any(target_os = "android", target_os = "ios")))]
 pub static MOUSE_MOVE_TIME: AtomicI64 = AtomicI64::new(0);
@@ -143,6 +151,13 @@ struct StartCmIpcPara {
     tx_cm_stream_ready: mpsc::Sender<()>,
 }
 
+#[derive(Debug, Copy, Clone, Eq, PartialEq)]
+pub enum AuthConnType {
+    Remote,
+    FileTransfer,
+    PortForward,
+}
+
 pub struct Connection {
     inner: ConnInner,
     stream: super::Stream,
@@ -205,6 +220,7 @@ pub struct Connection {
     #[cfg(not(any(target_os = "android", target_os = "ios")))]
     start_cm_ipc_para: Option<StartCmIpcPara>,
     auto_disconnect_timer: Option<(Instant, u64)>,
+    authed_conn_id: Option<self::raii::AuthedConnID>,
 }
 
 impl ConnInner {
@@ -345,6 +361,7 @@ impl Connection {
                 tx_cm_stream_ready,
             }),
             auto_disconnect_timer: None,
+            authed_conn_id: None,
         };
         let addr = hbb_common::try_into_v4(addr);
         if !conn.on_open(addr).await {
@@ -976,13 +993,17 @@ impl Connection {
         if self.authorized {
             return;
         }
-        let conn_type = if self.file_transfer.is_some() {
-            1
+        let (conn_type, auth_conn_type) = if self.file_transfer.is_some() {
+            (1, AuthConnType::FileTransfer)
         } else if self.port_forward_socket.is_some() {
-            2
+            (2, AuthConnType::PortForward)
         } else {
-            0
+            (0, AuthConnType::Remote)
         };
+        self.authed_conn_id = Some(self::raii::AuthedConnID::new(
+            self.inner.id(),
+            auth_conn_type,
+        ));
         self.post_conn_audit(
             json!({"peer": ((&self.lr.my_id, &self.lr.my_name)), "type": conn_type}),
         );
@@ -1117,6 +1138,7 @@ impl Connection {
                     *super::video_service::LAST_SYNC_DISPLAYS.write().unwrap() = displays;
                 }
             }
+            Self::on_remote_authorized();
         }
         let mut msg_out = Message::new();
         msg_out.set_login_response(res);
@@ -1155,6 +1177,29 @@ impl Connection {
         }
     }
 
+    fn on_remote_authorized() {
+        use std::sync::Once;
+        static ONCE: Once = Once::new();
+        #[cfg(any(target_os = "windows", target_os = "linux"))]
+        if !Config::get_option("allow-remove-wallpaper").is_empty() {
+            // multi connections set once
+            let mut wallpaper = WALLPAPER_REMOVER.lock().unwrap();
+            if wallpaper.is_none() {
+                match crate::platform::WallPaperRemover::new() {
+                    Ok(remover) => {
+                        *wallpaper = Some(remover);
+                        ONCE.call_once(|| {
+                            shutdown_hooks::add_shutdown_hook(shutdown_hook);
+                        });
+                    }
+                    Err(e) => {
+                        log::info!("create wallpaper remover failed:{:?}", e);
+                    }
+                }
+            }
+        }
+    }
+
     fn peer_keyboard_enabled(&self) -> bool {
         self.keyboard && !self.disable_keyboard
     }
@@ -2734,6 +2779,11 @@ impl LinuxHeadlessHandle {
     }
 }
 
+#[cfg(any(target_os = "windows", target_os = "linux"))]
+extern "C" fn shutdown_hook() {
+    *WALLPAPER_REMOVER.lock().unwrap() = None;
+}
+
 mod raii {
     use super::*;
     pub struct ConnectionID(i32);
@@ -2767,4 +2817,26 @@ mod raii {
                 .on_connection_close(self.0);
         }
     }
+
+    pub struct AuthedConnID(i32);
+
+    impl AuthedConnID {
+        pub fn new(id: i32, conn_type: AuthConnType) -> Self {
+            AUTHED_CONNS.lock().unwrap().push((id, conn_type));
+            Self(id)
+        }
+    }
+
+    impl Drop for AuthedConnID {
+        fn drop(&mut self) {
+            let mut lock = AUTHED_CONNS.lock().unwrap();
+            lock.retain(|&c| c.0 != self.0);
+            if lock.iter().filter(|c| c.1 == AuthConnType::Remote).count() == 0 {
+                #[cfg(any(target_os = "windows", target_os = "linux"))]
+                {
+                    *WALLPAPER_REMOVER.lock().unwrap() = None;
+                }
+            }
+        }
+    }
 }