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 AttackVector {
Physical,
Local,
Adjacent,
Network,
}
impl Default for AttackVector {
fn default() -> Self {
Self::Network
}
}
impl Metric for AttackVector {
const TYPE: MetricType = MetricType::AV;
fn as_str(self) -> &'static str {
match self {
AttackVector::Network => "N",
AttackVector::Adjacent => "A",
AttackVector::Local => "L",
AttackVector::Physical => "P",
}
}
}
impl fmt::Display for AttackVector {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}:{}", Self::name(), self.as_str())
}
}
impl FromStr for AttackVector {
type Err = Error;
fn from_str(s: &str) -> Result<Self> {
match s {
"N" => Ok(AttackVector::Network),
"A" => Ok(AttackVector::Adjacent),
"L" => Ok(AttackVector::Local),
"P" => Ok(AttackVector::Physical),
_ => 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::ModifiedAttackVector},
},
};
use alloc::borrow::ToOwned;
use core::str::FromStr;
#[derive(Copy, Clone, Debug, Eq, PartialEq, PartialOrd, Ord)]
pub(crate) enum MergedAttackVector {
Physical,
Local,
Adjacent,
Network,
}
impl Default for MergedAttackVector {
fn default() -> Self {
Self::Network
}
}
impl FromStr for MergedAttackVector {
type Err = Error;
fn from_str(s: &str) -> Result<Self> {
match s {
"P" => Ok(MergedAttackVector::Physical),
"L" => Ok(MergedAttackVector::Local),
"A" => Ok(MergedAttackVector::Adjacent),
"N" => Ok(MergedAttackVector::Network),
_ => Err(Error::InvalidMetricV4 {
metric_type: MetricType::AV,
value: s.to_owned(),
}),
}
}
}
impl MetricLevel for MergedAttackVector {
fn level(self) -> f64 {
match self {
Self::Physical => 0.3,
Self::Local => 0.2,
Self::Adjacent => 0.1,
Self::Network => 0.0,
}
}
}
impl AttackVector {
pub(crate) fn merge(self, value: Option<ModifiedAttackVector>) -> MergedAttackVector {
match value {
Some(ModifiedAttackVector::NotDefined) | None => match self {
Self::Network => MergedAttackVector::Network,
Self::Adjacent => MergedAttackVector::Adjacent,
Self::Local => MergedAttackVector::Local,
Self::Physical => MergedAttackVector::Physical,
},
Some(ModifiedAttackVector::Network) => MergedAttackVector::Network,
Some(ModifiedAttackVector::Adjacent) => MergedAttackVector::Adjacent,
Some(ModifiedAttackVector::Local) => MergedAttackVector::Local,
Some(ModifiedAttackVector::Physical) => MergedAttackVector::Physical,
}
}
}
}