Go crypto/rand.Read返回不足字节时的调用约束

用 Go 生成登录令牌、重置链接或临时密钥时，常见写法是准备一个固定长度的 []byte，再调用 crypto/rand.Read。这里有一个容易混淆的边界：顶层 rand.Read 的契约不是普通的“尽量读取”，而是填满整个切片；真正可能返回不足字节的，是你直接调用某个 io.Reader，或在测试中替换了随机源。

crypto/rand.Read不会悄悄返回半段令牌

标准库文档对 crypto/rand.Read 的描述很明确：它填充密码学安全的随机字节，不返回错误，并且总是完整填充参数切片。当前实现内部对随机源使用 io.ReadFull；如果底层随机源真的报错，程序会不可恢复地终止，而不是返回一个半成品给调用方。

所以固定长度令牌可以这样写。这里的 n 仍然保留，是为了把调用约束写清楚；对默认随机源而言，正常返回后它一定等于缓冲区长度。

package token

import (
	"crypto/rand"
	"encoding/base64"
)

// NewToken 生成固定长度的 URL 安全令牌。
func NewToken() (string, error) {
	buf := make([]byte, 32) // 32 字节提供固定的随机材料长度
	n, err := rand.Read(buf)
	if err != nil {
		return "", err // 保留错误分支，便于替换随机源或未来改用其他实现
	}
	if n != len(buf) {
		return "", ErrShortRandomRead // 完整令牌不能由半段随机数据组成
	}
	return base64.RawURLEncoding.EncodeToString(buf), nil
}

在默认实现下，err 不会以普通返回值出现，但显式检查仍能把“完整令牌”这个业务前提写进边界函数。若项目只接受标准随机源，也可以忽略 n，但不要把这条经验迁移到所有 io.Reader。

调用约束落在返回值和错误语义上

判断一次随机读取是否可交付，不能只看 n 是否大于零。对一个通用读取器，至少要区分下面三种结果：

io.Reader 的接口允许一次只返回部分数据。这个行为本身不等于随机源不安全，也不等于应该把已有字节补成令牌；它只说明“这次调用完成了部分工作”。是否继续读取，应由 io.ReadFull 或你的封装函数统一决定。

自定义Reader或底层io.Reader要用io.ReadFull

如果代码接收的是依赖注入的 io.Reader，不要复制顶层 crypto/rand.Read 的“总能填满”假设。用 io.ReadFull 把长度要求放在读取边界，并把错误交给调用者：

package token

import "io"

// ReadTokenBytes 从任意 Reader 读取完整令牌材料。
func ReadTokenBytes(r io.Reader, size int) ([]byte, error) {
	if size 

io.ReadFull 会持续读取，直到填满切片或出现错误；当输入提前结束时通常会得到 io.ErrUnexpectedEOF。因此它适合文件、网络流和测试替身等通用输入。对于默认 crypto/rand.Reader，则应直接使用 crypto/rand.Read 的标准契约，不要为了模拟短读在生产代码里自行拼接随机字节。

把令牌生成封装成可拒绝的边界

工程上更稳的做法是让上层只拿到“完整材料”或错误，不让它接触半成品。生产实现用 crypto/rand.Read，测试实现注入一个会短读的 Reader，再验证封装函数返回错误。这样可以覆盖调用约束，而不需要运行任何真实密钥流程。

还要注意两个细节：第一，crypto/rand 与 math/rand 的用途不同，密钥、会话令牌和重置链接应使用前者；第二，长度检查只能保证字节数，不负责令牌的过期、绑定用户、一次性消费或存储策略。

相关问题

rand.Read返回的n需要检查吗？

对默认的 crypto/rand.Read，正常返回时会填满切片，n 应为 len(b)。如果代码面对的是通用 Reader，则必须按 n 和 err 共同判断。

为什么不能只判断err为空？

因为普通 Reader 的接口允许短读；读取器可能返回部分数据而暂时没有错误。需要固定长度时使用 io.ReadFull，不要把部分缓冲区直接编码成令牌。

io.ReadFull和crypto/rand.Read怎么选？

直接从标准密码学随机源取字节时选 crypto/rand.Read；如果函数接收任意 io.Reader，或需要把短读转成可处理错误，就用 io.ReadFull。

一句话记忆：crypto/rand.Read 的正常返回意味着“整块已填满”，而通用 io.Reader 的一次返回只代表“本次读到这么多”。把这两个契约分开，令牌长度和错误处理就不会被偶然的短读带偏。