eive-tmtc/eive_tmtc/tmtc/payload/ploc_memory_dumper.py

# -*- coding: utf-8 -*-
"""
@file   ploc_memory_dumper.py
@brief  This file implements the command to dump memory sectors of the PLOC. Memories of the PLOC
    which can be dumped are one MRAM, two flash memories and the SRAM.
@author J. Meier
@date   31.08.2021
"""
import struct

from eive_tmtc.config.definitions import CustomServiceList
from spacepackets.ecss.tc import PusTelecommand
from tmtccmd.config import TmtcDefinitionWrapper
from tmtccmd.config.tmtc import tmtc_definitions_provider, OpCodeEntry
from tmtccmd.tc import DefaultPusQueueHelper
from tmtccmd.util import ObjectIdU32


class ActionId:
    DUMP_MRAM = 1


@tmtc_definitions_provider
def add_ploc_mem_dumper_cmds(defs: TmtcDefinitionWrapper):
    oce = OpCodeEntry()
    oce.add("0", "PLOC Memory Dumper: MRAM dump")
    defs.add_service(CustomServiceList.PLOC_MEMORY_DUMPER, "PLOC Memory Dumper", oce)


def pack_ploc_memory_dumper_cmd(
    object_id: ObjectIdU32, q: DefaultPusQueueHelper, op_code: str
):
    q.add_log_cmd(
        f"Testing PLOC memory dumper with object id: {object_id.as_hex_string}"
    )

    if op_code == "0":
        q.add_log_cmd("PLOC Supervisor: Dump MRAM")
        command = pack_mram_dump_cmd(object_id.as_bytes)
        q.add_pus_tc(PusTelecommand(service=8, subservice=128, app_data=command))


def pack_mram_dump_cmd(object_id: bytes) -> bytearray:
    start = int(input("Start address: 0x"), 16)
    end = int(input("End address: 0x"), 16)
    command = object_id + struct.pack("!I", ActionId.DUMP_MRAM)
    command = command + struct.pack("!I", start)
    command = command + struct.pack("!I", end)
    return bytearray(command)