mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-17 19:35:27 +00:00
c7c35ffb94
* fix mem leak * bump version tag * http: add global resp body read limit of 4MB * skip creating templateCtx in normal templates * fix mem leak via retryablehttp , fastdialer * go mod tidy * remove unused var * dep update --------- Co-authored-by: sandeep <8293321+ehsandeep@users.noreply.github.com>
34 lines
680 B
Go
34 lines
680 B
Go
package utils
|
|
|
|
import (
|
|
"io"
|
|
)
|
|
|
|
var (
|
|
MaxBodyRead = int64(1 << 22) // 4MB using shift operator
|
|
)
|
|
|
|
var _ io.ReadCloser = &LimitResponseBody{}
|
|
|
|
type LimitResponseBody struct {
|
|
io.Reader
|
|
io.Closer
|
|
}
|
|
|
|
// NewLimitResponseBody wraps response body with a limit reader.
|
|
// thus only allowing MaxBodyRead bytes to be read. i.e 4MB
|
|
func NewLimitResponseBody(body io.ReadCloser) io.ReadCloser {
|
|
if body == nil {
|
|
return nil
|
|
}
|
|
return &LimitResponseBody{
|
|
Reader: io.LimitReader(body, MaxBodyRead),
|
|
Closer: body,
|
|
}
|
|
}
|
|
|
|
// LimitBodyRead limits the body read to MaxBodyRead bytes.
|
|
func LimitBodyRead(r io.Reader) ([]byte, error) {
|
|
return io.ReadAll(io.LimitReader(r, MaxBodyRead))
|
|
}
|