Roll findings up per package and image, expandable to their CVEs
The findings table listed every CVE, which is thousands of rows on a real host. It now shows one row per affected package (host) or image (containers) with its severity split, how many findings it has and how many of them have a fix. Clicking a row loads and shows the CVEs of that group; collapsing keeps them cached. The rollup is a GROUP BY in SQLite behind a new groups endpoint, so the page loads a few dozen rows instead of the full finding list, and the flat list gained a package filter to expand one group. Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
@ -28,7 +28,7 @@ impl Modify for BearerAuth {
|
||||
crate::jobs::list, crate::jobs::kinds, crate::jobs::get_one, crate::jobs::run,
|
||||
crate::system::inventory, crate::system::upgrade,
|
||||
crate::cluster::overview, crate::cluster::restart, crate::cluster::scale, crate::cluster::set_image,
|
||||
crate::vulnerabilities::list, crate::vulnerabilities::summary, crate::vulnerabilities::targets, crate::vulnerabilities::set_status,
|
||||
crate::vulnerabilities::list, crate::vulnerabilities::groups, crate::vulnerabilities::summary, crate::vulnerabilities::targets, crate::vulnerabilities::set_status,
|
||||
crate::settings::get_notifications, crate::settings::put_notifications,
|
||||
crate::backups::list_targets, crate::backups::get_target, crate::backups::create_target, crate::backups::update_target,
|
||||
crate::backups::delete_target, crate::backups::test_target, crate::backups::list_strategies, crate::backups::get_strategy,
|
||||
|
||||
@ -3,7 +3,7 @@ use application::vuln_service::{Summary, TargetSummary};
|
||||
use axum::extract::{Path, Query, State};
|
||||
use axum::routing::{get, post};
|
||||
use axum::{Json, Router};
|
||||
use domain::vuln::{Finding, FindingFilter, FindingStatus, Severity, TargetKind};
|
||||
use domain::vuln::{Finding, FindingFilter, FindingGroup, FindingStatus, Severity, TargetKind};
|
||||
use domain::DomainError;
|
||||
use serde::{Deserialize, Serialize};
|
||||
use utoipa::ToSchema;
|
||||
@ -16,6 +16,7 @@ use crate::AppState;
|
||||
pub fn router() -> Router<AppState> {
|
||||
Router::new()
|
||||
.route("/", get(list))
|
||||
.route("/groups", get(groups))
|
||||
.route("/summary", get(summary))
|
||||
.route("/targets", get(targets))
|
||||
.route("/{id}/status", post(set_status))
|
||||
@ -26,36 +27,55 @@ pub struct ListQuery {
|
||||
pub min_severity: Option<String>,
|
||||
/// `host` (OS, packages, applications) or `container` (images in the cluster).
|
||||
pub scope: Option<String>,
|
||||
/// Restrict to one package; used to expand a host group.
|
||||
pub package: Option<String>,
|
||||
pub target: Option<String>,
|
||||
pub status: Option<String>,
|
||||
#[serde(default)]
|
||||
pub include_fixed: bool,
|
||||
}
|
||||
|
||||
impl ListQuery {
|
||||
fn into_filter(self) -> Result<FindingFilter, DomainError> {
|
||||
let target_kind = match self.scope.as_deref() {
|
||||
None => None,
|
||||
Some(s) => Some(TargetKind::parse_scope(s).ok_or_else(|| {
|
||||
DomainError::Validation(format!(
|
||||
"unknown scope '{s}', expected 'host' or 'container'"
|
||||
))
|
||||
})?),
|
||||
};
|
||||
Ok(FindingFilter {
|
||||
min_severity: self.min_severity.as_deref().map(Severity::parse),
|
||||
target_kind,
|
||||
target: self.target,
|
||||
package: self.package,
|
||||
status: self.status.as_deref().and_then(FindingStatus::parse),
|
||||
include_fixed: self.include_fixed,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
#[utoipa::path(get, path = "/api/vulnerabilities", tag = "vulnerabilities", security(("bearer" = [])),
|
||||
params(("min_severity" = Option<String>, Query), ("scope" = Option<String>, Query), ("target" = Option<String>, Query), ("status" = Option<String>, Query), ("include_fixed" = Option<bool>, Query)),
|
||||
params(("min_severity" = Option<String>, Query), ("scope" = Option<String>, Query), ("package" = Option<String>, Query), ("target" = Option<String>, Query), ("status" = Option<String>, Query), ("include_fixed" = Option<bool>, Query)),
|
||||
responses((status = 200, body = Vec<Object>)))]
|
||||
async fn list(
|
||||
State(state): State<AppState>,
|
||||
_: AuthUser,
|
||||
Query(q): Query<ListQuery>,
|
||||
) -> Result<Json<Vec<Finding>>, ApiError> {
|
||||
let target_kind = match q.scope.as_deref() {
|
||||
None => None,
|
||||
Some(s) => Some(TargetKind::parse_scope(s).ok_or_else(|| {
|
||||
DomainError::Validation(format!(
|
||||
"unknown scope '{s}', expected 'host' or 'container'"
|
||||
))
|
||||
})?),
|
||||
};
|
||||
let filter = FindingFilter {
|
||||
min_severity: q.min_severity.as_deref().map(Severity::parse),
|
||||
target_kind,
|
||||
target: q.target,
|
||||
status: q.status.as_deref().and_then(FindingStatus::parse),
|
||||
include_fixed: q.include_fixed,
|
||||
};
|
||||
Ok(Json(state.vulns.list(filter).await?))
|
||||
Ok(Json(state.vulns.list(q.into_filter()?).await?))
|
||||
}
|
||||
|
||||
#[utoipa::path(get, path = "/api/vulnerabilities/groups", tag = "vulnerabilities", security(("bearer" = [])),
|
||||
params(("scope" = Option<String>, Query), ("min_severity" = Option<String>, Query), ("target" = Option<String>, Query), ("status" = Option<String>, Query)),
|
||||
responses((status = 200, body = Vec<Object>), (status = 422)))]
|
||||
async fn groups(
|
||||
State(state): State<AppState>,
|
||||
_: AuthUser,
|
||||
Query(q): Query<ListQuery>,
|
||||
) -> Result<Json<Vec<FindingGroup>>, ApiError> {
|
||||
Ok(Json(state.vulns.groups(q.into_filter()?).await?))
|
||||
}
|
||||
|
||||
#[derive(Serialize, ToSchema)]
|
||||
|
||||
@ -257,3 +257,76 @@ async fn targets_carry_their_scope_and_open_count() {
|
||||
assert_eq!(image["kind"], "image");
|
||||
assert!(image["open"].as_u64().unwrap() >= 1);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn findings_are_rolled_up_per_package_and_image() {
|
||||
let app = test_app_with_admin().await;
|
||||
let token = common::login(&app, ADMIN, PW).await.access;
|
||||
run_scan(&app, &token).await;
|
||||
|
||||
let host = get(
|
||||
&app,
|
||||
"/api/vulnerabilities/groups?scope=host&min_severity=unknown",
|
||||
Some(&token),
|
||||
)
|
||||
.await;
|
||||
assert_eq!(host.status, StatusCode::OK, "{}", host.json);
|
||||
let groups = host.json.as_array().unwrap();
|
||||
let flat = get(
|
||||
&app,
|
||||
"/api/vulnerabilities?scope=host&min_severity=unknown",
|
||||
Some(&token),
|
||||
)
|
||||
.await;
|
||||
let flat_len = flat.json.as_array().unwrap().len();
|
||||
assert!(groups.len() < flat_len, "fewer rows than findings");
|
||||
assert_eq!(
|
||||
groups
|
||||
.iter()
|
||||
.map(|g| g["total"].as_u64().unwrap())
|
||||
.sum::<u64>() as usize,
|
||||
flat_len
|
||||
);
|
||||
let zlib = groups.iter().find(|g| g["key"] == "zlib1g").unwrap();
|
||||
assert_eq!(zlib["kind"], "os");
|
||||
assert_eq!(zlib["source"], "debian");
|
||||
assert_eq!(zlib["counts"]["critical"], 1);
|
||||
assert!(zlib["installed"].as_str().unwrap().starts_with("1:1.2.13"));
|
||||
|
||||
let images = get(
|
||||
&app,
|
||||
"/api/vulnerabilities/groups?scope=container&min_severity=unknown",
|
||||
Some(&token),
|
||||
)
|
||||
.await;
|
||||
let images = images.json.as_array().unwrap();
|
||||
let gitea = images
|
||||
.iter()
|
||||
.find(|g| g["key"].as_str().unwrap().contains("gitea"))
|
||||
.unwrap();
|
||||
assert_eq!(gitea["kind"], "image");
|
||||
assert!(gitea["total"].as_u64().unwrap() >= 1);
|
||||
assert!(gitea["packages"].as_u64().unwrap() >= 1);
|
||||
|
||||
// a group is expanded by filtering the flat list
|
||||
let pkg = get(
|
||||
&app,
|
||||
"/api/vulnerabilities?scope=host&package=zlib1g&min_severity=unknown",
|
||||
Some(&token),
|
||||
)
|
||||
.await;
|
||||
let pkg = pkg.json.as_array().unwrap();
|
||||
assert_eq!(pkg.len(), zlib["total"].as_u64().unwrap() as usize);
|
||||
assert!(pkg.iter().all(|f| f["package"] == "zlib1g"));
|
||||
|
||||
assert_eq!(
|
||||
get(&app, "/api/vulnerabilities/groups?scope=nope", Some(&token))
|
||||
.await
|
||||
.status,
|
||||
StatusCode::UNPROCESSABLE_ENTITY
|
||||
);
|
||||
assert_eq!(
|
||||
get(&app, "/api/vulnerabilities/groups", None).await.status,
|
||||
StatusCode::UNAUTHORIZED
|
||||
);
|
||||
}
|
||||
|
||||
@ -507,7 +507,8 @@ impl ClusterGateway for MemCluster {
|
||||
|
||||
use domain::ports::{FindingRepository, VulnerabilityScanner};
|
||||
use domain::vuln::{
|
||||
Finding, FindingFilter, FindingStatus, RawFinding, Severity, SeverityCounts, TargetKind,
|
||||
Finding, FindingFilter, FindingGroup, FindingStatus, RawFinding, Severity, SeverityCounts,
|
||||
TargetKind,
|
||||
};
|
||||
|
||||
pub fn raw(
|
||||
@ -630,6 +631,7 @@ impl FindingRepository for MemFindings {
|
||||
.filter(|f| filter.include_fixed || f.status != FindingStatus::Fixed)
|
||||
.filter(|f| filter.min_severity.is_none_or(|m| f.raw.severity >= m))
|
||||
.filter(|f| filter.target_kind.is_none_or(|k| f.target_kind == k))
|
||||
.filter(|f| filter.package.as_ref().is_none_or(|p| &f.raw.package == p))
|
||||
.filter(|f| filter.target.as_ref().is_none_or(|t| &f.target == t))
|
||||
.filter(|f| filter.status.is_none_or(|s| f.status == s))
|
||||
.cloned()
|
||||
@ -642,6 +644,71 @@ impl FindingRepository for MemFindings {
|
||||
});
|
||||
Ok(v)
|
||||
}
|
||||
async fn groups(&self, filter: &FindingFilter) -> Result<Vec<FindingGroup>, DomainError> {
|
||||
let per_image = filter.target_kind == Some(TargetKind::Image);
|
||||
let mut by_key: HashMap<String, FindingGroup> = HashMap::new();
|
||||
let mut packages: HashMap<String, std::collections::HashSet<String>> = HashMap::new();
|
||||
for f in self.list(filter).await? {
|
||||
let key = if per_image {
|
||||
f.target.clone()
|
||||
} else {
|
||||
f.raw.package.clone()
|
||||
};
|
||||
packages
|
||||
.entry(key.clone())
|
||||
.or_default()
|
||||
.insert(f.raw.package.clone());
|
||||
let g = by_key.entry(key.clone()).or_insert_with(|| FindingGroup {
|
||||
key,
|
||||
kind: f.target_kind,
|
||||
source: if per_image {
|
||||
String::new()
|
||||
} else {
|
||||
f.raw.source.clone()
|
||||
},
|
||||
installed: if per_image {
|
||||
String::new()
|
||||
} else {
|
||||
f.raw.installed_version.clone()
|
||||
},
|
||||
counts: SeverityCounts::default(),
|
||||
total: 0,
|
||||
fixable: 0,
|
||||
packages: 0,
|
||||
});
|
||||
g.counts.add(f.raw.severity);
|
||||
g.total += 1;
|
||||
if f.raw.fixed_version.is_some() {
|
||||
g.fixable += 1;
|
||||
}
|
||||
}
|
||||
let mut groups: Vec<FindingGroup> = by_key
|
||||
.into_values()
|
||||
.map(|mut g| {
|
||||
g.packages = packages.get(&g.key).map(|p| p.len()).unwrap_or(1);
|
||||
g
|
||||
})
|
||||
.collect();
|
||||
let worst = |g: &FindingGroup| {
|
||||
Severity::ALL
|
||||
.iter()
|
||||
.position(|s| match s {
|
||||
Severity::Critical => g.counts.critical > 0,
|
||||
Severity::High => g.counts.high > 0,
|
||||
Severity::Medium => g.counts.medium > 0,
|
||||
Severity::Low => g.counts.low > 0,
|
||||
Severity::Unknown => g.counts.unknown > 0,
|
||||
})
|
||||
.unwrap_or(usize::MAX)
|
||||
};
|
||||
groups.sort_by(|a, b| {
|
||||
worst(a)
|
||||
.cmp(&worst(b))
|
||||
.then(b.total.cmp(&a.total))
|
||||
.then(a.key.cmp(&b.key))
|
||||
});
|
||||
Ok(groups)
|
||||
}
|
||||
async fn counts(&self, kind: Option<TargetKind>) -> Result<SeverityCounts, DomainError> {
|
||||
let mut c = SeverityCounts::default();
|
||||
for f in
|
||||
|
||||
@ -417,17 +417,38 @@ async fn host_findings_are_grouped_per_package_and_containers_per_image() {
|
||||
.with(
|
||||
"os",
|
||||
Ok(vec![
|
||||
raw("CVE-1", "openssl", "3.0.1", Severity::Critical, Some("3.0.2")),
|
||||
raw(
|
||||
"CVE-1",
|
||||
"openssl",
|
||||
"3.0.1",
|
||||
Severity::Critical,
|
||||
Some("3.0.2"),
|
||||
),
|
||||
raw("CVE-2", "openssl", "3.0.1", Severity::Low, None),
|
||||
go,
|
||||
]),
|
||||
)
|
||||
.with(GITEA, Ok(vec![raw("CVE-3", "git", "2.39", Severity::High, None), raw("CVE-4", "curl", "7.8", Severity::Medium, None)]))
|
||||
.with(PG, Ok(vec![raw("CVE-5", "libssl3", "3.0", Severity::Low, None)]));
|
||||
.with(
|
||||
GITEA,
|
||||
Ok(vec![
|
||||
raw("CVE-3", "git", "2.39", Severity::High, None),
|
||||
raw("CVE-4", "curl", "7.8", Severity::Medium, None),
|
||||
]),
|
||||
)
|
||||
.with(
|
||||
PG,
|
||||
Ok(vec![raw("CVE-5", "libssl3", "3.0", Severity::Low, None)]),
|
||||
);
|
||||
let (_f, svc) = fixture(scanner);
|
||||
svc.scan(&VecLog::default()).await.unwrap();
|
||||
|
||||
let host = svc.groups(FindingFilter { target_kind: Some(TargetKind::Os), ..Default::default() }).await.unwrap();
|
||||
let host = svc
|
||||
.groups(FindingFilter {
|
||||
target_kind: Some(TargetKind::Os),
|
||||
..Default::default()
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(host.len(), 2, "one row per package, not per CVE");
|
||||
// worst severity first
|
||||
assert_eq!(host[0].key, "openssl");
|
||||
@ -440,30 +461,75 @@ async fn host_findings_are_grouped_per_package_and_containers_per_image() {
|
||||
assert_eq!(host[1].key, "stdlib");
|
||||
assert_eq!(host[1].source, "gobinary");
|
||||
|
||||
let images = svc.groups(FindingFilter { target_kind: Some(TargetKind::Image), ..Default::default() }).await.unwrap();
|
||||
assert_eq!(images.iter().map(|g| g.key.as_str()).collect::<Vec<_>>(), vec![GITEA, PG]);
|
||||
let images = svc
|
||||
.groups(FindingFilter {
|
||||
target_kind: Some(TargetKind::Image),
|
||||
..Default::default()
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(
|
||||
images.iter().map(|g| g.key.as_str()).collect::<Vec<_>>(),
|
||||
vec![GITEA, PG]
|
||||
);
|
||||
assert_eq!(images[0].total, 2);
|
||||
assert_eq!(images[0].packages, 2, "distinct packages in the image");
|
||||
|
||||
// the group list honours the other filters
|
||||
let high = svc
|
||||
.groups(FindingFilter { target_kind: Some(TargetKind::Os), min_severity: Some(Severity::High), ..Default::default() })
|
||||
.groups(FindingFilter {
|
||||
target_kind: Some(TargetKind::Os),
|
||||
min_severity: Some(Severity::High),
|
||||
..Default::default()
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(high.iter().map(|g| g.key.as_str()).collect::<Vec<_>>(), vec!["openssl", "stdlib"]);
|
||||
assert_eq!(high[0].total, 1, "the low finding is filtered out of the counts");
|
||||
assert_eq!(
|
||||
high.iter().map(|g| g.key.as_str()).collect::<Vec<_>>(),
|
||||
vec!["openssl", "stdlib"]
|
||||
);
|
||||
assert_eq!(
|
||||
high[0].total, 1,
|
||||
"the low finding is filtered out of the counts"
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn findings_of_one_group_can_be_listed() {
|
||||
let scanner = FakeScanner::default()
|
||||
.with("os", Ok(vec![raw("CVE-1", "openssl", "3.0.1", Severity::Critical, None), raw("CVE-2", "bash", "5.2", Severity::Low, None)]))
|
||||
.with(GITEA, Ok(vec![raw("CVE-3", "git", "2.39", Severity::High, None)]));
|
||||
.with(
|
||||
"os",
|
||||
Ok(vec![
|
||||
raw("CVE-1", "openssl", "3.0.1", Severity::Critical, None),
|
||||
raw("CVE-2", "bash", "5.2", Severity::Low, None),
|
||||
]),
|
||||
)
|
||||
.with(
|
||||
GITEA,
|
||||
Ok(vec![raw("CVE-3", "git", "2.39", Severity::High, None)]),
|
||||
);
|
||||
let (_f, svc) = fixture(scanner);
|
||||
svc.scan(&VecLog::default()).await.unwrap();
|
||||
|
||||
let pkg = svc.list(FindingFilter { package: Some("openssl".into()), ..Default::default() }).await.unwrap();
|
||||
assert_eq!(pkg.iter().map(|f| f.raw.cve_id.as_str()).collect::<Vec<_>>(), vec!["CVE-1"]);
|
||||
let image = svc.list(FindingFilter { target: Some(GITEA.into()), ..Default::default() }).await.unwrap();
|
||||
let pkg = svc
|
||||
.list(FindingFilter {
|
||||
package: Some("openssl".into()),
|
||||
..Default::default()
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(
|
||||
pkg.iter()
|
||||
.map(|f| f.raw.cve_id.as_str())
|
||||
.collect::<Vec<_>>(),
|
||||
vec!["CVE-1"]
|
||||
);
|
||||
let image = svc
|
||||
.list(FindingFilter {
|
||||
target: Some(GITEA.into()),
|
||||
..Default::default()
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(image.len(), 1);
|
||||
}
|
||||
|
||||
@ -7,8 +7,8 @@ use async_trait::async_trait;
|
||||
use chrono::{DateTime, Utc};
|
||||
use domain::ports::{ClusterGateway, FindingRepository, LineSink, VulnerabilityScanner};
|
||||
use domain::vuln::{
|
||||
Finding, FindingFilter, FindingStatus, RawFinding, ScanReport, Severity, SeverityCounts,
|
||||
TargetKind,
|
||||
Finding, FindingFilter, FindingGroup, FindingStatus, RawFinding, ScanReport, Severity,
|
||||
SeverityCounts, TargetKind,
|
||||
};
|
||||
use domain::DomainError;
|
||||
use uuid::Uuid;
|
||||
@ -237,6 +237,11 @@ impl VulnerabilityService {
|
||||
self.findings.list(&filter).await
|
||||
}
|
||||
|
||||
/// Findings rolled up per package (host) or per image (containers).
|
||||
pub async fn groups(&self, filter: FindingFilter) -> Result<Vec<FindingGroup>, DomainError> {
|
||||
self.findings.groups(&filter).await
|
||||
}
|
||||
|
||||
/// Scanned targets that still have open findings, host first.
|
||||
pub async fn targets(&self) -> Result<Vec<TargetSummary>, DomainError> {
|
||||
let mut by_target: std::collections::HashMap<(TargetKind, String), usize> =
|
||||
|
||||
@ -496,7 +496,8 @@ impl domain::ports::InventoryRepository for SqliteInventory {
|
||||
}
|
||||
|
||||
use domain::vuln::{
|
||||
Finding, FindingFilter, FindingStatus, RawFinding, Severity, SeverityCounts, TargetKind,
|
||||
Finding, FindingFilter, FindingGroup, FindingStatus, RawFinding, Severity, SeverityCounts,
|
||||
TargetKind,
|
||||
};
|
||||
|
||||
pub struct SqliteFindings(pub DbPool);
|
||||
@ -629,6 +630,9 @@ impl domain::ports::FindingRepository for SqliteFindings {
|
||||
if filter.target_kind.is_some() {
|
||||
sql.push_str(" AND target_kind = ?");
|
||||
}
|
||||
if filter.package.is_some() {
|
||||
sql.push_str(" AND package = ?");
|
||||
}
|
||||
if filter.min_severity.is_some() {
|
||||
sql.push_str(&format!(" AND {SEVERITY_RANK_SQL} >= ?"));
|
||||
}
|
||||
@ -645,6 +649,9 @@ impl domain::ports::FindingRepository for SqliteFindings {
|
||||
if let Some(k) = filter.target_kind {
|
||||
q = q.bind(k.as_str());
|
||||
}
|
||||
if let Some(p) = &filter.package {
|
||||
q = q.bind(p);
|
||||
}
|
||||
if let Some(m) = filter.min_severity {
|
||||
q = q.bind(severity_rank(m));
|
||||
}
|
||||
@ -653,6 +660,94 @@ impl domain::ports::FindingRepository for SqliteFindings {
|
||||
.map(|rows| rows.iter().map(finding_from_row).collect())
|
||||
.map_err(storage)
|
||||
}
|
||||
async fn groups(&self, filter: &FindingFilter) -> Result<Vec<FindingGroup>, DomainError> {
|
||||
// Host findings roll up per package, container findings per image.
|
||||
let per_image = filter.target_kind == Some(TargetKind::Image);
|
||||
let key = if per_image { "target" } else { "package" };
|
||||
let mut sql = format!(
|
||||
"SELECT {key} AS key, target_kind, \
|
||||
COALESCE(MAX(source), '') AS source, COALESCE(MAX(installed_version), '') AS installed, \
|
||||
COUNT(*) AS total, \
|
||||
SUM(fixed_version IS NOT NULL AND fixed_version != '') AS fixable, \
|
||||
COUNT(DISTINCT package) AS packages, \
|
||||
SUM(severity = 'critical') AS critical, SUM(severity = 'high') AS high, \
|
||||
SUM(severity = 'medium') AS medium, SUM(severity = 'low') AS low, \
|
||||
SUM(severity NOT IN ('critical', 'high', 'medium', 'low')) AS unknown \
|
||||
FROM findings WHERE 1=1"
|
||||
);
|
||||
if !filter.include_fixed {
|
||||
sql.push_str(" AND status != 'fixed'");
|
||||
}
|
||||
if filter.status.is_some() {
|
||||
sql.push_str(" AND status = ?");
|
||||
}
|
||||
if filter.target.is_some() {
|
||||
sql.push_str(" AND target = ?");
|
||||
}
|
||||
if filter.target_kind.is_some() {
|
||||
sql.push_str(" AND target_kind = ?");
|
||||
}
|
||||
if filter.package.is_some() {
|
||||
sql.push_str(" AND package = ?");
|
||||
}
|
||||
if filter.min_severity.is_some() {
|
||||
sql.push_str(&format!(" AND {SEVERITY_RANK_SQL} >= ?"));
|
||||
}
|
||||
// worst severity first, then most findings, then by name
|
||||
sql.push_str(&format!(
|
||||
" GROUP BY {key}, target_kind ORDER BY MAX({SEVERITY_RANK_SQL}) DESC, total DESC, key"
|
||||
));
|
||||
let mut q = sqlx::query(&sql);
|
||||
if let Some(st) = filter.status {
|
||||
q = q.bind(st.as_str());
|
||||
}
|
||||
if let Some(t) = &filter.target {
|
||||
q = q.bind(t);
|
||||
}
|
||||
if let Some(k) = filter.target_kind {
|
||||
q = q.bind(k.as_str());
|
||||
}
|
||||
if let Some(p) = &filter.package {
|
||||
q = q.bind(p);
|
||||
}
|
||||
if let Some(m) = filter.min_severity {
|
||||
q = q.bind(severity_rank(m));
|
||||
}
|
||||
let rows = q.fetch_all(&self.0).await.map_err(storage)?;
|
||||
Ok(rows
|
||||
.iter()
|
||||
.map(|r| {
|
||||
let n = |c: &str| r.get::<i64, _>(c) as usize;
|
||||
let kind = TargetKind::parse(r.get::<String, _>("target_kind").as_str())
|
||||
.unwrap_or(TargetKind::Os);
|
||||
FindingGroup {
|
||||
key: r.get("key"),
|
||||
kind,
|
||||
source: if kind == TargetKind::Image {
|
||||
String::new()
|
||||
} else {
|
||||
r.get("source")
|
||||
},
|
||||
installed: if kind == TargetKind::Image {
|
||||
String::new()
|
||||
} else {
|
||||
r.get("installed")
|
||||
},
|
||||
counts: SeverityCounts {
|
||||
critical: n("critical"),
|
||||
high: n("high"),
|
||||
medium: n("medium"),
|
||||
low: n("low"),
|
||||
unknown: n("unknown"),
|
||||
},
|
||||
total: n("total"),
|
||||
fixable: n("fixable"),
|
||||
packages: n("packages"),
|
||||
}
|
||||
})
|
||||
.collect())
|
||||
}
|
||||
|
||||
async fn counts(&self, kind: Option<TargetKind>) -> Result<SeverityCounts, DomainError> {
|
||||
let sql = match kind {
|
||||
Some(_) => "SELECT severity, COUNT(*) FROM findings WHERE status != 'fixed' AND target_kind = ? GROUP BY severity",
|
||||
@ -807,6 +902,77 @@ mod finding_tests {
|
||||
"source survives the roundtrip"
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn groups_roll_up_per_package_and_per_image() {
|
||||
let pool = crate::connect("sqlite::memory:").await.unwrap();
|
||||
let repo = SqliteFindings(pool);
|
||||
let mut a = finding("os", TargetKind::Os, "CVE-A", Severity::Critical);
|
||||
a.raw.package = "openssl".into();
|
||||
a.raw.fixed_version = Some("3.0.2".into());
|
||||
let mut b = finding("os", TargetKind::Os, "CVE-B", Severity::Low);
|
||||
b.raw.package = "openssl".into();
|
||||
let mut c = finding("os", TargetKind::Os, "CVE-C", Severity::High);
|
||||
c.raw.package = "stdlib".into();
|
||||
c.raw.source = "gobinary".into();
|
||||
let mut d = finding("img:1", TargetKind::Image, "CVE-D", Severity::Medium);
|
||||
d.raw.package = "curl".into();
|
||||
let mut e = finding("img:1", TargetKind::Image, "CVE-E", Severity::High);
|
||||
e.raw.package = "git".into();
|
||||
for f in [&a, &b, &c, &d, &e] {
|
||||
repo.insert(f).await.unwrap();
|
||||
}
|
||||
|
||||
let host = repo
|
||||
.groups(&FindingFilter {
|
||||
target_kind: Some(TargetKind::Os),
|
||||
..Default::default()
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(
|
||||
host.iter().map(|g| g.key.as_str()).collect::<Vec<_>>(),
|
||||
vec!["openssl", "stdlib"]
|
||||
);
|
||||
assert_eq!(host[0].counts.critical, 1);
|
||||
assert_eq!(host[0].counts.low, 1);
|
||||
assert_eq!(host[0].total, 2);
|
||||
assert_eq!(host[0].fixable, 1);
|
||||
assert_eq!(host[0].installed, "1");
|
||||
assert_eq!(host[1].source, "gobinary");
|
||||
|
||||
let images = repo
|
||||
.groups(&FindingFilter {
|
||||
target_kind: Some(TargetKind::Image),
|
||||
..Default::default()
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(images.len(), 1);
|
||||
assert_eq!(images[0].key, "img:1");
|
||||
assert_eq!(images[0].total, 2);
|
||||
assert_eq!(images[0].packages, 2);
|
||||
|
||||
// fixed findings stay out unless asked for, and the package filter drills in
|
||||
repo.set_status(b.id, FindingStatus::Fixed).await.unwrap();
|
||||
let host = repo
|
||||
.groups(&FindingFilter {
|
||||
target_kind: Some(TargetKind::Os),
|
||||
..Default::default()
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(host[0].total, 1);
|
||||
let drilled = repo
|
||||
.list(&FindingFilter {
|
||||
package: Some("stdlib".into()),
|
||||
..Default::default()
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(drilled.len(), 1);
|
||||
assert_eq!(drilled[0].raw.cve_id, "CVE-C");
|
||||
}
|
||||
}
|
||||
|
||||
use domain::backup::{BackupRecord, BackupSource, BackupStrategy, BackupTarget, StorageKind};
|
||||
|
||||
@ -202,6 +202,17 @@ impl VulnerabilityScanner for FakeScanner {
|
||||
url: "https://avd.aquasec.com/nvd/cve-2024-5535".into(),
|
||||
source: "debian".into(),
|
||||
},
|
||||
// a second flaw in the same package: the grouped view collapses both into one row
|
||||
RawFinding {
|
||||
cve_id: "CVE-2024-2511".into(),
|
||||
severity: Severity::Medium,
|
||||
package: "openssl".into(),
|
||||
installed_version: "3.0.15-1~deb12u1".into(),
|
||||
fixed_version: Some("3.0.16-1~deb12u1".into()),
|
||||
title: "openssl: unbounded memory growth in the session cache".into(),
|
||||
url: "https://avd.aquasec.com/nvd/cve-2024-2511".into(),
|
||||
source: "debian".into(),
|
||||
},
|
||||
RawFinding {
|
||||
cve_id: "CVE-2023-45853".into(),
|
||||
severity: Severity::Critical,
|
||||
|
||||
Reference in New Issue
Block a user