95 lines
2.3 KiB
Rust
95 lines
2.3 KiB
Rust
use itertools::Itertools;
|
|
use log::{info, warn};
|
|
use std::collections::BTreeSet as Set;
|
|
use tokio::process::Command;
|
|
|
|
use super::{format_err, retry_or_ignore, Config, Result};
|
|
use crate::{
|
|
bootstrap::config,
|
|
udev,
|
|
utils::{select_n_by_regex, NameAliases},
|
|
};
|
|
|
|
pub async fn setup(cfg: &Config) {
|
|
if cfg.networks.is_empty() {
|
|
warn!("no networks configured");
|
|
return;
|
|
}
|
|
|
|
let mut assigned = Set::new();
|
|
|
|
for net in &cfg.networks {
|
|
retry_or_ignore(async || setup_network(net, &mut assigned).await).await;
|
|
}
|
|
}
|
|
|
|
async fn setup_network(net: &config::Network, assigned: &mut Set<String>) -> Result<()> {
|
|
info!("setting up network {}", net.name);
|
|
|
|
let netdevs = get_interfaces()?
|
|
.filter(|dev| !assigned.contains(dev.name()))
|
|
.collect::<Vec<_>>();
|
|
|
|
for dev in &netdevs {
|
|
info!(
|
|
"- available network device: {}, aliases [{}]",
|
|
dev.name(),
|
|
dev.aliases().join(", ")
|
|
);
|
|
}
|
|
|
|
let mut cmd = Command::new("ash");
|
|
cmd.arg("-c");
|
|
cmd.arg(&net.script);
|
|
|
|
let mut selected = Vec::new();
|
|
|
|
for iface in &net.interfaces {
|
|
let var = &iface.var;
|
|
|
|
let netdevs = netdevs.iter().filter(|na| !assigned.contains(na.name()));
|
|
let if_names = select_n_by_regex(iface.n, &iface.regexps, netdevs);
|
|
|
|
if if_names.is_empty() {
|
|
return Err(format_err!("- no interface match for {var:?}"));
|
|
}
|
|
|
|
let value = if_names.join(" ");
|
|
info!("- {var}={value}");
|
|
cmd.env(var, value);
|
|
|
|
selected.extend(if_names);
|
|
}
|
|
|
|
info!("- running script");
|
|
let status = cmd.status().await?;
|
|
if !status.success() {
|
|
return Err(format_err!("setup script failed: {status}"));
|
|
}
|
|
|
|
assigned.extend(selected);
|
|
Ok(())
|
|
}
|
|
|
|
fn get_interfaces() -> Result<impl Iterator<Item = NameAliases>> {
|
|
Ok(udev::get_devices("net")?.into_iter().map(|dev| {
|
|
let mut na = NameAliases::new(dev.sysname().to_string());
|
|
|
|
for (property, value) in dev.properties() {
|
|
if [
|
|
"INTERFACE",
|
|
"ID_NET_NAME",
|
|
"ID_NET_NAME_PATH",
|
|
"ID_NET_NAME_MAC",
|
|
"ID_NET_NAME_SLOT",
|
|
]
|
|
.contains(&property)
|
|
{
|
|
na.push(value.to_string());
|
|
}
|
|
}
|
|
|
|
na
|
|
}))
|
|
}
|