mirror of
https://github.com/platomav/BIOSUtilities.git
synced 2025-05-13 14:44:46 -04:00
6de50c422f
Unified extracted output directory naming logic Multiple code fixes, refactors and improvements
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
|
#coding=utf-8
|
|
|
|
"""
|
|
Toshiba COM Extract
|
|
Toshiba BIOS COM Extractor
|
|
Copyright (C) 2018-2022 Plato Mavropoulos
|
|
"""
|
|
|
|
TITLE = 'Toshiba BIOS COM Extractor v2.0_a4'
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
|
|
# Stop __pycache__ generation
|
|
sys.dont_write_bytecode = True
|
|
|
|
from common.externals import get_comextract_path
|
|
from common.path_ops import make_dirs, path_stem, path_suffixes
|
|
from common.patterns import PAT_TOSHIBA_COM
|
|
from common.system import printer
|
|
from common.templates import BIOSUtility
|
|
from common.text_ops import file_to_bytes
|
|
|
|
# Check if input is Toshiba BIOS COM image
|
|
def is_toshiba_com(in_file):
|
|
buffer = file_to_bytes(in_file)
|
|
|
|
is_ext = path_suffixes(in_file)[-1].upper() == '.COM' if os.path.isfile(in_file) else True
|
|
|
|
is_com = PAT_TOSHIBA_COM.search(buffer)
|
|
|
|
return is_ext and is_com
|
|
|
|
# Parse & Extract Toshiba BIOS COM image
|
|
def toshiba_com_extract(input_file, extract_path, padding=0):
|
|
if not os.path.isfile(input_file):
|
|
printer('Error: Could not find input file path!', padding)
|
|
|
|
return 1
|
|
|
|
make_dirs(extract_path, delete=True)
|
|
|
|
output_name = path_stem(input_file)
|
|
output_file = os.path.join(extract_path, f'{output_name}.bin')
|
|
|
|
try:
|
|
subprocess.run([get_comextract_path(), input_file, output_file], check=True, stdout=subprocess.DEVNULL)
|
|
|
|
if not os.path.isfile(output_file):
|
|
raise Exception('EXTRACT_FILE_MISSING')
|
|
except Exception:
|
|
printer(f'Error: ToshibaComExtractor could not extract file {input_file}!', padding)
|
|
|
|
return 2
|
|
|
|
printer(f'Succesfull {output_name} extraction via ToshibaComExtractor!', padding)
|
|
|
|
return 0
|
|
|
|
if __name__ == '__main__':
|
|
BIOSUtility(TITLE, is_toshiba_com, toshiba_com_extract).run_utility()
|