add support for closure handling

This commit is contained in:
Robin Müller 2022-10-21 11:28:33 +02:00
parent 1dfc2fca2f
commit 3cbfeb7015
No known key found for this signature in database
GPG Key ID: 71B58F8A3CDFA9AC
3 changed files with 31 additions and 10 deletions

2
deps/tmtccmd vendored

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

View File

@ -20,8 +20,13 @@ from pus_tm.devs.gps import handle_gps_data
from pus_tm.devs.gyros import handle_gyros_hk_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 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 pus_tm.devs.syrlinks import handle_syrlinks_hk_data
from tmtc.acs.imtq import ImtqSetIds, handle_self_test_data, handle_eng_set, \ from tmtc.acs.imtq import (
handle_calibrated_mtm_measurement, handle_raw_mtm_measurement 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.defs import FsfwTmTcPrinter
from pus_tm.system.core import handle_core_hk_data from pus_tm.system.core import handle_core_hk_data
from pus_tm.devs.mgms import handle_mgm_hk_data from pus_tm.devs.mgms import handle_mgm_hk_data

View File

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