check head request response

04733438 · Michael Yang · 711e891f · 04733438
Commit 04733438 authored Oct 03, 2023 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

server/download.go server/download.go +5 -0

No files found.
--- a/server/download.go
+++ b/server/download.go
@@ -78,6 +78,11 @@ func (b *blobDownload) Prepare(ctx context.Context, requestURL *url.URL, opts *R
 		}
 		defer resp.Body.Close()

+		if resp.StatusCode >= http.StatusBadRequest {
+			body, _ := io.ReadAll(resp.Body)
+			return fmt.Errorf("registry responded with code %d: %v", resp.StatusCode, string(body))
+		}
+
 		b.Total, _ = strconv.ParseInt(resp.Header.Get("Content-Length"), 10, 64)

 		var offset int64