use crate::csaf::types::csaf_vuln_metric::CsafVulnerabilityMetric;
use crate::csaf_traits::{
ContentTrait, CsafTrait, MetricTrait, ProductStatusGroup, ProductStatusGroupMap, VulnerabilityTrait,
};
use crate::validation::ValidationError;
use std::collections::HashSet;
fn create_missing_cvss_v4_error(instance_path: String, cvss_versions: &[CsafVulnerabilityMetric]) -> ValidationError {
let versions_str = cvss_versions
.iter()
.map(|v| v.to_string())
.collect::<Vec<String>>()
.join(", ");
ValidationError {
message: format!("The metric contains {versions_str} but does not include a CVSS v4.0 score."),
instance_path,
}
}
fn create_affected_product_not_covered_error(product_id: &str, instance_path: String) -> ValidationError {
ValidationError {
message: format!("Affected product {product_id} is not covered by any CVSS score."),
instance_path,
}
}
pub fn test_6_3_12_missing_cvss_v4(doc: &impl CsafTrait) -> Result<(), Vec<ValidationError>> {
let mut errors: Option<Vec<ValidationError>> = None;
for (v_i, vulnerability) in doc.get_vulnerabilities().iter().enumerate() {
let mut products_covered_by_cvss: HashSet<String> = HashSet::new();
if let Some(metrics) = vulnerability.get_metrics() {
for (m_i, metric) in metrics.iter().enumerate() {
let content = metric.get_content();
if content.has_any_cvss() {
for product_id in metric.get_products() {
products_covered_by_cvss.insert(product_id.to_owned());
}
if !content.has_cvss_v4() {
let path = content.get_content_json_path(v_i, m_i);
let cvss_types = content.get_cvss_metric_types();
errors
.get_or_insert_default()
.push(create_missing_cvss_v4_error(path, &cvss_types));
}
}
}
}
if let Some(product_status) = vulnerability.get_product_status() {
let status_map = ProductStatusGroupMap::from(product_status);
if let Some(affected) = status_map.get(&ProductStatusGroup::Affected) {
for (product_id, entries) in affected {
if !products_covered_by_cvss.contains(product_id) {
for entry in entries {
errors
.get_or_insert_default()
.push(create_affected_product_not_covered_error(
product_id,
entry.json_path(v_i),
));
}
}
}
}
}
}
errors.map_or(Ok(()), Err)
}
crate::test_validation::impl_validator!(csaf2_1, ValidatorForTest6_3_12, test_6_3_12_missing_cvss_v4);
#[cfg(test)]
mod tests {
use super::*;
use crate::csaf2_1::testcases::TESTS_2_1;
#[test]
fn test_test_6_3_12() {
let case_01_cvss_v3_1_only = Err(vec![create_missing_cvss_v4_error(
"/vulnerabilities/0/metrics/0/content".to_string(),
&[CsafVulnerabilityMetric::CvssV3("3.1".to_string())],
)]);
let case_02_cvss_v3_0_only = Err(vec![create_missing_cvss_v4_error(
"/vulnerabilities/0/metrics/0/content".to_string(),
&[CsafVulnerabilityMetric::CvssV3("3.0".to_string())],
)]);
let case_03_cvss_v2_only = Err(vec![create_missing_cvss_v4_error(
"/vulnerabilities/0/metrics/0/content".to_string(),
&[CsafVulnerabilityMetric::CvssV2("2.0".to_string())],
)]);
let case_04_multiple_vulns_two_without_cvss_v4 = Err(vec![
create_missing_cvss_v4_error(
"/vulnerabilities/0/metrics/0/content".to_string(),
&[
CsafVulnerabilityMetric::CvssV2("2.0".to_string()),
CsafVulnerabilityMetric::CvssV3("3.1".to_string()),
],
),
create_missing_cvss_v4_error(
"/vulnerabilities/2/metrics/0/content".to_string(),
&[
CsafVulnerabilityMetric::CvssV2("2.0".to_string()),
CsafVulnerabilityMetric::CvssV3("3.1".to_string()),
],
),
]);
let case_05_uncovered_affected = Err(vec![
create_affected_product_not_covered_error(
"CSAFPID-9080701",
"/vulnerabilities/1/product_status/first_affected/1".to_string(),
),
create_missing_cvss_v4_error(
"/vulnerabilities/2/metrics/0/content".to_string(),
&[
CsafVulnerabilityMetric::CvssV2("2.0".to_string()),
CsafVulnerabilityMetric::CvssV3("3.1".to_string()),
],
),
create_affected_product_not_covered_error(
"CSAFPID-9080701",
"/vulnerabilities/2/product_status/known_affected/1".to_string(),
),
]);
let case_s01_last_affected_not_covered = Err(vec![create_affected_product_not_covered_error(
"CSAFPID-9080701",
"/vulnerabilities/0/product_status/last_affected/1".to_string(),
)]);
TESTS_2_1.test_6_3_12.expect(
case_01_cvss_v3_1_only,
case_02_cvss_v3_0_only,
case_03_cvss_v2_only,
case_04_multiple_vulns_two_without_cvss_v4,
case_05_uncovered_affected,
case_s01_last_affected_not_covered,
Ok(()),
Ok(()),
Ok(()),
Ok(()),
Ok(()),
Ok(()),
);
}
}