Mr.X

# Copyright (C) 2009-2010, 2020 Canonical Ltd.
# Copyright (C) 2012 Hewlett-Packard Development Company, L.P.
#
# Author: Scott Moser <scott.moser@canonical.com>
# Author: Juerg Haefliger <juerg.haefliger@hp.com>
# Author: Matthew Ruffell <matthew.ruffell@canonical.com>
#
# This file is part of cloud-init. See LICENSE file for license information.
"""Grub Dpkg: Configure grub debconf installation device"""

import logging
import os

from cloudinit import subp, util
from cloudinit.cloud import Cloud
from cloudinit.config import Config
from cloudinit.config.schema import MetaSchema
from cloudinit.settings import PER_INSTANCE
from cloudinit.subp import ProcessExecutionError

meta: MetaSchema = {
    "id": "cc_grub_dpkg",
    "distros": ["ubuntu", "debian"],
    "frequency": PER_INSTANCE,
    "activate_by_schema_keys": ["grub_dpkg", "grub-dpkg"],
}

LOG = logging.getLogger(__name__)


def fetch_idevs():
    """
    Fetches the /dev/disk/by-id device grub is installed to.
    Falls back to plain disk name if no by-id entry is present.
    """
    disk = ""
    devices = []

    # BIOS mode systems use /boot and the disk path,
    # EFI mode systems use /boot/efi and the partition path.
    probe_target = "disk"
    probe_mount = "/boot"
    if is_efi_booted():
        probe_target = "device"
        probe_mount = "/boot/efi"

    try:
        # get the root disk where the /boot directory resides.
        disk = subp.subp(
            ["grub-probe", "-t", probe_target, probe_mount], capture=True
        ).stdout.strip()
    except ProcessExecutionError as e:
        # grub-common may not be installed, especially on containers
        # FileNotFoundError is a nested exception of ProcessExecutionError
        if isinstance(e.reason, FileNotFoundError):
            LOG.debug("'grub-probe' not found in $PATH")
        # disks from the container host are present in /proc and /sys
        # which is where grub-probe determines where /boot is.
        # it then checks for existence in /dev, which fails as host disks
        # are not exposed to the container.
        elif "failed to get canonical path" in e.stderr:
            LOG.debug("grub-probe 'failed to get canonical path'")
        else:
            # something bad has happened, continue to log the error
            raise
    except Exception:
        util.logexc(LOG, "grub-probe failed to execute for grub_dpkg")

    if not disk or not os.path.exists(disk):
        # If we failed to detect a disk, we can return early
        return ""

    try:
        # check if disk exists and use udevadm to fetch symlinks
        devices = (
            subp.subp(
                ["udevadm", "info", "--root", "--query=symlink", disk],
                capture=True,
            )
            .stdout.strip()
            .split()
        )
    except Exception:
        util.logexc(
            LOG, "udevadm DEVLINKS symlink query failed for disk='%s'", disk
        )

    LOG.debug("considering these device symlinks: %s", ",".join(devices))
    # filter symlinks for /dev/disk/by-id entries
    devices = [dev for dev in devices if "disk/by-id" in dev]
    LOG.debug("filtered to these disk/by-id symlinks: %s", ",".join(devices))
    # select first device if there is one, else fall back to plain name
    idevs = sorted(devices)[0] if devices else disk
    LOG.debug("selected %s", idevs)

    return idevs


def is_efi_booted() -> bool:
    """
    Check if the system is booted in EFI mode.
    """
    try:
        return os.path.exists("/sys/firmware/efi")
    except OSError as e:
        LOG.error("Failed to determine if system is booted in EFI mode: %s", e)
        # If we can't determine if we're booted in EFI mode, assume we're not.
        return False


def handle(name: str, cfg: Config, cloud: Cloud, args: list) -> None:
    mycfg = cfg.get("grub_dpkg", cfg.get("grub-dpkg", {}))
    if not mycfg:
        mycfg = {}

    enabled = mycfg.get("enabled", False)
    if util.is_false(enabled):
        LOG.debug("%s disabled by config grub_dpkg/enabled=%s", name, enabled)
        return

    dconf_sel = get_debconf_config(mycfg)
    LOG.debug("Setting grub debconf-set-selections with '%s'", dconf_sel)

    try:
        subp.subp(["debconf-set-selections"], data=dconf_sel)
    except Exception as e:
        util.logexc(
            LOG, "Failed to run debconf-set-selections for grub_dpkg: %s", e
        )


def get_debconf_config(mycfg: Config) -> str:
    """
    Returns the debconf config for grub-pc or
    grub-efi depending on the systems boot mode.
    """
    if is_efi_booted():
        idevs = util.get_cfg_option_str(
            mycfg, "grub-efi/install_devices", None
        )

        if idevs is None:
            idevs = fetch_idevs()

        return "grub-pc grub-efi/install_devices string %s\n" % idevs
    else:
        idevs = util.get_cfg_option_str(mycfg, "grub-pc/install_devices", None)
        if idevs is None:
            idevs = fetch_idevs()

        idevs_empty = mycfg.get("grub-pc/install_devices_empty")
        if idevs_empty is None:
            idevs_empty = not idevs
        elif not isinstance(idevs_empty, bool):
            idevs_empty = util.translate_bool(idevs_empty)
        idevs_empty = str(idevs_empty).lower()

        # now idevs and idevs_empty are set to determined values
        # or, those set by user
        return (
            "grub-pc grub-pc/install_devices string %s\n"
            "grub-pc grub-pc/install_devices_empty boolean %s\n"
            % (idevs, idevs_empty)
        )

Name	Type	Size	Permission
__pycache__	Folder		0755
schemas	Folder		0755
__init__.py	File	40 B	0644
cc_ansible.py	File	7.68 KB	0644
cc_apk_configure.py	File	4.23 KB	0644
cc_apt_configure.py	File	38.08 KB	0644
cc_apt_pipelining.py	File	1.85 KB	0644
cc_bootcmd.py	File	1.65 KB	0644
cc_byobu.py	File	2.73 KB	0644
cc_ca_certs.py	File	8.67 KB	0644
cc_chef.py	File	12.19 KB	0644
cc_disable_ec2_metadata.py	File	1.6 KB	0644
cc_disk_setup.py	File	29.61 KB	0644
cc_fan.py	File	1.95 KB	0644
cc_final_message.py	File	2.51 KB	0644
cc_growpart.py	File	18.37 KB	0644
cc_grub_dpkg.py	File	5.42 KB	0644
cc_install_hotplug.py	File	3.19 KB	0644
cc_keyboard.py	File	1.49 KB	0644
cc_keys_to_console.py	File	2.09 KB	0644
cc_landscape.py	File	3.06 KB	0644
cc_locale.py	File	1.19 KB	0644
cc_lxd.py	File	13.47 KB	0644
cc_mcollective.py	File	4.11 KB	0644
cc_mounts.py	File	18.16 KB	0644
cc_ntp.py	File	18.97 KB	0644
cc_package_update_upgrade_install.py	File	3.79 KB	0644
cc_phone_home.py	File	3.75 KB	0644
cc_power_state_change.py	File	5.92 KB	0644
cc_puppet.py	File	10.72 KB	0644
cc_reset_rmc.py	File	4.33 KB	0644
cc_resizefs.py	File	10.63 KB	0644
cc_resolv_conf.py	File	3.17 KB	0644
cc_rh_subscription.py	File	15.21 KB	0644
cc_rsyslog.py	File	11.55 KB	0644
cc_runcmd.py	File	1.6 KB	0644
cc_salt_minion.py	File	3.96 KB	0644
cc_scripts_per_boot.py	File	1.27 KB	0644
cc_scripts_per_instance.py	File	1.26 KB	0644
cc_scripts_per_once.py	File	1.24 KB	0644
cc_scripts_user.py	File	1.25 KB	0644
cc_scripts_vendor.py	File	1.26 KB	0644
cc_seed_random.py	File	3 KB	0644
cc_set_hostname.py	File	3.4 KB	0644
cc_set_passwords.py	File	10.32 KB	0644
cc_snap.py	File	3.36 KB	0644
cc_spacewalk.py	File	2.64 KB	0644
cc_ssh.py	File	11 KB	0644
cc_ssh_authkey_fingerprints.py	File	3.76 KB	0644
cc_ssh_import_id.py	File	5.43 KB	0644
cc_timezone.py	File	1.14 KB	0644
cc_ubuntu_autoinstall.py	File	2.99 KB	0644
cc_ubuntu_drivers.py	File	4.08 KB	0644
cc_ubuntu_pro.py	File	13.65 KB	0644
cc_update_etc_hosts.py	File	2.46 KB	0644
cc_update_hostname.py	File	2.18 KB	0644
cc_users_groups.py	File	2.86 KB	0644
cc_wireguard.py	File	6.67 KB	0644
cc_write_files.py	File	6 KB	0644
cc_write_files_deferred.py	File	1.28 KB	0644
cc_yum_add_repo.py	File	4.53 KB	0644
cc_zypper_add_repo.py	File	4.93 KB	0644
modules.py	File	13.23 KB	0644
schema.py	File	50.26 KB	0644

Upload:

Command:

Filemanager

Server Info

System Info