ensure compressed file header accounts for overflow

ybirader · Sep 6, 2023 · 9c31929 · 9c31929
1 parent 1fa8d1d
commit 9c31929
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/archiver.go b/archiver.go
@@ -253,7 +253,7 @@ func (a *Archiver) populateHeader(file *pool.File) error {
 	} else {
 		header.Method = zip.Deflate
 		header.Flags |= 0x8 // will write data descriptor (crc32, comp, uncomp)
-		header.CompressedSize64 = uint64(file.CompressedData.Len())
+		header.CompressedSize64 = uint64(file.Written())
 	}
 
 	file.Header = header

diff --git a/archiver_test.go b/archiver_test.go
@@ -163,6 +163,7 @@ func TestCompress(t *testing.T) {
 		assert.Equal(t, file.CompressedData.Len(), bufCap)
 		assert.Equal(t, pool.FileFull, file.Status)
 		assertGreaterThan(t, file.Written(), int64(file.CompressedData.Len()))
+		assert.Equal(t, file.Written(), int64(file.Header.CompressedSize64))
 
 		assert.NotZero(t, file.Overflow)
 		overflowInfo := testutils.GetFileInfo(t, file.Overflow.Name())