mirror/rustdesk
1
0
Fork 0
mirror of https://github.com/rustdesk/rustdesk.git synced 2024-11-27 14:59:02 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix get_custom_server_from_string, relay= is not used (#9069)

Browse Source 
Signed-off-by: 21pages <sunboeasy@gmail.com>
This commit is contained in:
21pages 2024-08-15 09:57:12 +08:00 committed by GitHub
parent cc9b7e64eb
commit 85ae3916cb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 28 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/custom_server.rs
View File

@ -59,30 +59,30 @@ pub fn get_custom_server_from_string(s: &str) -> ResultType<CustomServer> {
if s.contains("host=") {
let stripped = &s[s.find("host=").unwrap_or(0)..s.len()];
let strs: Vec<&str> = stripped.split(",").collect();
let mut host = "";
let mut key = "";
let mut api = "";
let mut relay = "";
let mut host = String::default();
let mut key = String::default();
let mut api = String::default();
let mut relay = String::default();
let strs_iter = strs.iter();
for el in strs_iter {
if el.starts_with("host=") {
host = &el[5..el.len()];
host = el.chars().skip(5).collect();
}
if el.starts_with("key=") {
key = &el[4..el.len()];
key = el.chars().skip(4).collect();
}
if el.starts_with("api=") {
api = &el[4..el.len()];
api = el.chars().skip(4).collect();
}
if el.starts_with("relay=") {
relay = &el[4..el.len()];
relay = el.chars().skip(6).collect();
}
}
return Ok(CustomServer {
host: host.to_owned(),
key: key.to_owned(),
api: api.to_owned(),
relay: relay.to_owned(),
host,
key,
api,
relay,
});
} else {
let s = s
@ -146,8 +146,10 @@ mod test {
}
);
assert_eq!(
get_custom_server_from_string("rustdesk-host=server.example.net,key=Zm9vYmFyLiwyCg==,.exe")
.unwrap(),
get_custom_server_from_string(
"rustdesk-host=server.example.net,key=Zm9vYmFyLiwyCg==,.exe"
)
.unwrap(),
CustomServer {
host: "server.example.net".to_owned(),
key: "Zm9vYmFyLiwyCg==".to_owned(),
@ -155,6 +157,18 @@ mod test {
relay: "".to_owned(),
}
);
assert_eq!(
get_custom_server_from_string(
"rustdesk-host=server.example.net,key=Zm9vYmFyLiwyCg==,relay=server.example.net.exe"
)
.unwrap(),
CustomServer {
host: "server.example.net".to_owned(),
key: "Zm9vYmFyLiwyCg==".to_owned(),
api: "".to_owned(),
relay: "server.example.net".to_owned(),
}
);
let lic = CustomServer {
host: "1.1.1.1".to_owned(),
key: "5Qbwsde3unUcJBtrx9ZkvUmwFNoExHzpryHuPUdqlWM=".to_owned(),