189 lines
8.0 KiB
Python
189 lines
8.0 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# === This file is part of Calamares - <https://calamares.io> ===
|
|
#
|
|
# SPDX-FileCopyrightText: 2014 Aurélien Gâteau <agateau@kde.org>
|
|
# SPDX-FileCopyrightText: 2017 Alf Gaida <agaida@siduction.org>
|
|
# SPDX-FileCopyrightText: 2019 Adriaan de Groot <groot@kde.org>
|
|
# SPDX-FileCopyrightText: 2019 Kevin Kofler <kevin.kofler@chello.at>
|
|
# 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 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 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))
|
|
|
|
if fstype == "zfs" and partition["mountPoint"] == '/':
|
|
# Get the zfs dataset list from global storage
|
|
zfs = libcalamares.globalstorage.value("zfs")
|
|
|
|
if not zfs:
|
|
libcalamares.utils.warning("Failed to locate zfs dataset list")
|
|
|
|
# Set the canmount property for each dataset. This will effectively mount the dataset
|
|
for dataset in zfs:
|
|
if dataset['canmount'] == 'noauto' or dataset['canmount'] == 'on':
|
|
subprocess.check_call(['zfs', 'set', 'canmount=' + dataset['canmount'],
|
|
dataset['zpool'] + '/' + dataset['dsName']])
|
|
|
|
# It is common for the / mountpoint to be set to noauto since it is mounted by the initrd
|
|
# If this is the case we need to manually mount it here
|
|
if dataset['mountpoint'] == '/' and dataset['canmount'] == 'noauto':
|
|
subprocess.check_call(['zfs', 'mount', dataset['zpool'] + '/' + dataset['dsName']])
|
|
|
|
|
|
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 <pre>{!s}</pre> 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)
|