Move default symbol hooks to arch (#40)

Author: sledgeh4w

src/infernum/arch.py

@@ -1,4 +1,6 @@
 from unicorn import arm_const, arm64_const
+
+from . import hooks
 from .structs import Arch
 
 
@@ -17,6 +19,14 @@
         arm_const.UC_ARM_REG_R3,
     ],
     reg_ret=arm_const.UC_ARM_REG_R0,
+    symbol_hooks={
+        "__ctype_get_mb_cur_max": hooks.simply_return(1),
+        "malloc": hooks.hook_malloc,
+        "free": hooks.hook_free,
+        "getcwd": hooks.hook_getcwd,
+        "getpid": hooks.hook_getpid,
+        "gettid": hooks.hook_gettid,
+    },
 )
 
 
@@ -39,4 +49,12 @@
         arm64_const.UC_ARM64_REG_X7,
     ],
     reg_ret=arm64_const.UC_ARM64_REG_X0,
+    symbol_hooks={
+        **arch_arm.symbol_hooks,
+        "arc4random": hooks.hook_arc4random,
+        "clock_nanosleep": hooks.simply_return(0),
+        "nanosleep": hooks.simply_return(0),
+        "pthread_mutex_lock": hooks.simply_return(),
+        "pthread_mutex_unlock": hooks.simply_return(),
+    },
 )

src/infernum/core.py

@@ -21,7 +21,7 @@
 )
 from unicorn.unicorn import UC_HOOK_CODE_TYPE
 
-from . import const, hooks
+from . import const
 from .arch import arch_arm, arch_arm64
 from .exceptions import EmulatorCrashedException, SymbolMissingException
 from .log import get_logger
@@ -113,30 +113,7 @@ def _create_cs(self) -> Cs:
 
     def _init_symbol_hooks(self):
         """Initialize default symbol hooks."""
-        _hooks = {
-            "__ctype_get_mb_cur_max": hooks.simply_return(1),
-            "malloc": hooks.hook_malloc,
-            "free": hooks.hook_free,
-            "getcwd": hooks.hook_getcwd,
-            "getpid": hooks.hook_getpid,
-            "gettid": hooks.hook_gettid,
-        }
-
-        if self.arch == arch_arm:
-            _hooks.update({})
-
-        elif self.arch == arch_arm64:
-            _hooks.update(
-                {
-                    "arc4random": hooks.hook_arc4random,
-                    "clock_nanosleep": hooks.simply_return(0),
-                    "nanosleep": hooks.simply_return(0),
-                    "pthread_mutex_lock": hooks.simply_return(),
-                    "pthread_mutex_unlock": hooks.simply_return(),
-                }
-            )
-
-        self._symbol_hooks.update(_hooks)
+        self._symbol_hooks.update(self.arch.symbol_hooks)
 
     def _init_trap_memory(self):
         """Initialize trap area memory."""

src/infernum/structs.py

@@ -1,5 +1,7 @@
 from dataclasses import dataclass
-from typing import List, Optional
+from typing import Dict, List, Optional
+
+from unicorn.unicorn import UC_HOOK_CODE_TYPE
 
 
 @dataclass
@@ -60,3 +62,5 @@ class Arch:
 
     reg_args: List[int]
     reg_ret: int
+
+    symbol_hooks: Dict[str, UC_HOOK_CODE_TYPE]