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 再用应用侧读查询真打一遍， 迁移块排到依赖表之前这种错只有新库会炸，老库毫无症状。
121 lines
3.6 KiB
Go
121 lines
3.6 KiB
Go
package web
|
||
|
||
import (
|
||
"compress/gzip"
|
||
"net/http"
|
||
"strings"
|
||
"sync"
|
||
)
|
||
|
||
// 可压缩的响应类型。图片（迷雾 PNG、图标）本身已是压缩格式，再压纯属浪费 CPU。
|
||
func compressible(ct string) bool {
|
||
ct = strings.ToLower(ct)
|
||
if i := strings.IndexByte(ct, ';'); i >= 0 {
|
||
ct = ct[:i]
|
||
}
|
||
ct = strings.TrimSpace(ct)
|
||
switch ct {
|
||
case "text/html", "text/css", "text/plain", "text/javascript",
|
||
"application/javascript", "application/json", "application/x-javascript",
|
||
"image/svg+xml", "application/manifest+json", "application/xml":
|
||
return true
|
||
}
|
||
return strings.HasPrefix(ct, "text/") && ct != "text/cache-manifest"
|
||
}
|
||
|
||
type gzipWriter struct {
|
||
http.ResponseWriter
|
||
gz *gzip.Writer
|
||
comp bool
|
||
seen bool // 首部是否已定下来（WriteHeader 已调用，或首字节已写）
|
||
}
|
||
|
||
var gzipPool = sync.Pool{New: func() any {
|
||
return gzip.NewWriter(nil)
|
||
}}
|
||
|
||
// WriteHeader 必须在首部提交前决定压不压。静态文件由 http.FileServer 提供，
|
||
// 它会先设好 Content-Length 再显式 WriteHeader(200/206)；若拖到 Write 里再改首部，
|
||
// 首部早已提交，会出现「声明长度 ≠ 实际压缩后长度」，浏览器报
|
||
// ERR_CONTENT_LENGTH_MISMATCH 并把 CSS/JS 截断。只有 200 才压缩，
|
||
// 206（Range）/304 等自带长度语义，一律原样透传。
|
||
func (g *gzipWriter) WriteHeader(code int) {
|
||
if g.seen {
|
||
return
|
||
}
|
||
g.seen = true
|
||
if code == http.StatusOK && compressible(g.Header().Get("Content-Type")) {
|
||
g.comp = true
|
||
g.Header().Set("Content-Encoding", "gzip")
|
||
g.Header().Set("Vary", "Accept-Encoding")
|
||
g.Header().Del("Content-Length")
|
||
}
|
||
g.ResponseWriter.WriteHeader(code)
|
||
}
|
||
|
||
func (g *gzipWriter) Write(b []byte) (int, error) {
|
||
if !g.seen {
|
||
g.seen = true
|
||
// 上游没有显式 WriteHeader 时，首字节写入才能确定内容类型，此时决定压不压
|
||
ct := g.Header().Get("Content-Type")
|
||
if ct == "" {
|
||
ct = http.DetectContentType(b)
|
||
g.Header().Set("Content-Type", ct)
|
||
}
|
||
if compressible(ct) {
|
||
g.comp = true
|
||
g.Header().Set("Content-Encoding", "gzip")
|
||
g.Header().Set("Vary", "Accept-Encoding")
|
||
g.Header().Del("Content-Length")
|
||
}
|
||
}
|
||
if !g.comp {
|
||
return g.ResponseWriter.Write(b)
|
||
}
|
||
if g.gz == nil {
|
||
g.gz = gzipPool.Get().(*gzip.Writer)
|
||
g.gz.Reset(g.ResponseWriter)
|
||
}
|
||
return g.gz.Write(b)
|
||
}
|
||
|
||
func (g *gzipWriter) Flush() {
|
||
if g.gz != nil {
|
||
g.gz.Flush()
|
||
}
|
||
if f, ok := g.ResponseWriter.(http.Flusher); ok {
|
||
f.Flush()
|
||
}
|
||
}
|
||
|
||
// Unwrap 让 http.NewResponseController 能穿过这层包装拿到真正的 ResponseWriter
|
||
// （SetWriteDeadline / Flush / Hijack 都靠它逐层解包）。少了这个方法，
|
||
// 被包装的请求调 SetWriteDeadline 只会拿到 ErrNotSupported。
|
||
func (g *gzipWriter) Unwrap() http.ResponseWriter { return g.ResponseWriter }
|
||
|
||
func (g *gzipWriter) close() {
|
||
if g.gz != nil {
|
||
g.gz.Close()
|
||
gzipPool.Put(g.gz)
|
||
g.gz = nil
|
||
}
|
||
}
|
||
|
||
// gzipHandler 对声明了 gzip 且内容类型可压缩的响应做在线压缩。
|
||
func gzipHandler(next http.Handler) http.Handler {
|
||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
// 带 Range 的请求（浏览器校验 / 续传 immutable 缓存的静态资源）会返回 206 并自带
|
||
// Content-Length，gzip 无法在首部已提交后再改写长度，会导致浏览器把 CSS/JS 截断
|
||
// （ERR_CONTENT_LENGTH_MISMATCH）。这类请求直接透传、不压缩。
|
||
if !strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") ||
|
||
r.Header.Get("Upgrade") != "" ||
|
||
r.Header.Get("Range") != "" {
|
||
next.ServeHTTP(w, r)
|
||
return
|
||
}
|
||
gw := &gzipWriter{ResponseWriter: w}
|
||
defer gw.close()
|
||
next.ServeHTTP(gw, r)
|
||
})
|
||
}
|