From 12a6d3977e548c3ccdf280ec6a88215552819e25 Mon Sep 17 00:00:00 2001 From: Richard Webb Date: Fri, 14 Dec 2018 22:44:44 +0000 Subject: [PATCH] Return the DirectoryEndHeader from SeekableZipHeaderFactory.ReadSeekable so that it can be used by ZipArchive --- .../Common/Zip/SeekableZipHeaderFactory.cs | 24 ++++++++++++------- 1 file changed, 15 insertions(+), 9 deletions(-) diff --git a/src/SharpCompress/Common/Zip/SeekableZipHeaderFactory.cs b/src/SharpCompress/Common/Zip/SeekableZipHeaderFactory.cs index 897b5a0a..23f4cbe8 100644 --- a/src/SharpCompress/Common/Zip/SeekableZipHeaderFactory.cs +++ b/src/SharpCompress/Common/Zip/SeekableZipHeaderFactory.cs @@ -17,7 +17,7 @@ namespace SharpCompress.Common.Zip { } - internal IEnumerable ReadSeekableHeader(Stream stream) + internal IEnumerable ReadSeekableHeader(Stream stream) { var reader = new BinaryReader(stream); @@ -51,16 +51,22 @@ namespace SharpCompress.Common.Zip { stream.Position = position; uint signature = reader.ReadUInt32(); - var directoryEntryHeader = ReadHeader(signature, reader, _zip64) as DirectoryEntryHeader; + var nextHeader = ReadHeader(signature, reader, _zip64); position = stream.Position; - if (directoryEntryHeader == null) - { - yield break; - } - //entry could be zero bytes so we need to know that. - directoryEntryHeader.HasData = directoryEntryHeader.CompressedSize != 0; - yield return directoryEntryHeader; + if (nextHeader == null) + yield break; + + if (nextHeader is DirectoryEntryHeader entryHeader) + { + //entry could be zero bytes so we need to know that. + entryHeader.HasData = entryHeader.CompressedSize != 0; + yield return entryHeader; + } + else if (nextHeader is DirectoryEndHeader endHeader) + { + yield return endHeader; + } } }