ab/v2ray-proxy
1
0
Fork 0
mirror of https://github.com/house-of-vanity/v2-uri-parser.git synced 2025-12-15 22:47:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: resolve rust warning for protocols

Browse Source
This commit is contained in:
Keivan-sf
2023-10-09 18:05:39 +03:30
parent e4db6a92af
commit 3d114db44b
2 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main.rs
View File

@@ -4,7 +4,7 @@ mod parser;
fn main() {
let protocol = parser::get_uri_protocol("vmess://");
match protocol {
Some(parser::protocols::Vless) => {
Some(parser::Protocols::Vless) => {
println!("The protocol was Vless");
}
Some(_) => {

24
src/parser/mod.rs
View File

@@ -1,6 +1,6 @@
use regex::Regex;
pub enum protocols {
pub enum Protocols {
Vmess,
Vless,
Shadowsocks,
@@ -9,28 +9,28 @@ pub enum protocols {
Http,
}
pub fn get_uri_protocol(uri: &str) -> Option<protocols> {
pub fn get_uri_protocol(uri: &str) -> Option<Protocols> {
let uri_regex = Regex::new(r"^[a-z]+:\/\/.+$").unwrap();
if !uri_regex.is_match(uri) {
return None;
}
if uri.starts_with("vmess://") {
return Some(protocols::Vmess);
return Some(Protocols::Vmess);
}
if uri.starts_with("vless://") {
return Some(protocols::Vless);
return Some(Protocols::Vless);
}
if uri.starts_with("ss://") {
return Some(protocols::Shadowsocks);
return Some(Protocols::Shadowsocks);
}
if uri.starts_with("socks://") {
return Some(protocols::Socks);
return Some(Protocols::Socks);
}
if uri.starts_with("http://") {
return Some(protocols::Http);
return Some(Protocols::Http);
}
if uri.starts_with("trojan://") {
return Some(protocols::Trojan);
return Some(Protocols::Trojan);
}
return None;
}
@@ -46,21 +46,21 @@ mod tests {
#[test]
fn recognize_vless_protocol() {
let protocol = get_uri_protocol("vless://3d1c3f04-729d-59d3-bdb6-3f3f4352e173@root.ii.one:2083?security=reality&sni=www.spamhaus.org&fp=safari&pbk=7xhH4b_VkliBxGulljcyPOH-bYUA2dl-XAdZAsfhk04&sid=6ba85179e30d4fc2&type=tcp&flow=xtls-rprx-vision#Ha-ac").unwrap();
assert!(matches!(protocol, protocols::Vless));
assert!(matches!(protocol, Protocols::Vless));
}
#[test]
fn recognize_vmess_protocol() {
let protocol = get_uri_protocol("vmess://eyJhZGQiOiIxMjcuMC4wLjEiLCJhaWQiOiIwIiwiaG9zdCI6IiIsImlkIjoiOHM2OTdlMmMtZXMxNy00MDNkLTI0ZjMtZHMyYzYwc2I4ZjUiLCJuZXQiOiJ0Y3AiLCJwYXRoIjoiIiwicG9ydCI6IjgwODAiLCJwcyI6InRlc3QiLCJzY3kiOiJhdXRvIiwic25pIjoiIiwidGxzIjoiIiwidHlwZSI6Im5vbmUiLCJ2IjoiMiJ9").unwrap();
assert!(matches!(protocol, protocols::Vmess));
assert!(matches!(protocol, Protocols::Vmess));
}
#[test]
fn recognize_shadowsocks_protocol() {
let protocol = get_uri_protocol("ss://Y2hhY2hhMjAtaWV0Zi1wb2x5MTMwNTpXNzRYRkFMS0t1dzZtNUlB@www.outline.aasf.cyou:443#test").unwrap();
assert!(matches!(protocol, protocols::Shadowsocks));
assert!(matches!(protocol, Protocols::Shadowsocks));
}
#[test]
fn recognize_trojan_protocol() {
let protocol = get_uri_protocol("trojan://test-pw@13.50.100.84:22222?security=tls&sni=trj.rollingnext.co.uk&type=tcp#test").unwrap();
assert!(matches!(protocol, protocols::Trojan));
assert!(matches!(protocol, Protocols::Trojan));
}
}