* LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
* PURPOSE: Zip extraction
* COPYRIGHT: Copyright 2017-2019 Mark Jansen (mark.jansen@reactos.org)
+ * Copyright 2023 Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
*/
#include "precomp.h"
+#include <atlpath.h>
class CZipExtract :
public IZip
class CExtractSettingsPage : public CPropertyPageImpl<CExtractSettingsPage>
{
private:
+ HANDLE m_hExtractionThread;
+ bool m_bExtractionThreadCancel;
+
CZipExtract* m_pExtract;
CStringA* m_pPassword;
+ CStringW m_OldStatus;
public:
CExtractSettingsPage(CZipExtract* extract, CStringA* password)
:CPropertyPageImpl<CExtractSettingsPage>(MAKEINTRESOURCE(IDS_WIZ_TITLE))
+ ,m_hExtractionThread(NULL)
+ ,m_bExtractionThreadCancel(false)
,m_pExtract(extract)
,m_pPassword(password)
{
int OnWizardNext()
{
+ if (m_hExtractionThread != NULL)
+ {
+ /* We enter here when extraction has finished, and go to next page if it succeeded */
+ WaitForSingleObject(m_hExtractionThread, INFINITE);
+ CloseHandle(m_hExtractionThread);
+ m_hExtractionThread = NULL;
+ m_pExtract->Release();
+ if (!m_bExtractionThreadCancel)
+ {
+ return 0;
+ }
+ else
+ {
+ SetWindowLongPtr(DWLP_MSGRESULT, -1);
+ return TRUE;
+ }
+ }
+
+ /* We end up here if the user manually clicks Next: start extraction */
+ m_bExtractionThreadCancel = false;
+
+ /* Grey out every control during extraction to prevent user interaction */
::EnableWindow(GetDlgItem(IDC_BROWSE), FALSE);
::EnableWindow(GetDlgItem(IDC_DIRECTORY), FALSE);
::EnableWindow(GetDlgItem(IDC_PASSWORD), FALSE);
SetWizardButtons(0);
+ ::GetWindowTextW(GetDlgItem(IDC_STATUSTEXT), m_OldStatus.GetBuffer(MAX_PATH), MAX_PATH);
+ m_OldStatus.ReleaseBuffer();
+ CStringW strExtracting(MAKEINTRESOURCEW(IDS_EXTRACTING));
+ SetDlgItemTextW(IDC_STATUSTEXT, strExtracting);
+
if (m_pExtract->m_DirectoryChanged)
UpdateDirectory();
- if (!m_pExtract->Extract(m_hWnd, GetDlgItem(IDC_PROGRESS)))
+ m_pExtract->AddRef();
+
+ m_hExtractionThread = CreateThread(NULL, 0,
+ &CExtractSettingsPage::ExtractEntry,
+ this,
+ 0, NULL);
+ if (!m_hExtractionThread)
{
- /* Extraction failed, do not go to the next page */
+ /* Extraction thread creation failed, do not go to the next page */
+ DWORD err = GetLastError();
+ DPRINT1("ERROR, m_hExtractionThread: CreateThread failed: 0x%x\n", err);
+ m_pExtract->Release();
+
SetWindowLongPtr(DWLP_MSGRESULT, -1);
::EnableWindow(GetDlgItem(IDC_BROWSE), TRUE);
::EnableWindow(GetDlgItem(IDC_DIRECTORY), TRUE);
::EnableWindow(GetDlgItem(IDC_PASSWORD), TRUE);
SetWizardButtons(PSWIZB_NEXT);
+ }
+ return TRUE;
+ }
- return TRUE;
+ void WizardReset()
+ {
+ SetDlgItemTextW(IDC_STATUSTEXT, m_OldStatus);
+ }
+
+ static DWORD WINAPI ExtractEntry(LPVOID lpParam)
+ {
+ CExtractSettingsPage* pPage = (CExtractSettingsPage*)lpParam;
+ bool res = pPage->m_pExtract->Extract(pPage->m_hWnd, pPage->GetDlgItem(IDC_PROGRESS), &(pPage->m_bExtractionThreadCancel));
+ /* Failing and cancelling extraction both mean we stay on the same property page */
+ pPage->m_bExtractionThreadCancel = !res;
+
+ pPage->SetWizardButtons(PSWIZB_NEXT);
+ if (!res)
+ {
+ /* Extraction failed/cancelled: the page becomes interactive again */
+ ::EnableWindow(pPage->GetDlgItem(IDC_BROWSE), TRUE);
+ ::EnableWindow(pPage->GetDlgItem(IDC_DIRECTORY), TRUE);
+ ::EnableWindow(pPage->GetDlgItem(IDC_PASSWORD), TRUE);
+
+ /* Reset the progress bar's appearance */
+ CWindow Progress(pPage->GetDlgItem(IDC_PROGRESS));
+ Progress.SendMessage(PBM_SETRANGE32, 0, 1);
+ Progress.SendMessage(PBM_SETPOS, 0, 0);
+ pPage->WizardReset();
}
+ SendMessageCallback(pPage->GetParent().m_hWnd, PSM_PRESSBUTTON, PSBTN_NEXT, 0, NULL, NULL);
+
return 0;
}
+ BOOL OnQueryCancel()
+ {
+ if (m_hExtractionThread != NULL)
+ {
+ /* Extraction will check the value of m_bExtractionThreadCancel between each file in the archive */
+ m_bExtractionThreadCancel = true;
+ return TRUE;
+ }
+ return FALSE;
+ }
+
struct browse_info
{
HWND hWnd;
- LPCWSTR Directory;
+ PCWSTR Directory;
};
static INT CALLBACK s_BrowseCallbackProc(HWND hWnd, UINT uMsg, LPARAM lp, LPARAM pData)
{
SetWizardButtons(PSWIZB_FINISH);
CStringW Path = m_pExtract->m_Directory;
- PWSTR Ptr = Path.GetBuffer();
+ PWSTR Ptr = Path.GetBuffer(MAX_PATH);
RECT rc;
::GetWindowRect(GetDlgItem(IDC_DESTDIR), &rc);
HDC dc = GetDC();
};
+ /* NOTE: This callback is needed to set large icon correctly. */
+ static INT CALLBACK s_PropSheetCallbackProc(HWND hwndDlg, UINT uMsg, LPARAM lParam)
+ {
+ if (uMsg == PSCB_INITIALIZED)
+ {
+ HICON hIcon = LoadIconW(_AtlBaseModule.GetResourceInstance(), MAKEINTRESOURCEW(IDI_ZIPFLDR));
+ CWindow dlg(hwndDlg);
+ dlg.SetIcon(hIcon, TRUE);
+ }
+
+ return 0;
+ }
+
void runWizard()
{
PROPSHEETHEADERW psh = { sizeof(psh), 0 };
- psh.dwFlags = PSH_WIZARD97 | PSH_HEADER;
+ psh.dwFlags = PSH_WIZARD97 | PSH_HEADER | PSH_USEICONID | PSH_USECALLBACK;
psh.hInstance = _AtlBaseModule.GetResourceInstance();
CExtractSettingsPage extractPage(this, &m_Password);
psh.phpage = hpsp;
psh.nPages = _countof(hpsp);
+ psh.pszIcon = MAKEINTRESOURCE(IDI_ZIPFLDR);
+ psh.pszbmWatermark = MAKEINTRESOURCE(IDB_WATERMARK);
+ psh.pszbmHeader = MAKEINTRESOURCE(IDB_HEADER);
+ psh.pfnCallback = s_PropSheetCallbackProc;
PropertySheetW(&psh);
}
- bool Extract(HWND hDlg, HWND hProgress)
+ eZipExtractError ExtractSingle(
+ HWND hDlg,
+ PCWSTR FullPath,
+ bool is_dir,
+ unz_file_info64* Info,
+ CStringW Name,
+ CStringA Password,
+ bool* bOverwriteAll,
+ const bool* bCancel,
+ int* ErrorCode
+ )
+ {
+ int err;
+ BYTE Buffer[2048];
+ DWORD dwFlags = SHPPFW_DIRCREATE | (is_dir ? SHPPFW_NONE : SHPPFW_IGNOREFILENAME);
+ HRESULT hr = SHPathPrepareForWriteW(hDlg, NULL, FullPath, dwFlags);
+ if (FAILED_UNEXPECTEDLY(hr))
+ {
+ *ErrorCode = hr;
+ return eDirectoryError;
+ }
+ if (is_dir)
+ return eNoError;
+
+ if (Info->flag & MINIZIP_PASSWORD_FLAG)
+ {
+ eZipPasswordResponse Response = eAccept;
+ do
+ {
+ /* If there is a password set, try it */
+ if (!Password.IsEmpty())
+ {
+ err = unzOpenCurrentFilePassword(uf, Password);
+ if (err == UNZ_OK)
+ {
+ /* Try to read some bytes, because unzOpenCurrentFilePassword does not return failure */
+ char Buf[10];
+ err = unzReadCurrentFile(uf, Buf, sizeof(Buf));
+ unzCloseCurrentFile(uf);
+ if (err >= UNZ_OK)
+ {
+ /* 're'-open the file so that we can begin to extract */
+ err = unzOpenCurrentFilePassword(uf, Password);
+ break;
+ }
+ }
+ }
+ Response = _CZipAskPassword(hDlg, Name, Password);
+ } while (Response == eAccept);
+
+ if (Response == eSkip)
+ {
+ return eNoError;
+ }
+ else if (Response == eAbort)
+ {
+ return eExtractAbort;
+ }
+ }
+ else
+ {
+ err = unzOpenCurrentFile(uf);
+ }
+
+ if (err != UNZ_OK)
+ {
+ DPRINT1("ERROR, unzOpenCurrentFilePassword: 0x%x\n", err);
+ *ErrorCode = err;
+ return eOpenError;
+ }
+
+ HANDLE hFile = CreateFileW(FullPath, GENERIC_WRITE, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
+ if (hFile == INVALID_HANDLE_VALUE)
+ {
+ DWORD dwErr = GetLastError();
+ if (dwErr == ERROR_FILE_EXISTS)
+ {
+ bool bOverwrite = *bOverwriteAll;
+ if (!*bOverwriteAll)
+ {
+ eZipConfirmResponse Result = _CZipAskReplace(hDlg, FullPath);
+ switch (Result)
+ {
+ case eYesToAll:
+ *bOverwriteAll = true;
+ /* fall through */
+ case eYes:
+ bOverwrite = true;
+ break;
+ case eNo:
+ break;
+ case eCancel:
+ unzCloseCurrentFile(uf);
+ return eExtractAbort;
+ }
+ }
+
+ if (bOverwrite)
+ {
+ hFile = CreateFileW(FullPath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
+ if (hFile == INVALID_HANDLE_VALUE)
+ {
+ dwErr = GetLastError();
+ }
+ }
+ else
+ {
+ unzCloseCurrentFile(uf);
+ return eNoError;
+ }
+ }
+ if (hFile == INVALID_HANDLE_VALUE)
+ {
+ unzCloseCurrentFile(uf);
+ DPRINT1("ERROR, CreateFile: 0x%x (%s)\n", dwErr, *bOverwriteAll ? "Y" : "N");
+ *ErrorCode = dwErr;
+ return eFileError;
+ }
+ }
+
+ do
+ {
+ if (*bCancel)
+ {
+ CloseHandle(hFile);
+ BOOL deleteResult = DeleteFileW(FullPath);
+ if (!deleteResult)
+ DPRINT1("ERROR, DeleteFile: 0x%x\n", GetLastError());
+ return eExtractAbort;
+ }
+
+ err = unzReadCurrentFile(uf, Buffer, sizeof(Buffer));
+
+ if (err < 0)
+ {
+ DPRINT1("ERROR, unzReadCurrentFile: 0x%x\n", err);
+ break;
+ }
+ else if (err > 0)
+ {
+ DWORD dwWritten;
+ if (!WriteFile(hFile, Buffer, err, &dwWritten, NULL))
+ {
+ DPRINT1("ERROR, WriteFile: 0x%x\n", GetLastError());
+ break;
+ }
+ if (dwWritten != (DWORD)err)
+ {
+ DPRINT1("ERROR, WriteFile: dwWritten:%d err:%d\n", dwWritten, err);
+ break;
+ }
+ }
+
+ } while (err > 0);
+
+ /* Update Filetime */
+ FILETIME LocalFileTime;
+ DosDateTimeToFileTime((WORD)(Info->dosDate >> 16), (WORD)Info->dosDate, &LocalFileTime);
+ FILETIME FileTime;
+ LocalFileTimeToFileTime(&LocalFileTime, &FileTime);
+ SetFileTime(hFile, &FileTime, &FileTime, &FileTime);
+
+ /* Done */
+ CloseHandle(hFile);
+
+ if (err)
+ {
+ unzCloseCurrentFile(uf);
+ DPRINT1("ERROR, unzReadCurrentFile2: 0x%x\n", err);
+ *ErrorCode = err;
+ return eUnpackError;
+ }
+ else
+ {
+ err = unzCloseCurrentFile(uf);
+ if (err != UNZ_OK)
+ {
+ DPRINT1("ERROR(non-fatal), unzCloseCurrentFile: 0x%x\n", err);
+ }
+ }
+ return eNoError;
+ }
+
+ bool Extract(HWND hDlg, HWND hProgress, const bool* bCancel)
{
unz_global_info64 gi;
uf = unzOpen2_64(m_Filename.GetString(), &g_FFunc);
Progress.SendMessage(PBM_SETRANGE32, 0, gi.number_entry);
Progress.SendMessage(PBM_SETPOS, 0, 0);
- BYTE Buffer[2048];
- CStringA BaseDirectory = m_Directory;
- CStringA Name;
+ CStringW BaseDirectory = m_Directory;
+ CStringW Name;
CStringA Password = m_Password;
unz_file_info64 Info;
int CurrentFile = 0;
bool bOverwriteAll = false;
while (zipEnum.next(Name, Info))
{
- bool is_dir = Name.GetLength() > 0 && Name[Name.GetLength()-1] == '/';
-
- char CombinedPath[MAX_PATH * 2] = { 0 };
- PathCombineA(CombinedPath, BaseDirectory, Name);
- CStringA FullPath = CombinedPath;
- FullPath.Replace('/', '\\'); /* SHPathPrepareForWriteA does not handle '/' */
- DWORD dwFlags = SHPPFW_DIRCREATE | (is_dir ? SHPPFW_NONE : SHPPFW_IGNOREFILENAME);
- HRESULT hr = SHPathPrepareForWriteA(hDlg, NULL, FullPath, dwFlags);
- if (FAILED_UNEXPECTEDLY(hr))
+ if (*bCancel)
{
Close();
return false;
}
- CurrentFile++;
- if (is_dir)
- continue;
- if (Info.flag & MINIZIP_PASSWORD_FLAG)
+ bool is_dir = Name.GetLength() > 0 && Name[Name.GetLength()-1] == '/';
+
+ // Build a combined path
+ CPathW FullPath(BaseDirectory);
+ FullPath += Name;
+
+ // We use SHPathPrepareForWrite for this path.
+ // SHPathPrepareForWrite will prepare the necessary directories.
+ // Windows and ReactOS SHPathPrepareForWrite do not support '/'.
+ FullPath.m_strPath.Replace(L'/', L'\\');
+
+ Retry:
+ eZipExtractError Result = ExtractSingle(hDlg, FullPath, is_dir, &Info, Name, Password, &bOverwriteAll, bCancel, &err);
+ if (Result != eDirectoryError)
+ CurrentFile++;
+ switch (Result)
{
- eZipPasswordResponse Response = eAccept;
- do
- {
- /* If there is a password set, try it */
- if (!Password.IsEmpty())
- {
- err = unzOpenCurrentFilePassword(uf, Password);
- if (err == UNZ_OK)
- {
- /* Try to read some bytes, because unzOpenCurrentFilePassword does not return failure */
- char Buf[10];
- err = unzReadCurrentFile(uf, Buf, sizeof(Buf));
- unzCloseCurrentFile(uf);
- if (err >= UNZ_OK)
- {
- /* 're'-open the file so that we can begin to extract */
- err = unzOpenCurrentFilePassword(uf, Password);
- break;
- }
- }
- }
- Response = _CZipAskPassword(hDlg, Name, Password);
- } while (Response == eAccept);
+ case eNoError:
+ break;
- if (Response == eSkip)
+ case eExtractAbort:
+ case eUnpackError:
{
- Progress.SendMessage(PBM_SETPOS, CurrentFile, 0);
- continue;
+ Close();
+ return false;
}
- else if (Response == eAbort)
+
+ case eDirectoryError:
{
+ WCHAR StrippedPath[MAX_PATH] = { 0 };
+
+ StrCpyNW(StrippedPath, FullPath, _countof(StrippedPath));
+ if (!is_dir)
+ PathRemoveFileSpecW(StrippedPath);
+ PathStripPathW(StrippedPath);
+ if (ShowExtractError(hDlg, StrippedPath, err, eDirectoryError) == IDRETRY)
+ goto Retry;
Close();
return false;
}
- }
- else
- {
- err = unzOpenCurrentFile(uf);
- }
-
- if (err != UNZ_OK)
- {
- DPRINT1("ERROR, unzOpenCurrentFilePassword: 0x%x\n", err);
- Close();
- return false;
- }
- HANDLE hFile = CreateFileA(FullPath, GENERIC_WRITE, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
- if (hFile == INVALID_HANDLE_VALUE)
- {
- DWORD dwErr = GetLastError();
- if (dwErr == ERROR_FILE_EXISTS)
+ case eFileError:
{
- bool bOverwrite = bOverwriteAll;
- if (!bOverwriteAll)
+ int Result = ShowExtractError(hDlg, FullPath, err, eFileError);
+ switch (Result)
{
- eZipConfirmResponse Result = _CZipAskReplace(hDlg, FullPath);
- switch (Result)
- {
- case eYesToAll:
- bOverwriteAll = true;
- case eYes:
- bOverwrite = true;
- break;
- case eNo:
- break;
- case eCancel:
- unzCloseCurrentFile(uf);
- Close();
- return false;
- }
+ case IDABORT:
+ Close();
+ return false;
+ case IDRETRY:
+ CurrentFile--;
+ goto Retry;
+ case IDIGNORE:
+ break;
}
+ break;
+ }
- if (bOverwrite)
- {
- hFile = CreateFileA(FullPath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
- if (hFile == INVALID_HANDLE_VALUE)
- {
- dwErr = GetLastError();
- }
- }
- else
+ case eOpenError:
+ {
+ if (err == UNZ_BADZIPFILE &&
+ Info.compression_method != 0 &&
+ Info.compression_method != Z_DEFLATED &&
+ Info.compression_method != Z_BZIP2ED)
{
- unzCloseCurrentFile(uf);
- continue;
+ if (ShowExtractError(hDlg, FullPath, Info.compression_method, eOpenError) == IDYES)
+ break;
}
- }
- if (hFile == INVALID_HANDLE_VALUE)
- {
- unzCloseCurrentFile(uf);
- DPRINT1("ERROR, CreateFileA: 0x%x (%s)\n", dwErr, bOverwriteAll ? "Y" : "N");
Close();
return false;
}
}
+ if (Result == eNoError && is_dir)
+ continue;
+ Progress.SendMessage(PBM_SETPOS, CurrentFile, 0);
+ }
- do
- {
- err = unzReadCurrentFile(uf, Buffer, sizeof(Buffer));
-
- if (err < 0)
- {
- DPRINT1("ERROR, unzReadCurrentFile: 0x%x\n", err);
- break;
- }
- else if (err > 0)
- {
- DWORD dwWritten;
- if (!WriteFile(hFile, Buffer, err, &dwWritten, NULL))
- {
- DPRINT1("ERROR, WriteFile: 0x%x\n", GetLastError());
- break;
- }
- if (dwWritten != (DWORD)err)
- {
- DPRINT1("ERROR, WriteFile: dwWritten:%d err:%d\n", dwWritten, err);
- break;
- }
- }
+ Close();
+ return true;
+ }
- } while (err > 0);
+ int ShowExtractError(HWND hDlg, PCWSTR path, int Error, eZipExtractError ErrorType)
+ {
+ CStringW strTitle(MAKEINTRESOURCEW(IDS_ERRORTITLE));
+ CStringW strErr, strText;
+ PWSTR Win32ErrorString;
- /* Update Filetime */
- FILETIME LastAccessTime;
- GetFileTime(hFile, NULL, &LastAccessTime, NULL);
- FILETIME LocalFileTime;
- DosDateTimeToFileTime((WORD)(Info.dosDate >> 16), (WORD)Info.dosDate, &LocalFileTime);
- FILETIME FileTime;
- LocalFileTimeToFileTime(&LocalFileTime, &FileTime);
- SetFileTime(hFile, &FileTime, &LastAccessTime, &FileTime);
+ if (ErrorType == eFileError || ErrorType == eOpenError)
+ strText.LoadString(IDS_CANTEXTRACTFILE);
+ else
+ strText.LoadString(GetModuleHandleA("shell32.dll"), 128); // IDS_CREATEFOLDER_DENIED
- /* Done.. */
- CloseHandle(hFile);
+ strText.FormatMessage(strText.GetString(), path);
- if (err)
- {
- unzCloseCurrentFile(uf);
- DPRINT1("ERROR, unzReadCurrentFile2: 0x%x\n", err);
- Close();
- return false;
- }
- else
+ if (ErrorType == eFileError || HRESULT_FACILITY(Error) == FACILITY_WIN32)
+ {
+ if (FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
+ NULL, ErrorType == eFileError ? Error : HRESULT_CODE(Error), 0,
+ (PWSTR)&Win32ErrorString, 0, NULL) != 0)
{
- err = unzCloseCurrentFile(uf);
- if (err != UNZ_OK)
- {
- DPRINT1("ERROR(non-fatal), unzCloseCurrentFile: 0x%x\n", err);
- }
+ strErr.SetString(Win32ErrorString);
+ LocalFree(Win32ErrorString);
}
- Progress.SendMessage(PBM_SETPOS, CurrentFile, 0);
}
-
- Close();
- return true;
+ if (ErrorType == eOpenError)
+ strErr.Format(IDS_DECOMPRESSERROR, Error);
+ else if (strErr.GetLength() == 0)
+ strErr.Format(IDS_UNKNOWNERROR, Error);
+
+ strText.Append(L"\r\n\r\n" + strErr);
+
+ UINT mbFlags = MB_ICONWARNING;
+ if (ErrorType == eDirectoryError)
+ mbFlags |= MB_RETRYCANCEL;
+ else if (ErrorType == eFileError)
+ mbFlags |= MB_ABORTRETRYIGNORE;
+ else if (ErrorType == eOpenError)
+ mbFlags |= MB_YESNO;
+
+ return MessageBoxW(hDlg, strText, strTitle, mbFlags);
}
};
projects / reactos.git / blobdiff