lic draft

1 files changed, 7 insertions, 1 deletions

diff --git a/src/hbbr.rs b/src/hbbr.rs
index c2ac21b..f0633a2 100644
--- a/src/hbbr.rs
+++ b/src/hbbr.rs
@@ -3,14 +3,17 @@ mod relay_server;
 use hbb_common::{env_logger::*, ResultType};
 use relay_server::*;
 use std::sync::{Arc, Mutex};
+mod lic;
 
 fn main() -> ResultType<()> {
     init_from_env(Env::default().filter_or(DEFAULT_FILTER_ENV, "info"));
     let args = format!(
         "-p, --port=[NUMBER(default={})] 'Sets the listening port'
         -k, --key=[KEY] 'Only allow the client with the same key'
+        {}
         ",
-        DEFAULT_PORT
+        DEFAULT_PORT,
+        lic::EMAIL_ARG
     );
     let matches = App::new("hbbr")
         .version(hbbs::VERSION)
@@ -18,6 +21,9 @@ fn main() -> ResultType<()> {
         .about("RustDesk Relay Server")
         .args_from_usage(&args)
         .get_matches();
+    if !lic::check_lic(matches.value_of("email").unwrap_or("")) {
+        return Ok(());
+    }
     let stop: Arc<Mutex<bool>> = Default::default();
     start(
         matches.value_of("port").unwrap_or(DEFAULT_PORT),