本文介绍在 Go Web 服务中，如何通过包装 http.ResponseWriter 拦截响应体、计算其 SHA-256 校验和，并将结果作为自定义 HTTP 响应头（如 X-Content-SHA256）返回，实现下载前可验证文件完整性的能力。
本文介绍在 go web 服务中，如何通过包装 `http.responsewriter` 拦截响应体、计算其 sha-256 校验和，并将结果作为自定义 http 响应头（如 `x-content-sha256`）返回，实现下载前可验证文件完整性的能力。
在构建可信赖的 HTTP 文件分发服务时，向客户端提供响应内容的密码学校验和（如 SHA-256）至关重要——它使客户端能在下载开始前预知预期哈希值，从而主动校验完整性，防范传输损坏或中间人篡改。Go 标准库不直接支持“后置写入校验”，因为 http.ResponseWriter 的 Write 和 WriteHeader 是流式调用，一旦写出便无法回溯。解决方案是实现一个响应包装器（ResponseWriter wrapper），在真正写入底层连接前，先缓冲全部响应体并记录状态码，待 Handler 执行完毕后再统一计算哈希、设置头、转发响应。
以下是完整、可运行的实现：
package main
import (
"bytes"
"crypto/sha256"
"encoding/hex"
"flag"
"fmt"
"log"
"net/http"
"strconv"
)
// rwWrapper 捕获响应状态码与响应体字节
type rwWrapper struct {
http.ResponseWriter
buf bytes.Buffer
status int
}
// Write 拦截响应体写入，写入内部缓冲区
func (w *rwWrapper) Write(p []byte) (int, error) {
return w.buf.Write(p)
}
// WriteHeader 拦截状态码，仅缓存，不立即发送
func (w *rwWrapper) WriteHeader(status int) {
w.status = status
}
// Checksum 中间件：包装 Handler，注入校验和响应头
func Checksum(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// 创建包装器，初始状态设为 200（若 Handler 未显式调用 WriteHeader，则默认使用）
ww := &rwWrapper{
ResponseWriter: w,
status: http.StatusOK,
}
// 执行原始 Handler —— 所有 Write/WriteHeader 被重定向至 ww
h.ServeHTTP(ww, r)
// ✅ 此时响应体已完整存于 ww.buf 中，可安全计算哈希
hash := sha256.Sum256(ww.buf.Bytes())
checksum := hex.EncodeToString(hash[:])
// 设置自定义校验和头（推荐使用标准化名称，如 RFC 9286 提议的 Digest 头，此处用 X-Content-SHA256 保持清晰）
w.Header().Set("X-Content-SHA256", checksum)
// ⚠️ 必须先设置 Header，再调用 WriteHeader 和 Write（顺序不可逆）
w.WriteHeader(ww.status)
w.Write(ww.buf.Bytes())
})
}
func main() {
var listenAddr = flag.String("http", ":8080", "address to listen on for HTTP")
flag.Parse()
http.Handle("/", Checksum(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("X-Foo", "bar")
w.Header().Set("Content-Type", "text/plain")
w.Header().Set("Date", "Sun, 08 May 2016 14:04:53 GMT")
msg := "Curiosity is insubordination in its purest form.\n"
// 注意：此处无需手动设 Content-Length —— http.Server 会自动计算并覆盖
fmt.Fprintf(w, msg)
})))
log.Printf("Server listening on %s", *listenAddr)
log.Fatal(http.ListenAndServe(*listenAddr, nil))
}关键注意事项：
- ✅ 缓冲开销权衡：该方案将整个响应体暂存内存，适用于中小文件（hash.Hash +
io.TeeReader）或分块校验，但需修改业务逻辑。 - ✅ Header 设置时机：
w.Header().Set()必须在w.WriteHeader()之前调用，否则 Header 将被忽略。 - ✅ 状态码处理：包装器需准确捕获
WriteHeader调用；若 Handler 未调用，则http.Server默认使用200，因此初始化status: http.StatusOK是合理兜底。 - ✅ 标准化建议：生产环境推荐使用 RFC 9286 定义的
Digest响应头，例如：Digest: sha-256=47DEQpj8HBSa+/TImW+5JCeuQeRkm5NMpJWZG3hSuFU=（Base64 编码），而非自定义头，以提升互操作性。
通过此中间件，任何 http.Handler 都能透明获得内容校验和能力，客户端即可在发起 GET 请求时检查 X-Content-SHA256 头，下载后比对本地计算值，真正实现端到端可信交付。
