weyne/rustdesk
1
0
Fork 0
mirror of https://github.com/weyne85/rustdesk.git synced 2025-10-29 17:00:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

use more accurate regex for {domain}:{port}

Browse Source
This commit is contained in:
qiushiyang
2023-01-18 06:08:46 +00:00
parent 12d446b217
commit aa2cd37fb3
2 changed files with 29 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
libs/hbb_common/src/lib.rs
View File

@@ -322,10 +322,15 @@ pub fn is_ip_str(id: &str) -> bool {
}
#[inline]
pub fn is_hostname_port_str(id: &str) -> bool {
regex::Regex::new(r"^[\-.0-9a-zA-Z]+:\d{1,5}$")
.unwrap()
.is_match(id)
pub fn is_domain_port_str(id: &str) -> bool {
// modified regex for RFC1123 hostname. check https://stackoverflow.com/a/106223 for original version for hostname.
// according to [TLD List](https://data.iana.org/TLD/tlds-alpha-by-domain.txt) version 2023011700,
// there is no digits in TLD, and length is 2~63.
regex::Regex::new(
r"(?i)^([a-z0-9]([a-z0-9-]{0,61}[a-z0-9])?\.)+[a-z][a-z-]{0,61}[a-z]:\d{1,5}$",
)
.unwrap()
.is_match(id)
}
#[cfg(test)]
@@ -350,17 +355,22 @@ mod test_lib {
#[test]
fn test_hostname_port() {
assert_eq!(is_hostname_port_str("a:12"), true);
assert_eq!(is_hostname_port_str("a.b.c:12"), true);
assert_eq!(is_hostname_port_str("test.com:12"), true);
assert_eq!(is_hostname_port_str("1.2.3:12"), true);
assert_eq!(is_hostname_port_str("a.b.c:123456"), false);
// todo: should we also check for these edge case?
assert_eq!(is_domain_port_str("a:12"), false);
assert_eq!(is_domain_port_str("a.b.c:12"), false);
assert_eq!(is_domain_port_str("test.com:12"), true);
assert_eq!(is_domain_port_str("test-UPPER.com:12"), true);
assert_eq!(is_domain_port_str("some-other.domain.com:12"), true);
assert_eq!(is_domain_port_str("under_score:12"), false);
assert_eq!(is_domain_port_str("a@bc:12"), false);
assert_eq!(is_domain_port_str("1.1.1.1:12"), false);
assert_eq!(is_domain_port_str("1.2.3:12"), false);
assert_eq!(is_domain_port_str("1.2.3.45:12"), false);
assert_eq!(is_domain_port_str("a.b.c:123456"), false);
assert_eq!(is_domain_port_str("---:12"), false);
assert_eq!(is_domain_port_str(".:12"), false);
// todo: should we also check for these edge cases?
// out-of-range port
assert_eq!(is_hostname_port_str("test.com:0"), true);
assert_eq!(is_hostname_port_str("test.com:98989"), true);
// invalid hostname
assert_eq!(is_hostname_port_str("---:12"), true);
assert_eq!(is_hostname_port_str(".:12"), true);
assert_eq!(is_domain_port_str("test.com:0"), true);
assert_eq!(is_domain_port_str("test.com:98989"), true);
}
}