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 再用应用侧读查询真打一遍， 迁移块排到依赖表之前这种错只有新库会炸，老库毫无症状。
76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
package web
|
|
|
|
import (
|
|
"bytes"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"rond-with-you/internal/auth"
|
|
)
|
|
|
|
func uploadsIn(t *testing.T, dir string) map[string]bool {
|
|
t.Helper()
|
|
ents, err := os.ReadDir(dir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
out := map[string]bool{}
|
|
for _, e := range ents {
|
|
out[e.Name()] = true
|
|
}
|
|
return out
|
|
}
|
|
|
|
// D8 回归：备份包解析失败时，已经写进 uploads 的那份半成品必须删掉。
|
|
// 旧写法只在「sha 完全相同」那条路上 os.Remove，其余失败（rond.Open / rond.Parse /
|
|
// Import 任一）都把文件留在盘上 —— 反复上传坏包就是一堆垃圾，还占着 findSourceArchive
|
|
// 每次都要去算 sha256 的那个目录。
|
|
func TestUploadRemovesStoredFileOnFailure(t *testing.T) {
|
|
f := newLoginFixture(t)
|
|
dir := f.srv.cfg.UploadDir
|
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
before := uploadsIn(t, dir)
|
|
|
|
// 名字像备份包、内容是垃圾：必然走到「写盘成功 → 解析失败」这条路径。
|
|
var body bytes.Buffer
|
|
mw := multipart.NewWriter(&body)
|
|
fw, err := mw.CreateFormFile("backup", "zz-broken.rondbackup")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := fw.Write([]byte("这不是一个 zip 包")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
mw.Close()
|
|
|
|
r := httptest.NewRequest("POST", "/admin/upload", &body)
|
|
r.Header.Set("Content-Type", mw.FormDataContentType())
|
|
r = r.WithContext(ctxWithUser(&auth.User{ID: f.uid}))
|
|
w := httptest.NewRecorder()
|
|
f.srv.adminUpload(w, r)
|
|
|
|
if w.Code != http.StatusSeeOther {
|
|
t.Fatalf("应当以 303 回管理页，实际 %d", w.Code)
|
|
}
|
|
loc, _ := url.QueryUnescape(w.Header().Get("Location"))
|
|
// 前提守卫：必须真的是「文件已经落盘读过了、才在解析这一步失败」。
|
|
// 若落在更早的分支（读上传内容失败 / 类型不对），上传目录压根没动，
|
|
// 下面那条「没有残留」的断言只是空过。
|
|
if !strings.Contains(loc, "解析备份失败") {
|
|
t.Fatalf("没走到「写盘后解析失败」这条路，测试前提不成立: %q", loc)
|
|
}
|
|
|
|
for name := range uploadsIn(t, dir) {
|
|
if !before[name] {
|
|
t.Errorf("失败后残留在 uploads 里的半成品: %s", name)
|
|
}
|
|
}
|
|
}
|