From f558c12866c430d86254ad8d8afbe0960c1c37ce Mon Sep 17 00:00:00 2001
From: rustdesk <info@rustdesk.com>
Date: Tue, 3 Aug 2021 14:59:45 +0800
Subject: [PATCH] --password https://github.com/rustdesk/rustdesk/issues/187

---
 src/main.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index 53c30cf5a..9305c1742 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -76,7 +76,10 @@ fn main() {
             #[cfg(windows)]
             {
                 hbb_common::allow_err!(platform::uninstall_me());
-                hbb_common::allow_err!(platform::install_me("desktopicon startmenu"));
+                hbb_common::allow_err!(platform::install_me(
+                    "desktopicon startmenu",
+                    "".to_owned()
+                ));
                 return;
             }
         } else if args[0] == "--remove" {
@@ -99,6 +102,11 @@ fn main() {
                 hbb_common::config::Config::import(&args[1]);
             }
             return;
+        } else if args[0] == "--password" {
+            if args.len() == 2 {
+                hbb_common::config::Config::set_password(&args[1]);
+            }
+            return;
         }
     }
     ui::start(&mut args[..]);