#!/usr/bin/env python3 # -*- coding: utf-8 -*- # # === This file is part of Calamares - === # # SPDX-FileCopyrightText: 2014 Aurélien Gâteau # SPDX-FileCopyrightText: 2017 Alf Gaida # SPDX-FileCopyrightText: 2019 Adriaan de Groot # SPDX-FileCopyrightText: 2019 Kevin Kofler # SPDX-FileCopyrightText: 2019-2020 Collabora Ltd # SPDX-License-Identifier: GPL-3.0-or-later # # Calamares is Free Software: see the License-Identifier above. # import tempfile import subprocess import os import libcalamares import gettext _ = gettext.translation("calamares-python", localedir=libcalamares.utils.gettext_path(), languages=libcalamares.utils.gettext_languages(), fallback=True).gettext def pretty_name(): return _("Mounting partitions.") def get_btrfs_subvolumes(partitions): """ Gets the job-configuration for btrfs subvolumes, or if there is none given, returns a default configuration that matches the setup (/ and /home) from before configurability was introduced. @param partitions The partitions (from the partitioning module) that will exist on disk. This is used to filter out subvolumes that don't need to be created because they get a dedicated partition instead. """ btrfs_subvolumes = libcalamares.job.configuration.get("btrfsSubvolumes", None) # Warn if there's no configuration at all, and empty configurations are # replaced by a simple root-only layout. if btrfs_subvolumes is None: libcalamares.utils.warning("No configuration for btrfsSubvolumes") if not btrfs_subvolumes: btrfs_subvolumes = [ dict(mountPoint="/", subvolume="/@"), dict(mountPoint="/home", subvolume="/@home") ] # Filter out the subvolumes which have a dedicated partition non_root_partition_mounts = [ m for m in [ p.get("mountPoint", None) for p in partitions ] if m is not None and m != '/' ] btrfs_subvolumes = list(filter(lambda s : s["mountPoint"] not in non_root_partition_mounts, btrfs_subvolumes)) # If we have a swap **file**, give it a separate subvolume. swap_choice = libcalamares.globalstorage.value( "partitionChoices" ) if swap_choice and swap_choice.get( "swap", None ) == "file": btrfs_subvolumes.append({'mountPoint': '/swap', 'subvolume': '/@swap'}) return btrfs_subvolumes def parse_global_storage(gs_value): """ Something in the chain is converting on and off to true and false. This converts it back. :param gs_value: The value from global storage which needs to be fixed :return: """ if gs_value is True: return "on" elif gs_value is False: return "off" else: return gs_value def mount_zfs(root_mount_point, partition): """ Mounts a zfs partition at @p root_mount_point :param root_mount_point: The absolute path to the root of the install :param partition: The partition map from global storage for this partition :return: """ # Get the list of zpools from global storage zfs_pool_list = libcalamares.globalstorage.value("zfsPoolInfo") if not zfs_pool_list: libcalamares.utils.warning("Failed to locate zfsPoolInfo data in global storage") raise Exception("Internal error mounting zfs datasets") # Find the zpool matching this partition for zfs_pool in zfs_pool_list: if zfs_pool["mountpoint"] == partition["mountPoint"]: pool_name = zfs_pool["poolName"] ds_name = zfs_pool["dsName"] # import the zpool import_result = subprocess.run(['zpool', 'import', '-R', root_mount_point, pool_name]) if import_result.returncode != 0: raise Exception("Failed to import zpool") # Get the encrpytion information from global storage zfs_info_list = libcalamares.globalstorage.value("zfsInfo") encrypt = False if zfs_info_list: for zfs_info in zfs_info_list: if zfs_info["mountpoint"] == partition["mountPoint"] and zfs_info["encrypted"] is True: encrypt = True passphrase = zfs_info["passphrase"] if encrypt is True: # The zpool is encrypted, we need to unlock it loadkey_result = subprocess.run(['sh', '-c', 'echo "' + passphrase + '" | zfs load-key ' + pool_name]) if loadkey_result.returncode != 0: raise Exception("Failed to unlock zpool") if partition["mountPoint"] == '/': # Get the zfs dataset list from global storage zfs = libcalamares.globalstorage.value("zfsDatasets") if not zfs: libcalamares.utils.warning("Failed to locate zfs dataset list") raise Exception("Internal error mounting zfs datasets") # first we handle the / dataset if there is one for dataset in zfs: if dataset['mountpoint'] == '/': # Properly set the canmount field from global storage can_mount = parse_global_storage(dataset['canMount']) set_result = subprocess.run(['zfs', 'set', 'canmount=' + can_mount, dataset['zpool'] + '/' + dataset['dsName']]) if set_result.returncode != 0: raise Exception("Failed to set zfs mountpoint") if dataset['canMount'] == 'noauto': mount_result = subprocess.run(['zfs', 'mount', dataset['zpool'] + '/' + dataset['dsName']]) if mount_result.returncode != 0: raise Exception("Failed to mount root dataset") # Set the canmount property for each dataset. This will effectively mount the dataset for dataset in zfs: # We already handled the / mountpoint above if dataset['mountpoint'] != '/': can_mount = parse_global_storage(dataset['canMount']) set_result = subprocess.run(['zfs', 'set', 'canmount=' + can_mount, dataset['zpool'] + '/' + dataset['dsName']]) if set_result.returncode != 0: raise Exception("Failed to set zfs mountpoint") else: set_result = subprocess.run(['zfs', 'set', 'canmount=on', pool_name + '/' + ds_name]) if set_result.returncode != 0: raise Exception("Failed to set zfs mountpoint") def mount_partition(root_mount_point, partition, partitions): """ Do a single mount of @p partition inside @p root_mount_point. """ # Create mount point with `+` rather than `os.path.join()` because # `partition["mountPoint"]` starts with a '/'. raw_mount_point = partition["mountPoint"] if not raw_mount_point: return mount_point = root_mount_point + raw_mount_point # Ensure that the created directory has the correct SELinux context on # SELinux-enabled systems. os.makedirs(mount_point, exist_ok=True) try: subprocess.call(['chcon', '--reference=' + raw_mount_point, mount_point]) except FileNotFoundError as e: libcalamares.utils.warning(str(e)) except OSError: libcalamares.utils.error("Cannot run 'chcon' normally.") raise fstype = partition.get("fs", "").lower() if fstype == "unformatted": return if fstype == "fat16" or fstype == "fat32": fstype = "vfat" device = partition["device"] if "luksMapperName" in partition: device = os.path.join("/dev/mapper", partition["luksMapperName"]) if fstype == "zfs": mount_zfs(root_mount_point, partition) else: # fstype == "zfs" if libcalamares.utils.mount(device, mount_point, fstype, partition.get("options", "")) != 0: libcalamares.utils.warning("Cannot mount {}".format(device)) # Special handling for btrfs subvolumes. Create the subvolumes listed in mount.conf if fstype == "btrfs" and partition["mountPoint"] == '/': # Root has been mounted to btrfs volume -> create subvolumes from configuration btrfs_subvolumes = get_btrfs_subvolumes(partitions) # Store created list in global storage so it can be used in the fstab module libcalamares.globalstorage.insert("btrfsSubvolumes", btrfs_subvolumes) # Create the subvolumes that are in the completed list for s in btrfs_subvolumes: subprocess.check_call(['btrfs', 'subvolume', 'create', root_mount_point + s['subvolume']]) subprocess.check_call(["umount", "-v", root_mount_point]) device = partition["device"] if "luksMapperName" in partition: device = os.path.join("/dev/mapper", partition["luksMapperName"]) # Mount the subvolumes for s in btrfs_subvolumes: mount_option = "subvol={}".format(s['subvolume']) subvolume_mountpoint = mount_point[:-1] + s['mountPoint'] if libcalamares.utils.mount(device, subvolume_mountpoint, fstype, ",".join([mount_option, partition.get("options", "")])) != 0: libcalamares.utils.warning("Cannot mount {}".format(device)) def run(): """ Mount all the partitions from GlobalStorage and from the job configuration. Partitions are mounted in-lexical-order of their mountPoint. """ partitions = libcalamares.globalstorage.value("partitions") if not partitions: libcalamares.utils.warning("partitions is empty, {!s}".format(partitions)) return (_("Configuration Error"), _("No partitions are defined for 
{!s}
to use." ).format("mount")) root_mount_point = tempfile.mkdtemp(prefix="calamares-root-") # Guard against missing keys (generally a sign that the config file is bad) extra_mounts = libcalamares.job.configuration.get("extraMounts") or [] extra_mounts_efi = libcalamares.job.configuration.get("extraMountsEfi") or [] if not extra_mounts and not extra_mounts_efi: libcalamares.utils.warning("No extra mounts defined. Does mount.conf exist?") if libcalamares.globalstorage.value("firmwareType") == "efi": extra_mounts.extend(extra_mounts_efi) # Add extra mounts to the partitions list and sort by mount points. # This way, we ensure / is mounted before the rest, and every mount point # is created on the right partition (e.g. if a partition is to be mounted # under /tmp, we make sure /tmp is mounted before the partition) mountable_partitions = [ p for p in partitions + extra_mounts if "mountPoint" in p and p["mountPoint"] ] mountable_partitions.sort(key=lambda x: x["mountPoint"]) for partition in mountable_partitions: mount_partition(root_mount_point, partition, partitions) libcalamares.globalstorage.insert("rootMountPoint", root_mount_point) # Remember the extra mounts for the unpackfs module libcalamares.globalstorage.insert("extraMounts", extra_mounts)