验证图像文件的golang实现

知识点掌握了，还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战，手把手教大家学习《验证图像文件的golang实现》，在实现功能的过程中也带大家重新温习相关知识点，温故而知新，回头看看说不定又有不一样的感悟！

我们有一个 API，可以接受来自客户端的图像文件（jpeg、png 和 PDF）。避免 无限制文件上传漏洞，我们希望在服务器端实现文件内容验证。 golang 有规范的方法或库来做到这一点吗？我知道 http 包有 DetectContentType 函数，但我不确定它是否足够，因为它基于 MINE 嗅探并且只查看前 512 个字节。

正确答案

如果您只想接受列出的有效文件类型，那么您可以依靠 detectcontenttype。这些类型在其文件的开头具有众所周知的标记。事实上，大多数都可以通过查看远少于 512 字节的内容来识别。

组装 cli 工具并自行测试一些文件很容易。

这是我手头上的随机 jpg、pdf 和 gif 的前 5 个字节。

$ for i in {boo.jpg,AmStd_12-5062-01_final.pdf,tenor.gif}; \
     do echo "\n$i"; \
     head -c 5 $i | hexdump -C; \
done

boo.jpg
00000000  ff d8 ff e0 00                                    |.....|
00000005

AmStd_12-5062-01_final.pdf
00000000  25 50 44 46 2d                                    |%PDF-|
00000005

tenor.gif
00000000  47 49 46 38 39                                    |GIF89|
00000005

请注意，go 代码中定义的 jpg 标记为：[]byte("\xff\xd8\xff")，它们是 boo.jpg 输出的前 3 个字节。另外两个有 ascii 标记，因此更容易看到。

显然，文件可能会被篡改，使其具有与允许的文件类型之一匹配的签名，因此可能会被欺骗，并且可以上传“无效”的 pdf、jpg 或 gif。您如何使用该文件最终会影响您是否可以信任自动化系统。

编辑
看起来其他人已经制作了一个 cli 工具来检查文件。将文件名更改为标志，您就拥有了一个可以充当良好测试实用程序来验证正确性的工具。 https://golangcode.com/get-the-content-type-of-file/

终于介绍完啦！小伙伴们，这篇关于《验证图像文件的golang实现》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！