processes: distinguish thread class using arch_t

doronz88 · Mar 28, 2022 · 96525c1 · 96525c1
1 parent d6ecbe0
commit 96525c1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/rpcclient/rpcclient/darwin/processes.py b/src/rpcclient/rpcclient/darwin/processes.py
@@ -17,6 +17,7 @@
 from rpcclient.exceptions import BadReturnValueError, ArgumentError
 from rpcclient.processes import Processes
 from rpcclient.structs.consts import SIGTERM
+from rpcclient.protocol import arch_t
 
 FdStruct = namedtuple('FdStruct', 'fd struct')
 
@@ -156,9 +157,10 @@ def __init__(self, client, pid: int):
         self._client = client
         self._pid = pid
 
-        self._thread_class = IntelThread64
-        if self._client.uname.machine != 'x86_64':
+        if self._client.arch == arch_t.ARCH_ARM64:
             self._thread_class = ArmThread64
+        else:
+            self._thread_class = IntelThread64
 
     def kill(self, sig: int = SIGTERM):
         """ kill(pid, sig) at remote. read man for more details. """