Mohammadreza Khani
3a5c0d2aba
- Axum HTTP server with /dns-query endpoint (GET + POST per RFC 8484) - DNS wire-format parsing via hickory-proto - Upstream DoH client with round-robin selection - TOML config file with resolver definitions - SERVFAIL fallback when upstream is unreachable - /health endpoint - Unit tests (7) and integration tests (5) - Zero warnings, clippy clean, fmt clean
82 lines
2.4 KiB
Rust
82 lines
2.4 KiB
Rust
use std::path::PathBuf;
|
|
use std::sync::Arc;
|
|
|
|
use anyhow::Context;
|
|
use axum::routing::{get, post};
|
|
use axum::{Extension, Router};
|
|
use tokio::net::TcpListener;
|
|
use tracing::info;
|
|
use tracing_subscriber::EnvFilter;
|
|
|
|
use doh_forwarder::config::AppConfig;
|
|
use doh_forwarder::routes::{dns_query_get, dns_query_post, AppState};
|
|
use doh_forwarder::upstream::doh_client::DohClient;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> anyhow::Result<()> {
|
|
// Initialize tracing (RUST_LOG env filter, defaults to info)
|
|
tracing_subscriber::fmt()
|
|
.with_env_filter(
|
|
EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info")),
|
|
)
|
|
.init();
|
|
|
|
// Parse CLI args for config path
|
|
let config_path = parse_config_path();
|
|
info!(path = %config_path.display(), "loading configuration");
|
|
|
|
let config = AppConfig::load(&config_path)
|
|
.with_context(|| format!("failed to load config from {:?}", config_path))?;
|
|
|
|
info!(listen = %config.server.listen, "starting doh-forwarder");
|
|
info!(
|
|
resolvers = config
|
|
.upstream
|
|
.resolvers
|
|
.iter()
|
|
.map(|r| r.name.as_str())
|
|
.collect::<Vec<_>>()
|
|
.join(", "),
|
|
strategy = %config.upstream.strategy,
|
|
"upstream configuration"
|
|
);
|
|
|
|
let doh_client = DohClient::new(config.upstream.resolvers.clone());
|
|
|
|
let state = AppState {
|
|
doh_client,
|
|
rr_counter: Arc::new(std::sync::atomic::AtomicUsize::new(0)),
|
|
};
|
|
|
|
let app = Router::new()
|
|
.route("/dns-query", post(dns_query_post).get(dns_query_get))
|
|
.route("/health", get(health))
|
|
.layer(Extension(state));
|
|
|
|
let listener = TcpListener::bind(&config.server.listen)
|
|
.await
|
|
.with_context(|| format!("failed to bind to {}", config.server.listen))?;
|
|
|
|
info!(addr = %listener.local_addr()?, "listening");
|
|
|
|
axum::serve(listener, app).await.context("server error")?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Health-check endpoint.
|
|
async fn health() -> &'static str {
|
|
"ok"
|
|
}
|
|
|
|
/// Parse the config file path from CLI args, defaulting to `config/default.toml`.
|
|
fn parse_config_path() -> PathBuf {
|
|
let args: Vec<String> = std::env::args().collect();
|
|
if let Some(pos) = args.iter().position(|a| a == "--config") {
|
|
if let Some(path) = args.get(pos + 1) {
|
|
return PathBuf::from(path);
|
|
}
|
|
}
|
|
PathBuf::from("config/default.toml")
|
|
}
|