homelab/traefik
1
0
Fork 0
Code Activity

Handle responses without content length header

Browse source 
Co-authored-by: Kevin Pollet <pollet.kevin@gmail.com>
This commit is contained in:
Romain 2025-01-23 10:00:05 +01:00 • committed by GitHub
parent 4ce4bd7121
commit fb527dac1c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 57 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

18
pkg/proxy/fast/connpool.go
View file

@ -216,7 +216,9 @@ func (c *conn) handleResponse(r rwWithUpgrade) error {
return nil
}
if res.Header.ContentLength() == 0 {
hasContentLength := res.Header.Peek("Content-Length") != nil
if hasContentLength && res.Header.ContentLength() == 0 {
return nil
}
@ -226,6 +228,20 @@ func (c *conn) handleResponse(r rwWithUpgrade) error {
return nil
}
if !hasContentLength {
b := c.bufferPool.Get()
if b == nil {
b = make([]byte, bufferSize)
}
defer c.bufferPool.Put(b)
if _, err := io.CopyBuffer(r.RW, c.br, b); err != nil {
return err
}
return nil
}
// Chunked response, Content-Length is set to -1 by FastProxy when "Transfer-Encoding: chunked" header is received.
if res.Header.ContentLength() == -1 {
cbr := httputil.NewChunkedReader(c.br)