mirror of
https://github.com/platomav/BIOSUtilities.git
synced 2025-05-15 15:44:53 -04:00
f895fc208c
Added graceful exception hanlding during "main" flow Improved and cleaned 7-Zip and EFI compression logic Improved too aggressive extraction directory handling Fixed input name detection at VAIO Package Extractor Fixed Intel IBIOSI detection at Apple EFI Identifier
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
#!/usr/bin/env python3 -B
|
|
# coding=utf-8
|
|
|
|
"""
|
|
Toshiba COM Extract
|
|
Toshiba BIOS COM Extractor
|
|
Copyright (C) 2018-2024 Plato Mavropoulos
|
|
"""
|
|
|
|
import os
|
|
import subprocess
|
|
|
|
from biosutilities.common.externals import comextract_path
|
|
from biosutilities.common.paths import delete_file, is_file, make_dirs, path_stem
|
|
from biosutilities.common.patterns import PAT_TOSHIBA_COM
|
|
from biosutilities.common.system import printer
|
|
from biosutilities.common.templates import BIOSUtility
|
|
|
|
|
|
class ToshibaComExtract(BIOSUtility):
|
|
""" Toshiba BIOS COM Extractor """
|
|
|
|
TITLE: str = 'Toshiba BIOS COM Extractor'
|
|
|
|
def check_format(self) -> bool:
|
|
""" Check if input is Toshiba BIOS COM image """
|
|
|
|
return bool(PAT_TOSHIBA_COM.search(self.input_buffer, 0, 0x100))
|
|
|
|
def parse_format(self) -> bool:
|
|
""" Parse & Extract Toshiba BIOS COM image """
|
|
|
|
make_dirs(in_path=self.extract_path)
|
|
|
|
if isinstance(self.input_object, str) and is_file(in_path=self.input_object):
|
|
input_path: str = self.input_object
|
|
else:
|
|
input_path = os.path.join(self.extract_path, 'toshiba_bios.com')
|
|
|
|
with open(input_path, 'wb') as input_buffer:
|
|
input_buffer.write(self.input_buffer)
|
|
|
|
output_path: str = os.path.join(self.extract_path, f'{path_stem(in_path=input_path)}_extracted.bin')
|
|
|
|
comextract_res: subprocess.CompletedProcess[bytes] = subprocess.run(
|
|
[comextract_path(), input_path, output_path], check=False, stdout=subprocess.DEVNULL)
|
|
|
|
if input_path != self.input_object:
|
|
delete_file(in_path=input_path)
|
|
|
|
if comextract_res.returncode == 0 and is_file(in_path=output_path):
|
|
printer(message='Successful extraction via ToshibaComExtractor!', padding=self.padding)
|
|
|
|
return True
|
|
|
|
return False
|