fix(config): rename include-successes with include-non-failures (#1107)

This commit is contained in:
Teppei Fukuda
2021-07-11 16:55:13 +03:00
committed by GitHub
parent e6f7e556e8
commit d9883e4442
8 changed files with 73 additions and 70 deletions

View File

@@ -252,11 +252,11 @@ var (
EnvVars: []string{"TRIVY_POLICY_NAMESPACES"}, EnvVars: []string{"TRIVY_POLICY_NAMESPACES"},
} }
includeSuccesses = cli.BoolFlag{ includeNonFailures = cli.BoolFlag{
Name: "include-successes", Name: "include-non-failures",
Usage: "include successes of misconfigurations", Usage: "include successes and exceptions",
Value: false, Value: false,
EnvVars: []string{"TRIVY_INCLUDE_SUCCESSES"}, EnvVars: []string{"TRIVY_INCLUDE_NON_FAILURES"},
} }
traceFlag = cli.BoolFlag{ traceFlag = cli.BoolFlag{
@@ -297,8 +297,6 @@ var (
&cacheBackendFlag, &cacheBackendFlag,
stringSliceFlag(skipFiles), stringSliceFlag(skipFiles),
stringSliceFlag(skipDirs), stringSliceFlag(skipDirs),
stringSliceFlag(configPolicy),
stringSliceFlag(policyNamespaces),
} }
// deprecated options // deprecated options
@@ -477,7 +475,7 @@ func NewFilesystemCommand() *cli.Command {
stringSliceFlag(skipFiles), stringSliceFlag(skipFiles),
stringSliceFlag(skipDirs), stringSliceFlag(skipDirs),
stringSliceFlag(configPolicy), stringSliceFlag(configPolicy),
&policyNamespaces, stringSliceFlag(policyNamespaces),
}, },
} }
} }
@@ -613,7 +611,7 @@ func NewConfigCommand() *cli.Command {
stringSliceFlag(configPolicyAlias), stringSliceFlag(configPolicyAlias),
stringSliceFlag(filePatterns), stringSliceFlag(filePatterns),
stringSliceFlag(policyNamespaces), stringSliceFlag(policyNamespaces),
&includeSuccesses, &includeNonFailures,
&traceFlag, &traceFlag,
}, },
} }

View File

@@ -80,7 +80,7 @@ func runWithTimeout(ctx context.Context, opt Option, initializeScanner Initializ
Severities: opt.Severities, Severities: opt.Severities,
OutputTemplate: opt.Template, OutputTemplate: opt.Template,
Light: opt.Light, Light: opt.Light,
IncludeSuccesses: opt.IncludeSuccesses, IncludeNonFailures: opt.IncludeNonFailures,
Trace: opt.Trace, Trace: opt.Trace,
}); err != nil { }); err != nil {
return xerrors.Errorf("unable to write results: %w", err) return xerrors.Errorf("unable to write results: %w", err)
@@ -193,7 +193,7 @@ func filter(ctx context.Context, opt Option, report pkgReport.Report) (pkgReport
for i := range results { for i := range results {
resultClient.FillVulnerabilityInfo(results[i].Vulnerabilities, results[i].Type) resultClient.FillVulnerabilityInfo(results[i].Vulnerabilities, results[i].Type)
vulns, misconfSummary, misconfs, err := resultClient.Filter(ctx, results[i].Vulnerabilities, results[i].Misconfigurations, vulns, misconfSummary, misconfs, err := resultClient.Filter(ctx, results[i].Vulnerabilities, results[i].Misconfigurations,
opt.Severities, opt.IgnoreUnfixed, opt.IncludeSuccesses, opt.IgnoreFile, opt.IgnorePolicy) opt.Severities, opt.IgnoreUnfixed, opt.IncludeNonFailures, opt.IgnoreFile, opt.IgnorePolicy)
if err != nil { if err != nil {
return pkgReport.Report{}, xerrors.Errorf("unable to filter vulnerabilities: %w", err) return pkgReport.Report{}, xerrors.Errorf("unable to filter vulnerabilities: %w", err)
} }

View File

@@ -71,7 +71,7 @@ func runWithTimeout(ctx context.Context, opt Option) error {
results := report.Results results := report.Results
for i := range results { for i := range results {
vulns, misconfSummary, misconfs, err := resultClient.Filter(ctx, results[i].Vulnerabilities, results[i].Misconfigurations, vulns, misconfSummary, misconfs, err := resultClient.Filter(ctx, results[i].Vulnerabilities, results[i].Misconfigurations,
opt.Severities, opt.IgnoreUnfixed, opt.IncludeSuccesses, opt.IgnoreFile, opt.IgnorePolicy) opt.Severities, opt.IgnoreUnfixed, opt.IncludeNonFailures, opt.IgnoreFile, opt.IgnorePolicy)
if err != nil { if err != nil {
return xerrors.Errorf("filter error: %w", err) return xerrors.Errorf("filter error: %w", err)
} }
@@ -86,7 +86,7 @@ func runWithTimeout(ctx context.Context, opt Option) error {
Severities: opt.Severities, Severities: opt.Severities,
OutputTemplate: opt.Template, OutputTemplate: opt.Template,
Light: false, Light: false,
IncludeSuccesses: opt.IncludeSuccesses, IncludeNonFailures: opt.IncludeNonFailures,
Trace: opt.Trace, Trace: opt.Trace,
}); err != nil { }); err != nil {
return xerrors.Errorf("unable to write results: %w", err) return xerrors.Errorf("unable to write results: %w", err)

View File

@@ -7,7 +7,7 @@ import (
// ConfigOption holds the options for config scanning // ConfigOption holds the options for config scanning
type ConfigOption struct { type ConfigOption struct {
FilePatterns []string FilePatterns []string
IncludeSuccesses bool IncludeNonFailures bool
SkipPolicyUpdate bool SkipPolicyUpdate bool
Trace bool Trace bool
@@ -20,7 +20,7 @@ type ConfigOption struct {
// NewConfigOption is the factory method to return config scanning options // NewConfigOption is the factory method to return config scanning options
func NewConfigOption(c *cli.Context) ConfigOption { func NewConfigOption(c *cli.Context) ConfigOption {
return ConfigOption{ return ConfigOption{
IncludeSuccesses: c.Bool("include-successes"), IncludeNonFailures: c.Bool("include-non-failures"),
SkipPolicyUpdate: c.Bool("skip-policy-update"), SkipPolicyUpdate: c.Bool("skip-policy-update"),
Trace: c.Bool("trace"), Trace: c.Bool("trace"),
FilePatterns: c.StringSlice("file-patterns"), FilePatterns: c.StringSlice("file-patterns"),

View File

@@ -24,7 +24,7 @@ type TableWriter struct {
Light bool Light bool
// For misconfigurations // For misconfigurations
IncludeSuccesses bool IncludeNonFailures bool
Trace bool Trace bool
} }
@@ -125,7 +125,7 @@ func (tw TableWriter) writeMisconfigurations(table *tablewriter.Table, misconfs
tablewriter.ALIGN_CENTER, tablewriter.ALIGN_CENTER, tablewriter.ALIGN_LEFT} tablewriter.ALIGN_CENTER, tablewriter.ALIGN_CENTER, tablewriter.ALIGN_LEFT}
header := []string{"Type", "Misconf ID", "Check", "Severity", "Status", "Message"} header := []string{"Type", "Misconf ID", "Check", "Severity", "Status", "Message"}
if !tw.IncludeSuccesses { if !tw.IncludeNonFailures {
// Remove status // Remove status
statusPos := 4 statusPos := 4
alignment = append(alignment[:statusPos], alignment[statusPos+1:]...) alignment = append(alignment[:statusPos], alignment[statusPos+1:]...)
@@ -185,20 +185,24 @@ func (tw TableWriter) setMisconfRows(table *tablewriter.Table, misconfs []types.
} }
} }
var row []string severity := misconf.Severity
status := string(misconf.Status)
if tw.Output == os.Stdout { if tw.Output == os.Stdout {
if misconf.Status == types.StatusPassed { switch misconf.Status {
row = []string{misconf.Type, misconf.ID, misconf.Title, color.New(color.FgGreen).Sprint(misconf.Severity), case types.StatusPassed:
color.New(color.FgGreen).Sprint(misconf.Status), misconf.Message} severity = color.New(color.FgGreen).Sprint(misconf.Severity)
} else { status = color.New(color.FgGreen).Sprint(misconf.Status)
row = []string{misconf.Type, misconf.ID, misconf.Title, dbTypes.ColorizeSeverity(misconf.Severity), case types.StatusException:
color.New(color.FgRed).Sprint(misconf.Status), misconf.Message} severity = color.New(color.FgMagenta).Sprint(misconf.Severity)
status = color.New(color.FgMagenta).Sprint(misconf.Status)
case types.StatusFailure:
severity = dbTypes.ColorizeSeverity(severity)
status = color.New(color.FgRed).Sprint(misconf.Status)
} }
} else {
row = []string{misconf.Type, misconf.ID, misconf.Title, misconf.Severity, string(misconf.Status), misconf.Message}
} }
if !tw.IncludeSuccesses { row := []string{misconf.Type, misconf.ID, misconf.Title, severity, status, misconf.Message}
if !tw.IncludeNonFailures {
// Remove status // Remove status
row = append(row[:4], row[5:]...) row = append(row[:4], row[5:]...)
} }

View File

@@ -17,7 +17,7 @@ func TestReportWriter_Table(t *testing.T) {
results report.Results results report.Results
expectedOutput string expectedOutput string
light bool light bool
includeSuccesses bool includeNonFailures bool
}{ }{
{ {
name: "happy path full", name: "happy path full",
@@ -170,7 +170,7 @@ func TestReportWriter_Table(t *testing.T) {
}, },
{ {
name: "happy path misconfigurations with successes", name: "happy path misconfigurations with successes",
includeSuccesses: true, includeNonFailures: true,
results: report.Results{ results: report.Results{
{ {
Target: "test", Target: "test",
@@ -219,7 +219,7 @@ func TestReportWriter_Table(t *testing.T) {
Format: "table", Format: "table",
Output: &tableWritten, Output: &tableWritten,
Light: tc.light, Light: tc.light,
IncludeSuccesses: tc.includeSuccesses, IncludeNonFailures: tc.includeNonFailures,
}) })
assert.NoError(t, err) assert.NoError(t, err)
assert.Equal(t, tc.expectedOutput, tableWritten.String(), tc.name) assert.Equal(t, tc.expectedOutput, tableWritten.String(), tc.name)

View File

@@ -88,7 +88,7 @@ type Option struct {
Light bool Light bool
// For misconfigurations // For misconfigurations
IncludeSuccesses bool IncludeNonFailures bool
Trace bool Trace bool
} }
@@ -101,7 +101,7 @@ func Write(report Report, option Option) error {
Output: option.Output, Output: option.Output,
Severities: option.Severities, Severities: option.Severities,
Light: option.Light, Light: option.Light,
IncludeSuccesses: option.IncludeSuccesses, IncludeNonFailures: option.IncludeNonFailures,
Trace: option.Trace, Trace: option.Trace,
} }
case "json": case "json":

View File

@@ -140,12 +140,12 @@ func (c Client) getPrimaryURL(vulnID string, refs []string, source string) strin
// Filter filter out the vulnerabilities // Filter filter out the vulnerabilities
func (c Client) Filter(ctx context.Context, vulns []types.DetectedVulnerability, misconfs []types.DetectedMisconfiguration, func (c Client) Filter(ctx context.Context, vulns []types.DetectedVulnerability, misconfs []types.DetectedMisconfiguration,
severities []dbTypes.Severity, ignoreUnfixed, includeSuccesses bool, ignoreFile, policyFile string) ( severities []dbTypes.Severity, ignoreUnfixed, includeNonFailures bool, ignoreFile, policyFile string) (
[]types.DetectedVulnerability, *report.MisconfSummary, []types.DetectedMisconfiguration, error) { []types.DetectedVulnerability, *report.MisconfSummary, []types.DetectedMisconfiguration, error) {
ignoredIDs := getIgnoredIDs(ignoreFile) ignoredIDs := getIgnoredIDs(ignoreFile)
filteredVulns := filterVulnerabilities(vulns, severities, ignoreUnfixed, ignoredIDs) filteredVulns := filterVulnerabilities(vulns, severities, ignoreUnfixed, ignoredIDs)
misconfSummary, filteredMisconfs := filterMisconfigurations(misconfs, severities, includeSuccesses, ignoredIDs) misconfSummary, filteredMisconfs := filterMisconfigurations(misconfs, severities, includeNonFailures, ignoredIDs)
if policyFile != "" { if policyFile != "" {
var err error var err error
@@ -192,7 +192,7 @@ func filterVulnerabilities(vulns []types.DetectedVulnerability, severities []dbT
} }
func filterMisconfigurations(misconfs []types.DetectedMisconfiguration, severities []dbTypes.Severity, func filterMisconfigurations(misconfs []types.DetectedMisconfiguration, severities []dbTypes.Severity,
includeSuccesses bool, ignoredIDs []string) (*report.MisconfSummary, []types.DetectedMisconfiguration) { includeNonFailures bool, ignoredIDs []string) (*report.MisconfSummary, []types.DetectedMisconfiguration) {
var filtered []types.DetectedMisconfiguration var filtered []types.DetectedMisconfiguration
summary := new(report.MisconfSummary) summary := new(report.MisconfSummary)
@@ -204,9 +204,10 @@ func filterMisconfigurations(misconfs []types.DetectedMisconfiguration, severiti
continue continue
} }
// Count successes, failures, and exceptions
summarize(misconf.Status, summary) summarize(misconf.Status, summary)
if misconf.Status != types.StatusFailure && !includeSuccesses { if misconf.Status != types.StatusFailure && !includeNonFailures {
continue continue
} }
filtered = append(filtered, misconf) filtered = append(filtered, misconf)