148 lines
5.4 KiB
Python
148 lines
5.4 KiB
Python
from typing import Any
|
|
import uuid
|
|
import sqlite3
|
|
import logging
|
|
|
|
from spacepackets.ccsds import CdsShortTimestamp
|
|
from spacepackets.ecss import PusTm
|
|
from spacepackets.ecss.pus_17_test import Service17Tm
|
|
from spacepackets.ecss.pus_1_verification import Service1Tm, UnpackParams
|
|
from tmtccmd.logging.pus import RawTmtcTimedLogWrapper
|
|
from tmtccmd.pus import VerificationWrapper
|
|
from tmtccmd.tmtc import GenericApidHandlerBase
|
|
|
|
from opssat_tmtc.common import TM_DB_PATH, EventU32
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class PusHandler(GenericApidHandlerBase):
|
|
def __init__(
|
|
self,
|
|
file_logger: logging.Logger,
|
|
verif_wrapper: VerificationWrapper,
|
|
raw_logger: RawTmtcTimedLogWrapper,
|
|
):
|
|
super().__init__(None)
|
|
self.file_logger = file_logger
|
|
self.raw_logger = raw_logger
|
|
self.verif_wrapper = verif_wrapper
|
|
|
|
def handle_tm(self, apid: int, packet: bytes, _user_args: Any):
|
|
packet_uuid = uuid.uuid4()
|
|
try:
|
|
pus_tm = PusTm.unpack(
|
|
packet, timestamp_len=CdsShortTimestamp.TIMESTAMP_SIZE
|
|
)
|
|
except ValueError as e:
|
|
_LOGGER.warning("Could not generate PUS TM object from raw data")
|
|
_LOGGER.warning(f"Raw Packet: [{packet.hex(sep=',')}], REPR: {packet!r}")
|
|
raise e
|
|
timestamp = CdsShortTimestamp.unpack(pus_tm.timestamp)
|
|
db_con = sqlite3.connect(TM_DB_PATH)
|
|
self._store_packet_in_db(
|
|
db_con=db_con,
|
|
packet=packet,
|
|
tm_packet=pus_tm,
|
|
timestamp=timestamp,
|
|
packet_uuid=packet_uuid,
|
|
)
|
|
service = pus_tm.service
|
|
if service == 1:
|
|
tm_packet = Service1Tm.unpack(
|
|
data=packet, params=UnpackParams(CdsShortTimestamp.TIMESTAMP_SIZE, 1, 2)
|
|
)
|
|
res = self.verif_wrapper.add_tm(tm_packet)
|
|
if res is None:
|
|
_LOGGER.info(
|
|
f"Received Verification TM[{tm_packet.service}, {tm_packet.subservice}] "
|
|
f"with Request ID {tm_packet.tc_req_id.as_u32():#08x}"
|
|
)
|
|
_LOGGER.warning(
|
|
f"No matching telecommand found for {tm_packet.tc_req_id}"
|
|
)
|
|
else:
|
|
self.verif_wrapper.log_to_console(tm_packet, res)
|
|
self.verif_wrapper.log_to_file(tm_packet, res)
|
|
elif service == 3:
|
|
_LOGGER.info("No handling for HK packets implemented")
|
|
_LOGGER.info(f"Raw packet: 0x[{packet.hex(sep=',')}]")
|
|
pus_tm = PusTm.unpack(
|
|
packet, timestamp_len=CdsShortTimestamp.TIMESTAMP_SIZE
|
|
)
|
|
if pus_tm.subservice == 25:
|
|
if len(pus_tm.source_data) < 8:
|
|
raise ValueError("No addressable ID in HK packet")
|
|
json_str = pus_tm.source_data[8:]
|
|
_LOGGER.info(json_str)
|
|
elif service == 5:
|
|
tm_packet = PusTm.unpack(
|
|
packet, timestamp_len=CdsShortTimestamp.TIMESTAMP_SIZE
|
|
)
|
|
src_data = tm_packet.source_data
|
|
event_u32 = EventU32.unpack(src_data)
|
|
_LOGGER.info(f"Received event packet. Event: {event_u32}")
|
|
if event_u32.group_id == 0 and event_u32.unique_id == 0:
|
|
_LOGGER.info("Received test event")
|
|
elif service == 8:
|
|
if pus_tm.subservice == 130:
|
|
_LOGGER.info("Received Action Data Reply TM[8,130]")
|
|
reply = pus_tm.source_data
|
|
reply = reply[6:]
|
|
_LOGGER.info(f"Data Reply Content: {reply.decode()}")
|
|
elif service == 17:
|
|
tm_packet = Service17Tm.unpack(
|
|
packet, timestamp_len=CdsShortTimestamp.TIMESTAMP_SIZE
|
|
)
|
|
if tm_packet.subservice == 2:
|
|
self.file_logger.info("Received Ping Reply TM[17,2]")
|
|
_LOGGER.info("Received Ping Reply TM[17,2]")
|
|
else:
|
|
self.file_logger.info(
|
|
f"Received Test Packet with unknown subservice {tm_packet.subservice}"
|
|
)
|
|
_LOGGER.info(
|
|
f"Received Test Packet with unknown subservice {tm_packet.subservice}"
|
|
)
|
|
else:
|
|
_LOGGER.info(
|
|
f"The service {service} is not implemented in Telemetry Factory"
|
|
)
|
|
tm_packet = PusTm.unpack(
|
|
packet, timestamp_len=CdsShortTimestamp.TIMESTAMP_SIZE
|
|
)
|
|
self.raw_logger.log_tm(pus_tm)
|
|
|
|
def _store_packet_in_db(
|
|
self,
|
|
db_con: sqlite3.Connection,
|
|
packet: bytes,
|
|
timestamp: CdsShortTimestamp,
|
|
tm_packet: PusTm,
|
|
packet_uuid: uuid.UUID,
|
|
):
|
|
cursor = db_con.cursor()
|
|
cursor.execute(
|
|
"""
|
|
CREATE TABLE IF NOT EXISTS pus_tm(
|
|
packet_uuid TEXT PRIMARY KEY,
|
|
generation_time TEXT,
|
|
service NUM,
|
|
subservice NUM,
|
|
data_len NUM,
|
|
raw_data BLOB
|
|
)"""
|
|
)
|
|
cursor.execute(
|
|
"INSERT INTO pus_tm VALUES(?, ?, ?, ?, ?, ?)",
|
|
(
|
|
str(packet_uuid),
|
|
timestamp.as_datetime(),
|
|
tm_packet.service,
|
|
tm_packet.subservice,
|
|
len(packet),
|
|
packet,
|
|
),
|
|
)
|
|
db_con.commit()
|