/snap/core24/1587/usr/lib/python3/dist-packages/cloudinit/config
Edit: /snap/core24/1587/usr/lib/python3/dist-packages/cloudinit/config/cc_ubuntu_pro.py (13971B)
# This file is part of cloud-init. See LICENSE file for license information.
"""ubuntu_pro: Configure Ubuntu Pro support services"""
import json
import logging
import re
from typing import Any, List
from urllib.parse import urlparse
from cloudinit import performance, subp, util
from cloudinit.cloud import Cloud
from cloudinit.config import Config
from cloudinit.config.schema import MetaSchema
from cloudinit.settings import PER_INSTANCE
DEPRECATED_KEYS = set(["ubuntu-advantage", "ubuntu_advantage"])
meta: MetaSchema = {
"id": "cc_ubuntu_pro",
"distros": ["ubuntu"],
"frequency": PER_INSTANCE,
"activate_by_schema_keys": ["ubuntu_pro"] + list(DEPRECATED_KEYS),
}
LOG = logging.getLogger(__name__)
REDACTED = "REDACTED"
ERROR_MSG_SHOULD_AUTO_ATTACH = (
"Unable to determine if this is an Ubuntu Pro instance."
" Fallback to normal Pro attach."
)
KNOWN_PRO_CONFIG_PROPS = (
"http_proxy",
"https_proxy",
"global_apt_http_proxy",
"global_apt_https_proxy",
"ua_apt_http_proxy",
"ua_apt_https_proxy",
)
def validate_schema_features(pro_section: dict):
if "features" not in pro_section:
return
# Validate ubuntu_pro.features type
features = pro_section["features"]
if not isinstance(features, dict):
msg = (
f"'ubuntu_pro.features' should be a dict, not a"
f" {type(features).__name__}"
)
LOG.error(msg)
raise RuntimeError(msg)
# Validate ubuntu_pro.features.disable_auto_attach
if "disable_auto_attach" not in features:
return
disable_auto_attach = features["disable_auto_attach"]
if not isinstance(disable_auto_attach, bool):
msg = (
f"'ubuntu_pro.features.disable_auto_attach' should be a bool"
f", not a {type(disable_auto_attach).__name__}"
)
LOG.error(msg)
raise RuntimeError(msg)
def supplemental_schema_validation(pro_config: dict):
"""Validate user-provided ua:config option values.
This function supplements flexible jsonschema validation with specific
value checks to aid in triage of invalid user-provided configuration.
Note: It does not log/raise config values as they could be urls containing
sensitive auth info.
@param pro_config: Dictionary of config value under 'ubuntu_pro'.
@raises: ValueError describing invalid values provided.
"""
errors = []
for key, value in sorted(pro_config.items()):
if key not in KNOWN_PRO_CONFIG_PROPS:
LOG.warning(
"Not validating unknown ubuntu_pro.config.%s property",
key,
)
continue
elif value is None:
# key will be unset. No extra validation needed.
continue
try:
parsed_url = urlparse(value)
if parsed_url.scheme not in ("http", "https"):
errors.append(
f"Expected URL scheme http/https for ua:config:{key}"
)
except (AttributeError, ValueError):
errors.append(f"Expected a URL for ua:config:{key}")
if errors:
raise ValueError(
"Invalid ubuntu_pro configuration:\n{}".format("\n".join(errors))
)
def set_pro_config(pro_config: Any = None):
if pro_config is None:
return
if not isinstance(pro_config, dict):
raise RuntimeError(
f"ubuntu_pro: config should be a dict, not"
f" a {type(pro_config).__name__};"
" skipping enabling config parameters"
)
supplemental_schema_validation(pro_config)
enable_errors = []
for key, value in sorted(pro_config.items()):
redacted_key_value = None
subp_kwargs: dict = {}
if value is None:
LOG.debug("Disabling Pro config for %s", key)
config_cmd = ["pro", "config", "unset", key]
else:
redacted_key_value = f"{key}=REDACTED"
LOG.debug("Enabling Pro config %s", redacted_key_value)
if re.search(r"\s", value):
key_value = f"{key}={re.escape(value)}"
else:
key_value = f"{key}={value}"
config_cmd = ["pro", "config", "set", key_value]
subp_kwargs = {"logstring": config_cmd[:-1] + [redacted_key_value]}
try:
subp.subp(config_cmd, **subp_kwargs)
except subp.ProcessExecutionError as e:
err_msg = str(e)
if redacted_key_value is not None:
err_msg = err_msg.replace(value, REDACTED)
enable_errors.append((key, err_msg))
if enable_errors:
for param, error in enable_errors:
LOG.warning('Failure enabling/disabling "%s":\n%s', param, error)
raise RuntimeError(
"Failure enabling/disabling Ubuntu Pro config(s): {}".format(
", ".join('"{}"'.format(param) for param, _ in enable_errors)
)
)
def configure_pro(token, enable=None):
"""Call ua command line client to attach and/or enable services."""
if enable is None:
enable = []
elif isinstance(enable, str):
LOG.warning(
"ubuntu_pro: enable should be a list, not"
" a string; treating as a single enable"
)
enable = [enable]
elif not isinstance(enable, list):
LOG.warning(
"ubuntu_pro: enable should be a list, not"
" a %s; skipping enabling services",
type(enable).__name__,
)
enable = []
# Perform attach
if enable:
attach_cmd = ["pro", "attach", "--no-auto-enable", token]
else:
attach_cmd = ["pro", "attach", token]
redacted_cmd = attach_cmd[:-1] + [REDACTED]
LOG.debug("Attaching to Ubuntu Pro. %s", " ".join(redacted_cmd))
try:
# Allow `ua attach` to fail in already attached machines
subp.subp(attach_cmd, rcs={0, 2}, logstring=redacted_cmd)
except subp.ProcessExecutionError as e:
er = str(e).replace(token, REDACTED)
msg = f"Failure attaching Ubuntu Pro:\n{er}"
util.logexc(LOG, msg)
raise RuntimeError(msg) from e
# Enable services
if not enable:
return
cmd = ["pro", "enable", "--assume-yes", "--format", "json"] + enable
try:
enable_stdout, _ = subp.subp(cmd, capture=True, rcs={0, 1})
except subp.ProcessExecutionError as e:
raise RuntimeError(
"Error while enabling service(s): " + ", ".join(enable)
) from e
try:
enable_resp = json.loads(enable_stdout)
except json.JSONDecodeError as e:
raise RuntimeError(
f"Pro response was not json: {enable_stdout}"
) from e
# At this point we were able to load the json response from Pro. This
# response contains a list of errors under the key 'errors'. E.g.
#
# {
# "errors": [
# {
# "message": "UA Apps: ESM is already enabled ...",
# "message_code": "service-already-enabled",
# "service": "esm-apps",
# "type": "service"
# },
# {
# "message": "Cannot enable unknown service 'asdf' ...",
# "message_code": "invalid-service-or-failure",
# "service": null,
# "type": "system"
# }
# ]
# }
#
# From our pov there are two type of errors, service and non-service
# related. We can distinguish them by checking if `service` is non-null
# or null respectively.
enable_errors: List[dict] = []
for err in enable_resp.get("errors", []):
if err["message_code"] == "service-already-enabled":
LOG.debug("Service `%s` already enabled.", err["service"])
continue
enable_errors.append(err)
if enable_errors:
error_services: List[str] = []
for err in enable_errors:
service = err.get("service")
if service is not None:
error_services.append(service)
msg = f'Failure enabling `{service}`: {err["message"]}'
else:
msg = f'Failure of type `{err["type"]}`: {err["message"]}'
util.logexc(LOG, msg)
raise RuntimeError(
"Failure enabling Ubuntu Pro service(s): "
+ ", ".join(error_services)
)
def maybe_install_ua_tools(cloud: Cloud):
"""Install ubuntu-advantage-tools if not present."""
if subp.which("pro"):
return
try:
cloud.distro.update_package_sources()
except Exception:
util.logexc(LOG, "Package update failed")
raise
try:
cloud.distro.install_packages(["ubuntu-advantage-tools"])
except Exception:
util.logexc(LOG, "Failed to install ubuntu-advantage-tools")
raise
def _should_auto_attach(pro_section: dict) -> bool:
disable_auto_attach = bool(
pro_section.get("features", {}).get("disable_auto_attach", False)
)
if disable_auto_attach:
return False
# pylint: disable=import-error
from uaclient.api.exceptions import UserFacingError
from uaclient.api.u.pro.attach.auto.should_auto_attach.v1 import (
should_auto_attach,
)
# pylint: enable=import-error
try:
with performance.Timed(
"Checking if the instance can be attached to Ubuntu Pro",
log_mode="always",
):
result = should_auto_attach()
except UserFacingError as ex:
LOG.debug("Error during `should_auto_attach`: %s", ex)
LOG.warning(ERROR_MSG_SHOULD_AUTO_ATTACH)
return False
return result.should_auto_attach
def _attach(pro_section: dict):
token = pro_section.get("token")
if not token:
msg = "`ubuntu_pro.token` required in non-Pro Ubuntu instances."
LOG.error(msg)
raise RuntimeError(msg)
enable_beta = pro_section.get("enable_beta")
if enable_beta:
LOG.debug(
"Ignoring `ubuntu_pro.enable_beta` services in Pro attach: %s",
", ".join(enable_beta),
)
configure_pro(token=token, enable=pro_section.get("enable"))
def _auto_attach(pro_section: dict):
# pylint: disable=import-error
from uaclient.api.exceptions import AlreadyAttachedError, UserFacingError
from uaclient.api.u.pro.attach.auto.full_auto_attach.v1 import (
FullAutoAttachOptions,
full_auto_attach,
)
# pylint: enable=import-error
enable = pro_section.get("enable")
enable_beta = pro_section.get("enable_beta")
options = FullAutoAttachOptions(
enable=enable,
enable_beta=enable_beta,
)
try:
with performance.Timed("Attaching to Ubuntu Pro", log_mode="always"):
full_auto_attach(options=options)
except AlreadyAttachedError:
if enable_beta is not None or enable is not None:
# Only warn if the user defined some service to enable/disable.
LOG.warning(
"The instance is already attached to Pro. Leaving enabled"
" services untouched. Ignoring config directives"
" ubuntu_pro: enable and enable_beta"
)
except UserFacingError as ex:
msg = f"Error during `full_auto_attach`: {ex.msg}"
LOG.error(msg)
raise RuntimeError(msg) from ex
def handle(name: str, cfg: Config, cloud: Cloud, args: list) -> None:
pro_section = None
deprecated = list(DEPRECATED_KEYS.intersection(cfg))
if deprecated:
if len(deprecated) > 1:
raise RuntimeError(
"Unable to configure Ubuntu Pro. Multiple deprecated config"
" keys provided: %s" % ", ".join(deprecated)
)
LOG.warning(
"Deprecated configuration key(s) provided: %s."
' Expected "ubuntu_pro"; will attempt to continue.',
", ".join(deprecated),
)
pro_section = cfg[deprecated[0]]
if "ubuntu_pro" in cfg:
# Prefer ubuntu_pro over any deprecated keys when both exist
if deprecated:
LOG.warning(
"Ignoring deprecated key %s and preferring ubuntu_pro config",
deprecated[0],
)
pro_section = cfg["ubuntu_pro"]
if pro_section is None:
LOG.debug(
"Skipping module named %s, no 'ubuntu_pro' configuration found",
name,
)
return
elif not isinstance(pro_section, dict):
msg = (
f"'ubuntu_pro' should be a dict, not a"
f" {type(pro_section).__name__}"
)
LOG.error(msg)
raise RuntimeError(msg)
if "commands" in pro_section:
msg = (
'Deprecated configuration "ubuntu-advantage: commands" provided.'
' Expected "token"'
)
LOG.error(msg)
raise RuntimeError(msg)
maybe_install_ua_tools(cloud)
set_pro_config(pro_section.get("config"))
# ua-auto-attach.service had noop-ed as pro_section is not empty
validate_schema_features(pro_section)
LOG.debug(
"To discover more log info, please check /var/log/ubuntu-advantage.log"
)
if _should_auto_attach(pro_section):
_auto_attach(pro_section)
# If ua-auto-attach.service did noop, we did not auto-attach and more keys
# than `features` are given under `ubuntu_pro`, then try to attach.
# This supports the cases:
#
# 1) Previous attach behavior on non-pro instances.
# 2) Previous attach behavior on instances where ubuntu-advantage-tools
# is < v28.0 (Pro apis for should_auto-attach and auto-attach are not
# available.
# 3) The user wants to disable auto-attach and attach by giving:
# `{"ubuntu_pro": "features": {"disable_auto_attach": True}}`
elif not pro_section.keys() <= {"features"}:
_attach(pro_section)