add linux support

2025-10-03 17:29:58 +08:00
parent 4b8e3da812
commit 82eeee50f1
6 changed files with 1932 additions and 1 deletions
--- a/deploy_system.sh
+++ b/deploy_system.sh
@@ -0,0 +1,528 @@
+#!/bin/bash
+
+#
+# 智能网络监控脚本 - Ubuntu系统级部署脚本
+#
+# 功能描述：
+# 将脚本部署到系统级目录，配置为systemd服务，开机自启动
+#
+# 部署位置：
+# - 程序文件: /opt/smart-network-monitor/
+# - 配置文件: /etc/smart-network-monitor/
+# - 日志文件: /var/log/smart-network-monitor/
+# - 服务文件: /etc/systemd/system/smart-network-monitor.service
+#
+# 使用方法：
+# sudo ./deploy_system.sh
+#
+
+# 设置颜色输出
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+CYAN='\033[0;36m'
+NC='\033[0m' # No Color
+
+# 定义系统级路径
+PROGRAM_PATH="/opt/smart-network-monitor"
+CONFIG_PATH="/etc/smart-network-monitor"
+LOG_PATH="/var/log/smart-network-monitor"
+SERVICE_FILE="/etc/systemd/system/smart-network-monitor.service"
+SERVICE_NAME="smart-network-monitor"
+
+# 当前脚本目录
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+
+echo -e "${CYAN}==========================================${NC}"
+echo -e "${CYAN}智能网络监控脚本 - Ubuntu系统级部署${NC}"
+echo -e "${CYAN}==========================================${NC}"
+echo ""
+
+echo -e "${YELLOW}部署计划:${NC}"
+echo -e "- 程序目录: ${PROGRAM_PATH}"
+echo -e "- 配置目录: ${CONFIG_PATH}"
+echo -e "- 日志目录: ${LOG_PATH}"
+echo -e "- 服务文件: ${SERVICE_FILE}"
+echo ""
+
+# 检查root权限
+if [[ $EUID -ne 0 ]]; then
+    echo -e "${RED}错误：此脚本需要root权限执行${NC}"
+    echo -e "${YELLOW}请使用: sudo $0${NC}"
+    exit 1
+fi
+
+# 检查源文件
+REQUIRED_FILES=(
+    "smart_shutdown.sh"
+)
+
+for file in "${REQUIRED_FILES[@]}"; do
+    if [[ ! -f "$SCRIPT_DIR/$file" ]]; then
+        echo -e "${RED}错误：找不到必需文件: $file${NC}"
+        exit 1
+    fi
+done
+
+echo -e "${GREEN}[OK] 源文件检查完成${NC}"
+
+# 停止已存在的服务
+if systemctl is-active --quiet "$SERVICE_NAME"; then
+    echo -e "${YELLOW}正在停止现有服务...${NC}"
+    systemctl stop "$SERVICE_NAME"
+    echo -e "${GREEN}[OK] 服务已停止${NC}"
+fi
+
+# 创建目录结构
+echo -e "${YELLOW}正在创建目录结构...${NC}"
+
+# 创建程序目录
+if [[ ! -d "$PROGRAM_PATH" ]]; then
+    mkdir -p "$PROGRAM_PATH"
+    echo -e "${GREEN}[OK] 创建程序目录: $PROGRAM_PATH${NC}"
+fi
+
+# 创建配置目录
+if [[ ! -d "$CONFIG_PATH" ]]; then
+    mkdir -p "$CONFIG_PATH"
+    echo -e "${GREEN}[OK] 创建配置目录: $CONFIG_PATH${NC}"
+fi
+
+# 创建日志目录
+if [[ ! -d "$LOG_PATH" ]]; then
+    mkdir -p "$LOG_PATH"
+    # 设置适当的权限，允许服务用户写入
+    chmod 755 "$LOG_PATH"
+    echo -e "${GREEN}[OK] 创建日志目录: $LOG_PATH${NC}"
+fi
+
+# 创建系统优化版的主脚本
+echo -e "${YELLOW}正在创建系统版本的脚本...${NC}"
+
+cat > "$PROGRAM_PATH/smart_shutdown_system.sh" << 'EOF'
+#!/bin/bash
+
+#
+# 智能网络监控脚本 - 系统服务版本
+#
+# 功能描述：
+# 系统级网络监控脚本，在网络持续中断时自动关机
+#
+# 特点：
+# - 开机自启动，作为systemd服务运行
+# - 日志存储在系统日志目录
+# - 优化的错误处理和权限管理
+#
+
+# ==================== 系统级配置参数 ====================
+
+# 默认配置参数
+TARGET_IP="192.168.3.3"
+NORMAL_PING_INTERVAL=15
+MONITOR_WINDOW_SECONDS=180
+SHUTDOWN_COUNTDOWN=60
+COUNTDOWN_PING_INTERVAL=3
+PING_TIMEOUT=3
+
+# 系统级路径配置
+CONFIG_FILE="/etc/smart-network-monitor/config.json"
+LOG_DIRECTORY="/var/log/smart-network-monitor"
+MAX_LOG_DAYS=30
+
+# 正常连接时的日志记录间隔计数器
+NORMAL_LOG_INTERVAL=24  # 每24次循环记录一次状态 (约6分钟)
+NORMAL_LOG_COUNTER=0
+
+# ==================== 函数定义 ====================
+
+# 日志写入函数
+write_log() {
+    local level="$1"
+    local message="$2"
+    local timestamp=$(date '+%Y-%m-%d %H:%M:%S')
+    local log_file="$LOG_DIRECTORY/network_monitor_$(date '+%Y%m%d').log"
+    local log_entry="[$timestamp] [$level] $message"
+    
+    # 确保日志目录存在
+    mkdir -p "$LOG_DIRECTORY"
+    
+    # 写入日志文件
+    echo "$log_entry" >> "$log_file" 2>/dev/null
+    
+    # 同时写入系统日志
+    logger -t "smart-network-monitor" "$log_entry"
+    
+    # 对于关键信息，输出到标准输出（systemd会捕获）
+    if [[ "$level" == "CRITICAL" || "$level" == "SUCCESS" || "$level" == "WARN" ]]; then
+        echo "$log_entry"
+    fi
+}
+
+# 加载配置文件
+load_configuration() {
+    if [[ -f "$CONFIG_FILE" ]]; then
+        write_log "INFO" "加载配置文件: $CONFIG_FILE"
+        
+        # 使用jq解析JSON配置文件（如果可用）
+        if command -v jq >/dev/null 2>&1; then
+            local target_ip=$(jq -r '.TargetIP // empty' "$CONFIG_FILE" 2>/dev/null)
+            local monitor_window=$(jq -r '.MonitorWindowSeconds // empty' "$CONFIG_FILE" 2>/dev/null)
+            local shutdown_countdown=$(jq -r '.ShutdownCountdown // empty' "$CONFIG_FILE" 2>/dev/null)
+            local ping_interval=$(jq -r '.NormalPingInterval // empty' "$CONFIG_FILE" 2>/dev/null)
+            
+            [[ -n "$target_ip" ]] && TARGET_IP="$target_ip"
+            [[ -n "$monitor_window" ]] && MONITOR_WINDOW_SECONDS="$monitor_window"
+            [[ -n "$shutdown_countdown" ]] && SHUTDOWN_COUNTDOWN="$shutdown_countdown"
+            [[ -n "$ping_interval" ]] && NORMAL_PING_INTERVAL="$ping_interval"
+            
+            write_log "SUCCESS" "配置文件加载成功"
+        else
+            write_log "WARN" "jq未安装，无法解析JSON配置文件，使用默认配置"
+        fi
+    else
+        write_log "WARN" "配置文件不存在，使用默认配置"
+    fi
+}
+
+# 测试网络连接
+test_network_connection() {
+    local target_ip="$1"
+    ping -c 1 -W "$PING_TIMEOUT" "$target_ip" >/dev/null 2>&1
+    return $?
+}
+
+# 清理旧日志文件
+cleanup_old_logs() {
+    if [[ -d "$LOG_DIRECTORY" ]]; then
+        find "$LOG_DIRECTORY" -name "network_monitor_*.log" -type f -mtime +$MAX_LOG_DAYS -exec rm -f {} \; 2>/dev/null
+    fi
+}
+
+# 信号处理函数
+cleanup_and_exit() {
+    write_log "INFO" "接收到退出信号，正在清理..."
+    write_log "INFO" "========== 服务退出 =========="
+    exit 0
+}
+
+# ==================== 主程序开始 ====================
+
+# 设置信号处理
+trap cleanup_and_exit SIGINT SIGTERM
+
+# 清理旧日志
+cleanup_old_logs
+
+# 加载配置
+load_configuration
+
+# 获取系统信息
+running_user=$(whoami)
+local_ip=$(ip route get 8.8.8.8 2>/dev/null | awk '{for(i=1;i<=NF;i++) if($i=="src") print $(i+1)}' | head -1)
+[[ -z "$local_ip" ]] && local_ip="未知"
+
+write_log "INFO" "========== 系统级网络监控服务启动 =========="
+write_log "INFO" "运行用户: $running_user"
+write_log "INFO" "本机IP: $local_ip"
+write_log "INFO" "监控目标: $TARGET_IP"
+write_log "INFO" "监控窗口: ${MONITOR_WINDOW_SECONDS}秒"
+write_log "INFO" "关机倒计时: ${SHUTDOWN_COUNTDOWN}秒"
+write_log "INFO" "日志目录: $LOG_DIRECTORY"
+write_log "INFO" "配置文件: $CONFIG_FILE"
+
+# 主循环
+failure_start_time=""
+
+while true; do
+    # 测试网络连接
+    if test_network_connection "$TARGET_IP"; then
+        # 网络连接正常
+        if [[ -n "$failure_start_time" ]]; then
+            write_log "SUCCESS" "网络连接已恢复"
+            failure_start_time=""
+            NORMAL_LOG_COUNTER=0
+        else
+            # 减少正常连接时的日志频率
+            ((NORMAL_LOG_COUNTER++))
+            if [[ $NORMAL_LOG_COUNTER -ge $NORMAL_LOG_INTERVAL ]]; then
+                write_log "INFO" "网络连接正常（定期状态报告）"
+                NORMAL_LOG_COUNTER=0
+            fi
+        fi
+        
+        sleep "$NORMAL_PING_INTERVAL"
+        
+    else
+        # 网络连接失败
+        write_log "FAIL" "Ping失败 - 目标: $TARGET_IP"
+        
+        if [[ -z "$failure_start_time" ]]; then
+            failure_start_time=$(date +%s)
+            write_log "WARN" "网络首次中断，开始进入监控窗口计时"
+        fi
+        
+        current_time=$(date +%s)
+        failure_duration=$((current_time - failure_start_time))
+        write_log "INFO" "网络已持续中断 $failure_duration / ${MONITOR_WINDOW_SECONDS} 秒"
+        
+        if [[ $failure_duration -ge $MONITOR_WINDOW_SECONDS ]]; then
+            write_log "CRITICAL" "网络持续中断已超过 ${MONITOR_WINDOW_SECONDS} 秒，开始关机流程"
+            
+            shutdown_cancelled=false
+            countdown_end_time=$((current_time + SHUTDOWN_COUNTDOWN))
+            
+            while [[ $(date +%s) -lt $countdown_end_time && "$shutdown_cancelled" == "false" ]]; do
+                current_time=$(date +%s)
+                remaining_seconds=$((countdown_end_time - current_time))
+                
+                if [[ $remaining_seconds -le 0 ]]; then
+                    break
+                fi
+                
+                write_log "WARN" "距离关机还有 $remaining_seconds 秒... 正在快速检测网络"
+                
+                # 在倒计时中再次检测网络
+                if test_network_connection "$TARGET_IP"; then
+                    write_log "SUCCESS" "网络在倒计时期间恢复！取消关机"
+                    failure_start_time=""
+                    shutdown_cancelled=true
+                    break
+                fi
+                
+                sleep "$COUNTDOWN_PING_INTERVAL"
+            done
+            
+            if [[ "$shutdown_cancelled" == "false" ]]; then
+                write_log "CRITICAL" "关机倒计时完成，执行系统关机命令"
+                
+                # 执行实际关机命令
+                shutdown -h now
+                write_log "CRITICAL" "关机命令已执行"
+                
+                exit 0
+            fi
+        else
+            sleep "$NORMAL_PING_INTERVAL"
+        fi
+    fi
+done
+EOF
+
+# 设置脚本权限
+chmod +x "$PROGRAM_PATH/smart_shutdown_system.sh"
+echo -e "${GREEN}[OK] 创建系统脚本: $PROGRAM_PATH/smart_shutdown_system.sh${NC}"
+
+# 创建默认配置文件
+echo -e "${YELLOW}正在创建配置文件...${NC}"
+
+cat > "$CONFIG_PATH/config.json" << EOF
+{
+    "TargetIP": "192.168.3.3",
+    "MonitorWindowSeconds": 180,
+    "ShutdownCountdown": 60,
+    "NormalPingInterval": 15
+}
+EOF
+
+echo -e "${GREEN}[OK] 创建配置文件: $CONFIG_PATH/config.json${NC}"
+
+# 创建systemd服务文件
+echo -e "${YELLOW}正在配置systemd服务...${NC}"
+
+cat > "$SERVICE_FILE" << EOF
+[Unit]
+Description=Smart Network Monitor Service
+Documentation=https://github.com/example/smart-network-monitor
+After=network-online.target
+Wants=network-online.target
+
+[Service]
+Type=simple
+User=root
+Group=root
+ExecStart=$PROGRAM_PATH/smart_shutdown_system.sh
+Restart=always
+RestartSec=10
+StandardOutput=journal
+StandardError=journal
+
+# 安全设置
+NoNewPrivileges=false
+PrivateTmp=true
+ProtectHome=true
+ProtectSystem=strict
+ReadWritePaths=$LOG_PATH
+
+# 环境变量
+Environment=PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
+
+[Install]
+WantedBy=multi-user.target
+EOF
+
+echo -e "${GREEN}[OK] 创建systemd服务文件: $SERVICE_FILE${NC}"
+
+# 重新加载systemd配置
+echo -e "${YELLOW}正在重新加载systemd配置...${NC}"
+systemctl daemon-reload
+echo -e "${GREEN}[OK] systemd配置已重新加载${NC}"
+
+# 启用服务
+echo -e "${YELLOW}正在启用服务开机自启动...${NC}"
+systemctl enable "$SERVICE_NAME"
+echo -e "${GREEN}[OK] 服务已设置为开机自启动${NC}"
+
+# 创建管理脚本
+echo -e "${YELLOW}正在创建管理工具...${NC}"
+
+cat > "$PROGRAM_PATH/manage.sh" << 'EOF'
+#!/bin/bash
+
+#
+# 智能网络监控脚本 - 管理工具
+#
+
+# 颜色定义
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+CYAN='\033[0;36m'
+NC='\033[0m'
+
+SERVICE_NAME="smart-network-monitor"
+CONFIG_FILE="/etc/smart-network-monitor/config.json"
+LOG_PATH="/var/log/smart-network-monitor"
+
+echo -e "${CYAN}智能网络监控脚本 - 管理工具${NC}"
+echo -e "${CYAN}==============================${NC}"
+echo ""
+
+echo -e "${YELLOW}可用操作:${NC}"
+echo "1. 查看服务状态"
+echo "2. 启动服务"
+echo "3. 停止服务"
+echo "4. 重启服务"
+echo "5. 查看今天的日志"
+echo "6. 查看实时日志"
+echo "7. 查看配置文件"
+echo "8. 编辑配置文件"
+echo "9. 查看系统日志"
+echo ""
+
+read -p "请选择操作 (1-9): " choice
+
+case "$choice" in
+    "1")
+        systemctl status "$SERVICE_NAME"
+        ;;
+    "2")
+        sudo systemctl start "$SERVICE_NAME"
+        echo -e "${GREEN}服务已启动${NC}"
+        ;;
+    "3")
+        sudo systemctl stop "$SERVICE_NAME"
+        echo -e "${GREEN}服务已停止${NC}"
+        ;;
+    "4")
+        sudo systemctl restart "$SERVICE_NAME"
+        echo -e "${GREEN}服务已重启${NC}"
+        ;;
+    "5")
+        log_file="$LOG_PATH/network_monitor_$(date '+%Y%m%d').log"
+        if [[ -f "$log_file" ]]; then
+            cat "$log_file"
+        else
+            echo -e "${RED}今天的日志文件不存在${NC}"
+        fi
+        ;;
+    "6")
+        log_file="$LOG_PATH/network_monitor_$(date '+%Y%m%d').log"
+        if [[ -f "$log_file" ]]; then
+            tail -f "$log_file"
+        else
+            echo -e "${RED}今天的日志文件不存在，显示systemd日志：${NC}"
+            journalctl -u "$SERVICE_NAME" -f
+        fi
+        ;;
+    "7")
+        if [[ -f "$CONFIG_FILE" ]]; then
+            cat "$CONFIG_FILE"
+        else
+            echo -e "${RED}配置文件不存在${NC}"
+        fi
+        ;;
+    "8")
+        if [[ -f "$CONFIG_FILE" ]]; then
+            sudo nano "$CONFIG_FILE"
+            echo -e "${YELLOW}配置已修改，重启服务使配置生效：${NC}"
+            echo "sudo systemctl restart $SERVICE_NAME"
+        else
+            echo -e "${RED}配置文件不存在${NC}"
+        fi
+        ;;
+    "9")
+        journalctl -u "$SERVICE_NAME" -n 50
+        ;;
+    *)
+        echo -e "${RED}无效选择${NC}"
+        ;;
+esac
+
+echo ""
+echo -e "${YELLOW}常用命令:${NC}"
+echo -e "${BLUE}查看服务状态:${NC} systemctl status $SERVICE_NAME"
+echo -e "${BLUE}启动服务:${NC} sudo systemctl start $SERVICE_NAME"
+echo -e "${BLUE}停止服务:${NC} sudo systemctl stop $SERVICE_NAME"
+echo -e "${BLUE}重启服务:${NC} sudo systemctl restart $SERVICE_NAME"
+echo -e "${BLUE}查看今天日志:${NC} cat $LOG_PATH/network_monitor_\$(date '+%Y%m%d').log"
+echo -e "${BLUE}实时查看日志:${NC} tail -f $LOG_PATH/network_monitor_\$(date '+%Y%m%d').log"
+echo -e "${BLUE}查看系统日志:${NC} journalctl -u $SERVICE_NAME -f"
+echo -e "${BLUE}编辑配置:${NC} sudo nano $CONFIG_FILE"
+echo ""
+EOF
+
+chmod +x "$PROGRAM_PATH/manage.sh"
+echo -e "${GREEN}[OK] 创建管理脚本: $PROGRAM_PATH/manage.sh${NC}"
+
+# 创建符号链接到系统PATH
+if [[ ! -L "/usr/local/bin/smart-monitor" ]]; then
+    ln -s "$PROGRAM_PATH/manage.sh" "/usr/local/bin/smart-monitor"
+    echo -e "${GREEN}[OK] 创建管理工具快捷命令: smart-monitor${NC}"
+fi
+
+echo ""
+echo -e "${GREEN}[INFO] 系统级部署完成！${NC}"
+echo ""
+echo -e "${CYAN}部署摘要:${NC}"
+echo -e "${GREEN}[OK]${NC} 程序文件已部署到: $PROGRAM_PATH"
+echo -e "${GREEN}[OK]${NC} 配置目录已创建: $CONFIG_PATH"
+echo -e "${GREEN}[OK]${NC} 日志目录已创建: $LOG_PATH"
+echo -e "${GREEN}[OK]${NC} systemd服务已配置: $SERVICE_NAME"
+echo -e "${GREEN}[OK]${NC} 开机自启动已启用"
+echo ""
+echo -e "${YELLOW}管理工具:${NC}"
+echo -e "- 管理脚本: $PROGRAM_PATH/manage.sh"
+echo -e "- 快捷命令: ${CYAN}smart-monitor${NC}"
+echo -e "- 配置文件: $CONFIG_PATH/config.json"
+echo -e "- 今天日志: $LOG_PATH/network_monitor_$(date '+%Y%m%d').log"
+echo ""
+echo -e "${GREEN}下次重启后，服务将自动运行！${NC}"
+echo ""
+
+# 询问是否立即启动
+read -p "是否立即启动监控服务？(Y/n): " response
+if [[ "$response" == "Y" || "$response" == "y" || "$response" == "" ]]; then
+    if systemctl start "$SERVICE_NAME"; then
+        echo -e "${GREEN}[OK] 监控服务已启动${NC}"
+        echo -e "${CYAN}使用以下命令查看服务状态:${NC}"
+        echo -e "${BLUE}systemctl status $SERVICE_NAME${NC}"
+        echo -e "${BLUE}journalctl -u $SERVICE_NAME -f${NC}"
+    else
+        echo -e "${RED}[X] 启动服务失败${NC}"
+    fi
+fi
+
+echo ""
+echo -e "${CYAN}使用 'smart-monitor' 命令来管理服务${NC}"