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.
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
# Copyright 2020 Oliver Smith
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
import glob
|
|
import json
|
|
import logging
|
|
import os
|
|
import time
|
|
|
|
import pmb.helpers.mount
|
|
import pmb.helpers.run
|
|
import pmb.chroot
|
|
|
|
|
|
def init(args):
|
|
if not os.path.isdir("/sys/module/loop"):
|
|
pmb.helpers.run.root(args, ["modprobe", "loop"])
|
|
loopdevices = [loopdev for loopdev in glob.glob("/dev/loop*") if not os.path.isdir(loopdev)]
|
|
for loopdev in loopdevices:
|
|
pmb.helpers.mount.bind_file(args, loopdev,
|
|
args.work + "/chroot_native/" + loopdev)
|
|
|
|
|
|
def mount(args, img_path):
|
|
"""
|
|
:param img_path: Path to the img file inside native chroot.
|
|
"""
|
|
logging.debug("(native) mount " + img_path + " (loop)")
|
|
|
|
# Try to mount multiple times (let the kernel module initialize #1594)
|
|
for i in range(0, 5):
|
|
# Retry
|
|
if i > 0:
|
|
logging.debug("loop module might not be initialized yet, retry in"
|
|
" one second...")
|
|
time.sleep(1)
|
|
|
|
# Mount and return on success
|
|
init(args)
|
|
|
|
losetup_cmd = ["losetup", "-f", img_path]
|
|
sector_size = args.deviceinfo["rootfs_image_sector_size"]
|
|
if sector_size:
|
|
losetup_cmd += ["-b", str(int(sector_size))]
|
|
|
|
pmb.chroot.root(args, losetup_cmd, check=False)
|
|
if device_by_back_file(args, img_path):
|
|
return
|
|
|
|
# Failure: raise exception
|
|
raise RuntimeError("Failed to mount loop device: " + img_path)
|
|
|
|
|
|
def device_by_back_file(args, back_file):
|
|
"""
|
|
Get the /dev/loopX device, that points to a specific image file.
|
|
"""
|
|
|
|
# Get list from losetup
|
|
losetup_output = pmb.chroot.root(args, ["losetup", "--json",
|
|
"--list"], output_return=True)
|
|
if not losetup_output:
|
|
return None
|
|
|
|
# Find the back_file
|
|
losetup = json.loads(losetup_output)
|
|
for loopdevice in losetup["loopdevices"]:
|
|
if loopdevice["back-file"] == back_file:
|
|
return loopdevice["name"]
|
|
return None
|
|
|
|
|
|
def umount(args, img_path):
|
|
"""
|
|
:param img_path: Path to the img file inside native chroot.
|
|
"""
|
|
device = device_by_back_file(args, img_path)
|
|
if not device:
|
|
return
|
|
logging.debug("(native) umount " + device)
|
|
pmb.chroot.root(args, ["losetup", "-d", device])
|