rustdesk/libs/virtual_display/examples/idd_controller.rs

#[cfg(windows)]
use virtual_display::win10::{idd, DRIVER_INSTALL_PATH};

use std::{
    ffi::{CStr, CString},
    io::{self, Read},
    path::Path,
};

fn prompt_input() -> u8 {
    println!("Press  key                  execute:");
    println!("       1. 'x'               1. exit");
    println!("       2. 'i'               2. install or update driver");
    println!("       3. 'u'               3. uninstall driver");
    println!("       4. 'c'               4. create device");
    println!("       5. 'd'               5. destroy device");
    println!("       6. '1','2','3'       6. plug in monitor 0,1,2");
    println!("       7. '4','5','6'       7. plug out monitor 0,1,2");

    io::stdin()
        .bytes()
        .next()
        .and_then(|result| result.ok())
        .unwrap_or(0)
}

unsafe fn plug_in(index: idd::UINT, edid: idd::UINT) {
    println!("Plug in monitor begin");
    if idd::FALSE == idd::MonitorPlugIn(index, edid, 25) {
        println!("{}", CStr::from_ptr(idd::GetLastMsg()).to_str().unwrap());
    } else {
        println!("Plug in monitor done");

        let mut modes: Vec<idd::MonitorMode> = Vec::new();
        modes.push(idd::MonitorMode {
            width: 1920 as idd::DWORD,
            height: 1080 as idd::DWORD,
            sync: 60 as idd::DWORD,
        });
        modes.push(idd::MonitorMode {
            width: 1024 as idd::DWORD,
            height: 768 as idd::DWORD,
            sync: 60 as idd::DWORD,
        });
        if idd::FALSE == idd::MonitorModesUpdate(index, modes.len() as u32, modes.as_mut_ptr()) {
            println!("{}", CStr::from_ptr(idd::GetLastMsg()).to_str().unwrap());
        }
    }
}

unsafe fn plug_out(index: idd::UINT) {
    println!("Plug out monitor begin");
    if idd::FALSE == idd::MonitorPlugOut(index) {
        println!("{}", CStr::from_ptr(idd::GetLastMsg()).to_str().unwrap());
    } else {
        println!("Plug out monitor done");
    }
}

fn main() {
    let abs_path = Path::new(DRIVER_INSTALL_PATH).canonicalize().unwrap();
    let full_inf_path = abs_path.to_str().unwrap();

    unsafe {
        let invalid_device = 0 as idd::HSWDEVICE;
        let mut h_sw_device = invalid_device;
        let full_inf_path = CString::new(full_inf_path).unwrap().into_raw();
        loop {
            match prompt_input() as char {
                'x' => break,
                'i' => {
                    println!("Install or update driver begin");
                    let mut reboot_required = idd::FALSE;
                    if idd::InstallUpdate(full_inf_path, &mut reboot_required) == idd::FALSE {
                        println!("{}", CStr::from_ptr(idd::GetLastMsg()).to_str().unwrap());
                    } else {
                        println!(
                            "Install or update driver done, reboot is {} required",
                            if reboot_required == idd::FALSE {
                                "not"
                            } else {
                                ""
                            }
                        );
                    }
                }
                'u' => {
                    println!("Uninstall driver begin");
                    let mut reboot_required = idd::FALSE;
                    if idd::Uninstall(full_inf_path, &mut reboot_required) == idd::FALSE {
                        println!("{}", CStr::from_ptr(idd::GetLastMsg()).to_str().unwrap());
                    } else {
                        println!(
                            "Uninstall driver done, reboot is {} required",
                            if reboot_required == idd::FALSE {
                                "not"
                            } else {
                                ""
                            }
                        );
                    }
                }
                'c' => {
                    println!("Create device begin");
                    if h_sw_device != invalid_device {
                        println!("Device created before");
                        continue;
                    }
                    if idd::FALSE == idd::DeviceCreate(&mut h_sw_device) {
                        println!("{}", CStr::from_ptr(idd::GetLastMsg()).to_str().unwrap());
                        idd::DeviceClose(h_sw_device);
                        h_sw_device = invalid_device;
                    } else {
                        println!("Create device done");
                    }
                }
                'd' => {
                    println!("Close device begin");
                    idd::DeviceClose(h_sw_device);
                    h_sw_device = invalid_device;
                    println!("Close device done");
                }
                '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),
                _ => {}
            }
        }
        if !full_inf_path.is_null() {
            let _ = CString::from_raw(full_inf_path);
        }

        idd::DeviceClose(h_sw_device);
    }
}