我正在构建一个 Web 应用程序。
在其中一个页面上有一个上传表单，用户可以在其中上传文件。上传完成后，我想在服务器上检查上传的文件是不是图片。
除了简单的文件扩展名检查（即不假设 *.png 文件名实际上是 PNG 图像）之外，是否可以检查这一点？
例如，如果我编辑 JPEG 图像，在随机位置添加/编辑一个字节以制作无效的 JPEG 文件，我想检测它不再是 JPEG 图像。前段时间我曾经使用 GD 库通过 PHP 做这种类型的事情。
我想知道是否可以用 Go 做？
【问题讨论】：
通常要做的是检查文件是否具有您想要的图像文件格式的正确magic number。虽然这个测试不是非常准确，但通常已经足够好了。你可以使用这样的代码：
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 ""
}
【讨论】：
image/png、image/gif、image/jpeg、golang.org/x/image/bmp 等）并使用@987654328 @ 或 image.Decode 并检查错误。但请注意解码器中针对恶意数据的错误。
http 包可以为你做到这一点：
func DetectContentType(data []byte) stringDetectContentType 实现了在 http://mimesniff.spec.whatwg.org/ 确定 Content-Type 的 给定的数据。它最多考虑前 512 个字节的数据。 DetectContentType 总是返回一个有效的 MIME 类型：如果它不能 确定一个更具体的，它返回“application/octet-stream”。
【讨论】：
DetectContentType 比手动检查幻数要好得多。使用很简单：
clientFile, _, _ := r.FormFile("img") // or get your file from a file system
defer clientFile.Close()
buff := make([]byte, 512) // docs tell that it take only first 512 bytes into consideration
if _, err = clientFile.Read(buff); err != nil {
fmt.Println(err) // do something with that error
return
}
fmt.Println(http.DetectContentType(buff)) // do something based on your detection.
使用这种方法你需要知道你仍然不能保证有一个正确的文件。所以我建议对该文件进行一些图像处理（比如调整它的大小以确保这确实是一张图像）。
【讨论】：