🐛 Extend vulnmgmt for cnspec vuln command

Signed-off-by: Christian Zunker <[email protected]>
mondoohq · Dec 5, 2023 · 41292c2 · 41292c2
1 parent a37a388
commit 41292c2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/providers-sdk/v1/upstream/gql/vulnmgmt_gql.go b/providers-sdk/v1/upstream/gql/vulnmgmt_gql.go
@@ -219,6 +219,7 @@ func (c *MondooClient) GetIncognitoVulnReport(platform mondoogql.PlatformInput,
 				Packages []struct {
 					Package
 				}
+				Stats ReportStats
 			} `graphql:"... on AssetIncognitoVulnerabilityReport"`
 		} `graphql:"analyseIncognitoAssetVulnerabilities(input: $input)"`
 	}
@@ -236,6 +237,7 @@ func (c *MondooClient) GetIncognitoVulnReport(platform mondoogql.PlatformInput,
 		Advisories: make([]*Advisory, len(m.AssetVulnerabilityReportResponse.AssetIncognitoVulnerabilityReport.Advisories)),
 		Cves:       make([]*Cve, len(m.AssetVulnerabilityReportResponse.AssetIncognitoVulnerabilityReport.Cves)),
 		Packages:   make([]*Package, len(m.AssetVulnerabilityReportResponse.AssetIncognitoVulnerabilityReport.Packages)),
+		Stats:      &m.AssetVulnerabilityReportResponse.AssetIncognitoVulnerabilityReport.Stats,
 	}
 
 	for i := range m.AssetVulnerabilityReportResponse.AssetIncognitoVulnerabilityReport.Advisories {

diff --git a/providers/vsphere/resources/vulnmgmt.go b/providers/vsphere/resources/vulnmgmt.go
@@ -169,6 +169,7 @@ func (v *mqlVulnmgmt) getReport() (*gql.VulnReport, error) {
 }
 
 func (v *mqlVulnmgmt) getIncognitoReport(mondooClient *gql.MondooClient) (*gql.VulnReport, error) {
+	// FIXME: wrong connection
 	conn := v.MqlRuntime.Connection.(shared.Connection)
 	platform := conn.Asset().Platform