mirror of
https://github.com/rustdesk/rustdesk-server.git
synced 2025-07-01 23:35:38 +00:00
remove confusions
This commit is contained in:
parent
df2e4bb411
commit
cb26ce5bbc
@ -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(),
|
||||||
|
@ -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(())
|
||||||
}
|
}
|
||||||
|
@ -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;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user