Gecko [ theprintave.ca ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2024-09-18 19:49
schemas	[ DIR ]	drwxr-xr-x	2024-09-18 19:49
__init__.py	40 B	-rw-r--r--	2024-07-03 01:59
cc_ansible.py	7.66 KB	-rw-r--r--	2024-07-03 01:59
cc_apk_configure.py	4.25 KB	-rw-r--r--	2024-07-03 01:59
cc_apt_configure.py	38.07 KB	-rw-r--r--	2024-07-03 01:59
cc_apt_pipelining.py	1.86 KB	-rw-r--r--	2024-07-03 01:59
cc_bootcmd.py	1.67 KB	-rw-r--r--	2024-07-03 01:59
cc_byobu.py	2.66 KB	-rw-r--r--	2024-07-03 01:59
cc_ca_certs.py	8.35 KB	-rw-r--r--	2024-07-03 01:59
cc_chef.py	12.2 KB	-rw-r--r--	2024-07-03 01:59
cc_disable_ec2_metadata.py	1.62 KB	-rw-r--r--	2024-07-03 01:59
cc_disk_setup.py	29.99 KB	-rw-r--r--	2024-07-03 01:59
cc_fan.py	1.97 KB	-rw-r--r--	2024-07-03 01:59
cc_final_message.py	2.49 KB	-rw-r--r--	2024-07-03 01:59
cc_growpart.py	18.39 KB	-rw-r--r--	2024-07-03 01:59
cc_grub_dpkg.py	5.44 KB	-rw-r--r--	2024-07-03 01:59
cc_install_hotplug.py	3.21 KB	-rw-r--r--	2024-07-03 01:59
cc_keyboard.py	1.51 KB	-rw-r--r--	2024-07-03 01:59
cc_keys_to_console.py	2.04 KB	-rw-r--r--	2024-07-03 01:59
cc_landscape.py	3.12 KB	-rw-r--r--	2024-07-03 01:59
cc_locale.py	1.21 KB	-rw-r--r--	2024-07-03 01:59
cc_lxd.py	13.49 KB	-rw-r--r--	2024-07-11 21:27
cc_mcollective.py	4.12 KB	-rw-r--r--	2024-07-03 01:59
cc_mounts.py	16.35 KB	-rw-r--r--	2024-07-03 01:59
cc_ntp.py	18.74 KB	-rw-r--r--	2024-07-03 01:59
cc_package_update_upgrade_install.py	3.8 KB	-rw-r--r--	2024-07-03 01:59
cc_phone_home.py	3.75 KB	-rw-r--r--	2024-07-03 01:59
cc_power_state_change.py	5.94 KB	-rw-r--r--	2024-07-03 01:59
cc_puppet.py	10.73 KB	-rw-r--r--	2024-07-03 01:59
cc_reset_rmc.py	4.47 KB	-rw-r--r--	2024-07-03 01:59
cc_resizefs.py	10.82 KB	-rw-r--r--	2024-07-03 01:59
cc_resolv_conf.py	3.15 KB	-rw-r--r--	2024-07-03 01:59
cc_rh_subscription.py	15.23 KB	-rw-r--r--	2024-07-03 01:59
cc_rsyslog.py	12.7 KB	-rw-r--r--	2024-07-03 01:59
cc_runcmd.py	1.62 KB	-rw-r--r--	2024-07-03 01:59
cc_salt_minion.py	3.97 KB	-rw-r--r--	2024-07-03 01:59
cc_scripts_per_boot.py	1.28 KB	-rw-r--r--	2024-07-03 01:59
cc_scripts_per_instance.py	1.28 KB	-rw-r--r--	2024-07-03 01:59
cc_scripts_per_once.py	1.26 KB	-rw-r--r--	2024-07-03 01:59
cc_scripts_user.py	1.27 KB	-rw-r--r--	2024-07-03 01:59
cc_scripts_vendor.py	1.28 KB	-rw-r--r--	2024-07-03 01:59
cc_seed_random.py	3.01 KB	-rw-r--r--	2024-07-03 01:59
cc_set_hostname.py	3.42 KB	-rw-r--r--	2024-07-03 01:59
cc_set_passwords.py	8.68 KB	-rw-r--r--	2024-07-03 01:59
cc_snap.py	3.34 KB	-rw-r--r--	2024-07-11 21:27
cc_spacewalk.py	2.65 KB	-rw-r--r--	2024-07-03 01:59
cc_ssh.py	11 KB	-rw-r--r--	2024-07-03 01:59
cc_ssh_authkey_fingerprints.py	3.73 KB	-rw-r--r--	2024-07-03 01:59
cc_ssh_import_id.py	5.45 KB	-rw-r--r--	2024-07-03 01:59
cc_timezone.py	1.21 KB	-rw-r--r--	2024-07-03 01:59
cc_ubuntu_autoinstall.py	4.5 KB	-rw-r--r--	2024-07-11 21:27
cc_ubuntu_drivers.py	4.09 KB	-rw-r--r--	2024-07-03 01:59
cc_ubuntu_pro.py	13.74 KB	-rw-r--r--	2024-07-03 01:59
cc_update_etc_hosts.py	2.46 KB	-rw-r--r--	2024-07-03 01:59
cc_update_hostname.py	2.19 KB	-rw-r--r--	2024-07-03 01:59
cc_users_groups.py	2.88 KB	-rw-r--r--	2024-07-03 01:59
cc_wireguard.py	6.68 KB	-rw-r--r--	2024-07-03 01:59
cc_write_files.py	3.99 KB	-rw-r--r--	2024-07-03 01:59
cc_write_files_deferred.py	1.23 KB	-rw-r--r--	2024-07-03 01:59
cc_yum_add_repo.py	4.91 KB	-rw-r--r--	2024-07-03 01:59
cc_zypper_add_repo.py	4.95 KB	-rw-r--r--	2024-07-03 01:59
modules.py	12.96 KB	-rw-r--r--	2024-07-03 01:59
schema.py	66.38 KB	-rw-r--r--	2024-07-03 01:59

Rename

# 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

MODULE_DESCRIPTION = """\
"""
meta: MetaSchema = {
    "id": "cc_grub_dpkg",
    "distros": ["ubuntu", "debian"],
    "frequency": PER_INSTANCE,
    "activate_by_schema_keys": [],
}  # type: ignore

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", True)
    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)
        )