// ConsoleApplication1.cpp : 定義控制臺(tái)應(yīng)用程序的入口點(diǎn)。
//

#include "stdafx.h"

#define ZLIB_WINAPI
#include <string>
#include <iostream>
#include <vector>
#include <Shlwapi.h> 
#include <zip.h>
#include <unzip.h>
#include <zlib.h>

using namespace std;

#pragma comment(lib, "Shlwapi.lib")
#pragma comment(lib, "zlibstat.lib")

BOOL SaveToFile(wchar_t *pszFileName, BYTE *pData, DWORD dwDataSize)
{
	wchar_t  szSaveName[MAX_PATH] = { 0 };
	lstrcpy(szSaveName, pszFileName);

	HANDLE hFile = CreateFile(szSaveName, GENERIC_READ | GENERIC_WRITE,
		FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, CREATE_ALWAYS,
		FILE_ATTRIBUTE_ARCHIVE, NULL);
	if (INVALID_HANDLE_VALUE == hFile)
	{
		return FALSE;
	}

	DWORD dwRet = 0;
	WriteFile(hFile, pData, dwDataSize, &dwRet, NULL);

	CloseHandle(hFile);

	return TRUE;
}

int _tmain(int argc, _TCHAR* argv[])
{
	char szBuffer[1024] = { 0 };

	strcpy_s(szBuffer,sizeof("test__123123"), "test__123123");

	SaveToFile(L"F:\\vs2013_code\\ConsoleApplication2\\Debug\\1.txt", (BYTE *)szBuffer, sizeof(szBuffer));

	system("pause");
	return 0;
}

int compress(Bytef* dest, uLongf* destLen, const Bytef* source, uLong sourceLen);

// 單個(gè)文件限制大小為 100M 
#define MAX_SRC_FILE_SIZE (100*1024*1024)

/**
 * @brief 壓縮指定文件的數(shù)據(jù)
 *
 * @param pszCompressFileName 待壓縮文件的路徑
 * @param ppCompressData 保存壓縮后數(shù)據(jù)的指針
 * @param pdwCompressDataSize 傳入時(shí)為壓縮緩沖區(qū)的大小，傳出時(shí)為實(shí)際壓縮后數(shù)據(jù)的大小
 * @return 壓縮是否成功，成功返回 TRUE，否則返回 FALSE
 */
BOOL CompressData(wchar_t*pszCompressFileName, BYTE **ppCompressData, DWORD *pdwCompressDataSize)
{
  HANDLE hFile = CreateFile(pszCompressFileName, GENERIC_READ,
    FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
    FILE_ATTRIBUTE_ARCHIVE, NULL);

  // 檢查文件句柄是否有效
  if (INVALID_HANDLE_VALUE == hFile)
  {
    return FALSE;
  }

  // 獲取文件大小
  DWORD dwFileSize = GetFileSize(hFile, NULL);

  // 檢查文件大小是否超過限制
  if (MAX_SRC_FILE_SIZE < dwFileSize)
  {
    CloseHandle(hFile);
    return FALSE;
  }

  DWORD dwDestDataSize = dwFileSize;

  // 分配源數(shù)據(jù)和目標(biāo)數(shù)據(jù)的內(nèi)存
  BYTE *pSrcData = new BYTE[dwFileSize];
  if (NULL == pSrcData)
  {
    CloseHandle(hFile);
    return FALSE;
  }

  BYTE *pDestData = new BYTE[dwDestDataSize];
  if (NULL == pDestData)
  {
    delete[] pSrcData;
    CloseHandle(hFile);
    return FALSE;
  }

  DWORD dwRet = 0;

  // 讀取源數(shù)據(jù)
  ReadFile(hFile, pSrcData, dwFileSize, &dwRet, NULL);

  // 檢查讀取是否成功
  if ((0 >= dwRet) || (dwRet != dwFileSize))
  {
    delete[] pDestData;
    delete[] pSrcData;
    CloseHandle(hFile);
    return FALSE;
  }

  int iRet = 0;

  // 壓縮數(shù)據(jù)
  do
  {
    iRet = compress(pDestData, &dwDestDataSize, pSrcData, dwFileSize);

    // 壓縮成功，退出循環(huán)
    if (0 == iRet)
    {
      break;
    }
    // 輸出緩沖區(qū)不足，增加緩沖區(qū)大小并重試
    else if (-5 == iRet)
    {
      delete[] pDestData;
      pDestData = NULL;
      dwDestDataSize = dwDestDataSize + (100 * 1024);
      pDestData = new BYTE[dwDestDataSize];

      // 分配新的目標(biāo)數(shù)據(jù)內(nèi)存
      if (NULL == pDestData)
      {
        delete[] pSrcData;
        CloseHandle(hFile);
        return FALSE;
      }
    }
    // 壓縮失敗，釋放內(nèi)存并返回失敗
    else
    {
      delete[] pDestData;
      pDestData = NULL;
      delete[] pSrcData;
      pSrcData = NULL;
      CloseHandle(hFile);
      return FALSE;
    }
  } while (TRUE);

  // 保存壓縮后數(shù)據(jù)的指針和實(shí)際大小
  *ppCompressData = pDestData;
  *pdwCompressDataSize = dwDestDataSize;

  // 釋放源數(shù)據(jù)內(nèi)存
  delete[] pSrcData;

  // 關(guān)閉文件句柄
  CloseHandle(hFile);

  // 返回壓縮成功
  return TRUE;
}

int uncompress(Bytef* dest, uLongf* destLen, const Bytef* source, uLong sourceLen);

/**
 * @brief 解壓指定文件的數(shù)據(jù)
 *
 * @param pszUncompressFileName 待解壓文件的路徑
 * @param ppUncompressData 保存解壓后數(shù)據(jù)的指針
 * @param pdwUncompressDataSize 傳入時(shí)為解壓緩沖區(qū)的大小，傳出時(shí)為實(shí)際解壓后數(shù)據(jù)的大小
 * @return 解壓是否成功，成功返回 TRUE，否則返回 FALSE
 */
BOOL UncompressData(wchar_t*pszUncompressFileName, BYTE **ppUncompressData, DWORD *pdwUncompressDataSize)
{
  HANDLE hFile = CreateFile(pszUncompressFileName, GENERIC_READ,
    FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
    FILE_ATTRIBUTE_ARCHIVE, NULL);

  // 檢查文件句柄是否有效
  if (INVALID_HANDLE_VALUE == hFile)
  {
    return FALSE;
  }

  // 獲取文件大小
  DWORD dwFileSize = GetFileSize(hFile, NULL);

  // 設(shè)置目標(biāo)數(shù)據(jù)緩沖區(qū)大小
  DWORD dwDestDataSize = MAX_SRC_FILE_SIZE;

  // 分配源數(shù)據(jù)和目標(biāo)數(shù)據(jù)的內(nèi)存
  BYTE *pSrcData = new BYTE[dwFileSize];
  if (NULL == pSrcData)
  {
    CloseHandle(hFile);
    return FALSE;
  }

  BYTE *pDestData = new BYTE[dwDestDataSize];
  if (NULL == pDestData)
  {
    delete[] pSrcData;
    CloseHandle(hFile);
    return FALSE;
  }

  DWORD dwRet = 0;

  // 讀取源數(shù)據(jù)
  ReadFile(hFile, pSrcData, dwFileSize, &dwRet, NULL);

  // 檢查讀取是否成功
  if ((0 >= dwRet) || (dwRet != dwFileSize))
  {
    delete[] pDestData;
    delete[] pSrcData;
    CloseHandle(hFile);
    return FALSE;
  }

  int iRet = 0;

  // 解壓縮數(shù)據(jù)
  do
  {
    iRet = uncompress(pDestData, &dwDestDataSize, pSrcData, dwFileSize);

    // 解壓縮成功，退出循環(huán)
    if (0 == iRet)
    {
      break;
    }
    // 輸出緩沖區(qū)不足，增加緩沖區(qū)大小并重試
    else if (-5 == iRet)
    {
      delete[] pDestData;
      pDestData = NULL;
      dwDestDataSize = dwDestDataSize + (100 * 1024);
      pDestData = new BYTE[dwDestDataSize];

      // 分配新的目標(biāo)數(shù)據(jù)內(nèi)存
      if (NULL == pDestData)
      {
        delete[] pSrcData;
        CloseHandle(hFile);
        return FALSE;
      }
    }
    // 解壓縮失敗，釋放內(nèi)存并返回失敗
    else
    {
      delete[] pDestData;
      pDestData = NULL;
      delete[] pSrcData;
      pSrcData = NULL;
      CloseHandle(hFile);
      return FALSE;
    }
  } while (TRUE);

  // 保存解壓后數(shù)據(jù)的指針和實(shí)際大小
  *ppUncompressData = pDestData;
  *pdwUncompressDataSize = dwDestDataSize;

  // 釋放源數(shù)據(jù)內(nèi)存
  delete[] pSrcData;

  // 關(guān)閉文件句柄
  CloseHandle(hFile);

  // 返回解壓成功
  return TRUE;
}

int main(int argc, char* argv[])
{
	BOOL bRet = FALSE;
	BYTE *pCompressData = NULL;
	DWORD dwCompressDataSize = 0;

	// 壓縮文件
	bRet = CompressData(L"F:\\vs2013_code\\ConsoleApplication2\\Debug\\1.txt", &pCompressData, &dwCompressDataSize);
	if (TRUE == bRet)
	{
		std::cout << "已壓縮" << std::endl;
	}

	// 保存壓縮數(shù)據(jù)為文件
	bRet = SaveToFile(L"F:\\vs2013_code\\ConsoleApplication2\\Debug\\1.zlib", pCompressData, dwCompressDataSize);
	if (TRUE == bRet)
	{
		std::cout << "已保存到文件" << std::endl;
	}

	// 釋放內(nèi)存
	delete[]pCompressData;
	pCompressData = NULL;

	system("pause");
	return 0;
}

int main(int argc, char* argv[])
{
  BOOL bRet = FALSE;
  BYTE *pUncompressData = NULL;
  DWORD dwUncompressDataSize = 0;

  // 解壓文件
  bRet = UncompressData("F:\\vs2013_code\\ConsoleApplication2\\Debug\\1.zlib", &pUncompressData, &dwUncompressDataSize);
  if (TRUE == bRet)
  {
    std::cout << "已解壓" << std::endl;
  }

  // 保存解壓數(shù)據(jù)為文件
  bRet = SaveToFile("F:\\vs2013_code\\ConsoleApplication2\\Debug\\1.txt", pUncompressData, dwUncompressDataSize);
  if (TRUE == bRet)
  {
    std::cout << "已保存到文件" << std::endl;
  }

  // 釋放內(nèi)存
  delete[]pUncompressData;
  pUncompressData = NULL;

  system("pause");
  return 0;
}