登录
首页 >  Golang >  Go问答

使用 golang 时是否建议(进一步)限制表单的大小?

来源:Golang技术栈

时间:2023-04-10 14:54:46 211浏览 收藏

今日不肯埋头,明日何以抬头!每日一句努力自己的话哈哈~哈喽,今天我将给大家带来一篇《使用 golang 时是否建议(进一步)限制表单的大小?》,主要内容是讲解golang等等,感兴趣的朋友可以收藏或者有更好的建议在评论提出,我都会认真看的!大家一起进步,一起学习!

问题内容

我四处搜索,据我所知,POST 表单请求已经限制为 10MB ( http://golang.org/src/net/http/request.go#L721 )。

如果我要在我的ServeHTTP方法中减少这种情况,我不确定如何正确地做到这一点。我会尝试这样的事情:

r.Body = http.MaxBytesReader(w, r.Body, MaxFileSize) 
err := r.ParseForm()
if err != nil {
     //redirect to some error page
     return
}

但是返回错误也会关闭连接吗?我将如何避免阅读所有内容?我发现了这个:https ://stackoverflow.com/a/26393261/2202497 ,但是如果没有设置内容长度并且在阅读过程中我意识到文件太大了怎么办。

我将此用作安全措施,以防止有人占用我的服务器资源。

正确答案

限制请求正文大小的正确方法是按照您的建议进行操作:

r.Body = http.MaxBytesReader(w, r.Body, MaxFileSize) 
err := r.ParseForm()
if err != nil {
 // redirect or set error status code.
 return
}

达到限制时,MaxBytesReader在响应上设置一个标志。设置此标志时,服务器不会读取请求正文的其余部分,并且服务器会在处理程序返回时关闭连接。

如果您担心恶意客户端,那么您还应该设置Server.ReadTimeoutServer.WriteTimeout和可能的Server.MaxHeaderBytes

如果要为所有处理程序设置请求正文限制,请使用在委托给根处理程序之前设置限制的处理程序包装根处理程序:

 type maxBytesHandler struct {
     h http.Handler
     n int64
 }

 func (h *maxBytesHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
     r.Body = http.MaxBytesReader(w, r.Body, h.n) 
     h.h.ServeHTTP(w, r)
 }

调用 ListenAndServe 时包装根处理程序:

log.Fatal(http.ListenAndServe(":8080", &maxBytesHandler{h:mux, n:4096))

或者在配置服务器时:

s := http.Server{
    Addr: ":8080",
    Handler: &maxBytesReader{h:mux, n:4096},
}
log.Fatal(s.ListenAndServe())

不需要另一个答案中建议的补丁。MaxBytesReader 是官方限制请求体大小的方式。

文中关于golang的知识介绍,希望对你的学习有所帮助!若是受益匪浅,那就动动鼠标收藏这篇《使用 golang 时是否建议(进一步)限制表单的大小?》文章吧,也可关注golang学习网公众号了解相关技术文章。

声明:本文转载于:Golang技术栈 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>