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.
95 lines
4.1 KiB
Python
95 lines
4.1 KiB
Python
# Copyright 2020 Oliver Smith
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
import logging
|
|
import os
|
|
|
|
import pmb.build
|
|
import pmb.chroot.apk
|
|
import pmb.config
|
|
import pmb.flasher
|
|
import pmb.helpers.file
|
|
|
|
|
|
def odin(args, flavor, folder):
|
|
"""
|
|
Create Odin flashable tar file with kernel and initramfs for devices configured with
|
|
the flasher method 'heimdall-isorec' and with boot.img for devices with 'heimdall-bootimg'
|
|
"""
|
|
pmb.flasher.init(args)
|
|
suffix = "rootfs_" + args.device
|
|
|
|
# Validate method
|
|
method = args.deviceinfo["flash_method"]
|
|
if not method.startswith("heimdall-"):
|
|
raise RuntimeError("An odin flashable tar is not supported for the flash"
|
|
" method '" + method + "' specified in the current configuration."
|
|
" Only 'heimdall' methods are supported.")
|
|
|
|
# Partitions
|
|
partition_kernel = args.deviceinfo["flash_heimdall_partition_kernel"] or "KERNEL"
|
|
partition_initfs = args.deviceinfo["flash_heimdall_partition_initfs"] or "RECOVERY"
|
|
|
|
# Temporary folder
|
|
temp_folder = "/tmp/odin-flashable-tar"
|
|
if os.path.exists(args.work + "/chroot_native" + temp_folder):
|
|
pmb.chroot.root(args, ["rm", "-rf", temp_folder])
|
|
|
|
# Odin flashable tar generation script (because redirecting stdin/stdout is not allowed
|
|
# in pmbootstrap's chroot/shell functions for security reasons)
|
|
with open(args.work + "/chroot_rootfs_" + args.device + "/tmp/_odin.sh", "w") as handle:
|
|
odin_kernel_md5 = partition_kernel + ".bin.md5"
|
|
odin_initfs_md5 = partition_initfs + ".bin.md5"
|
|
odin_device_tar = args.device + ".tar"
|
|
odin_device_tar_md5 = args.device + ".tar.md5"
|
|
|
|
handle.write(
|
|
"#!/bin/sh\n"
|
|
"cd " + temp_folder + "\n")
|
|
if method == "heimdall-isorec":
|
|
handle.write(
|
|
# Kernel: copy and append md5
|
|
"cp /boot/vmlinuz-" + flavor + " " + odin_kernel_md5 + "\n"
|
|
"md5sum -t " + odin_kernel_md5 + " >> " + odin_kernel_md5 + "\n"
|
|
# Initramfs: recompress with lzop, append md5
|
|
"gunzip -c /boot/initramfs-" + flavor + " | lzop > " + odin_initfs_md5 + "\n"
|
|
"md5sum -t " + odin_initfs_md5 + " >> " + odin_initfs_md5 + "\n")
|
|
elif method == "heimdall-bootimg":
|
|
handle.write(
|
|
# boot.img: copy and append md5
|
|
"cp /boot/boot.img-" + flavor + " " + odin_kernel_md5 + "\n"
|
|
"md5sum -t " + odin_kernel_md5 + " >> " + odin_kernel_md5 + "\n")
|
|
handle.write(
|
|
# Create tar, remove included files and append md5
|
|
"tar -c -f " + odin_device_tar + " *.bin.md5\n"
|
|
"rm *.bin.md5\n"
|
|
"md5sum -t " + odin_device_tar + " >> " + odin_device_tar + "\n"
|
|
"mv " + odin_device_tar + " " + odin_device_tar_md5 + "\n")
|
|
|
|
commands = [["mkdir", "-p", temp_folder],
|
|
["cat", "/tmp/_odin.sh"], # for the log
|
|
["sh", "/tmp/_odin.sh"],
|
|
["rm", "/tmp/_odin.sh"]
|
|
]
|
|
for command in commands:
|
|
pmb.chroot.root(args, command, suffix)
|
|
|
|
# Move Odin flashable tar to native chroot and cleanup temp folder
|
|
pmb.chroot.user(args, ["mkdir", "-p", "/home/pmos/rootfs"])
|
|
pmb.chroot.root(args, ["mv", "/mnt/rootfs_" + args.device + temp_folder +
|
|
"/" + odin_device_tar_md5, "/home/pmos/rootfs/"]),
|
|
pmb.chroot.root(args, ["chown", "pmos:pmos",
|
|
"/home/pmos/rootfs/" + odin_device_tar_md5])
|
|
pmb.chroot.root(args, ["rmdir", temp_folder], suffix)
|
|
|
|
# Create the symlink
|
|
file = args.work + "/chroot_native/home/pmos/rootfs/" + odin_device_tar_md5
|
|
link = folder + "/" + odin_device_tar_md5
|
|
pmb.helpers.file.symlink(args, file, link)
|
|
|
|
# Display a readable message
|
|
msg = " * " + odin_device_tar_md5
|
|
if method == "heimdall-isorec":
|
|
msg += " (Odin flashable file, contains initramfs and kernel)"
|
|
elif method == "heimdall-bootimg":
|
|
msg += " (Odin flashable file, contains boot.img)"
|
|
logging.info(msg)
|