这篇文章主要讲解了“Golang怎么限制木马图片上传服务器”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“Golang怎么限制木马图片上传服务器”吧!

项目背景

自己开发了一个云对象存储服务,用于存储某个项目中所有的文件资源。

问题

在进行安全测试时,自己遇到了一个问题:无法限制木马病毒模拟的图片文件进行上传。看似是一个图片后缀的文件,其实是一个木马脚本。比如图片文件:muma.jpeg,我们使用 UltraEdit 编辑器打开这个文件,文件内容如下图所示:

然而,一个正常的 jpeg 图片文件的内容如下图所示:

解决

那么如何解决这个问题呢?毫无疑问,我们要做的就是区分正真的图片和木马图片。

经过我一顿猛如虎的调研,终于找到了解决办法,那就利用 Golang 语言的标准库 —— image。

但是,有一个问题,image 库并没有支持所有图片类型的校验,目前仅支持 jpeg、gif、png 三种图片格式。我们封装一个方法,同时支持三种图片格式的真实性校验。

参考代码如下:

funcCheckImageFile(path,stylestring)(string,error){f,err:=os.Open(path)iferr!=nil{fmt.Errorf("打开文件失败%s",err.Error())}switchstrings.ToUpper(style){case"JPG","JPEG":_,err=jpeg.Decode(f)case"PNG":_,err=png.Decode(f)case"GIF":_,err=gif.Decode(f)}iferr!=nil{fmt.Errorf("校验文件类型失败%s",err.Error())return"",err}return"",nil}

调用上述方法时,只需要传入图片路径和后缀类型,就可以校验文件的真实性。非常方便,本人已经验证通过。

感谢各位的阅读,以上就是“Golang怎么限制木马图片上传服务器”的内容了,经过本文的学习后,相信大家对Golang怎么限制木马图片上传服务器这一问题有了更深刻的体会,具体使用情况还需要大家实践验证。这里是亿速云,小编将为大家推送更多相关知识点的文章,欢迎关注!