#!/bin/bash
# A/B 对比：QUIC(UDP) vs HTTP/2(TCP)
#
# 背景：
#   隧道日志里全是 QUIC 超时：
#     Failed to dial a quic connection ... no recent network activity
#     Connection terminated error="control stream error"
#     全是 "Failed to accept QUIC stream"
#   而 cloudflared 自己回退成 protocol=http2 之后**注册成功了**。
#   所以怀疑这条 4G 线路在丢 UDP 包，TCP 反而稳。
#
# 每边跑 90 秒，每 2 秒探一次公网，对比不通率。
set -u

RELAY_DIR=/root/lanremote-relay
URL=https://relay.xxmxxs.top

test_protocol() {
    local label="$1"; shift
    local extra="$*"

    systemctl stop lanremote-tunnel 2>/dev/null || true
    pkill -f cloudflared 2>/dev/null || true
    sleep 3
    : > "$RELAY_DIR/tunnel.log"

    # shellcheck disable=SC2086
    nohup cloudflared tunnel --no-autoupdate --edge-ip-version 4 $extra \
        --config /root/.cloudflared/config.yml run lanremote \
        >> "$RELAY_DIR/tunnel.log" 2>&1 &

    echo "===== $label（$extra）====="
    echo "  等 35 秒建连…"
    sleep 35

    local ok=0 bad=0 i code
    for i in $(seq 1 45); do
        code=$(timeout 12 curl -4 -s -o /dev/null -w '%{http_code}' \
               -H 'User-Agent: LanRemote/1.4.0' "$URL/v1/health" 2>/dev/null)
        code=${code:-000}
        if [ "$code" = "200" ]; then ok=$((ok + 1)); else bad=$((bad + 1)); fi
        sleep 2
    done

    local conns drops dials
    conns=$(grep -c 'Registered tunnel connection' "$RELAY_DIR/tunnel.log" 2>/dev/null || echo 0)
    drops=$(grep -c 'Connection terminated' "$RELAY_DIR/tunnel.log" 2>/dev/null || echo 0)
    dials=$(grep -c 'Failed to dial' "$RELAY_DIR/tunnel.log" 2>/dev/null || echo 0)
    local pct=$((ok * 100 / (ok + bad)))

    printf "  >>> %-12s 可用 %2d/%2d (%d%%)  注册 %s 次  断开 %s 次  拨号失败 %s 次\n" \
        "$label" "$ok" "$((ok + bad))" "$pct" "$conns" "$drops" "$dials"
    echo "$label|$pct|$conns|$drops|$dials" >> /root/proto_ab.txt

    pkill -f cloudflared 2>/dev/null || true
    sleep 2
}

: > /root/proto_ab.txt
test_protocol "QUIC" ""
test_protocol "HTTP2" "--protocol http2"

echo
echo "===== 汇总 ====="
printf "  %-8s %-8s %-8s %-8s %s\n" "协议" "可用率" "注册" "断开" "拨号失败"
while IFS='|' read -r label pct conns drops dials; do
    printf "  %-8s %-8s %-8s %-8s %s\n" "$label" "$pct%" "$conns" "$drops" "$dials"
done < /root/proto_ab.txt

echo
echo "===== 结论 ====="
Q=$(grep '^QUIC' /root/proto_ab.txt | cut -d'|' -f2)
H=$(grep '^HTTP2' /root/proto_ab.txt | cut -d'|' -f2)
echo "  QUIC $Q%  vs  HTTP2 $H%"
if [ "$H" -gt "$Q" ]; then
    echo "  -> HTTP/2 明显更好，改用它"
elif [ "$Q" -gt "$H" ]; then
    echo "  -> QUIC 更好，保持"
else
    echo "  -> 两者差不多"
fi