Extended the wafv2_web_acl module with custom_response_bodies argument (

ansible-collections#721) Extended the wafv2_web_acl module with custom_response_bodies argument SUMMARY Extended the wafv2_web_acl module to also take the custom_response_bodies argument, improved docs and extended tests ISSUE TYPE Feature Pull Request COMPONENT NAME wafv2_web_acl ADDITIONAL INFORMATION Also touched docs of aws_waf_web_acl to make it easier to find the WAF v2 modules as I had trouble finding that at first. Reviewed-by: Markus Bergholz <[email protected]> Reviewed-by: Stefan Horning <None> Reviewed-by: Mark Chappell <None> Reviewed-by: Alina Buzachis <None>
abikouo · Feb 10, 2022 · 904a08c · 904a08c
1 parent bc0ca8a
commit 904a08c
Show file tree

Hide file tree

Showing 2 changed files with 119 additions and 31 deletions.
diff --git a/aws_waf_web_acl.py b/aws_waf_web_acl.py
@@ -8,11 +8,11 @@
 
 DOCUMENTATION = r'''
 module: aws_waf_web_acl
-short_description: Create and delete WAF Web ACLs.
+short_description: Create and delete WAF Web ACLs
 version_added: 1.0.0
 description:
-  - Read the AWS documentation for WAF
-    U(https://aws.amazon.com/documentation/waf/).
+  - Module for WAF classic, for WAF v2 use the I(wafv2_*) modules.
+  - Read the AWS documentation for WAF U(https://docs.aws.amazon.com/waf/latest/developerguide/classic-waf-chapter.html).
 
 author:
   - Mike Mochan (@mmochan)

diff --git a/wafv2_web_acl.py b/wafv2_web_acl.py
@@ -11,9 +11,10 @@
 version_added: 1.5.0
 author:
   - "Markus Bergholz (@markuman)"
-short_description: wafv2_web_acl
+short_description: Create and delete WAF Web ACLs
 description:
-  - Create, modify or delete a wafv2 web acl.
+  - Create, modify or delete AWS WAF v2 web ACLs (not for classic WAF).
+  - See docs at U(https://docs.aws.amazon.com/waf/latest/developerguide/waf-chapter.html)
 options:
     state:
       description:
@@ -28,9 +29,9 @@
       type: str
     scope:
       description:
-        - Scope of wafv2 web acl.
+        - Geographical scope of the web acl.
       required: true
-      choices: ["CLOUDFRONT","REGIONAL"]
+      choices: ["CLOUDFRONT", "REGIONAL"]
       type: str
     description:
       description:
@@ -39,7 +40,7 @@
     default_action:
       description:
         - Default action of the wafv2 web acl.
-      choices: ["Block","Allow"]
+      choices: ["Block", "Allow"]
       type: str
     sampled_requests:
       description:
@@ -87,6 +88,14 @@
           description:
             - Rule configuration.
           type: dict
+    custom_response_bodies:
+      description:
+        - A map of custom response keys and content bodies. Define response bodies here and reference them in the rules by providing
+        - the key of the body dictionary element.
+        - Each element must have a unique dict key and in the dict two keys for I(content_type) and I(content).
+        - Requires botocore >= 1.21.0
+      type: dict
+      version_added: 3.1.0
     purge_rules:
       description:
         - When set to C(no), keep the existing load balancer rules in place. Will modify and add, but will not delete.
@@ -100,16 +109,15 @@
 '''
 
 EXAMPLES = '''
-- name: create web acl
+- name: Create test web acl
   community.aws.wafv2_web_acl:
     name: test05
-    state: present
     description: hallo eins
     scope: REGIONAL
     default_action: Allow
     sampled_requests: no
     cloudwatch_metrics: yes
-    metric_name: blub
+    metric_name: test05-acl-metric
     rules:
       - name: zwei
         priority: 0
@@ -191,10 +199,56 @@
                       text_transformations:
                         - type: LOWERCASE
                           priority: 0
+    purge_rules: yes
     tags:
       A: B
       C: D
-  register: out
+    state: present
+
+- name: Create IP filtering web ACL
+  community.aws.wafv2_web_acl:
+    name: ip-filtering-traffic
+    description: ACL that filters web traffic based on rate limits and whitelists some IPs
+    scope: REGIONAL
+    default_action: Allow
+    sampled_requests: yes
+    cloudwatch_metrics: yes
+    metric_name: ip-filtering-traffic
+    rules:
+      - name: whitelist-own-IPs
+        priority: 0
+        action:
+          allow: {}
+        statement:
+          ip_set_reference_statement:
+            arn: 'arn:aws:wafv2:us-east-1:520789123123:regional/ipset/own-public-ips/1c4bdfc4-0f77-3b23-5222-123123123'
+        visibility_config:
+          sampled_requests_enabled: yes
+          cloud_watch_metrics_enabled: yes
+          metric_name: waf-acl-rule-whitelist-own-IPs
+      - name: rate-limit-per-IP
+        priority: 1
+        action:
+          block:
+            custom_response:
+              response_code: 429
+              custom_response_body_key: too_many_requests
+        statement:
+          rate_based_statement:
+            limit: 5000
+            aggregate_key_type: IP
+        visibility_config:
+          sampled_requests_enabled: yes
+          cloud_watch_metrics_enabled: yes
+          metric_name: waf-acl-rule-rate-limit-per-IP
+        purge_rules: yes
+    custom_response_bodies:
+      too_many_requests:
+        content_type: APPLICATION_JSON
+        content: '{ message: "Your request has been blocked due to too many HTTP requests coming from your IP" }'
+    region: us-east-1
+    state: present
+
 '''
 
 RETURN = """
@@ -218,6 +272,12 @@
   sample: test02
   returned: Always, as long as the web acl exists
   type: str
+default_action:
+  description: Default action of ACL
+  returned: Always, as long as the web acl exists
+  sample:
+    allow: {}
+  type: dict
 rules:
   description: Current rules of the web acl
   returned: Always, as long as the web acl exists
@@ -235,6 +295,14 @@
         cloud_watch_metrics_enabled: true
         metric_name: admin_protect
         sampled_requests_enabled: true
+custom_response_bodies:
+  description: Custom response body configurations to be used in rules
+  type: dict
+  sample:
+    too_many_requests:
+      content_type: APPLICATION_JSON
+      content: '{ message: "Your request has been blocked due to too many HTTP requests coming from your IP" }'
+  returned: Always, as long as the web acl exists
 visibility_config:
   description: Visibility config of the web acl
   returned: Always, as long as the web acl exists
@@ -267,22 +335,27 @@ def __init__(self, wafv2, name, scope, fail_json_aws):
         self.fail_json_aws = fail_json_aws
         self.existing_acl, self.id, self.locktoken = self.get_web_acl()
 
-    def update(self, default_action, description, rules, sampled_requests, cloudwatch_metrics, metric_name):
+    def update(self, default_action, description, rules, sampled_requests, cloudwatch_metrics, metric_name, custom_response_bodies):
+        req_obj = {
+            'Name': self.name,
+            'Scope': self.scope,
+            'Id': self.id,
+            'DefaultAction': default_action,
+            'Description': description,
+            'Rules': rules,
+            'VisibilityConfig': {
+                'SampledRequestsEnabled': sampled_requests,
+                'CloudWatchMetricsEnabled': cloudwatch_metrics,
+                'MetricName': metric_name
+            },
+            'LockToken': self.locktoken
+        }
+
+        if custom_response_bodies:
+            req_obj['CustomResponseBodies'] = custom_response_bodies
+
         try:
-            response = self.wafv2.update_web_acl(
-                Name=self.name,
-                Scope=self.scope,
-                Id=self.id,
-                DefaultAction=default_action,
-                Description=description,
-                Rules=rules,
-                VisibilityConfig={
-                    'SampledRequestsEnabled': sampled_requests,
-                    'CloudWatchMetricsEnabled': cloudwatch_metrics,
-                    'MetricName': metric_name
-                },
-                LockToken=self.locktoken
-            )
+            response = self.wafv2.update_web_acl(**req_obj)
         except (BotoCoreError, ClientError) as e:
             self.fail_json_aws(e, msg="Failed to update wafv2 web acl.")
         return response
@@ -331,7 +404,7 @@ def get_web_acl(self):
     def list(self):
         return wafv2_list_web_acls(self.wafv2, self.scope, self.fail_json_aws)
 
-    def create(self, default_action, rules, sampled_requests, cloudwatch_metrics, metric_name, tags, description):
+    def create(self, default_action, rules, sampled_requests, cloudwatch_metrics, metric_name, tags, description, custom_response_bodies):
         req_obj = {
             'Name': self.name,
             'Scope': self.scope,
@@ -343,6 +416,9 @@ def create(self, default_action, rules, sampled_requests, cloudwatch_metrics, me
                 'MetricName': metric_name
             }
         }
+
+        if custom_response_bodies:
+            req_obj['CustomResponseBodies'] = custom_response_bodies
         if description:
             req_obj['Description'] = description
         if tags:
@@ -370,6 +446,7 @@ def main():
         cloudwatch_metrics=dict(type='bool', default=True),
         metric_name=dict(type='str'),
         tags=dict(type='dict'),
+        custom_response_bodies=dict(type='dict'),
         purge_rules=dict(default=True, type='bool')
     )
 
@@ -392,6 +469,14 @@ def main():
     purge_rules = module.params.get("purge_rules")
     check_mode = module.check_mode
 
+    custom_response_bodies = module.params.get("custom_response_bodies")
+    if custom_response_bodies:
+        module.require_botocore_at_least('1.21.0', reason='to set custom response bodies')
+        custom_response_bodies = {}
+
+        for custom_name, body in module.params.get("custom_response_bodies").items():
+            custom_response_bodies[custom_name] = snake_dict_to_camel_dict(body, capitalize_first=True)
+
     if default_action == 'Block':
         default_action = {'Block': {}}
     elif default_action == 'Allow':
@@ -422,7 +507,8 @@ def main():
                     rules,
                     sampled_requests,
                     cloudwatch_metrics,
-                    metric_name
+                    metric_name,
+                    custom_response_bodies
                 )
 
             else:
@@ -438,7 +524,8 @@ def main():
                     cloudwatch_metrics,
                     metric_name,
                     tags,
-                    description
+                    description,
+                    custom_response_bodies
                 )
 
     elif state == 'absent':
@@ -453,7 +540,8 @@ def main():
                             rules,
                             sampled_requests,
                             cloudwatch_metrics,
-                            metric_name
+                            metric_name,
+                            custom_response_bodies
                         )
             else:
                 change = True