mirror of
https://gitlab.postmarketos.org/postmarketOS/pmbootstrap.git
synced 2025-07-15 04:15:13 +03:00
18d912d53d
Refactor the install code to stop using loop devices and instead create and manipulate a disk image directly. Both ext4 and vfat have mechanisms for formatting and populating partitions at an offset inside an image, other filesystems likely do as well but so far have not been implemented or tested. With this "pmbootstrap install" works for standard EFI disk images (e.g. QEMU, X64 or trailblazer) entirely rootless. Since the creation of the disk images happens in the same user namespace as everything else, the resulting disk images have correct ownership and permissions even though from the host perspective they are all subuids. This gets image building working properly *for the default case*. We can now build disk images! In particular, we can build disk images with a 4k sector size even on a host with a 512 byte sector size (or block size in the filesystem). This is surprisingly hard for some reason since not all libfdisk tools have the right flags. Thankfully sfdisk does. In addition, we now generate UUIDs ourselves, to break the loop between generating fstab and running mkfs (since we also populate the disk image /with/ mkfs, we need to already know the UUID when we run it...). Signed-off-by: Casey Connolly <kcxt@postmarketos.org>
99 lines
3.6 KiB
Python
99 lines
3.6 KiB
Python
# Copyright 2023 Oliver Smith
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
from pmb.core.arch import Arch
|
|
from pmb.helpers import logging
|
|
import socket
|
|
from contextlib import closing
|
|
|
|
import pmb.chroot
|
|
import pmb.helpers.mount
|
|
from pmb.core import Chroot, ChrootType
|
|
from pmb.core.context import get_context
|
|
|
|
|
|
def kill_adb() -> None:
|
|
"""
|
|
Kill adb daemon if it's running.
|
|
"""
|
|
port = 5038
|
|
with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as sock:
|
|
if sock.connect_ex(("127.0.0.1", port)) == 0:
|
|
pmb.chroot.root(["adb", "-P", str(port), "kill-server"])
|
|
|
|
|
|
def kill_sccache() -> None:
|
|
"""
|
|
Kill sccache daemon if it's running. Unlike ccache it automatically spawns
|
|
a daemon when you call it and exits after some time of inactivity.
|
|
"""
|
|
port = 4226
|
|
with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as sock:
|
|
if sock.connect_ex(("127.0.0.1", port)) == 0:
|
|
pmb.chroot.root(["sccache", "--stop-server"])
|
|
|
|
|
|
def shutdown_cryptsetup_device(name: str) -> None:
|
|
"""
|
|
:param name: cryptsetup device name, usually "pm_crypt" in pmbootstrap
|
|
"""
|
|
if not (Chroot.native() / "dev/mapper" / name).exists():
|
|
return
|
|
pmb.chroot.apk.install(["cryptsetup"], Chroot.native())
|
|
status = pmb.chroot.root(["cryptsetup", "status", name], output_return=True, check=False)
|
|
if not status:
|
|
logging.warning(
|
|
"WARNING: Failed to run cryptsetup to get the status"
|
|
" for " + name + ", assuming it is not mounted"
|
|
" (shutdown fails later if it is)!"
|
|
)
|
|
return
|
|
|
|
if status.startswith("/dev/mapper/" + name + " is active."):
|
|
pmb.chroot.root(["cryptsetup", "luksClose", name])
|
|
elif status.startswith("/dev/mapper/" + name + " is inactive."):
|
|
# When "cryptsetup status" fails, the device is not mounted and we
|
|
# have a left over file (#83)
|
|
pmb.chroot.root(["rm", "/dev/mapper/" + name])
|
|
else:
|
|
raise RuntimeError("Failed to parse 'cryptsetup status' output!")
|
|
|
|
|
|
def shutdown(only_install_related: bool = False) -> None:
|
|
# Stop daemons
|
|
kill_adb()
|
|
kill_sccache()
|
|
|
|
chroot = Chroot.native()
|
|
|
|
# Umount installation-related paths (order is important!)
|
|
pmb.helpers.mount.umount_all(chroot / "mnt/install")
|
|
shutdown_cryptsetup_device("pm_crypt")
|
|
|
|
# Remove "in-pmbootstrap" marker from all chroots. This marker indicates
|
|
# that pmbootstrap has set up all mount points etc. to run programs inside
|
|
# the chroots, but we want it gone afterwards (e.g. when the chroot
|
|
# contents get copied to a rootfs / installer image, or if creating an
|
|
# android recovery zip from its contents).
|
|
for marker in get_context().config.work.glob("chroot_*/in-pmbootstrap"):
|
|
pmb.helpers.run.root(["rm", marker])
|
|
|
|
# Umount device rootfs and installer chroots
|
|
if only_install_related:
|
|
for chroot_type in [ChrootType.ROOTFS, ChrootType.INSTALLER]:
|
|
chroot = Chroot(chroot_type, get_context().config.device)
|
|
if chroot.path.exists():
|
|
pmb.helpers.mount.umount_all(chroot.path)
|
|
return
|
|
|
|
# Umount all folders inside work dir
|
|
# The folders are explicitly iterated over, so folders symlinked inside
|
|
# work dir get umounted as well (used in test_pkgrel_bump.py, #1595)
|
|
for path in get_context().config.work.glob("*"):
|
|
pmb.helpers.mount.umount_all(path)
|
|
|
|
# Clean up the rest
|
|
for arch in Arch.supported():
|
|
if arch.cpu_emulation_required():
|
|
pmb.chroot.binfmt.unregister(arch)
|
|
|
|
logging.debug("Shutdown complete")
|