use std::path::{Path, PathBuf};
use anyhow::Result;
pub use appliance_pid1::net::DhcpSession;
pub const NET_VAR: &str = "GUNNAR_NET";
pub const T1_OVERRIDE_VAR: &str = "GUNNAR_DHCP_T1_OVERRIDE_SECS";
pub const HOSTNAME_PREFIX: &str = "gunnar";
pub const RESOLVER_PROBE: &str = "github.com";
pub const RESOLVER_PROBE_IN_THIS_BUILD: Option<&str> = if cfg!(feature = "resolver-probe") { Some(RESOLVER_PROBE) } else { None };
pub const PANIC_CODE: &str = "GN-99";
pub fn net_plans() -> Result<Vec<appliance_pid1::net::NetConfig>> {
appliance_pid1::net::NetConfig::from_env(NET_VAR)
}
pub fn role_bound(spec: gunnar_common::bindspec::BindSpec, net: Option<&str>, exists: impl Fn(&str) -> bool) -> Result<gunnar_common::bindspec::BindSpec> {
if !matches!(spec, gunnar_common::bindspec::BindSpec::Role { .. }) {
return Ok(spec);
}
let declared: Vec<String> = match net.map(str::trim).filter(|n| !n.is_empty()) {
Some(n) => appliance_pid1::net::NetConfig::parse_list(n)?.iter().map(|c| c.interface().to_string()).collect(),
None => Vec::new(),
};
spec.for_this_box(&declared, exists).map_err(|why| anyhow::anyhow!("{NET_VAR}: {why}"))
}
pub fn role_bound_here(spec: gunnar_common::bindspec::BindSpec) -> Result<gunnar_common::bindspec::BindSpec> {
let net = std::env::var(NET_VAR).ok();
role_bound(spec, net.as_deref(), |i| std::path::Path::new("/sys/class/net").join(i).exists())
}
pub fn bring_up_network() -> Result<Vec<DhcpSession>> {
let narrate = |note: &str| ship_boot_narration(note);
let how = appliance_pid1::net::Bringup {
hostname_prefix: HOSTNAME_PREFIX,
resolver_probe: RESOLVER_PROBE_IN_THIS_BUILD,
narrate: &narrate,
};
appliance_pid1::net::bring_up_network_with(net_plans()?, &how, late_lease())
}
pub fn late_lease() -> appliance_pid1::net::LateLease {
appliance_pid1::net::LateLease {
narrate: ship_boot_narration,
t1_override_var: Some(T1_OVERRIDE_VAR),
}
}
pub fn ship_boot_narration(network_note: &str) {
appliance_pid1::narration::ship_boot_narration(network_note, &|l: &str| gunnar_common::logqueue::tap_line(l));
}
#[cfg(feature = "dhcp-client")]
pub async fn dhcp_renewal(session: DhcpSession) {
appliance_pid1::runtime::dhcp_renewal(session, appliance_pid1::runtime::dhcp_t1_override(T1_OVERRIDE_VAR)).await
}
#[cfg(not(feature = "dhcp-client"))]
pub async fn dhcp_renewal(session: DhcpSession) {
match session {}
}
pub fn announce_install_mode(data_root: &Path) {
if !Path::new(appliance_pid1::install::INSTALL_RECORD).exists() {
return;
}
let set = crate::dashboard::probe::disk_kind(data_root)
.and_then(|k| k.set)
.map(|s| format!("{}\u{d7}{}", s.members, crate::dashboard::fmt_bytes(s.member_bytes)));
appliance_pid1::install::announce_install_mode(set.as_deref(), &install_mode_event);
}
pub fn install_mode_event(line: &str) {
let f = install_mode_fields(line);
tracing::info!(
event = "install.mode",
mode = %f[0],
kept = %f[1],
kept_partno = %f[2],
kept_gib = %f[3],
kept_set = %f[4],
"install record read off {}",
appliance_pid1::install::INSTALL_RECORD
);
}
pub fn install_mode_fields(line: &str) -> [String; 5] {
let get = |k: &str| {
line.split_whitespace()
.find_map(|t| t.strip_prefix(k).and_then(|r| r.strip_prefix('=')))
.unwrap_or("-")
.to_owned()
};
[get("mode"), get("kept"), get("kept_partno"), get("kept_gib"), get("kept_set")]
}
pub fn install_panic_fault(data: Option<PathBuf>, shout: fn(&str)) {
appliance_pid1::fault::narration_from(crate::narrate::memory);
let p = appliance_pid1::fault::PanicFault {
code: PANIC_CODE,
meaning: crate::narrate::code_meaning(PANIC_CODE).unwrap_or("the app panicked"),
data,
};
appliance_pid1::fault::install_panic_fault(
p,
|| appliance_pid1::net::is_init() && crate::narrate::narrator().is_some(),
move |line: &str| say_panic_line(line, shout),
);
}
fn say_panic_line(line: &str, shout: fn(&str)) {
if let Some(text) = line.strip_prefix("PANIC: ") {
let text = format!("\ngunnar PANIC (exit 101): {text}\n");
shout(&text);
if let Some(path) = crate::slow::record_crash(&text) {
shout(&format!("gunnar: crash record written to {}\n", path.display()));
}
return;
}
match crate::narrate::narrator() {
Some(n) => n.say(line),
None => shout(&format!("gunnar: {line}\n")),
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn the_utility_role_becomes_the_nic_this_box_has() {
use gunnar_common::bindspec::BindSpec;
let role: BindSpec = "utility:50051".parse().unwrap();
let iface = |i: &str| BindSpec::Interface { interface: i.into(), port: 50051 };
assert_eq!(role_bound(role.clone(), Some("eth0:dhcp,eth1:dhcp"), |_| true).unwrap(), iface("eth1"));
assert_eq!(role_bound(role.clone(), Some("eth0:dhcp,eth1:dhcp"), |i| i == "eth0").unwrap(), iface("eth0"));
assert!(role_bound(role.clone(), None, |_| true).unwrap_err().to_string().contains(NET_VAR));
let kept: BindSpec = "127.0.0.1:50051".parse().unwrap();
assert_eq!(role_bound(kept.clone(), None, |_| false).unwrap(), kept);
}
#[test]
fn gunnar_names_its_network_variable_its_host_and_its_panic() {
assert_eq!(NET_VAR, "GUNNAR_NET");
assert_eq!(T1_OVERRIDE_VAR, "GUNNAR_DHCP_T1_OVERRIDE_SECS");
assert_eq!(appliance_pid1::net::hostname_for(HOSTNAME_PREFIX, None, [0, 0x11, 0x22, 0xaa, 0xbb, 0xcc]), "gunnar-aabbcc");
assert_eq!(appliance_pid1::net::hostname_for(HOSTNAME_PREFIX, Some("vm1.example"), [0; 6]), "vm1.example");
assert_eq!(RESOLVER_PROBE, "github.com");
assert_eq!(PANIC_CODE, "GN-99");
assert_eq!(crate::narrate::code_meaning(PANIC_CODE), Some("the app panicked"));
}
#[test]
fn the_install_line_becomes_the_dashboards_fields() {
use appliance_pid1::install::InstallRecord;
let p = InstallRecord::parse("mode=upgrade\nkept=korp-data\nkept_partno=3\nkept_bytes=241591910400\n").line(None);
assert_eq!(install_mode_fields(&p), ["upgrade", "korp-data", "3", "225.0", "-"].map(String::from));
let s = InstallRecord::parse("mode=upgrade\nkept=data-set\n").line(Some("4\u{d7}75G"));
assert_eq!(install_mode_fields(&s), ["upgrade", "data-set", "-", "-", "4\u{d7}75G"].map(String::from));
let s = InstallRecord::parse("mode=upgrade\nkept=data-set\n").line(None);
assert_eq!(install_mode_fields(&s), ["upgrade", "data-set", "-", "-", "-"].map(String::from));
let f = InstallRecord::parse("mode=fresh\n").line(None);
assert_eq!(install_mode_fields(&format!("{f} (install record read off the ESP)")), ["fresh", "-", "-", "-", "-"].map(String::from));
}
#[test]
fn a_late_lease_is_narrated_into_the_installers_boot_log() {
let late = late_lease();
assert_eq!(late.t1_override_var, Some(T1_OVERRIDE_VAR));
#[cfg(feature = "wg-client")]
{
crate::installer_listener::capture();
(late.narrate)("network up: eth0 10.0.2.15/24 gw Some(10.0.2.2) via DHCP from 10.0.2.2 (lease 3600s) — after the boot's failure, 7 asks l10-late");
let log = crate::installer_listener::BOOT_LOG.since(0, usize::MAX).lines.join("\n");
assert!(log.contains("after the boot's failure, 7 asks l10-late"), "the late line reaches the boot log: {log}");
}
let src = include_str!("boot.rs");
let up = &src[src.find("pub fn bring_up_network()").unwrap()..];
let up = &up[..up.find("\n}\n").unwrap()];
assert!(up.contains("bring_up_network_with(net_plans()?, &how, late_lease())"), "{up}");
assert!(up.contains("resolver_probe: RESOLVER_PROBE_IN_THIS_BUILD,"), "{up}");
}
#[test]
fn gunnar_net_is_read_from_its_own_variable() {
if std::env::var_os(NET_VAR).is_none() {
assert!(net_plans().unwrap().is_empty());
}
let plans = appliance_pid1::net::NetConfig::parse_list("eth0:81.27.107.63/23:81.27.106.1,eth1:10.13.8.78/22").unwrap();
assert_eq!(plans.iter().map(|p| p.interface()).collect::<Vec<_>>(), ["eth0", "eth1"]);
}
}
#[cfg(test)]
mod airgap {
#[test]
fn the_bring_up_asks_the_name_this_build_names() {
let src = include_str!("boot.rs");
let up = &src[src.find("pub fn bring_up_network()").unwrap()..];
let up = &up[..up.find("\n}\n").unwrap()];
assert!(up.contains("resolver_probe: RESOLVER_PROBE_IN_THIS_BUILD,"), "{up}");
}
#[cfg(not(feature = "resolver-probe"))]
#[test]
fn a_default_build_proves_no_resolver_and_asks_no_public_name() {
assert_eq!(super::RESOLVER_PROBE_IN_THIS_BUILD, None);
}
#[cfg(feature = "resolver-probe")]
#[test]
fn a_resolver_probe_build_asks_github_com_as_before() {
assert_eq!(super::RESOLVER_PROBE_IN_THIS_BUILD, Some("github.com"));
}
}