82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
from os.path import join, exists
|
|
from re import sub
|
|
from cryptography.hazmat.primitives import serialization as crypto_serialization
|
|
from cryptography.hazmat.primitives.asymmetric import rsa
|
|
from cryptography.hazmat.primitives.serialization import load_pem_private_key
|
|
|
|
|
|
defaults = {
|
|
'apt': {
|
|
'packages': {
|
|
'opendkim': {},
|
|
'opendkim-tools': {},
|
|
},
|
|
},
|
|
'opendkim': {
|
|
'keys': {},
|
|
},
|
|
}
|
|
|
|
|
|
@metadata_reactor.provides(
|
|
'opendkim/keys',
|
|
)
|
|
def keys(metadata):
|
|
keys = {}
|
|
|
|
for domain in metadata.get('mailserver/domains'):
|
|
if domain in metadata.get(f'opendkim/keys'):
|
|
continue
|
|
|
|
privkey_path = join(repo.path, 'data', 'dkim', f'{domain}.privkey.enc')
|
|
|
|
if not exists(privkey_path):
|
|
with open(privkey_path, 'w') as file:
|
|
file.write(
|
|
repo.vault.encrypt(
|
|
rsa.generate_private_key(
|
|
public_exponent=65537,
|
|
key_size=2048
|
|
).private_bytes(
|
|
crypto_serialization.Encoding.PEM,
|
|
crypto_serialization.PrivateFormat.PKCS8,
|
|
crypto_serialization.NoEncryption()
|
|
).decode()
|
|
)
|
|
)
|
|
|
|
with open(privkey_path, 'r') as file:
|
|
privkey = str(repo.vault.decrypt(file.read()))
|
|
|
|
keys[domain] = {
|
|
'public': ''.join(
|
|
load_pem_private_key(privkey.encode(), password=None).public_key().public_bytes(
|
|
crypto_serialization.Encoding.PEM,
|
|
crypto_serialization.PublicFormat.SubjectPublicKeyInfo
|
|
).decode().split('\n')[1:-2]
|
|
),
|
|
'private': privkey,
|
|
}
|
|
|
|
return {
|
|
'opendkim': {
|
|
'keys': keys,
|
|
}
|
|
}
|
|
|
|
|
|
@metadata_reactor.provides(
|
|
'dns',
|
|
)
|
|
def dns(metadata):
|
|
dns = {}
|
|
|
|
for domain, keys in metadata.get('opendkim/keys').items():
|
|
raw_key = sub('^ssh-rsa ', '', keys['public'])
|
|
dns[f'mail._domainkey.{domain}'] = {
|
|
'TXT': [f'v=DKIM1; k=rsa; p={raw_key}'],
|
|
}
|
|
|
|
return {
|
|
'dns': dns,
|
|
}
|