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: create uri_identifier module in parser

Browse Source
This commit is contained in:
Keivan-sf
2023-10-10 15:01:12 +03:30
parent 3d114db44b
commit fc65da6458
3 changed files with 83 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/main.rs
View File

@@ -1,18 +1,5 @@
use std::process::exit;
mod parser;
fn main() {
let protocol = parser::get_uri_protocol("vmess://");
match protocol {
Some(parser::Protocols::Vless) => {
println!("The protocol was Vless");
}
Some(_) => {
println!("Some recognizable protocol")
}
None => {
println!("The protocol is not supported");
exit(0);
}
}
parser::parse("vmess://test");
}

78
src/parser/mod.rs
View File

@@ -1,66 +1,18 @@
use regex::Regex;
use std::process::exit;
mod uri_identifier;
pub enum Protocols {
Vmess,
Vless,
Shadowsocks,
Trojan,
Socks,
Http,
}
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);
}
if uri.starts_with("vless://") {
return Some(Protocols::Vless);
}
if uri.starts_with("ss://") {
return Some(Protocols::Shadowsocks);
}
if uri.starts_with("socks://") {
return Some(Protocols::Socks);
}
if uri.starts_with("http://") {
return Some(Protocols::Http);
}
if uri.starts_with("trojan://") {
return Some(Protocols::Trojan);
}
return None;
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn return_none_for_invalid_uri() {
let protocol = get_uri_protocol("123-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");
assert!(matches!(protocol, None));
}
#[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));
}
#[test]
fn recognize_vmess_protocol() {
let protocol = get_uri_protocol("vmess://eyJhZGQiOiIxMjcuMC4wLjEiLCJhaWQiOiIwIiwiaG9zdCI6IiIsImlkIjoiOHM2OTdlMmMtZXMxNy00MDNkLTI0ZjMtZHMyYzYwc2I4ZjUiLCJuZXQiOiJ0Y3AiLCJwYXRoIjoiIiwicG9ydCI6IjgwODAiLCJwcyI6InRlc3QiLCJzY3kiOiJhdXRvIiwic25pIjoiIiwidGxzIjoiIiwidHlwZSI6Im5vbmUiLCJ2IjoiMiJ9").unwrap();
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));
}
#[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));
pub fn parse(uri: &str) {
let protocol = uri_identifier::get_uri_protocol(uri);
match protocol {
Some(uri_identifier::Protocols::Vmess) => {
println!("The protocol was vmess");
}
Some(_) => {
println!("The protocol was recognized");
}
None => {
println!("The protcol is not supported");
exit(0);
}
}
}

66
src/parser/uri_identifier.rs Normal file
View File

@@ -0,0 +1,66 @@
use regex::Regex;
pub enum Protocols {
Vmess,
Vless,
Shadowsocks,
Trojan,
Socks,
Http,
}
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);
}
if uri.starts_with("vless://") {
return Some(Protocols::Vless);
}
if uri.starts_with("ss://") {
return Some(Protocols::Shadowsocks);
}
if uri.starts_with("socks://") {
return Some(Protocols::Socks);
}
if uri.starts_with("http://") {
return Some(Protocols::Http);
}
if uri.starts_with("trojan://") {
return Some(Protocols::Trojan);
}
return None;
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn return_none_for_invalid_uri() {
let protocol = get_uri_protocol("123-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");
assert!(matches!(protocol, None));
}
#[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));
}
#[test]
fn recognize_vmess_protocol() {
let protocol = get_uri_protocol("vmess://eyJhZGQiOiIxMjcuMC4wLjEiLCJhaWQiOiIwIiwiaG9zdCI6IiIsImlkIjoiOHM2OTdlMmMtZXMxNy00MDNkLTI0ZjMtZHMyYzYwc2I4ZjUiLCJuZXQiOiJ0Y3AiLCJwYXRoIjoiIiwicG9ydCI6IjgwODAiLCJwcyI6InRlc3QiLCJzY3kiOiJhdXRvIiwic25pIjoiIiwidGxzIjoiIiwidHlwZSI6Im5vbmUiLCJ2IjoiMiJ9").unwrap();
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));
}
#[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));
}
}