CAB文件压缩/解压1.0.0.2

      #include "Compress.h"
#include "Utility.h"

#include <io.h>
#include <fcntl.h>
#include <share.h>

namespace
{
FNFCIFILEPLACED(FCIFilePlaced)
{
	return 0;
}

FNFCIALLOC(FCIAlloc)
{
	return malloc(cb);
}

FNFCIFREE(FCIFree)
{
	return free(memory);
}

FNFCIOPEN(FCIOpen)
{
	INT_PTR hf = -1;
	*err = _sopen_s(&hf, pszFile, oflag | _O_RAW, _SH_DENYWR, pmode);
	return hf;
}

FNFCIREAD(FCIRead)
{
	return _read(hf, memory, cb);
}

FNFCIWRITE(FCIWrite)
{
	return _write(hf, memory, cb);
}

FNFCICLOSE(FCIClose)
{
	return _close(hf);
}

FNFCISEEK(FCISeek)
{
	return _lseek(hf, dist, seektype);
}

FNFCIDELETE(FCIDelete)
{
	return remove(pszFile);
}

FNFCIGETTEMPFILE(FCIGetTempFile)
{
	CHAR szTempPath[MAX_PATH];
	CHAR szTempFile[MAX_PATH];

	if(GetTempPathA(MAX_PATH, szTempPath) == 0)
	{
		return FALSE;
	}

	if(GetTempFileNameA(szTempPath, "CABINET", 0, szTempFile) == 0)
	{
		return FALSE;
	}

	if(!DeleteFileA(szTempFile))
	{
		return FALSE;
	}

	HRESULT hr = StringCbCopyA(pszTempName, cbTempName, szTempFile);
	return SUCCEEDED(hr);
}

FNFCIGETNEXTCABINET(FCIGetNextCabinet)
{
	HRESULT hr = StringCbPrintfA(pccab->szCab, CB_MAX_CABINET_NAME, reinterpret_cast<PCSTR>(pv), pccab->iCab);
	return SUCCEEDED(hr);
}

FNFCISTATUS(FCIStatus)
{
	return 0;
}

FNFCIGETOPENINFO(FCIGetOpenInfo)
{
	WIN32_FIND_DATAA FindFileData;
	HANDLE hFind = FindFirstFileA(pszName, &FindFileData);
	if(hFind != INVALID_HANDLE_VALUE)
	{
		FindClose(hFind);

		Cabinet::Utility::FileTimeToDosTime(FindFileData.ftLastWriteTime, pdate, ptime);
		*pattribs = static_cast<USHORT>(FindFileData.dwFileAttributes);
	}

	return FCIOpen(pszName, _O_RDONLY, 0, err, pv);
}
}	//anonymous namespace

Cabinet::CCompress::CCompress() : m_hFCI(NULL)
{
	m_szCab[0] = '\0';
}

Cabinet::CCompress::~CCompress()
{
	_Destroy();
}

BOOL Cabinet::CCompress::Create(ULONG cbCab, LPCSTR lpCabPath)
{
	_Destroy();

	ERF erf;

	CCAB ccab = {};
	ccab.cb = cbCab;

	PCCH pBackslash = strrchr(lpCabPath, '\\');
	if(pBackslash == NULL)
	{
		return FALSE;
	}

	if(cbCab == 0)
	{
		StringCbCopyA(ccab.szCab, CB_MAX_CABINET_NAME, pBackslash + 1);
	}
	else
	{
		PCCH pExtension = strrchr(pBackslash + 1, '.');
		if(pExtension == NULL)
		{
			StringCbCopyA(m_szCab, CB_MAX_CABINET_NAME, pBackslash + 1);
			StringCbCatA(m_szCab, CB_MAX_CABINET_NAME, "%02d");
		}
		else
		{
			StringCbCopyNA(m_szCab, CB_MAX_CABINET_NAME, pBackslash + 1, pExtension - pBackslash - 1);
			StringCbCatA(m_szCab, CB_MAX_CABINET_NAME, "%02d");
			StringCbCatA(m_szCab, CB_MAX_CABINET_NAME, pExtension);
		}

		FCIGetNextCabinet(&ccab, 0, m_szCab);
	}

	StringCbCopyNA(ccab.szCabPath, CB_MAX_CAB_PATH, lpCabPath, pBackslash - lpCabPath + 1);

	m_hFCI = FCICreate(&erf, FCIFilePlaced, FCIAlloc, FCIFree, FCIOpen, FCIRead, FCIWrite, FCIClose, FCISeek, FCIDelete, FCIGetTempFile, &ccab, m_szCab);
	return m_hFCI != NULL;
}

BOOL Cabinet::CCompress::AddFile(LPCSTR lpSourceFile) CONST
{
	PCCH pBackslash = strrchr(lpSourceFile, '\\');
	if(pBackslash == NULL)
	{
		return FALSE;
	}

	CHAR szSourceFile[CB_MAX_CAB_PATH];
	StringCbCopyA(szSourceFile, CB_MAX_CAB_PATH, lpSourceFile);

	CHAR szFileName[CB_MAX_CABINET_NAME];
	StringCbCopyA(szFileName, CB_MAX_CABINET_NAME, pBackslash + 1);

	return FCIAddFile(m_hFCI, szSourceFile, szFileName, FALSE, FCIGetNextCabinet, FCIStatus, FCIGetOpenInfo, tcompTYPE_MSZIP);
}

BOOL Cabinet::CCompress::AddFolder(LPCSTR lpSourceFolder) CONST
{
	CHAR szSourceFolder[MAX_PATH];
	StringCbCopyA(szSourceFolder, MAX_PATH, lpSourceFolder);
	Utility::PathRemoveBackslash(szSourceFolder);

	PCCH pBackslash = strrchr(szSourceFolder, '\\');
	if(pBackslash == NULL)
	{
		return FALSE;
	}

	return _AddFolder(szSourceFolder, pBackslash - szSourceFolder + 1);
}

BOOL Cabinet::CCompress::FlushCabinet() CONST
{
	return FCIFlushCabinet(m_hFCI, FALSE, FCIGetNextCabinet, FCIStatus);
}

BOOL Cabinet::CCompress::_AddFolder(LPCSTR lpSourceFolder, SIZE_T Offset) CONST
{
	CHAR szFindFile[MAX_PATH];
	StringCbCopyA(szFindFile, MAX_PATH, lpSourceFolder);
	StringCbCatA(szFindFile, MAX_PATH, "\\*.*");

	SIZE_T cb = strlen(lpSourceFolder) + 1;

	BOOL bRet = FALSE;

	WIN32_FIND_DATAA FindFileData;
	HANDLE hFind = FindFirstFileA(szFindFile, &FindFileData);
	if(hFind != INVALID_HANDLE_VALUE)
	{
		do
		{
			if(strcmp(FindFileData.cFileName, ".") == 0 || strcmp(FindFileData.cFileName, "..") == 0)
			{
				continue;
			}

			StringCbCopyA(szFindFile + cb, MAX_PATH, FindFileData.cFileName);

			if((FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY)
			{
				bRet = _AddFolder(szFindFile, Offset);
				if(!bRet)
				{
					break;
				}
			}
			else
			{
				bRet = FCIAddFile(m_hFCI, szFindFile, szFindFile + Offset, FALSE, FCIGetNextCabinet, FCIStatus, FCIGetOpenInfo, tcompTYPE_MSZIP);
				if(!bRet)
				{
					break;
				}
			}
		}
		while(FindNextFileA(hFind, &FindFileData));

		FindClose(hFind);
	}

	return bRet;
}

VOID Cabinet::CCompress::_Destroy()
{
	if(m_hFCI != NULL)
	{
		FCIDestroy(m_hFCI);
		m_hFCI = NULL;
	}
}