rustdesk/hbb_common
1
0
Fork 0
mirror of https://github.com/rustdesk/hbb_common.git synced 2025-07-01 15:36:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #147 from fufesou/feat/hostname_as_id

Browse Source 
feat: hostname as id
This commit is contained in:
RustDesk 2025-04-30 14:09:45 +08:00 committed by GitHub
commit bfddd5bb19
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 31 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Cargo.toml
View File

@ -64,6 +64,7 @@ rustls-platform-verifier = "0.5"
rustls-pki-types = "1.11"
[target.'cfg(any(target_os = "macos", target_os = "windows"))'.dependencies]
tokio-native-tls = "0.3"
whoami = "1.5.0"
[build-dependencies]
protobuf-codegen = { version = "3.7" }

32
src/config.rs
View File

@ -560,7 +560,7 @@ impl Config {
}
if !id_valid {
for _ in 0..3 {
if let Some(id) = Config::get_auto_id() {
if let Some(id) = Config::gen_id() {
config.id = id;
store = true;
break;
@ -822,6 +822,32 @@ impl Config {
std::cmp::max(CONFIG2.read().unwrap().serial, SERIAL)
}
#[cfg(any(target_os = "android", target_os = "ios"))]
fn gen_id() -> Option<String> {
Self::get_auto_id()
}
#[cfg(not(any(target_os = "android", target_os = "ios")))]
fn gen_id() -> Option<String> {
let hostname_as_id = BUILTIN_SETTINGS
.read()
.unwrap()
.get(keys::OPTION_ALLOW_HOSTNAME_AS_ID)
.map(|v| option2bool(keys::OPTION_ALLOW_HOSTNAME_AS_ID, v))
.unwrap_or(false);
if hostname_as_id {
match whoami::fallible::hostname() {
Ok(h) => Some(h.replace(" ", "-")),
Err(e) => {
log::warn!("Failed to get hostname, \"{}\", fallback to auto id", e);
Self::get_auto_id()
}
}
} else {
Self::get_auto_id()
}
}
fn get_auto_id() -> Option<String> {
#[cfg(any(target_os = "android", target_os = "ios"))]
{
@ -909,7 +935,7 @@ impl Config {
pub fn get_id() -> String {
let mut id = CONFIG.read().unwrap().id.clone();
if id.is_empty() {
if let Some(tmp) = Config::get_auto_id() {
if let Some(tmp) = Config::gen_id() {
id = tmp;
Config::set_id(&id);
}
@ -2369,6 +2395,7 @@ pub mod keys {
pub const OPTION_ALLOW_LOGON_SCREEN_PASSWORD: &str = "allow-logon-screen-password";
pub const OPTION_ONE_WAY_FILE_TRANSFER: &str = "one-way-file-transfer";
pub const OPTION_ALLOW_HTTPS_21114: &str = "allow-https-21114";
pub const OPTION_ALLOW_HOSTNAME_AS_ID: &str = "allow-hostname-as-id";
// flutter local options
pub const OPTION_FLUTTER_REMOTE_MENUBAR_STATE: &str = "remoteMenubarState";
@ -2528,6 +2555,7 @@ pub mod keys {
OPTION_ALLOW_LOGON_SCREEN_PASSWORD,
OPTION_ONE_WAY_FILE_TRANSFER,
OPTION_ALLOW_HTTPS_21114,
OPTION_ALLOW_HOSTNAME_AS_ID,
];
}