Initial commit for dissect.executable.pe

.gitignore

@@ -9,3 +9,5 @@ __pycache__/
 tests/docs/api
 tests/docs/build
 .tox/
+
+*.pyi

dissect/executable/__init__.py

@@ -1,5 +1,7 @@
 from dissect.executable.elf import ELF
+from dissect.executable.pe import PE
 
 __all__ = [
     "ELF",
+    "PE",
 ]

dissect/executable/elf/elf.py

@@ -287,7 +287,14 @@ def patch(self, new_data: bytes) -> None:
 
 
 class SegmentTable(Table[Segment]):
-    def __init__(self, fh: BinaryIO, offset: int, entries: int, size: int, c_elf: cstruct = c_elf_64):
+    def __init__(
+        self,
+        fh: BinaryIO,
+        offset: int,
+        entries: int,
+        size: int,
+        c_elf: cstruct = c_elf_64,
+    ):
         super().__init__(entries)
         self.fh = fh
         self.offset = offset

dissect/executable/exception.py

@@ -4,3 +4,28 @@ class Error(Exception):
 
 class InvalidSignatureError(Error):
     """Exception that occurs if the magic in the header does not match."""
+
+
+class InvalidPE(Error):
+    """Exception that occurs if the PE signature does not match."""
+
+
+class InvalidVA(Error):
+    """Exception that occurs when a virtual address is not found within the PE sections."""
+
+
+class InvalidAddress(Error):
+    """Exception that occurs when a raw address is not found within the PE file when translating from a virtual
+    address."""
+
+
+class InvalidArchitecture(Error):
+    """Exception that occurs when an invalid value is encountered for the PE architecture types."""
+
+
+class BuildSectionException(Error):
+    """Exception that occurs when the section to be build contains an error."""
+
+
+class ResourceException(Error):
+    """Exception that occurs when an error is thrown parsing the resources."""

dissect/executable/pe/__init__.py

@@ -0,0 +1,25 @@
+from dissect.executable.pe.builder import Builder
+from dissect.executable.pe.patcher import Patcher
+from dissect.executable.pe.pe import PE
+from dissect.executable.pe.sections.exports import ExportFunction, ExportManager
+from dissect.executable.pe.sections.imports import (
+    ImportFunction,
+    ImportManager,
+    ImportModule,
+)
+from dissect.executable.pe.sections.resources import Resource, ResourceManager
+from dissect.executable.pe.sections.sections import PESection
+
+__all__ = [
+    "PE",
+    "Builder",
+    "ExportFunction",
+    "ExportManager",
+    "ImportFunction",
+    "ImportManager",
+    "ImportModule",
+    "PESection",
+    "Patcher",
+    "Resource",
+    "ResourceManager",
+]