mirror of
https://github.com/platomav/BIOSUtilities.git
synced 2025-05-09 13:52:00 -04:00
35564f31b7
Re-written public attributes of Apple EFI ID Improved memory consumption for all utilities Adjusted README with consolidated requirements
62 lines
2.1 KiB
Python
62 lines
2.1 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_read, 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 """
|
|
|
|
if isinstance(self.input_object, str) and is_file_read(in_path=self.input_object):
|
|
with open(self.input_object, 'rb') as input_object:
|
|
check_buffer: bytes = input_object.read(0x100)
|
|
else:
|
|
check_buffer = self.input_buffer[:0x100]
|
|
|
|
return bool(PAT_TOSHIBA_COM.search(check_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_read(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_read(in_path=output_path):
|
|
printer(message='Successful extraction via ToshibaComExtractor!', padding=self.padding)
|
|
|
|
return True
|
|
|
|
return False
|