ospab
/
ostp
mirror of https://github.com/ospab/ostp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Prober now auto-reads DNS domain from config

This commit is contained in:
ospab 2026-06-19 17:34:37 +03:00
parent 87694c6218
commit 0cd189fb84
2 changed files with 46 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
ostp/src/dns_prober.rs
View File

@ -98,11 +98,52 @@ const PUBLIC_DNS_SERVERS: &[(&str, &str)] = &[
("Pishgaman", "5.160.25.25"),
];
pub async fn run_prober() {
println!("Enter your OSTP DNS Tunnel domain (e.g., tunnel.example.com):");
pub async fn run_prober(config_path: &std::path::Path) {
let mut target_domain = String::new();
if config_path.exists() {
if let Ok(content) = std::fs::read_to_string(config_path) {
let mut stripped = json_comments::StripComments::new(content.as_bytes());
if let Ok(json_val) = serde_json::from_reader::<_, serde_json::Value>(&mut stripped) {
// Check if it's a server config
if let Some(inbounds) = json_val.get("inbounds").and_then(|i| i.as_array()) {
for inbound in inbounds {
if inbound.get("protocol").and_then(|p| p.as_str()) == Some("dns") {
if let Some(domain) = inbound.get("domain").and_then(|d| d.as_str()) {
target_domain = domain.to_string();
break;
}
}
}
}
// Check if it's a client config
if target_domain.is_empty() {
if let Some(outbounds) = json_val.get("outbounds").and_then(|o| o.as_array()) {
for outbound in outbounds {
if let Some(transport) = outbound.get("transport") {
if transport.get("type").and_then(|t| t.as_str()) == Some("dns") {
if let Some(domain) = transport.get("domain").and_then(|d| d.as_str()) {
target_domain = domain.to_string();
break;
}
}
}
}
}
}
}
}
}
if target_domain.is_empty() {
println!("Could not find DNS Tunnel configuration in config.json.");
println!("Enter your OSTP DNS Tunnel domain (e.g., tunnel.example.com):");
std::io::stdin().read_line(&mut target_domain).unwrap();
let target_domain = target_domain.trim();
target_domain = target_domain.trim().to_string();
} else {
println!("Found DNS Tunnel domain in config.json: {}", target_domain);
}
if target_domain.is_empty() {
println!("Domain cannot be empty. Exiting prober.");
return;

2
ostp/src/main.rs
View File

@ -1607,7 +1607,7 @@ async fn run_app() -> Result<()> {
}
if args.prober {
dns_prober::run_prober().await;
dns_prober::run_prober(&args.config).await;
return Ok(());
}