mirror of https://github.com/SixLabors/ImageSharp
James Jackson-South
7 years ago
9 changed files with 317 additions and 669 deletions
@ -1,466 +1,155 @@ |
|||||
// Copyright (c) Six Labors and contributors.
|
// Copyright (c) Six Labors and contributors.
|
||||
// Licensed under the Apache License, Version 2.0.
|
// Licensed under the Apache License, Version 2.0.
|
||||
|
|
||||
// <auto-generated/>
|
|
||||
using System; |
using System; |
||||
using System.Collections.Generic; |
|
||||
using System.IO; |
using System.IO; |
||||
using System.Text; |
using SixLabors.Memory; |
||||
|
|
||||
namespace SixLabors.ImageSharp.Formats.Png.Zlib |
namespace SixLabors.ImageSharp.Formats.Png.Zlib |
||||
{ |
{ |
||||
/// <summary>
|
/// <summary>
|
||||
/// A special stream deflating or compressing the bytes that are
|
/// A special stream deflating or compressing the bytes that are
|
||||
/// written to it. It uses a Deflater to perform actual deflating.<br/>
|
/// written to it. It uses a Deflater to perform actual deflating.
|
||||
/// Authors of the original java version : Tom Tromey, Jochen Hoenicke
|
|
||||
/// </summary>
|
/// </summary>
|
||||
public class DeflaterOutputStream : Stream |
internal sealed class DeflaterOutputStream : Stream |
||||
{ |
{ |
||||
#region Constructors
|
private const int BufferLength = 512; |
||||
/// <summary>
|
private IManagedByteBuffer memoryOwner; |
||||
/// Creates a new DeflaterOutputStream with the given Deflater and
|
private readonly byte[] buffer; |
||||
/// default buffer size.
|
private Deflater deflater; |
||||
/// </summary>
|
private readonly Stream rawStream; |
||||
/// <param name="baseOutputStream">
|
private bool isDisposed; |
||||
/// the output stream where deflated output should be written.
|
|
||||
/// </param>
|
|
||||
/// <param name="deflater">
|
|
||||
/// the underlying deflater.
|
|
||||
/// </param>
|
|
||||
public DeflaterOutputStream(Stream baseOutputStream, Deflater deflater) |
|
||||
: this(baseOutputStream, deflater, 512) |
|
||||
{ |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
/// <summary>
|
||||
/// Creates a new DeflaterOutputStream with the given Deflater and
|
/// Initializes a new instance of the <see cref="DeflaterOutputStream"/> class.
|
||||
/// buffer size.
|
|
||||
/// </summary>
|
/// </summary>
|
||||
/// <param name="baseOutputStream">
|
/// <param name="memoryAllocator">The memory allocator to use for buffer allocations.</param>
|
||||
/// The output stream where deflated output is written.
|
/// <param name="rawStream">The output stream where deflated output is written.</param>
|
||||
/// </param>
|
/// <param name="compressionLevel">The compression level.</param>
|
||||
/// <param name="deflater">
|
public DeflaterOutputStream(MemoryAllocator memoryAllocator, Stream rawStream, int compressionLevel) |
||||
/// The underlying deflater to use
|
|
||||
/// </param>
|
|
||||
/// <param name="bufferSize">
|
|
||||
/// The buffer size in bytes to use when deflating (minimum value 512)
|
|
||||
/// </param>
|
|
||||
/// <exception cref="ArgumentOutOfRangeException">
|
|
||||
/// bufsize is less than or equal to zero.
|
|
||||
/// </exception>
|
|
||||
/// <exception cref="ArgumentException">
|
|
||||
/// baseOutputStream does not support writing
|
|
||||
/// </exception>
|
|
||||
/// <exception cref="ArgumentNullException">
|
|
||||
/// deflater instance is null
|
|
||||
/// </exception>
|
|
||||
public DeflaterOutputStream(Stream baseOutputStream, Deflater deflater, int bufferSize) |
|
||||
{ |
{ |
||||
if (baseOutputStream == null) |
this.rawStream = rawStream; |
||||
{ |
this.memoryOwner = memoryAllocator.AllocateManagedByteBuffer(BufferLength); |
||||
throw new ArgumentNullException(nameof(baseOutputStream)); |
this.buffer = this.memoryOwner.Array; |
||||
} |
this.deflater = new Deflater(memoryAllocator, compressionLevel); |
||||
|
|
||||
if (baseOutputStream.CanWrite == false) |
|
||||
{ |
|
||||
throw new ArgumentException("Must support writing", nameof(baseOutputStream)); |
|
||||
} |
|
||||
|
|
||||
if (bufferSize < 512) |
|
||||
{ |
|
||||
throw new ArgumentOutOfRangeException(nameof(bufferSize)); |
|
||||
} |
|
||||
|
|
||||
baseOutputStream_ = baseOutputStream; |
|
||||
buffer_ = new byte[bufferSize]; |
|
||||
deflater_ = deflater ?? throw new ArgumentNullException(nameof(deflater)); |
|
||||
} |
} |
||||
|
|
||||
#endregion Constructors
|
/// <inheritdoc/>
|
||||
|
public override bool CanRead => false; |
||||
|
|
||||
#region Public API
|
/// <inheritdoc/>
|
||||
|
public override bool CanSeek => false; |
||||
|
|
||||
/// <summary>
|
/// <inheritdoc/>
|
||||
/// Finishes the stream by calling finish() on the deflater.
|
public override bool CanWrite => this.rawStream.CanWrite; |
||||
/// </summary>
|
|
||||
/// <exception cref="ImageFormatException">
|
|
||||
/// Not all input is deflated
|
|
||||
/// </exception>
|
|
||||
public virtual void Finish() |
|
||||
{ |
|
||||
deflater_.Finish(); |
|
||||
while (!deflater_.IsFinished) |
|
||||
{ |
|
||||
int len = deflater_.Deflate(buffer_, 0, buffer_.Length); |
|
||||
if (len <= 0) |
|
||||
{ |
|
||||
break; |
|
||||
} |
|
||||
|
|
||||
baseOutputStream_.Write(buffer_, 0, len); |
/// <inheritdoc/>
|
||||
} |
public override long Length => this.rawStream.Length; |
||||
|
|
||||
if (!deflater_.IsFinished) |
/// <inheritdoc/>
|
||||
|
public override long Position |
||||
|
{ |
||||
|
get |
||||
{ |
{ |
||||
throw new ImageFormatException("Can't deflate all input?"); |
return this.rawStream.Position; |
||||
} |
} |
||||
|
|
||||
baseOutputStream_.Flush(); |
set |
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Gets or sets a flag indicating ownership of underlying stream.
|
|
||||
/// When the flag is true <see cref="Stream.Dispose()" /> will close the underlying stream also.
|
|
||||
/// </summary>
|
|
||||
/// <remarks>The default value is true.</remarks>
|
|
||||
public bool IsStreamOwner { get; set; } = true; |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Allows client to determine if an entry can be patched after its added
|
|
||||
/// </summary>
|
|
||||
public bool CanPatchEntries |
|
||||
{ |
|
||||
get |
|
||||
{ |
{ |
||||
return baseOutputStream_.CanSeek; |
throw new NotSupportedException(); |
||||
} |
} |
||||
} |
} |
||||
|
|
||||
#endregion Public API
|
/// <inheritdoc/>
|
||||
|
public override long Seek(long offset, SeekOrigin origin) => throw new NotSupportedException(); |
||||
//#region Encryption
|
|
||||
|
|
||||
//private string password;
|
|
||||
|
|
||||
//private ICryptoTransform cryptoTransform_;
|
|
||||
|
|
||||
///// <summary>
|
/// <inheritdoc/>
|
||||
///// Returns the 10 byte AUTH CODE to be appended immediately following the AES data stream.
|
public override void SetLength(long value) => throw new NotSupportedException(); |
||||
///// </summary>
|
|
||||
//protected byte[] AESAuthCode;
|
|
||||
|
|
||||
///// <summary>
|
/// <inheritdoc/>
|
||||
///// Get/set the password used for encryption.
|
public override int ReadByte() => throw new NotSupportedException(); |
||||
///// </summary>
|
|
||||
///// <remarks>When set to null or if the password is empty no encryption is performed</remarks>
|
|
||||
//public string Password
|
|
||||
//{
|
|
||||
// get
|
|
||||
// {
|
|
||||
// return password;
|
|
||||
// }
|
|
||||
// set
|
|
||||
// {
|
|
||||
// if ((value != null) && (value.Length == 0))
|
|
||||
// {
|
|
||||
// password = null;
|
|
||||
// }
|
|
||||
// else
|
|
||||
// {
|
|
||||
// password = value;
|
|
||||
// }
|
|
||||
// }
|
|
||||
//}
|
|
||||
|
|
||||
///// <summary>
|
/// <inheritdoc/>
|
||||
///// Encrypt a block of data
|
public override int Read(byte[] buffer, int offset, int count) => throw new NotSupportedException(); |
||||
///// </summary>
|
|
||||
///// <param name="buffer">
|
|
||||
///// Data to encrypt. NOTE the original contents of the buffer are lost
|
|
||||
///// </param>
|
|
||||
///// <param name="offset">
|
|
||||
///// Offset of first byte in buffer to encrypt
|
|
||||
///// </param>
|
|
||||
///// <param name="length">
|
|
||||
///// Number of bytes in buffer to encrypt
|
|
||||
///// </param>
|
|
||||
//protected void EncryptBlock(byte[] buffer, int offset, int length)
|
|
||||
//{
|
|
||||
// cryptoTransform_.TransformBlock(buffer, 0, length, buffer, 0);
|
|
||||
//}
|
|
||||
|
|
||||
///// <summary>
|
/// <inheritdoc/>
|
||||
///// Initializes encryption keys based on given <paramref name="password"/>.
|
public override void Flush() |
||||
///// </summary>
|
{ |
||||
///// <param name="password">The password.</param>
|
this.deflater.Flush(); |
||||
//protected void InitializePassword(string password)
|
this.Deflate(true); |
||||
//{
|
this.rawStream.Flush(); |
||||
// var pkManaged = new PkzipClassicManaged();
|
} |
||||
// byte[] key = PkzipClassic.GenerateKeys(ZipStrings.ConvertToArray(password));
|
|
||||
// cryptoTransform_ = pkManaged.CreateEncryptor(key, null);
|
|
||||
//}
|
|
||||
|
|
||||
///// <summary>
|
|
||||
///// Initializes encryption keys based on given password.
|
|
||||
///// </summary>
|
|
||||
//protected void InitializeAESPassword(ZipEntry entry, string rawPassword,
|
|
||||
// out byte[] salt, out byte[] pwdVerifier)
|
|
||||
//{
|
|
||||
// salt = new byte[entry.AESSaltLen];
|
|
||||
// // Salt needs to be cryptographically random, and unique per file
|
|
||||
// if (_aesRnd == null)
|
|
||||
// _aesRnd = RandomNumberGenerator.Create();
|
|
||||
// _aesRnd.GetBytes(salt);
|
|
||||
// int blockSize = entry.AESKeySize / 8; // bits to bytes
|
|
||||
|
|
||||
// cryptoTransform_ = new ZipAESTransform(rawPassword, salt, blockSize, true);
|
|
||||
// pwdVerifier = ((ZipAESTransform)cryptoTransform_).PwdVerifier;
|
|
||||
//}
|
|
||||
|
|
||||
//#endregion Encryption
|
|
||||
|
|
||||
#region Deflation Support
|
|
||||
|
|
||||
/// <summary>
|
/// <inheritdoc/>
|
||||
/// Deflates everything in the input buffers. This will call
|
public override void Write(byte[] buffer, int offset, int count) |
||||
/// <code>def.deflate()</code> until all bytes from the input buffers
|
|
||||
/// are processed.
|
|
||||
/// </summary>
|
|
||||
protected void Deflate() |
|
||||
{ |
{ |
||||
Deflate(false); |
this.deflater.SetInput(buffer, offset, count); |
||||
|
this.Deflate(); |
||||
} |
} |
||||
|
|
||||
|
private void Deflate() => this.Deflate(false); |
||||
|
|
||||
private void Deflate(bool flushing) |
private void Deflate(bool flushing) |
||||
{ |
{ |
||||
while (flushing || !deflater_.IsNeedingInput) |
while (flushing || !this.deflater.IsNeedingInput) |
||||
{ |
{ |
||||
int deflateCount = deflater_.Deflate(buffer_, 0, buffer_.Length); |
int deflateCount = this.deflater.Deflate(this.buffer, 0, BufferLength); |
||||
|
|
||||
if (deflateCount <= 0) |
if (deflateCount <= 0) |
||||
{ |
{ |
||||
break; |
break; |
||||
} |
} |
||||
//if (cryptoTransform_ != null)
|
|
||||
//{
|
|
||||
// EncryptBlock(buffer_, 0, deflateCount);
|
|
||||
//}
|
|
||||
|
|
||||
baseOutputStream_.Write(buffer_, 0, deflateCount); |
this.rawStream.Write(this.buffer, 0, deflateCount); |
||||
} |
} |
||||
|
|
||||
if (!deflater_.IsNeedingInput) |
if (!this.deflater.IsNeedingInput) |
||||
{ |
{ |
||||
throw new ImageFormatException("DeflaterOutputStream can't deflate all input?"); |
DeflateThrowHelper.ThrowNoDeflate(); |
||||
} |
} |
||||
} |
} |
||||
|
|
||||
#endregion Deflation Support
|
private void Finish() |
||||
|
|
||||
#region Stream Overrides
|
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Gets value indicating stream can be read from
|
|
||||
/// </summary>
|
|
||||
public override bool CanRead |
|
||||
{ |
{ |
||||
get |
this.deflater.Finish(); |
||||
|
while (!this.deflater.IsFinished) |
||||
{ |
{ |
||||
return false; |
int len = this.deflater.Deflate(this.buffer, 0, BufferLength); |
||||
} |
if (len <= 0) |
||||
} |
{ |
||||
|
break; |
||||
|
} |
||||
|
|
||||
/// <summary>
|
this.rawStream.Write(this.buffer, 0, len); |
||||
/// Gets a value indicating if seeking is supported for this stream
|
|
||||
/// This property always returns false
|
|
||||
/// </summary>
|
|
||||
public override bool CanSeek |
|
||||
{ |
|
||||
get |
|
||||
{ |
|
||||
return false; |
|
||||
} |
} |
||||
} |
|
||||
|
|
||||
/// <summary>
|
if (!this.deflater.IsFinished) |
||||
/// Get value indicating if this stream supports writing
|
|
||||
/// </summary>
|
|
||||
public override bool CanWrite |
|
||||
{ |
|
||||
get |
|
||||
{ |
{ |
||||
return baseOutputStream_.CanWrite; |
DeflateThrowHelper.ThrowNoDeflate(); |
||||
} |
} |
||||
} |
|
||||
|
|
||||
/// <summary>
|
this.rawStream.Flush(); |
||||
/// Get current length of stream
|
|
||||
/// </summary>
|
|
||||
public override long Length |
|
||||
{ |
|
||||
get |
|
||||
{ |
|
||||
return baseOutputStream_.Length; |
|
||||
} |
|
||||
} |
} |
||||
|
|
||||
/// <summary>
|
/// <inheritdoc/>
|
||||
/// Gets the current position within the stream.
|
protected override void Dispose(bool disposing) |
||||
/// </summary>
|
|
||||
/// <exception cref="NotSupportedException">Any attempt to set position</exception>
|
|
||||
public override long Position |
|
||||
{ |
{ |
||||
get |
if (this.isDisposed) |
||||
{ |
|
||||
return baseOutputStream_.Position; |
|
||||
} |
|
||||
set |
|
||||
{ |
{ |
||||
throw new NotSupportedException("Position property not supported"); |
return; |
||||
} |
} |
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Sets the current position of this stream to the given value. Not supported by this class!
|
|
||||
/// </summary>
|
|
||||
/// <param name="offset">The offset relative to the <paramref name="origin"/> to seek.</param>
|
|
||||
/// <param name="origin">The <see cref="SeekOrigin"/> to seek from.</param>
|
|
||||
/// <returns>The new position in the stream.</returns>
|
|
||||
/// <exception cref="NotSupportedException">Any access</exception>
|
|
||||
public override long Seek(long offset, SeekOrigin origin) |
|
||||
{ |
|
||||
throw new NotSupportedException("DeflaterOutputStream Seek not supported"); |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Sets the length of this stream to the given value. Not supported by this class!
|
|
||||
/// </summary>
|
|
||||
/// <param name="value">The new stream length.</param>
|
|
||||
/// <exception cref="NotSupportedException">Any access</exception>
|
|
||||
public override void SetLength(long value) |
|
||||
{ |
|
||||
throw new NotSupportedException("DeflaterOutputStream SetLength not supported"); |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Read a byte from stream advancing position by one
|
|
||||
/// </summary>
|
|
||||
/// <returns>The byte read cast to an int. THe value is -1 if at the end of the stream.</returns>
|
|
||||
/// <exception cref="NotSupportedException">Any access</exception>
|
|
||||
public override int ReadByte() |
|
||||
{ |
|
||||
throw new NotSupportedException("DeflaterOutputStream ReadByte not supported"); |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Read a block of bytes from stream
|
|
||||
/// </summary>
|
|
||||
/// <param name="buffer">The buffer to store read data in.</param>
|
|
||||
/// <param name="offset">The offset to start storing at.</param>
|
|
||||
/// <param name="count">The maximum number of bytes to read.</param>
|
|
||||
/// <returns>The actual number of bytes read. Zero if end of stream is detected.</returns>
|
|
||||
/// <exception cref="NotSupportedException">Any access</exception>
|
|
||||
public override int Read(byte[] buffer, int offset, int count) |
|
||||
{ |
|
||||
throw new NotSupportedException("DeflaterOutputStream Read not supported"); |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Flushes the stream by calling <see cref="DeflaterOutputStream.Flush">Flush</see> on the deflater and then
|
|
||||
/// on the underlying stream. This ensures that all bytes are flushed.
|
|
||||
/// </summary>
|
|
||||
public override void Flush() |
|
||||
{ |
|
||||
deflater_.Flush(); |
|
||||
Deflate(true); |
|
||||
baseOutputStream_.Flush(); |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
if (disposing) |
||||
/// Calls <see cref="Finish"/> and closes the underlying
|
|
||||
/// stream when <see cref="IsStreamOwner"></see> is true.
|
|
||||
/// </summary>
|
|
||||
protected override void Dispose(bool disposing) |
|
||||
{ |
|
||||
if (!isClosed_) |
|
||||
{ |
{ |
||||
isClosed_ = true; |
this.Finish(); |
||||
|
this.deflater.Dispose(); |
||||
try |
this.memoryOwner.Dispose(); |
||||
{ |
|
||||
Finish(); |
|
||||
} |
|
||||
finally |
|
||||
{ |
|
||||
if (IsStreamOwner) |
|
||||
{ |
|
||||
baseOutputStream_.Dispose(); |
|
||||
} |
|
||||
} |
|
||||
} |
} |
||||
} |
|
||||
|
|
||||
///// <summary>
|
|
||||
///// Get the Auth code for AES encrypted entries
|
|
||||
///// </summary>
|
|
||||
//protected void GetAuthCodeIfAES()
|
|
||||
//{
|
|
||||
// if (cryptoTransform_ is ZipAESTransform)
|
|
||||
// {
|
|
||||
// AESAuthCode = ((ZipAESTransform)cryptoTransform_).GetAuthCode();
|
|
||||
// }
|
|
||||
//}
|
|
||||
|
|
||||
/// <summary>
|
this.deflater = null; |
||||
/// Writes a single byte to the compressed output stream.
|
this.memoryOwner = null; |
||||
/// </summary>
|
this.isDisposed = true; |
||||
/// <param name="value">
|
base.Dispose(disposing); |
||||
/// The byte value.
|
|
||||
/// </param>
|
|
||||
public override void WriteByte(byte value) |
|
||||
{ |
|
||||
byte[] b = new byte[1]; |
|
||||
b[0] = value; |
|
||||
Write(b, 0, 1); |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Writes bytes from an array to the compressed stream.
|
|
||||
/// </summary>
|
|
||||
/// <param name="buffer">
|
|
||||
/// The byte array
|
|
||||
/// </param>
|
|
||||
/// <param name="offset">
|
|
||||
/// The offset into the byte array where to start.
|
|
||||
/// </param>
|
|
||||
/// <param name="count">
|
|
||||
/// The number of bytes to write.
|
|
||||
/// </param>
|
|
||||
public override void Write(byte[] buffer, int offset, int count) |
|
||||
{ |
|
||||
deflater_.SetInput(buffer, offset, count); |
|
||||
Deflate(); |
|
||||
} |
} |
||||
|
|
||||
#endregion Stream Overrides
|
|
||||
|
|
||||
#region Instance Fields
|
|
||||
|
|
||||
/// <summary>
|
|
||||
/// This buffer is used temporarily to retrieve the bytes from the
|
|
||||
/// deflater and write them to the underlying output stream.
|
|
||||
/// </summary>
|
|
||||
private byte[] buffer_; |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// The deflater which is used to deflate the stream.
|
|
||||
/// </summary>
|
|
||||
protected Deflater deflater_; |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Base stream the deflater depends on.
|
|
||||
/// </summary>
|
|
||||
protected Stream baseOutputStream_; |
|
||||
|
|
||||
private bool isClosed_; |
|
||||
|
|
||||
#endregion Instance Fields
|
|
||||
|
|
||||
#region Static Fields
|
|
||||
|
|
||||
// Static to help ensure that multiple files within a zip will get different random salt
|
|
||||
//private static RandomNumberGenerator _aesRnd = RandomNumberGenerator.Create();
|
|
||||
|
|
||||
#endregion Static Fields
|
|
||||
} |
} |
||||
} |
} |
||||
|
|||||
@ -1,2 +1,5 @@ |
|||||
Adler32.cs and Crc32.cs have been copied from |
Deflatestream implementation adapted from |
||||
https://github.com/ygrenier/SharpZipLib.Portable |
|
||||
|
https://github.com/icsharpcode/SharpZipLib |
||||
|
|
||||
|
LIcensed under MIT |
||||
|
|||||
Loading…
Reference in new issue