mirror of
https://github.com/certbot/certbot.git
synced 2026-01-23 07:20:55 +03:00
- to make "run" the default, we need to add it to the args - which means we need to pass that back up to the actual argparse call (This is ugly... probably HelpfulArgParser needs to actuall inherit from argparse...)
801 lines
30 KiB
Python
801 lines
30 KiB
Python
"""Let's Encrypt CLI."""
|
|
# TODO: Sanity check all input. Be sure to avoid shell code etc...
|
|
import argparse
|
|
import atexit
|
|
import functools
|
|
import logging
|
|
import logging.handlers
|
|
import os
|
|
import pkg_resources
|
|
import sys
|
|
import time
|
|
import traceback
|
|
|
|
import configargparse
|
|
import zope.component
|
|
import zope.interface.exceptions
|
|
import zope.interface.verify
|
|
|
|
import letsencrypt
|
|
|
|
from letsencrypt import account
|
|
from letsencrypt import configuration
|
|
from letsencrypt import constants
|
|
from letsencrypt import client
|
|
from letsencrypt import errors
|
|
from letsencrypt import interfaces
|
|
from letsencrypt import le_util
|
|
from letsencrypt import log
|
|
from letsencrypt import reporter
|
|
|
|
from letsencrypt.display import util as display_util
|
|
from letsencrypt.display import ops as display_ops
|
|
from letsencrypt.plugins import disco as plugins_disco
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# Argparse's help formatting has a lot of unhelpful peculiarities, so we want
|
|
# to replace as much of it as we can...
|
|
|
|
# This is the stub to include in help generated by argparse
|
|
|
|
SHORT_USAGE = """
|
|
letsencrypt [SUBCOMMAND] [options] [domains]
|
|
|
|
The Let's Encrypt agent can obtain and install HTTPS/TLS/SSL certificates. By
|
|
default, it will attempt to use a webserver both for obtaining and installing
|
|
the cert. """
|
|
|
|
# This is the short help for letsencrypt --help, where we disable argparse
|
|
# altogether
|
|
USAGE = SHORT_USAGE + """Major SUBCOMMANDS are:
|
|
|
|
(default) everything Obtain & install a cert in your current webserver
|
|
auth Authenticate & obtain cert, but do not install it
|
|
install Install a previously obtained cert in a server
|
|
revoke Revoke a previously obtained certificate
|
|
rollback Rollback server configuration changes made during install
|
|
config_changes Show changes made to server config during installation
|
|
|
|
Choice of server for authentication/installation:
|
|
|
|
--apache Use the Apache plugin for authentication & installation
|
|
--nginx Use the Nginx plugin for authentication & installation
|
|
--standalone Run a standalone HTTPS server (for authentication only)
|
|
OR:
|
|
--authenticator standalone --installer nginx
|
|
|
|
More detailed help:
|
|
|
|
-h, --help [topic] print this message, or detailed help on a topic;
|
|
the available topics are:
|
|
|
|
all, apache, automation, nginx, paths, security, testing, or any of the
|
|
subcommands
|
|
"""
|
|
|
|
|
|
def _find_domains(args, installer):
|
|
if args.domains is None:
|
|
domains = display_ops.choose_names(installer)
|
|
else:
|
|
domains = args.domains
|
|
|
|
if not domains:
|
|
raise errors.Error("Please specify --domains, or --installer that "
|
|
"will help in domain names autodiscovery")
|
|
|
|
return domains
|
|
|
|
|
|
def _determine_account(args, config):
|
|
"""Determine which account to use.
|
|
|
|
In order to make the renewer (configuration de/serialization) happy,
|
|
if ``args.account`` is ``None``, it will be updated based on the
|
|
user input. Same for ``args.email``.
|
|
|
|
:param argparse.Namespace args: CLI arguments
|
|
:param letsencrypt.interface.IConfig config: Configuration object
|
|
:param .AccountStorage account_storage: Account storage.
|
|
|
|
:returns: Account and optionally ACME client API (biproduct of new
|
|
registration).
|
|
:rtype: `tuple` of `letsencrypt.account.Account` and
|
|
`acme.client.Client`
|
|
|
|
"""
|
|
account_storage = account.AccountFileStorage(config)
|
|
acme = None
|
|
|
|
if args.account is not None:
|
|
acc = account_storage.load(args.account)
|
|
else:
|
|
accounts = account_storage.find_all()
|
|
if len(accounts) > 1:
|
|
acc = display_ops.choose_account(accounts)
|
|
elif len(accounts) == 1:
|
|
acc = accounts[0]
|
|
else: # no account registered yet
|
|
if args.email is None:
|
|
args.email = display_ops.get_email()
|
|
if not args.email: # get_email might return ""
|
|
args.email = None
|
|
|
|
def _tos_cb(regr):
|
|
if args.tos:
|
|
return True
|
|
msg = ("Please read the Terms of Service at {0}. You "
|
|
"must agree in order to register with the ACME "
|
|
"server at {1}".format(
|
|
regr.terms_of_service, config.server))
|
|
return zope.component.getUtility(interfaces.IDisplay).yesno(
|
|
msg, "Agree", "Cancel")
|
|
|
|
try:
|
|
acc, acme = client.register(
|
|
config, account_storage, tos_cb=_tos_cb)
|
|
except errors.Error as error:
|
|
logger.debug(error, exc_info=True)
|
|
raise errors.Error(
|
|
"Unable to register an account with ACME server")
|
|
|
|
args.account = acc.id
|
|
return acc, acme
|
|
|
|
|
|
def _init_le_client(args, config, authenticator, installer):
|
|
if authenticator is not None:
|
|
# if authenticator was given, then we will need account...
|
|
acc, acme = _determine_account(args, config)
|
|
logger.debug("Picked account: %r", acc)
|
|
# XXX
|
|
#crypto_util.validate_key_csr(acc.key)
|
|
else:
|
|
acc, acme = None, None
|
|
|
|
return client.Client(config, acc, authenticator, installer, acme=acme)
|
|
|
|
|
|
def run(args, config, plugins):
|
|
"""Obtain a certificate and install."""
|
|
if args.configurator is not None and (args.installer is not None or
|
|
args.authenticator is not None):
|
|
return ("Either --configurator or --authenticator/--installer"
|
|
"pair, but not both, is allowed")
|
|
|
|
if args.authenticator is not None or args.installer is not None:
|
|
installer = display_ops.pick_installer(
|
|
config, args.installer, plugins)
|
|
authenticator = display_ops.pick_authenticator(
|
|
config, args.authenticator, plugins)
|
|
else:
|
|
# TODO: this assumes that user doesn't want to pick authenticator
|
|
# and installer separately...
|
|
authenticator = installer = display_ops.pick_configurator(
|
|
config, args.configurator, plugins)
|
|
|
|
if installer is None or authenticator is None:
|
|
return "Configurator could not be determined"
|
|
|
|
domains = _find_domains(args, installer)
|
|
# TODO: Handle errors from _init_le_client?
|
|
le_client = _init_le_client(args, config, authenticator, installer)
|
|
lineage = le_client.obtain_and_enroll_certificate(
|
|
domains, authenticator, installer, plugins)
|
|
if not lineage:
|
|
return "Certificate could not be obtained"
|
|
le_client.deploy_certificate(
|
|
domains, lineage.privkey, lineage.cert, lineage.chain)
|
|
le_client.enhance_config(domains, args.redirect)
|
|
|
|
|
|
def auth(args, config, plugins):
|
|
"""Authenticate & obtain cert, but do not install it."""
|
|
# XXX: Update for renewer / RenewableCert
|
|
|
|
if args.domains is not None and args.csr is not None:
|
|
# TODO: --csr could have a priority, when --domains is
|
|
# supplied, check if CSR matches given domains?
|
|
return "--domains and --csr are mutually exclusive"
|
|
|
|
authenticator = display_ops.pick_authenticator(
|
|
config, args.authenticator, plugins)
|
|
if authenticator is None:
|
|
return "Authenticator could not be determined"
|
|
|
|
if args.installer is not None:
|
|
installer = display_ops.pick_installer(config, args.installer, plugins)
|
|
else:
|
|
installer = None
|
|
|
|
# TODO: Handle errors from _init_le_client?
|
|
le_client = _init_le_client(args, config, authenticator, installer)
|
|
|
|
if args.csr is not None:
|
|
certr, chain = le_client.obtain_certificate_from_csr(le_util.CSR(
|
|
file=args.csr[0], data=args.csr[1], form="der"))
|
|
le_client.save_certificate(
|
|
certr, chain, args.cert_path, args.chain_path)
|
|
else:
|
|
domains = _find_domains(args, installer)
|
|
if not le_client.obtain_and_enroll_certificate(
|
|
domains, authenticator, installer, plugins):
|
|
return "Certificate could not be obtained"
|
|
|
|
|
|
def install(args, config, plugins):
|
|
"""Install a previously obtained cert in a server."""
|
|
# XXX: Update for renewer/RenewableCert
|
|
installer = display_ops.pick_installer(config, args.installer, plugins)
|
|
if installer is None:
|
|
return "Installer could not be determined"
|
|
domains = _find_domains(args, installer)
|
|
le_client = _init_le_client(
|
|
args, config, authenticator=None, installer=installer)
|
|
assert args.cert_path is not None # required=True in the subparser
|
|
le_client.deploy_certificate(
|
|
domains, args.key_path, args.cert_path, args.chain_path)
|
|
le_client.enhance_config(domains, args.redirect)
|
|
|
|
|
|
def revoke(args, unused_config, unused_plugins):
|
|
"""Revoke a previously obtained certificate."""
|
|
if args.cert_path is None and args.key_path is None:
|
|
return "At least one of --cert-path or --key-path is required"
|
|
|
|
# This depends on the renewal config and cannot be completed yet.
|
|
zope.component.getUtility(interfaces.IDisplay).notification(
|
|
"Revocation is not available with the new Boulder server yet.")
|
|
#client.revoke(args.installer, config, plugins, args.no_confirm,
|
|
# args.cert_path, args.key_path)
|
|
|
|
|
|
def rollback(args, config, plugins):
|
|
"""Rollback server configuration changes made during install."""
|
|
client.rollback(args.installer, args.checkpoints, config, plugins)
|
|
|
|
|
|
def config_changes(unused_args, config, unused_plugins):
|
|
"""Show changes made to server config during installation
|
|
|
|
View checkpoints and associated configuration changes.
|
|
|
|
"""
|
|
client.view_config_changes(config)
|
|
|
|
|
|
def plugins_cmd(args, config, plugins): # TODO: Use IDisplay rather than print
|
|
"""List server software plugins."""
|
|
logger.debug("Expected interfaces: %s", args.ifaces)
|
|
|
|
ifaces = [] if args.ifaces is None else args.ifaces
|
|
filtered = plugins.ifaces(ifaces)
|
|
logger.debug("Filtered plugins: %r", filtered)
|
|
|
|
if not args.init and not args.prepare:
|
|
print str(filtered)
|
|
return
|
|
|
|
filtered.init(config)
|
|
verified = filtered.verify(ifaces)
|
|
logger.debug("Verified plugins: %r", verified)
|
|
|
|
if not args.prepare:
|
|
print str(verified)
|
|
return
|
|
|
|
verified.prepare()
|
|
available = verified.available()
|
|
logger.debug("Prepared plugins: %s", available)
|
|
print str(available)
|
|
|
|
|
|
def read_file(filename, mode="rb"):
|
|
"""Returns the given file's contents.
|
|
|
|
:param str filename: Filename
|
|
:param str mode: open mode (see `open`)
|
|
|
|
:returns: A tuple of filename and its contents
|
|
:rtype: tuple
|
|
|
|
:raises argparse.ArgumentTypeError: File does not exist or is not readable.
|
|
|
|
"""
|
|
try:
|
|
return filename, open(filename, mode).read()
|
|
except IOError as exc:
|
|
raise argparse.ArgumentTypeError(exc.strerror)
|
|
|
|
|
|
def flag_default(name):
|
|
"""Default value for CLI flag."""
|
|
return constants.CLI_DEFAULTS[name]
|
|
|
|
|
|
def config_help(name, hidden=False):
|
|
"""Help message for `.IConfig` attribute."""
|
|
if hidden:
|
|
return argparse.SUPPRESS
|
|
else:
|
|
return interfaces.IConfig[name].__doc__
|
|
|
|
|
|
class SilentParser(object): # pylint: disable=too-few-public-methods
|
|
"""Silent wrapper around argparse.
|
|
|
|
A mini parser wrapper that doesn't print help for its
|
|
arguments. This is needed for the use of callbacks to define
|
|
arguments within plugins.
|
|
|
|
"""
|
|
def __init__(self, parser):
|
|
self.parser = parser
|
|
def add_argument(self, *args, **kwargs):
|
|
"""Wrap, but silence help"""
|
|
kwargs["help"] = argparse.SUPPRESS
|
|
self.parser.add_argument(*args, **kwargs)
|
|
|
|
|
|
HELP_TOPICS = ["all", "security", "paths", "automation", "testing", "plugins"]
|
|
|
|
|
|
class HelpfulArgumentParser(object):
|
|
"""Argparse Wrapper.
|
|
|
|
This class wraps argparse, adding the ability to make --help less
|
|
verbose, and request help on specific subcategories at a time, eg
|
|
'letsencrypt --help security' for security options.
|
|
|
|
"""
|
|
def __init__(self, args, plugins):
|
|
print args
|
|
plugin_names = [name for name, _p in plugins.iteritems()]
|
|
self.help_topics = HELP_TOPICS + plugin_names + [None]
|
|
self.parser = configargparse.ArgParser(
|
|
usage=SHORT_USAGE,
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
|
|
args_for_setting_config_path=["-c", "--config"],
|
|
default_config_files=flag_default("config_files"))
|
|
|
|
# This is the only way to turn off overly verbose config flag documentation
|
|
self.parser._add_config_file_help = False # pylint: disable=protected-access
|
|
self.silent_parser = SilentParser(self.parser)
|
|
|
|
self.args = self.preprocess_args(args)
|
|
help1 = self.prescan_for_flag("-h", self.help_topics)
|
|
help2 = self.prescan_for_flag("--help", self.help_topics)
|
|
assert max(True, "a") == "a", "Gravity changed direction"
|
|
help_arg = max(help1, help2)
|
|
if help_arg == True:
|
|
# just --help with no topic; avoid argparse altogether
|
|
print USAGE
|
|
sys.exit(0)
|
|
self.visible_topics = self.determine_help_topics(help_arg)
|
|
#print self.visible_topics
|
|
self.groups = {} # elements are added by .add_group()
|
|
|
|
def preprocess_args(self, args):
|
|
"""Work around some limitations in argparse.
|
|
|
|
Currently, add the default verb "run" as a default.
|
|
"""
|
|
|
|
for token in args:
|
|
if token in VERBS:
|
|
return args
|
|
return ["run"] + args
|
|
|
|
def prescan_for_flag(self, flag, possible_arguments):
|
|
"""Checks cli input for flags.
|
|
|
|
Check for a flag, which accepts a fixed set of possible arguments, in
|
|
the command line; we will use this information to configure argparse's
|
|
help correctly. Return the flag's argument, if it has one that matches
|
|
the sequence @possible_arguments; otherwise return whether the flag is
|
|
present.
|
|
|
|
"""
|
|
if flag not in self.args:
|
|
return False
|
|
pos = self.args.index(flag)
|
|
try:
|
|
nxt = self.args[pos + 1]
|
|
if nxt in possible_arguments:
|
|
return nxt
|
|
except IndexError:
|
|
pass
|
|
return True
|
|
|
|
def add(self, topic, *args, **kwargs):
|
|
"""Add a new command line argument.
|
|
|
|
@topic is required, to indicate which part of the help will document
|
|
it, but can be None for `always documented'.
|
|
|
|
"""
|
|
if self.visible_topics[topic]:
|
|
if topic in self.groups:
|
|
group = self.groups[topic]
|
|
group.add_argument(*args, **kwargs)
|
|
else:
|
|
self.parser.add_argument(*args, **kwargs)
|
|
else:
|
|
kwargs["help"] = argparse.SUPPRESS
|
|
self.parser.add_argument(*args, **kwargs)
|
|
|
|
def add_group(self, topic, **kwargs):
|
|
"""
|
|
|
|
This has to be called once for every topic; but we leave those calls
|
|
next to the argument definitions for clarity. Return something
|
|
arguments can be added to if necessary, either the parser or an argument
|
|
group.
|
|
|
|
"""
|
|
if self.visible_topics[topic]:
|
|
#print "Adding visible group " + topic
|
|
group = self.parser.add_argument_group(topic, **kwargs)
|
|
self.groups[topic] = group
|
|
return group
|
|
else:
|
|
#print "Invisible group " + topic
|
|
return self.silent_parser
|
|
|
|
def add_plugin_args(self, plugins):
|
|
"""
|
|
|
|
Let each of the plugins add its own command line arguments, which
|
|
may or may not be displayed as help topics.
|
|
|
|
"""
|
|
for name, plugin_ep in plugins.iteritems():
|
|
parser_or_group = self.add_group(name, description=plugin_ep.description)
|
|
#print parser_or_group
|
|
plugin_ep.plugin_cls.inject_parser_options(parser_or_group, name)
|
|
|
|
def determine_help_topics(self, chosen_topic):
|
|
"""
|
|
|
|
The user may have requested help on a topic, return a dict of which
|
|
topics to display. @chosen_topic has prescan_for_flag's return type
|
|
|
|
:returns: dict
|
|
|
|
"""
|
|
# topics maps each topic to whether it should be documented by
|
|
# argparse on the command line
|
|
if chosen_topic == "all":
|
|
return dict([(t, True) for t in self.help_topics])
|
|
elif not chosen_topic:
|
|
return dict([(t, False) for t in self.help_topics])
|
|
else:
|
|
return dict([(t, t == chosen_topic) for t in self.help_topics])
|
|
|
|
|
|
def create_parser(plugins, args):
|
|
"""Create parser."""
|
|
helpful = HelpfulArgumentParser(args, plugins)
|
|
|
|
# --help is automatically provided by argparse
|
|
helpful.add(
|
|
None, "-v", "--verbose", dest="verbose_count", action="count",
|
|
default=flag_default("verbose_count"), help="This flag can be used "
|
|
"multiple times to incrementally increase the verbosity of output, "
|
|
"e.g. -vvv.")
|
|
helpful.add(
|
|
None, "-t", "--text", dest="text_mode", action="store_true",
|
|
help="Use the text output instead of the curses UI.")
|
|
helpful.add(None, "-m", "--email", help=config_help("email"))
|
|
# positional arg shadows --domains, instead of appending, and
|
|
# --domains is useful, because it can be stored in config
|
|
#for subparser in parser_run, parser_auth, parser_install:
|
|
# subparser.add_argument("domains", nargs="*", metavar="domain")
|
|
helpful.add(None, "-d", "--domains", metavar="DOMAIN", action="append")
|
|
|
|
helpful.add_group(
|
|
"automation",
|
|
description="Arguments for automating execution & other tweaks")
|
|
helpful.add(
|
|
"automation", "--version", action="version",
|
|
version="%(prog)s {0}".format(letsencrypt.__version__),
|
|
help="show program's version number and exit")
|
|
helpful.add(
|
|
"automation", "--no-confirm", dest="no_confirm", action="store_true",
|
|
help="Turn off confirmation screens, currently used for --revoke")
|
|
helpful.add(
|
|
"automation", "--agree-eula", dest="eula", action="store_true",
|
|
help="Agree to the Let's Encrypt Developer Preview EULA")
|
|
helpful.add(
|
|
"automation", "--agree-tos", dest="tos", action="store_true",
|
|
help="Agree to the Let's Encrypt Subscriber Agreement")
|
|
helpful.add(
|
|
"automation", "--account", metavar="ACCOUNT_ID",
|
|
help="Account ID to use")
|
|
|
|
helpful.add_group(
|
|
"testing", description="The following flags are meant for "
|
|
"testing purposes only! Do NOT change them, unless you "
|
|
"really know what you're doing!")
|
|
helpful.add(
|
|
"testing", "--debug", action="store_true",
|
|
help="Show tracebacks if the program exits abnormally")
|
|
helpful.add(
|
|
"testing", "--no-verify-ssl", action="store_true",
|
|
help=config_help("no_verify_ssl"),
|
|
default=flag_default("no_verify_ssl"))
|
|
helpful.add( # TODO: apache plugin does NOT respect it (#479)
|
|
"testing", "--dvsni-port", type=int, default=flag_default("dvsni_port"),
|
|
help=config_help("dvsni_port"))
|
|
helpful.add("testing", "--simple-http-port", type=int,
|
|
help=config_help("simple_http_port"))
|
|
helpful.add("testing", "--no-simple-http-tls", action="store_true",
|
|
help=config_help("no_simple_http_tls"))
|
|
|
|
helpful.add_group(
|
|
"security", description="Security parameters & server settings")
|
|
helpful.add(
|
|
"security", "-B", "--rsa-key-size", type=int, metavar="N",
|
|
default=flag_default("rsa_key_size"), help=config_help("rsa_key_size"))
|
|
# TODO: resolve - assumes binary logic while client.py assumes ternary.
|
|
helpful.add(
|
|
"security", "-r", "--redirect", action="store_true",
|
|
help="Automatically redirect all HTTP traffic to HTTPS for the newly "
|
|
"authenticated vhost.")
|
|
|
|
_paths_parser(helpful)
|
|
# _plugins_parsing should be the last thing to act upon the main
|
|
# parser (--help should display plugin-specific options last)
|
|
_plugins_parsing(helpful, plugins)
|
|
|
|
_create_subparsers(helpful)
|
|
|
|
return helpful.parser, helpful.args
|
|
|
|
# For now unfortunately this constant just needs to match the code below;
|
|
# there isn't an elegant way to autogenerate it in time.
|
|
VERBS = ["run", "auth", "install", "revoke", "rollback", "config_changes",\
|
|
"plugins"]
|
|
|
|
|
|
def _create_subparsers(helpful):
|
|
subparsers = helpful.parser.add_subparsers(metavar="SUBCOMMAND")
|
|
def add_subparser(name, func): # pylint: disable=missing-docstring
|
|
subparser = subparsers.add_parser(
|
|
name, help=func.__doc__.splitlines()[0], description=func.__doc__)
|
|
subparser.set_defaults(func=func)
|
|
return subparser
|
|
|
|
# the order of add_subparser() calls is important: it defines the
|
|
# order in which subparser names will be displayed in --help
|
|
add_subparser("run", run)
|
|
parser_auth = add_subparser("auth", auth)
|
|
parser_install = add_subparser("install", install)
|
|
parser_revoke = add_subparser("revoke", revoke)
|
|
parser_rollback = add_subparser("rollback", rollback)
|
|
add_subparser("config_changes", config_changes)
|
|
parser_plugins = add_subparser("plugins", plugins_cmd)
|
|
|
|
parser_auth.add_argument(
|
|
"--csr", type=read_file, help="Path to a Certificate Signing "
|
|
"Request (CSR) in DER format.")
|
|
parser_auth.add_argument(
|
|
"--cert-path", default=flag_default("auth_cert_path"),
|
|
help="When using --csr this is where certificate is saved.")
|
|
parser_auth.add_argument(
|
|
"--chain-path", default=flag_default("auth_chain_path"),
|
|
help="When using --csr this is where certificate chain is saved.")
|
|
|
|
parser_install.add_argument(
|
|
"--cert-path", required=True, help="Path to a certificate that "
|
|
"is going to be installed.")
|
|
parser_install.add_argument(
|
|
"--key-path", required=True, help="Accompynying private key")
|
|
parser_install.add_argument(
|
|
"--chain-path", help="Accompanying path to a certificate chain.")
|
|
parser_revoke.add_argument(
|
|
"--cert-path", type=read_file, help="Revoke a specific certificate.")
|
|
parser_revoke.add_argument(
|
|
"--key-path", type=read_file,
|
|
help="Revoke all certs generated by the provided authorized key.")
|
|
|
|
parser_rollback.add_argument(
|
|
"--checkpoints", type=int, metavar="N",
|
|
default=flag_default("rollback_checkpoints"),
|
|
help="Revert configuration N number of checkpoints.")
|
|
|
|
parser_plugins.add_argument(
|
|
"--init", action="store_true", help="Initialize plugins.")
|
|
parser_plugins.add_argument(
|
|
"--prepare", action="store_true",
|
|
help="Initialize and prepare plugins.")
|
|
parser_plugins.add_argument(
|
|
"--authenticators", action="append_const", dest="ifaces",
|
|
const=interfaces.IAuthenticator,
|
|
help="Limit to authenticator plugins only.")
|
|
parser_plugins.add_argument(
|
|
"--installers", action="append_const", dest="ifaces",
|
|
const=interfaces.IInstaller, help="Limit to installer plugins only.")
|
|
|
|
|
|
def _paths_parser(helpful):
|
|
add = helpful.add
|
|
helpful.add_group(
|
|
"paths", description="Arguments changing execution paths & servers")
|
|
add("paths", "--config-dir", default=flag_default("config_dir"),
|
|
help=config_help("config_dir"))
|
|
add("paths", "--work-dir", default=flag_default("work_dir"),
|
|
help=config_help("work_dir"))
|
|
add("paths", "--logs-dir", default=flag_default("logs_dir"),
|
|
help="Logs directory.")
|
|
add("paths", "--server", default=flag_default("server"),
|
|
help=config_help("server"))
|
|
|
|
|
|
def _plugins_parsing(helpful, plugins):
|
|
helpful.add_group(
|
|
"plugins", description="Let's Encrypt client supports an "
|
|
"extensible plugins architecture. See '%(prog)s plugins' for a "
|
|
"list of all available plugins and their names. You can force "
|
|
"a particular plugin by setting options provided below. Futher "
|
|
"down this help message you will find plugin-specific options "
|
|
"(prefixed by --{plugin_name}).")
|
|
helpful.add(
|
|
"plugins", "-a", "--authenticator", help="Authenticator plugin name.")
|
|
helpful.add(
|
|
"plugins", "-i", "--installer", help="Installer plugin name.")
|
|
helpful.add(
|
|
"plugins", "--configurator", help="Name of the plugin that is "
|
|
"both an authenticator and an installer. Should not be used "
|
|
"together with --authenticator or --installer.")
|
|
|
|
# things should not be reorder past/pre this comment:
|
|
# plugins_group should be displayed in --help before plugin
|
|
# specific groups (so that plugins_group.description makes sense)
|
|
|
|
helpful.add_plugin_args(plugins)
|
|
|
|
|
|
def _setup_logging(args):
|
|
level = -args.verbose_count * 10
|
|
fmt = "%(asctime)s:%(levelname)s:%(name)s:%(message)s"
|
|
if args.text_mode:
|
|
handler = logging.StreamHandler()
|
|
handler.setFormatter(logging.Formatter(fmt))
|
|
else:
|
|
handler = log.DialogHandler()
|
|
# dialog box is small, display as less as possible
|
|
handler.setFormatter(logging.Formatter("%(message)s"))
|
|
handler.setLevel(level)
|
|
|
|
# TODO: use fileConfig?
|
|
|
|
# unconditionally log to file for debugging purposes
|
|
# TODO: change before release?
|
|
log_file_name = os.path.join(args.logs_dir, 'letsencrypt.log')
|
|
file_handler = logging.handlers.RotatingFileHandler(
|
|
log_file_name, maxBytes=2**20, backupCount=10)
|
|
# rotate on each invocation, rollover only possible when maxBytes
|
|
# is nonzero and backupCount is nonzero, so we set maxBytes as big
|
|
# as possible not to overrun in single CLI invocation (1MB).
|
|
file_handler.doRollover() # TODO: creates empty letsencrypt.log.1 file
|
|
file_handler.setLevel(logging.DEBUG)
|
|
file_handler_formatter = logging.Formatter(fmt=fmt)
|
|
file_handler_formatter.converter = time.gmtime # don't use localtime
|
|
file_handler.setFormatter(file_handler_formatter)
|
|
|
|
root_logger = logging.getLogger()
|
|
root_logger.setLevel(logging.DEBUG) # send all records to handlers
|
|
root_logger.addHandler(handler)
|
|
root_logger.addHandler(file_handler)
|
|
|
|
logger.debug("Root logging level set at %d", level)
|
|
logger.info("Saving debug log to %s", log_file_name)
|
|
|
|
|
|
def _handle_exception(exc_type, exc_value, trace, args):
|
|
"""Logs exceptions and reports them to the user.
|
|
|
|
Args is used to determine how to display exceptions to the user. In
|
|
general, if args.debug is True, then the full exception and traceback is
|
|
shown to the user, otherwise it is suppressed. If args itself is None,
|
|
then the traceback and exception is attempted to be written to a logfile.
|
|
If this is successful, the traceback is suppressed, otherwise it is shown
|
|
to the user. sys.exit is always called with a nonzero status.
|
|
|
|
"""
|
|
logger.debug(
|
|
"Exiting abnormally:\n%s",
|
|
"".join(traceback.format_exception(exc_type, exc_value, trace)))
|
|
|
|
if issubclass(exc_type, Exception) and (args is None or not args.debug):
|
|
if args is None:
|
|
logfile = "letsencrypt.log"
|
|
try:
|
|
with open(logfile, "w") as logfd:
|
|
traceback.print_exception(
|
|
exc_type, exc_value, trace, file=logfd)
|
|
except: # pylint: disable=bare-except
|
|
sys.exit("".join(
|
|
traceback.format_exception(exc_type, exc_value, trace)))
|
|
|
|
if issubclass(exc_type, errors.Error):
|
|
sys.exit(exc_value)
|
|
elif args is None:
|
|
sys.exit(
|
|
"An unexpected error occurred. Please see the logfile '{0}' "
|
|
"for more details.".format(logfile))
|
|
else:
|
|
sys.exit(
|
|
"An unexpected error occurred. Please see the logfiles in {0} "
|
|
"for more details.".format(args.logs_dir))
|
|
else:
|
|
sys.exit("".join(
|
|
traceback.format_exception(exc_type, exc_value, trace)))
|
|
|
|
|
|
def main(cli_args=sys.argv[1:]):
|
|
"""Command line argument parsing and main script execution."""
|
|
sys.excepthook = functools.partial(_handle_exception, args=None)
|
|
|
|
# note: arg parser internally handles --help (and exits afterwards)
|
|
plugins = plugins_disco.PluginsRegistry.find_all()
|
|
parser,tweaked_cli_args = create_parser(plugins, cli_args)
|
|
args = parser.parse_args(tweaked_cli_args)
|
|
config = configuration.NamespaceConfig(args)
|
|
|
|
# Setup logging ASAP, otherwise "No handlers could be found for
|
|
# logger ..." TODO: this should be done before plugins discovery
|
|
for directory in config.config_dir, config.work_dir:
|
|
le_util.make_or_verify_dir(
|
|
directory, constants.CONFIG_DIRS_MODE, os.geteuid())
|
|
# TODO: logs might contain sensitive data such as contents of the
|
|
# private key! #525
|
|
le_util.make_or_verify_dir(args.logs_dir, 0o700, os.geteuid())
|
|
_setup_logging(args)
|
|
|
|
# do not log `args`, as it contains sensitive data (e.g. revoke --key)!
|
|
logger.debug("Arguments: %r", cli_args)
|
|
logger.debug("Discovered plugins: %r", plugins)
|
|
|
|
sys.excepthook = functools.partial(_handle_exception, args=args)
|
|
|
|
# Displayer
|
|
if args.text_mode:
|
|
displayer = display_util.FileDisplay(sys.stdout)
|
|
else:
|
|
displayer = display_util.NcursesDisplay()
|
|
zope.component.provideUtility(displayer)
|
|
|
|
# Reporter
|
|
report = reporter.Reporter()
|
|
zope.component.provideUtility(report)
|
|
atexit.register(report.atexit_print_messages)
|
|
|
|
# TODO: remove developer EULA prompt for the launch
|
|
if not config.eula:
|
|
eula = pkg_resources.resource_string("letsencrypt", "EULA")
|
|
if not zope.component.getUtility(interfaces.IDisplay).yesno(
|
|
eula, "Agree", "Cancel"):
|
|
raise errors.Error("Must agree to TOS")
|
|
|
|
if not os.geteuid() == 0:
|
|
logger.warning(
|
|
"Root (sudo) is required to run most of letsencrypt functionality.")
|
|
# check must be done after arg parsing as --help should work
|
|
# w/o root; on the other hand, e.g. "letsencrypt run
|
|
# --authenticator dns" or "letsencrypt plugins" does not
|
|
# require root as well
|
|
#return (
|
|
# "{0}Root is required to run letsencrypt. Please use sudo.{0}"
|
|
# .format(os.linesep))
|
|
|
|
return args.func(args, config, plugins)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main()) # pragma: no cover
|