首先限制文件大小，通过r.ParseMultipartForm设置内存和总大小阈值，超出则报错；接着用http.DetectContentType读取前512字节检测MIME类型，防止仅依赖扩展名；再检查r.MultipartForm.File字段长度以限制文件数量，最多允许5个；最后建议校验Content-Length、使用随机文件名并解析图像头确保有效性，综合措施提升上传安全性。
在使用Golang处理表单文件上传时，限制文件大小、类型和数量是保障服务安全和稳定的关键措施。直接读取用户上传的文件而不加控制，可能导致服务器资源耗尽或恶意文件注入。下面介绍如何在标准库net/http基础上实现常见上传限制。
限制上传文件大小
Go的http.Request提供了ParseMultipartForm方法，可以预先设置内存中缓存的最大字节数，超出部分将写入临时文件。
通过设置该参数，可有效防止过大的文件占用内存：
- 调用
r.ParseMultipartForm(maxMemory)，其中maxMemory是你允许在内存中存储的最大字节数（如10MB） - 如果请求体超过此值，多余部分会自动写入磁盘临时文件
- 若整体文件超过你设定的总上限，可在解析后检查
r.MultipartForm.File中的文件大小
示例代码：
立即学习“go语言免费学习笔记（深入）”；
func uploadHandler(w http.ResponseWriter, r *http.Request) {
// 允许内存中最多10MB，整个请求不超过20MB
err := r.ParseMultipartForm(10 << 20)
if err != nil {
if err == http.ErrContentLengthExceeded {
http.Error(w, "上传文件过大", http.StatusBadRequest)
return
}
http.Error(w, "解析表单失败", http.StatusInternalServerError)
return
}
file, handler, err := r.FormFile("uploadFile")
if err != nil {
http.Error(w, "获取文件失败", http.StatusBadRequest)
return
}
defer file.Close()
// 检查文件实际大小
if handler.Size > 20<<20 {
http.Error(w, "文件不能超过20MB", http.StatusBadRequest)
return
}
// 正常处理文件...
}
限制文件类型（MIME类型检测）
仅靠文件扩展名判断类型容易被绕过，应读取文件头部几个字节进行MIME类型识别。
Go标准库http.DetectContentType可以根据前512个字节推断内容类型。
- 从
multipart.File读取前512字节 - 使用
http.DetectContentType获取MIME类型 - 比对是否在允许列表中
示例：
fileBytes := make([]byte, 512)
_, err = file.Read(fileBytes)
if err != nil {
http.Error(w, "读取文件出错", http.StatusInternalServerError)
return
}
contentType := http.DetectContentType(fileBytes)
allowedTypes := map[string]bool{
"image/jpeg": true,
"image/png": true,
"image/gif": true,
}
if !allowedTypes[contentType] {
http.Error(w, "不支持的文件类型", http.StatusBadRequest)
return
}
// 注意：Read后需要重置文件指针
file.Seek(0, 0)
限制上传文件数量
多个文件上传时，可通过遍历FormFile字段来计数并逐一校验。
例如，限制一次最多上传5个文件：
files := r.MultipartForm.File["uploadFiles"]
if len(files) > 5 {
http.Error(w, "最多上传5个文件", http.StatusBadRequest)
return
}
再结合循环对每个文件做大小和类型检查。
其他建议
除了上述基本限制，还可考虑：
- 设置HTTP请求头
Content-Length初步判断，若超过阈值直接拒绝（需注意客户端可能伪造） - 保存文件时使用随机文件名，避免路径遍历或覆盖系统文件
- 对图片类文件可进一步用
image.DecodeConfig确认是否为有效图像
基本上就这些。合理组合大小、类型和数量限制，能有效提升文件上传接口的安全性。不复杂但容易忽略细节。
