lazymio/unicorn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Release mmio callbacks on a best-effort basis

Browse Source
This commit is contained in:
elicn
2022-10-20 14:18:49 +03:00
parent d6e5e6a19f
commit 99d6745461
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
bindings/python/unicorn/unicorn.py
View File

@@ -641,7 +641,14 @@ class Uc(RegStateManager):
# TODO: this is where mmio callbacks need to be released from cache, # TODO: this is where mmio callbacks need to be released from cache,
# but we cannot tell whether this is an mmio range. also, memory ranges # but we cannot tell whether this is an mmio range. also, memory ranges
# might be splitted by 'map_protect' after they were mapped, so the # might be splitted by 'map_protect' after they were mapped, so the
# (start, end) tuple may not be suitable for retrieving the callbacks # (start, end) tuple may not be suitable for retrieving the callbacks.
#
# here we try to do that on a best-effort basis:
rng = (address, address + size)
if rng in self._mmio_callbacks:
del self._mmio_callbacks[rng]
def mem_protect(self, address: int, size: int, perms: int = uc.UC_PROT_ALL) -> None: def mem_protect(self, address: int, size: int, perms: int = uc.UC_PROT_ALL) -> None:
"""Modify access protection bitmask of a mapped memory range. """Modify access protection bitmask of a mapped memory range.
@@ -686,10 +693,9 @@ class Uc(RegStateManager):
raise UcError(status) raise UcError(status)
# hold a reference to mmio callbacks # hold a reference to mmio callbacks
rng_starts = address rng = (address, address + size)
rng_ends = address + size
self._mmio_callbacks[(rng_starts, rng_ends)] = (read_cb_fptr, write_cb_fptr) self._mmio_callbacks[rng] = (read_cb_fptr, write_cb_fptr)
def mem_regions(self) -> Iterator[Tuple[int, int, int]]: def mem_regions(self) -> Iterator[Tuple[int, int, int]]:
"""Iterate through mapped memory regions. """Iterate through mapped memory regions.