Files
easyai-ai-gateway/scripts/cluster/bootstrap-wireguard.sh
T
wangbo ed3b970178 refactor(cluster): 通用化第四个 Worker 节点接入
将原宁波专用接入脚本改为由环境变量控制节点名称、站点和 WireGuard 端口,支持香港双物理节点 Worker 池。增加候选节点启用开关和服务商网络许可门禁,未明确批准时禁止任何 WireGuard 或 K3s 写操作。同步调整生产验收、巡检、拓扑校验和运行手册。\n\n验证:\n- scripts/cluster 全部 Shell 脚本 bash -n\n- scripts/cluster 全部 ShellCheck\n- git diff --check\n- 本地凭据值扫描\n- 当前三节点 WireGuard/K3s/PostgreSQL 只读基线检查
2026-08-01 11:23:19 +08:00

120 lines
3.5 KiB
Bash
Executable File

#!/usr/bin/env bash
set -euo pipefail
script_dir=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
# shellcheck source=scripts/cluster/common.sh
source "$script_dir/common.sh"
load_cluster_env
require_commands ssh scp
hosts=("$CLUSTER_NINGBO_HOST" "$CLUSTER_HONGKONG_HOST" "$CLUSTER_LOS_ANGELES_HOST")
public_endpoints=("${CLUSTER_NINGBO_HOST#*@}" "${CLUSTER_HONGKONG_HOST#*@}" "${CLUSTER_LOS_ANGELES_HOST#*@}")
wireguard_ips=(10.77.0.1 10.77.0.2 10.77.0.3)
wireguard_ports=(51820 51820 51820)
if [[ -n $CLUSTER_WORKER_NODE_4_HOST ]]; then
[[ $CLUSTER_WORKER_NODE_4_NETWORK_APPROVED == true ]] || {
echo 'dedicated Worker private-network setup is blocked until the provider approves this non-proxy Kubernetes use in writing' >&2
exit 1
}
hosts+=("$CLUSTER_WORKER_NODE_4_HOST")
public_endpoints+=("${CLUSTER_WORKER_NODE_4_HOST#*@}")
wireguard_ips+=(10.77.0.4)
wireguard_ports+=("$CLUSTER_WORKER_NODE_4_WIREGUARD_PORT")
fi
public_keys=()
for host in "${hosts[@]}"; do
echo "[wireguard] preparing $host"
public_key=$(cluster_ssh "$host" 'bash -s' <<'REMOTE' | tail -n 1
set -euo pipefail
export DEBIAN_FRONTEND=noninteractive
if ! command -v wg >/dev/null 2>&1; then
apt-get update -qq
apt-get install -y -qq wireguard-tools
fi
install -d -m 0700 /etc/wireguard
if [[ ! -s /etc/wireguard/privatekey ]]; then
umask 077
wg genkey >/etc/wireguard/privatekey
fi
chmod 0600 /etc/wireguard/privatekey
wg pubkey </etc/wireguard/privatekey >/etc/wireguard/publickey
cat /etc/wireguard/publickey
REMOTE
)
[[ $public_key =~ ^[A-Za-z0-9+/]{43}=$ ]] || {
echo "invalid WireGuard public key returned by $host" >&2
exit 1
}
public_keys+=("$public_key")
done
for local_index in "${!hosts[@]}"; do
peer_arguments=()
for candidate in "${!hosts[@]}"; do
if [[ $candidate -ne $local_index ]]; then
peer_arguments+=(
"${public_keys[$candidate]}"
"${public_endpoints[$candidate]}"
"${wireguard_ports[$candidate]}"
"${wireguard_ips[$candidate]}"
)
fi
done
echo "[wireguard] configuring ${hosts[$local_index]} as ${wireguard_ips[$local_index]}"
cluster_ssh "${hosts[$local_index]}" bash -s -- \
"${wireguard_ips[$local_index]}" \
"${wireguard_ports[$local_index]}" \
"${peer_arguments[@]}" <<'REMOTE'
set -euo pipefail
local_ip=$1
local_port=$2
shift 2
private_key=$(cat /etc/wireguard/privatekey)
umask 077
temporary_config=$(mktemp /etc/wireguard/wg0.conf.XXXXXX)
cat >"$temporary_config" <<EOF
[Interface]
Address = ${local_ip}/24
ListenPort = ${local_port}
PrivateKey = ${private_key}
EOF
while (( $# >= 4 )); do
cat >>"$temporary_config" <<EOF
[Peer]
PublicKey = $1
Endpoint = $2:$3
AllowedIPs = $4/32
PersistentKeepalive = 25
EOF
shift 4
done
(( $# == 0 ))
chmod 0600 "$temporary_config"
mv "$temporary_config" /etc/wireguard/wg0.conf
systemctl enable wg-quick@wg0 >/dev/null
systemctl restart wg-quick@wg0
REMOTE
done
for source_index in "${!hosts[@]}"; do
for target_index in "${!hosts[@]}"; do
[[ $source_index -eq $target_index ]] && continue
cluster_ssh "${hosts[$source_index]}" \
"ping -c 3 -W 2 ${wireguard_ips[$target_index]} >/dev/null"
done
done
for host in "${hosts[@]}"; do
latest_handshake=$(cluster_ssh "$host" \
"wg show wg0 latest-handshakes | awk '\$2 > 0 { count++ } END { print count + 0 }'")
[[ $latest_handshake == $((${#hosts[@]} - 1)) ]] || {
echo "WireGuard handshake verification failed on $host" >&2
exit 1
}
done
echo "wireguard_bootstrap=PASS peers=${#hosts[@]} routing=direct"