ntmemoryapi 1.6.3__tar.gz → 1.7.1__tar.gz

{ntmemoryapi-1.6.3 → ntmemoryapi-1.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ntmemoryapi
-Version: 1.6.3
+Version: 1.7.1
 Summary: Simple library for Windows to manipulate process virtual memory with stelthy syscall wraps
 Author: Xenely
 Requires-Dist: psutil>=7.1.3

{ntmemoryapi-1.6.3 → ntmemoryapi-1.7.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ntmemoryapi"
-version = "1.6.3"
+version = "1.7.1"
 description = "Simple library for Windows to manipulate process virtual memory with stelthy syscall wraps"
 authors = [
     {name = "Xenely"}

{ntmemoryapi-1.6.3 → ntmemoryapi-1.7.1}/src/ntmemoryapi/__init__.py

@@ -20,7 +20,6 @@ syscall_wrapper = DirectSyscallWrapper()
 
 PROCESS_ID = 1
 PROCESS_NAME = 2
-PROCESS_USERNAME = 4
 
 
 # ==-------------------------------------------------------------------== #
@@ -238,28 +237,21 @@ _nt_protect_virtual_memory = syscall_wrapper.wrap("NtProtectVirtualMemory", resu
 # ==-------------------------------------------------------------------== #
 # Functions                                                               #
 # ==-------------------------------------------------------------------== #
-def list_processes(include_process_information: int = PROCESS_ID | PROCESS_NAME | PROCESS_USERNAME, current_username_only: bool = True) -> list[dict[str, int | str]]:
+def list_processes(include_process_information: int = PROCESS_ID | PROCESS_NAME) -> list[dict[str, int | str]]:
     """List all of the currently active system processes."""
 
     # If `include_id` and `include_name` disabled both
     if not include_process_information & PROCESS_ID and not include_process_information & PROCESS_NAME:
         raise Exception("Unable to disable ID and name including at once")
 
-    # Current process username
-    current_username = psutil.Process().username()
-
     # Result list of processes
-    processes = list()
+    processes = []
 
     # Process iteration
     for process in psutil.process_iter():
 
-        # If process started not from current username
-        if current_username_only and process.username() != current_username:
-            continue
-
         # Save process to list
-        process_dict = dict()
+        process_dict = {}
 
         # If process ID including required
         if include_process_information & PROCESS_ID:
@@ -269,10 +261,6 @@ def list_processes(include_process_information: int = PROCESS_ID | PROCESS_NAME
         if include_process_information & PROCESS_NAME:
             process_dict |= {"name": process.name()}
 
-        # If process username including required
-        if include_process_information & PROCESS_USERNAME:
-            process_dict |= {"username": process.username()}
-
         # Save process information to list
         processes.append(process_dict)
 
@@ -289,17 +277,17 @@ class Process:
     # Methods                                                                 #
     # ==-------------------------------------------------------------------== #
 
-    def __init__(self, name_or_pid: int | str, access: int = PROCESS_ALL_ACCESS, current_username_only: bool = True) -> None:
+    def __init__(self, name_or_pid: int | str, access: int = PROCESS_ALL_ACCESS) -> None:
         """Initialize instance to manipulate process."""
 
         # Open process by it's ID or it's name
         match name_or_pid:
 
-            case pid if type(name_or_pid) is int:
-                self.handle, self.name, self.pid = self.__init_with_pid(pid, access, current_username_only)
+            case int():
+                self.handle, self.pid, self.name = self.__init_with_pid(name_or_pid, access)
 
-            case name if type(name_or_pid) is str:
-                self.handle, self.name, self.pid = self.__init_with_name(name, access, current_username_only)
+            case str():
+                self.handle, self.pid, self.name = self.__init_with_name(name_or_pid, access)
 
             case _:
                 raise Exception("Invalid `pid_or_name` argument value, have to be `int` or `str` type")
@@ -331,7 +319,7 @@ class Process:
             raise Exception("Unable to create snapshot to iterate process modules")
 
         # Result list of process modules
-        process_modules = list()
+        process_modules = []
 
         # Process modules enumeration
         try:
@@ -364,11 +352,11 @@ class Process:
 
         return process_modules
 
-    def list_memory_regions(self, allowed_states: list[int] = list(), allowed_protects: list[int] = list(), allowed_types: list[int] = list(), memory_regions_filter: typing.Callable[[MEMORY_BASIC_INFORMATION], bool] | None = None) -> list[MEMORY_BASIC_INFORMATION]:
+    def list_memory_regions(self, allowed_states: list[int] = [], allowed_protects: list[int] = [], allowed_types: list[int] = [], memory_regions_filter: typing.Callable[[MEMORY_BASIC_INFORMATION], bool] | None = None) -> list[MEMORY_BASIC_INFORMATION]:
         """List all of the aviable process memory regions."""
 
         # Result list of memory regions
-        memory_regions = list()
+        memory_regions = []
 
         # Iterate all of the process memory regions
         current_address = 0
@@ -487,7 +475,7 @@ class Process:
 
         return memory_old_protect.value or 0
 
-    def pattern_scan(self, pattern: str, allowed_states: list[int] = [MEM_COMMIT], allowed_protects: list[int] = [PAGE_READWRITE], allowed_types: list[int] = list(), start_address: int | None = None, end_address: int | None = None, return_first: int | None = None, memory_regions_filter: typing.Callable[[MEMORY_BASIC_INFORMATION], bool] | None = None) -> list[int]:
+    def pattern_scan(self, pattern: str, allowed_states: list[int] = [MEM_COMMIT], allowed_protects: list[int] = [PAGE_READWRITE], allowed_types: list[int] = [], start_address: int | None = None, end_address: int | None = None, return_first: int | None = None, memory_regions_filter: typing.Callable[[MEMORY_BASIC_INFORMATION], bool] | None = None) -> list[int]:
         """Scan process and return address that validates given pattern hex byte mask, use `??` to wildcard byte, for example - "14 00 00 00 DB FF ?? ?? FF FF 00 00"."""
 
         # Validate given pattern
@@ -504,7 +492,7 @@ class Process:
             raise Exception("Invalid pattern: `%s`" % pattern)
 
         # Result list of found addresses
-        found_addresses = list()
+        found_addresses = []
 
         # iterate memory regions and finding addresses at them
         for region in self.list_memory_regions(allowed_states, allowed_protects, allowed_types, memory_regions_filter):
@@ -758,11 +746,11 @@ class Process:
     # ==-------------------------------------------------------------------== #
     # Private methods                                                         #
     # ==-------------------------------------------------------------------== #
-    def __init_with_pid(self, pid: int, access: int, current_username_only: bool, process_name: str | None = None) -> int:
+    def __init_with_pid(self, pid: int, access: int, listed_processes: list[dict[str, int | str]] | None = None) -> tuple[int, int, str]:
         """Open process handle by it's ID with desired access."""
 
-        # Iterate all of the processes if name not defined
-        for process in list_processes(current_username_only=current_username_only):
+        # Iterate all of the processes
+        for process in listed_processes if listed_processes is not None else list_processes():
 
             # If process have a reqired name
             if process["id"] == pid:
@@ -790,16 +778,16 @@ class Process:
             else:
                 raise Exception("NtOpenProcess failed with status: 0x%s" % hex(result)[2:].upper())
 
-        return handle.value, process_name, pid
+        return handle.value, pid, process_name
 
-    def __init_with_name(self, name: str, access: int, current_username_only: bool) -> int:
+    def __init_with_name(self, name: str, access: int) -> tuple[int, int, str]:
         """Open process hanle by it's name with desired access."""
 
         # Iterate all of the processes using snapshot
-        for process in list_processes(current_username_only=current_username_only):
+        for process in (listed_processes := list_processes()):
 
             # If process have a reqired name
             if process["name"].lower() == name.strip().lower():
-                return self.__init_with_pid(process["id"], access, current_username_only, process["name"])
+                return self.__init_with_pid(process["id"], access, listed_processes)
 
         raise Exception("Process with `%s` name not found" % name)