remove confusions

This commit is contained in:
rustdesk 2022-05-13 09:56:44 +08:00
parent df2e4bb411
commit cb26ce5bbc
3 changed files with 8 additions and 17 deletions

View File

@ -55,11 +55,11 @@ impl Database {
if !std::path::Path::new(url).exists() { if !std::path::Path::new(url).exists() {
std::fs::File::create(url).ok(); std::fs::File::create(url).ok();
} }
let n: usize = std::env::var("MAX_CONNECTIONS") let n: usize = std::env::var("MAX_DATABASE_CONNECTIONS")
.unwrap_or("1".to_owned()) .unwrap_or("1".to_owned())
.parse() .parse()
.unwrap_or(1); .unwrap_or(1);
log::info!("MAX_CONNECTIONS={}", n); log::debug!("MAX_DATABASE_CONNECTIONS={}", n);
let pool = Pool::new( let pool = Pool::new(
DbPool { DbPool {
url: url.to_owned(), url: url.to_owned(),

View File

@ -20,7 +20,6 @@ fn main() -> ResultType<()> {
-R, --rendezvous-servers=[HOSTS] 'Sets rendezvous servers, seperated by colon' -R, --rendezvous-servers=[HOSTS] 'Sets rendezvous servers, seperated by colon'
-u, --software-url=[URL] 'Sets download url of RustDesk software of newest version' -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' -r, --relay-servers=[HOST] 'Sets the default relay servers, seperated by colon'
-C, --change-id=[BOOL(default=Y)] 'Sets if support to change id'
-M, --rmem=[NUMBER(default={})] 'Sets UDP recv buffer size, set system rmem_max first, e.g., sudo sysctl -w net.core.rmem_max=52428800. vi /etc/sysctl.conf, net.core.rmem_max=52428800, sudo sysctl p' -M, --rmem=[NUMBER(default={})] 'Sets UDP recv buffer size, set system rmem_max first, e.g., sudo sysctl -w net.core.rmem_max=52428800. vi /etc/sysctl.conf, net.core.rmem_max=52428800, sudo sysctl p'
-k, --key=[KEY] 'Only allow the client with the same key'", -k, --key=[KEY] 'Only allow the client with the same key'",
RENDEZVOUS_PORT, RENDEZVOUS_PORT,
@ -33,7 +32,6 @@ fn main() -> ResultType<()> {
} }
let rmem = get_arg("rmem").parse::<usize>().unwrap_or(RMEM); let rmem = get_arg("rmem").parse::<usize>().unwrap_or(RMEM);
let serial: i32 = get_arg("serial").parse().unwrap_or(0); let serial: i32 = get_arg("serial").parse().unwrap_or(0);
let id_change_support: bool = get_arg_or("change-id", "Y".to_owned()).to_uppercase() == "Y"; RendezvousServer::start(port, serial, &get_arg("key"), rmem)?;
RendezvousServer::start(port, serial, &get_arg("key"), id_change_support, rmem)?;
Ok(()) Ok(())
} }

View File

@ -85,7 +85,6 @@ impl RendezvousServer {
port: i32, port: i32,
serial: i32, serial: i32,
key: &str, key: &str,
id_change_support: bool,
rmem: usize, rmem: usize,
) -> ResultType<()> { ) -> ResultType<()> {
let addr = format!("0.0.0.0:{}", port); let addr = format!("0.0.0.0:{}", port);
@ -97,7 +96,6 @@ impl RendezvousServer {
log::info!("Listening on tcp/udp {}", addr); log::info!("Listening on tcp/udp {}", addr);
log::info!("Listening on tcp {}, extra port for NAT test", addr2); log::info!("Listening on tcp {}, extra port for NAT test", addr2);
log::info!("Listening on websocket {}", addr3); log::info!("Listening on websocket {}", addr3);
log::info!("change-id={:?}", id_change_support);
let mut socket = FramedSocket::new_with_buf_size(&addr, rmem).await?; let mut socket = FramedSocket::new_with_buf_size(&addr, rmem).await?;
let (tx, mut rx) = mpsc::unbounded_channel::<Data>(); let (tx, mut rx) = mpsc::unbounded_channel::<Data>();
let software_url = get_arg("software-url"); let software_url = get_arg("software-url");
@ -163,7 +161,6 @@ impl RendezvousServer {
&mut listener3, &mut listener3,
&mut socket, &mut socket,
&key, &key,
id_change_support,
) )
.await .await
{ {
@ -195,7 +192,6 @@ impl RendezvousServer {
listener3: &mut TcpListener, listener3: &mut TcpListener,
socket: &mut FramedSocket, socket: &mut FramedSocket,
key: &str, key: &str,
id_change_support: bool,
) -> LoopFailure { ) -> LoopFailure {
let mut timer_check_relay = interval(Duration::from_millis(CHECK_RELAY_TIMEOUT)); let mut timer_check_relay = interval(Duration::from_millis(CHECK_RELAY_TIMEOUT));
loop { loop {
@ -249,7 +245,7 @@ impl RendezvousServer {
match res { match res {
Ok((stream, addr)) => { Ok((stream, addr)) => {
stream.set_nodelay(true).ok(); stream.set_nodelay(true).ok();
self.handle_listener(stream, addr, key, id_change_support, true).await; self.handle_listener(stream, addr, key, true).await;
} }
Err(err) => { Err(err) => {
log::error!("listener3.accept failed: {}", err); log::error!("listener3.accept failed: {}", err);
@ -261,7 +257,7 @@ impl RendezvousServer {
match res { match res {
Ok((stream, addr)) => { Ok((stream, addr)) => {
stream.set_nodelay(true).ok(); stream.set_nodelay(true).ok();
self.handle_listener(stream, addr, key, id_change_support, false).await; self.handle_listener(stream, addr, key, false).await;
} }
Err(err) => { Err(err) => {
log::error!("listener.accept failed: {}", err); log::error!("listener.accept failed: {}", err);
@ -446,7 +442,6 @@ impl RendezvousServer {
sink: &mut Option<Sink>, sink: &mut Option<Sink>,
addr: SocketAddr, addr: SocketAddr,
key: &str, key: &str,
id_change_support: bool,
ws: bool, ws: bool,
) -> bool { ) -> bool {
if let Ok(msg_in) = RendezvousMessage::parse_from_bytes(&bytes) { if let Ok(msg_in) = RendezvousMessage::parse_from_bytes(&bytes) {
@ -1022,7 +1017,6 @@ impl RendezvousServer {
stream: TcpStream, stream: TcpStream,
addr: SocketAddr, addr: SocketAddr,
key: &str, key: &str,
id_change_support: bool,
ws: bool, ws: bool,
) { ) {
log::debug!("Tcp connection from {:?}, ws: {}", addr, ws); log::debug!("Tcp connection from {:?}, ws: {}", addr, ws);
@ -1030,7 +1024,7 @@ impl RendezvousServer {
let key = key.to_owned(); let key = key.to_owned();
tokio::spawn(async move { tokio::spawn(async move {
allow_err!( allow_err!(
rs.handle_listener_inner(stream, addr, &key, id_change_support, ws) rs.handle_listener_inner(stream, addr, &key, ws)
.await .await
); );
}); });
@ -1042,7 +1036,6 @@ impl RendezvousServer {
stream: TcpStream, stream: TcpStream,
addr: SocketAddr, addr: SocketAddr,
key: &str, key: &str,
id_change_support: bool,
ws: bool, ws: bool,
) -> ResultType<()> { ) -> ResultType<()> {
let mut sink; let mut sink;
@ -1054,7 +1047,7 @@ impl RendezvousServer {
match msg { match msg {
tungstenite::Message::Binary(bytes) => { tungstenite::Message::Binary(bytes) => {
if !self if !self
.handle_tcp(&bytes, &mut sink, addr, key, id_change_support, ws) .handle_tcp(&bytes, &mut sink, addr, key, ws)
.await .await
{ {
break; break;
@ -1068,7 +1061,7 @@ impl RendezvousServer {
sink = Some(Sink::TcpStream(a)); sink = Some(Sink::TcpStream(a));
while let Ok(Some(Ok(bytes))) = timeout(30_000, b.next()).await { while let Ok(Some(Ok(bytes))) = timeout(30_000, b.next()).await {
if !self if !self
.handle_tcp(&bytes, &mut sink, addr, key, id_change_support, ws) .handle_tcp(&bytes, &mut sink, addr, key, ws)
.await .await
{ {
break; break;