Blizzard
ac38d5e663
部署前生产级审计（可靠性/数据层/安全三路）后，清掉 7 处代码级硬伤： A1 后台定时器 goroutine 无 panic recover → 单个 DB panic 崩整个 gateway。加 safeGo/ safeCall，包住订阅/掉单补偿/微信推送/探针 goroutine，单轮 tick 再兜一层。 A2 提示词控制面(建/激活/停用，热广播全服务)只 RequireAuth → 任意登录用户改全局提示词。 三写端点+列表挂 RequireAdmin。 A3 HITL 审批端点无角色门 → viewer 可放行烧钱执行。加 RequireTenantRole(member)。 A4 审计/护栏列表 limit 无校验，limit=-1 让 gorm 取消 LIMIT 全表扫。加 clampLimit/ clampOffset，AdminTasks/AdminSpaces 补上界。 A5 限流 Redis 一挂就完全放行(fail-open)。加进程内固定窗口兜底(fail-safe) + 登录/注册 按 IP 专用严限流(10/min)。 A6 公开 by-id 端点(stream/exec/report导出/kb导入流)无鉴权无租户过滤。加 AuthFromHeaderOrQuery(从 ?token= 取 JWT) + task/report 按 owner 归属校验；桌面端 5 处 EventSource/下载 URL 经 tokenQuery 附 JWT。 A7 文件上传无大小上限(整文件进内存 OOM 面) → 50MB 闸(KB_MAX_UPLOAD_BYTES)+ LimitReader； http.Server 加 ReadHeaderTimeout/ReadTimeout/MaxHeaderBytes(不设 WriteTimeout 保 SSE)。 带单测：clampLimit/safeCall/procLimiter/AuthFromHeaderOrQuery/TaskOwner。 build+vet+全量 test 绿；desktop tsc 绿。B(迁移工具/实时探针/出网韧性/登录锁定/leader选举) 与 C(TLS/PG HA/K8s/备份自动化/可观测)分期后做，参照 production_readiness.md。 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
38 lines
1.2 KiB
Go
38 lines
1.2 KiB
Go
package middleware
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/sundynix/sundynix-gateway/internal/auth"
|
|
)
|
|
|
|
// AuthFromHeaderOrQuery 是给「公开 by-id」端点（SSE 流 / 报告下载）用的强制鉴权。
|
|
// 这些端点由 EventSource / <a download> 发起，**带不了 Authorization 头**——此前只靠随机
|
|
// task_id 寻址、无 authz、无租户过滤（capability-URL，泄露即无第二道门）。
|
|
// 本中间件先读 Bearer 头、没有再读 `?token=` / `?access_token=` 查询参数（EventSource 能带 query），
|
|
// 校验通过注入 CtxUserID（同 Auth），失败即 401。归属校验由各 handler 用 uid 再做（见 requireTaskOwner）。
|
|
func AuthFromHeaderOrQuery() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
tok := ""
|
|
if h := c.GetHeader("Authorization"); strings.HasPrefix(h, "Bearer ") {
|
|
tok = strings.TrimSpace(h[len("Bearer "):])
|
|
}
|
|
if tok == "" {
|
|
tok = c.Query("token")
|
|
}
|
|
if tok == "" {
|
|
tok = c.Query("access_token")
|
|
}
|
|
uid, err := auth.Parse(tok)
|
|
if err != nil || uid == "" {
|
|
c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"error": "需要登录"})
|
|
return
|
|
}
|
|
c.Set(CtxUserID, uid)
|
|
c.Next()
|
|
}
|
|
}
|