src/config/network.rs: read/write /etc/network/interfaces
Start implementing a recursive descent parser.
This commit is contained in:
Dietmar Maurer
289
src/config/network.rs
Normal file
289
src/config/network.rs
Normal file
@ -0,0 +1,289 @@
|
||||
use std::io::{Write};
|
||||
use std::collections::{HashSet, HashMap};
|
||||
|
||||
use anyhow::{Error, bail};
|
||||
|
||||
mod helper;
|
||||
//pub use helper::*;
|
||||
|
||||
mod lexer;
|
||||
pub use lexer::*;
|
||||
|
||||
mod parser;
|
||||
pub use parser::*;
|
||||
|
||||
#[derive(Debug, PartialEq)]
|
||||
pub enum AddressFamily {
|
||||
Inet4and6, // both v4 and v6
|
||||
Inet, // v4
|
||||
Inet6, // v6
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct Interface {
|
||||
pub name: String,
|
||||
pub address_family: AddressFamily,
|
||||
pub method_v4: Option<Token>,
|
||||
pub method_v6: Option<Token>,
|
||||
pub address_v4: Option<String>,
|
||||
pub gateway_v4: Option<String>,
|
||||
pub netmask_v4: Option<u8>,
|
||||
pub address_v6: Option<String>,
|
||||
pub gateway_v6: Option<String>,
|
||||
pub netmask_v6: Option<u8>,
|
||||
pub options_v4: Vec<String>,
|
||||
pub options_v6: Vec<String>,
|
||||
}
|
||||
|
||||
impl Interface {
|
||||
|
||||
pub fn new(
|
||||
name: String,
|
||||
address_family: AddressFamily,
|
||||
config_method: Option<Token>,
|
||||
) -> Self {
|
||||
let config_method_v4 = match address_family {
|
||||
AddressFamily::Inet | AddressFamily::Inet4and6 => Some(config_method.unwrap_or(Token::Static)),
|
||||
_ => None,
|
||||
};
|
||||
let config_method_v6 = match address_family {
|
||||
AddressFamily::Inet6 | AddressFamily::Inet4and6 => Some(config_method.unwrap_or(Token::Static)),
|
||||
_ => None,
|
||||
};
|
||||
|
||||
Self {
|
||||
name,
|
||||
address_family,
|
||||
method_v4: config_method_v4,
|
||||
method_v6: config_method_v6,
|
||||
address_v4: None,
|
||||
gateway_v4: None,
|
||||
netmask_v4: None,
|
||||
address_v6: None,
|
||||
gateway_v6: None,
|
||||
netmask_v6: None,
|
||||
options_v4: Vec::new(),
|
||||
options_v6: Vec::new(),
|
||||
}
|
||||
}
|
||||
|
||||
fn set_address_v4(&mut self, address: String) -> Result<(), Error> {
|
||||
if self.address_v4.is_none() {
|
||||
self.address_v4 = Some(address);
|
||||
} else {
|
||||
bail!("duplicate IPv4 address.");
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn set_gateway_v4(&mut self, gateway: String) -> Result<(), Error> {
|
||||
if self.gateway_v4.is_none() {
|
||||
self.gateway_v4 = Some(gateway);
|
||||
} else {
|
||||
bail!("duplicate IPv4 gateway.");
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn set_netmask_v4(&mut self, mask: u8) -> Result<(), Error> {
|
||||
if self.netmask_v4.is_none() {
|
||||
if mask > 0 && mask <= 32 {
|
||||
self.netmask_v4 = Some(mask);
|
||||
} else {
|
||||
bail!("invalid ipv4 netmaks '{}'", mask);
|
||||
}
|
||||
} else {
|
||||
bail!("duplicate IPv4 netmask.");
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn set_address_v6(&mut self, address: String) -> Result<(), Error> {
|
||||
if self.address_v6.is_none() {
|
||||
self.address_v6 = Some(address);
|
||||
} else {
|
||||
bail!("duplicate IPv6 address.");
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn set_gateway_v6(&mut self, gateway: String) -> Result<(), Error> {
|
||||
if self.gateway_v6.is_none() {
|
||||
self.gateway_v6 = Some(gateway);
|
||||
} else {
|
||||
bail!("duplicate IPv4 gateway.");
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn set_netmask_v6(&mut self, mask: u8) -> Result<(), Error> {
|
||||
if self.netmask_v6.is_none() {
|
||||
if mask > 0 && mask <= 128 {
|
||||
self.netmask_v6 = Some(mask);
|
||||
} else {
|
||||
bail!("invalid ipv6 netmaks '{}'", mask);
|
||||
}
|
||||
} else {
|
||||
bail!("duplicate IPv6 netmask.");
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn push_addon_option(&mut self, text: String) {
|
||||
match self.address_family {
|
||||
AddressFamily::Inet | AddressFamily::Inet4and6 => self.options_v4.push(text),
|
||||
AddressFamily::Inet6 => self.options_v6.push(text),
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
enum NetworkOrderEntry {
|
||||
Iface(String),
|
||||
Comment(String),
|
||||
Option(String),
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct NetworkConfig {
|
||||
pub auto_flag: HashSet<String>,
|
||||
interfaces: HashMap<String, Interface>,
|
||||
order: Vec<NetworkOrderEntry>,
|
||||
}
|
||||
|
||||
impl NetworkConfig {
|
||||
|
||||
pub fn new() -> Self {
|
||||
Self {
|
||||
auto_flag: HashSet::new(),
|
||||
interfaces: HashMap::new(),
|
||||
order: Vec::new(),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn write_config(&self, w: &mut dyn Write) -> Result<(), Error> {
|
||||
|
||||
fn method_to_str(method: &Option<Token>) -> &str {
|
||||
match method {
|
||||
None => "static",
|
||||
Some(method) => {
|
||||
match method {
|
||||
Token::Static => "static",
|
||||
Token::Loopback => "loopback",
|
||||
Token::Manual => "manual",
|
||||
Token::DHCP => "dhcp",
|
||||
_ => unreachable!(),
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn write_attributes_v4(w: &mut dyn Write, interface: &Interface) -> Result<(), Error> {
|
||||
if let Some(address) = &interface.address_v4 {
|
||||
if let Some(netmask) = interface.netmask_v4 {
|
||||
writeln!(w, " address {}/{}", address, netmask)?;
|
||||
} else {
|
||||
writeln!(w, " address {}", address)?;
|
||||
}
|
||||
}
|
||||
if let Some(gateway) = &interface.gateway_v4 {
|
||||
writeln!(w, " gateway {}", gateway)?;
|
||||
}
|
||||
for option in &interface.options_v4 {
|
||||
writeln!(w, " {}", option)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
};
|
||||
|
||||
fn write_attributes_v6(w: &mut dyn Write, interface: &Interface) -> Result<(), Error> {
|
||||
if let Some(address) = &interface.address_v6 {
|
||||
if let Some(netmask) = interface.netmask_v6 {
|
||||
writeln!(w, " address {}/{}", address, netmask)?;
|
||||
} else {
|
||||
writeln!(w, " address {}", address)?;
|
||||
}
|
||||
}
|
||||
if let Some(gateway) = &interface.gateway_v6 {
|
||||
writeln!(w, " gateway {}", gateway)?;
|
||||
}
|
||||
for option in &interface.options_v6 {
|
||||
writeln!(w, " {}", option)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
};
|
||||
|
||||
fn write_interface(w: &mut dyn Write, config: &NetworkConfig, name: &str, interface: &Interface) -> Result<(), Error> {
|
||||
if config.auto_flag.contains(name) {
|
||||
writeln!(w, "auto {}", name)?;
|
||||
}
|
||||
|
||||
if interface.address_family == AddressFamily::Inet4and6 && interface.method_v4 == interface.method_v6 {
|
||||
writeln!(w, "iface {} {}", name, method_to_str(&interface.method_v4))?;
|
||||
write_attributes_v4(w, &interface)?;
|
||||
write_attributes_v6(w, &interface)?;
|
||||
writeln!(w)?;
|
||||
} else if interface.address_family == AddressFamily::Inet4and6 {
|
||||
writeln!(w, "iface {} inet {}", name, method_to_str(&interface.method_v4))?;
|
||||
write_attributes_v4(w, &interface)?;
|
||||
writeln!(w)?;
|
||||
writeln!(w, "iface {} inet6 {}", name, method_to_str(&interface.method_v6))?;
|
||||
write_attributes_v6(w, &interface)?;
|
||||
writeln!(w)?;
|
||||
} else if interface.address_family == AddressFamily::Inet {
|
||||
writeln!(w, "iface {} inet {}", name, method_to_str(&interface.method_v4))?;
|
||||
write_attributes_v4(w, &interface)?;
|
||||
writeln!(w)?;
|
||||
} else if interface.address_family == AddressFamily::Inet6 {
|
||||
writeln!(w, "iface {} inet {}", name, method_to_str(&interface.method_v6))?;
|
||||
write_attributes_v6(w, &interface)?;
|
||||
writeln!(w)?;
|
||||
} else {
|
||||
unreachable!();
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
let mut done = HashSet::new();
|
||||
|
||||
let mut last_entry_was_comment = false;
|
||||
|
||||
for entry in self.order.iter() {
|
||||
match entry {
|
||||
NetworkOrderEntry::Comment(comment) => {
|
||||
writeln!(w, "#{}", comment)?;
|
||||
last_entry_was_comment = true;
|
||||
}
|
||||
NetworkOrderEntry::Option(option) => {
|
||||
if last_entry_was_comment { writeln!(w)?; }
|
||||
last_entry_was_comment = false;
|
||||
writeln!(w, "{}", option)?;
|
||||
writeln!(w)?;
|
||||
}
|
||||
NetworkOrderEntry::Iface(name) => {
|
||||
let interface = match self.interfaces.get(name) {
|
||||
Some(interface) => interface,
|
||||
None => continue,
|
||||
};
|
||||
|
||||
if last_entry_was_comment { writeln!(w)?; }
|
||||
last_entry_was_comment = false;
|
||||
|
||||
if done.contains(name) { continue; }
|
||||
done.insert(name);
|
||||
|
||||
write_interface(w, self, name, interface)?;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for (name, interface) in &self.interfaces {
|
||||
if done.contains(name) { continue; }
|
||||
write_interface(w, self, name, interface)?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user