|
|
|
@ -1,241 +1,393 @@ |
|
|
|
|
#!/usr/bin/env python3 |
|
|
|
|
|
|
|
|
|
import grp |
|
|
|
|
import logging |
|
|
|
|
import os |
|
|
|
|
import pwd |
|
|
|
|
import subprocess |
|
|
|
|
import sys |
|
|
|
|
from datetime import datetime |
|
|
|
|
from typing import List, Dict |
|
|
|
|
|
|
|
|
|
import josepy as jose |
|
|
|
|
import acme |
|
|
|
|
from OpenSSL import crypto |
|
|
|
|
import josepy as jose |
|
|
|
|
import requests |
|
|
|
|
import yaml |
|
|
|
|
from acme import client, messages, challenges, crypto_util |
|
|
|
|
from OpenSSL import crypto |
|
|
|
|
|
|
|
|
|
SSL_CONFIG_DIR="/etc/ssl" |
|
|
|
|
SSL_CONFIG_DIR = "/etc/ssl" |
|
|
|
|
SSL_CONFIG_FILE = os.path.join(SSL_CONFIG_DIR, "acme.yml") |
|
|
|
|
SSL_ACCOUNTS_DIR = os.path.join(SSL_CONFIG_DIR, "accounts") |
|
|
|
|
SSL_KEYS_DIR = os.path.join(SSL_CONFIG_DIR, "private") |
|
|
|
|
SSL_CSR_DIR = os.path.join(SSL_CONFIG_DIR, "csr") |
|
|
|
|
SSL_CERTS_DIR = os.path.join(SSL_CONFIG_DIR, "certs") |
|
|
|
|
|
|
|
|
|
try: |
|
|
|
|
SSL_GROUP = grp.getgrnam("ssl-cert").gr_name |
|
|
|
|
except KeyError: |
|
|
|
|
SSL_GROUP = "root" |
|
|
|
|
|
|
|
|
|
ACME_ACCOUNT_KEY = os.path.join(SSL_ACCOUNTS_DIR, "acme_account.key") |
|
|
|
|
ACME_CHALLENGE_DIR = "/var/www/acme/.well-known/acme-challenge" |
|
|
|
|
LETSENCRYPT_DIRECTORY_URL = "https://acme-v02.api.letsencrypt.org/directory" |
|
|
|
|
OPENSSL_DATE_FORMAT = '%Y%m%d%H%M%SZ' |
|
|
|
|
USER_AGENT="python-acme" |
|
|
|
|
|
|
|
|
|
def get_cert_expiration_days(cert_path: str): |
|
|
|
|
"""Calculate remaining number of days before certificate's expiration. |
|
|
|
|
CERT_REMAINING_DAYS = 30 |
|
|
|
|
USER_AGENT = "python-acme" |
|
|
|
|
|
|
|
|
|
OS_OWNERS = {} |
|
|
|
|
OS_GROUPS = {} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def chown(path: str, owner: str, group: str, **kwargs) -> None: |
|
|
|
|
if owner not in OS_OWNERS: |
|
|
|
|
OS_OWNERS[owner] = pwd.getpwnam(owner).pw_uid |
|
|
|
|
if group not in OS_GROUPS: |
|
|
|
|
OS_GROUPS[group] = grp.getgrnam(group).gr_gid |
|
|
|
|
os.chown(path, OS_OWNERS[owner], OS_GROUPS[group], **kwargs) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class PrivateKey: |
|
|
|
|
def __init__(self, private_key: str, owner: str, group: str): |
|
|
|
|
self.private_key = private_key |
|
|
|
|
self.owner = owner |
|
|
|
|
self.group = group |
|
|
|
|
|
|
|
|
|
def load_or_create_private_key(self) -> crypto.PKey: |
|
|
|
|
logging.info(f"Loading private key from {self.private_key}") |
|
|
|
|
try: |
|
|
|
|
with open(self.private_key, 'rb') as pem_in: |
|
|
|
|
private_key_pem = crypto.load_privatekey(crypto.FILETYPE_PEM, |
|
|
|
|
pem_in.read()) |
|
|
|
|
return private_key_pem |
|
|
|
|
except IOError as e: |
|
|
|
|
logging.warning(f"Unable to load private key: {e}") |
|
|
|
|
|
|
|
|
|
logging.info("Generating a new private key") |
|
|
|
|
private_key_pem = crypto.PKey() |
|
|
|
|
private_key_pem.generate_key(crypto.TYPE_RSA, 4096) |
|
|
|
|
|
|
|
|
|
logging.info(f"Writing private key into {self.private_key}") |
|
|
|
|
private_key_dir = os.path.dirname(self.private_key) |
|
|
|
|
os.makedirs(private_key_dir, mode=0o750, exist_ok=True) |
|
|
|
|
chown(private_key_dir, self.owner, self.group) |
|
|
|
|
ofd = os.open(self.private_key, os.O_CREAT | os.O_WRONLY, 0o640) |
|
|
|
|
with open(ofd, 'wb') as pem_out: |
|
|
|
|
pem_out.write(crypto.dump_privatekey(crypto.FILETYPE_PEM, |
|
|
|
|
private_key_pem)) |
|
|
|
|
chown(self.private_key, self.owner, self.group) |
|
|
|
|
return private_key_pem |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Account(PrivateKey): |
|
|
|
|
def __init__(self, private_key: str, email: str, owner: str, group: str): |
|
|
|
|
super().__init__(private_key, owner, group) |
|
|
|
|
self.private_key_pem = self.load_or_create_private_key() |
|
|
|
|
self.email = email |
|
|
|
|
|
|
|
|
|
self.jwk = jose.JWKRSA(key=self.private_key_pem.to_cryptography_key()) |
|
|
|
|
self.jwk_pub = self.jwk.public_key() |
|
|
|
|
|
|
|
|
|
def register(self, directory_url: str) -> acme.client.ClientV2: |
|
|
|
|
client_network = client.ClientNetwork(self.jwk, user_agent=USER_AGENT) |
|
|
|
|
|
|
|
|
|
directory_json = client_network.get(directory_url).json() |
|
|
|
|
directory = messages.Directory.from_json(directory_json) |
|
|
|
|
acme_client = client.ClientV2(directory, net=client_network) |
|
|
|
|
|
|
|
|
|
try: |
|
|
|
|
logging.info(f"Registering with a new ACME account") |
|
|
|
|
registration = messages.NewRegistration.from_data( |
|
|
|
|
email=self.email, |
|
|
|
|
terms_of_service_agreed=True) |
|
|
|
|
acme_client.new_account(registration) |
|
|
|
|
except acme.errors.ConflictError as e: |
|
|
|
|
logging.warning(f"Unable to create a new ACME account: {e}") |
|
|
|
|
logging.info("Registering with an existing ACME account") |
|
|
|
|
registration_message = messages.NewRegistration( |
|
|
|
|
key=self.jwk_pub, |
|
|
|
|
only_return_existing=True) |
|
|
|
|
registration_response = acme_client._post(directory["newAccount"], |
|
|
|
|
registration_message) |
|
|
|
|
account = acme_client._regr_from_response(registration_response) |
|
|
|
|
acme_client.net.account = account |
|
|
|
|
return acme_client |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Challenge: |
|
|
|
|
def __init__(self, path: str, acme_client: acme.client.ClientV2): |
|
|
|
|
self.path = path |
|
|
|
|
self.acme_client = acme_client |
|
|
|
|
|
|
|
|
|
def run(self, csr_pem: str) -> str: |
|
|
|
|
logging.info("Ordering ACME challenge") |
|
|
|
|
order = self.acme_client.new_order(csr_pem) |
|
|
|
|
|
|
|
|
|
logging.info("Selecting HTTP-01 ACME challenge") |
|
|
|
|
challenge = self.select_http01_challenge(order) |
|
|
|
|
|
|
|
|
|
logging.info("Performing HTTP-01 ACME challenge") |
|
|
|
|
fullchain_pem = self.perform_http01_challenge(challenge, order) |
|
|
|
|
return fullchain_pem |
|
|
|
|
|
|
|
|
|
@staticmethod |
|
|
|
|
def select_http01_challenge(order: messages.OrderResource) \ |
|
|
|
|
-> challenges.Challenge: |
|
|
|
|
"""Select the HTTP-01 challenge from a given order |
|
|
|
|
|
|
|
|
|
:param messages.OrderResource order: Order containing the challenges. |
|
|
|
|
|
|
|
|
|
:returns: The HTTP-01 challenge. |
|
|
|
|
:rtype: challenges.Challenge |
|
|
|
|
""" |
|
|
|
|
for auth in order.authorizations: |
|
|
|
|
for challenge in auth.body.challenges: |
|
|
|
|
if isinstance(challenge.chall, challenges.HTTP01): |
|
|
|
|
return challenge |
|
|
|
|
raise Exception("HTTP-01 challenge was not offered by the CA server.") |
|
|
|
|
|
|
|
|
|
def perform_http01_challenge(self, challenge: challenges.Challenge, |
|
|
|
|
order: messages.OrderResource) -> str: |
|
|
|
|
"""Perform the HTTP-01 challenge in a given directory |
|
|
|
|
|
|
|
|
|
:param challenges.Challenge challenge: A ACME challenge |
|
|
|
|
:param messages.OrderResource order: An ACME order |
|
|
|
|
|
|
|
|
|
:returns: The fullchain certificate in PEM file format |
|
|
|
|
:rtype: str |
|
|
|
|
""" |
|
|
|
|
account_key = self.acme_client.net.key |
|
|
|
|
response, validation = challenge.response_and_validation(account_key) |
|
|
|
|
validation_filename, _ = validation.split('.') |
|
|
|
|
|
|
|
|
|
challenge_path = os.path.join(self.path, validation_filename) |
|
|
|
|
logging.info(f"Writing challenge into {challenge_path} directory") |
|
|
|
|
os.makedirs(self.path, mode=0o755, exist_ok=True) |
|
|
|
|
with open(challenge_path, 'w') as ofd: |
|
|
|
|
ofd.write(validation) |
|
|
|
|
|
|
|
|
|
self.acme_client.answer_challenge(challenge, response) |
|
|
|
|
|
|
|
|
|
fullchain_pem = self.acme_client.poll_and_finalize(order).fullchain_pem |
|
|
|
|
os.remove(challenge_path) |
|
|
|
|
return fullchain_pem |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Domain(PrivateKey): |
|
|
|
|
def __init__(self, name: str, owner: str, group: str, |
|
|
|
|
alt_names: List[str] = None, |
|
|
|
|
remaining_days: int = CERT_REMAINING_DAYS, |
|
|
|
|
hooks: List[str] = None): |
|
|
|
|
private_key = os.path.join(SSL_KEYS_DIR, name + '.key') |
|
|
|
|
super().__init__(private_key, owner, group) |
|
|
|
|
|
|
|
|
|
self.name = name |
|
|
|
|
self.alt_names = [] if alt_names is None else alt_names |
|
|
|
|
self.remaining_days = remaining_days |
|
|
|
|
self.hooks = [] if hooks is None else hooks |
|
|
|
|
|
|
|
|
|
certs_dir = os.path.join(SSL_CERTS_DIR, name + '.d') |
|
|
|
|
self.csr = os.path.join(SSL_CSR_DIR, name + '.csr') |
|
|
|
|
self.cert = os.path.join(certs_dir, 'cert.pem') |
|
|
|
|
self.chain_cert = os.path.join(certs_dir, 'chain.pem') |
|
|
|
|
self.fullchain_cert = os.path.join(certs_dir, 'fullchain.pem') |
|
|
|
|
self.cert_expiration_days = None |
|
|
|
|
|
|
|
|
|
def check_certificate_expiration_date(self) -> bool: |
|
|
|
|
"""Indicate whether the certificate will expire soon or not. |
|
|
|
|
|
|
|
|
|
:returns: True if the certificate expires soon. |
|
|
|
|
:rtype: bool |
|
|
|
|
""" |
|
|
|
|
if self.cert_expiration_days is None: |
|
|
|
|
logging.debug(f"Checking '{self.cert}' certificate expiration " |
|
|
|
|
"date") |
|
|
|
|
try: |
|
|
|
|
with open(self.cert, 'rb') as pem_in: |
|
|
|
|
cert_pem = crypto.load_certificate(crypto.FILETYPE_PEM, |
|
|
|
|
pem_in.read()) |
|
|
|
|
except IOError as e: |
|
|
|
|
logging.warning(f"Unable to load certificate: {e}") |
|
|
|
|
self.cert_expiration_days = float("-inf") |
|
|
|
|
return True |
|
|
|
|
|
|
|
|
|
now = datetime.now() |
|
|
|
|
cert_raw_expiration_date = cert_pem.get_notAfter().decode() |
|
|
|
|
cert_expiration_date = datetime.strptime(cert_raw_expiration_date, |
|
|
|
|
OPENSSL_DATE_FORMAT) |
|
|
|
|
self.cert_expiration_days = (cert_expiration_date - now).days |
|
|
|
|
logging.info(f"Certificate expires in {self.cert_expiration_days} " |
|
|
|
|
"days.") |
|
|
|
|
return self.cert_expiration_days <= self.remaining_days |
|
|
|
|
|
|
|
|
|
def renew_cert(self, challenge: Challenge, force: bool = False) -> None: |
|
|
|
|
if not (force or self.check_certificate_expiration_date()): |
|
|
|
|
return |
|
|
|
|
logging.info(f"Renewing {self.name} certificates") |
|
|
|
|
|
|
|
|
|
csr_pem = self.load_or_create_csr() |
|
|
|
|
fullchain_pem = challenge.run(csr_pem) |
|
|
|
|
|
|
|
|
|
logging.info(f"Saving {self.name} certificates") |
|
|
|
|
certs_pem = [] |
|
|
|
|
for line in fullchain_pem.split('\n'): |
|
|
|
|
if 'BEGIN CERTIFICATE' in line: |
|
|
|
|
cert_pem = '' |
|
|
|
|
|
|
|
|
|
cert_pem += line + '\n' |
|
|
|
|
|
|
|
|
|
if 'END CERTIFICATE' in line: |
|
|
|
|
certs_pem.append(cert_pem) |
|
|
|
|
|
|
|
|
|
logging.info(f"Writing '{self.cert}' certificate") |
|
|
|
|
cert_dir = os.path.dirname(self.cert) |
|
|
|
|
os.makedirs(cert_dir, mode=0o755, exist_ok=True) |
|
|
|
|
with open(self.cert, 'w') as pem_out: |
|
|
|
|
pem_out.write(certs_pem[0]) |
|
|
|
|
|
|
|
|
|
logging.info(f"Writing '{self.chain_cert}' chain certificate") |
|
|
|
|
chain_cert_dir = os.path.dirname(self.chain_cert) |
|
|
|
|
os.makedirs(chain_cert_dir, mode=0o755, exist_ok=True) |
|
|
|
|
with open(self.chain_cert, 'w') as pem_out: |
|
|
|
|
pem_out.write(''.join(certs_pem[1:])) |
|
|
|
|
|
|
|
|
|
logging.info(f"Writing '{self.fullchain_cert}' full chain certificate") |
|
|
|
|
fullchain_cert_dir = os.path.dirname(self.fullchain_cert) |
|
|
|
|
os.makedirs(fullchain_cert_dir, mode=0o755, exist_ok=True) |
|
|
|
|
with open(self.fullchain_cert, 'w') as pem_out: |
|
|
|
|
pem_out.write(fullchain_pem) |
|
|
|
|
|
|
|
|
|
self.run_hooks() |
|
|
|
|
|
|
|
|
|
def load_or_create_csr(self) -> str: |
|
|
|
|
"""Load or create a CSR for the domain. |
|
|
|
|
|
|
|
|
|
:returns: The CSR in PEM file format. |
|
|
|
|
:rtype: str |
|
|
|
|
""" |
|
|
|
|
logging.info(f"Loading {self.name} CSR file from '{self.csr}'") |
|
|
|
|
try: |
|
|
|
|
with open(self.csr, 'rb') as pem_in: |
|
|
|
|
csr = pem_in.read() |
|
|
|
|
csr_pem = crypto.load_certificate_request(crypto.FILETYPE_PEM, csr) |
|
|
|
|
except IOError as e: |
|
|
|
|
logging.warning(f"Unable to load CSR file: {e}") |
|
|
|
|
csr = self.create_csr() |
|
|
|
|
|
|
|
|
|
logging.info("Checking if CSR contains all the domains") |
|
|
|
|
domain_names = set() |
|
|
|
|
for ext in csr_pem.get_extensions(): |
|
|
|
|
if ext.get_short_name().decode() != "subjectAltName": |
|
|
|
|
continue |
|
|
|
|
subject_domains = str(ext).split(',') |
|
|
|
|
for subject_domain in subject_domains: |
|
|
|
|
_, domain = str(subject_domain).split(':') |
|
|
|
|
domain_names.add(domain) |
|
|
|
|
domain_names_diff = domain_names & set([self.name] + self.alt_names) |
|
|
|
|
if len(domain_names_diff) > 0: |
|
|
|
|
logging.warning(f"Differences found in CSR: {domain_names_diff}") |
|
|
|
|
csr = self.create_csr() |
|
|
|
|
|
|
|
|
|
return csr |
|
|
|
|
|
|
|
|
|
def create_csr(self): |
|
|
|
|
private_key_pem = self.load_or_create_private_key() |
|
|
|
|
private_key = crypto.dump_privatekey(crypto.FILETYPE_PEM, |
|
|
|
|
private_key_pem) |
|
|
|
|
|
|
|
|
|
logging.info(f"Generating a new CSR") |
|
|
|
|
csr = crypto_util.make_csr(private_key, [self.name] + self.alt_names) |
|
|
|
|
|
|
|
|
|
logging.info(f"Writing CSR file into '{self.csr}'") |
|
|
|
|
csr_dir = os.path.dirname(self.csr) |
|
|
|
|
os.makedirs(csr_dir, mode=0o755, exist_ok=True) |
|
|
|
|
with open(self.csr, 'wb') as pem_out: |
|
|
|
|
pem_out.write(csr) |
|
|
|
|
return csr |
|
|
|
|
|
|
|
|
|
def run_hooks(self) -> None: |
|
|
|
|
for hook in self.hooks: |
|
|
|
|
logging.info(f"Running hook: '{hook}'") |
|
|
|
|
subprocess.run(hook.split()) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def main(): |
|
|
|
|
parser = argparse.ArgumentParser( |
|
|
|
|
description="Renew ACME certificates for a list of domains. This " |
|
|
|
|
"script assumes you already have an account on the CA " |
|
|
|
|
"server and a private key for each certificate.", |
|
|
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter) |
|
|
|
|
parser.add_argument("--config", "-c", default=SSL_CONFIG_FILE, |
|
|
|
|
help="Path to the config file.") |
|
|
|
|
parser.add_argument("--quiet", "-q", action="store_true", |
|
|
|
|
help="Quiet mode.") |
|
|
|
|
parser.add_argument("--verbose", "-v", action="store_true", |
|
|
|
|
help="Increase verbosity.") |
|
|
|
|
parser.add_argument("--force", "-f", action="store_true", |
|
|
|
|
help="Force certificates renewal without checking " |
|
|
|
|
"their expiration date.") |
|
|
|
|
args = parser.parse_args() |
|
|
|
|
|
|
|
|
|
:param str cert_path: Path to a certificate file. |
|
|
|
|
if args.quiet: |
|
|
|
|
log_level = logging.WARNING |
|
|
|
|
elif args.verbose: |
|
|
|
|
log_level = logging.DEBUG |
|
|
|
|
else: |
|
|
|
|
log_level = logging.INFO |
|
|
|
|
|
|
|
|
|
:returns: Number of remaining days before certificate's expiration. |
|
|
|
|
:rtype: int |
|
|
|
|
""" |
|
|
|
|
now = datetime.now() |
|
|
|
|
cert_expiration_date = now |
|
|
|
|
logging.basicConfig(stream=sys.stdout, level=log_level, |
|
|
|
|
format="%(levelname)s:%(message)s") |
|
|
|
|
|
|
|
|
|
try: |
|
|
|
|
with open(cert_path, 'rb') as pem_in: |
|
|
|
|
fullchain_pem = crypto.load_certificate(crypto.FILETYPE_PEM, pem_in.read()) |
|
|
|
|
cert_expiration_date = datetime.strptime(fullchain_pem.get_notAfter().decode(), OPENSSL_DATE_FORMAT) |
|
|
|
|
with open(args.config, 'r') as ifd: |
|
|
|
|
config = yaml.safe_load(ifd) |
|
|
|
|
except IOError as e: |
|
|
|
|
logging.warning(f"Unable to load certificate: {e}") |
|
|
|
|
|
|
|
|
|
cert_expiration_days = (cert_expiration_date - now).days |
|
|
|
|
return cert_expiration_days |
|
|
|
|
logging.error(f"Unable to load config file: {e}") |
|
|
|
|
return 1 |
|
|
|
|
|
|
|
|
|
def load_or_create_csr(domain: str, csr_path: str, privkey_path: str): |
|
|
|
|
"""Load or create a CSR for a given domain. |
|
|
|
|
|
|
|
|
|
:param str domain: A domain name. |
|
|
|
|
:param str csr_path: Path to a CSR file. |
|
|
|
|
:param str privkey_path: Path to a private key. It is used as fallback when the CSR doesn't exist. |
|
|
|
|
|
|
|
|
|
:returns: The CSR in PEM file format. |
|
|
|
|
:rtype: str |
|
|
|
|
""" |
|
|
|
|
logging.info(f"Loading {domain} CSR file from {csr_path}…") |
|
|
|
|
try: |
|
|
|
|
with open(csr_path, 'r') as pem_in: |
|
|
|
|
csr_pem = pem_in.read() |
|
|
|
|
except IOError as e: |
|
|
|
|
logging.warning(f"Unable to load CSR file: {e}") |
|
|
|
|
logging.info(f"Loading {domain} private key from {privkey_path}…") |
|
|
|
|
with open(privkey_path, 'r') as pem_in: |
|
|
|
|
privkey_pem = pem_in.read() |
|
|
|
|
|
|
|
|
|
logging.info(f"Generating a new CSR…") |
|
|
|
|
csr_pem = crypto_util.make_csr(privkey_pem, [domain]) |
|
|
|
|
return csr_pem |
|
|
|
|
|
|
|
|
|
def select_http01_challenge(order: messages.OrderResource): |
|
|
|
|
"""Select the HTTP-01 challenge from a given order |
|
|
|
|
|
|
|
|
|
:param messages.OrderResource order: ACME order containing the challenges. |
|
|
|
|
|
|
|
|
|
:returns: The HTTP-01 challenge. |
|
|
|
|
:rtype: challenges.Challenge |
|
|
|
|
""" |
|
|
|
|
for auth in order.authorizations: |
|
|
|
|
for challenge in auth.body.challenges: |
|
|
|
|
if isinstance(challenge.chall, challenges.HTTP01): |
|
|
|
|
return challenge |
|
|
|
|
raise Exception("HTTP-01 challenge was not offered by the CA server.") |
|
|
|
|
|
|
|
|
|
def perform_http01_challenge( |
|
|
|
|
client_acme: client.ClientV2, |
|
|
|
|
challenge: challenges.Challenge, |
|
|
|
|
order: messages.OrderResource, |
|
|
|
|
challenge_dir: str): |
|
|
|
|
"""Perform the HTTP-01 challenge in a given directory |
|
|
|
|
|
|
|
|
|
:param client.ClientV2 client_acme: A ACME v2 client |
|
|
|
|
:param challenges.Challenge challenge: A ACME challenge |
|
|
|
|
:param messages.OrderResource order: An ACME order |
|
|
|
|
:param str challenge_dir: The directory containing the challenge |
|
|
|
|
|
|
|
|
|
:returns: The fullchain certificate in PEM file format |
|
|
|
|
:rtype: str |
|
|
|
|
""" |
|
|
|
|
response, validation = challenge.response_and_validation(client_acme.net.key) |
|
|
|
|
validation_filename, _ = validation.split('.') |
|
|
|
|
|
|
|
|
|
challenge_path = os.path.join(challenge_dir, validation_filename) |
|
|
|
|
with open(challenge_path, 'w') as f_out: |
|
|
|
|
f_out.write(validation) |
|
|
|
|
|
|
|
|
|
client_acme.answer_challenge(challenge, response) |
|
|
|
|
|
|
|
|
|
fullchain_pem = client_acme.poll_and_finalize(order).fullchain_pem |
|
|
|
|
|
|
|
|
|
os.remove(challenge_path) |
|
|
|
|
|
|
|
|
|
return fullchain_pem |
|
|
|
|
|
|
|
|
|
def renew_cert(domain: str, |
|
|
|
|
account_key_path: str, |
|
|
|
|
privkey_path: str, |
|
|
|
|
csr_path: str, |
|
|
|
|
certs_dir: str, |
|
|
|
|
challenge_dir: str, |
|
|
|
|
directory_url: str, |
|
|
|
|
days_before_renewal: int, |
|
|
|
|
force: bool = False): |
|
|
|
|
logging.info(f"Checking {domain} certificate's expiration date…") |
|
|
|
|
fullchain_path = os.path.join(certs_dir, "fullchain.pem") |
|
|
|
|
|
|
|
|
|
cert_expiration_days = get_cert_expiration_days(fullchain_path) |
|
|
|
|
if not force and (cert_expiration_days >= days_before_renewal): |
|
|
|
|
logging.info(f"Certificate expires in {cert_expiration_days} days. Nothing to do.") |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
logging.info(f"Certificate expires in {cert_expiration_days} days! Renewing certificate…") |
|
|
|
|
|
|
|
|
|
logging.info(f"Loading account key from {account_key_path}…") |
|
|
|
|
with open(account_key_path, 'rb') as pem_in: |
|
|
|
|
account_pem = crypto.load_privatekey(crypto.FILETYPE_PEM, pem_in.read()) |
|
|
|
|
account_jwk = jose.JWKRSA(key=account_pem.to_cryptography_key()) |
|
|
|
|
account_jwk_pub = account_jwk.public_key() |
|
|
|
|
|
|
|
|
|
csr_pem = load_or_create_csr(domain, csr_path, privkey_path) |
|
|
|
|
|
|
|
|
|
client_network = client.ClientNetwork(account_jwk, user_agent=USER_AGENT) |
|
|
|
|
directory = messages.Directory.from_json(client_network.get(directory_url).json()) |
|
|
|
|
client_acme = client.ClientV2(directory, net=client_network) |
|
|
|
|
|
|
|
|
|
logging.info("Registering with ACME account…") |
|
|
|
|
# Here we assume we already have an account |
|
|
|
|
registration_message = messages.NewRegistration( |
|
|
|
|
key=account_jwk_pub, |
|
|
|
|
only_return_existing=True) |
|
|
|
|
registration_response = client_acme._post(directory["newAccount"], registration_message) |
|
|
|
|
account = client_acme._regr_from_response(registration_response) |
|
|
|
|
client_acme.net.account = account |
|
|
|
|
|
|
|
|
|
logging.info("Ordering ACME challenge…") |
|
|
|
|
order = client_acme.new_order(csr_pem) |
|
|
|
|
|
|
|
|
|
logging.info("Selecting HTTP-01 ACME challenge…") |
|
|
|
|
challenge = select_http01_challenge(order) |
|
|
|
|
|
|
|
|
|
logging.info("Performing HTTP-01 ACME challenge…") |
|
|
|
|
fullchain_pem = perform_http01_challenge(client_acme, challenge, order, challenge_dir) |
|
|
|
|
|
|
|
|
|
logging.info(f"Writing {domain} certificates into {certs_dir}…") |
|
|
|
|
certs_pem = [] |
|
|
|
|
for line in fullchain_pem.split('\n'): |
|
|
|
|
if 'BEGIN CERTIFICATE' in line: |
|
|
|
|
cert_pem = '' |
|
|
|
|
|
|
|
|
|
cert_pem += line + '\n' |
|
|
|
|
|
|
|
|
|
if 'END CERTIFICATE' in line: |
|
|
|
|
certs_pem.append(cert_pem) |
|
|
|
|
|
|
|
|
|
cert_path = os.path.join(certs_dir, "cert.pem") |
|
|
|
|
with open(cert_path, 'w') as pem_out: |
|
|
|
|
pem_out.write(certs_pem[0]) |
|
|
|
|
|
|
|
|
|
chain_path = os.path.join(certs_dir, "chain.pem") |
|
|
|
|
with open(chain_path, 'w') as pem_out: |
|
|
|
|
pem_out.write(''.join(certs_pem[1:])) |
|
|
|
|
|
|
|
|
|
with open(fullchain_path, 'w') as pem_out: |
|
|
|
|
pem_out.write(fullchain_pem) |
|
|
|
|
domains_to_renew = [] |
|
|
|
|
for domain_name, domain_details in config["domains"].items(): |
|
|
|
|
remaining_days = domain_details.get("remaining_days", |
|
|
|
|
CERT_REMAINING_DAYS) |
|
|
|
|
domain = Domain(domain_name, |
|
|
|
|
owner=domain_details.get("owner", "root"), |
|
|
|
|
group=domain_details.get("group", SSL_GROUP), |
|
|
|
|
alt_names=domain_details.get("alt_names"), |
|
|
|
|
remaining_days=remaining_days, |
|
|
|
|
hooks=domain_details.get("hooks")) |
|
|
|
|
|
|
|
|
|
if not (args.force or domain.check_certificate_expiration_date()): |
|
|
|
|
logging.info(f"{domain.name} certificate will not be renewed.") |
|
|
|
|
continue |
|
|
|
|
|
|
|
|
|
logging.info(f"{domain.name} certificate will be renewed.") |
|
|
|
|
domains_to_renew.append(domain) |
|
|
|
|
|
|
|
|
|
if len(domains_to_renew) == 0: |
|
|
|
|
logging.info("No domain to renew. Aborting.") |
|
|
|
|
return 0 |
|
|
|
|
|
|
|
|
|
account_config = config["account"] |
|
|
|
|
account = Account( |
|
|
|
|
private_key=account_config.get("private_key", ACME_ACCOUNT_KEY), |
|
|
|
|
email=account_config["email"], |
|
|
|
|
owner=account_config.get("owner", "root"), |
|
|
|
|
group=account_config.get("group", "root")) |
|
|
|
|
|
|
|
|
|
directory_url = config.get("directory_url", LETSENCRYPT_DIRECTORY_URL) |
|
|
|
|
acme_client = account.register(directory_url) |
|
|
|
|
|
|
|
|
|
challenge_dir = config.get("challenge_dir", ACME_CHALLENGE_DIR) |
|
|
|
|
challenge = Challenge(challenge_dir, acme_client) |
|
|
|
|
|
|
|
|
|
for domain in domains_to_renew: |
|
|
|
|
domain.renew_cert(challenge, force=args.force) |
|
|
|
|
except Exception as e: |
|
|
|
|
logging.exception(e) |
|
|
|
|
return 2 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
|
|
import argparse |
|
|
|
|
|
|
|
|
|
parser = argparse.ArgumentParser( |
|
|
|
|
description="Renew ACME certificates for a list of domains. This script assumes you already have an account on the CA server and a private key for each certificate.", |
|
|
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter) |
|
|
|
|
parser.add_argument("domains", nargs='+', |
|
|
|
|
help="List of domain names for which to renew the certificate.") |
|
|
|
|
parser.add_argument("--account_key", "-a", |
|
|
|
|
default=os.path.join(SSL_CONFIG_DIR, "accounts", "acme_account.key"), |
|
|
|
|
help="Path to the account key.") |
|
|
|
|
parser.add_argument("--privkey", "-p", |
|
|
|
|
default=os.path.join(SSL_CONFIG_DIR, "private", "{domain}.pem"), |
|
|
|
|
help="Path to the private certificate.") |
|
|
|
|
parser.add_argument("--csr", "-r", |
|
|
|
|
default=os.path.join(SSL_CONFIG_DIR, "csr", "{domain}.pem"), |
|
|
|
|
help="Path to the CSR file. If the file doesn't exist, it will be generated from the private key and the domain name.") |
|
|
|
|
parser.add_argument("--certs", "-o", |
|
|
|
|
default=os.path.join(SSL_CONFIG_DIR, "certs", "{domain}.d"), |
|
|
|
|
help="Path to the certificates directory.") |
|
|
|
|
parser.add_argument("--challenge", "-c", |
|
|
|
|
default="/var/www/html/.well-known/acme-challenge", |
|
|
|
|
help="Path to the challenge directory.") |
|
|
|
|
parser.add_argument("--directory_url", "-u", |
|
|
|
|
default="https://acme-v02.api.letsencrypt.org/directory", |
|
|
|
|
help="Directory URL on which performing ACME challenges. Only ACME v2 is supported.") |
|
|
|
|
parser.add_argument("--days", "-d", type=int, |
|
|
|
|
default=30, |
|
|
|
|
help="Days before attempting to renew the certificates.") |
|
|
|
|
parser.add_argument("--quiet", "-q", |
|
|
|
|
action="store_true", |
|
|
|
|
help="Quiet mode.") |
|
|
|
|
parser.add_argument("--verbose", "-v", |
|
|
|
|
action="store_true", |
|
|
|
|
help="Increase verbosity.") |
|
|
|
|
parser.add_argument("--force", "-f", |
|
|
|
|
action="store_true", |
|
|
|
|
help="Force certificates renewal without checking their expiration date.") |
|
|
|
|
args = parser.parse_args() |
|
|
|
|
|
|
|
|
|
if args.quiet: |
|
|
|
|
log_level = logging.WARNING |
|
|
|
|
elif args.verbose: |
|
|
|
|
log_level = logging.DEBUG |
|
|
|
|
else: |
|
|
|
|
log_level = logging.INFO |
|
|
|
|
|
|
|
|
|
logging.basicConfig(stream=sys.stdout, level=log_level, format="%(levelname)s:%(message)s") |
|
|
|
|
|
|
|
|
|
for domain in args.domains: |
|
|
|
|
account_key = args.account_key.format(domain=domain) |
|
|
|
|
privkey = args.privkey.format(domain=domain) |
|
|
|
|
csr = args.csr.format(domain=domain) |
|
|
|
|
certs_dir = args.certs.format(domain=domain) |
|
|
|
|
challenge_dir = args.challenge.format(domain=domain) |
|
|
|
|
|
|
|
|
|
renew_cert(domain, |
|
|
|
|
account_key_path=account_key, |
|
|
|
|
privkey_path=privkey, |
|
|
|
|
csr_path=csr, |
|
|
|
|
certs_dir=certs_dir, |
|
|
|
|
challenge_dir=challenge_dir, |
|
|
|
|
directory_url=args.directory_url, |
|
|
|
|
days_before_renewal=args.days, |
|
|
|
|
force=args.force) |
|
|
|
|
exit(main()) |
|
|
|
|