Go image.DecodeConfig 如何只读取图片尺寸不解码像素

处理用户上传图片时，我通常不会一上来就调用 image.Decode。更稳妥的做法是先用 image.DecodeConfig 读取图片头部，拿到格式、宽度、高度和颜色模型；只有像素总量在可接受范围内，才继续完整解码。这样既能提前拒绝超大图片，也能避免一次普通尺寸校验就分配整张像素缓冲区。

DecodeConfig 读取的是什么，为什么适合放在入口

image.Config包含 ColorModel、Width 和 Height。它描述的是图片的基本参数，不是已经展开的像素矩阵。因此入口层可以先计算 Width × Height，再决定是否允许继续处理。Go 官方文档也提醒，任意图片完整解码可能造成资源耗尽，处理不可信输入时应先调用 DecodeConfig。

需要区分两个结果：配置读取成功，只说明头部能被对应格式的解码器识别；它不等于整张图片已经通过完整性检查。真正需要像素时仍要调用 image.Decode 并处理它的错误。

先读头部，再用像素预算决定是否解码

下面的函数只演示入口判断。它把宽高转换成无符号整数后做乘法，并先检查乘法可能溢出的情况；业务中的上限应结合图片用途和进程内存设置，而不是照搬一个固定数字。

package main

import (
    "fmt"
    "image"
    _ "image/gif"  // 注册 GIF 解码器，让 DecodeConfig 能识别 GIF。
    _ "image/jpeg" // 注册 JPEG 解码器，让 DecodeConfig 能识别 JPEG。
    _ "image/png"  // 注册 PNG 解码器，让 DecodeConfig 能识别 PNG。
    "io"
    "math"
)

const maxPixels uint64 = 24_000_000 // 示例上限：按服务内存和业务图片类型调整。

func inspectHeader(r io.Reader) (image.Config, string, error) {
    cfg, format, err := image.DecodeConfig(r)
    if err != nil {
        return image.Config{}, "", fmt.Errorf("读取图片头部失败: %w", err) // 不把坏输入伪装成尺寸不合格。
    }
    if cfg.Width  math.MaxUint64/height || width*height > maxPixels {
        return image.Config{}, format, fmt.Errorf("图片像素量过大: %dx%d", cfg.Width, cfg.Height) // 在完整解码前止损。
    }
    return cfg, format, nil
}

这里的 format 是解码器注册时使用的格式名，例如 jpeg、png 或 gif。如果返回 “image: unknown format”，通常不是尺寸问题，而是对应格式包没有被导入，或输入头部并不是受支持的格式。

Reader 被消费后，完整 Decode 要重新定位

这是最容易在代码评审中漏掉的细节。DecodeConfig 为了识别头部会读取 Reader；如果随后直接对同一个普通文件 Reader 调用 image.Decode，读取位置可能已经不在开头。对可 Seek 的文件可以回到零点，更简单的上传流程则可以在通过配置检查后重新打开文件。

把格式、尺寸和完整解码分成三道判断

上传接口可以按“能否识别格式、像素量是否合规、完整解码是否成功”三层返回错误。第一层失败多半是输入格式或注册问题，第二层是业务限制，第三层才是文件内容损坏或解码器发现的异常。分开记录这三类原因，比统一返回“图片无效”更容易排查。

最后再强调一次：DecodeConfig 适合做尺寸和资源预算的前置检查，不是图片安全扫描器，也不能替代完整解码。对外部输入保留大小限制、超时和错误处理；需要真正读像素时，重新定位 Reader 后再调用 Decode。

相关问题

只导入 image 包为什么识别不了 PNG？

格式解码器通常通过包初始化注册。需要 PNG 时加入空白导入 _ "image/png"，JPEG 和 GIF 同理。

DecodeConfig 成功后还需要 Decode 吗？

如果只需要宽高和格式，不需要；如果要缩放、读取像素或保存新图片，仍要完整调用 image.Decode。

宽高相乘为什么还要检查溢出？

外部输入的尺寸可能异常巨大，直接相乘可能绕回小数值。先做除法边界检查，再与像素上限比较更稳妥。

参考资料：https://pkg.go.dev/image#DecodeConfig