fix(query): handle list-form access_logs in ELBv2 LB Access Log Disabled

assets/queries/terraform/aws/elb_v2_lb_access_log_disabled/query.rego

@@ -25,43 +25,71 @@ CxPolicy[result] {
     load_balancer := get_load_balancer(input.document[i].resource)
     resource := input.document[i].resource[load_balancer][name]
 
-    not common_lib.valid_key(resource.access_logs, "enabled")
+    log_info := get_access_logs(resource)[_]
+    not common_lib.valid_key(log_info.block, "enabled")
 
     result := {
         "documentId": input.document[i].id,
         "resourceType": load_balancer,
 		"resourceName": tf_lib.get_specific_resource_name(resource, load_balancer, name),
-        "searchKey": sprintf("%s[%s].access_logs", [load_balancer,name]),
+        "searchKey": get_search_key(load_balancer, name, log_info.index, "access_logs"),
         "issueType": "MissingAttribute",
         "keyExpectedValue": "'access_logs.enabled' should be defined and set to true",
         "keyActualValue": "'access_logs.enabled' is undefined",
-        "searchLine": common_lib.build_search_line(["resource", load_balancer, name, "access_logs"], []),
+        "searchLine": get_search_line(load_balancer, name, log_info.index, ["access_logs"]),
     }
 }
 
 CxPolicy[result] {
     load_balancer := get_load_balancer(input.document[i].resource)
     resource := input.document[i].resource[load_balancer][name]
 
-    resource.access_logs.enabled != true
+    log_info := get_access_logs(resource)[_]
+    log_info.block.enabled != true
 
     result := {
         "documentId": input.document[i].id,
         "resourceType": load_balancer,
 		"resourceName": tf_lib.get_specific_resource_name(resource, load_balancer, name),
-        "searchKey": sprintf("%s[%s].access_logs.enabled", [load_balancer,name]),
+        "searchKey": get_search_key(load_balancer, name, log_info.index, "access_logs.enabled"),
         "issueType": "IncorrectValue",
         "keyExpectedValue": "'access_logs.enabled' should be defined and set to true",
         "keyActualValue": "'access_logs.enabled' is not set to true",
-        "searchLine": common_lib.build_search_line(["resource", load_balancer, name, "access_logs", "enabled"], []),
+        "searchLine": get_search_line(load_balancer, name, log_info.index, ["access_logs", "enabled"]),
     }
 }
 
+# access_logs can be parsed as a single block (object) or, when written as a
+# list (access_logs = [{ ... }]) or as multiple blocks, as an array. Normalize
+# both shapes to a list of {block, index} so the checks above behave the same.
+get_access_logs(resource) = logs {
+    is_array(resource.access_logs)
+    logs := [{"block": block, "index": idx} | block := resource.access_logs[idx]]
+} else = logs {
+    logs := [{"block": resource.access_logs, "index": null}]
+}
+
+get_search_key(lb, name, index, suffix) = searchKey {
+    index != null
+    searchKey := sprintf("%s[%s].access_logs[%d].%s", [lb, name, index, trim_prefix(suffix, "access_logs.")])
+} else = searchKey {
+    searchKey := sprintf("%s[%s].%s", [lb, name, suffix])
+}
+
+get_search_line(lb, name, index, path_elements) = searchLine {
+    index != null
+    full_path := array.concat(["resource", lb, name], path_elements)
+    path_with_index := array.concat(array.slice(full_path, 0, 4), array.concat([index], array.slice(full_path, 4, count(full_path))))
+    searchLine := common_lib.build_search_line(path_with_index, [])
+} else = searchLine {
+    full_path := array.concat(["resource", lb, name], path_elements)
+    searchLine := common_lib.build_search_line(full_path, [])
+}
+
 get_load_balancer(resource) = lb {
     common_lib.valid_key(resource,"aws_lb")
     lb = "aws_lb"
 } else = lb {
     common_lib.valid_key(resource,"aws_alb")
     lb = "aws_alb"
 }
-

assets/queries/terraform/aws/elb_v2_lb_access_log_disabled/test/negative3.tf

@@ -0,0 +1,18 @@
+# access_logs written as a list of objects (= [{ ... }]) instead of a block
+resource "aws_lb" "test" {
+  name               = "test-lb-tf"
+  internal           = false
+  load_balancer_type = "network"
+
+  enable_deletion_protection = true
+
+  access_logs = [{
+    bucket  = aws_s3_bucket.lb_logs.id
+    prefix  = "test-lb"
+    enabled = true
+  }]
+
+  tags = {
+    Environment = "production"
+  }
+}

assets/queries/terraform/aws/elb_v2_lb_access_log_disabled/test/positive7.tf

@@ -0,0 +1,18 @@
+# access_logs as a list with logging explicitly disabled - still a finding
+resource "aws_lb" "test" {
+  name               = "test-lb-tf"
+  internal           = false
+  load_balancer_type = "network"
+
+  enable_deletion_protection = true
+
+  access_logs = [{
+    bucket  = aws_s3_bucket.lb_logs.id
+    prefix  = "test-lb"
+    enabled = false
+  }]
+
+  tags = {
+    Environment = "production"
+  }
+}

assets/queries/terraform/aws/elb_v2_lb_access_log_disabled/test/positive_expected_result.json

@@ -34,5 +34,11 @@
     "severity": "MEDIUM",
     "line": 2,
     "fileName": "positive6.tf"
+  },
+  {
+    "queryName": "ELBv2 LB Access Log Disabled",
+    "severity": "MEDIUM",
+    "line": 12,
+    "fileName": "positive7.tf"
   }
 ]