diff --git a/master/tg_master.sh b/master/tg_master.sh index 3b38ff8..bed9e14 100644 --- a/master/tg_master.sh +++ b/master/tg_master.sh @@ -1,8 +1,8 @@ #!/bin/bash # ========================================================== -# 脚本名称: tg_master.sh (Master 端调度枢纽 V1.2.1) -# 核心功能: 监听 TG、操作 SQLite、Webhook 调度、僵尸节点清理 +# 脚本名称: tg_master.sh (Master 端调度枢纽 V2.0 模块化适配版) +# 核心功能: 监听 TG、操作 SQLite、Webhook 精准调度、403权限拦截、僵尸节点清理 # ========================================================== CONF="/opt/ip_sentinel_master/master.conf" @@ -45,46 +45,39 @@ while true; do TEXT=$(echo "$UPDATE" | jq -r '.message.text // .callback_query.data') # ========================================== - # 1. 节点注册通道 (处理 Agent 发来的注册暗号) - # 格式: #REGISTER#||| + # 1. 节点注册通道 # ========================================== if [[ "$TEXT" == *"#REGISTER#"* ]]; then - # 提取包含暗号的那一行,并剔除可能误复制的反引号和空格 REG_LINE=$(echo "$TEXT" | grep "#REGISTER#" | head -n 1 | tr -d '\` ') IFS='|' read -r MAGIC NODE_NAME AGENT_IP AGENT_PORT <<< "$REG_LINE" - # UPSERT 逻辑 db_exec "INSERT INTO nodes (chat_id, node_name, agent_ip, agent_port, last_seen) VALUES ('$CHAT_ID', '$NODE_NAME', '$AGENT_IP', '$AGENT_PORT', CURRENT_TIMESTAMP) ON CONFLICT(chat_id, node_name) DO UPDATE SET agent_ip='$AGENT_IP', agent_port='$AGENT_PORT', last_seen=CURRENT_TIMESTAMP;" send_msg "$CHAT_ID" "✅ 司令部已确认!节点接入成功: \`$NODE_NAME\` ($AGENT_IP:$AGENT_PORT)" continue fi # ========================================== - # 2. 交互菜单与下发通道 (主控逻辑) + # 2. 交互菜单与下发通道 # ========================================== case "$TEXT" in - "/start"|"/menu") - # 【升级】新增 [🚀 全节点日报汇总] 按钮 + "/start"|"/menu") BTNS="[[{\"text\":\"🖥️ 我的节点列表\",\"callback_data\":\"list_nodes\"}], [{\"text\":\"🚀 全节点日报汇总\",\"callback_data\":\"all_reports\"}], [{\"text\":\"🛠️ 全节点一键维护\",\"callback_data\":\"all_run\"}]]" send_ui "$CHAT_ID" "🛡️ **IP-Sentinel 司令部**\n欢迎回来,长官。请下达指令:" "$BTNS" ;; "all_reports") - # 【新增】遍历该用户下的所有节点,并发触发 report Webhook NODE_DATA=$(db_exec "SELECT node_name, agent_ip, agent_port FROM nodes WHERE chat_id='$CHAT_ID';") if [ -z "$NODE_DATA" ]; then send_msg "$CHAT_ID" "⚠️ 您名下暂无在线节点。" else send_msg "$CHAT_ID" "📢 **司令部指令下达:正在召唤所有哨兵回传简报...**" echo "$NODE_DATA" | while IFS='|' read -r NNAME AIP APORT; do - # 挂入后台并发请求,防止节点过多时阻塞主循环 curl -s -m 5 "http://${AIP}:${APORT}/trigger_report" > /dev/null & done fi ;; "list_nodes") - # 从 SQLite 查询属于该 CHAT_ID 的节点 NODE_LIST=$(db_exec "SELECT node_name FROM nodes WHERE chat_id='$CHAT_ID';") if [ -z "$NODE_LIST" ]; then send_msg "$CHAT_ID" "⚠️ 您名下暂无在线节点,请先在边缘机执行部署。" @@ -100,18 +93,16 @@ while true; do manage:*) TARGET_NODE=${TEXT#*:} - # 【升级点 1】重构战术面板排版,新增 [剔除失联节点] 按钮,采用 3 行优雅布局 - BTNS="[[{\"text\":\"▶️ 执行深度伪装\",\"callback_data\":\"run:$TARGET_NODE\"}, {\"text\":\"📜 查看实时日志\",\"callback_data\":\"log:$TARGET_NODE\"}], [{\"text\":\"📊 索要统计战报\",\"callback_data\":\"report:$TARGET_NODE\"}, {\"text\":\"🗑️ 剔除失联节点\",\"callback_data\":\"del:$TARGET_NODE\"}], [{\"text\":\"⬅️ 返回主列表\",\"callback_data\":\"list_nodes\"}]]" + # 【核心升级】拆分下发按钮,精准对应 Google 与 Trust 两个模块,并排版为 3 行 2 列 + BTNS="[[{\"text\":\"📍 Google 纠偏\",\"callback_data\":\"google:$TARGET_NODE\"}, {\"text\":\"🛡️ 信用净化\",\"callback_data\":\"trust:$TARGET_NODE\"}], [{\"text\":\"📜 实时日志\",\"callback_data\":\"log:$TARGET_NODE\"}, {\"text\":\"📊 统计战报\",\"callback_data\":\"report:$TARGET_NODE\"}], [{\"text\":\"🗑️ 剔除失联节点\",\"callback_data\":\"del:$TARGET_NODE\"}, {\"text\":\"⬅️ 返回主列表\",\"callback_data\":\"list_nodes\"}]]" send_ui "$CHAT_ID" "⚙️ **目标锁定**: \`$TARGET_NODE\`\n请选择战术动作:" "$BTNS" ;; del:*) - # 【升级点 2】执行数据库硬删除,并自动刷新节点列表 TARGET_NODE=${TEXT#*:} db_exec "DELETE FROM nodes WHERE chat_id='$CHAT_ID' AND node_name='$TARGET_NODE';" send_msg "$CHAT_ID" "🗑️ 节点 \`$TARGET_NODE\` 的档案已从司令部彻底销毁!" - # 删除后自动重新渲染节点列表展示 NODE_LIST=$(db_exec "SELECT node_name FROM nodes WHERE chat_id='$CHAT_ID';") if [ -z "$NODE_LIST" ]; then send_msg "$CHAT_ID" "⚠️ 当前司令部已无任何节点挂载。" @@ -125,11 +116,11 @@ while true; do fi ;; - run:*|report:*|log:*) + # 【核心升级】增加拦截规则,支持 google 和 trust 前缀 + google:*|trust:*|run:*|report:*|log:*) ACTION_TYPE=$(echo "$TEXT" | cut -d':' -f1) TARGET_NODE=$(echo "$TEXT" | cut -d':' -f2) - # 从 DB 提取 IP 和 Port AGENT_INFO=$(db_exec "SELECT agent_ip, agent_port FROM nodes WHERE chat_id='$CHAT_ID' AND node_name='$TARGET_NODE' LIMIT 1;") AGENT_IP=$(echo "$AGENT_INFO" | cut -d'|' -f1) AGENT_PORT=$(echo "$AGENT_INFO" | cut -d'|' -f2) @@ -137,14 +128,20 @@ while true; do if [ -n "$AGENT_IP" ] && [ -n "$AGENT_PORT" ]; then send_msg "$CHAT_ID" "⏳ 正在向 \`$TARGET_NODE\` ($AGENT_IP) 下发 [$ACTION_TYPE] 指令,请稍候..." - # 向 Agent 的开放端口发送动态 Webhook 唤醒指令 + # 触发 Webhook RESPONSE=$(curl -s -m 5 "http://${AGENT_IP}:${AGENT_PORT}/trigger_${ACTION_TYPE}" || echo "FAILED") + # 【核心升级】处理 Agent 传回来的 403 拦截状态码 if [ "$RESPONSE" == "FAILED" ]; then send_msg "$CHAT_ID" "❌ 指令下发超时或失败!请检查节点公网 IP 或防火墙端口 ($AGENT_PORT) 是否放行。" + elif [[ "$RESPONSE" == *"403"* ]]; then + send_msg "$CHAT_ID" "⚠️ **拒绝执行**:该节点未在本地开启此模块,请检查安装时的配置!" else - if [ "$ACTION_TYPE" == "run" ]; then - send_msg "$CHAT_ID" "✅ 节点 \`$TARGET_NODE\` 回应: 指令已接收,伪装程序启动。" + # 细化成功提示 + if [ "$ACTION_TYPE" == "google" ] || [ "$ACTION_TYPE" == "run" ]; then + send_msg "$CHAT_ID" "✅ 节点 \`$TARGET_NODE\` 回应: 📍 Google 纠偏程序启动。" + elif [ "$ACTION_TYPE" == "trust" ]; then + send_msg "$CHAT_ID" "✅ 节点 \`$TARGET_NODE\` 回应: 🛡️ IP 信用净化程序启动。" fi fi else