Merge pull request #71 from mrexodia/primitive-types

Add a few missing primitive types

Author: mrexodia

src/dumpulator/dumpulator.py

@@ -881,7 +881,7 @@ def read(self, addr, size):
     def write(self, addr, data):
         if not isinstance(addr, int):
             addr = int(addr)
-        self._pages.write(addr, data)
+        self._pages.write(addr, bytes(data))
 
     def call(self, addr, args: List[int] = None, regs: dict = None, count=0):
         if args is None:

src/dumpulator/modules.py

@@ -74,7 +74,7 @@ def find_export(self, key: Union[str, int]):
         raise TypeError()
 
     def __repr__(self):
-        return f"Module({hex(self.base)}, {hex(self.size)}, {repr(self.path)})"
+        return f"Module({hex(self.base)}, {hex(self.size)}, {self.path})"
 
     def __hash__(self):
         return hash((self.base, self.size, self.path))

src/dumpulator/native.py

@@ -460,7 +460,6 @@ def to_regs(self, regs):
         assert regs.gs == self.SegGs & 0xFFFF
 
         # TODO: implement xmm
-
 assert ctypes.sizeof(WOW64_CONTEXT) == 0x2cc
 
 class EXCEPTION_RECORD32(ctypes.Structure):

src/dumpulator/ntprimitives.py

@@ -392,6 +392,9 @@ class KAFFINITY(ULONG_PTR):
 class KPRIORITY(ULONG_PTR):
     pass
 
+class KIRQL(UCHAR):
+    pass
+
 # TODO: should probably be bool
 class BOOLEAN(BYTE):
     pass
@@ -408,6 +411,15 @@ class PSID(PVOID):
 class PWSTR(PVOID):
     pass
 
+class CCHAR(CHAR):
+    pass
+
+class CSHORT(SHORT):
+    pass
+
+class CLONG(ULONG):
+    pass
+
 # Some unsupported enum
 class LATENCY_TIME(Enum):
     LT_DONT_CARE = 0