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

Allow setting CPU model on Uc initialization

Browse Source
This commit is contained in:
elicn 2023-07-14 00:44:58 +03:00
parent cef2b3394d
commit 0a876d30ea
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -439,7 +439,7 @@ class Uc(RegStateManager):
return (uc_maj, uc_min) == (bnd_maj, bnd_min) return (uc_maj, uc_min) == (bnd_maj, bnd_min)
def __new__(cls, arch: int, mode: int): def __new__(cls, arch: int, mode: int, cpu: Optional[int] = None):
# verify version compatibility with the core before doing anything # verify version compatibility with the core before doing anything
if not Uc.__is_compliant(): if not Uc.__is_compliant():
raise UcError(uc.UC_ERR_VERSION) raise UcError(uc.UC_ERR_VERSION)
@ -479,12 +479,13 @@ class Uc(RegStateManager):
# return the appropriate unicorn subclass type # return the appropriate unicorn subclass type
return super(Uc, cls).__new__(subclass) return super(Uc, cls).__new__(subclass)
def __init__(self, arch: int, mode: int) -> None: def __init__(self, arch: int, mode: int, cpu: Optional[int] = None) -> None:
"""Initialize a Unicorn engine instance. """Initialize a Unicorn engine instance.
Args: Args:
arch: emulated architecture identifier (see UC_ARCH_* constants) arch: emulated architecture identifier (see UC_ARCH_* constants)
mode: emulated processor mode (see UC_MODE_* constants) mode: emulated processor mode (see UC_MODE_* constants)
cpu: emulated cpu model (see UC_CPU_* constants) [optional]
""" """
self._arch = arch self._arch = arch
@ -498,6 +499,9 @@ class Uc(RegStateManager):
self._uch = None self._uch = None
raise UcError(status) raise UcError(status)
if cpu is not None:
self.ctl_set_cpu_model(cpu)
# we have to keep a reference to the callbacks so they do not get gc-ed # we have to keep a reference to the callbacks so they do not get gc-ed
# see: https://docs.python.org/3/library/ctypes.html#callback-functions # see: https://docs.python.org/3/library/ctypes.html#callback-functions
self._callbacks: MutableMapping[int, ctypes._FuncPointer] = {} self._callbacks: MutableMapping[int, ctypes._FuncPointer] = {}