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 再用应用侧读查询真打一遍， 迁移块排到依赖表之前这种错只有新库会炸，老库毫无症状。
101 lines
3.5 KiB
Go
101 lines
3.5 KiB
Go
package auth
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
_ "github.com/jackc/pgx/v5/stdlib"
|
|
)
|
|
|
|
// TestSessionTokenBindsPasswordHash 钉住「改密后旧会话立即作废」这条契约。
|
|
// 令牌的 MAC 必须真的绑上口令摘要：曾经绑的是摘要的前 16 个字符，而 PBKDF2
|
|
// 编码形如 `pbkdf2$sha256$210000$<salt>$<dk>`，前 16 字节恒为 `pbkdf2$sha256$21`
|
|
// ——等于绑了个常量，改密后别的设备上的旧 cookie 照样能用。
|
|
// 不碰数据库：SignToken/ParseToken 只用到 Secret 与 Days。
|
|
func TestSessionTokenBindsPasswordHash(t *testing.T) {
|
|
st := &Store{Secret: []byte("test-secret"), Days: 7}
|
|
before, err := HashPassword("old-password-123456")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
after, err := HashPassword("new-password-123456")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tok := st.SignToken(1, before)
|
|
if _, ok := st.ParseToken(tok, before); !ok {
|
|
t.Fatal("用签名时的口令摘要应能验签")
|
|
}
|
|
if _, ok := st.ParseToken(tok, after); ok {
|
|
t.Error("改密后旧会话仍验签通过：令牌 MAC 没有真的绑上口令摘要")
|
|
}
|
|
}
|
|
|
|
// TestChangeUsername 覆盖改名的正向路径、口令校验与占用冲突。
|
|
// 会临时建一个用户并在结束时删掉，所以需要一个可连的库：用 ROND_DSN 指过来，
|
|
// 未设置时跳过（与项目里其它按环境变量门控的测试一致）。
|
|
func TestChangeUsername(t *testing.T) {
|
|
dsn := os.Getenv("ROND_DSN")
|
|
if dsn == "" {
|
|
t.Skip("设置 ROND_DSN 后运行改名检查")
|
|
}
|
|
db, err := sql.Open("pgx", dsn)
|
|
if err != nil {
|
|
t.Fatalf("连接数据库: %v", err)
|
|
}
|
|
defer db.Close()
|
|
ctx := context.Background()
|
|
|
|
name := fmt.Sprintf("tmp_rename_%d", time.Now().UnixNano())
|
|
hash, err := HashPassword("tmp-pass-123456")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var id int64
|
|
if err := db.QueryRowContext(ctx,
|
|
`INSERT INTO users (username, password_hash) VALUES ($1,$2) RETURNING id`, name, hash).Scan(&id); err != nil {
|
|
t.Fatalf("建临时用户: %v", err)
|
|
}
|
|
defer db.ExecContext(ctx, `DELETE FROM users WHERE id=$1`, id)
|
|
|
|
st := &Store{DB: db, Secret: []byte("test-secret"), Days: 7}
|
|
// 改名不该影响已登录会话：cookie 认的是 id 与口令摘要，都不含用户名
|
|
tok := st.SignToken(id, hash)
|
|
|
|
next := name + "_new"
|
|
if err := st.ChangeUsername(ctx, id, "wrong-password", next); !errors.Is(err, ErrInvalidCredentials) {
|
|
t.Errorf("口令错误应返回 ErrInvalidCredentials，实际 %v", err)
|
|
}
|
|
|
|
// 换成一个已存在的用户名（库里除自己之外随便取一个）
|
|
var other string
|
|
if err := db.QueryRowContext(ctx, `SELECT username FROM users WHERE id<>$1 LIMIT 1`, id).Scan(&other); err == nil {
|
|
if err := st.ChangeUsername(ctx, id, "tmp-pass-123456", other); !errors.Is(err, ErrUsernameTaken) {
|
|
t.Errorf("占用他人用户名应返回 ErrUsernameTaken，实际 %v", err)
|
|
}
|
|
}
|
|
|
|
if err := st.ChangeUsername(ctx, id, "tmp-pass-123456", next); err != nil {
|
|
t.Fatalf("正常改名失败: %v", err)
|
|
}
|
|
u, err := st.UserByID(ctx, id)
|
|
if err != nil {
|
|
t.Fatalf("回读用户: %v", err)
|
|
}
|
|
if u.Username != next {
|
|
t.Errorf("用户名应为 %q，实际 %q", next, u.Username)
|
|
}
|
|
if got, ok := st.ParseToken(tok, hash); !ok || got != id {
|
|
t.Errorf("改名后旧会话应仍有效，实际 ok=%v id=%d", ok, got)
|
|
}
|
|
// 密码没被动过：改完还能用原口令登录
|
|
if _, err := st.Authenticate(ctx, next, "tmp-pass-123456"); err != nil {
|
|
t.Errorf("改名后原口令应仍可登录: %v", err)
|
|
}
|
|
}
|