Improve results handling

Fraunhofer-AISEC · Nov 13, 2024 · 1e81ee8 · 1e81ee8
1 parent d304e78
commit 1e81ee8
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/src/gallia/commands/primitive/uds/rdbi.py b/src/gallia/commands/primitive/uds/rdbi.py
@@ -33,6 +33,7 @@ class ReadByIdentifierPrimitive(UDSScanner):
     def __init__(self, config: ReadByIdentifierPrimitiveConfig):
         super().__init__(config)
         self.config: ReadByIdentifierPrimitiveConfig = config
+        self.result: bytes | None = None
 
     async def main(self) -> None:
         try:

diff --git a/src/gallia/commands/scan/uds/services.py b/src/gallia/commands/scan/uds/services.py
@@ -49,8 +49,9 @@ def __init__(self, config: ServicesScannerConfig):
         super().__init__(config)
         self.config: ServicesScannerConfig = config
 
-    async def main(self) -> None:
         self.result: list[tuple[int, int]] = []
+
+    async def main(self) -> None:
         self.ecu.max_retry = 0
         found: dict[int, dict[int, Any]] = {}
 

diff --git a/src/gallia/commands/scan/uds/sessions.py b/src/gallia/commands/scan/uds/sessions.py
@@ -50,6 +50,8 @@ def __init__(self, config: SessionsScannerConfig):
         super().__init__(config)
         self.config: SessionsScannerConfig = config
 
+        self.result: list[int] = []
+
     async def set_session_with_hooks_handling(
         self, session: int, use_hooks: bool
     ) -> NegativeResponse | DiagnosticSessionControlResponse:
@@ -100,7 +102,6 @@ async def recover_stack(self, stack: list[int], use_hooks: bool) -> bool:
         return True
 
     async def main(self) -> None:
-        self.result: list[int] = []
         found: dict[int, list[list[int]]] = {0: [[0x01]]}
         positive_results: list[dict[str, Any]] = []
         negative_results: list[dict[str, Any]] = []