Shofel2_T124_python/venv/lib/python3.10/site-packages/qiling/loader/loader.py

64 lines
1.8 KiB
Python
Raw Normal View History

2024-05-25 16:45:07 +00:00
#!/usr/bin/env python3
#
# Cross Platform and Multi Architecture Advanced Binary Emulation Framework
#
from __future__ import annotations
import os
from typing import TYPE_CHECKING, Any, Mapping, MutableSequence, NamedTuple, Optional
if TYPE_CHECKING:
from qiling import Qiling
class Image(NamedTuple):
base: int
end: int
path: str
class QlLoader:
def __init__(self, ql: Qiling):
self.ql = ql
self.env = self.ql.env
self.argv = self.ql.argv
self.images: MutableSequence[Image] = []
self.skip_exit_check = False
def find_containing_image(self, address: int) -> Optional[Image]:
"""Retrieve the image object that contains the specified address.
Returns: image containing the specified address, or `None` if not found
"""
return next((image for image in self.images if image.base <= address < image.end), None)
def get_image_by_name(self, name: str, *, casefold: bool = False) -> Optional[Image]:
"""Retrieve an image by its basename.
Args:
name : image base name
casefold : whether name matching should be case-insensitive (default is case-sensitive)
Returns: image object whose basename match to the one given, or `None` if not found
"""
cf = str.casefold if casefold else lambda s: s
return next((image for image in self.images if cf(os.path.basename(image.path)) == cf(name)), None)
def save(self) -> Mapping[str, Any]:
saved_state = {
'images': [tuple(img) for img in self.images]
}
return saved_state
def restore(self, saved_state: Mapping[str, Any]):
self.images = [Image(*img) for img in saved_state['images']]
# loader main method; derivatives must implement one of their own
def run(self) -> None:
raise NotImplementedError