From 6707a01a846c459aa2f4bb51ff725193e4a7a6d3 Mon Sep 17 00:00:00 2001 From: r Date: Sun, 24 Sep 2023 10:38:28 +0000 Subject: Use a custom LimitedReader instead of http.MaxBytesReader Fixes compatibility with older Go versions. --- mastodon/http.go | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) (limited to 'mastodon') diff --git a/mastodon/http.go b/mastodon/http.go index ca4c771..7d1c1c4 100644 --- a/mastodon/http.go +++ b/mastodon/http.go @@ -9,15 +9,19 @@ import ( type lr struct { io.ReadCloser + n int64 r *http.Request } func (r *lr) Read(p []byte) (n int, err error) { - n, err = r.ReadCloser.Read(p) - // override the generic error returned by the MaxBytesReader - if _, ok := err.(*http.MaxBytesError); ok { - err = fmt.Errorf("%s \"%s\": response body too large", r.r.Method, r.r.URL) + if r.n <= 0 { + return 0, fmt.Errorf("%s \"%s\": response body too large", r.r.Method, r.r.URL) + } + if int64(len(p)) > r.n { + p = p[0:r.n] } + n, err = r.ReadCloser.Read(p) + r.n -= int64(n) return } @@ -28,7 +32,7 @@ type transport struct { func (t *transport) RoundTrip(r *http.Request) (*http.Response, error) { resp, err := t.t.RoundTrip(r) if resp != nil && resp.Body != nil { - resp.Body = &lr{http.MaxBytesReader(nil, resp.Body, 8<<20), r} + resp.Body = &lr{resp.Body, 8 << 20, r} } return resp, err } -- cgit v1.2.3