use crate::{
Error, Result,
v4::metric::{Metric, MetricType},
};
use alloc::borrow::ToOwned;
use core::{fmt, str::FromStr};
#[derive(Copy, Clone, Debug, Eq, PartialEq, PartialOrd, Ord)]
pub enum AvailabilityImpactToTheVulnerableSystem {
None,
Low,
High,
}
impl Default for AvailabilityImpactToTheVulnerableSystem {
fn default() -> Self {
Self::High
}
}
impl Metric for AvailabilityImpactToTheVulnerableSystem {
const TYPE: MetricType = MetricType::VA;
fn as_str(self) -> &'static str {
match self {
AvailabilityImpactToTheVulnerableSystem::None => "N",
AvailabilityImpactToTheVulnerableSystem::Low => "L",
AvailabilityImpactToTheVulnerableSystem::High => "H",
}
}
}
impl fmt::Display for AvailabilityImpactToTheVulnerableSystem {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}:{}", Self::name(), self.as_str())
}
}
impl FromStr for AvailabilityImpactToTheVulnerableSystem {
type Err = Error;
fn from_str(s: &str) -> Result<Self> {
match s {
"N" => Ok(AvailabilityImpactToTheVulnerableSystem::None),
"L" => Ok(AvailabilityImpactToTheVulnerableSystem::Low),
"H" => Ok(AvailabilityImpactToTheVulnerableSystem::High),
_ => Err(Error::InvalidMetricV4 {
metric_type: Self::TYPE,
value: s.to_owned(),
}),
}
}
}
#[cfg(feature = "std")]
pub(crate) mod merge {
use super::*;
use crate::{
Error,
v4::{
MetricType,
metric::{MetricLevel, environmental::ModifiedAvailabilityImpactToTheVulnerableSystem},
},
};
use alloc::borrow::ToOwned;
use core::str::FromStr;
#[derive(Copy, Clone, Debug, Eq, PartialEq, PartialOrd, Ord)]
pub(crate) enum MergedAvailabilityImpactToTheVulnerableSystem {
High,
Low,
None,
}
impl Default for MergedAvailabilityImpactToTheVulnerableSystem {
fn default() -> Self {
Self::High
}
}
impl FromStr for MergedAvailabilityImpactToTheVulnerableSystem {
type Err = Error;
fn from_str(s: &str) -> Result<Self> {
match s {
"H" => Ok(MergedAvailabilityImpactToTheVulnerableSystem::High),
"L" => Ok(MergedAvailabilityImpactToTheVulnerableSystem::Low),
"N" => Ok(MergedAvailabilityImpactToTheVulnerableSystem::None),
_ => Err(Error::InvalidMetricV4 {
metric_type: MetricType::VA,
value: s.to_owned(),
}),
}
}
}
impl MetricLevel for MergedAvailabilityImpactToTheVulnerableSystem {
fn level(self) -> f64 {
match self {
Self::High => 0.0,
Self::Low => 0.1,
Self::None => 0.2,
}
}
}
impl AvailabilityImpactToTheVulnerableSystem {
pub(crate) fn merge(
self,
value: Option<ModifiedAvailabilityImpactToTheVulnerableSystem>,
) -> MergedAvailabilityImpactToTheVulnerableSystem {
match value {
Some(ModifiedAvailabilityImpactToTheVulnerableSystem::NotDefined) | None => {
match self {
Self::High => MergedAvailabilityImpactToTheVulnerableSystem::High,
Self::Low => MergedAvailabilityImpactToTheVulnerableSystem::Low,
Self::None => MergedAvailabilityImpactToTheVulnerableSystem::None,
}
}
Some(ModifiedAvailabilityImpactToTheVulnerableSystem::High) => {
MergedAvailabilityImpactToTheVulnerableSystem::High
}
Some(ModifiedAvailabilityImpactToTheVulnerableSystem::Low) => {
MergedAvailabilityImpactToTheVulnerableSystem::Low
}
Some(ModifiedAvailabilityImpactToTheVulnerableSystem::None) => {
MergedAvailabilityImpactToTheVulnerableSystem::None
}
}
}
}
}