mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2026-02-01 10:31:54 +08:00
feat!: 移除内置公共服务器
- 移除公共 RustDesk ID 服务器 (用户需自行配置) - 移除公共 TURN 服务器 (仅保留 Google STUN) - 清理废弃代码: PublicServerInfo, is_using_public_server 等 - 更新前端 UI 和国际化文本 - 重新生成 TypeScript 类型 破坏性变更: 不再提供内置公共服务器。用户必须配置自己的 RustDesk 服务器和 TURN 服务器才能在生产环境中使用。
This commit is contained in:
@@ -5,8 +5,6 @@
|
||||
use serde::{Deserialize, Serialize};
|
||||
use typeshare::typeshare;
|
||||
|
||||
use crate::secrets;
|
||||
|
||||
/// RustDesk configuration
|
||||
#[typeshare]
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
@@ -15,9 +13,8 @@ pub struct RustDeskConfig {
|
||||
/// Enable RustDesk protocol
|
||||
pub enabled: bool,
|
||||
|
||||
/// Rendezvous server address (hbbs), e.g., "rs.example.com" or "192.168.1.100"
|
||||
/// Port defaults to 21116 if not specified
|
||||
/// If empty, uses the public server from secrets.toml
|
||||
/// Rendezvous server address (hbbs), e.g., "rs.example.com" or "192.168.1.100:21116"
|
||||
/// Required for RustDesk to function
|
||||
pub rendezvous_server: String,
|
||||
|
||||
/// Relay server address (hbbr), if different from rendezvous server
|
||||
@@ -79,39 +76,17 @@ impl Default for RustDeskConfig {
|
||||
|
||||
impl RustDeskConfig {
|
||||
/// Check if the configuration is valid for starting the service
|
||||
/// Returns true if enabled and has a valid server (user-configured or public)
|
||||
/// Returns true if enabled and has a valid server
|
||||
pub fn is_valid(&self) -> bool {
|
||||
self.enabled
|
||||
&& !self.effective_rendezvous_server().is_empty()
|
||||
&& !self.rendezvous_server.is_empty()
|
||||
&& !self.device_id.is_empty()
|
||||
&& !self.device_password.is_empty()
|
||||
}
|
||||
|
||||
/// Check if using the public server (user left rendezvous_server empty)
|
||||
pub fn is_using_public_server(&self) -> bool {
|
||||
self.rendezvous_server.is_empty() && secrets::rustdesk::has_public_server()
|
||||
}
|
||||
|
||||
/// Get the effective rendezvous server (user-configured or public fallback)
|
||||
/// Get the rendezvous server (user-configured)
|
||||
pub fn effective_rendezvous_server(&self) -> &str {
|
||||
if self.rendezvous_server.is_empty() {
|
||||
secrets::rustdesk::PUBLIC_SERVER
|
||||
} else {
|
||||
&self.rendezvous_server
|
||||
}
|
||||
}
|
||||
|
||||
/// Get public server info for display (server address and public key)
|
||||
/// Returns None if no public server is configured
|
||||
pub fn public_server_info() -> Option<PublicServerInfo> {
|
||||
if secrets::rustdesk::has_public_server() {
|
||||
Some(PublicServerInfo {
|
||||
server: secrets::rustdesk::PUBLIC_SERVER.to_string(),
|
||||
public_key: secrets::rustdesk::PUBLIC_KEY.to_string(),
|
||||
})
|
||||
} else {
|
||||
None
|
||||
}
|
||||
&self.rendezvous_server
|
||||
}
|
||||
|
||||
/// Generate a new random device ID
|
||||
@@ -148,8 +123,10 @@ impl RustDeskConfig {
|
||||
|
||||
/// Get the rendezvous server address with default port
|
||||
pub fn rendezvous_addr(&self) -> String {
|
||||
let server = self.effective_rendezvous_server();
|
||||
if server.contains(':') {
|
||||
let server = &self.rendezvous_server;
|
||||
if server.is_empty() {
|
||||
String::new()
|
||||
} else if server.contains(':') {
|
||||
server.to_string()
|
||||
} else {
|
||||
format!("{}:21116", server)
|
||||
@@ -165,8 +142,8 @@ impl RustDeskConfig {
|
||||
format!("{}:21117", s)
|
||||
}
|
||||
}).or_else(|| {
|
||||
// Default: same host as effective rendezvous server
|
||||
let server = self.effective_rendezvous_server();
|
||||
// Default: same host as rendezvous server
|
||||
let server = &self.rendezvous_server;
|
||||
if !server.is_empty() {
|
||||
let host = server.split(':').next().unwrap_or("");
|
||||
if !host.is_empty() {
|
||||
@@ -181,16 +158,6 @@ impl RustDeskConfig {
|
||||
}
|
||||
}
|
||||
|
||||
/// Public server information for display to users
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
#[typeshare]
|
||||
pub struct PublicServerInfo {
|
||||
/// Public server address
|
||||
pub server: String,
|
||||
/// Public key for client connection
|
||||
pub public_key: String,
|
||||
}
|
||||
|
||||
/// Generate a random 9-digit device ID
|
||||
pub fn generate_device_id() -> String {
|
||||
use rand::Rng;
|
||||
@@ -239,6 +206,10 @@ mod tests {
|
||||
|
||||
config.rendezvous_server = "example.com:21116".to_string();
|
||||
assert_eq!(config.rendezvous_addr(), "example.com:21116");
|
||||
|
||||
// Empty server returns empty string
|
||||
config.rendezvous_server = String::new();
|
||||
assert_eq!(config.rendezvous_addr(), "");
|
||||
}
|
||||
|
||||
#[test]
|
||||
@@ -252,6 +223,10 @@ mod tests {
|
||||
// Explicit relay server
|
||||
config.relay_server = Some("relay.example.com".to_string());
|
||||
assert_eq!(config.relay_addr(), Some("relay.example.com:21117".to_string()));
|
||||
|
||||
// No rendezvous server, relay is None
|
||||
config.rendezvous_server = String::new();
|
||||
assert_eq!(config.relay_addr(), None);
|
||||
}
|
||||
|
||||
#[test]
|
||||
@@ -262,10 +237,8 @@ mod tests {
|
||||
config.rendezvous_server = "custom.example.com".to_string();
|
||||
assert_eq!(config.effective_rendezvous_server(), "custom.example.com");
|
||||
|
||||
// When empty, falls back to public server (if configured)
|
||||
// When empty, returns empty
|
||||
config.rendezvous_server = String::new();
|
||||
// This will return PUBLIC_SERVER from secrets
|
||||
let effective = config.effective_rendezvous_server();
|
||||
assert!(!effective.is_empty() || !secrets::rustdesk::has_public_server());
|
||||
assert_eq!(config.effective_rendezvous_server(), "");
|
||||
}
|
||||
}
|
||||
|
||||
@@ -239,16 +239,10 @@ impl RustDeskService {
|
||||
let config = service_config_punch.clone();
|
||||
|
||||
tokio::spawn(async move {
|
||||
// Get relay_key from config, or use public server's relay_key if using public server
|
||||
// Get relay_key from config (no public server fallback)
|
||||
let relay_key = {
|
||||
let cfg = config.read();
|
||||
cfg.relay_key.clone().unwrap_or_else(|| {
|
||||
if cfg.is_using_public_server() {
|
||||
crate::secrets::rustdesk::RELAY_KEY.to_string()
|
||||
} else {
|
||||
String::new()
|
||||
}
|
||||
})
|
||||
cfg.relay_key.clone().unwrap_or_default()
|
||||
};
|
||||
|
||||
// Try P2P direct connection first
|
||||
@@ -295,16 +289,10 @@ impl RustDeskService {
|
||||
let config = service_config.clone();
|
||||
|
||||
tokio::spawn(async move {
|
||||
// Get relay_key from config, or use public server's relay_key if using public server
|
||||
// Get relay_key from config (no public server fallback)
|
||||
let relay_key = {
|
||||
let cfg = config.read();
|
||||
cfg.relay_key.clone().unwrap_or_else(|| {
|
||||
if cfg.is_using_public_server() {
|
||||
crate::secrets::rustdesk::RELAY_KEY.to_string()
|
||||
} else {
|
||||
String::new()
|
||||
}
|
||||
})
|
||||
cfg.relay_key.clone().unwrap_or_default()
|
||||
};
|
||||
|
||||
if let Err(e) = handle_relay_request(
|
||||
|
||||
Reference in New Issue
Block a user