From 3194c4c62ead09fbeb8af9421fe00840a12361f2 Mon Sep 17 00:00:00 2001
From: Jan Heylen <jan.heylen@nokia.com>
Date: Mon, 8 Feb 2021 12:17:48 +0100
Subject: [PATCH] partialzip: cope with big zip directories

---
 partialzip.go | 52 ++++++++++++++++++++++++++++++++-------------------
 1 file changed, 33 insertions(+), 19 deletions(-)

diff --git a/partialzip.go b/partialzip.go
index 016a259..b3c25e1 100644
--- a/partialzip.go
+++ b/partialzip.go
@@ -36,39 +36,53 @@ func New(url string) (*PartialZip, error) {
 
 func (p *PartialZip) init() error {
 	var client http.Client
-	var chuck int64 = 10 * 1024
+	var chuck int64 = 1024
+	var offset int64 = -1
+	var err error
+	var end *directoryEnd
+	var r *bytes.Reader
 
 	// get remote zip size
 	req, _ := http.NewRequest("HEAD", p.URL, nil)
 	resp, _ := client.Do(req)
 	p.Size = resp.ContentLength
 
-	// pull chuck from end of remote zip
-	reqRange := fmt.Sprintf("bytes=%d-%d", p.Size-chuck, p.Size)
-	req, _ = http.NewRequest("GET", p.URL, nil)
-	req.Header.Add("Range", reqRange)
-	resp, _ = client.Do(req)
+	for offset < 0 && chuck < (100 * 1024) {
+		//increase until we have enough to hold in the offset
+		chuck = 10 * chuck
 
-	body, err := ioutil.ReadAll(resp.Body)
-	if err != nil {
-		return errors.Wrap(err, "failed to read http response body")
-	}
-	defer resp.Body.Close()
+		// pull chuck from end of remote zip
+		reqRange := fmt.Sprintf("bytes=%d-%d", p.Size-chuck, p.Size)
+		req, _ = http.NewRequest("GET", p.URL, nil)
+		req.Header.Add("Range", reqRange)
+		resp, _ = client.Do(req)
+
+		body, err := ioutil.ReadAll(resp.Body)
+		if err != nil {
+			return errors.Wrap(err, "failed to read http response body")
+		}
+		defer resp.Body.Close()
 
-	r := bytes.NewReader(body)
+		r = bytes.NewReader(body)
 
-	// parse zip's directory end
-	end, err := readDirectoryEnd(r, chuck)
-	if err != nil {
-		return errors.Wrap(err, "failed to read directory end from remote zip")
+		// parse zip's directory end
+		end, err = readDirectoryEnd(r, chuck)
+		if err != nil {
+			return errors.Wrap(err, "failed to read directory end from remote zip")
+		}
+
+		// z.r = r
+		p.Files = make([]*File, 0, end.directoryRecords)
+
+		offset = int64(chuck - (p.Size - int64(end.directoryOffset)))
 	}
 
-	// z.r = r
-	p.Files = make([]*File, 0, end.directoryRecords)
 	// z.Comment = end.comment
 	rs := io.NewSectionReader(r, 0, chuck)
 
-	offset := int64(chuck - (p.Size - int64(end.directoryOffset)))
+	if offset < 0 {
+		return errors.Wrap(nil, "ivalid offset to begining of directory")
+	}
 	if _, err = rs.Seek(offset, io.SeekStart); err != nil {
 		return errors.Wrap(err, "failed to seek to begining of directory")
 	}