From 16ca7be2ea4bd6bd6c84961462483f0eaa76de96 Mon Sep 17 00:00:00 2001
From: fufesou <shuanglongchen@yeah.net>
Date: Wed, 24 May 2023 19:54:44 +0800
Subject: [PATCH] copy runtime broker on portable

Signed-off-by: fufesou <shuanglongchen@yeah.net>
---
 libs/portable/src/main.rs | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/libs/portable/src/main.rs b/libs/portable/src/main.rs
index 13dd0c3dc..ca335363f 100644
--- a/libs/portable/src/main.rs
+++ b/libs/portable/src/main.rs
@@ -30,6 +30,8 @@ fn setup(reader: BinaryReader, dir: Option<PathBuf>, clear: bool) -> Option<Path
     for file in reader.files.iter() {
         file.write_to_file(&dir);
     }
+    #[cfg(windows)]
+    windows::copy_runtime_broker(&dir);
     #[cfg(linux)]
     reader.configure_permission(&dir);
     Some(dir.join(&reader.exe))
@@ -80,3 +82,30 @@ fn main() {
         execute(exe, args);
     }
 }
+
+mod windows {
+    use std::{fs, path::PathBuf, process::Command};
+
+    // Used for privacy mode(magnifier impl).
+    pub const RUNTIME_BROKER_EXE: &'static str = "C:\\Windows\\System32\\RuntimeBroker.exe";
+    pub const WIN_MAG_INJECTED_PROCESS_EXE: &'static str = "RuntimeBroker_rustdesk.exe";
+
+    pub(super) fn copy_runtime_broker(dir: &PathBuf) {
+        let src = RUNTIME_BROKER_EXE;
+        let tgt = WIN_MAG_INJECTED_PROCESS_EXE;
+        let target_file = dir.join(tgt);
+        if target_file.exists() {
+            if let (Ok(src_file), Ok(tgt_file)) = (fs::read(src), fs::read(&target_file)) {
+                let src_md5 = format!("{:x}", md5::compute(&src_file));
+                let tgt_md5 = format!("{:x}", md5::compute(&tgt_file));
+                if src_md5 == tgt_md5 {
+                    return;
+                }
+            }
+        }
+        let _allow_err = Command::new("taskkill")
+            .args(&["/F", "/IM", "RuntimeBroker_rustdesk.exe"])
+            .output();
+        let _allow_err = std::fs::copy(src, &format!("{}\\{}", dir.to_string_lossy(), tgt));
+    }
+}