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 8
 .vscode
9 9
 src/version.rs
10 10
 db_v2.sqlite3
11
+test.*

+ 1 - 1
libs/hbb_common/build.rs

@@ -2,7 +2,7 @@ fn main() {
2 2
     let out_dir = format!("{}/protos", std::env::var("OUT_DIR").unwrap());
3 3
 
4 4
     std::fs::create_dir_all(&out_dir).unwrap();
5
-    
5
+
6 6
     protobuf_codegen::Codegen::new()
7 7
         .pure()
8 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 135
             (pk, sk) = gen_func();
136 136
         }
137
-        let pub_file = format!("{}.pub", sk_file);
137
+        let pub_file = format!("{sk_file}.pub");
138 138
         if let Ok(mut f) = std::fs::File::create(&pub_file) {
139 139
             f.write_all(pk.as_bytes()).ok();
140 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 13
         .write_mode(WriteMode::Async)
14 14
         .start()?;
15 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 17
         -k, --key=[KEY] 'Only allow the client with the same key'
18 18
         ",
19
-        RELAY_PORT,
20 19
     );
21 20
     let matches = App::new("hbbr")
22 21
         .version(version::VERSION)

+ 1 - 0
src/lib.rs

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

+ 2 - 4
src/main.rs

@@ -15,16 +15,14 @@ fn main() -> ResultType<()> {
15 15
         .start()?;
16 16
     let args = format!(
17 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 19
         -s, --serial=[NUMBER(default=0)] 'Sets configure update serial number'
20 20
         -R, --rendezvous-servers=[HOSTS] 'Sets rendezvous servers, seperated by colon'
21 21
         -u, --software-url=[URL] 'Sets download url of RustDesk software of newest version'
22 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 24
         , --mask=[MASK] 'Determine if the connection comes from LAN, e.g. 192.168.0.0/16'
25 25
         -k, --key=[KEY] 'Only allow the client with the same key'",
26
-        RENDEZVOUS_PORT,
27
-        RMEM,
28 26
     );
29 27
     init_args(&args, "hbbs", "RustDesk ID/Rendezvous Server");
30 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 78
             #[cfg(not(windows))]
79 79
             {
80
-                db = format!("./{}", db);
80
+                db = format!("./{db}");
81 81
             }
82 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 241
             } else {
242 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 93
     #[tokio::main(flavor = "multi_thread")]
94 94
     pub async fn start(port: i32, serial: i32, key: &str, rmem: usize) -> ResultType<()> {
95 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 97
         let nat_port = port - 1;
98 98
         let ws_port = port + 2;
99 99
         let pm = PeerMap::new().await?;

+ 8 - 9
src/utils.rs

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