cleanup tester present sender

polybassa · May 3, 2024 · 9e1cc08 · 9e1cc08
1 parent 1b2780d
commit 9e1cc08
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/scapy/contrib/automotive/scanner/enumerator.py b/scapy/contrib/automotive/scanner/enumerator.py
@@ -191,6 +191,12 @@ def _get_initial_requests(self, **kwargs):
 
     def __reduce__(self):  # type: ignore
         f, t, d = super(ServiceEnumerator, self).__reduce__()  # type: ignore
+
+        try:
+            del d["_tester_present_sender"]
+        except KeyError:
+            pass
+
         try:
             for k, v in d["_request_iterators"].items():
                 d["_request_iterators"][k] = list(v)
@@ -287,6 +293,10 @@ def pre_execute(self, socket, state, global_configuration):
         except KeyError:
             self._tester_present_sender = None
 
+    def post_execute(self, socket, state, global_configuration):
+        # type: (_SocketUnion, EcuState, AutomotiveTestCaseExecutorConfiguration) -> None
+        self._tester_present_sender = None
+
     def execute(self, socket, state, **kwargs):
         # type: (_SocketUnion, EcuState, Any) -> None
         self.check_kwargs(kwargs)