Skip to content
This repository has been archived by the owner on Oct 14, 2024. It is now read-only.

Commit

Permalink
Fix relation counters in demo data (#170)
Browse files Browse the repository at this point in the history 
Update counters across different related objects in the demo data. This is a temporary fix as the demo data should post the scan results and the post-processing logic should adjust the counters across all the dependencies.
  • Loading branch information
@akpsgit
akpsgit authored Mar 21, 2023
1 parent 78d8535 commit f03ca42
Showing 1 changed file with 73 additions and 71 deletions.
144 changes: 73 additions & 71 deletions backend/pkg/database/demo.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -427,58 +427,58 @@ func createScopes() (models.Scopes, error) {
func createTargets() []models.Target {
return []models.Target{
{
ScansCount: utils.PointerTo(100),
ScansCount: utils.PointerTo(1),
Summary: &models.ScanFindingsSummary{
TotalExploits: utils.PointerTo(1),
TotalMalware: utils.PointerTo(2),
TotalMisconfigurations: utils.PointerTo(3),
TotalPackages: utils.PointerTo(4),
TotalRootkits: utils.PointerTo(5),
TotalSecrets: utils.PointerTo(6),
TotalExploits: utils.PointerTo(0),
TotalMalware: utils.PointerTo(0),
TotalMisconfigurations: utils.PointerTo(0),
TotalPackages: utils.PointerTo(2),
TotalRootkits: utils.PointerTo(0),
TotalSecrets: utils.PointerTo(3),
TotalVulnerabilities: &models.VulnerabilityScanSummary{
TotalCriticalVulnerabilities: utils.PointerTo(7),
TotalHighVulnerabilities: utils.PointerTo(8),
TotalLowVulnerabilities: utils.PointerTo(9),
TotalMediumVulnerabilities: utils.PointerTo(10),
TotalNegligibleVulnerabilities: utils.PointerTo(11),
TotalCriticalVulnerabilities: utils.PointerTo(1),
TotalHighVulnerabilities: utils.PointerTo(1),
TotalLowVulnerabilities: utils.PointerTo(1),
TotalMediumVulnerabilities: utils.PointerTo(0),
TotalNegligibleVulnerabilities: utils.PointerTo(0),
},
},
TargetInfo: createVMInfo(awsInstanceEUCentral11, awsRegionEUCentral1, models.AWS),
},
{
ScansCount: utils.PointerTo(102),
ScansCount: utils.PointerTo(1),
Summary: &models.ScanFindingsSummary{
TotalExploits: utils.PointerTo(12),
TotalMalware: utils.PointerTo(22),
TotalMisconfigurations: utils.PointerTo(32),
TotalPackages: utils.PointerTo(42),
TotalRootkits: utils.PointerTo(52),
TotalSecrets: utils.PointerTo(62),
TotalExploits: utils.PointerTo(0),
TotalMalware: utils.PointerTo(0),
TotalMisconfigurations: utils.PointerTo(0),
TotalPackages: utils.PointerTo(2),
TotalRootkits: utils.PointerTo(0),
TotalSecrets: utils.PointerTo(3),
TotalVulnerabilities: &models.VulnerabilityScanSummary{
TotalCriticalVulnerabilities: utils.PointerTo(72),
TotalHighVulnerabilities: utils.PointerTo(82),
TotalLowVulnerabilities: utils.PointerTo(92),
TotalMediumVulnerabilities: utils.PointerTo(102),
TotalNegligibleVulnerabilities: utils.PointerTo(112),
TotalCriticalVulnerabilities: utils.PointerTo(1),
TotalHighVulnerabilities: utils.PointerTo(1),
TotalLowVulnerabilities: utils.PointerTo(1),
TotalMediumVulnerabilities: utils.PointerTo(0),
TotalNegligibleVulnerabilities: utils.PointerTo(0),
},
},
TargetInfo: createVMInfo(awsInstanceEUCentral12, awsRegionEUCentral1, models.AWS),
},
{
ScansCount: utils.PointerTo(103),
ScansCount: utils.PointerTo(1),
Summary: &models.ScanFindingsSummary{
TotalExploits: utils.PointerTo(13),
TotalMalware: utils.PointerTo(23),
TotalMisconfigurations: utils.PointerTo(33),
TotalPackages: utils.PointerTo(43),
TotalRootkits: utils.PointerTo(53),
TotalSecrets: utils.PointerTo(63),
TotalExploits: utils.PointerTo(2),
TotalMalware: utils.PointerTo(3),
TotalMisconfigurations: utils.PointerTo(3),
TotalPackages: utils.PointerTo(0),
TotalRootkits: utils.PointerTo(3),
TotalSecrets: utils.PointerTo(0),
TotalVulnerabilities: &models.VulnerabilityScanSummary{
TotalCriticalVulnerabilities: utils.PointerTo(73),
TotalHighVulnerabilities: utils.PointerTo(83),
TotalLowVulnerabilities: utils.PointerTo(93),
TotalMediumVulnerabilities: utils.PointerTo(103),
TotalNegligibleVulnerabilities: utils.PointerTo(113),
TotalCriticalVulnerabilities: utils.PointerTo(0),
TotalHighVulnerabilities: utils.PointerTo(0),
TotalLowVulnerabilities: utils.PointerTo(0),
TotalMediumVulnerabilities: utils.PointerTo(0),
TotalNegligibleVulnerabilities: utils.PointerTo(0),
},
},
TargetInfo: createVMInfo(awsInstanceUSEast11, awsRegionUSEast1, models.AWS),
Expand Down Expand Up @@ -642,14 +642,14 @@ func createScanConfigs() []models.ScanConfig {

return []models.ScanConfig{
{
Name: utils.PointerTo("demo scan 1"),
Name: utils.PointerTo("Scan Config 1"),
ScanFamiliesConfig: scanFamiliesConfig1,
Scheduled: &scheduled1,
Scope: &scanScopeType1,
MaxParallelScanners: utils.PointerTo(2),
},
{
Name: utils.PointerTo("demo scan 2"),
Name: utils.PointerTo("Scan Config 2"),
ScanFamiliesConfig: scanFamiliesConfig2,
Scheduled: &scanConfig2Scheduled,
Scope: &scanScopeType2,
Expand All @@ -665,57 +665,59 @@ func createScans(targets []models.Target, scanConfigs []models.ScanConfig) []mod
scan1Targets := []string{*targets[0].Id, *targets[1].Id}

scan1Summary := &models.ScanSummary{
JobsCompleted: utils.PointerTo[int](23),
JobsCompleted: utils.PointerTo[int](2),
JobsLeftToRun: utils.PointerTo[int](0),
TotalExploits: utils.PointerTo[int](14),
TotalMalware: utils.PointerTo[int](44),
TotalMisconfigurations: utils.PointerTo[int](9),
TotalPackages: utils.PointerTo[int](4221),
TotalRootkits: utils.PointerTo[int](1),
TotalSecrets: utils.PointerTo[int](0),
TotalExploits: utils.PointerTo[int](0),
TotalMalware: utils.PointerTo[int](0),
TotalMisconfigurations: utils.PointerTo[int](0),
TotalPackages: utils.PointerTo[int](4),
TotalRootkits: utils.PointerTo[int](0),
TotalSecrets: utils.PointerTo[int](6),
TotalVulnerabilities: &models.VulnerabilityScanSummary{
TotalCriticalVulnerabilities: utils.PointerTo[int](9),
TotalHighVulnerabilities: utils.PointerTo[int](12),
TotalLowVulnerabilities: utils.PointerTo[int](424),
TotalMediumVulnerabilities: utils.PointerTo[int](1551),
TotalNegligibleVulnerabilities: utils.PointerTo[int](132),
TotalCriticalVulnerabilities: utils.PointerTo[int](2),
TotalHighVulnerabilities: utils.PointerTo[int](2),
TotalLowVulnerabilities: utils.PointerTo[int](2),
TotalMediumVulnerabilities: utils.PointerTo[int](0),
TotalNegligibleVulnerabilities: utils.PointerTo[int](0),
},
}

scan1ConfigSnapshot := &models.ScanConfigData{
Name: utils.PointerTo[string]("Scan Config 1"),
ScanFamiliesConfig: scanConfigs[0].ScanFamiliesConfig,
Scheduled: scanConfigs[0].Scheduled,
Scope: scanConfigs[0].Scope,
MaxParallelScanners: scanConfigs[0].MaxParallelScanners,
Name: utils.PointerTo[string]("Scan Config 1"),
ScanFamiliesConfig: scanConfigs[0].ScanFamiliesConfig,
Scheduled: scanConfigs[0].Scheduled,
Scope: scanConfigs[0].Scope,
}

// Create scan 2: Running
scan2Start := time.Now().Add(-5 * time.Minute)
scan2Targets := []string{*targets[2].Id}

scan2Summary := &models.ScanSummary{
JobsCompleted: utils.PointerTo[int](77),
JobsLeftToRun: utils.PointerTo[int](98),
TotalExploits: utils.PointerTo[int](6),
TotalMalware: utils.PointerTo[int](0),
TotalMisconfigurations: utils.PointerTo[int](75),
TotalPackages: utils.PointerTo[int](9778),
TotalRootkits: utils.PointerTo[int](5),
TotalSecrets: utils.PointerTo[int](557),
JobsCompleted: utils.PointerTo[int](1),
JobsLeftToRun: utils.PointerTo[int](1),
TotalExploits: utils.PointerTo[int](2),
TotalMalware: utils.PointerTo[int](3),
TotalMisconfigurations: utils.PointerTo[int](3),
TotalPackages: utils.PointerTo[int](0),
TotalRootkits: utils.PointerTo[int](3),
TotalSecrets: utils.PointerTo[int](0),
TotalVulnerabilities: &models.VulnerabilityScanSummary{
TotalCriticalVulnerabilities: utils.PointerTo[int](11),
TotalHighVulnerabilities: utils.PointerTo[int](52),
TotalLowVulnerabilities: utils.PointerTo[int](241),
TotalMediumVulnerabilities: utils.PointerTo[int](8543),
TotalNegligibleVulnerabilities: utils.PointerTo[int](73),
TotalCriticalVulnerabilities: utils.PointerTo[int](0),
TotalHighVulnerabilities: utils.PointerTo[int](0),
TotalLowVulnerabilities: utils.PointerTo[int](0),
TotalMediumVulnerabilities: utils.PointerTo[int](0),
TotalNegligibleVulnerabilities: utils.PointerTo[int](0),
},
}

scan2ConfigSnapshot := &models.ScanConfigData{
Name: utils.PointerTo[string]("Scan Config 2"),
ScanFamiliesConfig: scanConfigs[1].ScanFamiliesConfig,
Scheduled: scanConfigs[1].Scheduled,
Scope: scanConfigs[1].Scope,
MaxParallelScanners: scanConfigs[1].MaxParallelScanners,
Name: utils.PointerTo[string]("Scan Config 2"),
ScanFamiliesConfig: scanConfigs[1].ScanFamiliesConfig,
Scheduled: scanConfigs[1].Scheduled,
Scope: scanConfigs[1].Scope,
}

return []models.Scan{
Expand Down

0 comments on commit f03ca42

Please sign in to comment.