CAB文件压缩/解压C++源码

#include "Compress.h" #include "Utility.h" #include <io.h> #include <fcntl.h> #include <share.h> #include <strsafe.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() { if(m_hFCI != NULL) { FCIDestroy(m_hFCI); } } BOOL Cabinet::CCompress::Create(ULONG cbCab, LPCSTR lpCabPath) { 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) { 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::FlushCabinet() { return FCIFlushCabinet(m_hFCI, FALSE, FCIGetNextCabinet, FCIStatus); }