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.
78 lines
3.5 KiB
Python
78 lines
3.5 KiB
Python
# Copyright 2020 Oliver Smith
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
import os
|
|
import logging
|
|
import pmb
|
|
|
|
|
|
def is_dtb(path):
|
|
if not os.path.isfile(path):
|
|
return False
|
|
with open(path, 'rb') as f:
|
|
# Check FDT magic identifier (0xd00dfeed)
|
|
return f.read(4) == b'\xd0\x0d\xfe\xed'
|
|
|
|
|
|
def bootimg(args, path):
|
|
if not os.path.exists(path):
|
|
raise RuntimeError("Could not find file '" + path + "'")
|
|
|
|
logging.info("NOTE: You will be prompted for your sudo password, so we can set"
|
|
" up a chroot to extract and analyze your boot.img file")
|
|
pmb.chroot.apk.install(args, ["file", "unpackbootimg"])
|
|
|
|
temp_path = pmb.chroot.other.tempfolder(args, "/tmp/bootimg_parser")
|
|
bootimg_path = args.work + "/chroot_native" + temp_path + "/boot.img"
|
|
|
|
# Copy the boot.img into the chroot temporary folder
|
|
pmb.helpers.run.root(args, ["cp", path, bootimg_path])
|
|
|
|
file_output = pmb.chroot.user(args, ["file", "-b", "boot.img"],
|
|
working_dir=temp_path,
|
|
output_return=True).rstrip()
|
|
if "android bootimg" not in file_output.lower():
|
|
if "force" in args and args.force:
|
|
logging.warning("WARNING: boot.img file seems to be invalid, but"
|
|
" proceeding anyway (-f specified)")
|
|
else:
|
|
logging.info("NOTE: If you are sure that your file is a valid"
|
|
" boot.img file, you could force the analysis"
|
|
" with: 'pmbootstrap bootimg_analyze " + path +
|
|
" -f'")
|
|
if ("linux kernel" in file_output.lower() or
|
|
"ARM OpenFirmware FORTH Dictionary" in file_output):
|
|
raise RuntimeError("File is a Kernel image, you might need the"
|
|
" 'heimdall-isorec' flash method. See also:"
|
|
" <https://wiki.postmarketos.org/wiki/"
|
|
"Deviceinfo_flash_methods>")
|
|
else:
|
|
raise RuntimeError("File is not an Android boot.img. (" +
|
|
file_output + ")")
|
|
|
|
# Extract all the files
|
|
pmb.chroot.user(args, ["unpackbootimg", "-i", "boot.img"], working_dir=temp_path)
|
|
|
|
output = {}
|
|
# Get base, offsets, pagesize, cmdline and qcdt info
|
|
with open(bootimg_path + "-base", 'r') as f:
|
|
output["base"] = ("0x%08x" % int(f.read().replace('\n', ''), 16))
|
|
with open(bootimg_path + "-kerneloff", 'r') as f:
|
|
output["kernel_offset"] = ("0x%08x" % int(f.read().replace('\n', ''), 16))
|
|
with open(bootimg_path + "-ramdiskoff", 'r') as f:
|
|
output["ramdisk_offset"] = ("0x%08x" % int(f.read().replace('\n', ''), 16))
|
|
with open(bootimg_path + "-secondoff", 'r') as f:
|
|
output["second_offset"] = ("0x%08x" % int(f.read().replace('\n', ''), 16))
|
|
with open(bootimg_path + "-tagsoff", 'r') as f:
|
|
output["tags_offset"] = ("0x%08x" % int(f.read().replace('\n', ''), 16))
|
|
with open(bootimg_path + "-pagesize", 'r') as f:
|
|
output["pagesize"] = f.read().replace('\n', '')
|
|
with open(bootimg_path + "-cmdline", 'r') as f:
|
|
output["cmdline"] = f.read().replace('\n', '')
|
|
output["qcdt"] = ("true" if os.path.isfile(bootimg_path + "-dt") and
|
|
os.path.getsize(bootimg_path + "-dt") > 0 else "false")
|
|
output["dtb_second"] = ("true" if is_dtb(bootimg_path + "-second") else "false")
|
|
|
|
# Cleanup
|
|
pmb.chroot.root(args, ["rm", "-r", temp_path])
|
|
|
|
return output
|