139 lines
4.6 KiB
Python
139 lines
4.6 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# Cross Platform and Multi Architecture Advanced Binary Emulation Framework
|
|
#
|
|
|
|
import ctypes
|
|
|
|
from qiling.core import Qiling
|
|
from qiling.hw.peripheral import QlPeripheral
|
|
from qiling.utils import ql_get_module_function
|
|
from qiling.exception import QlErrorModuleFunctionNotFound
|
|
|
|
|
|
class QlHwManager:
|
|
def __init__(self, ql: Qiling):
|
|
self.ql = ql
|
|
|
|
self.entity = {}
|
|
self.region = {}
|
|
|
|
self.stepable = {}
|
|
|
|
def create(self, label: str, struct: str=None, base: int=None, kwargs: dict={}) -> "QlPeripheral":
|
|
""" Create the peripheral accroding the label and envs.
|
|
|
|
struct: Structure of the peripheral. Use defualt ql structure if not provide.
|
|
base: Base address. Use defualt address if not provide.
|
|
"""
|
|
|
|
if struct is None:
|
|
struct, base, kwargs = self.load_env(label.upper())
|
|
|
|
try:
|
|
|
|
entity = ql_get_module_function('qiling.hw', struct)(self.ql, label, **kwargs)
|
|
|
|
self.entity[label] = entity
|
|
if hasattr(entity, 'step'):
|
|
self.stepable[label] = entity
|
|
|
|
self.region[label] = [(lbound + base, rbound + base) for (lbound, rbound) in entity.region]
|
|
|
|
|
|
return entity
|
|
except QlErrorModuleFunctionNotFound:
|
|
self.ql.log.debug(f'The {struct}({label}) has not been implemented')
|
|
|
|
def delete(self, label: str):
|
|
""" Remove the peripheral
|
|
"""
|
|
if label in self.entity:
|
|
self.entity.pop(label)
|
|
self.region.pop(label)
|
|
if label in self.stepable:
|
|
self.stepable.pop(label)
|
|
|
|
def load_env(self, label: str):
|
|
""" Get peripheral information (structure, base address, initialization list) from env.
|
|
|
|
Args:
|
|
label (str): Peripheral Label
|
|
|
|
"""
|
|
args = self.ql.env[label]
|
|
|
|
return args['struct'], args['base'], args.get("kwargs", {})
|
|
|
|
def load_all(self):
|
|
for label, args in self.ql.env.items():
|
|
if args['type'] == 'peripheral':
|
|
self.create(label.lower(), args['struct'], args['base'], args.get("kwargs", {}))
|
|
|
|
def find(self, address: int):
|
|
""" Find the peripheral at `address`
|
|
"""
|
|
|
|
for label in self.entity.keys():
|
|
for lbound, rbound in self.region[label]:
|
|
if lbound <= address < rbound:
|
|
return self.entity[label]
|
|
|
|
def step(self):
|
|
""" Update all peripheral's state
|
|
"""
|
|
for entity in self.stepable.values():
|
|
entity.step()
|
|
|
|
def setup_mmio(self, begin, size, info=""):
|
|
mmio = ctypes.create_string_buffer(size)
|
|
|
|
def mmio_read_cb(ql, offset, size):
|
|
address = begin + offset
|
|
hardware = self.find(address)
|
|
|
|
if hardware:
|
|
return hardware.read(address - hardware.base, size)
|
|
else:
|
|
ql.log.debug('%s Read non-mapped hardware [0x%08x]' % (info, address))
|
|
|
|
buf = ctypes.create_string_buffer(size)
|
|
ctypes.memmove(buf, ctypes.addressof(mmio) + offset, size)
|
|
return int.from_bytes(buf.raw, byteorder='little')
|
|
|
|
def mmio_write_cb(ql, offset, size, value):
|
|
address = begin + offset
|
|
hardware = self.find(address)
|
|
|
|
if hardware:
|
|
hardware.write(address - hardware.base, size, value)
|
|
else:
|
|
ql.log.debug('%s Write non-mapped hardware [0x%08x] = 0x%08x' % (info, address, value))
|
|
ctypes.memmove(ctypes.addressof(mmio) + offset, (value).to_bytes(size, 'little'), size)
|
|
|
|
self.ql.mem.map_mmio(begin, size, mmio_read_cb, mmio_write_cb, info=info)
|
|
|
|
def show_info(self):
|
|
self.ql.log.info(f'{"Start":8s} {"End":8s} {"Label":8s} {"Class"}')
|
|
|
|
for label, region in self.region.items():
|
|
for lbound, ubound in region:
|
|
classname = self.entity[label].__class__.__name__
|
|
self.ql.log.info(f'{lbound:08x} - {ubound:08x} {label.upper():8s} {classname}')
|
|
|
|
def __getitem__(self, key):
|
|
return self.entity[key]
|
|
|
|
def __setitem__(self, key, value):
|
|
self.entity[key] = value
|
|
|
|
def __getattr__(self, key):
|
|
return self.entity.get(key)
|
|
|
|
def save(self):
|
|
return {label : entity.save() for label, entity in self.entity.items()}
|
|
|
|
def restore(self, saved_state):
|
|
for label, data in saved_state.items():
|
|
self.entity[label].restore(data)
|