Update 02

02

@@ -1,300 +1,213 @@
 #!/bin/bash
 
-# 完整客户端控制器 v3.0
-# 支持自动连接、命令执行、开机自启
+# 多方案端口检测脚本
+# 用法: ./port_check.sh IP地址 端口号 [超时时间]
 
-SERVER_IP="159.138.58.239"
-SERVER_PORT=25555
-CLIENT_PORT=5556
-HEARTBEAT_INTERVAL=30
-CONFIG_FILE="/tmp/controller_client.conf"
-PID_FILE="/tmp/controller_client.pid"
+TARGET_HOST="${1}"
+TARGET_PORT="${2}"
+TIMEOUT="${3:-5}"  # 默认超时5秒
 
 # 颜色定义
 RED='\033[0;31m'
 GREEN='\033[0;32m'
 YELLOW='\033[1;33m'
 BLUE='\033[0;34m'
-NC='\033[0m'
-
-# 初始化客户端
-init_client() {
-    # 生成序列号
-    if [[ -r /etc/machine-id ]]; then
-        SERIAL=$(cat /etc/machine-id | md5sum | cut -c1-8)
-    else
-        SERIAL=$(date +%s | md5sum | cut -c1-8)
-    fi
-    
-    HOSTNAME=$(hostname 2>/dev/null || echo "unknown")
-    
-    # 获取系统信息
-    if [[ -r /etc/os-release ]]; then
-        source /etc/os-release
-        SYSTEM_INFO="${ID:-unknown} ${VERSION_ID:-unknown}"
-    else
-        SYSTEM_INFO="unknown"
-    fi
-    
-    echo "=== 客户端控制器 v3.0 ==="
-    echo "序列号: $SERIAL"
-    echo "服务器: $SERVER_IP:$SERVER_PORT"
-    echo "系统: $SYSTEM_INFO"
-}
+NC='\033[0m' # No Color
 
 # 日志函数
-log() {
-    local timestamp=$(date '+%Y-%m-%d %H:%M:%S')
-    echo -e "[$timestamp] $1" >&2
+log_info() {
+    echo -e "${BLUE}[INFO]${NC} $1"
 }
 
-# Bash TCP发送函数
-bash_tcp_send() {
-    local data="$1"
-    local ip="$2"
-    local port="$3"
-    
-    {
-        exec 3<>/dev/tcp/$ip/$port 2>/dev/null
-        [[ $? -ne 0 ]] && return 1
-        
-        echo "$data" >&3 2>/dev/null
-        sleep 0.5
-        exec 3>&-
-        exec 3<&-
-        return 0
-    } 2>/dev/null
+log_success() {
+    echo -e "${GREEN}[SUCCESS]${NC} $1"
 }
 
-# Bash TCP监听函数
-bash_tcp_listen() {
-    local port="$1"
-    
-    {
-        timeout 5 bash -c "
-            exec 3<>/dev/tcp/0.0.0.0/$port 2>/dev/null
-            [[ \$? -ne 0 ]] && exit 1
-            read -r data <&3
-            echo \"\$data\"
-            exec 3>&-
-            exit 0
-        " 2>/dev/null
-    }
+log_warning() {
+    echo -e "${YELLOW}[WARNING]${NC} $1"
 }
 
-# 发送心跳
-send_heartbeat() {
-    local heartbeat_data="HEARTBEAT|$SERIAL|$HOSTNAME|$SYSTEM_INFO"
+log_error() {
+    echo -e "${RED}[ERROR]${NC} $1"
+}
 
-    if bash_tcp_send "$heartbeat_data" "$SERVER_IP" "$SERVER_PORT"; then
-        log "✅ 心跳成功"
+# 检查依赖工具
+check_dependencies() {
+    local tools=("nc" "telnet" "curl" "timeout" "nmap")
+    local missing=()
+    
+    for tool in "${tools[@]}"; do
+        if ! command -v "$tool" &> /dev/null; then
+            missing+=("$tool")
+        fi
+    done
+    
+    if [ ${#missing[@]} -gt 0 ]; then
+        log_warning "以下工具未安装: ${missing[*]}"
+        log_info "部分检测功能可能受限"
+    fi
+}
+
+# 方案1: 使用nc (netcat) 检测端口
+check_with_nc() {
+    log_info "方案1: 使用 netcat 检测端口..."
+    
+    if command -v nc &> /dev/null; then
+        if timeout "$TIMEOUT" nc -z -w "$TIMEOUT" "$TARGET_HOST" "$TARGET_PORT" &> /dev/null; then
+            log_success "nc检测: 端口 $TARGET_PORT 可达"
             return 0
         else
-        log "❌ 心跳失败"
+            log_error "nc检测: 端口 $TARGET_PORT 不可达"
+            return 1
+        fi
+    else
+        log_warning "nc 未安装，跳过此方案"
+        return 2
+    fi
+}
+
+# 方案2: 使用telnet检测端口
+check_with_telnet() {
+    log_info "方案2: 使用 telnet 检测端口..."
+    
+    if command -v telnet &> /dev/null; then
+        # 创建telnet命令脚本
+        local telnet_cmd="echo 'quit' | timeout $TIMEOUT telnet $TARGET_HOST $TARGET_PORT"
+        
+        if eval "$telnet_cmd" 2>&1 | grep -q "Connected\|Escape character"; then
+            log_success "telnet检测: 端口 $TARGET_PORT 可达"
+            return 0
+        else
+            log_error "telnet检测: 端口 $TARGET_PORT 不可达"
+            return 1
+        fi
+    else
+        log_warning "telnet 未安装，跳过此方案"
+        return 2
+    fi
+}
+
+# 方案3: 使用curl检测HTTP/HTTPS端口
+check_with_curl() {
+    log_info "方案3: 使用 curl 检测HTTP服务..."
+    
+    if command -v curl &> /dev/null; then
+        local protocols=("http" "https")
+        
+        for proto in "${protocols[@]}"; do
+            log_info "尝试 $proto 协议..."
+            if timeout "$TIMEOUT" curl -s -I "${proto}://${TARGET_HOST}:${TARGET_PORT}/" &> /dev/null; then
+                log_success "curl检测: $proto 服务在端口 $TARGET_PORT 可达"
+                return 0
+            fi
+        done
+        
+        log_error "curl检测: HTTP/HTTPS服务在端口 $TARGET_PORT 不可达"
+        return 1
+    else
+        log_warning "curl 未安装，跳过此方案"
+        return 2
+    fi
+}
+
+# 方案4: 使用nmap检测端口
+check_with_nmap() {
+    log_info "方案4: 使用 nmap 检测端口..."
+    
+    if command -v nmap &> /dev/null; then
+        if timeout "$TIMEOUT" nmap -p "$TARGET_PORT" "$TARGET_HOST" 2>&1 | grep -q "$TARGET_PORT/tcp open"; then
+            log_success "nmap检测: 端口 $TARGET_PORT 开放"
+            return 0
+        else
+            log_error "nmap检测: 端口 $TARGET_PORT 关闭"
+            return 1
+        fi
+    else
+        log_warning "nmap 未安装，跳过此方案"
+        return 2
+    fi
+}
+
+# 方案5: 使用/dev/tcp检测 (bash内置)
+check_with_bash_tcp() {
+    log_info "方案5: 使用 bash /dev/tcp 检测..."
+    
+    if timeout "$TIMEOUT" bash -c "echo > /dev/tcp/$TARGET_HOST/$TARGET_PORT" &> /dev/null; then
+        log_success "bash /dev/tcp检测: 端口 $TARGET_PORT 可达"
+        return 0
+    else
+        log_error "bash /dev/tcp检测: 端口 $TARGET_PORT 不可达"
         return 1
     fi
 }
 
-# 执行命令
-execute_command() {
-    local command="$1"
-    log "执行命令: $command"
+# 方案6: 使用ping检测主机可达性
+check_ping() {
+    log_info "方案6: 检测主机网络连通性..."
     
-    # 在子进程中执行
-    (
-        result=$(eval "$command" 2>&1)
-        log "命令结果: $result"
-        echo "$result"
-    ) &
+    if ping -c 2 -W 2 "$TARGET_HOST" &> /dev/null; then
+        log_success "ping检测: 主机 $TARGET_HOST 网络可达"
+        return 0
+    else
+        log_error "ping检测: 主机 $TARGET_HOST 网络不可达"
+        return 1
+    fi
 }
 
-# 执行脚本
-execute_script() {
-    local script_name="$1"
+# 主检测函数
+main_check() {
+    log_info "开始检测 $TARGET_HOST:$TARGET_PORT ..."
+    log_info "超时时间: $TIMEOUT 秒"
+    echo "========================================"
     
-    case "$script_name" in
-        "system_info.sh")
-            echo "=== 系统信息 ==="
-            echo "主机名: $HOSTNAME"
-            echo "序列号: $SERIAL" 
-            echo "系统: $SYSTEM_INFO"
-            echo "内核: $(uname -r)"
-            echo "架构: $(uname -m)"
-            echo "上线时间: $(uptime -p 2>/dev/null || uptime)"
-            echo "内存:"
-            free -h 2>/dev/null || cat /proc/meminfo | head -3
-            ;;
-        "restart_services.sh")
-            echo "重启系统服务..."
-            systemctl restart networking 2>/dev/null || systemctl restart network 2>/dev/null
-            systemctl restart ssh 2>/dev/null || systemctl restart sshd 2>/dev/null
-            echo "服务重启完成"
-            ;;
-        "update_system.sh")
-            echo "开始系统更新..."
-            if command -v apt-get &>/dev/null; then
-                apt-get update && apt-get upgrade -y
-            elif command -v yum &>/dev/null; then
-                yum update -y
-            elif command -v dnf &>/dev/null; then
-                dnf update -y
+    # 检查依赖
+    check_dependencies
+    echo "----------------------------------------"
+    
+    local results=()
+    local methods=("check_with_nc" "check_with_telnet" "check_with_curl" "check_with_nmap" "check_with_bash_tcp")
+    
+    # 先检查主机连通性
+    if ! check_ping; then
+        log_error "主机网络不可达，可能无法连接端口"
     fi
-            echo "系统更新完成"
-            ;;
-        "network_info.sh")
-            echo "=== 网络信息 ==="
-            ip addr show 2>/dev/null || ifconfig 2>/dev/null || echo "无法获取网络信息"
-            ;;
-        *)
-            echo "未知脚本: $script_name"
-            echo "可用脚本: system_info.sh, restart_services.sh, update_system.sh, network_info.sh"
-            ;;
+    echo "----------------------------------------"
+    
+    # 执行各种检测方法
+    for method in "${methods[@]}"; do
+        if $method; then
+            results+=("$method: 成功")
+        else
+            case $? in
+                1) results+=("$method: 失败") ;;
+                2) results+=("$method: 跳过") ;;
             esac
-}
-
-# 处理命令
-process_command() {
-    local command_data="$1"
-    
-    if [[ "$command_data" == COMMAND:* ]]; then
-        local cmd="${command_data#COMMAND:}"
-        execute_command "$cmd"
-    elif [[ "$command_data" == SCRIPT:* ]]; then
-        local script_name="${command_data#SCRIPT:}"
-        execute_script "$script_name"
-    else
-        log "未知命令: $command_data"
         fi
-}
+        echo "----------------------------------------"
+    done
     
-# 命令监听器
-start_listener() {
-    log "启动命令监听器端口: $CLIENT_PORT"
-    
-    while true; do
-        local command_data=$(bash_tcp_listen "$CLIENT_PORT")
-        
-        if [[ -n "$command_data" ]]; then
-            log "收到命令: $command_data"
-            process_command "$command_data"
-        fi
-        
-        sleep 1
+    # 汇总结果
+    log_info "检测结果汇总:"
+    for result in "${results[@]}"; do
+        echo "  - $result"
     done
 }
 
-# 主守护进程
-start_daemon() {
-    log "启动客户端守护进程"
-    
-    local heartbeat_count=0
-    local success_count=0
-    
-    # 启动命令监听器
-    start_listener &
-    local listener_pid=$!
-    
-    # 主心跳循环
-    while true; do
-        heartbeat_count=$((heartbeat_count + 1))
-        
-        if send_heartbeat; then
-            success_count=$((success_count + 1))
-        fi
-        
-        # 每10次心跳显示状态
-        if [[ $((heartbeat_count % 10)) -eq 0 ]]; then
-            local rate=$((success_count * 100 / heartbeat_count))
-            log "运行状态: ${heartbeat_count}次心跳, ${success_count}次成功 (${rate}%)"
-        fi
-        
-        sleep "$HEARTBEAT_INTERVAL"
-    done
-    
-    # 清理
-    kill $listener_pid 2>/dev/null
+# 使用说明
+usage() {
+    echo "多方案端口检测脚本"
+    echo "用法: $0 <IP地址/域名> <端口号> [超时时间]"
+    echo "示例:"
+    echo "  $0 google.com 80"
+    echo "  $0 192.168.1.1 22 10"
+    echo "  $0 example.com 443 3"
 }
 
-# 设置开机自启
-setup_autostart() {
-    log "设置开机自启..."
-    
-    # 方法1: systemd
-    if command -v systemctl &>/dev/null && [[ -w /etc/systemd/system ]]; then
-        local service_file="/etc/systemd/system/controller-client.service"
-        cat > "$service_file" << EOF
-[Unit]
-Description=Controller Client
-After=network.target
-
-[Service]
-Type=simple
-ExecStart=/bin/bash -c "curl -sSL https://raw.githubusercontent.com/xzx3344521/dock/refs/heads/main/client.sh | bash"
-Restart=always
-RestartSec=30
-
-[Install]
-WantedBy=multi-user.target
-EOF
-        systemctl daemon-reload
-        systemctl enable controller-client
-        log "✅ systemd自启设置完成"
-        return 0
-    fi
-    
-    # 方法2: rc.local
-    if [[ -w /etc/rc.local ]]; then
-        grep -q "controller-client" /etc/rc.local || {
-            echo "curl -sSL https://raw.githubusercontent.com/xzx3344521/dock/refs/heads/main/client.sh | bash &" >> /etc/rc.local
-        }
-        log "✅ rc.local自启设置完成"
-        return 0
-    fi
-    
-    # 方法3: crontab
-    if command -v crontab &>/dev/null; then
-        (crontab -l 2>/dev/null | grep -v "controller-client"; \
-         echo "@reboot curl -sSL https://raw.githubusercontent.com/xzx3344521/dock/refs/heads/main/client.sh | bash") | crontab -
-        log "✅ crontab自启设置完成"
-        return 0
-    fi
-    
-    log "⚠️ 无法设置开机自启"
-    return 1
-}
-
-# 测试连接
-test_connection() {
-    log "测试服务器连接..."
-    
-    if bash_tcp_send "TEST|$SERIAL" "$SERVER_IP" "$SERVER_PORT"; then
-        log "✅ 服务器连接测试成功"
-        return 0
-    else
-        log "❌ 服务器连接测试失败"
-        return 1
-    fi
-}
-
-# 主函数
-main() {
-    init_client
-    
-    # 设置开机自启
-    setup_autostart
-    
-    # 测试连接
-    if test_connection; then
-        log "🎉 连接成功! 开始守护进程..."
-        start_daemon
-    else
-        log "💥 连接失败，退出"
+# 参数检查
+if [ $# -lt 2 ]; then
+    usage
     exit 1
 fi
-}
 
-# 信号处理
-trap 'log "客户端停止"; exit 0' INT TERM
-
-# 启动
-main
+# 运行主检测
+main_check