mirror of
https://github.com/platomav/BIOSUtilities.git
synced 2024-09-19 07:18:53 -04:00
bdf926f6fd
Added ability to parse nested Panasonic BIOS update executable directly Restructured logic to allow more flexibility on input executable parsing Populated code type hints and applied multiple small improvements
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
#!/usr/bin/env python3 -B
|
|
# coding=utf-8
|
|
|
|
"""
|
|
Copyright (C) 2022-2024 Plato Mavropoulos
|
|
"""
|
|
|
|
import pefile
|
|
|
|
from common.system import printer
|
|
from common.text_ops import file_to_bytes
|
|
|
|
|
|
def is_pe_file(in_file: str | bytes) -> bool:
|
|
""" Check if input is a PE file """
|
|
|
|
return bool(get_pe_file(in_file, silent=True))
|
|
|
|
|
|
def get_pe_file(in_file: str | bytes, padding: int = 0, fast: bool = True, silent: bool = False) -> pefile.PE | None:
|
|
""" Get pefile object from PE file """
|
|
|
|
pe_file: pefile.PE | None = None
|
|
|
|
try:
|
|
# Analyze detected MZ > PE image buffer
|
|
pe_file = pefile.PE(data=file_to_bytes(in_file), fast_load=fast)
|
|
except Exception as error: # pylint: disable=broad-except
|
|
if not silent:
|
|
filename: str = in_file if isinstance(in_file, str) else 'buffer'
|
|
|
|
printer(f'Error: Could not get pefile object from {filename}: {error}!', padding)
|
|
|
|
return pe_file
|
|
|
|
|
|
def get_pe_desc(pe_file: pefile.PE, padding: int = 0, silent: bool = False) -> bytes:
|
|
""" Get PE description from pefile object info """
|
|
|
|
return get_pe_info(pe_file, padding, silent).get(b'FileDescription', b'')
|
|
|
|
|
|
def get_pe_info(pe_file: pefile.PE, padding: int = 0, silent: bool = False) -> dict:
|
|
""" Get PE info from pefile object """
|
|
|
|
pe_info: dict = {}
|
|
|
|
try:
|
|
# When fast_load is used, IMAGE_DIRECTORY_ENTRY_RESOURCE must be parsed prior to FileInfo > StringTable
|
|
pe_file.parse_data_directories(directories=[pefile.DIRECTORY_ENTRY['IMAGE_DIRECTORY_ENTRY_RESOURCE']])
|
|
|
|
# Retrieve MZ > PE > FileInfo > StringTable information
|
|
pe_info = pe_file.FileInfo[0][0].StringTable[0].entries
|
|
except Exception as error: # pylint: disable=broad-except
|
|
if not silent:
|
|
printer(f'Error: Could not get PE info from pefile object: {error}!', padding)
|
|
|
|
return pe_info
|
|
|
|
|
|
def show_pe_info(pe_file: pefile.PE, padding: int = 0) -> None:
|
|
""" Print PE info from pefile StringTable """
|
|
|
|
pe_info: dict = get_pe_info(pe_file=pe_file, padding=padding)
|
|
|
|
if isinstance(pe_info, dict):
|
|
for title, value in pe_info.items():
|
|
info_title: str = title.decode('utf-8', 'ignore').strip()
|
|
info_value: str = value.decode('utf-8', 'ignore').strip()
|
|
|
|
if info_title and info_value:
|
|
printer(f'{info_title}: {info_value}', padding, new_line=False)
|