mirror of
https://github.com/platomav/BIOSUtilities.git
synced 2024-09-16 13:58:52 -04:00
d85a7f82dc
Added AMI PFAT nested detection at each file Added Award BIOS payload naming at each file Switched Panasonic BIOS LZNT1 external library Improved Panasonic LZNT1 detection and length Improved Dell PFS code structure and fixed bugs Improved code exception handling (raise, catch) Improved code definitions (PEP8, docs, types) Fixed some arguments missing from help screens
84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
#!/usr/bin/env python3 -B
|
|
# coding=utf-8
|
|
|
|
"""
|
|
Award BIOS Extract
|
|
Award BIOS Module Extractor
|
|
Copyright (C) 2018-2024 Plato Mavropoulos
|
|
"""
|
|
|
|
import os
|
|
import stat
|
|
|
|
from common.comp_szip import szip_decompress
|
|
from common.path_ops import get_extract_path, make_dirs, safe_name
|
|
from common.patterns import PAT_AWARD_LZH
|
|
from common.system import printer
|
|
from common.templates import BIOSUtility
|
|
from common.text_ops import file_to_bytes
|
|
|
|
TITLE = 'Award BIOS Module Extractor v3.0'
|
|
|
|
|
|
def is_award_bios(in_file):
|
|
""" Check if input is Award BIOS image """
|
|
|
|
in_buffer = file_to_bytes(in_file)
|
|
|
|
return bool(PAT_AWARD_LZH.search(in_buffer))
|
|
|
|
|
|
def award_bios_extract(input_file, extract_path, padding=0):
|
|
""" Parse & Extract Award BIOS image """
|
|
|
|
input_buffer = file_to_bytes(input_file)
|
|
|
|
make_dirs(extract_path, delete=True)
|
|
|
|
for lzh_match in PAT_AWARD_LZH.finditer(input_buffer):
|
|
lzh_type = lzh_match.group(0).decode('utf-8')
|
|
|
|
lzh_text = f'LZH-{lzh_type.strip("-").upper()}'
|
|
|
|
lzh_bgn = lzh_match.start()
|
|
|
|
mod_bgn = lzh_bgn - 0x2
|
|
hdr_len = input_buffer[mod_bgn]
|
|
mod_len = int.from_bytes(input_buffer[mod_bgn + 0x7:mod_bgn + 0xB], 'little')
|
|
mod_end = lzh_bgn + hdr_len + mod_len
|
|
|
|
mod_bin = input_buffer[mod_bgn:mod_end]
|
|
|
|
if len(mod_bin) != 0x2 + hdr_len + mod_len:
|
|
printer(f'Error: Skipped incomplete LZH stream at 0x{mod_bgn:X}!', padding, False)
|
|
|
|
continue
|
|
|
|
tag_txt = safe_name(mod_bin[0x16:0x16 + mod_bin[0x15]].decode('utf-8', 'ignore').strip())
|
|
|
|
printer(f'{lzh_text} > {tag_txt} [0x{mod_bgn:06X}-0x{mod_end:06X}]', padding)
|
|
|
|
mod_path = os.path.join(extract_path, tag_txt)
|
|
|
|
lzh_path = f'{mod_path}.lzh'
|
|
|
|
with open(lzh_path, 'wb') as lzh_file:
|
|
lzh_file.write(mod_bin) # Store LZH archive
|
|
|
|
# 7-Zip returns critical exit code (i.e. 2) if LZH CRC is wrong, do not check result
|
|
szip_decompress(lzh_path, extract_path, lzh_text, padding + 4, check=False)
|
|
|
|
# Manually check if 7-Zip extracted LZH due to its CRC check issue
|
|
if os.path.isfile(mod_path):
|
|
os.chmod(lzh_path, stat.S_IWRITE)
|
|
|
|
os.remove(lzh_path) # Successful extraction, delete LZH archive
|
|
|
|
# Extract any nested LZH archives
|
|
if is_award_bios(mod_path):
|
|
# Recursively extract nested Award BIOS modules
|
|
award_bios_extract(mod_path, get_extract_path(mod_path), padding + 8)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
BIOSUtility(title=TITLE, check=is_award_bios, main=award_bios_extract).run_utility()
|