如何检查文件是否为有效图像？

Question

我正在构建一个 Web 应用程序。在其中一个页面上有一个上传表单，用户可以在其中上传文件。上传完成后，我想在服务器上检查上传的文件是否为图像。除了简单的文件扩展名检查（即不假设*.png文件名实际上是 PNG 图像）之外，是否可以对此进行检查？例如，如果我编辑 JPEG 图像，在随机位置添加/编辑一个字节以制作无效的 JPEG 文件，我想检测它不再是 JPEG 图像。前段时间我曾经使用 GD 库通过 PHP 做这种类型的事情。我想知道是否可以用 Go 来做？

Answer 1

通常所做的是检查文件是否具有您想要的图像文件格式的正确幻数。虽然这个测试不是非常准确，但通常已经足够了。你可以使用这样的代码：

package foo

import "strings"

// image formats and magic numbers

var magicTable = map[string]string{

    "\xff\xd8\xff":      "image/jpeg",

    "\x89PNG\r\n\x1a\n": "image/png",

    "GIF87a":            "image/gif",

    "GIF89a":            "image/gif",

}

// mimeFromIncipit returns the mime type of an image file from its first few

// bytes or the empty string if the file does not look like a known file type

func mimeFromIncipit(incipit []byte) string {

    incipitStr := []byte(incipit)

    for magic, mime := range magicTable {

        if strings.HasPrefix(incipitStr, magic) {

            return mime

        }

    }

    return ""

}