forked from Mirror/pmbootstrap
866e5bcfab
Move pmb/parse/arch.py over to core and refactor it as an Arch type, similar to how Chroot was done. Fix all the uses (that I can find) of arch in the codebase that need adjusting. The new Arch type is an Enum, making it clear what architectures can be represented and making it much easier to reason about. Since we support ~5 (kinda) different representations of an Architecture (Alpine, Kernel, target triple, platform, and QEMU), we now formalise that the Alpine format is what we represent internally, with methods to convert to any of the others as-needed. Signed-off-by: Caleb Connolly <caleb@postmarketos.org>
105 lines
4 KiB
Python
105 lines
4 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
|
|
import pmb.install.losetup
|
|
from pmb.core import Chroot, ChrootType, get_context
|
|
|
|
|
|
def kill_adb():
|
|
"""
|
|
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():
|
|
"""
|
|
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):
|
|
"""
|
|
: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=False):
|
|
# 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")
|
|
|
|
# Umount all losetup mounted images
|
|
if pmb.helpers.mount.ismount(chroot / "dev/loop-control"):
|
|
for path_outside in (chroot / "/home/pmos/rootfs").glob("*.img"):
|
|
path = path_outside.relative_to(chroot.path)
|
|
pmb.install.losetup.umount(path)
|
|
|
|
# Umount device rootfs and installer chroots
|
|
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)
|
|
|
|
# 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])
|
|
|
|
if not only_install_related:
|
|
for path in Chroot.glob():
|
|
if path.exists():
|
|
pmb.helpers.mount.umount_all(path)
|
|
# 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")
|