123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216 |
- using System;
- using System.IO;
- using ICSharpCode.SharpZipLib.Checksum;
- using ICSharpCode.SharpZipLib.Zip.Compression;
- using ICSharpCode.SharpZipLib.Zip.Compression.Streams;
- namespace ICSharpCode.SharpZipLib.GZip
- {
- /// <summary>
- /// This filter stream is used to compress a stream into a "GZIP" stream.
- /// The "GZIP" format is described in RFC 1952.
- ///
- /// author of the original java version : John Leuner
- /// </summary>
- /// <example> This sample shows how to gzip a file
- /// <code>
- /// using System;
- /// using System.IO;
- ///
- /// using ICSharpCode.SharpZipLib.GZip;
- /// using ICSharpCode.SharpZipLib.Core;
- ///
- /// class MainClass
- /// {
- /// public static void Main(string[] args)
- /// {
- /// using (Stream s = new GZipOutputStream(File.Create(args[0] + ".gz")))
- /// using (FileStream fs = File.OpenRead(args[0])) {
- /// byte[] writeData = new byte[4096];
- /// Streamutils.Copy(s, fs, writeData);
- /// }
- /// }
- /// }
- /// }
- /// </code>
- /// </example>
- public class GZipOutputStream : DeflaterOutputStream
- {
- enum OutputState
- {
- Header,
- Footer,
- Finished,
- Closed,
- };
- #region Instance Fields
- /// <summary>
- /// CRC-32 value for uncompressed data
- /// </summary>
- protected Crc32 crc = new Crc32();
- OutputState state_ = OutputState.Header;
- #endregion
- #region Constructors
- /// <summary>
- /// Creates a GzipOutputStream with the default buffer size
- /// </summary>
- /// <param name="baseOutputStream">
- /// The stream to read data (to be compressed) from
- /// </param>
- public GZipOutputStream(Stream baseOutputStream)
- : this(baseOutputStream, 4096)
- {
- }
- /// <summary>
- /// Creates a GZipOutputStream with the specified buffer size
- /// </summary>
- /// <param name="baseOutputStream">
- /// The stream to read data (to be compressed) from
- /// </param>
- /// <param name="size">
- /// Size of the buffer to use
- /// </param>
- public GZipOutputStream(Stream baseOutputStream, int size) : base(baseOutputStream, new Deflater(Deflater.DEFAULT_COMPRESSION, true), size)
- {
- }
- #endregion
- #region Public API
- /// <summary>
- /// Sets the active compression level (1-9). The new level will be activated
- /// immediately.
- /// </summary>
- /// <param name="level">The compression level to set.</param>
- /// <exception cref="ArgumentOutOfRangeException">
- /// Level specified is not supported.
- /// </exception>
- /// <see cref="Deflater"/>
- public void SetLevel(int level)
- {
- if (level < Deflater.BEST_SPEED) {
- throw new ArgumentOutOfRangeException("nameof(level)");
- }
- deflater_.SetLevel(level);
- }
- /// <summary>
- /// Get the current compression level.
- /// </summary>
- /// <returns>The current compression level.</returns>
- public int GetLevel()
- {
- return deflater_.GetLevel();
- }
- #endregion
- #region Stream overrides
- /// <summary>
- /// Write given buffer to output updating crc
- /// </summary>
- /// <param name="buffer">Buffer to write</param>
- /// <param name="offset">Offset of first byte in buf to write</param>
- /// <param name="count">Number of bytes to write</param>
- public override void Write(byte[] buffer, int offset, int count)
- {
- if (state_ == OutputState.Header) {
- WriteHeader();
- }
- if (state_ != OutputState.Footer) {
- throw new InvalidOperationException("Write not permitted in current state");
- }
- crc.Update(buffer, offset, count);
- base.Write(buffer, offset, count);
- }
- /// <summary>
- /// Writes remaining compressed output data to the output stream
- /// and closes it.
- /// </summary>
- protected override void Dispose(bool disposing)
- {
- try {
- Finish();
- } finally {
- if (state_ != OutputState.Closed) {
- state_ = OutputState.Closed;
- if (IsStreamOwner) {
- baseOutputStream_.Dispose();
- }
- }
- }
- }
- #endregion
- #region DeflaterOutputStream overrides
- /// <summary>
- /// Finish compression and write any footer information required to stream
- /// </summary>
- public override void Finish()
- {
- // If no data has been written a header should be added.
- if (state_ == OutputState.Header) {
- WriteHeader();
- }
- if (state_ == OutputState.Footer) {
- state_ = OutputState.Finished;
- base.Finish();
- var totalin = (uint)(deflater_.TotalIn & 0xffffffff);
- var crcval = (uint)(crc.Value & 0xffffffff);
- byte[] gzipFooter;
- unchecked {
- gzipFooter = new byte[] {
- (byte) crcval, (byte) (crcval >> 8),
- (byte) (crcval >> 16), (byte) (crcval >> 24),
- (byte) totalin, (byte) (totalin >> 8),
- (byte) (totalin >> 16), (byte) (totalin >> 24)
- };
- }
- baseOutputStream_.Write(gzipFooter, 0, gzipFooter.Length);
- }
- }
- #endregion
- #region Support Routines
- void WriteHeader()
- {
- if (state_ == OutputState.Header) {
- state_ = OutputState.Footer;
- var mod_time = (int)((DateTime.Now.Ticks - new DateTime(1970, 1, 1).Ticks) / 10000000L); // Ticks give back 100ns intervals
- byte[] gzipHeader = {
- // The two magic bytes
- (byte) (GZipConstants.GZIP_MAGIC >> 8), (byte) (GZipConstants.GZIP_MAGIC & 0xff),
- // The compression type
- (byte) Deflater.DEFLATED,
- // The flags (not set)
- 0,
- // The modification time
- (byte) mod_time, (byte) (mod_time >> 8),
- (byte) (mod_time >> 16), (byte) (mod_time >> 24),
- // The extra flags
- 0,
- // The OS type (unknown)
- (byte) 255
- };
- baseOutputStream_.Write(gzipHeader, 0, gzipHeader.Length);
- }
- }
- #endregion
- }
- }
|