256 lines
8.3 KiB
Python
256 lines
8.3 KiB
Python
import enum
|
|
|
|
from config.definitions import CustomServiceList
|
|
from tmtccmd.config import TmtcDefinitionWrapper
|
|
|
|
from tmtccmd.tc import DefaultPusQueueHelper
|
|
from tmtccmd.tc.pus_8_funccmd import make_fsfw_action_cmd
|
|
from tmtccmd.logging import get_console_logger
|
|
from tmtccmd.tc.pus_3_fsfw_hk import make_sid, generate_one_hk_command
|
|
from tmtccmd.config.tmtc import OpCodeEntry, tmtc_definitions_provider
|
|
from config.object_ids import CORE_CONTROLLER_ID
|
|
|
|
|
|
LOGGER = get_console_logger()
|
|
|
|
|
|
class ActionIds(enum.IntEnum):
|
|
LIST_DIR_INTO_FILE = 0
|
|
SWITCH_REBOOT_FILE_HANDLING = 5
|
|
RESET_ALL_REBOOT_COUNTERS = 6
|
|
RESET_REBOOT_COUNTER_00 = 7
|
|
RESET_REBOOT_COUNTER_01 = 8
|
|
RESET_REBOOT_COUNTER_10 = 9
|
|
RESET_REBOOT_COUNTER_11 = 10
|
|
SET_MAX_REBOOT_CNT = 11
|
|
XSC_REBOOT = 32
|
|
FULL_REBOOT = 34
|
|
|
|
|
|
class SetIds(enum.IntEnum):
|
|
HK = 5
|
|
|
|
|
|
class OpCodes:
|
|
REBOOT_XSC = ["0", "reboot_xsc"]
|
|
XSC_REBOOT_SELF = ["1", "reboot_self"]
|
|
XSC_REBOOT_0_0 = ["2", "reboot_00"]
|
|
XSC_REBOOT_0_1 = ["3", "reboot_01"]
|
|
XSC_REBOOT_1_0 = ["4", "reboot_10"]
|
|
XSC_REBOOT_1_1 = ["5", "reboot_11"]
|
|
REBOOT_FULL = ["6", "reboot_regular"]
|
|
GET_HK = ["7", "get_hk"]
|
|
ENABLE_REBOOT_FILE_HANDLING = ["32", "rbh_off"]
|
|
DISABLE_REBOOT_FILE_HANDLING = ["33", "rbh_on"]
|
|
RESET_ALL_REBOOT_COUNTERS = ["34", "rbh_reset_a"]
|
|
RESET_REBOOT_COUNTER_00 = ["35", "rbh_reset_00"]
|
|
RESET_REBOOT_COUNTER_01 = ["36", "rbh_reset_01"]
|
|
RESET_REBOOT_COUNTER_10 = ["37", "rbh_reset_10"]
|
|
RESET_REBOOT_COUNTER_11 = ["38", "rbh_reset_11"]
|
|
SET_MAX_REBOOT_CNT = ["39", "rbh_max_cnt"]
|
|
|
|
|
|
class Info:
|
|
REBOOT_XSC = "XSC reboot with prompt"
|
|
REBOOT_FULL = "Full regular reboot"
|
|
|
|
|
|
class Chip(enum.IntEnum):
|
|
CHIP_0 = 0
|
|
CHIP_1 = 1
|
|
NONE = 2
|
|
|
|
|
|
class Copy(enum.IntEnum):
|
|
COPY_0_NOM = 0
|
|
COPY_1_GOLD = 1
|
|
NONE = 2
|
|
|
|
|
|
@tmtc_definitions_provider
|
|
def add_core_controller_definitions(defs: TmtcDefinitionWrapper):
|
|
oce = OpCodeEntry()
|
|
oce.add(keys=OpCodes.REBOOT_XSC, info=Info.REBOOT_XSC)
|
|
oce.add(keys=OpCodes.REBOOT_XSC, info=Info.REBOOT_XSC)
|
|
oce.add(keys=OpCodes.REBOOT_FULL, info=Info.REBOOT_FULL)
|
|
oce.add(keys=OpCodes.XSC_REBOOT_SELF, info="Reboot Self")
|
|
oce.add(keys=OpCodes.XSC_REBOOT_0_0, info="Reboot 0 0")
|
|
oce.add(keys=OpCodes.XSC_REBOOT_0_1, info="Reboot 0 1")
|
|
oce.add(keys=OpCodes.XSC_REBOOT_1_0, info="Reboot 1 0")
|
|
oce.add(keys=OpCodes.XSC_REBOOT_1_1, info="Reboot 1 1")
|
|
oce.add(
|
|
keys=OpCodes.GET_HK,
|
|
info="Request housekeeping set",
|
|
)
|
|
oce.add(
|
|
keys=OpCodes.ENABLE_REBOOT_FILE_HANDLING,
|
|
info="Enable reboot file handling",
|
|
)
|
|
oce.add(
|
|
keys=OpCodes.DISABLE_REBOOT_FILE_HANDLING,
|
|
info="Disable reboot file handling",
|
|
)
|
|
oce.add(
|
|
keys=OpCodes.RESET_ALL_REBOOT_COUNTERS,
|
|
info="Reset all reboot counters",
|
|
)
|
|
oce.add(
|
|
keys=OpCodes.RESET_REBOOT_COUNTER_00,
|
|
info="Reset reboot counter 0 0",
|
|
)
|
|
oce.add(
|
|
keys=OpCodes.RESET_REBOOT_COUNTER_01,
|
|
info="Reset reboot counter 0 1",
|
|
)
|
|
oce.add(
|
|
keys=OpCodes.RESET_REBOOT_COUNTER_10,
|
|
info="Reset reboot counter 1 0",
|
|
)
|
|
oce.add(
|
|
keys=OpCodes.RESET_REBOOT_COUNTER_11,
|
|
info="Reset reboot counter 1 1",
|
|
)
|
|
defs.add_service(CustomServiceList.CORE.value, "Core Controller", oce)
|
|
|
|
|
|
def pack_core_commands(q: DefaultPusQueueHelper, op_code: str):
|
|
if op_code in OpCodes.REBOOT_XSC:
|
|
reboot_self, chip_select, copy_select = determine_reboot_params()
|
|
perform_reboot_cmd(
|
|
q=q,
|
|
reboot_self=reboot_self,
|
|
chip=chip_select,
|
|
copy=copy_select,
|
|
)
|
|
if op_code in OpCodes.REBOOT_FULL:
|
|
q.add_log_cmd(f"Core Command: {Info.REBOOT_FULL}")
|
|
q.add_pus_tc(
|
|
make_fsfw_action_cmd(
|
|
object_id=CORE_CONTROLLER_ID, action_id=ActionIds.FULL_REBOOT
|
|
)
|
|
)
|
|
if op_code in OpCodes.XSC_REBOOT_SELF:
|
|
perform_reboot_cmd(q=q, reboot_self=True)
|
|
if op_code in OpCodes.XSC_REBOOT_0_0:
|
|
perform_reboot_cmd(
|
|
q=q, reboot_self=False, chip=Chip.CHIP_0, copy=Copy.COPY_0_NOM
|
|
)
|
|
if op_code in OpCodes.XSC_REBOOT_0_1:
|
|
perform_reboot_cmd(
|
|
q=q,
|
|
reboot_self=False,
|
|
chip=Chip.CHIP_0,
|
|
copy=Copy.COPY_1_GOLD,
|
|
)
|
|
if op_code in OpCodes.XSC_REBOOT_1_0:
|
|
perform_reboot_cmd(
|
|
q=q, reboot_self=False, chip=Chip.CHIP_1, copy=Copy.COPY_0_NOM
|
|
)
|
|
if op_code in OpCodes.XSC_REBOOT_1_1:
|
|
perform_reboot_cmd(
|
|
q=q,
|
|
reboot_self=False,
|
|
chip=Chip.CHIP_1,
|
|
copy=Copy.COPY_1_GOLD,
|
|
)
|
|
if op_code in OpCodes.DISABLE_REBOOT_FILE_HANDLING:
|
|
q.add_log_cmd("Disabling reboot file handling")
|
|
app_data = bytearray([0])
|
|
make_fsfw_action_cmd(
|
|
object_id=CORE_CONTROLLER_ID,
|
|
action_id=ActionIds.SWITCH_REBOOT_FILE_HANDLING,
|
|
user_data=app_data,
|
|
)
|
|
if op_code in OpCodes.ENABLE_REBOOT_FILE_HANDLING:
|
|
q.add_log_cmd("Enabling reboot file handling")
|
|
app_data = bytearray([1])
|
|
make_fsfw_action_cmd(
|
|
object_id=CORE_CONTROLLER_ID,
|
|
action_id=ActionIds.SWITCH_REBOOT_FILE_HANDLING,
|
|
user_data=app_data,
|
|
)
|
|
if op_code in OpCodes.RESET_ALL_REBOOT_COUNTERS:
|
|
q.add_log_cmd("Resetting all reboot counters")
|
|
make_fsfw_action_cmd(
|
|
object_id=CORE_CONTROLLER_ID, action_id=ActionIds.RESET_ALL_REBOOT_COUNTERS
|
|
)
|
|
if op_code in OpCodes.RESET_REBOOT_COUNTER_00:
|
|
q.add_log_cmd("Resetting reboot counter 0 0")
|
|
make_fsfw_action_cmd(
|
|
object_id=CORE_CONTROLLER_ID, action_id=ActionIds.RESET_REBOOT_COUNTER_00
|
|
)
|
|
if op_code in OpCodes.RESET_REBOOT_COUNTER_01:
|
|
q.add_log_cmd("Resetting reboot counter 0 1")
|
|
make_fsfw_action_cmd(
|
|
object_id=CORE_CONTROLLER_ID, action_id=ActionIds.RESET_REBOOT_COUNTER_01
|
|
)
|
|
if op_code in OpCodes.RESET_REBOOT_COUNTER_10:
|
|
q.add_log_cmd("Resetting reboot counter 1 0")
|
|
make_fsfw_action_cmd(
|
|
object_id=CORE_CONTROLLER_ID, action_id=ActionIds.RESET_REBOOT_COUNTER_10
|
|
)
|
|
if op_code in OpCodes.RESET_REBOOT_COUNTER_11:
|
|
q.add_log_cmd("Resetting reboot counter 1 1")
|
|
make_fsfw_action_cmd(
|
|
object_id=CORE_CONTROLLER_ID, action_id=ActionIds.RESET_REBOOT_COUNTER_11
|
|
)
|
|
if op_code in OpCodes.GET_HK:
|
|
q.add_log_cmd("Requesting housekeeping set")
|
|
sid = make_sid(object_id=CORE_CONTROLLER_ID, set_id=SetIds.HK)
|
|
q.add_pus_tc(generate_one_hk_command(sid))
|
|
|
|
|
|
def determine_reboot_params() -> (bool, Chip, Copy):
|
|
chip_select = -1
|
|
copy_select = -1
|
|
reboot_self = input("Reboot self? [y/n]: ")
|
|
if reboot_self in ["y", "yes", "1"]:
|
|
LOGGER.info("Rebooting currently running image")
|
|
return True, chip_select, copy_select
|
|
LOGGER.info("Rebooting image specified by chip and copy")
|
|
while True:
|
|
chip_select = input("Chip select [0/1]: ")
|
|
if chip_select in ["0", "1"]:
|
|
if chip_select == "0":
|
|
chip_select = Chip.CHIP_0
|
|
else:
|
|
chip_select = Chip.CHIP_1
|
|
break
|
|
else:
|
|
LOGGER.warning("Invalid chip select value. Try again")
|
|
while True:
|
|
copy_select = input("Copy select [0/1]: ")
|
|
if copy_select in ["0", "1"]:
|
|
if copy_select == "0":
|
|
copy_select = Copy.COPY_0_NOM
|
|
else:
|
|
copy_select = Copy.COPY_1_GOLD
|
|
break
|
|
else:
|
|
LOGGER.warning("Invalid copy select value. Try again")
|
|
return False, chip_select, copy_select
|
|
|
|
|
|
def perform_reboot_cmd(
|
|
q: DefaultPusQueueHelper,
|
|
reboot_self: bool,
|
|
chip: Chip = Chip.NONE,
|
|
copy: Copy = Copy.NONE,
|
|
):
|
|
tc_data = bytearray()
|
|
if reboot_self:
|
|
q.add_log_cmd("Packing reboot command for current image")
|
|
tc_data.append(True)
|
|
else:
|
|
tc_data.append(False)
|
|
tc_data.append(chip)
|
|
tc_data.append(copy)
|
|
q.add_log_cmd(f"Packing reboot command for chip {chip} and copy {copy}")
|
|
q.add_pus_tc(
|
|
make_fsfw_action_cmd(
|
|
object_id=CORE_CONTROLLER_ID,
|
|
action_id=ActionIds.XSC_REBOOT,
|
|
user_data=tc_data,
|
|
)
|
|
)
|