add support for closure handling

deps/tmtccmd

@@ -1 +1 @@
-Subproject commit 229a2b5032b0c1becd93fe37f34ddc93a6b425c4
+Subproject commit f3c9501891b582f5b4c61aec5eb939d05a2e24d2

pus_tm/hk_handling.py

@@ -20,8 +20,13 @@ from pus_tm.devs.gps import handle_gps_data
 from pus_tm.devs.gyros import handle_gyros_hk_data
 from tmtc.power.tm import handle_pdu_data, handle_p60_hk_data, handle_acu_hk_data
 from pus_tm.devs.syrlinks import handle_syrlinks_hk_data
-from tmtc.acs.imtq import ImtqSetIds, handle_self_test_data, handle_eng_set, \
-    handle_calibrated_mtm_measurement, handle_raw_mtm_measurement
+from tmtc.acs.imtq import (
+    ImtqSetIds,
+    handle_self_test_data,
+    handle_eng_set,
+    handle_calibrated_mtm_measurement,
+    handle_raw_mtm_measurement,
+)
 from pus_tm.defs import FsfwTmTcPrinter
 from pus_tm.system.core import handle_core_hk_data
 from pus_tm.devs.mgms import handle_mgm_hk_data

tmtcc.py

@@ -7,12 +7,15 @@ from pathlib import Path
 from typing import cast
 
 from spacepackets import SpacePacketHeader, SpacePacket
+from spacepackets.ccsds import SPACE_PACKET_HEADER_SIZE
 from spacepackets.cfdp import (
     ConditionCode,
     ChecksumType,
     TransmissionMode,
     PduHolder,
     DirectiveType,
+    PduFactory,
+    PduType,
 )
 from tmtccmd.cfdp import CfdpUserBase, TransactionId
 from tmtccmd.cfdp.defs import CfdpRequestType
@@ -182,11 +185,20 @@ class CfdpInCcsdsWrapper(SpecificApidHandlerBase):
         self.handler = cfdp_in_ccsds_handler
 
     def handle_tm(self, packet: bytes, _user_args: any):
-        ccsds_header_raw = packet[0:6]
-        sp_header = SpacePacketHeader.unpack(ccsds_header_raw)
-        pdu = packet[6:]
-        sp = SpacePacket(sp_header, sec_header=None, user_data=pdu)
-        self.handler.pass_packet(sp)
+        # Ignore the space packet header. Its only purpose is to use the same protocol and
+        # have a seaprate APID for space packets. If this function is called, the APID is correct.
+        pdu = packet[SPACE_PACKET_HEADER_SIZE:]
+        pdu_base = PduFactory.from_raw(pdu)
+        if pdu_base.pdu_type == PduType.FILE_DATA:
+            LOGGER.info("Received File Data PDU TM")
+        else:
+            if pdu_base.directive_type == DirectiveType.FINISHED_PDU:
+                LOGGER.info(f"Received Finished PDU TM")
+            else:
+                LOGGER.info(
+                    f"Received File Directive PDU with type {pdu_base.directive_type!r} TM"
+                )
+        self.handler.pass_pdu_packet(pdu_base)
 
 
 class TcHandler(TcHandlerBase):
@@ -207,7 +219,6 @@ class TcHandler(TcHandlerBase):
         self.file_logger = file_logger
         self.raw_logger = raw_logger
         self.gui = gui
-        self.cfdp_done = False
         self.queue_helper = DefaultPusQueueHelper(
             queue_wrapper=None,
             pus_apid=PUS_APID,
@@ -216,6 +227,9 @@ class TcHandler(TcHandlerBase):
         )
         self.cfdp_in_ccsds_wrapper = cfdp_in_ccsds_wrapper
 
+    def cfdp_done(self) -> bool:
+        return not self.cfdp_in_ccsds_wrapper.handler.put_request_pending()
+
     def feed_cb(self, info: ProcedureWrapper, wrapper: FeedWrapper):
         self.queue_helper.queue_wrapper = wrapper.queue_wrapper
         if info.proc_type == TcProcedureType.DEFAULT:
@@ -298,7 +312,6 @@ class TcHandler(TcHandlerBase):
                 )
             elif info.proc_type == TcProcedureType.CFDP:
                 LOGGER.info(f"Finished CFDP queue")
-                self.cfdp_done = True
 
 
 def setup_params() -> SetupWrapper:
@@ -429,6 +442,9 @@ def main():
                 LOGGER.info("TMTC Client in IDLE mode")
                 time.sleep(3.0)
             elif state.request == BackendRequest.DELAY_LISTENER:
+                if tc_handler.cfdp_done():
+                    LOGGER.info("CFDP transaction done, closing client")
+                    sys.exit(0)
                 time.sleep(0.5)
             elif state.request == BackendRequest.DELAY_CUSTOM:
                 if state.next_delay.total_seconds() < 0.5: