mirror of
https://github.com/rustdesk/rustdesk-server.git
synced 2024-12-18 21:47:47 +08:00
84 lines
3.1 KiB
Rust
84 lines
3.1 KiB
Rust
// https://tools.ietf.org/rfc/rfc5128.txt
|
|
// https://blog.csdn.net/bytxl/article/details/44344855
|
|
|
|
use clap::App;
|
|
use hbb_common::{env_logger::*, log, ResultType};
|
|
use hbbs::*;
|
|
mod lic;
|
|
use ini::Ini;
|
|
use std::sync::{Arc, Mutex};
|
|
|
|
fn main() -> ResultType<()> {
|
|
init_from_env(Env::default().filter_or(DEFAULT_FILTER_ENV, "info"));
|
|
let args = format!(
|
|
"-c --config=[FILE] +takes_value 'Sets a custom config file'
|
|
-p, --port=[NUMBER(default={})] 'Sets the listening port'
|
|
-s, --serial=[NUMBER(default=0)] 'Sets configure update serial number'
|
|
-R, --rendezvous-servers=[HOSTS] 'Sets rendezvous servers, seperated by colon'
|
|
-u, --software-url=[URL] 'Sets download url of RustDesk software of newest version'
|
|
-r, --relay-servers=[HOST] 'Sets the default relay servers, seperated by colon'
|
|
-C, --change-id=[BOOL(default=Y)] 'Sets if support to change id'
|
|
{}
|
|
-k, --key=[KEY] 'Only allow the client with the same key'",
|
|
DEFAULT_PORT,
|
|
lic::EMAIL_ARG
|
|
);
|
|
let matches = App::new("hbbs")
|
|
.version(crate::VERSION)
|
|
.author("CarrieZ Studio<info@rustdesk.com>")
|
|
.about("RustDesk ID/Rendezvous Server")
|
|
.args_from_usage(&args)
|
|
.get_matches();
|
|
let mut section = None;
|
|
let conf; // for holding section
|
|
if let Some(config) = matches.value_of("config") {
|
|
if let Ok(v) = Ini::load_from_file(config) {
|
|
conf = v;
|
|
section = conf.section(None::<String>);
|
|
}
|
|
}
|
|
let get_arg = |name: &str, default: &str| -> String {
|
|
if let Some(v) = matches.value_of(name) {
|
|
return v.to_owned();
|
|
} else if let Some(section) = section {
|
|
if let Some(v) = section.get(name) {
|
|
return v.to_owned();
|
|
}
|
|
}
|
|
return default.to_owned();
|
|
};
|
|
let port = get_arg("port", DEFAULT_PORT);
|
|
let relay_servers: Vec<String> = get_arg("relay-servers", "")
|
|
.split(",")
|
|
.filter(|x| !x.is_empty() && test_if_valid_server(x, "relay-server").is_ok())
|
|
.map(|x| x.to_owned())
|
|
.collect();
|
|
let serial: i32 = get_arg("serial", "").parse().unwrap_or(0);
|
|
let id_change_support: bool = get_arg("change-id", "Y").to_uppercase() == "Y";
|
|
let rendezvous_servers: Vec<String> = get_arg("rendezvous-servers", "")
|
|
.split(",")
|
|
.filter(|x| !x.is_empty() && test_if_valid_server(x, "rendezvous-server").is_ok())
|
|
.map(|x| x.to_owned())
|
|
.collect();
|
|
let addr = format!("0.0.0.0:{}", port);
|
|
let addr2 = format!("0.0.0.0:{}", port.parse::<i32>().unwrap_or(0) - 1);
|
|
log::info!("serial={}", serial);
|
|
log::info!("rendezvous-servers={:?}", rendezvous_servers);
|
|
let stop: Arc<Mutex<bool>> = Default::default();
|
|
if !lic::check_lic(&get_arg("email", "")) {
|
|
return Ok(());
|
|
}
|
|
RendezvousServer::start(
|
|
&addr,
|
|
&addr2,
|
|
relay_servers,
|
|
serial,
|
|
rendezvous_servers,
|
|
get_arg("software-url", ""),
|
|
&get_arg("key", ""),
|
|
stop,
|
|
id_change_support,
|
|
)?;
|
|
Ok(())
|
|
}
|