Merge pull request #235 from dbaumber/Issue-230

Issue #230: preserve the compression method when getting a compressed…
adamhathcock · May 23, 2017 · 7b81d18 · 7b81d18
2 parents 0f2d325 + 7d0acbc
commit 7b81d18
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 1 deletion.
diff --git a/src/SharpCompress/Archives/Zip/ZipArchiveEntry.cs b/src/SharpCompress/Archives/Zip/ZipArchiveEntry.cs
@@ -14,7 +14,15 @@ internal ZipArchiveEntry(ZipArchive archive, SeekableZipFilePart part)
 
         public virtual Stream OpenEntryStream()
         {
-            return Parts.Single().GetCompressedStream();
+            var filePart = Parts.Single() as ZipFilePart;
+            var compressionMethod = filePart.Header.CompressionMethod;
+            var stream = filePart.GetCompressedStream();
+            if (filePart.Header.CompressionMethod != compressionMethod)
+            {
+                filePart.Header.CompressionMethod = compressionMethod;
+            }
+
+            return stream; 
         }
 
         #region IArchiveEntry Members

diff --git a/tests/SharpCompress.Test/Zip/ZipArchiveTests.cs b/tests/SharpCompress.Test/Zip/ZipArchiveTests.cs
@@ -324,6 +324,24 @@ public void Zip_Deflate_WinzipAES_Read()
             VerifyFiles();
         }
 
+        [Fact]
+        public void Zip_Deflate_WinzipAES_MultiOpenEntryStream()
+        {
+            ResetScratch();
+            using (var reader = ZipArchive.Open(Path.Combine(TEST_ARCHIVES_PATH, "Zip.deflate.WinzipAES2.zip"), new ReaderOptions()
+            {
+                Password = "test"
+            }))
+            {
+                foreach (var entry in reader.Entries.Where(x => !x.IsDirectory))
+                {
+                    var stream = entry.OpenEntryStream();
+                    Assert.NotNull(stream);
+                    var ex = Record.Exception(() => stream = entry.OpenEntryStream());
+                    Assert.Null(ex);
+                }
+            }
+        }
 
         [Fact]
         public void Zip_BZip2_Pkware_Read()

diff --git a/tests/TestArchives/Archives/Zip.deflate.WinzipAES2.zip b/tests/TestArchives/Archives/Zip.deflate.WinzipAES2.zip