forked from Mirror/pmbootstrap
f21c216a26
While at it, also remove unnecessary "#!/usr/bin/env python3" in files that only get imported, and adjust other empty/comment lines in the beginnings of the files for consistency. This makes files easier to read, and makes the pmbootstrap codebase more consistent with the build.postmarketos.org codebase.
126 lines
5.2 KiB
Python
126 lines
5.2 KiB
Python
# Copyright 2020 Oliver Smith
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
import logging
|
|
import os
|
|
import glob
|
|
import pmb.helpers.mount
|
|
import pmb.install.losetup
|
|
import pmb.helpers.cli
|
|
import pmb.config
|
|
|
|
|
|
def previous_install(args):
|
|
"""
|
|
Search the sdcard for possible existence of a previous installation of pmOS.
|
|
We temporarily mount the possible pmOS_boot partition as /dev/sdcardp1 inside
|
|
the native chroot to check the label from there.
|
|
"""
|
|
label = ""
|
|
for blockdevice_outside in [args.sdcard + "1", args.sdcard + "p1"]:
|
|
if not os.path.exists(blockdevice_outside):
|
|
continue
|
|
blockdevice_inside = "/dev/sdcardp1"
|
|
pmb.helpers.mount.bind_file(args, blockdevice_outside,
|
|
args.work + "/chroot_native" + blockdevice_inside)
|
|
label = pmb.chroot.root(args, ["blkid", "-s", "LABEL", "-o", "value",
|
|
blockdevice_inside], output_return=True)
|
|
pmb.helpers.run.root(args, ["umount", args.work + "/chroot_native" + blockdevice_inside])
|
|
return "pmOS_boot" in label
|
|
|
|
|
|
def mount_sdcard(args):
|
|
# Sanity checks
|
|
if args.deviceinfo["external_storage"] != "true":
|
|
raise RuntimeError("According to the deviceinfo, this device does"
|
|
" not support a sdcard installation.")
|
|
if not os.path.exists(args.sdcard):
|
|
raise RuntimeError("The sdcard device does not exist: " +
|
|
args.sdcard)
|
|
for path in glob.glob(args.sdcard + "*"):
|
|
if pmb.helpers.mount.ismount(path):
|
|
raise RuntimeError(path + " is mounted! We will not attempt"
|
|
" to format this!")
|
|
logging.info("(native) mount /dev/install (host: " + args.sdcard + ")")
|
|
pmb.helpers.mount.bind_file(args, args.sdcard,
|
|
args.work + "/chroot_native/dev/install")
|
|
if previous_install(args):
|
|
if not pmb.helpers.cli.confirm(args, "WARNING: This device has a"
|
|
" previous installation of pmOS."
|
|
" CONTINUE?"):
|
|
raise RuntimeError("Aborted.")
|
|
else:
|
|
if not pmb.helpers.cli.confirm(args, "EVERYTHING ON " + args.sdcard +
|
|
" WILL BE ERASED! CONTINUE?"):
|
|
raise RuntimeError("Aborted.")
|
|
|
|
|
|
def create_and_mount_image(args, size_boot, size_root):
|
|
"""
|
|
Create a new image file, and mount it as /dev/install.
|
|
|
|
:param size_boot: size of the boot partition in bytes
|
|
:param size_root: size of the root partition in bytes
|
|
"""
|
|
# Short variables for paths
|
|
chroot = args.work + "/chroot_native"
|
|
img_path_prefix = "/home/pmos/rootfs/" + args.device
|
|
img_path_full = img_path_prefix + ".img"
|
|
img_path_boot = img_path_prefix + "-boot.img"
|
|
img_path_root = img_path_prefix + "-root.img"
|
|
|
|
# Umount and delete existing images
|
|
for img_path in [img_path_full, img_path_boot, img_path_root]:
|
|
outside = chroot + img_path
|
|
if os.path.exists(outside):
|
|
pmb.helpers.mount.umount_all(args, chroot + "/mnt")
|
|
pmb.install.losetup.umount(args, img_path)
|
|
pmb.chroot.root(args, ["rm", img_path])
|
|
|
|
# Make sure there is enough free space
|
|
size_mb = round((size_boot + size_root) / (1024**2))
|
|
disk_data = os.statvfs(args.work)
|
|
free = round((disk_data.f_bsize * disk_data.f_bavail) / (1024**2))
|
|
if size_mb > free:
|
|
raise RuntimeError("Not enough free space to create rootfs image! (free: " + str(free) + "M, required: " + str(size_mb) + "M)")
|
|
|
|
# Create empty image files
|
|
pmb.chroot.user(args, ["mkdir", "-p", "/home/pmos/rootfs"])
|
|
size_mb_full = str(size_mb) + "M"
|
|
size_mb_boot = str(round(size_boot / (1024**2))) + "M"
|
|
size_mb_root = str(round(size_root / (1024**2))) + "M"
|
|
images = {img_path_full: size_mb_full}
|
|
if args.split:
|
|
images = {img_path_boot: size_mb_boot,
|
|
img_path_root: size_mb_root}
|
|
for img_path, size_mb in images.items():
|
|
logging.info("(native) create " + os.path.basename(img_path) + " (" + size_mb + ")")
|
|
pmb.chroot.root(args, ["truncate", "-s", size_mb, img_path])
|
|
|
|
# Mount to /dev/install
|
|
mount_image_paths = {img_path_full: "/dev/install"}
|
|
if args.split:
|
|
mount_image_paths = {img_path_boot: "/dev/installp1",
|
|
img_path_root: "/dev/installp2"}
|
|
|
|
for img_path, mount_point in mount_image_paths.items():
|
|
logging.info("(native) mount " + mount_point +
|
|
" (" + os.path.basename(img_path) + ")")
|
|
pmb.install.losetup.mount(args, img_path)
|
|
device = pmb.install.losetup.device_by_back_file(args, img_path)
|
|
pmb.helpers.mount.bind_file(args, device,
|
|
args.work + "/chroot_native" + mount_point)
|
|
|
|
|
|
def create(args, size_boot, size_root):
|
|
"""
|
|
Create /dev/install (the "install blockdevice").
|
|
|
|
:param size_boot: size of the boot partition in bytes
|
|
:param size_root: size of the root partition in bytes
|
|
"""
|
|
pmb.helpers.mount.umount_all(
|
|
args, args.work + "/chroot_native/dev/install")
|
|
if args.sdcard:
|
|
mount_sdcard(args)
|
|
else:
|
|
create_and_mount_image(args, size_boot, size_root)
|