feat: add DNS override settings (listen, fake-ip-range)

src/config.rs

@@ -78,6 +78,8 @@ pub struct MihomoConfig {
     pub geo_auto_update: Option<bool>,
     pub geo_update_interval: Option<u16>,
     pub geox_url: Option<GeoxUrl>,
+    pub dns: Option<MihomoDnsConfig>,
+    pub tun: Option<MihomoTunConfig>,
 }
 
 impl Default for MihomoConfig {
@@ -109,6 +111,8 @@ impl Default for MihomoConfig {
                     "https://testingcf.jsdelivr.net/gh/MetaCubeX/meta-rules-dat@release/country.mmdb",
                 ),
             }),
+            dns: Some(MihomoDnsConfig::default()),
+            tun: Some(MihomoTunConfig::disabled()),
         }
     }
 }
@@ -144,6 +148,61 @@ pub struct GeoxUrl {
     pub mmdb: String,
 }
 
+/// DNS configuration for mihomo.
+///
+/// Referenced from https://wiki.metacubex.one/config/dns
+#[derive(Serialize, Deserialize, Debug, Clone)]
+#[serde(default)]
+pub struct MihomoDnsConfig {
+    pub enable: Option<bool>,
+    pub listen: Option<String>,
+    pub fake_ip_range: Option<String>,
+}
+
+impl Default for MihomoDnsConfig {
+    fn default() -> Self {
+        MihomoDnsConfig {
+            enable: Some(true),
+            listen: Some(String::from("0.0.0.0:5353")),
+            fake_ip_range: Some(String::from("198.18.0.1/16")),
+        }
+    }
+}
+
+/// TUN configuration for mihomo.
+///
+/// Referenced from https://wiki.metacubex.one/config/tun
+#[derive(Serialize, Deserialize, Debug, Clone)]
+#[serde(default)]
+pub struct MihomoTunConfig {
+    pub enable: Option<bool>,
+    pub stack: Option<String>,
+    pub auto_route: Option<bool>,
+    pub auto_detect_interface: Option<bool>,
+    pub dns_hijack: Option<Vec<String>>,
+}
+
+impl MihomoTunConfig {
+    pub fn disabled() -> Self {
+        MihomoTunConfig {
+            enable: Some(false),
+            ..Default::default()
+        }
+    }
+}
+
+impl Default for MihomoTunConfig {
+    fn default() -> Self {
+        MihomoTunConfig {
+            enable: Some(true),
+            stack: Some(String::from("mixed")),
+            auto_route: Some(true),
+            auto_detect_interface: Some(true),
+            dns_hijack: Some(vec![String::from("any:53"), String::from("tcp://any:53")]),
+        }
+    }
+}
+
 impl Config {
     pub fn new() -> Config {
         Config::default()
@@ -279,6 +338,51 @@ pub struct MihomoYamlConfig {
     #[serde(rename = "geox-url", skip_serializing_if = "Option::is_none")]
     geox_url: Option<GeoxUrl>,
 
+    #[serde(skip_serializing_if = "Option::is_none")]
+    dns: Option<MihomoDnsYamlConfig>,
+
+    #[serde(skip_serializing_if = "Option::is_none")]
+    tun: Option<MihomoTunYamlConfig>,
+
+    #[serde(flatten)]
+    extra: HashMap<String, serde_yaml::Value>,
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+pub struct MihomoDnsYamlConfig {
+    #[serde(skip_serializing_if = "Option::is_none")]
+    enable: Option<bool>,
+
+    #[serde(skip_serializing_if = "Option::is_none")]
+    listen: Option<String>,
+
+    #[serde(rename = "fake-ip-range", skip_serializing_if = "Option::is_none")]
+    fake_ip_range: Option<String>,
+
+    #[serde(flatten)]
+    extra: HashMap<String, serde_yaml::Value>,
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+pub struct MihomoTunYamlConfig {
+    #[serde(skip_serializing_if = "Option::is_none")]
+    enable: Option<bool>,
+
+    #[serde(skip_serializing_if = "Option::is_none")]
+    stack: Option<String>,
+
+    #[serde(rename = "auto-route", skip_serializing_if = "Option::is_none")]
+    auto_route: Option<bool>,
+
+    #[serde(
+        rename = "auto-detect-interface",
+        skip_serializing_if = "Option::is_none"
+    )]
+    auto_detect_interface: Option<bool>,
+
+    #[serde(rename = "dns-hijack", skip_serializing_if = "Option::is_none")]
+    dns_hijack: Option<Vec<String>>,
+
     #[serde(flatten)]
     extra: HashMap<String, serde_yaml::Value>,
 }
@@ -313,6 +417,34 @@ pub fn apply_mihomo_override(path: &str, override_config: &MihomoConfig) -> Resu
     mihomo_yaml.geo_update_interval = override_config.geo_update_interval;
     mihomo_yaml.geox_url = override_config.geox_url.clone();
 
+    if let Some(ref dns_override) = override_config.dns {
+        let dns = mihomo_yaml.dns.get_or_insert_with(|| MihomoDnsYamlConfig {
+            enable: None,
+            listen: None,
+            fake_ip_range: None,
+            extra: HashMap::new(),
+        });
+        dns.enable = dns_override.enable;
+        dns.listen = dns_override.listen.clone();
+        dns.fake_ip_range = dns_override.fake_ip_range.clone();
+    }
+
+    if let Some(ref tun_override) = override_config.tun {
+        let tun = mihomo_yaml.tun.get_or_insert_with(|| MihomoTunYamlConfig {
+            enable: None,
+            stack: None,
+            auto_route: None,
+            auto_detect_interface: None,
+            dns_hijack: None,
+            extra: HashMap::new(),
+        });
+        tun.enable = tun_override.enable;
+        tun.stack = tun_override.stack.clone();
+        tun.auto_route = tun_override.auto_route;
+        tun.auto_detect_interface = tun_override.auto_detect_interface;
+        tun.dns_hijack = tun_override.dns_hijack.clone();
+    }
+
     // Write to file
     let serialized_mihomo_yaml = serde_yaml::to_string(&mihomo_yaml)?;
     fs::write(path, serialized_mihomo_yaml)?;