atmt: also use Session in IO supersockets

AI-Assisted: no

Author: gpotter2

scapy/automaton.py

@@ -681,6 +681,9 @@ def recv(self, n=MTU, **kwargs):  # type: ignore
         r = self.spb.recv(n)
         if self.proto is not None and r is not None:
             r = self.proto(r, **kwargs)
+        if self.atmt.atmt_session is not None:
+            # Apply session if provided
+            r = self.atmt.atmt_session.process(r)
         return r
 
     def close(self):
@@ -962,7 +965,8 @@ def parse_args(self, debug=0, store=0, session=None, **kargs):
         self.debug_level = debug
         if debug:
             conf.logLevel = logging.DEBUG
-        self.atmt_session = session
+        if session:
+            self.atmt_session = session
         self.socket_kargs = kargs
         self.store_packets = store