rustdesk лет назад: 3
Родитель
Сommit
cb26ce5bbc
3 измененных файлов с 8 добавлено и 17 удалено
  1. 2 2
      src/database.rs
  2. 1 3
      src/main.rs
  3. 5 12
      src/rendezvous_server.rs

+ 2 - 2
src/database.rs

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

+ 1 - 3
src/main.rs

@@ -20,7 +20,6 @@ fn main() -> ResultType<()> {
20
         -R, --rendezvous-servers=[HOSTS] 'Sets rendezvous servers, seperated by colon'
20
         -R, --rendezvous-servers=[HOSTS] 'Sets rendezvous servers, seperated by colon'
21
         -u, --software-url=[URL] 'Sets download url of RustDesk software of newest version'
21
         -u, --software-url=[URL] 'Sets download url of RustDesk software of newest version'
22
         -r, --relay-servers=[HOST] 'Sets the default relay servers, seperated by colon'
22
         -r, --relay-servers=[HOST] 'Sets the default relay servers, seperated by colon'
23
-        -C, --change-id=[BOOL(default=Y)] 'Sets if support to change id'
24
         -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'
23
         -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'
25
         -k, --key=[KEY] 'Only allow the client with the same key'",
24
         -k, --key=[KEY] 'Only allow the client with the same key'",
26
         RENDEZVOUS_PORT,
25
         RENDEZVOUS_PORT,
@@ -33,7 +32,6 @@ fn main() -> ResultType<()> {
33
     }
32
     }
34
     let rmem = get_arg("rmem").parse::<usize>().unwrap_or(RMEM);
33
     let rmem = get_arg("rmem").parse::<usize>().unwrap_or(RMEM);
35
     let serial: i32 = get_arg("serial").parse().unwrap_or(0);
34
     let serial: i32 = get_arg("serial").parse().unwrap_or(0);
36
-    let id_change_support: bool = get_arg_or("change-id", "Y".to_owned()).to_uppercase() == "Y";
37
-    RendezvousServer::start(port, serial, &get_arg("key"), id_change_support, rmem)?;
35
+    RendezvousServer::start(port, serial, &get_arg("key"), rmem)?;
38
     Ok(())
36
     Ok(())
39
 }
37
 }

+ 5 - 12
src/rendezvous_server.rs

@@ -85,7 +85,6 @@ impl RendezvousServer {
85
         port: i32,
85
         port: i32,
86
         serial: i32,
86
         serial: i32,
87
         key: &str,
87
         key: &str,
88
-        id_change_support: bool,
89
         rmem: usize,
88
         rmem: usize,
90
     ) -> ResultType<()> {
89
     ) -> ResultType<()> {
91
         let addr = format!("0.0.0.0:{}", port);
90
         let addr = format!("0.0.0.0:{}", port);
@@ -97,7 +96,6 @@ impl RendezvousServer {
97
         log::info!("Listening on tcp/udp {}", addr);
96
         log::info!("Listening on tcp/udp {}", addr);
98
         log::info!("Listening on tcp {}, extra port for NAT test", addr2);
97
         log::info!("Listening on tcp {}, extra port for NAT test", addr2);
99
         log::info!("Listening on websocket {}", addr3);
98
         log::info!("Listening on websocket {}", addr3);
100
-        log::info!("change-id={:?}", id_change_support);
101
         let mut socket = FramedSocket::new_with_buf_size(&addr, rmem).await?;
99
         let mut socket = FramedSocket::new_with_buf_size(&addr, rmem).await?;
102
         let (tx, mut rx) = mpsc::unbounded_channel::<Data>();
100
         let (tx, mut rx) = mpsc::unbounded_channel::<Data>();
103
         let software_url = get_arg("software-url");
101
         let software_url = get_arg("software-url");
@@ -163,7 +161,6 @@ impl RendezvousServer {
163
                     &mut listener3,
161
                     &mut listener3,
164
                     &mut socket,
162
                     &mut socket,
165
                     &key,
163
                     &key,
166
-                    id_change_support,
167
                 )
164
                 )
168
                 .await
165
                 .await
169
             {
166
             {
@@ -195,7 +192,6 @@ impl RendezvousServer {
195
         listener3: &mut TcpListener,
192
         listener3: &mut TcpListener,
196
         socket: &mut FramedSocket,
193
         socket: &mut FramedSocket,
197
         key: &str,
194
         key: &str,
198
-        id_change_support: bool,
199
     ) -> LoopFailure {
195
     ) -> LoopFailure {
200
         let mut timer_check_relay = interval(Duration::from_millis(CHECK_RELAY_TIMEOUT));
196
         let mut timer_check_relay = interval(Duration::from_millis(CHECK_RELAY_TIMEOUT));
201
         loop {
197
         loop {
@@ -249,7 +245,7 @@ impl RendezvousServer {
249
                     match res {
245
                     match res {
250
                         Ok((stream, addr))  => {
246
                         Ok((stream, addr))  => {
251
                             stream.set_nodelay(true).ok();
247
                             stream.set_nodelay(true).ok();
252
-                            self.handle_listener(stream, addr, key, id_change_support, true).await;
248
+                            self.handle_listener(stream, addr, key, true).await;
253
                         }
249
                         }
254
                         Err(err) => {
250
                         Err(err) => {
255
                            log::error!("listener3.accept failed: {}", err);
251
                            log::error!("listener3.accept failed: {}", err);
@@ -261,7 +257,7 @@ impl RendezvousServer {
261
                     match res {
257
                     match res {
262
                         Ok((stream, addr)) => {
258
                         Ok((stream, addr)) => {
263
                             stream.set_nodelay(true).ok();
259
                             stream.set_nodelay(true).ok();
264
-                            self.handle_listener(stream, addr, key, id_change_support, false).await;
260
+                            self.handle_listener(stream, addr, key, false).await;
265
                         }
261
                         }
266
                        Err(err) => {
262
                        Err(err) => {
267
                            log::error!("listener.accept failed: {}", err);
263
                            log::error!("listener.accept failed: {}", err);
@@ -446,7 +442,6 @@ impl RendezvousServer {
446
         sink: &mut Option<Sink>,
442
         sink: &mut Option<Sink>,
447
         addr: SocketAddr,
443
         addr: SocketAddr,
448
         key: &str,
444
         key: &str,
449
-        id_change_support: bool,
450
         ws: bool,
445
         ws: bool,
451
     ) -> bool {
446
     ) -> bool {
452
         if let Ok(msg_in) = RendezvousMessage::parse_from_bytes(&bytes) {
447
         if let Ok(msg_in) = RendezvousMessage::parse_from_bytes(&bytes) {
@@ -1022,7 +1017,6 @@ impl RendezvousServer {
1022
         stream: TcpStream,
1017
         stream: TcpStream,
1023
         addr: SocketAddr,
1018
         addr: SocketAddr,
1024
         key: &str,
1019
         key: &str,
1025
-        id_change_support: bool,
1026
         ws: bool,
1020
         ws: bool,
1027
     ) {
1021
     ) {
1028
         log::debug!("Tcp connection from {:?}, ws: {}", addr, ws);
1022
         log::debug!("Tcp connection from {:?}, ws: {}", addr, ws);
@@ -1030,7 +1024,7 @@ impl RendezvousServer {
1030
         let key = key.to_owned();
1024
         let key = key.to_owned();
1031
         tokio::spawn(async move {
1025
         tokio::spawn(async move {
1032
             allow_err!(
1026
             allow_err!(
1033
-                rs.handle_listener_inner(stream, addr, &key, id_change_support, ws)
1027
+                rs.handle_listener_inner(stream, addr, &key, ws)
1034
                     .await
1028
                     .await
1035
             );
1029
             );
1036
         });
1030
         });
@@ -1042,7 +1036,6 @@ impl RendezvousServer {
1042
         stream: TcpStream,
1036
         stream: TcpStream,
1043
         addr: SocketAddr,
1037
         addr: SocketAddr,
1044
         key: &str,
1038
         key: &str,
1045
-        id_change_support: bool,
1046
         ws: bool,
1039
         ws: bool,
1047
     ) -> ResultType<()> {
1040
     ) -> ResultType<()> {
1048
         let mut sink;
1041
         let mut sink;
@@ -1054,7 +1047,7 @@ impl RendezvousServer {
1054
                 match msg {
1047
                 match msg {
1055
                     tungstenite::Message::Binary(bytes) => {
1048
                     tungstenite::Message::Binary(bytes) => {
1056
                         if !self
1049
                         if !self
1057
-                            .handle_tcp(&bytes, &mut sink, addr, key, id_change_support, ws)
1050
+                            .handle_tcp(&bytes, &mut sink, addr, key, ws)
1058
                             .await
1051
                             .await
1059
                         {
1052
                         {
1060
                             break;
1053
                             break;
@@ -1068,7 +1061,7 @@ impl RendezvousServer {
1068
             sink = Some(Sink::TcpStream(a));
1061
             sink = Some(Sink::TcpStream(a));
1069
             while let Ok(Some(Ok(bytes))) = timeout(30_000, b.next()).await {
1062
             while let Ok(Some(Ok(bytes))) = timeout(30_000, b.next()).await {
1070
                 if !self
1063
                 if !self
1071
-                    .handle_tcp(&bytes, &mut sink, addr, key, id_change_support, ws)
1064
+                    .handle_tcp(&bytes, &mut sink, addr, key, ws)
1072
                     .await
1065
                     .await
1073
                 {
1066
                 {
1074
                     break;
1067
                     break;