bug: Attack tactics and Platform enums were not parsed properly

DataDog · Jan 19, 2022 · a838ec1 · a838ec1
1 parent 1be2715
commit a838ec1
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/pkg/stratus/mitreattack/tactics.go b/pkg/stratus/mitreattack/tactics.go
@@ -1,6 +1,9 @@
 package mitreattack
 
-import "errors"
+import (
+	"errors"
+	"strings"
+)
 
 type Tactic int
 
@@ -33,8 +36,9 @@ const (
 )
 
 func AttackTacticFromString(name string) (Tactic, error) {
+	lowerName := strings.ToLower(name)
 	for i := range tactics {
-		if tactics[i] == name {
+		if strings.ToLower(tactics[i]) == lowerName {
 			return Tactic(i), nil
 		}
 	}

diff --git a/pkg/stratus/platform.go b/pkg/stratus/platform.go
@@ -1,6 +1,9 @@
 package stratus
 
-import "errors"
+import (
+	"errors"
+	"strings"
+)
 
 type Platform string
 
@@ -9,8 +12,8 @@ const (
 )
 
 func PlatformFromString(name string) (Platform, error) {
-	switch name {
-	case AWS:
+	switch strings.ToLower(name) {
+	case strings.ToLower(AWS):
 		return AWS, nil
 	default:
 		return "", errors.New("unknown platform: " + name)