如何使用Golang实现文件上传限制_控制文件大小和类型

冬伟姑娘_5609

冬伟姑娘_5609

2025-12-21

382人浏览

原创

Go中文件上传限制需在解析阶段介入,通过ParseMultipartForm设置maxMemory(内存阈值)和maxFormSize(总请求体上限)实现;超出maxFormSize直接返回400错误。

如何使用golang实现文件上传限制_控制文件大小和类型

在 Go 中实现文件上传限制,关键在于解析 multipart 表单时主动校验,而不是等文件写入磁盘后再判断。Go 的 http.Request.ParseMultipartForm 默认会把整个文件读入内存或临时磁盘,若不加控制,大文件可能耗尽内存或填满磁盘。真正有效的限制需在读取阶段介入。

设置 ParseMultipartForm 的 maxMemory 和 maxFormSize

这是第一道防线。调用 r.ParseMultipartForm 前,必须显式指定内存阈值和总表单大小上限:

  • maxMemory:控制多少字节以内保留在内存(默认 32MB),超过则写入临时磁盘;设为 0 表示全部写磁盘,但无法阻止超大上传
  • maxFormSize:限制整个 multipart 请求体的总字节数(含文件+字段),单位是字节;超出会直接返回 http.ErrMissingFile 或 400 错误

示例:

r.ParseMultipartForm(32 r.ParseMultipartForm(0) // 禁用内存缓冲 → 仍无总大小限制
✅ 正确做法:
r.ParseMultipartForm(10 r.MultipartReader().NextPart() // 需配合后续校验

手动读取并校验文件头与 MIME 类型

浏览器提交的 Content-Type 可被伪造,仅依赖表单中声明的类型不安全。应在读取文件流时,用前 512 字节检测真实类型:

Golang Uber Dig
Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器,提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象,支持命名依赖等

下载
  • 调用 part.Header.Get("Content-Type") 获取客户端声称的类型(参考用)
  • 用 io.LimitReader(part, 512) 读取开头字节,传给 http.DetectContentType
  • 比对白名单(如 "image/jpeg", "application/pdf"),不匹配立即拒绝

注意:不要用文件扩展名判断类型,扩展名完全不可信。

边读边计数,实时拦截超大文件

即使设置了 maxFormSize,它只限制整个请求体,而单个文件可能占绝大部分。更稳妥的方式是在读取每个 multipart.Part 时,用 io.TeeReader 或自定义 reader 实时累加已读字节数:

  • 初始化计数器 size := int64(0)
  • 循环读取 part:每次 n, err := part.Read(buf) 后执行 size += int64(n)
  • 一旦 size > 10 * 1024 * 1024(如 10MB),立刻中断读取、清理资源、返回错误响应
  • 避免使用 part.Size 字段——它来自 header,可被篡改,不可信

完整流程建议顺序

  • 设置合理的 http.Server.ReadTimeout 和 MaxHeaderBytes 防慢速攻击
  • 调用 r.ParseMultipartForm 并指定 maxMemory(如 16MB)
  • 遍历 r.MultipartReader 获取每个 part,跳过非文件字段
  • 对每个文件 part:检测真实 MIME、实时计数大小、检查扩展名(辅助)、验证是否在白名单内
  • 任一校验失败,立即 http.Error(w, "Invalid file", http.StatusBadRequest) 并关闭 part
  • 全部通过后,再保存或处理文件

不复杂但容易忽略。核心是别让未经校验的数据流进你的处理逻辑。

golang免费学习笔记(深入):立即使用
在学习笔记中,你将探索golang的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

golang 字节

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.02.23

519

5

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.02.23

616

5

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.02.23

1216

5

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.05

791

5

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.05.21

1786

6

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

2025.06.09

5054

18

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

2025.06.10

2334

13

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

2025.06.17

4226

17

golang数组大全
golang数组大全

本专题整合了golang数组相关问题解决方法,想了解更多相关内容,请阅读下面的文章。

2025.06.20

5346

22

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程