Go bufio.Reader.Peek 怎样判断输入是否满足边界

判断 bufio.Reader.Peek 是否满足固定边界，核心条件是同时检查返回的切片长度和错误：err == nil && len(data) == n。如果 n 大于 reader 的缓冲区容量，结果不是“继续等待”，而是 bufio.ErrBufferFull；如果输入提前结束，通常得到不足长度的切片和 io.EOF。因此，边界判断要先限制请求，再分类处理错误。

先把“边界满足”定义成可审计条件

这里的边界是“当前流至少有 n 个连续字节可供本次判断”。它不是底层连接一次返回了多少字节，也不是 Buffered() 当前显示多少字节。Peek 会在必要时继续从底层 reader 填充缓冲区，但它不能突破自身的缓冲区容量。

从资产和攻击路径看，资产是协议头或长度字段的完整性，风险来自外部输入提前结束、请求长度过大，以及把临时切片保存到下一次读操作之后。第一张结构说明图把这几个对象的关系画在同一条边界上。

先拦截超过缓冲区的请求

Reader.Size() 返回底层缓冲区大小，适合做固定帧头或协议前缀的上限检查。把上限放在 Peek 之前有两个好处：错误语义更清楚，也不会让调用方误以为 reader 会自动扩容。

package main

import (
    "bufio"
    "errors"
    "fmt"
    "io"
)

// peekExact 只判断当前流是否具备 n 个字节，不消费这些字节。
func peekExact(r *bufio.Reader, n int) ([]byte, error) {
    // 固定长度边界必须是正数，避免把 Peek(0) 当成有效协议头。
    if n  r.Size() {
        return nil, fmt.Errorf("请求 %d 字节超过缓冲区 %d：%w", n, r.Size(), bufio.ErrBufferFull)
    }
    data, err := r.Peek(n)
    if err != nil {
        // data 可能是短切片；保留原错误，便于区分 EOF 和底层 I/O 故障。
        return data, fmt.Errorf("边界未满足，需要 %d 字节，当前只有 %d：%w", n, len(data), err)
    }
    return data, nil
}

func readFrameHeader(r *bufio.Reader) error {
    header, err := peekExact(r, 4)
    if err != nil {
        if errors.Is(err, bufio.ErrBufferFull) {
            return fmt.Errorf("协议头长度配置过大：%w", err)
        }
        if errors.Is(err, io.EOF) {
            return fmt.Errorf("输入提前结束：%w", err)
        }
        return err
    }
    // Peek 没有推进位置；确认魔数后再消费协议头。
    if string(header) != "G001" {
        return fmt.Errorf("协议头不匹配：%q", header)
    }
    _, err = r.Discard(len(header))
    return err
}

示例中的 data 只在当前判断和比较中使用。若业务要跨越下一次读取保存它，应复制到新切片；否则底层填充缓冲区后，原切片内容可能变化。

用返回值区分输入不足和配置错误

判断时不要只写 len(data) == n，也不要只判断 err == nil。完整边界应满足两者；短切片说明已经拿到部分输入，但错误才说明为什么没拿齐。第二张边界说明图将三个常见分支与处理动作对应起来。

消费顺序和验证清单

成功 Peek 后，先完成魔数、版本或类型字段的比较，再用 Read 或 Discard 消费。不要先读走再回退，因为 Peek 会影响 UnreadByte 和 UnreadRune 的使用条件。生产代码至少记录请求长度、reader 缓冲区大小、返回长度和错误分类，便于定位是对端截断还是本地配置过小。

• 空输入：确认不会把 EOF 当成正常的零长度头部。
• 刚好 n 字节：确认比较成功后，消费位置只前进一次。
• 少于 n 字节：确认短切片不会被转换成合法消息。
• n 超过 Size()：确认快速返回 ErrBufferFull。
• 下一次读取后：确认没有继续引用旧的 Peek 切片。

相关问题

Peek 会不会移动 Reader 的读取位置？

不会。它只返回当前位置开始的视图；需要推进位置时再调用 Read 或 Discard。

为什么输入明明很长，Peek 仍然返回 ErrBufferFull？

因为 n 超过了 reader 自身缓冲区大小。底层输入总长度大，不代表一次 Peek 可以跨越这个容量限制。

Peek 返回的字节能长期保存吗？

不能直接依赖。下一次读操作后这段切片可能被覆盖；要长期保存就用 append([]byte(nil), data...) 复制。