k2safe/go-omni
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: apply default HTTP2 body limit

Browse Source
This commit is contained in:
k2safe
2026-09-22 22:43:07 +08:00
parent 1487a96f67
commit 46c5c9406c
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
transport/http2/server.go
View File

@@ -209,15 +209,17 @@ func (s *Server) Serve(ctx context.Context, listener net.Listener) error {
} }
s.mu.Unlock() s.mu.Unlock()
maxBodyBytes := s.MaxBodyBytes
if maxBodyBytes <= 0 {
maxBodyBytes = defaultMaxBodyBytes
}
requestHandler := http.Handler(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) { requestHandler := http.Handler(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
if s.MaxBodyBytes > 0 { if request.ContentLength > maxBodyBytes {
if request.ContentLength > s.MaxBodyBytes { _ = request.Body.Close()
_ = request.Body.Close() http.Error(writer, "request body too large\n", http.StatusRequestEntityTooLarge)
http.Error(writer, "request body too large\n", http.StatusRequestEntityTooLarge) return
return
}
request.Body = http.MaxBytesReader(writer, request.Body, s.MaxBodyBytes)
} }
request.Body = http.MaxBytesReader(writer, request.Body, maxBodyBytes)
s.handler.ServeHTTP(writer, request) s.handler.ServeHTTP(writer, request)
})) }))
gate := &requestGate{server: s, next: requestHandler, slots: s.requestSlots} gate := &requestGate{server: s, next: requestHandler, slots: s.requestSlots}