Add an optional parameter to allow specifying a domain, which will take precedence over both the DHCP-supplied domain (if any) and the hardcoded default domain.
Signed-off-by: Christoph Heiss <c.he...@proxmox.com> --- proxmox-auto-installer/src/utils.rs | 2 +- proxmox-installer-common/src/options.rs | 95 ++++++++++++++++++++----- proxmox-tui-installer/src/main.rs | 2 +- 3 files changed, 80 insertions(+), 19 deletions(-) diff --git a/proxmox-auto-installer/src/utils.rs b/proxmox-auto-installer/src/utils.rs index 655fecb..cf072b8 100644 --- a/proxmox-auto-installer/src/utils.rs +++ b/proxmox-auto-installer/src/utils.rs @@ -24,7 +24,7 @@ fn get_network_settings( runtime_info: &RuntimeInfo, setup_info: &SetupInfo, ) -> Result<NetworkOptions> { - let mut network_options = NetworkOptions::defaults_from(setup_info, &runtime_info.network); + let mut network_options = NetworkOptions::defaults_from(setup_info, &runtime_info.network, None); info!("Setting network configuration"); diff --git a/proxmox-installer-common/src/options.rs b/proxmox-installer-common/src/options.rs index 28f2971..562d285 100644 --- a/proxmox-installer-common/src/options.rs +++ b/proxmox-installer-common/src/options.rs @@ -408,10 +408,18 @@ pub struct NetworkOptions { impl NetworkOptions { const DEFAULT_DOMAIN: &'static str = "example.invalid"; - pub fn defaults_from(setup: &SetupInfo, network: &NetworkInfo) -> Self { + pub fn defaults_from( + setup: &SetupInfo, + network: &NetworkInfo, + default_domain: Option<&str>, + ) -> Self { let mut this = Self { ifname: String::new(), - fqdn: Self::construct_fqdn(network, setup.config.product.default_hostname()), + fqdn: Self::construct_fqdn( + network, + setup.config.product.default_hostname(), + default_domain, + ), // Safety: The provided mask will always be valid. address: CidrAddress::new(Ipv4Addr::UNSPECIFIED, 0).unwrap(), gateway: Ipv4Addr::UNSPECIFIED.into(), @@ -454,14 +462,23 @@ impl NetworkOptions { this } - fn construct_fqdn(network: &NetworkInfo, default_hostname: &str) -> Fqdn { + pub fn construct_fqdn( + network: &NetworkInfo, + default_hostname: &str, + default_domain: Option<&str>, + ) -> Fqdn { let hostname = network.hostname.as_deref().unwrap_or(default_hostname); - let domain = network - .dns - .domain - .as_deref() - .unwrap_or(Self::DEFAULT_DOMAIN); + // First, use the provided default domain if provided. If that is unset, + // use the one from the host network configuration, i.e. as and if provided by DHCP. + // As last fallback, use [`Self::DEFAULT_DOMAIN`]. + let domain = default_domain.unwrap_or_else(|| { + network + .dns + .domain + .as_deref() + .unwrap_or(Self::DEFAULT_DOMAIN) + }); Fqdn::from(&format!("{hostname}.{domain}")).unwrap_or_else(|_| { // Safety: This will always result in a valid FQDN, as we control & know @@ -527,10 +544,7 @@ mod tests { } } - #[test] - fn network_options_from_setup_network_info() { - let setup = SetupInfo::mocked(); - + fn mock_setup_network() -> (SetupInfo, NetworkInfo) { let mut interfaces = BTreeMap::new(); interfaces.insert( "eth0".to_owned(), @@ -545,7 +559,7 @@ mod tests { }, ); - let mut info = NetworkInfo { + let info = NetworkInfo { dns: Dns { domain: Some("bar.com".to_owned()), dns: Vec::new(), @@ -561,8 +575,15 @@ mod tests { hostname: Some("foo".to_owned()), }; + (SetupInfo::mocked(), info) + } + + #[test] + fn network_options_from_setup_network_info() { + let (setup, mut info) = mock_setup_network(); + pretty_assertions::assert_eq!( - NetworkOptions::defaults_from(&setup, &info), + NetworkOptions::defaults_from(&setup, &info, None), NetworkOptions { ifname: "eth0".to_owned(), fqdn: Fqdn::from("foo.bar.com").unwrap(), @@ -574,7 +595,7 @@ mod tests { info.hostname = None; pretty_assertions::assert_eq!( - NetworkOptions::defaults_from(&setup, &info), + NetworkOptions::defaults_from(&setup, &info, None), NetworkOptions { ifname: "eth0".to_owned(), fqdn: Fqdn::from("pve.bar.com").unwrap(), @@ -586,7 +607,7 @@ mod tests { info.dns.domain = None; pretty_assertions::assert_eq!( - NetworkOptions::defaults_from(&setup, &info), + NetworkOptions::defaults_from(&setup, &info, None), NetworkOptions { ifname: "eth0".to_owned(), fqdn: Fqdn::from("pve.example.invalid").unwrap(), @@ -598,7 +619,7 @@ mod tests { info.hostname = Some("foo".to_owned()); pretty_assertions::assert_eq!( - NetworkOptions::defaults_from(&setup, &info), + NetworkOptions::defaults_from(&setup, &info, None), NetworkOptions { ifname: "eth0".to_owned(), fqdn: Fqdn::from("foo.example.invalid").unwrap(), @@ -608,4 +629,44 @@ mod tests { } ); } + + #[test] + fn network_options_correctly_handles_user_supplied_default_domain() { + let (setup, mut info) = mock_setup_network(); + + pretty_assertions::assert_eq!( + NetworkOptions::defaults_from(&setup, &info, None), + NetworkOptions { + ifname: "eth0".to_owned(), + fqdn: Fqdn::from("foo.bar.com").unwrap(), + address: CidrAddress::new(Ipv4Addr::new(192, 168, 0, 2), 24).unwrap(), + gateway: IpAddr::V4(Ipv4Addr::new(192, 168, 0, 1)), + dns_server: Ipv4Addr::UNSPECIFIED.into(), + } + ); + + info.dns.domain = None; + pretty_assertions::assert_eq!( + NetworkOptions::defaults_from(&setup, &info, Some("custom.local")), + NetworkOptions { + ifname: "eth0".to_owned(), + fqdn: Fqdn::from("foo.custom.local").unwrap(), + address: CidrAddress::new(Ipv4Addr::new(192, 168, 0, 2), 24).unwrap(), + gateway: IpAddr::V4(Ipv4Addr::new(192, 168, 0, 1)), + dns_server: Ipv4Addr::UNSPECIFIED.into(), + } + ); + + info.dns.domain = Some("some.domain.local".to_owned()); + pretty_assertions::assert_eq!( + NetworkOptions::defaults_from(&setup, &info, Some("custom.local")), + NetworkOptions { + ifname: "eth0".to_owned(), + fqdn: Fqdn::from("foo.custom.local").unwrap(), + address: CidrAddress::new(Ipv4Addr::new(192, 168, 0, 2), 24).unwrap(), + gateway: IpAddr::V4(Ipv4Addr::new(192, 168, 0, 1)), + dns_server: Ipv4Addr::UNSPECIFIED.into(), + } + ); + } } diff --git a/proxmox-tui-installer/src/main.rs b/proxmox-tui-installer/src/main.rs index 341e60c..57a334f 100644 --- a/proxmox-tui-installer/src/main.rs +++ b/proxmox-tui-installer/src/main.rs @@ -169,7 +169,7 @@ fn main() { bootdisk: BootdiskOptions::defaults_from(&runtime_info.disks[0]), timezone: TimezoneOptions::defaults_from(&runtime_info, &locales), password: Default::default(), - network: NetworkOptions::defaults_from(&setup_info, &runtime_info.network), + network: NetworkOptions::defaults_from(&setup_info, &runtime_info.network, None), autoreboot: true, }, setup_info, -- 2.48.1 _______________________________________________ pve-devel mailing list pve-devel@lists.proxmox.com https://lists.proxmox.com/cgi-bin/mailman/listinfo/pve-devel