当前位置：首页 > 文章列表 > Golang > Go教程 > Go bufio.Reader.Peek 怎样判断输入是否满足边界

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

来源：17golang原创 2026-11-03 17:14:09 0浏览收藏

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

要点速览

Peek(n) 只观察接下来的 n 个字节，不推进读取位置。
n > Reader.Size() 时优先按 ErrBufferFull 处理，不能把它当成普通 EOF。
返回的切片只适合立即比较；下一次读取后它就可能被覆盖。

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

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

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

Go bufio.Reader.Peek 输入流、缓冲区窗口和未推进读取位置的关系说明图 — 图1：结构说明图，展示输入流、bufio.Reader 缓冲区窗口与 Peek 返回视图的关系；这是静态说明图，不是运行截图。

先拦截超过缓冲区的请求

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。完整边界应满足两者；短切片说明已经拿到部分输入，但错误才说明为什么没拿齐。第二张边界说明图将三个常见分支与处理动作对应起来。

Go bufio.Reader.Peek 的完整边界、io.EOF、ErrBufferFull 和底层错误分类说明图 — 图2：结果说明图，把 n 与缓冲区容量、实际可读字节数和错误类型对应起来；这是静态说明图，不是运行结果截图。

情况	返回特征	处理建议
刚好或超过 n 个字节	长度为 n，err 为 nil	可以比较头部；随后再消费
输入提前结束	长度小于 n，常见为 io.EOF	记录为不完整输入，不把短头部当合法帧
n 大于缓冲区	bufio.ErrBufferFull	调整帧头上限或改用分段读取
底层读取失败	保留底层错误	进入重试、断连或审计分支，不能静默吞掉

消费顺序和验证清单

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