123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188 |
- using System;
- using System.Collections.Generic;
- using System.IO;
- using System.Linq;
- using SharpCompress.Common;
- using SharpCompress.Common.GZip;
- using SharpCompress.Readers;
- using SharpCompress.Readers.GZip;
- using SharpCompress.Writers;
- using SharpCompress.Writers.GZip;
- namespace SharpCompress.Archives.GZip
- {
- public class GZipArchive : AbstractWritableArchive<GZipArchiveEntry, GZipVolume>
- {
- #if !NO_FILE
- /// <summary>
- /// Constructor expects a filepath to an existing file.
- /// </summary>
- /// <param name="filePath"></param>
- /// <param name="readerOptions"></param>
- public static GZipArchive Open(string filePath, ReaderOptions readerOptions = null)
- {
- filePath.CheckNotNullOrEmpty("filePath");
- return Open(new FileInfo(filePath), readerOptions ?? new ReaderOptions());
- }
- /// <summary>
- /// Constructor with a FileInfo object to an existing file.
- /// </summary>
- /// <param name="fileInfo"></param>
- /// <param name="readerOptions"></param>
- public static GZipArchive Open(FileInfo fileInfo, ReaderOptions readerOptions = null)
- {
- fileInfo.CheckNotNull("fileInfo");
- return new GZipArchive(fileInfo, readerOptions ?? new ReaderOptions());
- }
- #endif
- /// <summary>
- /// Takes a seekable Stream as a source
- /// </summary>
- /// <param name="stream"></param>
- /// <param name="readerOptions"></param>
- public static GZipArchive Open(Stream stream, ReaderOptions readerOptions = null)
- {
- stream.CheckNotNull("stream");
- return new GZipArchive(stream, readerOptions ?? new ReaderOptions());
- }
- public static GZipArchive Create()
- {
- return new GZipArchive();
- }
- #if !NO_FILE
- /// <summary>
- /// Constructor with a FileInfo object to an existing file.
- /// </summary>
- /// <param name="fileInfo"></param>
- /// <param name="options"></param>
- internal GZipArchive(FileInfo fileInfo, ReaderOptions options)
- : base(ArchiveType.GZip, fileInfo, options)
- {
- }
- protected override IEnumerable<GZipVolume> LoadVolumes(FileInfo file)
- {
- return new GZipVolume(file, ReaderOptions).AsEnumerable();
- }
- public static bool IsGZipFile(string filePath)
- {
- return IsGZipFile(new FileInfo(filePath));
- }
- public static bool IsGZipFile(FileInfo fileInfo)
- {
- if (!fileInfo.Exists)
- {
- return false;
- }
- using (Stream stream = fileInfo.OpenRead())
- {
- return IsGZipFile(stream);
- }
- }
- public void SaveTo(string filePath)
- {
- SaveTo(new FileInfo(filePath));
- }
- public void SaveTo(FileInfo fileInfo)
- {
- using (var stream = fileInfo.Open(FileMode.Create, FileAccess.Write))
- {
- SaveTo(stream, new WriterOptions(CompressionType.GZip));
- }
- }
- #endif
- public static bool IsGZipFile(Stream stream)
- {
- // read the header on the first read
- byte[] header = new byte[10];
- // workitem 8501: handle edge case (decompress empty stream)
- if (!stream.ReadFully(header))
- {
- return false;
- }
- if (header[0] != 0x1F || header[1] != 0x8B || header[2] != 8)
- {
- return false;
- }
- return true;
- }
- /// <summary>
- /// Takes multiple seekable Streams for a multi-part archive
- /// </summary>
- /// <param name="stream"></param>
- /// <param name="options"></param>
- internal GZipArchive(Stream stream, ReaderOptions options)
- : base(ArchiveType.GZip, stream, options)
- {
- }
- internal GZipArchive()
- : base(ArchiveType.GZip)
- {
- }
- protected override GZipArchiveEntry CreateEntryInternal(string filePath, Stream source, long size, DateTime? modified,
- bool closeStream)
- {
- if (Entries.Any())
- {
- throw new InvalidOperationException("Only one entry is allowed in a GZip Archive");
- }
- return new GZipWritableArchiveEntry(this, source, filePath, size, modified, closeStream);
- }
- protected override void SaveTo(Stream stream, WriterOptions options,
- IEnumerable<GZipArchiveEntry> oldEntries,
- IEnumerable<GZipArchiveEntry> newEntries)
- {
- if (Entries.Count > 1)
- {
- throw new InvalidOperationException("Only one entry is allowed in a GZip Archive");
- }
- using (var writer = new GZipWriter(stream, new GZipWriterOptions(options)))
- {
- foreach (var entry in oldEntries.Concat(newEntries)
- .Where(x => !x.IsDirectory))
- {
- using (var entryStream = entry.OpenEntryStream())
- {
- writer.Write(entry.Key, entryStream, entry.LastModifiedTime);
- }
- }
- }
- }
- protected override IEnumerable<GZipVolume> LoadVolumes(IEnumerable<Stream> streams)
- {
- return new GZipVolume(streams.First(), ReaderOptions).AsEnumerable();
- }
- protected override IEnumerable<GZipArchiveEntry> LoadEntries(IEnumerable<GZipVolume> volumes)
- {
- Stream stream = volumes.Single().Stream;
- yield return new GZipArchiveEntry(this, new GZipFilePart(stream, ReaderOptions.ArchiveEncoding));
- }
- protected override IReader CreateReaderForSolidExtraction()
- {
- var stream = Volumes.Single().Stream;
- stream.Position = 0;
- return GZipReader.Open(stream);
- }
- }
- }
|