Add handler for no rules threat

teler-sh · Dec 16, 2021 · 46209b5 · 46209b5
1 parent 9184a44
commit 46209b5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/internal/runner/validator.go b/internal/runner/validator.go
@@ -89,6 +89,11 @@ func customs(options *common.Options) {
 		cat[custom[i].Name] = true
 
 		rules := custom[i].Rules
+		if len(rules) < 1 {
+			err = strings.Replace(errors.ErrNoThreatRules, ":category", custom[i].Name, -1)
+			errors.Exit(err)
+		}
+
 		for j := 0; j < len(rules); j++ {
 			matchers.IsBlank(rules[j].Element, "Custom threat rules element")
 			elm := fmt.Sprint("$", rules[j].Element)

diff --git a/pkg/errors/constants.go b/pkg/errors/constants.go
@@ -15,6 +15,7 @@ const (
 	ErrNoInputConfig  = "No config file specified"
 	ErrNoInputLog     = "No input logs provided"
 	ErrNoPassZinc     = "No password provided for Zinc log server; " + ErrCheckConfig
+	ErrNoThreatRules  = "No rules for ':category' threat category; " + ErrCheckConfig
 	ErrNoUserZinc     = "No username provided for Zinc log server; " + ErrCheckConfig
 	ErrParseConfig    = "Can't parse config file: "
 )