ntmemoryapi 1.1.0__tar.gz → 1.2.0__tar.gz

{ntmemoryapi-1.1.0 → ntmemoryapi-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ntmemoryapi
-Version: 1.1.0
+Version: 1.2.0
 Summary: Simple library for Windows to manipulate process virtual memory with stelthy syscall wraps
 Author: Xenely
 Requires-Python: >=3.10
@@ -9,6 +9,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: psutil (>=7.1.0,<8.0.0)
 Description-Content-Type: text/markdown
 
 

{ntmemoryapi-1.1.0 → ntmemoryapi-1.2.0}/ntmemoryapi/__init__.py

@@ -7,6 +7,7 @@
 
 import os
 import ctypes
+import psutil
 
 # Local imports
 from .misc import *
@@ -83,35 +84,6 @@ class OBJECT_ATTRIBUTES(ctypes.Structure):
     ]
 
 
-class PROCESSENTRY32(ctypes.Structure):
-    """Process entry structure for processes enumeration."""
-
-    _fields_ = [
-        ("dw_size", ctypes.c_ulong),
-        ("cnt_usage", ctypes.c_ulong),
-        ("th32_process_id", ctypes.c_ulong),
-        ("th32_default_heap_id", ctypes.c_void_p),
-        ("th32_module_id", ctypes.c_ulong),
-        ("cnt_threads", ctypes.c_ulong),
-        ("th32_parent_process_id", ctypes.c_ulong),
-        ("pc_pri_class_base", ctypes.c_long),
-        ("dw_flags", ctypes.c_ulong),
-        ("sz_exe_file", ctypes.c_char * 260)
-    ]
-
-    @property
-    def pid(self) -> int:
-        """Process ID."""
-
-        return self.th32_process_id
-
-    @property
-    def name(self) -> str:
-        """Process name."""
-
-        return self.sz_exe_file.decode()
-
-
 class MODULEENTRY32(ctypes.Structure):
     """Module entry structure, describes an entry from a list of the modules belonging to the specified process."""
 
@@ -254,61 +226,43 @@ _nt_protect_virtual_memory = syscall("NtProtectVirtualMemory", result_type=ctype
 # ==-------------------------------------------------------------------== #
 # Functions                                                               #
 # ==-------------------------------------------------------------------== #
-def list_processes(include_id: bool = True, include_name: bool = True) -> list[dict[str, int | str]]:
+def list_processes(include_id: bool = True, include_name: bool = True, include_username: bool = True, current_username_only: bool = True) -> list[dict[str, int | str]]:
     """List all of the currently active system processes."""
 
     # If `include_id` and `include_name` disabled both
     if not include_id and not include_name:
         raise Exception("Unable to disable ID and name including at once")
 
-    # Process snapshot functions
-    close_handle = ctypes.windll.kernel32.CloseHandle
-    process32_next = ctypes.windll.kernel32.Process32Next
-    process32_first = ctypes.windll.kernel32.Process32First
-    create_tool_help32_snapshot = ctypes.windll.kernel32.CreateToolhelp32Snapshot
-
-    # Create snapshot to iterate processes
-    if (snapshot := create_tool_help32_snapshot(TH32CS_SNAPPROCESS, 0)) == -1:
-        raise Exception("Unable to create snapshot to iterate processes")
+    # Current process username
+    current_username = psutil.Process().username()
 
     # Result list of processes
     processes = list()
 
-    # Process enumeration
-    try:
-
-        # Process entry to iterate processes
-        process_entry = PROCESSENTRY32()
-        process_entry.dw_size = ctypes.sizeof(PROCESSENTRY32)
-
-        # Retrieve first process snapshot
-        if not process32_first(snapshot, ctypes.byref(process_entry)):
-            raise Exception("Unable to get first process to save to snapshot")
+    # Process iteration
+    for process in psutil.process_iter():
 
-        # Iterate all of the processes using snapshot
-        while True:
-
-            # Save process to list
-            process_dict = dict()
+        # If process started not from current username
+        if current_username_only and process.username() != current_username:
+            continue
 
-            # If process ID including required
-            if include_id:
-                process_dict |= {"id": process_entry.pid}
+        # Save process to list
+        process_dict = dict()
 
-            # If process ID including required
-            if include_name:
-                process_dict |= {"name": process_entry.name}
+        # If process ID including required
+        if include_id:
+            process_dict |= {"id": process.pid}
 
-            processes.append(process_dict)
-
-            # Snapshot is over
-            if not process32_next(snapshot, ctypes.byref(process_entry)):
-                break
+        # If process ID including required
+        if include_name:
+            process_dict |= {"name": process.name()}
 
-    finally:
+        # If process username including required
+        if include_username:
+            process_dict |= {"username": process.username()}
 
-        # Close snapshot handle
-        close_handle(snapshot)
+        # Save process information to list
+        processes.append(process_dict)
 
     return processes
 
@@ -323,17 +277,17 @@ class Process:
     # Methods                                                                 #
     # ==-------------------------------------------------------------------== #
 
-    def __init__(self, pid_or_name: int | str, access: int = PROCESS_ALL_ACCESS) -> None:
+    def __init__(self, pid_or_name: int | str, access: int = PROCESS_ALL_ACCESS, current_username_only: bool = True) -> None:
         """Initialize instance to manipulate process."""
 
         # Open process by it's ID or it's name
         match pid_or_name:
 
             case pid if type(pid_or_name) is int:
-                self.handle, self.name, self.pid = self.__init_with_pid(pid, access)
+                self.handle, self.name, self.pid = self.__init_with_pid(pid, access, current_username_only)
 
             case name if type(pid_or_name) is str:
-                self.handle, self.name, self.pid = self.__init_with_name(name, access)
+                self.handle, self.name, self.pid = self.__init_with_name(name, access, current_username_only)
 
             case _:
                 raise Exception("Invalid `pid_or_name` argument value, have to be `int` or `str` type")
@@ -874,11 +828,11 @@ class Process:
     # ==-------------------------------------------------------------------== #
     # Private methods                                                         #
     # ==-------------------------------------------------------------------== #
-    def __init_with_pid(self, pid: int, access: int, process_name: str | None = None) -> int:
+    def __init_with_pid(self, pid: int, access: int, current_username_only: bool, process_name: str | None = None) -> int:
         """Open process handle by it's ID with desired access."""
 
         # Iterate all of the processes if name not defined
-        for process in list_processes():
+        for process in list_processes(current_username_only):
 
             # If process have a reqired name
             if process["id"] == pid:
@@ -908,14 +862,14 @@ class Process:
 
         return handle.value, process_name, pid
 
-    def __init_with_name(self, name: str, access: int) -> int:
+    def __init_with_name(self, name: str, access: int, current_username_only: bool) -> int:
         """Open process hanle by it's name with desired access."""
 
         # Iterate all of the processes using snapshot
-        for process in list_processes():
+        for process in list_processes(current_username_only):
 
             # If process have a reqired name
             if process["name"].lower() == name.strip().lower():
-                return self.__init_with_pid(process["id"], access, process["name"])
+                return self.__init_with_pid(process["id"], access, current_username_only, process["name"])
 
         raise Exception("Process with `%s` name not found" % name)

{ntmemoryapi-1.1.0 → ntmemoryapi-1.2.0}/pyproject.toml

@@ -1,14 +1,18 @@
 [project]
 name = "ntmemoryapi"
-version = "1.1.0"
+version = "1.2.0"
 description = "Simple library for Windows to manipulate process virtual memory with stelthy syscall wraps"
 authors = [
     {name = "Xenely"}
 ]
 readme = "README.md"
 requires-python = ">=3.10"
-dependencies = []
+dependencies = [
+    "psutil (>=7.1.0,<8.0.0)"
+]
 
 [build-system]
-requires = ["poetry-core>=2.0.0,<3.0.0"]
 build-backend = "poetry.core.masonry.api"
+requires = [
+    "poetry-core>=2.0.0,<3.0.0"
+]