using System;
using System.IO;
using System.Collections;
using ICSharpCode.SharpZipLib.Checksum;
using ICSharpCode.SharpZipLib.Zip.Compression;
using ICSharpCode.SharpZipLib.Zip.Compression.Streams;
using System.Collections.Generic;
namespace ICSharpCode.SharpZipLib.Zip
{
	/// 
	/// This is a DeflaterOutputStream that writes the files into a zip
	/// archive one after another.  It has a special method to start a new
	/// zip entry.  The zip entries contains information about the file name
	/// size, compressed size, CRC, etc.
	///
	/// It includes support for Stored and Deflated entries.
	/// This class is not thread safe.
	/// 
	/// 
Author of the original java version : Jochen Hoenicke
	/// 
	///  This sample shows how to create a zip file
	/// 
	/// using System;
	/// using System.IO;
	///
	/// using ICSharpCode.SharpZipLib.Core;
	/// using ICSharpCode.SharpZipLib.Zip;
	///
	/// class MainClass
	/// {
	/// 	public static void Main(string[] args)
	/// 	{
	/// 		string[] filenames = Directory.GetFiles(args[0]);
	/// 		byte[] buffer = new byte[4096];
	///
	/// 		using ( ZipOutputStream s = new ZipOutputStream(File.Create(args[1])) ) {
	///
	/// 			s.SetLevel(9); // 0 - store only to 9 - means best compression
	///
	/// 			foreach (string file in filenames) {
	/// 				ZipEntry entry = new ZipEntry(file);
	/// 				s.PutNextEntry(entry);
	///
	/// 				using (FileStream fs = File.OpenRead(file)) {
	///						StreamUtils.Copy(fs, s, buffer);
	/// 				}
	/// 			}
	/// 		}
	/// 	}
	/// }
	/// 
	/// 
	public class ZipOutputStream : DeflaterOutputStream
	{
		#region Constructors
		/// 
		/// Creates a new Zip output stream, writing a zip archive.
		/// 
		/// 
		/// The output stream to which the archive contents are written.
		/// 
		public ZipOutputStream(Stream baseOutputStream)
			: base(baseOutputStream, new Deflater(Deflater.DEFAULT_COMPRESSION, true))
		{
		}
		/// 
		/// Creates a new Zip output stream, writing a zip archive.
		/// 
		/// The output stream to which the archive contents are written.
		/// Size of the buffer to use.
		public ZipOutputStream(Stream baseOutputStream, int bufferSize)
			: base(baseOutputStream, new Deflater(Deflater.DEFAULT_COMPRESSION, true), bufferSize)
		{
		}
		#endregion
		/// 
		/// Gets a flag value of true if the central header has been added for this archive; false if it has not been added.
		/// 
		/// No further entries can be added once this has been done.
		public bool IsFinished {
			get {
				return entries == null;
			}
		}
		/// 
		/// Set the zip file comment.
		/// 
		/// 
		/// The comment text for the entire archive.
		/// 
		/// 
		/// The converted comment is longer than 0xffff bytes.
		/// 
		public void SetComment(string comment)
		{
			// TODO: Its not yet clear how to handle unicode comments here.
			byte[] commentBytes = ZipConstants.ConvertToArray(comment);
			if (commentBytes.Length > 0xffff) {
				throw new ArgumentOutOfRangeException("nameof(comment)");
			}
			zipComment = commentBytes;
		}
		/// 
		/// Sets the compression level.  The new level will be activated
		/// immediately.
		/// 
		/// The new compression level (1 to 9).
		/// 
		/// Level specified is not supported.
		/// 
		/// 
		public void SetLevel(int level)
		{
			deflater_.SetLevel(level);
			defaultCompressionLevel = level;
		}
		/// 
		/// Get the current deflater compression level
		/// 
		/// The current compression level
		public int GetLevel()
		{
			return deflater_.GetLevel();
		}
		/// 
		/// Get / set a value indicating how Zip64 Extension usage is determined when adding entries.
		/// 
		/// Older archivers may not understand Zip64 extensions.
		/// If backwards compatability is an issue be careful when adding entries to an archive.
		/// Setting this property to off is workable but less desirable as in those circumstances adding a file
		/// larger then 4GB will fail.
		public UseZip64 UseZip64 {
			get { return useZip64_; }
			set { useZip64_ = value; }
		}
		/// 
		/// Write an unsigned short in little endian byte order.
		/// 
		private void WriteLeShort(int value)
		{
			unchecked {
				baseOutputStream_.WriteByte((byte)(value & 0xff));
				baseOutputStream_.WriteByte((byte)((value >> 8) & 0xff));
			}
		}
		/// 
		/// Write an int in little endian byte order.
		/// 
		private void WriteLeInt(int value)
		{
			unchecked {
				WriteLeShort(value);
				WriteLeShort(value >> 16);
			}
		}
		/// 
		/// Write an int in little endian byte order.
		/// 
		private void WriteLeLong(long value)
		{
			unchecked {
				WriteLeInt((int)value);
				WriteLeInt((int)(value >> 32));
			}
		}
		/// 
		/// Starts a new Zip entry. It automatically closes the previous
		/// entry if present.
		/// All entry elements bar name are optional, but must be correct if present.
		/// If the compression method is stored and the output is not patchable
		/// the compression for that entry is automatically changed to deflate level 0
		/// 
		/// 
		/// the entry.
		/// 
		/// 
		/// if entry passed is null.
		/// 
		/// 
		/// if an I/O error occured.
		/// 
		/// 
		/// if stream was finished
		/// 
		/// 
		/// Too many entries in the Zip file
		/// Entry name is too long
		/// Finish has already been called
		/// 
		public void PutNextEntry(ZipEntry entry)
		{
			if (entry == null) {
				throw new ArgumentNullException("nameof(entry)");
			}
			if (entries == null) {
				throw new InvalidOperationException("ZipOutputStream was finished");
			}
			if (curEntry != null) {
				CloseEntry();
			}
			if (entries.Count == int.MaxValue) {
				throw new ZipException("Too many entries for Zip file");
			}
			CompressionMethod method = entry.CompressionMethod;
			int compressionLevel = defaultCompressionLevel;
			// Clear flags that the library manages internally
			entry.Flags &= (int)GeneralBitFlags.UnicodeText;
			patchEntryHeader = false;
			bool headerInfoAvailable;
			// No need to compress - definitely no data.
			if (entry.Size == 0) {
				entry.CompressedSize = entry.Size;
				entry.Crc = 0;
				method = CompressionMethod.Stored;
				headerInfoAvailable = true;
			} else {
				headerInfoAvailable = (entry.Size >= 0) && entry.HasCrc && entry.CompressedSize >= 0;
				// Switch to deflation if storing isnt possible.
				if (method == CompressionMethod.Stored) {
					if (!headerInfoAvailable) {
						if (!CanPatchEntries) {
							// Can't patch entries so storing is not possible.
							method = CompressionMethod.Deflated;
							compressionLevel = 0;
						}
					} else // entry.size must be > 0
					  {
						entry.CompressedSize = entry.Size;
						headerInfoAvailable = entry.HasCrc;
					}
				}
			}
			if (headerInfoAvailable == false) {
				if (CanPatchEntries == false) {
					// Only way to record size and compressed size is to append a data descriptor
					// after compressed data.
					// Stored entries of this form have already been converted to deflating.
					entry.Flags |= 8;
				} else {
					patchEntryHeader = true;
				}
			}
			if (Password != null) {
				entry.IsCrypted = true;
				if (entry.Crc < 0) {
					// Need to append a data descriptor as the crc isnt available for use
					// with encryption, the date is used instead.  Setting the flag
					// indicates this to the decompressor.
					entry.Flags |= 8;
				}
			}
			entry.Offset = offset;
			entry.CompressionMethod = (CompressionMethod)method;
			curMethod = method;
			sizePatchPos = -1;
			if ((useZip64_ == UseZip64.On) || ((entry.Size < 0) && (useZip64_ == UseZip64.Dynamic))) {
				entry.ForceZip64();
			}
			// Write the local file header
			WriteLeInt(ZipConstants.LocalHeaderSignature);
			WriteLeShort(entry.Version);
			WriteLeShort(entry.Flags);
			WriteLeShort((byte)entry.CompressionMethodForHeader);
			WriteLeInt((int)entry.DosTime);
			// TODO: Refactor header writing.  Its done in several places.
			if (headerInfoAvailable) {
				WriteLeInt((int)entry.Crc);
				if (entry.LocalHeaderRequiresZip64) {
					WriteLeInt(-1);
					WriteLeInt(-1);
				} else {
					WriteLeInt(entry.IsCrypted ? (int)entry.CompressedSize + ZipConstants.CryptoHeaderSize : (int)entry.CompressedSize);
					WriteLeInt((int)entry.Size);
				}
			} else {
				if (patchEntryHeader) {
					crcPatchPos = baseOutputStream_.Position;
				}
				WriteLeInt(0);  // Crc
				if (patchEntryHeader) {
					sizePatchPos = baseOutputStream_.Position;
				}
				// For local header both sizes appear in Zip64 Extended Information
				if (entry.LocalHeaderRequiresZip64 || patchEntryHeader) {
					WriteLeInt(-1);
					WriteLeInt(-1);
				} else {
					WriteLeInt(0);  // Compressed size
					WriteLeInt(0);  // Uncompressed size
				}
			}
			byte[] name = ZipConstants.ConvertToArray(entry.Flags, entry.Name);
			if (name.Length > 0xFFFF) {
				throw new ZipException("Entry name too long.");
			}
			var ed = new ZipExtraData(entry.ExtraData);
			if (entry.LocalHeaderRequiresZip64) {
				ed.StartNewEntry();
				if (headerInfoAvailable) {
					ed.AddLeLong(entry.Size);
					ed.AddLeLong(entry.CompressedSize);
				} else {
					ed.AddLeLong(-1);
					ed.AddLeLong(-1);
				}
				ed.AddNewEntry(1);
				if (!ed.Find(1)) {
					throw new ZipException("Internal error cant find extra data");
				}
				if (patchEntryHeader) {
					sizePatchPos = ed.CurrentReadIndex;
				}
			} else {
				ed.Delete(1);
			}
			if (entry.AESKeySize > 0) {
				AddExtraDataAES(entry, ed);
			}
			byte[] extra = ed.GetEntryData();
			WriteLeShort(name.Length);
			WriteLeShort(extra.Length);
			if (name.Length > 0) {
				baseOutputStream_.Write(name, 0, name.Length);
			}
			if (entry.LocalHeaderRequiresZip64 && patchEntryHeader) {
				sizePatchPos += baseOutputStream_.Position;
			}
			if (extra.Length > 0) {
				baseOutputStream_.Write(extra, 0, extra.Length);
			}
			offset += ZipConstants.LocalHeaderBaseSize + name.Length + extra.Length;
			// Fix offsetOfCentraldir for AES
			if (entry.AESKeySize > 0)
				offset += entry.AESOverheadSize;
			// Activate the entry.
			curEntry = entry;
			crc.Reset();
			if (method == CompressionMethod.Deflated) {
				deflater_.Reset();
				deflater_.SetLevel(compressionLevel);
			}
			size = 0;
			if (entry.IsCrypted) {
				if (entry.AESKeySize > 0) {
					WriteAESHeader(entry);
				} else {
					if (entry.Crc < 0) {            // so testing Zip will says its ok
						WriteEncryptionHeader(entry.DosTime << 16);
					} else {
						WriteEncryptionHeader(entry.Crc);
					}
				}
			}
		}
		/// 
		/// Closes the current entry, updating header and footer information as required
		/// 
		/// 
		/// An I/O error occurs.
		/// 
		/// 
		/// No entry is active.
		/// 
		public void CloseEntry()
		{
			if (curEntry == null) {
				throw new InvalidOperationException("No open entry");
			}
			long csize = size;
			// First finish the deflater, if appropriate
			if (curMethod == CompressionMethod.Deflated) {
				if (size >= 0) {
					base.Finish();
					csize = deflater_.TotalOut;
				} else {
					deflater_.Reset();
				}
			}
			// Write the AES Authentication Code (a hash of the compressed and encrypted data)
			if (curEntry.AESKeySize > 0) {
				baseOutputStream_.Write(AESAuthCode, 0, 10);
			}
			if (curEntry.Size < 0) {
				curEntry.Size = size;
			} else if (curEntry.Size != size) {
				throw new ZipException("size was " + size + ", but I expected " + curEntry.Size);
			}
			if (curEntry.CompressedSize < 0) {
				curEntry.CompressedSize = csize;
			} else if (curEntry.CompressedSize != csize) {
				throw new ZipException("compressed size was " + csize + ", but I expected " + curEntry.CompressedSize);
			}
			if (curEntry.Crc < 0) {
				curEntry.Crc = crc.Value;
			} else if (curEntry.Crc != crc.Value) {
				throw new ZipException("crc was " + crc.Value + ", but I expected " + curEntry.Crc);
			}
			offset += csize;
			if (curEntry.IsCrypted) {
				if (curEntry.AESKeySize > 0) {
					curEntry.CompressedSize += curEntry.AESOverheadSize;
				} else {
					curEntry.CompressedSize += ZipConstants.CryptoHeaderSize;
				}
			}
			// Patch the header if possible
			if (patchEntryHeader) {
				patchEntryHeader = false;
				long curPos = baseOutputStream_.Position;
				baseOutputStream_.Seek(crcPatchPos, SeekOrigin.Begin);
				WriteLeInt((int)curEntry.Crc);
				if (curEntry.LocalHeaderRequiresZip64) {
					if (sizePatchPos == -1) {
						throw new ZipException("Entry requires zip64 but this has been turned off");
					}
					baseOutputStream_.Seek(sizePatchPos, SeekOrigin.Begin);
					WriteLeLong(curEntry.Size);
					WriteLeLong(curEntry.CompressedSize);
				} else {
					WriteLeInt((int)curEntry.CompressedSize);
					WriteLeInt((int)curEntry.Size);
				}
				baseOutputStream_.Seek(curPos, SeekOrigin.Begin);
			}
			// Add data descriptor if flagged as required
			if ((curEntry.Flags & 8) != 0) {
				WriteLeInt(ZipConstants.DataDescriptorSignature);
				WriteLeInt(unchecked((int)curEntry.Crc));
				if (curEntry.LocalHeaderRequiresZip64) {
					WriteLeLong(curEntry.CompressedSize);
					WriteLeLong(curEntry.Size);
					offset += ZipConstants.Zip64DataDescriptorSize;
				} else {
					WriteLeInt((int)curEntry.CompressedSize);
					WriteLeInt((int)curEntry.Size);
					offset += ZipConstants.DataDescriptorSize;
				}
			}
			entries.Add(curEntry);
			curEntry = null;
		}
		void WriteEncryptionHeader(long crcValue)
		{
			offset += ZipConstants.CryptoHeaderSize;
			InitializePassword(Password);
			byte[] cryptBuffer = new byte[ZipConstants.CryptoHeaderSize];
			var rnd = new Random();
			rnd.NextBytes(cryptBuffer);
			cryptBuffer[11] = (byte)(crcValue >> 24);
			EncryptBlock(cryptBuffer, 0, cryptBuffer.Length);
			baseOutputStream_.Write(cryptBuffer, 0, cryptBuffer.Length);
		}
		private static void AddExtraDataAES(ZipEntry entry, ZipExtraData extraData)
		{
			// Vendor Version: AE-1 IS 1. AE-2 is 2. With AE-2 no CRC is required and 0 is stored.
			const int VENDOR_VERSION = 2;
			// Vendor ID is the two ASCII characters "AE".
			const int VENDOR_ID = 0x4541; //not 6965;
			extraData.StartNewEntry();
			// Pack AES extra data field see http://www.winzip.com/aes_info.htm
			//extraData.AddLeShort(7);							// Data size (currently 7)
			extraData.AddLeShort(VENDOR_VERSION);               // 2 = AE-2
			extraData.AddLeShort(VENDOR_ID);                    // "AE"
			extraData.AddData(entry.AESEncryptionStrength);     //  1 = 128, 2 = 192, 3 = 256
			extraData.AddLeShort((int)entry.CompressionMethod); // The actual compression method used to compress the file
			extraData.AddNewEntry(0x9901);
		}
		// Replaces WriteEncryptionHeader for AES
		//
		private void WriteAESHeader(ZipEntry entry)
		{
			byte[] salt;
			byte[] pwdVerifier;
			InitializeAESPassword(entry, Password, out salt, out pwdVerifier);
			// File format for AES:
			// Size (bytes)   Content
			// ------------   -------
			// Variable       Salt value
			// 2              Password verification value
			// Variable       Encrypted file data
			// 10             Authentication code
			//
			// Value in the "compressed size" fields of the local file header and the central directory entry
			// is the total size of all the items listed above. In other words, it is the total size of the
			// salt value, password verification value, encrypted data, and authentication code.
			baseOutputStream_.Write(salt, 0, salt.Length);
			baseOutputStream_.Write(pwdVerifier, 0, pwdVerifier.Length);
		}
		/// 
		/// Writes the given buffer to the current entry.
		/// 
		/// The buffer containing data to write.
		/// The offset of the first byte to write.
		/// The number of bytes to write.
		/// Archive size is invalid
		/// No entry is active.
		public override void Write(byte[] buffer, int offset, int count)
		{
			if (curEntry == null) {
				throw new InvalidOperationException("No open entry.");
			}
			if (buffer == null) {
				throw new ArgumentNullException("nameof(buffer)");
			}
			if (offset < 0) {
				throw new ArgumentOutOfRangeException("nameof(offset)", "Cannot be negative");
			}
			if (count < 0) {
				throw new ArgumentOutOfRangeException("nameof(count)", "Cannot be negative");
			}
			if ((buffer.Length - offset) < count) {
				throw new ArgumentException("Invalid offset/count combination");
			}
			crc.Update(buffer, offset, count);
			size += count;
			switch (curMethod) {
				case CompressionMethod.Deflated:
					base.Write(buffer, offset, count);
					break;
				case CompressionMethod.Stored:
					if (Password != null) {
						CopyAndEncrypt(buffer, offset, count);
					} else {
						baseOutputStream_.Write(buffer, offset, count);
					}
					break;
			}
		}
		void CopyAndEncrypt(byte[] buffer, int offset, int count)
		{
			const int CopyBufferSize = 4096;
			byte[] localBuffer = new byte[CopyBufferSize];
			while (count > 0) {
				int bufferCount = (count < CopyBufferSize) ? count : CopyBufferSize;
				Array.Copy(buffer, offset, localBuffer, 0, bufferCount);
				EncryptBlock(localBuffer, 0, bufferCount);
				baseOutputStream_.Write(localBuffer, 0, bufferCount);
				count -= bufferCount;
				offset += bufferCount;
			}
		}
		/// 
		/// Finishes the stream.  This will write the central directory at the
		/// end of the zip file and flush the stream.
		/// 
		/// 
		/// This is automatically called when the stream is closed.
		/// 
		/// 
		/// An I/O error occurs.
		/// 
		/// 
		/// Comment exceeds the maximum length
		/// Entry name exceeds the maximum length
		/// 
		public override void Finish()
		{
			if (entries == null) {
				return;
			}
			if (curEntry != null) {
				CloseEntry();
			}
			long numEntries = entries.Count;
			long sizeEntries = 0;
			foreach (ZipEntry entry in entries) {
				WriteLeInt(ZipConstants.CentralHeaderSignature);
				WriteLeShort(ZipConstants.VersionMadeBy);
				WriteLeShort(entry.Version);
				WriteLeShort(entry.Flags);
				WriteLeShort((short)entry.CompressionMethodForHeader);
				WriteLeInt((int)entry.DosTime);
				WriteLeInt((int)entry.Crc);
				if (entry.IsZip64Forced() ||
					(entry.CompressedSize >= uint.MaxValue)) {
					WriteLeInt(-1);
				} else {
					WriteLeInt((int)entry.CompressedSize);
				}
				if (entry.IsZip64Forced() ||
					(entry.Size >= uint.MaxValue)) {
					WriteLeInt(-1);
				} else {
					WriteLeInt((int)entry.Size);
				}
				byte[] name = ZipConstants.ConvertToArray(entry.Flags, entry.Name);
				if (name.Length > 0xffff) {
					throw new ZipException("Name too long.");
				}
				var ed = new ZipExtraData(entry.ExtraData);
				if (entry.CentralHeaderRequiresZip64) {
					ed.StartNewEntry();
					if (entry.IsZip64Forced() ||
						(entry.Size >= 0xffffffff)) {
						ed.AddLeLong(entry.Size);
					}
					if (entry.IsZip64Forced() ||
						(entry.CompressedSize >= 0xffffffff)) {
						ed.AddLeLong(entry.CompressedSize);
					}
					if (entry.Offset >= 0xffffffff) {
						ed.AddLeLong(entry.Offset);
					}
					ed.AddNewEntry(1);
				} else {
					ed.Delete(1);
				}
				if (entry.AESKeySize > 0) {
					AddExtraDataAES(entry, ed);
				}
				byte[] extra = ed.GetEntryData();
				byte[] entryComment =
					(entry.Comment != null) ?
					ZipConstants.ConvertToArray(entry.Flags, entry.Comment) :
					new byte[0];
				if (entryComment.Length > 0xffff) {
					throw new ZipException("Comment too long.");
				}
				WriteLeShort(name.Length);
				WriteLeShort(extra.Length);
				WriteLeShort(entryComment.Length);
				WriteLeShort(0);    // disk number
				WriteLeShort(0);    // internal file attributes
									// external file attributes
				if (entry.ExternalFileAttributes != -1) {
					WriteLeInt(entry.ExternalFileAttributes);
				} else {
					if (entry.IsDirectory) {                         // mark entry as directory (from nikolam.AT.perfectinfo.com)
						WriteLeInt(16);
					} else {
						WriteLeInt(0);
					}
				}
				if (entry.Offset >= uint.MaxValue) {
					WriteLeInt(-1);
				} else {
					WriteLeInt((int)entry.Offset);
				}
				if (name.Length > 0) {
					baseOutputStream_.Write(name, 0, name.Length);
				}
				if (extra.Length > 0) {
					baseOutputStream_.Write(extra, 0, extra.Length);
				}
				if (entryComment.Length > 0) {
					baseOutputStream_.Write(entryComment, 0, entryComment.Length);
				}
				sizeEntries += ZipConstants.CentralHeaderBaseSize + name.Length + extra.Length + entryComment.Length;
			}
			using (ZipHelperStream zhs = new ZipHelperStream(baseOutputStream_)) {
				zhs.WriteEndOfCentralDirectory(numEntries, sizeEntries, offset, zipComment);
			}
			entries = null;
		}
		#region Instance Fields
		/// 
		/// The entries for the archive.
		/// 
		List entries = new List();
		/// 
		/// Used to track the crc of data added to entries.
		/// 
		Crc32 crc = new Crc32();
		/// 
		/// The current entry being added.
		/// 
		ZipEntry curEntry;
		int defaultCompressionLevel = Deflater.DEFAULT_COMPRESSION;
		CompressionMethod curMethod = CompressionMethod.Deflated;
		/// 
		/// Used to track the size of data for an entry during writing.
		/// 
		long size;
		/// 
		/// Offset to be recorded for each entry in the central header.
		/// 
		long offset;
		/// 
		/// Comment for the entire archive recorded in central header.
		/// 
		byte[] zipComment = new byte[0];
		/// 
		/// Flag indicating that header patching is required for the current entry.
		/// 
		bool patchEntryHeader;
		/// 
		/// Position to patch crc
		/// 
		long crcPatchPos = -1;
		/// 
		/// Position to patch size.
		/// 
		long sizePatchPos = -1;
		// Default is dynamic which is not backwards compatible and can cause problems
		// with XP's built in compression which cant read Zip64 archives.
		// However it does avoid the situation were a large file is added and cannot be completed correctly.
		// NOTE: Setting the size for entries before they are added is the best solution!
		UseZip64 useZip64_ = UseZip64.Dynamic;
		#endregion
	}
}