From 2d456fd1fc1aab0c323f1985a2f3a52a835afaa3 Mon Sep 17 00:00:00 2001 From: fufesou Date: Sun, 26 Nov 2023 12:30:14 +0800 Subject: [PATCH] --install-idd with cert Signed-off-by: fufesou --- src/core_main.rs | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/src/core_main.rs b/src/core_main.rs index 53957fe45..6c588a48a 100644 --- a/src/core_main.rs +++ b/src/core_main.rs @@ -219,15 +219,18 @@ pub fn core_main() -> Option> { #[cfg(windows)] hbb_common::allow_err!(crate::platform::windows::uninstall_cert()); return None; - } else if args[0] == "--install-update-idd" { - // --install-cert must be called before this. + } else if args[0] == "--install-idd" { + // Install cert if cert file is provided (2rd arg). + if args.len() == 2 { + #[cfg(windows)] + hbb_common::allow_err!(crate::platform::windows::install_cert(&args[1])); + } #[cfg(all(windows, feature = "virtual_display_driver"))] if crate::virtual_display_manager::is_virtual_display_supported() { hbb_common::allow_err!(crate::virtual_display_manager::install_update_driver()); } return None; - } - else if args[0] == "--portable-service" { + } else if args[0] == "--portable-service" { crate::platform::elevate_or_run_as_system( click_setup, _is_elevate,