feat: dns bind addr

config/config.yml

@@ -1,3 +1,6 @@
+# default 0.0.0.0
+# bind: 0.0.0.0
+
 # default 53
 dns_port: 53
 

src/config/load.rs

@@ -203,6 +203,7 @@ mod test {
         })
         .unwrap();
 
+        assert_eq!(config.bind, "0.0.0.0".to_string());
         assert_eq!(config.dns_port, 53);
         assert_eq!(config.network, "10.89.0.1/16".to_string());
     }

src/config/setting.rs

@@ -8,6 +8,7 @@ use super::{dns_table::DnsTable, hosts::Hosts};
 #[derive(Debug, Deserialize)]
 #[serde(default)]
 pub struct Setting {
+    pub bind: String,
     pub dns_port: u32,
     pub dns_upstream: Vec<String>,
     pub network: String,
@@ -26,6 +27,7 @@ pub struct Setting {
 impl Default for Setting {
     fn default() -> Self {
         Self {
+            bind: "0.0.0.0".to_string(),
             dns_port: 53,
             dns_upstream: vec![String::from("1.2.4.8"), String::from("8.8.8.8")],
             network: String::from("10.89.0.1/16"),

src/dns/dns_server.rs

@@ -68,9 +68,10 @@ pub(crate) async fn build_dns_server(setting: ArcSetting) -> Result<ServerFuture
 
     let mut server = ServerFuture::new(Handler { catalog });
     log::info!("dns listen port: {}", setting.dns_port);
-    server.register_socket(UdpSocket::bind(format!("0.0.0.0:{}", setting.dns_port)).await?);
+    server
+        .register_socket(UdpSocket::bind(format!("{}:{}", setting.bind, setting.dns_port)).await?);
     server.register_listener(
-        TcpListener::bind(format!("0.0.0.0:{}", setting.dns_port)).await?,
+        TcpListener::bind(format!("{}:{}", setting.bind, setting.dns_port)).await?,
         Duration::from_secs(5),
     );