From 0d956562cb29849e59f51789628c3e3d5479428a Mon Sep 17 00:00:00 2001 From: ReneKlever Date: Mon, 24 Mar 2025 20:56:41 +0100 Subject: [PATCH] Update command_handlers.py adde security check before enter the bbs --- command_handlers.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/command_handlers.py b/command_handlers.py index 5696d0f..3b76a72 100644 --- a/command_handlers.py +++ b/command_handlers.py @@ -61,11 +61,16 @@ def handle_help_command(sender_id, interface, menu_name=None): response = build_menu(bbs_menu_items, "📰BBS Menu📰") elif menu_name == 'utilities': response = build_menu(utilities_menu_items, "🛠️Utilities Menu🛠️") + send_message(response, sender_id, interface) else: - update_user_state(sender_id, {'command': 'MAIN_MENU', 'step': 1}) # Reset to main menu state - mail = get_mail(get_node_id_from_num(sender_id, interface)) - response = build_menu(main_menu_items, f"💾TC² BBS💾 (✉️:{len(mail)})") - send_message(response, sender_id, interface) + node_id = get_node_id_from_num(sender_id, interface) + allowed2bbs_nodes = interface.allowed2bbs_nodes + if allowed2bbs_nodes and node_id not in allowed2bbs_nodes: + logging.info(f"The node_id: {node_id} is not allowed to enter this bbs") # Debug statement + elif update_user_state(sender_id, {'command': 'MAIN_MENU', 'step': 1}) # Reset to main menu state + mail = get_mail(get_node_id_from_num(sender_id, interface)) + response = build_menu(main_menu_items, f"💾TC² BBS💾 (✉️:{len(mail)})") + send_message(response, sender_id, interface) def get_node_name(node_id, interface): node_info = interface.nodes.get(node_id)