Merge pull request #1370 from maticnetwork/mardizzone/fixes

Limit bytes during response body read
This commit is contained in:
Marcello Ardizzone 2024-10-31 23:10:04 +01:00 committed by GitHub
commit 70f3205cd9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -31,9 +31,10 @@ var (
)
const (
stateFetchLimit = 50
apiHeimdallTimeout = 5 * time.Second
retryCall = 5 * time.Second
heimdallAPIBodyLimit = 128 * 1024 * 1024 // 128 MB
stateFetchLimit = 50
apiHeimdallTimeout = 5 * time.Second
retryCall = 5 * time.Second
)
type StateSyncEventsResponse struct {
@ -455,8 +456,11 @@ func internalFetch(ctx context.Context, client http.Client, u *url.URL) ([]byte,
return nil, nil
}
// Limit the number of bytes read from the response body
limitedBody := http.MaxBytesReader(nil, res.Body, heimdallAPIBodyLimit)
// get response
body, err := io.ReadAll(res.Body)
body, err := io.ReadAll(limitedBody)
if err != nil {
return nil, err
}