feat: implement M1 — core DoH forwarding
- 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
This commit is contained in:
81
src/main.rs
Normal file
81
src/main.rs
Normal file
@@ -0,0 +1,81 @@
|
||||
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")
|
||||
}
|
||||
Reference in New Issue
Block a user