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

feat: parse all vless data in a single function

Browse Source
This commit is contained in:
Keivan-sf
2023-10-11 21:07:43 +03:30
parent 497300e23c
commit 8898654b6f
1 changed files with 27 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
src/parser/vless.rs
View File

@@ -2,8 +2,7 @@ use std::process::exit;
use querystring; use querystring;
#[derive(PartialEq, Eq)] struct VlessQuery {
pub struct VlessQuery {
security: String, security: String,
sni: String, sni: String,
fp: String, fp: String,
@@ -24,13 +23,18 @@ pub struct VlessQuery {
spx: String, spx: String,
} }
pub struct VlessAddress { struct VlessAddress {
uuid: String, uuid: String,
address: String, address: String,
port: u16, port: u16,
} }
pub fn get_vless_data(uri: &str) { pub struct VlessData {
query: VlessQuery,
address_data: VlessAddress,
}
pub fn get_vless_data(uri: &str) -> VlessData {
let data = uri.split_once("vless://").unwrap().1; let data = uri.split_once("vless://").unwrap().1;
let query_and_name = uri.split_once("?").unwrap().1; let query_and_name = uri.split_once("?").unwrap().1;
let query = query_and_name let query = query_and_name
@@ -38,10 +42,14 @@ pub fn get_vless_data(uri: &str) {
.unwrap_or((query_and_name, "")) .unwrap_or((query_and_name, ""))
.0; .0;
let parsed_query = parse_vless_query(query); let parsed_query = parse_vless_query(query);
println!("{0}", parsed_query.flow); let parsed_address = parse_vless_address(data.split_once("?").unwrap().0);
return VlessData {
query: parsed_query,
address_data: parsed_address,
};
} }
pub fn parse_vless_address(raw_data: &str) -> VlessAddress { fn parse_vless_address(raw_data: &str) -> VlessAddress {
let (uuid, raw_address): (String, &str) = match raw_data.split_once("@") { let (uuid, raw_address): (String, &str) = match raw_data.split_once("@") {
None => { None => {
println!("Wrong vless format, no `@` found in the address"); println!("Wrong vless format, no `@` found in the address");
@@ -68,7 +76,7 @@ pub fn parse_vless_address(raw_data: &str) -> VlessAddress {
}; };
} }
pub fn parse_vless_query(raw_query: &str) -> VlessQuery { fn parse_vless_query(raw_query: &str) -> VlessQuery {
let query: Vec<(&str, &str)> = querystring::querify(raw_query); let query: Vec<(&str, &str)> = querystring::querify(raw_query);
let a = VlessQuery { let a = VlessQuery {
@@ -108,9 +116,19 @@ mod tests {
use super::*; use super::*;
#[test] #[test]
fn vless_test() { fn vless_test() {
let v = "vless://4d2c3e35-749d-52e3-bdb6-3f3f4950c183@tre.test.one:2053?security=reality&sni=bench.sh&fp=chrome&pbk=7xhH4b_VkliBxGulljcyPOH-bYUA2dl-XAdZAsfhk04&sid=6ba85179e30d4fc2&type=tcp&flow=xtls-rprx-vision#test-name"; let v = "vless://4d2c3e35-749d-52e3-bdb6-3f3f4950c183@tre.test.one:2053?security=reality&type=tcp&flow=xtls-rprx-vision#test-name";
get_vless_data(v); let data = get_vless_data(v);
assert_eq!(data.address_data.address, "tre.test.one");
assert_eq!(
data.address_data.uuid,
"4d2c3e35-749d-52e3-bdb6-3f3f4950c183"
);
assert_eq!(data.address_data.port, 2053);
assert_eq!(data.query.flow, "xtls-rprx-vision");
assert_eq!(data.query.security, "reality");
assert_eq!(data.query.r#type, "tcp");
} }
#[test] #[test]
fn parse_vless_query_data() { fn parse_vless_query_data() {
let query = "security=reality&sni=bench.sh&fp=chrome&pbk=7xhH4b_VkliBxGulljcyPOH-bYUA2dl-XAdZAsfhk04&sid=6ba85179e30d4fc2&type=tcp&flow=xtls-rprx-vision&path=/"; let query = "security=reality&sni=bench.sh&fp=chrome&pbk=7xhH4b_VkliBxGulljcyPOH-bYUA2dl-XAdZAsfhk04&sid=6ba85179e30d4fc2&type=tcp&flow=xtls-rprx-vision&path=/";