fix(prod): 后端生产级 A 类硬伤全清(7 项:授权/崩溃点/全表扫/限流/鉴权/限额)
部署前生产级审计(可靠性/数据层/安全三路)后,清掉 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>
This commit is contained in:
@@ -82,3 +82,19 @@ func TestListInvites_FiltersDeadCodes(t *testing.T) {
|
||||
t.Fatalf("全量列表应 3 条,得 %d", len(all))
|
||||
}
|
||||
}
|
||||
|
||||
// A6:TaskOwner 供公开 by-id 端点判权——按 task_id 返回提交者,跨租户可查,不存在返回空。
|
||||
func TestTaskOwner(t *testing.T) {
|
||||
p := newTestStore(t)
|
||||
seedTenant(t, p, "t1")
|
||||
ctx := context.Background()
|
||||
if err := p.SaveTask(ctx, "user-1", "task-abc", "{}"); err != nil {
|
||||
t.Fatalf("建任务失败: %v", err)
|
||||
}
|
||||
if got := p.TaskOwner(ctx, "task-abc"); got != "user-1" {
|
||||
t.Fatalf("owner 应 user-1,得 %q", got)
|
||||
}
|
||||
if got := p.TaskOwner(ctx, "nope"); got != "" {
|
||||
t.Fatalf("不存在任务应返回空,得 %q", got)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -98,6 +98,18 @@ type AdminEval struct {
|
||||
Corrected bool `json:"corrected"`
|
||||
}
|
||||
|
||||
// TaskOwner 按 task_id 返回提交者 user.id(供 SSE/导出等公开 by-id 端点做归属校验)。
|
||||
// 跨租户查(WithoutTenant):这些端点无租户上下文,靠 owner 判权。不存在返回空串。
|
||||
func (p *Postgres) TaskOwner(ctx context.Context, taskID string) string {
|
||||
if p.db == nil || taskID == "" {
|
||||
return ""
|
||||
}
|
||||
var owner string
|
||||
p.db.WithContext(WithoutTenant(ctx)).Model(&Task{}).
|
||||
Where("task_id = ? and deleted_at is null", taskID).Select("owner").Scan(&owner)
|
||||
return owner
|
||||
}
|
||||
|
||||
// TaskDetail 按 task_id 取单条任务的完整下钻数据(管理端,跨租户)。
|
||||
// 必须 WithoutTenant:Task/Eval 都在租户插件作用域内,用请求 ctx 查别的租户的任务
|
||||
// 不会报错,而是静默返回空——看起来像"这任务没产出",比报错更难排查。
|
||||
|
||||
Reference in New Issue
Block a user