minor update: GetCurrentThreadSystemID & GetProcessIdsByIndex (#16)

Implementation of additional APIs
dshikashio · Feb 23, 2024 · 50cac40 · 50cac40
1 parent e58cad3
commit 50cac40
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/pybag/dbgeng/idebugsystemobjects.py b/pybag/dbgeng/idebugsystemobjects.py
@@ -90,7 +90,10 @@ def GetThreadIdByTeb(self):
         raise exception.E_NOTIMPL_Error
 
     def GetCurrentThreadSystemId(self):
-        raise exception.E_NOTIMPL_Error
+        sysid = c_ulong()
+        hr = self._sys.GetCurrentThreadSystemId(byref(sysid))
+        exception.check_err(hr)
+        return sysid.value
 
     def GetThreadIdBySystemId(self):
         raise exception.E_NOTIMPL_Error
@@ -107,8 +110,14 @@ def GetNumberProcesses(self):
         exception.check_err(hr)
         return number.value
 
-    def GetProcessIdsByIndex(self):
-        raise exception.E_NOTIMPL_Error
+    def GetProcessIdsByIndex(self, count=0):
+        if count == 0:
+            count = self.GetNumberProcesses()
+        ids = (c_ulong * count)()
+        sysids = (c_ulong * count)()
+        hr = self._sys.GetProcessIdsByIndex(0, count, ids, sysids)
+        exception.check_err(hr)
+        return (tuple(ids), tuple(sysids))
 
     def GetCurrentProcessDataOffset(self):
         offset = c_ulonglong()