package com.ninemax.demo.zip.decrypt;
import java.io.File;
import java.io.IOException;
import java.util.List;
import java.util.zip.DataFormatException;
import org.apache.commons.io.FileUtils;
import de.idyl.winzipaes.AesZipFileDecrypter;
import de.idyl.winzipaes.AesZipFileEncrypter;
import de.idyl.winzipaes.impl.AESDecrypter;
import de.idyl.winzipaes.impl.AESDecrypterBC;
import de.idyl.winzipaes.impl.AESEncrypter;
import de.idyl.winzipaes.impl.AESEncrypterBC;
import de.idyl.winzipaes.impl.ExtZipEntry; 
/**
 * 壓縮指定文件或目錄為ZIP格式壓縮文件
 * 支持中文(修改源碼后)
 * 支持密碼(僅支持256bit的AES加密解密)
 * 依賴bcprov項目(bcprov-jdk16-140.jar)
 * 
 * @author zyh
 */
public class DecryptionZipUtil {	
	/**
	 * 使用指定密碼將給定文件或文件夾壓縮成指定的輸出ZIP文件
	 * @param srcFile 需要壓縮的文件或文件夾
	 * @param destPath 輸出路徑
	 * @param passwd 壓縮文件使用的密碼
	 */
	public static void zip(String srcFile,String destPath,String passwd) {
		AESEncrypter encrypter = new AESEncrypterBC();
		AesZipFileEncrypter zipFileEncrypter = null;
		try {
			zipFileEncrypter = new AesZipFileEncrypter(destPath, encrypter);
			/**
			 * 此方法是修改源碼后添加,用以支持中文文件名
			 */
			zipFileEncrypter.setEncoding("utf8");
			File sFile = new File(srcFile);
			/**
			 * AesZipFileEncrypter提供了重載的添加Entry的方法,其中:
			 * add(File f, String passwd) 
			 * 			方法是將文件直接添加進(jìn)壓縮文件
			 * 
			 * add(File f, String pathForEntry, String passwd)
			 * 			方法是按指定路徑將文件添加進(jìn)壓縮文件
			 * pathForEntry - to be used for addition of the file (path within zip file)
			 */
			doZip(sFile, zipFileEncrypter, "", passwd);
		} catch (IOException e) {
			e.printStackTrace();
		} finally {
			try {
				zipFileEncrypter.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
	
	/**
	 * 具體壓縮方法,將給定文件添加進(jìn)壓縮文件中,并處理壓縮文件中的路徑
	 * @param file 給定磁盤文件(是文件直接添加,是目錄遞歸調(diào)用添加)
	 * @param encrypter AesZipFileEncrypter實例,用于輸出加密ZIP文件
	 * @param pathForEntry ZIP文件中的路徑
	 * @param passwd 壓縮密碼
	 * @throws IOException
	 */
	private static void doZip(File file, AesZipFileEncrypter encrypter,
			String pathForEntry, String passwd) throws IOException {
		if (file.isFile()) {
			pathForEntry += file.getName();
			encrypter.add(file, pathForEntry, passwd);
			return;
		}
		pathForEntry += file.getName() + File.separator;
		for(File subFile : file.listFiles()) {
			doZip(subFile, encrypter, pathForEntry, passwd);
		}
	}
	
	/**
	 * 使用給定密碼解壓指定壓縮文件到指定目錄
	 * @param inFile 指定Zip文件
	 * @param outDir 解壓目錄
	 * @param passwd 解壓密碼
	 */
	public static void unzip(String inFile, String outDir, String passwd) {
		File outDirectory = new File(outDir);
		if (!outDirectory.exists()) {
			outDirectory.mkdir();
		}
		AESDecrypter decrypter = new AESDecrypterBC();
		AesZipFileDecrypter zipDecrypter = null;
		try {
			zipDecrypter = new AesZipFileDecrypter(new File(inFile), decrypter);
			AesZipFileDecrypter.charset = "utf-8";
			/**
			 * 得到ZIP文件中所有Entry,但此處好像與JDK里不同,目錄不視為Entry
			 * 需要創(chuàng)建文件夾,entry.isDirectory()方法同樣不適用,不知道是不是自己使用錯誤
			 * 處理文件夾問題處理可能不太好
			 */
			List<ExtZipEntry> entryList = zipDecrypter.getEntryList();
			for(ExtZipEntry entry : entryList) {
				String eName = entry.getName();
				String dir = eName.substring(0, eName.lastIndexOf(File.separator) + 1);
				File extractDir = new File(outDir, dir);
				if (!extractDir.exists()) {
					FileUtils.forceMkdir(extractDir);
				}
				/**
				 * 抽出文件
				 */
				File extractFile = new File(outDir + File.separator + eName);
				zipDecrypter.extractEntry(entry, extractFile, passwd);
			}
		} catch (IOException e) {
			e.printStackTrace();
		} catch (DataFormatException e) {
			e.printStackTrace();
		} finally {
			try {
				zipDecrypter.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	} 
	/**
	 * 測試
	 * @param args
	 */
	public static void main(String[] args) {
		/**
		 * 壓縮測試
		 * 可以傳文件或者目錄
		 */
//		zip("M:\\ZIP\\test\\bb\\a\\t.txt", "M:\\ZIP\\test\\temp1.zip", "zyh");
//		zip("M:\\ZIP\\test\\bb", "M:\\ZIP\\test\\temp2.zip", "zyh");		
		unzip("M:\\ZIP\\test\\temp2.zip", "M:\\ZIP\\test\\temp", "zyh");
	}
}

protected String encoding = "iso-8859-1";	
public boolean utf8Flg = false;
	public void setEncoding(String encoding) {
		this.encoding = encoding;
		utf8Flg |= isUTF8(encoding);
	}
	protected boolean isUTF8(String encoding) {
    if (encoding == null) {
      // check platform's default encoding
      encoding = System.getProperty("file.encoding");
    }
    return "UTF8".equalsIgnoreCase(encoding)
      || "UTF-8".equalsIgnoreCase(encoding);
  }

writeShort(entry.getName().getBytes(encoding).length); // file name length

public void initEncryptedEntry(boolean utf8Flag) {
	setCrc(0); // CRC-32 / for encrypted files it's 0 as AES/MAC checks integritiy 
	this.flag |= 1; // bit0 - encrypted
	if (utf8Flag) {
		this.flag |=(1 << 11);
	}
	// flag |= 8; // bit3 - use data descriptor
	this.primaryCompressionMethod = 0x63;
 
	byte[] extraBytes = new byte[11];
	extraBytes = new byte[11];
	// extra data header ID for AES encryption is 0x9901
	extraBytes[0] = 0x01;
	extraBytes[1] = (byte)0x99; 
	// data size (currently 7, but subject to possible increase in the
	// future)
	extraBytes[2] = 0x07; // data size
	extraBytes[3] = 0x00; // data size
	// Integer version number specific to the zip vendor
	extraBytes[4] = 0x02; // version number
	extraBytes[5] = 0x00; // version number
 
	// 2-character vendor ID
	extraBytes[6] = 0x41; // vendor id
	extraBytes[7] = 0x45; // vendor id 
	// AES encryption strength - 1=128, 2=192, 3=256
	extraBytes[8] = 0x03; 
	// actual compression method - 0x0000==stored (no compression) - 2 bytes
	extraBytes[9] = (byte) (getMethod() & 0xff);
	extraBytes[10] = (byte) ((getMethod() & 0xff00) >> 8);
 
	setExtra(extraBytes);
}

if (utf8Flag) {
this.flag |=(1 << 11);
}