Files
rond-with-you/internal/auth/code_replay_test.go
T
tamakyi b4c2a8452f 审计修复:后端(安全 / 数据一致性 / 性能)
audit-2026-09-21.md 里的后端条目落地。每条都配了回归守卫,并逐条做过变异验证
(注入原 bug → 确认按正确理由变红 → 还原 → 变绿)。

安全
- CSRF 只加同源 Origin/Referer 守卫(csrf.go),不动模板
- 会话 cookie 的 Secure 按请求判断;两步验证两闸(loginGuard + 验证码一次性)
- 跳转收口到 safeInternalPath;clientIP 只认 XFF 最后一段
- GPX 按属主圈定且查 RowsAffected;下载名走 downloadDisposition
- 上传口统一 MaxBytesReader;落盘后的失败路径清掉半成品

数据一致性
- 区域判定两端各过一遍再 AND(apiTrack 与 movementsWhere 对齐)
- Filter.Verdict 全域一致(下推到 visitsWhere);聚合刷新必须返回 error
- 附属数据改挂坐标键,Go 与 SQL 两套表达式逐位一致
- 脏备份的伪时间/越界坐标不再进领域模型;rows 循环统一 drainRows
- 合并与导出:writeRawRow 上抛错误、悬空引用清理、原生列补全

性能
- placeCols 相关子查询带 dataset_id 前导列;findSourceArchive 先按大小筛
- 批量勾选改 ANY($3::bigint[]);fillupCache 加上限;同一查询不再查两遍

另:新增「全新库」冒烟测试 —— 空库跑完 Migrate 再用应用侧读查询真打一遍,
迁移块排到依赖表之前这种错只有新库会炸,老库毫无症状。
2026-09-24 20:37:16 +08:00

48 lines
1.3 KiB
Go

package auth
import (
"testing"
"time"
)
// TestCodeReplayRejectsReuseInWindow 同一个码在窗口内只能用一次。
func TestCodeReplayRejectsReuseInWindow(t *testing.T) {
r := NewCodeReplay()
base := time.Date(2026, 9, 21, 10, 0, 0, 0, time.UTC)
r.now = func() time.Time { return base }
if !r.Consume(7, "123456") {
t.Fatal("首次使用应当通过")
}
if r.Consume(7, "123456") {
t.Error("同一个码被重放了")
}
// 换个用户号的同一个数字码不受影响
if !r.Consume(8, "123456") {
t.Error("不同用户的同一个码不该互相影响")
}
}
// TestCodeReplayExpiresAfterWindow 条目只活过一个接受窗口(±1 步 = 90 秒),
// 过期后同一个码可以再被收下——否则条目会越攒越多,且数字复用会误伤。
func TestCodeReplayExpiresAfterWindow(t *testing.T) {
r := NewCodeReplay()
now := time.Date(2026, 9, 21, 10, 0, 0, 0, time.UTC)
r.now = func() time.Time { return now }
if !r.Consume(7, "123456") {
t.Fatal("首次使用应当通过")
}
now = now.Add(replayTTL - time.Second)
if r.Consume(7, "123456") {
t.Error("窗口内不该放行")
}
now = now.Add(2 * time.Second) // 越过 replayTTL
if !r.Consume(7, "123456") {
t.Error("过期后应当放行")
}
if n := len(r.used); n != 1 {
t.Errorf("过期条目没被清掉,还剩 %d 条", n)
}
}