projectdiscovery/httpx
1
1
Fork 0
mirror of https://github.com/projectdiscovery/httpx.git synced 2024-11-28 22:01:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixing content-length calculation

Browse Source
This commit is contained in:
mzack 2021-09-03 15:50:17 +02:00
parent d350217bce
commit fc77c54395
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
common/httpx/httpx.go
View File

@ -194,13 +194,18 @@ get_response:
respbodystr = h.htmlPolicy.Sanitize(respbodystr)
}
if contentLength, ok := resp.Headers["Content-Length"]; ok {
contentLengthInt, err := strconv.Atoi(strings.Join(contentLength, ""))
if err != nil {
resp.ContentLength = utf8.RuneCountInString(respbodystr)
} else {
// if content length is not defined
if resp.ContentLength <= 0 {
// check if it's in the header and convert to int
if contentLength, ok := resp.Headers["Content-Length"]; ok {
contentLengthInt, _ := strconv.Atoi(strings.Join(contentLength, ""))
resp.ContentLength = contentLengthInt
}
// if we have a body, then use the number of bytes in the body if the length is still zero
if resp.ContentLength <= 0 && len(respbodystr) > 0 {
resp.ContentLength = utf8.RuneCountInString(respbodystr)
}
}
resp.Data = respbody