external/PotatoNV-crossplatform
1
0
Fork 0
mirror of https://github.com/mashed-potatoes/PotatoNV-crossplatform.git synced 2024-11-10 01:02:17 +01:00
Code Issues Projects Releases Packages Wiki Activity

bootloaders: Load bootloader manifests from XMLs

Browse source 
* HisiBootloaders repo uses XML instead of JSON

Change-Id: I54d8dc3d78ef29133304cb631d6710313e768848
Signed-off-by: Woomymy <woomy@woomy.be>
This commit is contained in:
Woomymy 2023-11-01 16:38:40 +01:00 committed by Andrey Smirnoff
parent fddf3f6329
commit 9e9f437c76
2 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
requirements.txt
View file

@ -1,5 +1,4 @@
adb==1.3.0
lxml==4.4.1
pychalk==2.0.1
pyserial==3.4
inquirerpy==0.3.4

18
usrlock/main.py
View file

@ -1,8 +1,8 @@
import sys
import chalk
import argparse
import json
import hashlib
import xml.etree.ElementTree as ET
from os import path
from glob import glob
from . import ui
@ -34,7 +34,7 @@ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLI
'type': 'list',
'name': 'bootloader',
'message': 'Select bootloader:',
'choices': list(map(lambda x: path.split(x)[-1], glob('bootloaders/*')))
'choices': list(map(lambda x: path.basename(path.split(x)[-2]), glob('bootloaders/*/manifest.xml')))
})['bootloader']
if not args.key:
@ -45,7 +45,7 @@ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLI
'validate': lambda val: len(val) == 16 or 'Excepted 16 symbols'
})['key']
args.manifest = "./bootloaders/%s/manifest.json".format(args.bootloader)
args.manifest = f"./bootloaders/{args.bootloader}/manifest.xml"
if len(args.key) != 16:
ui.error("Invalid key length!", critical=True)
@ -62,7 +62,7 @@ def flash_images(data: dict):
for image in data["images"]:
ui.progress(title="Flashing {}".format(image['role']))
flasher.download_from_disk("./bootloaders/{}/{}"
.format(data['name'], image['filename']), int(image['address'], 16))
.format(data['name'], image['path']), int(image['address'], 16))
ui.success("Bootloader uploaded.")
@ -82,8 +82,12 @@ def write_nvme(key: str):
def main():
args = setup()
if not args.skip_bootloader:
with open(args.manifest) as json_file:
data = json.load(json_file)
data["name"] = args.bootloader
xmltree = ET.parse(args.manifest)
bootloader = xmltree.getroot()
data = bootloader.attrib
data["images"] = list(map(lambda img: img.attrib, bootloader.findall('image')))
data["name"] = args.bootloader
flash_images(data)
write_nvme(args.key)