From 5effc34b2eb4987ec6b0654e58f9ed3b4fe7e5d3 Mon Sep 17 00:00:00 2001
From: TAKANO Mitsuhiro <takano32@gmail.com>
Date: Tue, 27 Feb 2024 21:45:45 +0900
Subject: [PATCH] WAF: Access Control with IPv4 in CloudFront

---
 cloudfront.tf |  3 ++-
 waf.tf        | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+), 1 deletion(-)
 create mode 100644 waf.tf

diff --git a/cloudfront.tf b/cloudfront.tf
index 86db011..beb60bb 100644
--- a/cloudfront.tf
+++ b/cloudfront.tf
@@ -13,6 +13,7 @@ module "acm" {
 
 resource "aws_cloudfront_distribution" "this" {
   http_version = "http2and3"
+  web_acl_id = aws_wafv2_web_acl.web_acl.arn
   origin {
     domain_name = module.alb.lb_dns_name
     origin_id   = "alb"
@@ -26,7 +27,7 @@ resource "aws_cloudfront_distribution" "this" {
   }
 
   enabled         = true
-  is_ipv6_enabled = true
+  is_ipv6_enabled = false
   comment         = var.site_domain
 
   aliases = [var.site_domain]
diff --git a/waf.tf b/waf.tf
new file mode 100644
index 0000000..2a58db1
--- /dev/null
+++ b/waf.tf
@@ -0,0 +1,48 @@
+resource "aws_wafv2_ip_set" "ip_set" {
+  provider           = aws.virginia
+  name               = "allow-ip-set"
+  description        = "allow ip set"
+  scope              = "CLOUDFRONT"
+  ip_address_version = "IPV4"
+  addresses = [
+    "116.0.0.0/6",
+    "116.82.102.140/32",
+    "113.43.73.18/32"
+  ]
+}
+
+resource "aws_wafv2_web_acl" "web_acl" {
+  provider    = aws.virginia
+  name        = "only-from-allow-ip-set"
+  description = "Web ACL that blocks all traffic except for a allow IP set"
+  scope       = "CLOUDFRONT"
+  default_action {
+    block {}
+  }
+
+  rule {
+    name     = "allow-ips-in-ip_set"
+    priority = 1
+    action {
+      allow {}
+    }
+
+    statement {
+      ip_set_reference_statement {
+        arn = aws_wafv2_ip_set.ip_set.arn
+      }
+    }
+
+    visibility_config {
+      sampled_requests_enabled   = false
+      cloudwatch_metrics_enabled = false
+      metric_name                = "wordpress-allow-ips-in-ip_set"
+    }
+  }
+
+  visibility_config {
+    cloudwatch_metrics_enabled = false
+    metric_name                = "wordpress-wafv2-web-acl"
+    sampled_requests_enabled   = false
+  }
+}