diff --git a/src/parser/mod.rs b/src/parser/mod.rs index 07a61fa..d4fb81d 100644 --- a/src/parser/mod.rs +++ b/src/parser/mod.rs @@ -28,7 +28,7 @@ pub fn create_outbound_object(uri: &str) -> config_models::Outbound { let protocol = uri_identifier::get_uri_protocol(uri); match protocol { Some(uri_identifier::Protocols::Vless) => { - let vless_data = vless::get_vless_data(uri); + let vless_data = vless::data::get_data(uri); let outbound_object = vless::create_outbound_object(vless_data); return outbound_object; } diff --git a/src/parser/vless/data.rs b/src/parser/vless/data.rs new file mode 100644 index 0000000..bd03a36 --- /dev/null +++ b/src/parser/vless/data.rs @@ -0,0 +1,183 @@ +use crate::parser::vless::models; +use crate::utils::{get_parameter_value, url_decode}; +use http::Uri; + +pub fn get_data(uri: &str) -> models::VlessData { + let data = uri.split_once("vless://").unwrap().1; + let query_and_name = uri.split_once("?").unwrap().1; + let query = query_and_name + .split_once("#") + .unwrap_or((query_and_name, "")) + .0; + let parsed_query = parse_vless_query(query); + let parsed_address = parse_vless_address(data.split_once("?").unwrap().0); + return models::VlessData { + query: parsed_query, + address_data: parsed_address, + }; +} + +fn parse_vless_address(raw_data: &str) -> models::VlessAddress { + let (uuid, raw_address): (String, &str) = match raw_data.split_once("@") { + None => { + panic!("Wrong vless format, no `@` found in the address"); + } + Some(data) => (String::from(data.0), data.1), + }; + let address_wo_slash = raw_address.strip_suffix("/").unwrap_or(raw_address); + + let parsed = address_wo_slash.parse::().unwrap(); + + return models::VlessAddress { + uuid: url_decode(Some(uuid)).unwrap(), + address: parsed.host().unwrap().to_string(), + port: parsed.port().unwrap().as_u16(), + }; +} + +fn parse_vless_query(raw_query: &str) -> models::VlessQuery { + let query: Vec<(&str, &str)> = querystring::querify(raw_query); + + let a = models::VlessQuery { + alpn: url_decode(get_parameter_value(&query, "alpn")), + path: url_decode(get_parameter_value(&query, "path")), + authority: url_decode(get_parameter_value(&query, "authority")), + pbk: url_decode(get_parameter_value(&query, "pbk")), + security: get_parameter_value(&query, "security"), + sid: url_decode(get_parameter_value(&query, "sid")), + flow: get_parameter_value(&query, "flow"), + sni: get_parameter_value(&query, "sni"), + fp: url_decode(get_parameter_value(&query, "fp")), + r#type: get_parameter_value(&query, "type"), + encryption: get_parameter_value(&query, "encryption"), + header_type: get_parameter_value(&query, "headerType"), + host: url_decode(get_parameter_value(&query, "host")), + seed: url_decode(get_parameter_value(&query, "seed")), + quic_security: get_parameter_value(&query, "quicSecurity"), + key: get_parameter_value(&query, "key"), + mode: url_decode(get_parameter_value(&query, "mode")), + service_name: url_decode(get_parameter_value(&query, "serviceName")), + slpn: get_parameter_value(&query, "slpn"), + spx: url_decode(get_parameter_value(&query, "spx")), + extra: url_decode(get_parameter_value(&query, "extra")), + }; + return a; +} + +#[cfg(test)] +mod tests { + use crate::parser::vless::create_outbound_object; + + use super::*; + + #[test] + fn vless_tcp_header_test() { + let v = "vless://1010501a-ca9a-479c-84d0-1308d97789b5@104.21.25.109:443?security=reality&alpn=http%2F1.1&encryption=none&pbk=testpub&host=mehr14-n.gowow31220.workers.dev&headerType=http&fp=chrome&spx=spideex&type=tcp&flow=xtls-rprx-vision&sni=mehr14-iran-mehr14-iran-mehr14-iran-mehr14-iran-mehr14-iran.gowow31220.workers.dev&sid=testshort#%E2%AD%90%EF%B8%8F%20Telegram%20%3D%20%40z_v2ray"; + let data = create_outbound_object(get_data(v)); + assert_eq!( + data.streamSettings + .tcpSettings + .unwrap() + .header + .unwrap() + .r#type, + Some(String::from("http")) + ) + } + + #[test] + fn vless_test_2() { + let v = "vless://2dc56709-sdfs-sdfs-2234-128904@nwarne.fast-ip.com:80/?type=ws&encryption=none&host=Shuposipet.com&path=%2Fde%3Fed%3D1048#[test]@test"; + let data = get_data(v); + assert_eq!(data.address_data.address, "nwarne.fast-ip.com"); + assert_eq!(data.address_data.uuid, "2dc56709-sdfs-sdfs-2234-128904"); + assert_eq!(data.address_data.port, 80); + assert_eq!(data.query.encryption.unwrap(), "none"); + assert_eq!(data.query.r#type.unwrap(), "ws"); + } + + #[test] + fn vless_test() { + let v = "vless://4d2c3e35-749d-52e3-bdb6-3f3f4950c183@tre.test.one:2053?security=reality&type=tcp&flow=xtls-rprx-vision#test-name"; + let data = get_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.unwrap(), "xtls-rprx-vision"); + assert_eq!(data.query.security.unwrap(), "reality"); + assert_eq!(data.query.r#type.unwrap(), "tcp"); + } + + #[test] + 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&alpn=http%2F1.1&path=/"; + let parsed_query = parse_vless_query(query); + assert_eq!(parsed_query.sni.unwrap(), "bench.sh"); + assert_eq!(parsed_query.security.unwrap(), "reality"); + assert_eq!(parsed_query.fp.unwrap(), "chrome"); + assert_eq!( + parsed_query.pbk.unwrap(), + "7xhH4b_VkliBxGulljcyPOH-bYUA2dl-XAdZAsfhk04" + ); + assert_eq!(parsed_query.sid.unwrap(), "6ba85179e30d4fc2"); + assert_eq!(parsed_query.r#type.unwrap(), "tcp"); + assert_eq!(parsed_query.flow.unwrap(), "xtls-rprx-vision"); + assert_eq!(parsed_query.alpn.unwrap(), "http/1.1"); + assert_eq!(parsed_query.path.unwrap(), "/"); + assert_eq!(parsed_query.encryption, None); + assert_eq!(parsed_query.header_type, None); + assert_eq!(parsed_query.host, None); + assert_eq!(parsed_query.seed, None); + assert_eq!(parsed_query.quic_security, None); + assert_eq!(parsed_query.key, None); + assert_eq!(parsed_query.mode, None); + assert_eq!(parsed_query.service_name, None); + assert_eq!(parsed_query.slpn, None); + assert_eq!(parsed_query.spx, None); + } + + #[test] + fn parse_vless_query_with_defaults() { + let query = ""; + let parsed_query = parse_vless_query(query); + assert_eq!(parsed_query.sni, None); + assert_eq!(parsed_query.security, None); + assert_eq!(parsed_query.fp, None); + assert_eq!(parsed_query.pbk, None); + assert_eq!(parsed_query.sid, None); + assert_eq!(parsed_query.r#type, None); + assert_eq!(parsed_query.r#flow, None); + assert_eq!(parsed_query.path, None); + assert_eq!(parsed_query.encryption, None); + assert_eq!(parsed_query.header_type, None); + assert_eq!(parsed_query.host, None); + assert_eq!(parsed_query.seed, None); + assert_eq!(parsed_query.quic_security, None); + assert_eq!(parsed_query.key, None); + assert_eq!(parsed_query.mode, None); + assert_eq!(parsed_query.service_name, None); + assert_eq!(parsed_query.slpn, None); + assert_eq!(parsed_query.spx, None); + } + + #[test] + fn parse_vless_host() { + let raw_host = "uu0id@127.0.0.1:3012"; + let parsed = parse_vless_address(raw_host); + assert_eq!(parsed.address, "127.0.0.1"); + assert_eq!(parsed.port, 3012); + assert_eq!(parsed.uuid, "uu0id"); + } + + #[test] + fn parse_vless_ipv6_host() { + let raw_host = "uu0id@[2a06:98c1:3120::1]:443"; + let parsed = parse_vless_address(raw_host); + assert_eq!(parsed.port, 443); + assert_eq!(parsed.address, "[2a06:98c1:3120::1]"); + assert_eq!(parsed.uuid, "uu0id"); + } +} diff --git a/src/parser/vless/mod.rs b/src/parser/vless/mod.rs index 51daefa..1ac1b00 100644 --- a/src/parser/vless/mod.rs +++ b/src/parser/vless/mod.rs @@ -1,8 +1,6 @@ -use querystring; +pub mod data; mod models; -use crate::config_models::*; -use http::Uri; -use std::process::exit; +use crate::{config_models::*, utils::parse_raw_json}; pub fn create_outbound_object(data: models::VlessData) -> Outbound { let network_type = data.query.r#type.clone().unwrap_or(String::from("")); @@ -118,206 +116,3 @@ pub fn create_outbound_object(data: models::VlessData) -> Outbound { }), }; } - -pub fn get_vless_data(uri: &str) -> models::VlessData { - let data = uri.split_once("vless://").unwrap().1; - let query_and_name = uri.split_once("?").unwrap().1; - let query = query_and_name - .split_once("#") - .unwrap_or((query_and_name, "")) - .0; - let parsed_query = parse_vless_query(query); - let parsed_address = parse_vless_address(data.split_once("?").unwrap().0); - return models::VlessData { - query: parsed_query, - address_data: parsed_address, - }; -} - -fn parse_vless_address(raw_data: &str) -> models::VlessAddress { - let (uuid, raw_address): (String, &str) = match raw_data.split_once("@") { - None => { - println!("Wrong vless format, no `@` found in the address"); - exit(0); - } - Some(data) => (String::from(data.0), data.1), - }; - let address_wo_slash = raw_address.strip_suffix("/").unwrap_or(raw_address); - - let parsed = address_wo_slash.parse::().unwrap(); - - return models::VlessAddress { - uuid: url_decode(Some(uuid)).unwrap(), - address: parsed.host().unwrap().to_string(), - port: parsed.port().unwrap().as_u16(), - }; -} - -fn parse_vless_query(raw_query: &str) -> models::VlessQuery { - let query: Vec<(&str, &str)> = querystring::querify(raw_query); - - let a = models::VlessQuery { - alpn: url_decode(get_parameter_value(&query, "alpn")), - path: url_decode(get_parameter_value(&query, "path")), - authority: url_decode(get_parameter_value(&query, "authority")), - pbk: url_decode(get_parameter_value(&query, "pbk")), - security: get_parameter_value(&query, "security"), - sid: url_decode(get_parameter_value(&query, "sid")), - flow: get_parameter_value(&query, "flow"), - sni: get_parameter_value(&query, "sni"), - fp: url_decode(get_parameter_value(&query, "fp")), - r#type: get_parameter_value(&query, "type"), - encryption: get_parameter_value(&query, "encryption"), - header_type: get_parameter_value(&query, "headerType"), - host: url_decode(get_parameter_value(&query, "host")), - seed: url_decode(get_parameter_value(&query, "seed")), - quic_security: get_parameter_value(&query, "quicSecurity"), - key: get_parameter_value(&query, "key"), - mode: url_decode(get_parameter_value(&query, "mode")), - service_name: url_decode(get_parameter_value(&query, "serviceName")), - slpn: get_parameter_value(&query, "slpn"), - spx: url_decode(get_parameter_value(&query, "spx")), - extra: url_decode(get_parameter_value(&query, "extra")), - }; - return a; -} - -fn url_decode(value: Option) -> Option { - return value.and_then(|s| { - urlencoding::decode(&s) - .ok() - .map(|decoded| decoded.into_owned()) - }); -} - -fn parse_raw_json(input: &str) -> Option { - serde_json::from_str::(input) - .ok() - .and_then(|v| match v { - serde_json::Value::Object(_) => Some(v), - _ => None, - }) -} - -fn get_parameter_value(query: &Vec<(&str, &str)>, param: &str) -> Option { - return query - .iter() - .find(|q| String::from(q.0) == String::from(param)) - .map(|q| q.1.to_string()); -} - -#[cfg(test)] -mod tests { - use super::*; - - #[test] - fn vless_tcp_header_test() { - let v = "vless://1010501a-ca9a-479c-84d0-1308d97789b5@104.21.25.109:443?security=reality&alpn=http%2F1.1&encryption=none&pbk=testpub&host=mehr14-n.gowow31220.workers.dev&headerType=http&fp=chrome&spx=spideex&type=tcp&flow=xtls-rprx-vision&sni=mehr14-iran-mehr14-iran-mehr14-iran-mehr14-iran-mehr14-iran.gowow31220.workers.dev&sid=testshort#%E2%AD%90%EF%B8%8F%20Telegram%20%3D%20%40z_v2ray"; - let data = create_outbound_object(get_vless_data(v)); - assert_eq!( - data.streamSettings - .tcpSettings - .unwrap() - .header - .unwrap() - .r#type, - Some(String::from("http")) - ) - } - - #[test] - fn vless_test_2() { - let v = "vless://2dc56709-sdfs-sdfs-2234-128904@nwarne.fast-ip.com:80/?type=ws&encryption=none&host=Shuposipet.com&path=%2Fde%3Fed%3D1048#[test]@test"; - let data = get_vless_data(v); - assert_eq!(data.address_data.address, "nwarne.fast-ip.com"); - assert_eq!(data.address_data.uuid, "2dc56709-sdfs-sdfs-2234-128904"); - assert_eq!(data.address_data.port, 80); - assert_eq!(data.query.encryption.unwrap(), "none"); - assert_eq!(data.query.r#type.unwrap(), "ws"); - } - - #[test] - fn vless_test() { - let v = "vless://4d2c3e35-749d-52e3-bdb6-3f3f4950c183@tre.test.one:2053?security=reality&type=tcp&flow=xtls-rprx-vision#test-name"; - 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.unwrap(), "xtls-rprx-vision"); - assert_eq!(data.query.security.unwrap(), "reality"); - assert_eq!(data.query.r#type.unwrap(), "tcp"); - } - - #[test] - 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&alpn=http%2F1.1&path=/"; - let parsed_query = parse_vless_query(query); - assert_eq!(parsed_query.sni.unwrap(), "bench.sh"); - assert_eq!(parsed_query.security.unwrap(), "reality"); - assert_eq!(parsed_query.fp.unwrap(), "chrome"); - assert_eq!( - parsed_query.pbk.unwrap(), - "7xhH4b_VkliBxGulljcyPOH-bYUA2dl-XAdZAsfhk04" - ); - assert_eq!(parsed_query.sid.unwrap(), "6ba85179e30d4fc2"); - assert_eq!(parsed_query.r#type.unwrap(), "tcp"); - assert_eq!(parsed_query.flow.unwrap(), "xtls-rprx-vision"); - assert_eq!(parsed_query.alpn.unwrap(), "http/1.1"); - assert_eq!(parsed_query.path.unwrap(), "/"); - assert_eq!(parsed_query.encryption, None); - assert_eq!(parsed_query.header_type, None); - assert_eq!(parsed_query.host, None); - assert_eq!(parsed_query.seed, None); - assert_eq!(parsed_query.quic_security, None); - assert_eq!(parsed_query.key, None); - assert_eq!(parsed_query.mode, None); - assert_eq!(parsed_query.service_name, None); - assert_eq!(parsed_query.slpn, None); - assert_eq!(parsed_query.spx, None); - } - - #[test] - fn parse_vless_query_with_defaults() { - let query = ""; - let parsed_query = parse_vless_query(query); - assert_eq!(parsed_query.sni, None); - assert_eq!(parsed_query.security, None); - assert_eq!(parsed_query.fp, None); - assert_eq!(parsed_query.pbk, None); - assert_eq!(parsed_query.sid, None); - assert_eq!(parsed_query.r#type, None); - assert_eq!(parsed_query.r#flow, None); - assert_eq!(parsed_query.path, None); - assert_eq!(parsed_query.encryption, None); - assert_eq!(parsed_query.header_type, None); - assert_eq!(parsed_query.host, None); - assert_eq!(parsed_query.seed, None); - assert_eq!(parsed_query.quic_security, None); - assert_eq!(parsed_query.key, None); - assert_eq!(parsed_query.mode, None); - assert_eq!(parsed_query.service_name, None); - assert_eq!(parsed_query.slpn, None); - assert_eq!(parsed_query.spx, None); - } - - #[test] - fn parse_vless_host() { - let raw_host = "uu0id@127.0.0.1:3012"; - let parsed = parse_vless_address(raw_host); - assert_eq!(parsed.address, "127.0.0.1"); - assert_eq!(parsed.port, 3012); - assert_eq!(parsed.uuid, "uu0id"); - } - - #[test] - fn parse_vless_ipv6_host() { - let raw_host = "uu0id@[2a06:98c1:3120::1]:443"; - let parsed = parse_vless_address(raw_host); - assert_eq!(parsed.port, 443); - assert_eq!(parsed.address, "[2a06:98c1:3120::1]"); - assert_eq!(parsed.uuid, "uu0id"); - } -} diff --git a/src/utils/mod.rs b/src/utils/mod.rs index 62e230a..e40bd46 100644 --- a/src/utils/mod.rs +++ b/src/utils/mod.rs @@ -1 +1,25 @@ pub mod inbound_generator; + +pub fn url_decode(value: Option) -> Option { + return value.and_then(|s| { + urlencoding::decode(&s) + .ok() + .map(|decoded| decoded.into_owned()) + }); +} + +pub fn parse_raw_json(input: &str) -> Option { + serde_json::from_str::(input) + .ok() + .and_then(|v| match v { + serde_json::Value::Object(_) => Some(v), + _ => None, + }) +} + +pub fn get_parameter_value(query: &Vec<(&str, &str)>, param: &str) -> Option { + return query + .iter() + .find(|q| String::from(q.0) == String::from(param)) + .map(|q| q.1.to_string()); +}