fix panic during quota sync (#137)

apigee · Jun 7, 2023 · 777c318 · 777c318
1 parent c9a3d3e
commit 777c318
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/analytics/legacy_analytics.go b/analytics/legacy_analytics.go
@@ -64,7 +64,11 @@ func (oa *legacyAnalytics) SendRecords(authContext *auth.Context, records []Reco
 	}
 	defer resp.Body.Close()
 
-	buf := bytes.NewBuffer(make([]byte, 0, resp.ContentLength))
+	bufLen := resp.ContentLength
+	if bufLen < bytes.MinRead {
+		bufLen = bytes.MinRead
+	}
+	buf := bytes.NewBuffer(make([]byte, 0, bufLen))
 	_, err = buf.ReadFrom(resp.Body)
 	if err != nil {
 		return err

diff --git a/quota/bucket.go b/quota/bucket.go
@@ -162,7 +162,11 @@ func (b *bucket) sync() error {
 	}
 	defer resp.Body.Close()
 
-	buf := bytes.NewBuffer(make([]byte, 0, resp.ContentLength))
+	bufLen := resp.ContentLength
+	if bufLen < bytes.MinRead {
+		bufLen = bytes.MinRead
+	}
+	buf := bytes.NewBuffer(make([]byte, 0, bufLen))
 	if _, err = buf.ReadFrom(resp.Body); err != nil {
 		return errors.Wrap(err, "read body")
 	}