test/modules/md/dns01.py

#!/usr/bin/env python3

import os
import subprocess
import sys

from configparser import ConfigParser, ExtendedInterpolation

challtestsrv = "localhost:8055"


def curl_bin():
    """The curl binary the test framework is configured to use."""
    if 'CURL' in os.environ:
        return os.environ['CURL']
    pyhttpd_dir = os.path.join(os.path.dirname(__file__), '..', '..', 'pyhttpd')
    config_ini = os.getenv("PYHTTPD_CONFIG", os.path.join(pyhttpd_dir, 'config.ini'))
    config = ConfigParser(interpolation=ExtendedInterpolation())
    config.read(config_ini)
    return config.get('global', 'curl_bin', fallback='curl')


curl = curl_bin()


def run(args):
    sys.stderr.write(f"run: {' '.join(args)}\n")
    p = subprocess.Popen(args, stdin=None, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    output, errput = p.communicate(None)
    rv = p.wait()
    if rv != 0:
        sys.stderr.write(errput.decode())
    sys.stdout.write(output.decode())
    return rv


def teardown(domain):
    rv = run([curl, '-s', '-d', f'{{"host":"_acme-challenge.{domain}"}}',
              f'{challtestsrv}/clear-txt'])
    if rv == 0:
        rv = run([curl, '-s', '-d', f'{{"host":"{domain}"}}',
                  f'{challtestsrv}/set-txt'])
    return rv


def setup(domain, challenge):
    teardown(domain)
    rv = run([curl, '-s', '-d', f'{{"host":"{domain}", "addresses":["127.0.0.1"]}}',
              f'{challtestsrv}/set-txt'])
    if rv == 0:
        rv = run([curl, '-s', '-d', f'{{"host":"_acme-challenge.{domain}.", "value":"{challenge}"}}',
                  f'{challtestsrv}/set-txt'])
    return rv


def main(argv):
    if len(argv) > 1:
        if argv[1] == 'setup':
            if len(argv) != 4:
                sys.stderr.write("wrong number of arguments: dns01.py setup <domain> <challenge>\n")
                sys.exit(2)
            rv = setup(argv[2], argv[3])
        elif argv[1] == 'teardown':
            if len(argv) != 3:
                sys.stderr.write("wrong number of arguments: dns01.py teardown <domain>\n")
                sys.exit(1)
            rv = teardown(argv[2])
        else:
            sys.stderr.write(f"unknown option {argv[1]}\n")
            rv = 2
    else:
        sys.stderr.write("dns01.py wrong number of arguments\n")
        rv = 2
    sys.exit(rv)
    

if __name__ == "__main__":
    main(sys.argv)