add virtual display

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

libs/hbb_common/src/lib.rs

@@ -27,13 +27,14 @@ pub use anyhow::{self, bail};
 pub use futures_util;
 pub mod config;
 pub mod fs;
+pub use lazy_static;
+pub use mac_address;
 pub use rand;
 pub use regex;
 pub use sodiumoxide;
 pub use tokio_socks;
 pub use tokio_socks::IntoTargetAddr;
 pub use tokio_socks::TargetAddr;
-pub use mac_address;
 
 #[cfg(feature = "quic")]
 pub type Stream = quic::Connection;

libs/virtual_display/Cargo.toml

@@ -13,3 +13,4 @@ thiserror = "1.0.30"
 lazy_static = "1.4"
 serde = "1.0"
 serde_derive = "1.0"
+hbb_common = { path = "../hbb_common" }

libs/virtual_display/examples/idd_controller.rs

@@ -1,5 +1,5 @@
 #[cfg(windows)]
-use virtual_display::win10::idd;
+use virtual_display::win10::{idd, DRIVER_INSTALL_PATH};
 
 use std::{
     ffi::{CStr, CString},
@@ -24,9 +24,9 @@ fn prompt_input() -> u8 {
         .unwrap_or(0)
 }
 
-unsafe fn plug_in(index: idd::UINT) {
+unsafe fn plug_in(index: idd::UINT, edid: idd::UINT) {
     println!("Plug in monitor begin");
-    if idd::FALSE == idd::MonitorPlugIn(index, 25) {
+    if idd::FALSE == idd::MonitorPlugIn(index, edid, 25) {
         println!("{}", CStr::from_ptr(idd::GetLastMsg()).to_str().unwrap());
     } else {
         println!("Plug in monitor done");
@@ -58,8 +58,7 @@ unsafe fn plug_out(index: idd::UINT) {
 }
 
 fn main() {
-    let relative_path = "RustDeskIddDriver/RustDeskIddDriver.inf";
-    let abs_path = Path::new(relative_path).canonicalize().unwrap();
+    let abs_path = Path::new(DRIVER_INSTALL_PATH).canonicalize().unwrap();
     let full_inf_path = abs_path.to_str().unwrap();
 
     unsafe {
@@ -121,9 +120,9 @@ fn main() {
                     h_sw_device = invalid_device;
                     println!("Close device done");
                 }
-                '1' => plug_in(0),
-                '2' => plug_in(1),
-                '3' => plug_in(2),
+                '1' => plug_in(0, 0),
+                '2' => plug_in(1, 0),
+                '3' => plug_in(2, 0),
                 '4' => plug_out(0),
                 '5' => plug_out(1),
                 '6' => plug_out(2),
@@ -133,5 +132,7 @@ fn main() {
         if !full_inf_path.is_null() {
             let _ = CString::from_raw(full_inf_path);
         }
+
+        idd::DeviceClose(h_sw_device);
     }
 }

libs/virtual_display/src/lib.rs

@@ -1,2 +1,103 @@
 #[cfg(windows)]
 pub mod win10;
+
+use hbb_common::{bail, lazy_static, ResultType};
+use std::{
+    ffi::{CStr, CString},
+    path::Path,
+    sync::Mutex,
+};
+
+lazy_static::lazy_static! {
+    #[cfg(windows)]
+    static ref H_SW_DEVICE: Mutex<u64> = Mutex::new(0);
+}
+
+pub fn download_driver() -> ResultType<()> {
+    #[cfg(windows)]
+    let _download_url = win10::DRIVER_DOWNLOAD_URL;
+    #[cfg(target_os = "linux")]
+    let _download_url = "";
+
+    // process download and report progress
+
+    Ok(())
+}
+
+pub fn install_update_driver() -> ResultType<()> {
+    #[cfg(windows)]
+    let install_path = win10::DRIVER_INSTALL_PATH;
+    #[cfg(not(windows))]
+    let install_path = "";
+
+    let abs_path = Path::new(install_path).canonicalize()?;
+    if !abs_path.exists() {
+        bail!("{} not exists", install_path)
+    }
+
+    let full_install_path = match abs_path.to_str() {
+        Some(p) => CString::new(p)?.into_raw(),
+        None => bail!("{} not exists", install_path),
+    };
+
+    unsafe {
+        #[cfg(windows)]
+        {
+            let mut reboot_required = win10::idd::FALSE;
+            if win10::idd::InstallUpdate(full_install_path, &mut reboot_required)
+                == win10::idd::FALSE
+            {
+                bail!("{}", CStr::from_ptr(win10::idd::GetLastMsg()).to_str()?);
+            }
+        }
+    }
+
+    Ok(())
+}
+
+pub fn is_device_created() -> bool {
+    #[cfg(windows)]
+    return *H_SW_DEVICE.lock().unwrap() != 0;
+    #[cfg(not(windows))]
+    return false;
+}
+
+#[cfg(windows)]
+pub fn create_device() -> ResultType<()> {
+    unsafe {
+        let mut h_sw_device = *H_SW_DEVICE.lock().unwrap() as win10::idd::HSWDEVICE;
+        if win10::idd::DeviceCreate(&mut h_sw_device) == win10::idd::FALSE {
+            bail!("{}", CStr::from_ptr(win10::idd::GetLastMsg()).to_str()?);
+        } else {
+            *H_SW_DEVICE.lock().unwrap() = h_sw_device as u64;
+            Ok(())
+        }
+    }
+}
+
+#[cfg(windows)]
+pub fn close_device() {
+    unsafe {
+        win10::idd::DeviceClose(*H_SW_DEVICE.lock().unwrap() as win10::idd::HSWDEVICE);
+    }
+}
+
+#[cfg(windows)]
+pub fn plug_in_monitor() -> ResultType<()> {
+    unsafe {
+        if win10::idd::MonitorPlugIn(0, 0, 30) == win10::idd::FALSE {
+            bail!("{}", CStr::from_ptr(win10::idd::GetLastMsg()).to_str()?);
+        }
+        Ok(())
+    }
+}
+
+#[cfg(windows)]
+pub fn plug_out_monitor() -> ResultType<()> {
+    unsafe {
+        if win10::idd::MonitorPlugOut(0) == win10::idd::FALSE {
+            bail!("{}", CStr::from_ptr(win10::idd::GetLastMsg()).to_str()?);
+        }
+        Ok(())
+    }
+}

libs/virtual_display/src/win10/IddController.c

@@ -282,7 +282,7 @@ VOID DeviceClose(HSWDEVICE hSwDevice)
     }
 }
 
-BOOL MonitorPlugIn(UINT index, INT retries)
+BOOL MonitorPlugIn(UINT index, UINT edid, INT retries)
 {
     SetLastMsg("Sucess");
 
@@ -315,6 +315,7 @@ BOOL MonitorPlugIn(UINT index, INT retries)
     DWORD junk = 0;
     CtlPlugIn plugIn;
     plugIn.ConnectorIndex = index;
+    plugIn.MonitorEDID = edid;
     HRESULT hr = CoCreateGuid(&plugIn.ContainerId);
     if (!SUCCEEDED(hr))
     {

libs/virtual_display/src/win10/IddController.h

@@ -70,6 +70,9 @@ VOID DeviceClose(HSWDEVICE hSwDevice);
  * @brief Plug in monitor.
  *
  * @param index [in] Monitor index, should be 0, 1, 2.
+ * @param edid  [in] Monitor edid.
+ *                   0 Modified EDID from Dell S2719DGF
+ *                   1 Modified EDID from Lenovo Y27fA
  * @param retries [in] Retry times. Retry 1 time / sec. 25~30 seconds may be good choices.
  *                     -1 is invalid.
  *                     0 means doing once and no retries.
@@ -83,7 +86,7 @@ VOID DeviceClose(HSWDEVICE hSwDevice);
  *          System need some time to prepare the device.
  *
  */
-BOOL MonitorPlugIn(UINT index, INT retries);
+BOOL MonitorPlugIn(UINT index, UINT edid, INT retries);
 
 /**
  * @brief Plug out monitor.

libs/virtual_display/src/win10/Public.h

@@ -26,8 +26,12 @@
 #define STATUS_ERROR_MONITOR_INVALID_PARAM (3 << 30) + 53
 #define STATUS_ERROR_MONITOR_OOM           (3 << 30) + 54
 
+#define MONITOR_EDID_MOD_DELL_S2719DGF 0
+#define MONITOR_EDID_MOD_LENOVO_Y27fA  1
+
 typedef struct _CtlPlugIn {
     UINT ConnectorIndex;
+    UINT MonitorEDID;
     GUID ContainerId;
 } CtlPlugIn, *PCtlPlugIn;
 

libs/virtual_display/src/win10/idd.rs

@@ -206,7 +206,7 @@ extern "C" {
 
 extern "C" {
     pub fn Uninstall(fullInfPath: LPCTSTR, rebootRequired: PBOOL) -> BOOL;
-    pub fn MonitorPlugIn(index: UINT, retries: INT) -> BOOL;
+    pub fn MonitorPlugIn(index: UINT, edid: UINT, retries: INT) -> BOOL;
     pub fn MonitorPlugOut(index: UINT) -> BOOL;
     pub fn MonitorModesUpdate(index: UINT, modeCount: UINT, modes: PMonitorMode) -> BOOL;
 

libs/virtual_display/src/win10/mod.rs

@@ -1 +1,4 @@
-pub mod idd;
+pub mod idd;
+
+pub const DRIVER_INSTALL_PATH: &str = "RustDeskIddDriver/RustDeskIddDriver.inf";
+pub const DRIVER_DOWNLOAD_URL: &str = "";