Bladeren bron

Check source content length before getting download buffer

DarthSim 6 jaren geleden
bovenliggende
commit
c6145a8e69
1 gewijzigde bestanden met toevoegingen van 4 en 0 verwijderingen
  1. 4 0
      download.go

+ 4 - 0
download.go

@@ -125,6 +125,10 @@ func readAndCheckImage(ctx context.Context, res *http.Response) (context.Context
 
 
 	if res.ContentLength > 0 {
 	if res.ContentLength > 0 {
 		contentLength = int(res.ContentLength)
 		contentLength = int(res.ContentLength)
+
+		if conf.MaxSrcFileSize > 0 && contentLength > conf.MaxSrcFileSize {
+			return ctx, func() {}, errSourceFileTooBig
+		}
 	}
 	}
 
 
 	buf := downloadBufPool.Get(contentLength)
 	buf := downloadBufPool.Get(contentLength)