145 lines
4.5 KiB
Python
Executable File
145 lines
4.5 KiB
Python
Executable File
#!/usr/bin/python3
|
|
"""
|
|
PowerDNS <-> Mailcow DKIM Public Key patching script.
|
|
|
|
Created and Last modified: 13/09/2021 by Georg Pfuetzenreuter <georg@lysergic.dev>
|
|
|
|
Example use cases:
|
|
- Administration failure caused individual zones to contain faulty DKIM records
|
|
- Automation failure caused individual zones to contain faulty DKIM records
|
|
- Security incident required replacing DKIM private keys
|
|
- Security audit required manipulating the DKIM records of selected zones
|
|
|
|
The public key provided by Mailcow is considered as the source of truth.
|
|
"""
|
|
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')
|
|
|
|
# MAILCOW SETTINGS
|
|
ENDPOINT_MAILCOW = os.environ.get('ENDPOINT_MAILCOW')
|
|
APIKEY_MAILCOW = os.environ.get('APIKEY_MAILCOW')
|
|
|
|
if None in (ENDPOINT_PDNS, APIKEY_PDNS, ENDPOINT_MAILCOW, APIKEY_MAILCOW):
|
|
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 '._domainkey' in record:
|
|
#print(record)
|
|
txtis = record.split('"')[1]
|
|
try:
|
|
txtis
|
|
except NameError:
|
|
print("No DKIM TXT record found.")
|
|
#else:
|
|
#print(txtis)
|
|
except requests.exceptions.ConnectionError as err:
|
|
print("Connection failed.")
|
|
sys.exit(1)
|
|
except requests.exceptions.HTTPError as err:
|
|
print(err)
|
|
sys.exit(1)
|
|
|
|
# QUERY MAILCOW (can I put cow emoji in comment?)
|
|
server = ENDPOINT_MAILCOW
|
|
api_key = APIKEY_MAILCOW
|
|
api = '/api/v1'
|
|
get = api + '/get'
|
|
URL = server + get + '/dkim/' + domain
|
|
try:
|
|
response = requests.get(
|
|
URL,
|
|
headers = {'accept': 'application/json', 'X-API-Key': api_key},
|
|
)
|
|
data = response.json()
|
|
selector = data['dkim_selector']
|
|
txtshould = data['dkim_txt']
|
|
length = data['length']
|
|
pubkey = data['pubkey']
|
|
#print(f"Domain: {domain}\nSelector: {selector}\nTXT: {txtshould}\nPublic Key: {pubkey}")
|
|
#print(txtshould)
|
|
except KeyError:
|
|
print("No or faulty DKIM lookup from Mailcow. ABORTING.")
|
|
sys.exit(1)
|
|
except requests.exceptions.ConnectionError as err:
|
|
print("Connection failed.")
|
|
sys.exit(1)
|
|
except requests.exceptions.HTTPError as err:
|
|
print(err)
|
|
sys.exit(1)
|
|
# COMPARE
|
|
if txtis == txtshould:
|
|
print(domain + ": All good!")
|
|
if txtis != txtshould:
|
|
print("Mismatch!")
|
|
print("Patching SPF and DMARC ...")
|
|
URL = ENDPOINT_PDNS + '/api/v1/servers/localhost/zones/' + domain + "."
|
|
payload = {
|
|
"rrsets": [{"name": domain + ".", "type": "TXT", "ttl": "3600", "changetype": "REPLACE", "records": [{"content": "\"v=spf1 mx a -all\"", "disabled": False, "name": domain + "."}, {"content": "\"v=DMARC1; p=reject; rua=mailto:system@lysergic.dev\"", "disabled": False, "name": domain + "."}]}]
|
|
}
|
|
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("SPF/DMARC: OK!")
|
|
elif status == 422:
|
|
print("SPF/DMARC: Failed:")
|
|
print(response.json())
|
|
sys.exit(1)
|
|
else:
|
|
print("Unhandled error.")
|
|
print(status)
|
|
print(response.json())
|
|
sys.exit(1)
|
|
print("Patching DKIM ...")
|
|
payload = {
|
|
"rrsets": [{"name": selector + "._domainkey." + domain + ".", "type": "TXT", "ttl": "3600", "changetype": "REPLACE", "records": [{"content": "\""+ txtshould + "\"", "disabled": False, "name": selector + "._domainkey." + domain + "."}]}]
|
|
}
|
|
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("DKIM: OK!")
|
|
elif status == 422:
|
|
print("DKIM: Failed:")
|
|
print(response.json())
|
|
sys.exit(1)
|
|
else:
|
|
print("Unhandled error.")
|
|
print(status)
|
|
print(response.json())
|
|
sys.exit(1)
|
|
print("Done.")
|
|
sys.exit(0)
|
|
|