mirror of
https://github.com/LibreQoE/LibreQoS.git
synced 2024-11-29 11:23:49 -06:00
Move previous configuration file into its own space.
This commit is contained in:
Herbert Wolverson
parent
ff04f3718e
commit
a0c18164ff
317
src/rust/lqos_config/src/etc/etclqos_migration.rs
Normal file
317
src/rust/lqos_config/src/etc/etclqos_migration.rs
Normal file
@ -0,0 +1,317 @@
|
||||
//! Manages the `/etc/lqos.conf` file.
|
||||
use log::error;
|
||||
use serde::{Deserialize, Serialize};
|
||||
use toml_edit::{Document, value};
|
||||
use std::{fs, path::Path};
|
||||
use thiserror::Error;
|
||||
|
||||
/// Represents the top-level of the `/etc/lqos.conf` file. Serialization
|
||||
/// structure.
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct EtcLqos {
|
||||
/// The directory in which LibreQoS is installed.
|
||||
pub lqos_directory: String,
|
||||
|
||||
/// How frequently should `lqosd` read the `tc show qdisc` data?
|
||||
/// In ms.
|
||||
pub queue_check_period_ms: u64,
|
||||
|
||||
/// If present, provides a unique ID for the node. Used for
|
||||
/// anonymous stats (to identify nodes without providing an actual
|
||||
/// identity), and long-term stas.
|
||||
pub node_id: Option<String>,
|
||||
|
||||
/// If present, provide a name for the node.
|
||||
pub node_name: Option<String>,
|
||||
|
||||
/// If present, defines how the Bifrost XDP bridge operates.
|
||||
pub bridge: Option<BridgeConfig>,
|
||||
|
||||
/// If present, defines the values for various `sysctl` and `ethtool`
|
||||
/// tweaks.
|
||||
pub tuning: Option<Tunables>,
|
||||
|
||||
/// If present, defined anonymous usage stat sending
|
||||
pub usage_stats: Option<UsageStats>,
|
||||
|
||||
/// Defines for how many seconds a libpcap compatible capture should
|
||||
/// run. Short times are good, there's a real performance penalty to
|
||||
/// capturing high-throughput streams. Defaults to 10 seconds.
|
||||
pub packet_capture_time: Option<usize>,
|
||||
|
||||
/// Long-term statistics retention settings.
|
||||
pub long_term_stats: Option<LongTermStats>,
|
||||
}
|
||||
|
||||
/// Represents a set of `sysctl` and `ethtool` tweaks that may be
|
||||
/// applied (in place of the previous version's offload service)
|
||||
#[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq)]
|
||||
pub struct Tunables {
|
||||
/// Should the `irq_balance` system service be stopped?
|
||||
pub stop_irq_balance: bool,
|
||||
|
||||
/// Set the netdev budget (usecs)
|
||||
pub netdev_budget_usecs: u32,
|
||||
|
||||
/// Set the netdev budget (packets)
|
||||
pub netdev_budget_packets: u32,
|
||||
|
||||
/// Set the RX side polling frequency
|
||||
pub rx_usecs: u32,
|
||||
|
||||
/// Set the TX side polling frequency
|
||||
pub tx_usecs: u32,
|
||||
|
||||
/// Disable RXVLAN offloading? You generally want to do this.
|
||||
pub disable_rxvlan: bool,
|
||||
|
||||
/// Disable TXVLAN offloading? You generally want to do this.
|
||||
pub disable_txvlan: bool,
|
||||
|
||||
/// A list of `ethtool` offloads to be disabled.
|
||||
/// The default list is: [ "gso", "tso", "lro", "sg", "gro" ]
|
||||
pub disable_offload: Vec<String>,
|
||||
}
|
||||
|
||||
/// Defines the BiFrost XDP bridge accelerator parameters
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct BridgeConfig {
|
||||
/// Should the XDP bridge be enabled?
|
||||
pub use_xdp_bridge: bool,
|
||||
|
||||
/// A list of interface mappings.
|
||||
pub interface_mapping: Vec<BridgeInterface>,
|
||||
|
||||
/// A list of VLAN mappings.
|
||||
pub vlan_mapping: Vec<BridgeVlan>,
|
||||
}
|
||||
|
||||
/// An interface within the Bifrost XDP bridge.
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct BridgeInterface {
|
||||
/// The interface name. It *must* match an interface name
|
||||
/// findable by Linux.
|
||||
pub name: String,
|
||||
|
||||
/// Should Bifrost read VLAN tags and determine redirect
|
||||
/// policy from there?
|
||||
pub scan_vlans: bool,
|
||||
|
||||
/// The outbound interface - data that arrives in the interface
|
||||
/// defined by `name` will be redirected to this interface.
|
||||
///
|
||||
/// If you are using an "on a stick" configuration, this will
|
||||
/// be the same as `name`.
|
||||
pub redirect_to: String,
|
||||
}
|
||||
|
||||
/// If `scan_vlans` is enabled for an interface, then VLANs
|
||||
/// are examined on the way through the XDP BiFrost bridge.
|
||||
///
|
||||
/// If a VLAN is on the `parent` interface, and matches `tag` - it
|
||||
/// will be moved to VLAN `redirect_to`.
|
||||
///
|
||||
/// You often need to make reciprocal pairs of these.
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct BridgeVlan {
|
||||
/// The parent interface name on which the VLAN occurs.
|
||||
pub parent: String,
|
||||
|
||||
/// The VLAN tag number to redirect if matched.
|
||||
pub tag: u32,
|
||||
|
||||
/// The destination VLAN tag number if matched.
|
||||
pub redirect_to: u32,
|
||||
}
|
||||
|
||||
/// Definitions for anonymous usage submission
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct UsageStats {
|
||||
/// Are we allowed to send stats at all?
|
||||
pub send_anonymous: bool,
|
||||
|
||||
/// Where do we send them?
|
||||
pub anonymous_server: String,
|
||||
}
|
||||
|
||||
/// Long Term Data Retention
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct LongTermStats {
|
||||
/// Should we store long-term stats at all?
|
||||
pub gather_stats: bool,
|
||||
|
||||
/// How frequently should stats be accumulated into a long-term
|
||||
/// min/max/avg format per tick?
|
||||
pub collation_period_seconds: u32,
|
||||
|
||||
/// The license key for submitting stats to a LibreQoS hosted
|
||||
/// statistics server
|
||||
pub license_key: Option<String>,
|
||||
|
||||
/// UISP reporting period (in seconds). UISP queries can be slow,
|
||||
/// so hitting it every second or 10 seconds is going to cause problems
|
||||
/// for some people. A good default may be 5 minutes. Not specifying this
|
||||
/// disabled UISP integration.
|
||||
pub uisp_reporting_interval_seconds: Option<u64>,
|
||||
}
|
||||
|
||||
impl EtcLqos {
|
||||
/// Loads `/etc/lqos.conf`.
|
||||
pub fn load() -> Result<Self, EtcLqosError> {
|
||||
if !Path::new("/etc/lqos.conf").exists() {
|
||||
error!("/etc/lqos.conf does not exist!");
|
||||
return Err(EtcLqosError::ConfigDoesNotExist);
|
||||
}
|
||||
if let Ok(raw) = std::fs::read_to_string("/etc/lqos.conf") {
|
||||
let document = raw.parse::<Document>();
|
||||
match document {
|
||||
Err(e) => {
|
||||
error!("Unable to parse TOML from /etc/lqos.conf");
|
||||
error!("Full error: {:?}", e);
|
||||
Err(EtcLqosError::CannotParseToml)
|
||||
}
|
||||
Ok(mut config_doc) => {
|
||||
let cfg = toml_edit::de::from_document::<EtcLqos>(config_doc.clone());
|
||||
match cfg {
|
||||
Ok(mut cfg) => {
|
||||
check_config(&mut config_doc, &mut cfg);
|
||||
Ok(cfg)
|
||||
}
|
||||
Err(e) => {
|
||||
error!("Unable to parse TOML from /etc/lqos.conf");
|
||||
error!("Full error: {:?}", e);
|
||||
Err(EtcLqosError::CannotParseToml)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
error!("Unable to read contents of /etc/lqos.conf");
|
||||
Err(EtcLqosError::CannotReadFile)
|
||||
}
|
||||
}
|
||||
|
||||
/// Saves changes made to /etc/lqos.conf
|
||||
/// Copies current configuration into /etc/lqos.conf.backup first
|
||||
pub fn save(&self, document: &mut Document) -> Result<(), EtcLqosError> {
|
||||
let cfg_path = Path::new("/etc/lqos.conf");
|
||||
let backup_path = Path::new("/etc/lqos.conf.backup");
|
||||
if let Err(e) = std::fs::copy(cfg_path, backup_path) {
|
||||
log::error!("Unable to backup /etc/lqos.conf");
|
||||
log::error!("{e:?}");
|
||||
return Err(EtcLqosError::BackupFail);
|
||||
}
|
||||
let new_cfg = document.to_string();
|
||||
if let Err(e) = fs::write(cfg_path, new_cfg) {
|
||||
log::error!("Unable to write to /etc/lqos.conf");
|
||||
log::error!("{e:?}");
|
||||
return Err(EtcLqosError::WriteFail);
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Run this if you've received the OK from the licensing server, and been
|
||||
/// sent a license key. This appends a [long_term_stats] section to your
|
||||
/// config file - ONLY if one doesn't already exist.
|
||||
pub fn enable_long_term_stats(license_key: String) {
|
||||
if let Ok(raw) = std::fs::read_to_string("/etc/lqos.conf") {
|
||||
let document = raw.parse::<Document>();
|
||||
match document {
|
||||
Err(e) => {
|
||||
error!("Unable to parse TOML from /etc/lqos.conf");
|
||||
error!("Full error: {:?}", e);
|
||||
return;
|
||||
}
|
||||
Ok(mut config_doc) => {
|
||||
let cfg = toml_edit::de::from_document::<EtcLqos>(config_doc.clone());
|
||||
match cfg {
|
||||
Ok(cfg) => {
|
||||
// Now we enable LTS if its not present
|
||||
if let Ok(isp_config) = crate::LibreQoSConfig::load() {
|
||||
if cfg.long_term_stats.is_none() {
|
||||
|
||||
let mut new_section = toml_edit::table();
|
||||
new_section["gather_stats"] = value(true);
|
||||
new_section["collation_period_seconds"] = value(60);
|
||||
new_section["license_key"] = value(license_key);
|
||||
if isp_config.automatic_import_uisp {
|
||||
new_section["uisp_reporting_interval_seconds"] = value(300);
|
||||
}
|
||||
config_doc["long_term_stats"] = new_section;
|
||||
|
||||
let new_cfg = config_doc.to_string();
|
||||
if let Err(e) = fs::write(Path::new("/etc/lqos.conf"), new_cfg) {
|
||||
log::error!("Unable to write to /etc/lqos.conf");
|
||||
log::error!("{e:?}");
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Err(e) => {
|
||||
error!("Unable to parse TOML from /etc/lqos.conf");
|
||||
error!("Full error: {:?}", e);
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn check_config(cfg_doc: &mut Document, cfg: &mut EtcLqos) {
|
||||
use sha2::digest::Update;
|
||||
use sha2::Digest;
|
||||
|
||||
if cfg.node_id.is_none() {
|
||||
if let Ok(machine_id) = std::fs::read_to_string("/etc/machine-id") {
|
||||
let hash = sha2::Sha256::new().chain(machine_id).finalize();
|
||||
cfg.node_id = Some(format!("{:x}", hash));
|
||||
cfg_doc["node_id"] = value(format!("{:x}", hash));
|
||||
println!("Updating");
|
||||
if let Err(e) = cfg.save(cfg_doc) {
|
||||
log::error!("Unable to save /etc/lqos.conf");
|
||||
log::error!("{e:?}");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Error, Debug)]
|
||||
pub enum EtcLqosError {
|
||||
#[error(
|
||||
"/etc/lqos.conf not found. You must setup this file to use LibreQoS."
|
||||
)]
|
||||
ConfigDoesNotExist,
|
||||
#[error("Unable to read contents of /etc/lqos.conf.")]
|
||||
CannotReadFile,
|
||||
#[error("Unable to parse TOML in /etc/lqos.conf")]
|
||||
CannotParseToml,
|
||||
#[error("Unable to backup /etc/lqos.conf to /etc/lqos.conf.backup")]
|
||||
BackupFail,
|
||||
#[error("Unable to serialize new configuration")]
|
||||
SerializeFail,
|
||||
#[error("Unable to write to /etc/lqos.conf")]
|
||||
WriteFail,
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod test {
|
||||
const EXAMPLE_LQOS_CONF: &str = include_str!("../../../../lqos.example");
|
||||
|
||||
#[test]
|
||||
fn round_trip_toml() {
|
||||
let doc = EXAMPLE_LQOS_CONF.parse::<toml_edit::Document>().unwrap();
|
||||
let reserialized = doc.to_string();
|
||||
assert_eq!(EXAMPLE_LQOS_CONF, reserialized);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn add_node_id() {
|
||||
let mut doc = EXAMPLE_LQOS_CONF.parse::<toml_edit::Document>().unwrap();
|
||||
doc["node_id"] = toml_edit::value("test");
|
||||
let reserialized = doc.to_string();
|
||||
assert!(reserialized.contains("node_id = \"test\""));
|
||||
}
|
||||
}
|
||||
@ -1,318 +1,4 @@
|
||||
//! Manages the `/etc/lqos.conf` file.
|
||||
use log::error;
|
||||
use serde::{Deserialize, Serialize};
|
||||
use toml_edit::{Document, value};
|
||||
use std::{fs, path::Path};
|
||||
use thiserror::Error;
|
||||
mod python_migration;
|
||||
|
||||
/// Represents the top-level of the `/etc/lqos.conf` file. Serialization
|
||||
/// structure.
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct EtcLqos {
|
||||
/// The directory in which LibreQoS is installed.
|
||||
pub lqos_directory: String,
|
||||
|
||||
/// How frequently should `lqosd` read the `tc show qdisc` data?
|
||||
/// In ms.
|
||||
pub queue_check_period_ms: u64,
|
||||
|
||||
/// If present, provides a unique ID for the node. Used for
|
||||
/// anonymous stats (to identify nodes without providing an actual
|
||||
/// identity), and long-term stas.
|
||||
pub node_id: Option<String>,
|
||||
|
||||
/// If present, provide a name for the node.
|
||||
pub node_name: Option<String>,
|
||||
|
||||
/// If present, defines how the Bifrost XDP bridge operates.
|
||||
pub bridge: Option<BridgeConfig>,
|
||||
|
||||
/// If present, defines the values for various `sysctl` and `ethtool`
|
||||
/// tweaks.
|
||||
pub tuning: Option<Tunables>,
|
||||
|
||||
/// If present, defined anonymous usage stat sending
|
||||
pub usage_stats: Option<UsageStats>,
|
||||
|
||||
/// Defines for how many seconds a libpcap compatible capture should
|
||||
/// run. Short times are good, there's a real performance penalty to
|
||||
/// capturing high-throughput streams. Defaults to 10 seconds.
|
||||
pub packet_capture_time: Option<usize>,
|
||||
|
||||
/// Long-term statistics retention settings.
|
||||
pub long_term_stats: Option<LongTermStats>,
|
||||
}
|
||||
|
||||
/// Represents a set of `sysctl` and `ethtool` tweaks that may be
|
||||
/// applied (in place of the previous version's offload service)
|
||||
#[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq)]
|
||||
pub struct Tunables {
|
||||
/// Should the `irq_balance` system service be stopped?
|
||||
pub stop_irq_balance: bool,
|
||||
|
||||
/// Set the netdev budget (usecs)
|
||||
pub netdev_budget_usecs: u32,
|
||||
|
||||
/// Set the netdev budget (packets)
|
||||
pub netdev_budget_packets: u32,
|
||||
|
||||
/// Set the RX side polling frequency
|
||||
pub rx_usecs: u32,
|
||||
|
||||
/// Set the TX side polling frequency
|
||||
pub tx_usecs: u32,
|
||||
|
||||
/// Disable RXVLAN offloading? You generally want to do this.
|
||||
pub disable_rxvlan: bool,
|
||||
|
||||
/// Disable TXVLAN offloading? You generally want to do this.
|
||||
pub disable_txvlan: bool,
|
||||
|
||||
/// A list of `ethtool` offloads to be disabled.
|
||||
/// The default list is: [ "gso", "tso", "lro", "sg", "gro" ]
|
||||
pub disable_offload: Vec<String>,
|
||||
}
|
||||
|
||||
/// Defines the BiFrost XDP bridge accelerator parameters
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct BridgeConfig {
|
||||
/// Should the XDP bridge be enabled?
|
||||
pub use_xdp_bridge: bool,
|
||||
|
||||
/// A list of interface mappings.
|
||||
pub interface_mapping: Vec<BridgeInterface>,
|
||||
|
||||
/// A list of VLAN mappings.
|
||||
pub vlan_mapping: Vec<BridgeVlan>,
|
||||
}
|
||||
|
||||
/// An interface within the Bifrost XDP bridge.
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct BridgeInterface {
|
||||
/// The interface name. It *must* match an interface name
|
||||
/// findable by Linux.
|
||||
pub name: String,
|
||||
|
||||
/// Should Bifrost read VLAN tags and determine redirect
|
||||
/// policy from there?
|
||||
pub scan_vlans: bool,
|
||||
|
||||
/// The outbound interface - data that arrives in the interface
|
||||
/// defined by `name` will be redirected to this interface.
|
||||
///
|
||||
/// If you are using an "on a stick" configuration, this will
|
||||
/// be the same as `name`.
|
||||
pub redirect_to: String,
|
||||
}
|
||||
|
||||
/// If `scan_vlans` is enabled for an interface, then VLANs
|
||||
/// are examined on the way through the XDP BiFrost bridge.
|
||||
///
|
||||
/// If a VLAN is on the `parent` interface, and matches `tag` - it
|
||||
/// will be moved to VLAN `redirect_to`.
|
||||
///
|
||||
/// You often need to make reciprocal pairs of these.
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct BridgeVlan {
|
||||
/// The parent interface name on which the VLAN occurs.
|
||||
pub parent: String,
|
||||
|
||||
/// The VLAN tag number to redirect if matched.
|
||||
pub tag: u32,
|
||||
|
||||
/// The destination VLAN tag number if matched.
|
||||
pub redirect_to: u32,
|
||||
}
|
||||
|
||||
/// Definitions for anonymous usage submission
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct UsageStats {
|
||||
/// Are we allowed to send stats at all?
|
||||
pub send_anonymous: bool,
|
||||
|
||||
/// Where do we send them?
|
||||
pub anonymous_server: String,
|
||||
}
|
||||
|
||||
/// Long Term Data Retention
|
||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
||||
pub struct LongTermStats {
|
||||
/// Should we store long-term stats at all?
|
||||
pub gather_stats: bool,
|
||||
|
||||
/// How frequently should stats be accumulated into a long-term
|
||||
/// min/max/avg format per tick?
|
||||
pub collation_period_seconds: u32,
|
||||
|
||||
/// The license key for submitting stats to a LibreQoS hosted
|
||||
/// statistics server
|
||||
pub license_key: Option<String>,
|
||||
|
||||
/// UISP reporting period (in seconds). UISP queries can be slow,
|
||||
/// so hitting it every second or 10 seconds is going to cause problems
|
||||
/// for some people. A good default may be 5 minutes. Not specifying this
|
||||
/// disabled UISP integration.
|
||||
pub uisp_reporting_interval_seconds: Option<u64>,
|
||||
}
|
||||
|
||||
impl EtcLqos {
|
||||
/// Loads `/etc/lqos.conf`.
|
||||
pub fn load() -> Result<Self, EtcLqosError> {
|
||||
if !Path::new("/etc/lqos.conf").exists() {
|
||||
error!("/etc/lqos.conf does not exist!");
|
||||
return Err(EtcLqosError::ConfigDoesNotExist);
|
||||
}
|
||||
if let Ok(raw) = std::fs::read_to_string("/etc/lqos.conf") {
|
||||
let document = raw.parse::<Document>();
|
||||
match document {
|
||||
Err(e) => {
|
||||
error!("Unable to parse TOML from /etc/lqos.conf");
|
||||
error!("Full error: {:?}", e);
|
||||
Err(EtcLqosError::CannotParseToml)
|
||||
}
|
||||
Ok(mut config_doc) => {
|
||||
let cfg = toml_edit::de::from_document::<EtcLqos>(config_doc.clone());
|
||||
match cfg {
|
||||
Ok(mut cfg) => {
|
||||
check_config(&mut config_doc, &mut cfg);
|
||||
Ok(cfg)
|
||||
}
|
||||
Err(e) => {
|
||||
error!("Unable to parse TOML from /etc/lqos.conf");
|
||||
error!("Full error: {:?}", e);
|
||||
Err(EtcLqosError::CannotParseToml)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
error!("Unable to read contents of /etc/lqos.conf");
|
||||
Err(EtcLqosError::CannotReadFile)
|
||||
}
|
||||
}
|
||||
|
||||
/// Saves changes made to /etc/lqos.conf
|
||||
/// Copies current configuration into /etc/lqos.conf.backup first
|
||||
pub fn save(&self, document: &mut Document) -> Result<(), EtcLqosError> {
|
||||
let cfg_path = Path::new("/etc/lqos.conf");
|
||||
let backup_path = Path::new("/etc/lqos.conf.backup");
|
||||
if let Err(e) = std::fs::copy(cfg_path, backup_path) {
|
||||
log::error!("Unable to backup /etc/lqos.conf");
|
||||
log::error!("{e:?}");
|
||||
return Err(EtcLqosError::BackupFail);
|
||||
}
|
||||
let new_cfg = document.to_string();
|
||||
if let Err(e) = fs::write(cfg_path, new_cfg) {
|
||||
log::error!("Unable to write to /etc/lqos.conf");
|
||||
log::error!("{e:?}");
|
||||
return Err(EtcLqosError::WriteFail);
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Run this if you've received the OK from the licensing server, and been
|
||||
/// sent a license key. This appends a [long_term_stats] section to your
|
||||
/// config file - ONLY if one doesn't already exist.
|
||||
pub fn enable_long_term_stats(license_key: String) {
|
||||
if let Ok(raw) = std::fs::read_to_string("/etc/lqos.conf") {
|
||||
let document = raw.parse::<Document>();
|
||||
match document {
|
||||
Err(e) => {
|
||||
error!("Unable to parse TOML from /etc/lqos.conf");
|
||||
error!("Full error: {:?}", e);
|
||||
return;
|
||||
}
|
||||
Ok(mut config_doc) => {
|
||||
let cfg = toml_edit::de::from_document::<EtcLqos>(config_doc.clone());
|
||||
match cfg {
|
||||
Ok(cfg) => {
|
||||
// Now we enable LTS if its not present
|
||||
if let Ok(isp_config) = crate::LibreQoSConfig::load() {
|
||||
if cfg.long_term_stats.is_none() {
|
||||
|
||||
let mut new_section = toml_edit::table();
|
||||
new_section["gather_stats"] = value(true);
|
||||
new_section["collation_period_seconds"] = value(60);
|
||||
new_section["license_key"] = value(license_key);
|
||||
if isp_config.automatic_import_uisp {
|
||||
new_section["uisp_reporting_interval_seconds"] = value(300);
|
||||
}
|
||||
config_doc["long_term_stats"] = new_section;
|
||||
|
||||
let new_cfg = config_doc.to_string();
|
||||
if let Err(e) = fs::write(Path::new("/etc/lqos.conf"), new_cfg) {
|
||||
log::error!("Unable to write to /etc/lqos.conf");
|
||||
log::error!("{e:?}");
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Err(e) => {
|
||||
error!("Unable to parse TOML from /etc/lqos.conf");
|
||||
error!("Full error: {:?}", e);
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn check_config(cfg_doc: &mut Document, cfg: &mut EtcLqos) {
|
||||
use sha2::digest::Update;
|
||||
use sha2::Digest;
|
||||
|
||||
if cfg.node_id.is_none() {
|
||||
if let Ok(machine_id) = std::fs::read_to_string("/etc/machine-id") {
|
||||
let hash = sha2::Sha256::new().chain(machine_id).finalize();
|
||||
cfg.node_id = Some(format!("{:x}", hash));
|
||||
cfg_doc["node_id"] = value(format!("{:x}", hash));
|
||||
println!("Updating");
|
||||
if let Err(e) = cfg.save(cfg_doc) {
|
||||
log::error!("Unable to save /etc/lqos.conf");
|
||||
log::error!("{e:?}");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Error, Debug)]
|
||||
pub enum EtcLqosError {
|
||||
#[error(
|
||||
"/etc/lqos.conf not found. You must setup this file to use LibreQoS."
|
||||
)]
|
||||
ConfigDoesNotExist,
|
||||
#[error("Unable to read contents of /etc/lqos.conf.")]
|
||||
CannotReadFile,
|
||||
#[error("Unable to parse TOML in /etc/lqos.conf")]
|
||||
CannotParseToml,
|
||||
#[error("Unable to backup /etc/lqos.conf to /etc/lqos.conf.backup")]
|
||||
BackupFail,
|
||||
#[error("Unable to serialize new configuration")]
|
||||
SerializeFail,
|
||||
#[error("Unable to write to /etc/lqos.conf")]
|
||||
WriteFail,
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod test {
|
||||
const EXAMPLE_LQOS_CONF: &str = include_str!("../../../lqos.example");
|
||||
|
||||
#[test]
|
||||
fn round_trip_toml() {
|
||||
let doc = EXAMPLE_LQOS_CONF.parse::<toml_edit::Document>().unwrap();
|
||||
let reserialized = doc.to_string();
|
||||
assert_eq!(EXAMPLE_LQOS_CONF, reserialized);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn add_node_id() {
|
||||
let mut doc = EXAMPLE_LQOS_CONF.parse::<toml_edit::Document>().unwrap();
|
||||
doc["node_id"] = toml_edit::value("test");
|
||||
let reserialized = doc.to_string();
|
||||
assert!(reserialized.contains("node_id = \"test\""));
|
||||
}
|
||||
}
|
||||
mod etclqos_migration;
|
||||
pub use etclqos_migration::*;
|
||||
|
||||
Loading…
Reference in New Issue
Block a user