rustdesk лет назад: 3
Родитель
Сommit
7e307a5a1c
11 измененных файлов с 19 добавлено и 21 удалено
  1. 1 0
      .gitignore
  2. 1 1
      libs/hbb_common/build.rs
  3. 1 1
      libs/hbb_common/src/protos/mod.rs
  4. 1 1
      src/common.rs
  5. 1 2
      src/hbbr.rs
  6. 1 0
      src/lib.rs
  7. 2 4
      src/main.rs
  8. 1 1
      src/peer.rs
  9. 1 1
      src/relay_server.rs
  10. 1 1
      src/rendezvous_server.rs
  11. 8 9
      src/utils.rs

+ 1 - 0
.gitignore

@@ -8,3 +8,4 @@ debian/debhelper-build-stamp
8
 .vscode
8
 .vscode
9
 src/version.rs
9
 src/version.rs
10
 db_v2.sqlite3
10
 db_v2.sqlite3
11
+test.*

+ 1 - 1
libs/hbb_common/build.rs

@@ -2,7 +2,7 @@ fn main() {
2
     let out_dir = format!("{}/protos", std::env::var("OUT_DIR").unwrap());
2
     let out_dir = format!("{}/protos", std::env::var("OUT_DIR").unwrap());
3
 
3
 
4
     std::fs::create_dir_all(&out_dir).unwrap();
4
     std::fs::create_dir_all(&out_dir).unwrap();
5
-    
5
+
6
     protobuf_codegen::Codegen::new()
6
     protobuf_codegen::Codegen::new()
7
         .pure()
7
         .pure()
8
         .out_dir(out_dir)
8
         .out_dir(out_dir)

+ 1 - 1
libs/hbb_common/src/protos/mod.rs

@@ -1 +1 @@
1
-include!(concat!(env!("OUT_DIR"), "/protos/mod.rs"));
1
+include!(concat!(env!("OUT_DIR"), "/protos/mod.rs"));

+ 1 - 1
src/common.rs

@@ -134,7 +134,7 @@ pub fn gen_sk(wait: u64) -> (String, Option<sign::SecretKey>) {
134
             }
134
             }
135
             (pk, sk) = gen_func();
135
             (pk, sk) = gen_func();
136
         }
136
         }
137
-        let pub_file = format!("{}.pub", sk_file);
137
+        let pub_file = format!("{sk_file}.pub");
138
         if let Ok(mut f) = std::fs::File::create(&pub_file) {
138
         if let Ok(mut f) = std::fs::File::create(&pub_file) {
139
             f.write_all(pk.as_bytes()).ok();
139
             f.write_all(pk.as_bytes()).ok();
140
             if let Ok(mut f) = std::fs::File::create(sk_file) {
140
             if let Ok(mut f) = std::fs::File::create(sk_file) {

+ 1 - 2
src/hbbr.rs

@@ -13,10 +13,9 @@ fn main() -> ResultType<()> {
13
         .write_mode(WriteMode::Async)
13
         .write_mode(WriteMode::Async)
14
         .start()?;
14
         .start()?;
15
     let args = format!(
15
     let args = format!(
16
-        "-p, --port=[NUMBER(default={})] 'Sets the listening port'
16
+        "-p, --port=[NUMBER(default={RELAY_PORT})] 'Sets the listening port'
17
         -k, --key=[KEY] 'Only allow the client with the same key'
17
         -k, --key=[KEY] 'Only allow the client with the same key'
18
         ",
18
         ",
19
-        RELAY_PORT,
20
     );
19
     );
21
     let matches = App::new("hbbr")
20
     let matches = App::new("hbbr")
22
         .version(version::VERSION)
21
         .version(version::VERSION)

+ 1 - 0
src/lib.rs

@@ -3,4 +3,5 @@ pub use rendezvous_server::*;
3
 pub mod common;
3
 pub mod common;
4
 mod database;
4
 mod database;
5
 mod peer;
5
 mod peer;
6
+#[rustfmt::skip]
6
 mod version;
7
 mod version;

+ 2 - 4
src/main.rs

@@ -15,16 +15,14 @@ fn main() -> ResultType<()> {
15
         .start()?;
15
         .start()?;
16
     let args = format!(
16
     let args = format!(
17
         "-c --config=[FILE] +takes_value 'Sets a custom config file'
17
         "-c --config=[FILE] +takes_value 'Sets a custom config file'
18
-        -p, --port=[NUMBER(default={})] 'Sets the listening port'
18
+        -p, --port=[NUMBER(default={RENDEZVOUS_PORT})] 'Sets the listening port'
19
         -s, --serial=[NUMBER(default=0)] 'Sets configure update serial number'
19
         -s, --serial=[NUMBER(default=0)] 'Sets configure update serial number'
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
-        -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={RMEM})] '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'
24
         , --mask=[MASK] 'Determine if the connection comes from LAN, e.g. 192.168.0.0/16'
24
         , --mask=[MASK] 'Determine if the connection comes from LAN, e.g. 192.168.0.0/16'
25
         -k, --key=[KEY] 'Only allow the client with the same key'",
25
         -k, --key=[KEY] 'Only allow the client with the same key'",
26
-        RENDEZVOUS_PORT,
27
-        RMEM,
28
     );
26
     );
29
     init_args(&args, "hbbs", "RustDesk ID/Rendezvous Server");
27
     init_args(&args, "hbbs", "RustDesk ID/Rendezvous Server");
30
     let port = get_arg_or("port", RENDEZVOUS_PORT.to_string()).parse::<i32>()?;
28
     let port = get_arg_or("port", RENDEZVOUS_PORT.to_string()).parse::<i32>()?;

+ 1 - 1
src/peer.rs

@@ -77,7 +77,7 @@ impl PeerMap {
77
             }
77
             }
78
             #[cfg(not(windows))]
78
             #[cfg(not(windows))]
79
             {
79
             {
80
-                db = format!("./{}", db);
80
+                db = format!("./{db}");
81
             }
81
             }
82
             db
82
             db
83
         });
83
         });

+ 1 - 1
src/relay_server.rs

@@ -240,7 +240,7 @@ async fn check_cmd(cmd: &str, limiter: Limiter) -> String {
240
                 }
240
                 }
241
             } else {
241
             } else {
242
                 unsafe {
242
                 unsafe {
243
-                    res = format!("{}\n", DOWNGRADE_THRESHOLD);
243
+                    res = format!("{DOWNGRADE_THRESHOLD}\n");
244
                 }
244
                 }
245
             }
245
             }
246
         }
246
         }

+ 1 - 1
src/rendezvous_server.rs

@@ -93,7 +93,7 @@ impl RendezvousServer {
93
     #[tokio::main(flavor = "multi_thread")]
93
     #[tokio::main(flavor = "multi_thread")]
94
     pub async fn start(port: i32, serial: i32, key: &str, rmem: usize) -> ResultType<()> {
94
     pub async fn start(port: i32, serial: i32, key: &str, rmem: usize) -> ResultType<()> {
95
         let (key, sk) = Self::get_server_sk(key);
95
         let (key, sk) = Self::get_server_sk(key);
96
-        let addr = format!("0.0.0.0:{}", port);
96
+        let addr = format!("0.0.0.0:{port}");
97
         let nat_port = port - 1;
97
         let nat_port = port - 1;
98
         let ws_port = port + 2;
98
         let ws_port = port + 2;
99
         let pm = PeerMap::new().await?;
99
         let pm = PeerMap::new().await?;

+ 8 - 9
src/utils.rs

@@ -20,7 +20,7 @@ Available Commands:
20
 }
20
 }
21
 
21
 
22
 fn error_then_help(msg: &str) {
22
 fn error_then_help(msg: &str) {
23
-    println!("ERROR: {}\n", msg);
23
+    println!("ERROR: {msg}\n");
24
     print_help();
24
     print_help();
25
 }
25
 }
26
 
26
 
@@ -74,7 +74,7 @@ fn validate_keypair(pk: &str, sk: &str) -> ResultType<()> {
74
 
74
 
75
 fn doctor_tcp(address: std::net::IpAddr, port: &str, desc: &str) {
75
 fn doctor_tcp(address: std::net::IpAddr, port: &str, desc: &str) {
76
     let start = std::time::Instant::now();
76
     let start = std::time::Instant::now();
77
-    let conn = format!("{}:{}", address, port);
77
+    let conn = format!("{address}:{port}");
78
     if let Ok(_stream) = TcpStream::connect(conn.as_str()) {
78
     if let Ok(_stream) = TcpStream::connect(conn.as_str()) {
79
         let elapsed = std::time::Instant::now().duration_since(start);
79
         let elapsed = std::time::Instant::now().duration_since(start);
80
         println!(
80
         println!(
@@ -84,12 +84,12 @@ fn doctor_tcp(address: std::net::IpAddr, port: &str, desc: &str) {
84
             elapsed.as_millis()
84
             elapsed.as_millis()
85
         );
85
         );
86
     } else {
86
     } else {
87
-        println!("TCP Port {} ({}): ERROR", port, desc);
87
+        println!("TCP Port {port} ({desc}): ERROR");
88
     }
88
     }
89
 }
89
 }
90
 
90
 
91
 fn doctor_ip(server_ip_address: std::net::IpAddr, server_address: Option<&str>) {
91
 fn doctor_ip(server_ip_address: std::net::IpAddr, server_address: Option<&str>) {
92
-    println!("\nChecking IP address: {}", server_ip_address);
92
+    println!("\nChecking IP address: {server_ip_address}");
93
     println!("Is IPV4: {}", server_ip_address.is_ipv4());
93
     println!("Is IPV4: {}", server_ip_address.is_ipv4());
94
     println!("Is IPV6: {}", server_ip_address.is_ipv6());
94
     println!("Is IPV6: {}", server_ip_address.is_ipv6());
95
 
95
 
@@ -98,11 +98,10 @@ fn doctor_ip(server_ip_address: std::net::IpAddr, server_address: Option<&str>)
98
     let reverse = lookup_addr(&server_ip_address).unwrap();
98
     let reverse = lookup_addr(&server_ip_address).unwrap();
99
     if let Some(server_address) = server_address {
99
     if let Some(server_address) = server_address {
100
         if reverse == server_address {
100
         if reverse == server_address {
101
-            println!("Reverse DNS lookup: '{}' MATCHES server address", reverse);
101
+            println!("Reverse DNS lookup: '{reverse}' MATCHES server address");
102
         } else {
102
         } else {
103
             println!(
103
             println!(
104
-                "Reverse DNS lookup: '{}' DOESN'T MATCH server address '{}'",
105
-                reverse, server_address
104
+                "Reverse DNS lookup: '{reverse}' DOESN'T MATCH server address '{server_address}'"
106
             );
105
             );
107
         }
106
         }
108
     }
107
     }
@@ -124,7 +123,7 @@ fn doctor(server_address_unclean: &str) {
124
     let server_address3 = server_address_unclean.trim();
123
     let server_address3 = server_address_unclean.trim();
125
     let server_address2 = server_address3.to_lowercase();
124
     let server_address2 = server_address3.to_lowercase();
126
     let server_address = server_address2.as_str();
125
     let server_address = server_address2.as_str();
127
-    println!("Checking server:  {}\n", server_address);
126
+    println!("Checking server:  {server_address}\n");
128
     if let Ok(server_ipaddr) = server_address.parse::<IpAddr>() {
127
     if let Ok(server_ipaddr) = server_address.parse::<IpAddr>() {
129
         // user requested an ip address
128
         // user requested an ip address
130
         doctor_ip(server_ipaddr, None);
129
         doctor_ip(server_ipaddr, None);
@@ -155,7 +154,7 @@ fn main() {
155
             }
154
             }
156
             let res = validate_keypair(args[2].as_str(), args[3].as_str());
155
             let res = validate_keypair(args[2].as_str(), args[3].as_str());
157
             if let Err(e) = res {
156
             if let Err(e) = res {
158
-                println!("{}", e);
157
+                println!("{e}");
159
                 process::exit(0x0001);
158
                 process::exit(0x0001);
160
             }
159
             }
161
             println!("Key pair is VALID");
160
             println!("Key pair is VALID");