5bef6177ac
Support distributions that ship a full .img (Raspberry Pi OS, moode, RetroPie, Manjaro ARM) rather than a rootfs tarball, by loop-mounting the image and copying its boot + root partitions into a fresh LUKS container. - distributions.yml/catalog.py/choosers.py: raspios catalog (lite64/desktop64/ lite32 via the stable _latest redirects) + choose_raspios. - plan.py: source_url override so a _latest redirect downloads under an .img.xz name that decompress_command recognises. - transfer.py: transfer_disk_image (loop-mount -> repartition -> LUKS -> rsync copy with progress, cp fallback -> fix boot fstab); transfer_image gains interactive= and routes encrypted non-arch images here; download_image gains force_prompt. - loopimg.py: losetup attach/detach/partition helper. - fsutil.drop_fstab_mount + register_encrypted_root: replace a stock image's existing / fstab line instead of colliding with it. - encryption.configure_encryption returns the onion address. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
118 lines
5.1 KiB
Python
118 lines
5.1 KiB
Python
"""initramfs-tools backend (Debian / Raspberry Pi OS).
|
|
|
|
Debian does not use mkinitcpio; the LUKS unlock stack is cryptsetup-initramfs
|
|
(reads /etc/crypttab, option ``initramfs``) + dropbear-initramfs (serves the
|
|
passphrase prompt over SSH). Raspberry Pi OS ships without an initramfs, so we
|
|
enable one in config.txt and regenerate with update-initramfs.
|
|
|
|
NOTE: the exact dropbear-initramfs key path, the boot-partition mount point
|
|
(/boot vs /boot/firmware) and the initramfs enable knob vary across Raspberry
|
|
Pi OS releases; verify on the target release. This path is not exercised by the
|
|
QEMU harness yet.
|
|
"""
|
|
|
|
import re
|
|
from pathlib import Path
|
|
|
|
from lim import config, fsutil, packages, runner, ui
|
|
from lim.image.initramfs import keygen
|
|
from lim.image.initramfs.base import InitramfsBackend
|
|
from lim.image.plan import ImagePlan
|
|
from lim.image.session import ImageSession, chroot_bash, install_packages
|
|
|
|
|
|
class InitramfsToolsBackend(InitramfsBackend):
|
|
def luks_package_collection(self) -> str:
|
|
return "server/luks-debian"
|
|
|
|
def install_authorized_key(self, root: Path, authorized_keys: Path) -> None:
|
|
# dropbear-initramfs (bookworm) reads /etc/dropbear/initramfs/authorized_keys.
|
|
target = root / "etc/dropbear/initramfs/authorized_keys"
|
|
ui.info(f"Adding {authorized_keys} to {target}...")
|
|
runner.run(
|
|
["install", "-D", "-m", "0600", str(authorized_keys), str(target)],
|
|
sudo=True,
|
|
)
|
|
|
|
def install_tor_unlock(self, plan: ImagePlan, root: Path) -> str:
|
|
ui.info("Setting up remote unlock via Tor onion service (initramfs-tools)...")
|
|
install_packages(
|
|
plan.distribution, root, " ".join(packages.get_packages("server/tor-debian"))
|
|
)
|
|
source_dir = config.CONFIGURATION_PATH / "initramfs-tools"
|
|
for source, target, mode in (
|
|
(source_dir / "tor_hook", root / "etc/initramfs-tools/hooks/tor", "0755"),
|
|
(
|
|
source_dir / "tor_premount",
|
|
root / "etc/initramfs-tools/scripts/init-premount/tor",
|
|
"0755",
|
|
),
|
|
(
|
|
source_dir / "tor_bottom",
|
|
root / "etc/initramfs-tools/scripts/init-bottom/tor",
|
|
"0755",
|
|
),
|
|
(source_dir / "torrc", root / keygen.TORRC_STAGING_PATH, "0644"),
|
|
):
|
|
ui.info(f"Installing {target}...")
|
|
runner.run(["install", "-D", "-m", mode, str(source), str(target)], sudo=True)
|
|
onion_address = keygen.generate_onion_keys(root)
|
|
ui.success(f"Onion unlock address: {onion_address}")
|
|
ui.info(f"Unlock later with: torsocks ssh root@{onion_address}")
|
|
return onion_address
|
|
|
|
def register_encrypted_root(self, plan: ImagePlan, session: ImageSession, root: Path) -> None:
|
|
fstab_path = root / "etc/fstab"
|
|
fstab_line = (
|
|
f"/dev/mapper/{session.root_mapper_name} / {plan.root_filesystem}"
|
|
" defaults,noatime 0 1"
|
|
)
|
|
ui.info(f"Configuring {fstab_path}...")
|
|
# A stock image already mounts / by PARTUUID; drop it so the mapper wins.
|
|
fsutil.drop_fstab_mount("/", fstab_path)
|
|
fsutil.ensure_line_in_file(fstab_line, fstab_path)
|
|
|
|
crypttab_path = root / "etc/crypttab"
|
|
# `initramfs` bakes the mapping into the initramfs so cryptsetup-initramfs
|
|
# unlocks the root before the pivot.
|
|
crypttab_line = (
|
|
f"{session.root_mapper_name} UUID={session.root_partition_uuid} none luks,initramfs"
|
|
)
|
|
ui.info(f"Configuring {crypttab_path}...")
|
|
fsutil.ensure_line_in_file(crypttab_line, crypttab_path)
|
|
|
|
def configure_initramfs(self, plan: ImagePlan, root: Path) -> None: # noqa: ARG002
|
|
ui.info("Regenerating the Debian initramfs (update-initramfs)...")
|
|
# Create for all installed kernels when none exists yet (Raspberry Pi OS
|
|
# ships without one), else update the existing one. `-k all` avoids the
|
|
# build host's `uname -r` leaking into the chroot.
|
|
chroot_bash(
|
|
root,
|
|
"update-initramfs -c -k all 2>/dev/null || update-initramfs -u -k all",
|
|
)
|
|
|
|
def configure_bootloader(
|
|
self,
|
|
plan: ImagePlan, # noqa: ARG002
|
|
session: ImageSession,
|
|
root: Path,
|
|
) -> None:
|
|
hostname = (root / "etc/hostname").read_text().strip()
|
|
boot = session.boot_mount_path
|
|
|
|
config_txt = boot / "config.txt"
|
|
if config_txt.is_file():
|
|
ui.info(f"Enabling an initramfs in {config_txt}...")
|
|
fsutil.ensure_line_in_file("auto_initramfs=1", config_txt)
|
|
|
|
cmdline_txt = boot / "cmdline.txt"
|
|
ui.info(f"Configuring {cmdline_txt}...")
|
|
text = cmdline_txt.read_text().strip()
|
|
text = re.sub(r"root=\S+", f"root=/dev/mapper/{session.root_mapper_name}", text)
|
|
if "ip=" not in text:
|
|
text = f"{text} ip=::::{hostname}:eth0:dhcp"
|
|
if "net.ifnames=0" not in text:
|
|
text = f"{text} net.ifnames=0"
|
|
cmdline_txt.write_text(text + "\n")
|
|
ui.info(f"Content of {cmdline_txt}:{cmdline_txt.read_text()}")
|