use std::default::Default; use std::path::{Path, PathBuf}; use std::collections::HashMap; use serde::{Serialize, Deserialize, Deserializer, Serializer}; use std::fmt::Formatter; use std::net::{SocketAddr, IpAddr, ToSocketAddrs}; use std::str::FromStr; use serde::de::Error; use crate::authorization::permissions::PermRule; use crate::authorization::roles::RoleIdentifier; type Result = std::result::Result; pub fn read(path: &Path) -> Result { serde_dhall::from_file(path) .parse() .map_err(Into::into) } #[derive(Debug, Clone, Serialize, Deserialize)] pub struct Config { /// A list of address/port pairs to listen on. // TODO: This should really be a variant type; that is something that can figure out itself if // it contains enough information to open a socket (i.e. it checks if it's a valid path (=> // Unix socket) or IPv4/v6 address) pub listens: Vec, /// Machine descriptions to load //pub machines: HashMap, /// Actors to load and their configuration options pub actors: HashMap, /// Initiators to load and their configuration options pub initiators: HashMap, pub mqtt_url: String, pub actor_connections: Vec<(String, String)>, pub init_connections: Vec<(String, String)>, pub db_path: PathBuf, pub roles: HashMap, #[serde(flatten)] pub tlsconfig: TlsListen, #[serde(default, skip_serializing_if = "Option::is_none")] pub tlskeylog: Option, #[serde(default, skip)] pub verbosity: isize, #[serde(default, skip)] pub log_format: String, } impl Config { pub fn is_quiet(&self) -> bool { self.verbosity < 0 } } #[derive(Debug, Clone, Serialize, Deserialize)] pub struct RoleConfig { #[serde(default = "Vec::new")] pub parents: Vec, #[serde(default = "Vec::new")] pub permissions: Vec, } #[derive(Debug, Clone, Serialize, Deserialize)] pub struct ModuleConfig { pub module: String, pub params: HashMap } pub struct ListenSock { listen: Listen, tls_config: TlsListen, } #[derive(Debug, Clone)] pub struct Listen { address: String, port: Option, } #[derive(Debug, Clone, Serialize, Deserialize, Default)] pub struct TlsListen { pub certfile: PathBuf, pub keyfile: PathBuf, #[serde(default, skip_serializing_if = "Option::is_none")] pub ciphers: Option, #[serde(default, skip_serializing_if = "Option::is_none")] pub tls_min_version: Option, #[serde(default = "Vec::new", skip_serializing_if = "Vec::is_empty")] pub protocols: Vec, } impl Listen { pub fn to_tuple(&self) -> (&str, u16) { (self.address.as_str(), self.port.unwrap_or(DEFAULT_PORT)) } } impl std::fmt::Display for Listen { fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result { write!(f, "{}:{}", &self.address, self.port.unwrap_or(DEFAULT_PORT)) } } impl ToSocketAddrs for Listen { type Iter = <(String, u16) as ToSocketAddrs>::Iter; fn to_socket_addrs(&self) -> std::io::Result { if let Some(port) = self.port { (self.address.as_str(), port).to_socket_addrs() } else { (self.address.as_str(), DEFAULT_PORT).to_socket_addrs() } } } impl<'de> serde::Deserialize<'de> for Listen { fn deserialize(deserializer: D) -> std::result::Result where D: Deserializer<'de> { deserializer.deserialize_str(ListenVisitor) } } impl serde::Serialize for Listen { fn serialize(&self, serializer: S) -> std::result::Result where S: Serializer { if let Some(port) = self.port { serializer.serialize_str(&format!("{}:{}", self.address, port)) } else { serializer.serialize_str(&self.address) } } } struct ListenVisitor; impl<'de> serde::de::Visitor<'de> for ListenVisitor { type Value = Listen; fn expecting(&self, formatter: &mut Formatter) -> std::fmt::Result { write!(formatter, "A string encoding a valid IP or Hostname (e.g. 127.0.0.1 or [::1]) with \ or without a defined port") } fn visit_str(self, v: &str) -> std::result::Result where E: Error { let sockaddr = SocketAddr::from_str(v); if let Ok(address) = sockaddr { return Ok(Listen { address: address.ip().to_string(), port: Some(address.port()), }) } let ipaddr = IpAddr::from_str(v); if let Ok(address) = ipaddr { return Ok(Listen { address: address.to_string(), port: None, }) } let mut split = v.split(':'); let address = split.next() .expect("str::split should always return at least one element") .to_string(); let port = if let Some(port) = split.next() { let port: u16 = port.parse() .map_err(|_| { E::custom(&format!("Expected valid ip address or hostname with or without \ port. Failed to parse \"{}\".", v)) })?; Some(port) } else { None }; Ok(Listen { address, port }) } } impl Default for Config { fn default() -> Self { let mut actors: HashMap:: = HashMap::new(); let mut initiators: HashMap:: = HashMap::new(); actors.insert("Actor".to_string(), ModuleConfig { module: "Shelly".to_string(), params: HashMap::new(), }); initiators.insert("Initiator".to_string(), ModuleConfig { module: "TCP-Listen".to_string(), params: HashMap::new(), }); Config { listens: vec![ Listen { address: "127.0.0.1".to_string(), port: None, } ], actors, initiators, mqtt_url: "tcp://localhost:1883".to_string(), actor_connections: vec![ ("Testmachine".to_string(), "Actor".to_string()), ], init_connections: vec![ ("Initiator".to_string(), "Testmachine".to_string()), ], db_path: PathBuf::from("/run/bffh/database"), roles: HashMap::new(), tlsconfig: TlsListen { certfile: PathBuf::from("./bffh.crt"), keyfile: PathBuf::from("./bffh.key"), .. Default::default() }, tlskeylog: None, verbosity: 0, log_format: "Full".to_string(), } } } // The default port in the non-assignable i.e. free-use area pub const DEFAULT_PORT: u16 = 59661;