From 127b77c802f35b69f3f0e96bf07ed9b7dfda51b5 Mon Sep 17 00:00:00 2001
From: Georg <georg@lysergic.dev>
Date: Mon, 13 Sep 2021 11:25:46 +0200
Subject: [PATCH] CAA Patcher

Signed-off-by: Georg <georg@lysergic.dev>
---
 scripts/python/powerdns_caa_patcher.py | 83 ++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)
 create mode 100755 scripts/python/powerdns_caa_patcher.py

diff --git a/scripts/python/powerdns_caa_patcher.py b/scripts/python/powerdns_caa_patcher.py
new file mode 100755
index 0000000..59855e5
--- /dev/null
+++ b/scripts/python/powerdns_caa_patcher.py
@@ -0,0 +1,83 @@
+#!/usr/bin/python3
+"""
+PowerDNS CAA Patching/Rollout script.
+
+Created and Last modified: 13/09/2021 by Georg Pfuetzenreuter <georg@lysergic.dev>
+"""
+import requests
+import sys
+import os
+from dotenv import load_dotenv
+
+if len(sys.argv) > 1:
+    domain = sys.argv[1]
+else:
+    print("Specify the domain name.")
+    sys.exit(1)
+
+load_dotenv()
+
+# POWERDNS SETTINGS
+ENDPOINT_PDNS = os.environ.get('ENDPOINT_PDNS')
+APIKEY_PDNS = os.environ.get('APIKEY_PDNS')
+
+# CAA SETTINGS
+ca = 'letsencrypt.org'
+iodef = 'system@lysergic.dev'
+
+if None in (ENDPOINT_PDNS, APIKEY_PDNS):
+    print("Could not load environment variables. Please check your .env file.")
+    sys.exit(0)
+
+print("Scanning " + domain)
+
+# QUERY POWERDNS
+URL = ENDPOINT_PDNS + '/api/v1/servers/localhost/zones/' + domain + './export'
+try:
+    response = requests.get(
+    URL,
+    headers = {'accept': 'text/plain', 'X-API-Key': APIKEY_PDNS},
+    )
+    data = response.text
+    #print(data)
+    for record in data.split('\n'):
+        if '*' in record:
+            #print(record)
+            wildcard = True
+            break
+        else:
+            wildcard = False
+    print('Wildcards found: ', wildcard)
+except requests.exceptions.ConnectionError as err:
+    print("Connection failed.")
+    sys.exit(1)
+except requests.exceptions.HTTPError as err:
+    print(err)
+    sys.exit(1)
+print("Patching CAA ...")
+URL = ENDPOINT_PDNS + '/api/v1/servers/localhost/zones/' + domain + "."
+if wildcard == True:
+    issuewild = 'letsencrypt.org'
+else:
+    issuewild = ';'
+payload = {
+"rrsets": [{"name": domain + ".", "type": "CAA", "ttl": "3600", "changetype": "REPLACE", "records": [{"content": "0 iodef \"" + iodef + "\"", "disabled": False, "name": domain + "."}, {"content": "0 issue \"" + ca + "\"" , "disabled": False, "name": domain + "."}, {"content": "0 issuewild \"" + issuewild + "\""}]}]
+}
+response = requests.patch(
+URL,
+headers = {'accept': 'application/json', 'X-API-Key': APIKEY_PDNS, 'Content-Type': 'application/json'},
+json = payload,
+)
+status = response.status_code
+if status == 204:
+    print("OK!")
+elif status == 422:
+    print("Failed:")
+    print(response.json())
+    sys.exit(1)
+else:
+    print("Unhandled error.")
+    print(status)
+    print(response.json())
+    sys.exit(1)
+sys.exit(0)