forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-07-27 07:06:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #1370 from maticnetwork/mardizzone/fixes

Browse source 
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
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
consensus/bor/heimdall/client.go
View file

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