feat(cloudformation): add support for the length function

pkg/scanners/cloudformation/parser/fn_length.go

@@ -0,0 +1,24 @@
+package parser
+
+import "github.com/aquasecurity/defsec/pkg/scanners/cloudformation/cftypes"
+
+func ResolveLength(property *Property) (*Property, bool) {
+	if !property.isFunction() {
+		return property, true
+	}
+
+	val := property.AsMap()["Fn::Length"]
+	if val.IsList() {
+		return property.deriveResolved(cftypes.Int, val.Len()), true
+	} else if val.IsMap() {
+		resolved, _ := val.resolveValue()
+
+		if resolved.IsList() {
+			return property.deriveResolved(cftypes.Int, resolved.Len()), true
+		}
+		return resolved, false
+	}
+
+	return property, false
+
+}

pkg/scanners/cloudformation/parser/fn_length_test.go

@@ -0,0 +1,99 @@
+package parser
+
+import (
+	"testing"
+
+	"github.com/aquasecurity/defsec/pkg/scanners/cloudformation/cftypes"
+	"github.com/stretchr/testify/require"
+)
+
+func Test_ResolveLength_WhenPropIsArray(t *testing.T) {
+	prop := &Property{
+		Inner: PropertyInner{
+			Type: cftypes.Map,
+			Value: map[string]*Property{
+				"Fn::Length": {
+					Inner: PropertyInner{
+						Type: cftypes.List,
+						Value: []*Property{
+							{
+								Inner: PropertyInner{
+									Type:  cftypes.Int,
+									Value: 1,
+								},
+							},
+							{
+								Inner: PropertyInner{
+									Type:  cftypes.String,
+									Value: "IntParameter",
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+	resolved, ok := ResolveIntrinsicFunc(prop)
+	require.True(t, ok)
+	require.True(t, resolved.IsInt())
+	require.Equal(t, 2, resolved.AsInt())
+}
+
+func Test_ResolveLength_WhenPropIsIntrinsicFunction(t *testing.T) {
+	fctx := &FileContext{
+		Parameters: map[string]*Parameter{
+			"SomeParameter": {
+				inner: parameterInner{
+					Type:    "string",
+					Default: "a|b|c|d",
+				},
+			},
+		},
+	}
+	prop := &Property{
+		Inner: PropertyInner{
+			Type: cftypes.Map,
+			Value: map[string]*Property{
+				"Fn::Length": {
+					Inner: PropertyInner{
+						Type: cftypes.Map,
+						Value: map[string]*Property{
+							"Fn::Split": {
+								Inner: PropertyInner{
+									Type: cftypes.List,
+									Value: []*Property{
+										{
+											Inner: PropertyInner{
+												Type:  cftypes.String,
+												Value: "|",
+											},
+										},
+										{
+											ctx: fctx,
+											Inner: PropertyInner{
+												Type: cftypes.Map,
+												Value: map[string]*Property{
+													"Ref": {
+														Inner: PropertyInner{
+															Type:  cftypes.String,
+															Value: "SomeParameter",
+														},
+													},
+												},
+											},
+										},
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+	resolved, ok := ResolveIntrinsicFunc(prop)
+	require.True(t, ok)
+	require.True(t, resolved.IsInt())
+	require.Equal(t, 4, resolved.AsInt())
+}

pkg/scanners/cloudformation/parser/intrinsics.go

@@ -27,6 +27,7 @@ func init() {
 		"Fn::And":         ResolveAnd,
 		"Fn::Or":          ResolveOr,
 		"Fn::Not":         ResolveNot,
+		"Fn::Length":      ResolveLength,
 	}
 }
 

pkg/scanners/cloudformation/parser/property_helpers.go

@@ -186,6 +186,10 @@ func (p *Property) AsList() []*Property {
 	return nil
 }
 
+func (p *Property) Len() int {
+	return len(p.AsList())
+}
+
 func (p *Property) EqualTo(checkValue interface{}, equalityOptions ...EqualityOptions) bool {
 	var ignoreCase bool
 	for _, option := range equalityOptions {