本文介绍三种标准库方案（io.limitedreader、io.readfull、http.maxbytesreader），帮助你在 go 中从 tcp 连接安全读取最多 n 字节数据，避免内存溢出或 dos 攻击。
在 Go 网络编程中，直接使用 ioutil.ReadAll（现已迁移至 io.ReadAll）读取 net.TCPConn 的全部数据存在严重风险：攻击者可发送超长流导致内存耗尽，构成典型的拒绝服务（DOS）漏洞。因此，必须对单次读取长度施加硬性上限。Go 标准库提供了多个轻量、高效且语义明确的工具来实现“读取至 EOF 或 n 字节，以先到者为准”。
✅ 推荐方案：io.LimitedReader
io.LimitedReader 是最贴合需求的选项——它包装任意 io.Reader，并在累计读取 N 字节后自动返回 io.EOF（后续读取返回 0, io.EOF），不会阻塞等待更多数据，也不会超额读取。
conn, _ := net.Dial("tcp", "example.com:80")
defer conn.Close()
const maxBytes = 1024 * 1024 // 1MB 限制
limited := &io.LimitedReader{R: conn, N: maxBytes}
data, err := io.ReadAll(limited) // 最多读取 maxBytes，遇 EOF 提前结束
if err != nil && err != io.EOF {
log.Fatal("read error:", err)
}
// len(data) <= maxBytes，安全可控⚠️ 注意：io.LimitedReader.N 是剩余可读字节数，每次成功读取后自动递减；若需复用，应重新构造实例。
✅ 替代方案：io.ReadFull（适用于精确长度）
若需严格读满 n 字节（不足则返回 io.ErrUnexpectedEOF），而非“最多 n 字节”，请使用 io.ReadFull：
buf := make([]byte, 1024)
n, err := io.ReadFull(conn, buf) // 必须填满 buf，否则失败
if err == io.ErrUnexpectedEOF {
// 实际只读了 n < len(buf) 字节
} else if err != nil {
// 其他错误
}此方式不适用于流式场景，仅适合协议头等固定长度字段。
✅ HTTP 场景专用：http.MaxBytesReader
当处理 HTTP 请求体时，http.MaxBytesReader 是最佳实践——它不仅限流，还自动注入 400 Bad Request 响应头，并在超限时关闭连接：
func handler(w http.ResponseWriter, r *http.Request) {
limitedBody := http.MaxBytesReader(w, r.Body, 5*1024*1024) // 5MB
data, err := io.ReadAll(limitedBody)
if err == http.ErrBodyReadAfterClose {
http.Error(w, "body read after close", http.StatusBadRequest)
return
}
// ...
}? 自定义实现（按需扩展）
若标准方案无法满足特殊逻辑（如动态限流、带超时的限流），可参考 io.LimitedReader 源码（仅约 20 行）快速定制：
type SafeLimitedReader struct {
R io.Reader
N int64
Err error // 可选：自定义终止错误
}
func (l *SafeLimitedReader) Read(p []byte) (int, error) {
if l.N <= 0 {
return 0, io.EOF
}
if int64(len(p)) > l.N {
p = p[:l.N]
}
n, err := l.R.Read(p)
l.N -= int64(n)
return n, err
}✅ 总结
| 方案 | 适用场景 | 是否推荐 |
|---|---|---|
| io.LimitedReader | 通用 TCP/UDP 流，读取「≤n 字节」 | ✅ 强烈推荐 |
| io.ReadFull | 协议头等需精确长度的场景 | ⚠️ 按需选用 |
| http.MaxBytesReader | HTTP 服务端请求体防护 | ✅ HTTP 场景首选 |
始终避免无限制读取网络数据。选择 io.LimitedReader 作为默认方案，配合合理的 N 值（如根据业务预期最大包大小设定），即可兼顾安全性与简洁性。
