ntmemoryapi 2.0.1__tar.gz → 2.0.3__tar.gz

{ntmemoryapi-2.0.1 → ntmemoryapi-2.0.3}/PKG-INFO

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

{ntmemoryapi-2.0.1 → ntmemoryapi-2.0.3}/pyproject.toml

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

{ntmemoryapi-2.0.1 → ntmemoryapi-2.0.3}/src/ntmemoryapi/__init__.py

@@ -76,7 +76,7 @@ class CLIENT_ID(ctypes.Structure):
 
     _fields_ = [
         ("unique_process", ctypes.c_void_p),
-        ("unique_thread", ctypes.c_void_p)
+        ("unique_thread", ctypes.c_void_p),
     ]
 
 
@@ -89,7 +89,7 @@ class OBJECT_ATTRIBUTES(ctypes.Structure):
         ("object_name", ctypes.c_void_p),
         ("attributes", ctypes.c_ulong),
         ("security_descriptor", ctypes.c_void_p),
-        ("security_quality_of_service", ctypes.c_void_p)
+        ("security_quality_of_service", ctypes.c_void_p),
     ]
 
 
@@ -106,7 +106,7 @@ class MODULEENTRY32(ctypes.Structure):
         ("mod_base_size", ctypes.c_ulong),
         ("h_module", ctypes.c_void_p),
         ("sz_module", ctypes.c_char * 256),
-        ("sz_exe_path", ctypes.c_char * 260)
+        ("sz_exe_path", ctypes.c_char * 260),
     ]
 
     @property
@@ -251,23 +251,23 @@ def list_processes(include_process_information: int = PROCESS_ID | PROCESS_NAME)
     for process in psutil.process_iter():
 
         # Save process to list
-        process_dict = {}
+        process_info = {}
 
         # If process ID including required
         if include_process_information & PROCESS_ID:
-            process_dict |= {"id": process.pid}
+            process_info["id"] = process.pid
 
         # If process ID including required
         if include_process_information & PROCESS_NAME:
-            process_dict |= {"name": process.name()}
+            process_info["name"] = process.name()
 
         # Save process information to list
-        processes.append(process_dict)
+        processes.append(process_info)
 
     return processes
 
 
-def __get_be_buffer(soure_c_type: typing.Any) -> ctypes.BigEndianStructure:
+def _get_be_buffer(soure_c_type: typing.Any) -> ctypes.BigEndianStructure:
     """Create buffer to hold data in big-endian format."""
 
     class BigEndianValue(ctypes.BigEndianStructure):
@@ -309,7 +309,7 @@ class Process:
         # Try create file at temp directory to load SIMD KMP .dll (Module to blazingly fast pattern scaning)
         try:
 
-            # Write library bytes directry from python list
+            # Write library bytes directly from python list
             with open("%s\\simdkmp.dll" % (appdata := os.getenv("APPDATA")), "wb") as file:
                 file.write(bytes(embed.kmp))
 
@@ -493,7 +493,7 @@ class Process:
         """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
-        for byte in pattern.strip().split():
+        for byte in pattern.lower().strip().split():
 
             # If pattern byte is not wildcard and valid
             if len(byte) == 2 and [item in "0123456789ABCDEFabcdef" for item in byte].count(True) == 2:
@@ -570,7 +570,7 @@ class Process:
         """Read 2 byte signed integer value located at given address."""
 
         # If result failed
-        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_int16)() if big_endian else ctypes.c_int16()), ctypes.sizeof(buffer), None)):
+        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_int16)() if big_endian else ctypes.c_int16()), ctypes.sizeof(buffer), None)):
             raise Exception("NtReadVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
         return buffer.value
@@ -579,7 +579,7 @@ class Process:
         """Read 4 byte signed integer value located at given address."""
 
         # If result failed
-        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_int32)() if big_endian else ctypes.c_int32()), ctypes.sizeof(buffer), None)):
+        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_int32)() if big_endian else ctypes.c_int32()), ctypes.sizeof(buffer), None)):
             raise Exception("NtReadVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
         return buffer.value
@@ -588,7 +588,7 @@ class Process:
         """Read 8 byte signed integer value located at given address."""
 
         # If result failed
-        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_int64)() if big_endian else ctypes.c_int64()), ctypes.sizeof(buffer), None)):
+        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_int64)() if big_endian else ctypes.c_int64()), ctypes.sizeof(buffer), None)):
             raise Exception("NtReadVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
         return buffer.value
@@ -606,7 +606,7 @@ class Process:
         """Read 2 byte unsigned integer value located at given address."""
 
         # If result failed
-        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_uint16)() if big_endian else ctypes.c_uint16()), ctypes.sizeof(buffer), None)):
+        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_uint16)() if big_endian else ctypes.c_uint16()), ctypes.sizeof(buffer), None)):
             raise Exception("NtReadVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
         return buffer.value
@@ -615,7 +615,7 @@ class Process:
         """Read 4 byte unsigned integer value located at given address."""
 
         # If result failed
-        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_uint32)() if big_endian else ctypes.c_uint32()), ctypes.sizeof(buffer), None)):
+        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_uint32)() if big_endian else ctypes.c_uint32()), ctypes.sizeof(buffer), None)):
             raise Exception("NtReadVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
         return buffer.value
@@ -624,7 +624,7 @@ class Process:
         """Read 8 byte unsigned integer value located at given address."""
 
         # If result failed
-        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_uint64)() if big_endian else ctypes.c_uint64()), ctypes.sizeof(buffer), None)):
+        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_uint64)() if big_endian else ctypes.c_uint64()), ctypes.sizeof(buffer), None)):
             raise Exception("NtReadVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
         return buffer.value
@@ -633,7 +633,7 @@ class Process:
         """Read 4 byte floating-point digit value located at given address."""
 
         # If result failed
-        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_float)() if big_endian else ctypes.c_float()), ctypes.sizeof(buffer), None)):
+        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_float)() if big_endian else ctypes.c_float()), ctypes.sizeof(buffer), None)):
             raise Exception("NtReadVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
         return buffer.value
@@ -642,7 +642,7 @@ class Process:
         """Read 8 byte floating-point digit value located at given address."""
 
         # If result failed
-        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_double)() if big_endian else ctypes.c_double()), ctypes.sizeof(buffer), None)):
+        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_double)() if big_endian else ctypes.c_double()), ctypes.sizeof(buffer), None)):
             raise Exception("NtReadVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
         return buffer.value
@@ -651,7 +651,7 @@ class Process:
         """Read bytes array of variadic size located at given address."""
 
         # If result failed
-        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_int8 * size)() if big_endian else (ctypes.c_int8 * size)()), ctypes.sizeof(buffer), None)):
+        if (result := _nt_read_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_int8 * size)() if big_endian else (ctypes.c_int8 * size)()), ctypes.sizeof(buffer), None)):
             raise Exception("NtReadVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
         return bytes(buffer)
@@ -676,21 +676,21 @@ class Process:
         """Write 2 byte signed integer value at given address."""
 
         # If result failed
-        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_int16(value)) if big_endian else ctypes.c_int16(value)), ctypes.sizeof(buffer), None)):
+        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_int16(value)) if big_endian else ctypes.c_int16(value)), ctypes.sizeof(buffer), None)):
             raise Exception("NtWriteVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
     def write_int32(self, address: int, value: int, big_endian: bool = False) -> None:
         """Write 4 byte signed integer value at given address."""
 
         # If result failed
-        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_int32(value)) if big_endian else ctypes.c_int32(value)), ctypes.sizeof(buffer), None)):
+        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_int32(value)) if big_endian else ctypes.c_int32(value)), ctypes.sizeof(buffer), None)):
             raise Exception("NtWriteVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
     def write_int64(self, address: int, value: int, big_endian: bool = False) -> None:
         """Write 8 byte signed integer value at given address."""
 
         # If result failed
-        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_int64(value)) if big_endian else ctypes.c_int64(value)), ctypes.sizeof(buffer), None)):
+        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_int64(value)) if big_endian else ctypes.c_int64(value)), ctypes.sizeof(buffer), None)):
             raise Exception("NtWriteVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
     def write_uint8(self, address: int, value: int) -> None:
@@ -704,35 +704,35 @@ class Process:
         """Write 2 byte unsigned integer value at given address."""
 
         # If result failed
-        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_uint16(value)) if big_endian else ctypes.c_uint16(value)), ctypes.sizeof(buffer), None)):
+        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_uint16(value)) if big_endian else ctypes.c_uint16(value)), ctypes.sizeof(buffer), None)):
             raise Exception("NtWriteVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
     def write_uint32(self, address: int, value: int, big_endian: bool = False) -> None:
         """Write 4 byte unsigned integer value at given address."""
 
         # If result failed
-        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_uint32(value)) if big_endian else ctypes.c_uint32(value)), ctypes.sizeof(buffer), None)):
+        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_uint32(value)) if big_endian else ctypes.c_uint32(value)), ctypes.sizeof(buffer), None)):
             raise Exception("NtWriteVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
     def write_uint64(self, address: int, value: int, big_endian: bool = False) -> None:
         """Write 8 byte unsigned integer value at given address."""
 
         # If result failed
-        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_uint64(value)) if big_endian else ctypes.c_uint64(value)), ctypes.sizeof(buffer), None)):
+        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_uint64(value)) if big_endian else ctypes.c_uint64(value)), ctypes.sizeof(buffer), None)):
             raise Exception("NtWriteVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
     def write_float32(self, address: int, value: float | int, big_endian: bool = False) -> float:
         """Write 4 byte floating-point digit value at given address."""
 
         # If result failed
-        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_float(value)) if big_endian else ctypes.c_float(value)), ctypes.sizeof(buffer), None)):
+        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_float(value)) if big_endian else ctypes.c_float(value)), ctypes.sizeof(buffer), None)):
             raise Exception("NtWriteVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
     def write_float64(self, address: int, value: float | int, big_endian: bool = False) -> float:
         """Write 8 byte floating-point digit value at given address."""
 
         # If result failed
-        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := __get_be_buffer(ctypes.c_double(value)) if big_endian else ctypes.c_double(value)), ctypes.sizeof(buffer), None)):
+        if (result := _nt_write_virtual_memory(self.handle, address, ctypes.byref(buffer := _get_be_buffer(ctypes.c_double(value)) if big_endian else ctypes.c_double(value)), ctypes.sizeof(buffer), None)):
             raise Exception("NtWriteVirtualMemory failed with status: 0x%s" % hex(result)[2:].upper())
 
     def write_bytes(self, address: int, value: bytes, big_endian: bool = False) -> None: