mirror of
https://github.com/Xevion/easy7zip.git
synced 2025-12-12 09:10:00 -06:00
4.44 beta
This commit is contained in:
Igor Pavlov
committed by
Kornel Lesiński
Kornel Lesiński
parent
804edc5756
commit
d9666cf046
382
CPP/7zip/FileManager/ExtractCallback.cpp
Executable file
382
CPP/7zip/FileManager/ExtractCallback.cpp
Executable file
@@ -0,0 +1,382 @@
|
||||
// ExtractCallback.h
|
||||
|
||||
#include "StdAfx.h"
|
||||
|
||||
#include "ExtractCallback.h"
|
||||
|
||||
#include "Windows/FileFind.h"
|
||||
#include "Windows/FileDir.h"
|
||||
#include "Windows/Error.h"
|
||||
|
||||
#include "Resource/OverwriteDialog/OverwriteDialog.h"
|
||||
#ifndef _NO_CRYPTO
|
||||
#include "Resource/PasswordDialog/PasswordDialog.h"
|
||||
#endif
|
||||
#include "Resource/MessagesDialog/MessagesDialog.h"
|
||||
#include "../UI/Resource/Extract/resource.h"
|
||||
#include "../UI/GUI/resource.h"
|
||||
|
||||
#include "Common/Wildcard.h"
|
||||
#include "Common/StringConvert.h"
|
||||
|
||||
#include "FormatUtils.h"
|
||||
|
||||
#include "../Common/FilePathAutoRename.h"
|
||||
|
||||
using namespace NWindows;
|
||||
using namespace NFile;
|
||||
using namespace NFind;
|
||||
|
||||
CExtractCallbackImp::~CExtractCallbackImp()
|
||||
{
|
||||
if (ShowMessages && !Messages.IsEmpty())
|
||||
{
|
||||
CMessagesDialog messagesDialog;
|
||||
messagesDialog.Messages = &Messages;
|
||||
messagesDialog.Create(ParentWindow);
|
||||
}
|
||||
}
|
||||
|
||||
void CExtractCallbackImp::Init()
|
||||
{
|
||||
Messages.Clear();
|
||||
NumArchiveErrors = 0;
|
||||
}
|
||||
|
||||
void CExtractCallbackImp::AddErrorMessage(LPCWSTR message)
|
||||
{
|
||||
Messages.Add(message);
|
||||
}
|
||||
|
||||
STDMETHODIMP CExtractCallbackImp::SetTotal(UInt64 total)
|
||||
{
|
||||
ProgressDialog.ProgressSynch.SetProgress(total, 0);
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
STDMETHODIMP CExtractCallbackImp::SetCompleted(const UInt64 *completeValue)
|
||||
{
|
||||
for (;;)
|
||||
{
|
||||
if(ProgressDialog.ProgressSynch.GetStopped())
|
||||
return E_ABORT;
|
||||
if(!ProgressDialog.ProgressSynch.GetPaused())
|
||||
break;
|
||||
::Sleep(100);
|
||||
}
|
||||
if (completeValue != NULL)
|
||||
ProgressDialog.ProgressSynch.SetPos(*completeValue);
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
STDMETHODIMP CExtractCallbackImp::AskOverwrite(
|
||||
const wchar_t *existName, const FILETIME *existTime, const UInt64 *existSize,
|
||||
const wchar_t *newName, const FILETIME *newTime, const UInt64 *newSize,
|
||||
Int32 *answer)
|
||||
{
|
||||
COverwriteDialog dialog;
|
||||
|
||||
dialog.OldFileInfo.Time = *existTime;
|
||||
dialog.OldFileInfo.SizeIsDefined = (existSize != NULL);
|
||||
if (dialog.OldFileInfo.SizeIsDefined)
|
||||
dialog.OldFileInfo.Size = *existSize;
|
||||
dialog.OldFileInfo.Name = existName;
|
||||
|
||||
if (newTime == 0)
|
||||
dialog.NewFileInfo.TimeIsDefined = false;
|
||||
else
|
||||
{
|
||||
dialog.NewFileInfo.TimeIsDefined = true;
|
||||
dialog.NewFileInfo.Time = *newTime;
|
||||
}
|
||||
|
||||
dialog.NewFileInfo.SizeIsDefined = (newSize != NULL);
|
||||
if (dialog.NewFileInfo.SizeIsDefined)
|
||||
dialog.NewFileInfo.Size = *newSize;
|
||||
dialog.NewFileInfo.Name = newName;
|
||||
|
||||
/*
|
||||
NOverwriteDialog::NResult::EEnum writeAnswer =
|
||||
NOverwriteDialog::Execute(oldFileInfo, newFileInfo);
|
||||
*/
|
||||
INT_PTR writeAnswer = dialog.Create(NULL); // ParentWindow doesn't work with 7z
|
||||
|
||||
switch(writeAnswer)
|
||||
{
|
||||
case IDCANCEL:
|
||||
return E_ABORT;
|
||||
// askResult = NAskOverwriteAnswer::kCancel;
|
||||
// break;
|
||||
case IDNO:
|
||||
*answer = NOverwriteAnswer::kNo;
|
||||
break;
|
||||
case IDC_BUTTON_OVERWRITE_NO_TO_ALL:
|
||||
*answer = NOverwriteAnswer::kNoToAll;
|
||||
break;
|
||||
case IDC_BUTTON_OVERWRITE_YES_TO_ALL:
|
||||
*answer = NOverwriteAnswer::kYesToAll;
|
||||
break;
|
||||
case IDC_BUTTON_OVERWRITE_AUTO_RENAME:
|
||||
*answer = NOverwriteAnswer::kAutoRename;
|
||||
break;
|
||||
case IDYES:
|
||||
*answer = NOverwriteAnswer::kYes;
|
||||
break;
|
||||
default:
|
||||
throw 20413;
|
||||
}
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
|
||||
STDMETHODIMP CExtractCallbackImp::PrepareOperation(const wchar_t *name, Int32 /* askExtractMode */, const UInt64 * /* position */)
|
||||
{
|
||||
return SetCurrentFilePath(name);
|
||||
}
|
||||
|
||||
STDMETHODIMP CExtractCallbackImp::MessageError(const wchar_t *message)
|
||||
{
|
||||
AddErrorMessage(message);
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
STDMETHODIMP CExtractCallbackImp::ShowMessage(const wchar_t *message)
|
||||
{
|
||||
AddErrorMessage(message);
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
STDMETHODIMP CExtractCallbackImp::SetOperationResult(Int32 operationResult, bool encrypted)
|
||||
{
|
||||
switch(operationResult)
|
||||
{
|
||||
case NArchive::NExtract::NOperationResult::kOK:
|
||||
break;
|
||||
default:
|
||||
{
|
||||
UINT messageID;
|
||||
UInt32 langID;
|
||||
switch(operationResult)
|
||||
{
|
||||
case NArchive::NExtract::NOperationResult::kUnSupportedMethod:
|
||||
messageID = IDS_MESSAGES_DIALOG_EXTRACT_MESSAGE_UNSUPPORTED_METHOD;
|
||||
langID = 0x02000A91;
|
||||
break;
|
||||
case NArchive::NExtract::NOperationResult::kDataError:
|
||||
messageID = encrypted ?
|
||||
IDS_MESSAGES_DIALOG_EXTRACT_MESSAGE_DATA_ERROR_ENCRYPTED:
|
||||
IDS_MESSAGES_DIALOG_EXTRACT_MESSAGE_DATA_ERROR;
|
||||
langID = encrypted ? 0x02000A94 : 0x02000A92;
|
||||
break;
|
||||
case NArchive::NExtract::NOperationResult::kCRCError:
|
||||
messageID = encrypted ?
|
||||
IDS_MESSAGES_DIALOG_EXTRACT_MESSAGE_CRC_ENCRYPTED:
|
||||
IDS_MESSAGES_DIALOG_EXTRACT_MESSAGE_CRC;
|
||||
langID = encrypted ? 0x02000A95 : 0x02000A93;
|
||||
break;
|
||||
default:
|
||||
return E_FAIL;
|
||||
}
|
||||
if (_needWriteArchivePath)
|
||||
{
|
||||
AddErrorMessage(_currentArchivePath);
|
||||
_needWriteArchivePath = false;
|
||||
}
|
||||
AddErrorMessage(
|
||||
MyFormatNew(messageID,
|
||||
#ifdef LANG
|
||||
langID,
|
||||
#endif
|
||||
_currentFilePath));
|
||||
}
|
||||
}
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
////////////////////////////////////////
|
||||
// IExtractCallbackUI
|
||||
|
||||
HRESULT CExtractCallbackImp::BeforeOpen(const wchar_t *name)
|
||||
{
|
||||
#ifndef _SFX
|
||||
ProgressDialog.ProgressSynch.SetTitleFileName(name);
|
||||
#endif
|
||||
_currentArchivePath = name;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT CExtractCallbackImp::SetCurrentFilePath(const wchar_t *path)
|
||||
{
|
||||
_currentFilePath = path;
|
||||
#ifndef _SFX
|
||||
ProgressDialog.ProgressSynch.SetCurrentFileName(path);
|
||||
#endif
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT CExtractCallbackImp::OpenResult(const wchar_t *name, HRESULT result, bool encrypted)
|
||||
{
|
||||
if (result != S_OK)
|
||||
{
|
||||
MessageError(MyFormatNew(encrypted ? IDS_CANT_OPEN_ENCRYPTED_ARCHIVE : IDS_CANT_OPEN_ARCHIVE,
|
||||
#ifdef LANG
|
||||
(encrypted ? 0x0200060A : 0x02000609),
|
||||
#endif
|
||||
name));
|
||||
NumArchiveErrors++;
|
||||
}
|
||||
_currentArchivePath = name;
|
||||
_needWriteArchivePath = true;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT CExtractCallbackImp::ThereAreNoFiles()
|
||||
{
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT CExtractCallbackImp::ExtractResult(HRESULT result)
|
||||
{
|
||||
if (result == S_OK)
|
||||
return result;
|
||||
NumArchiveErrors++;
|
||||
if (result == E_ABORT || result == ERROR_DISK_FULL)
|
||||
return result;
|
||||
MessageError(_currentFilePath);
|
||||
MessageError(NError::MyFormatMessageW(result));
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT CExtractCallbackImp::SetPassword(const UString &password)
|
||||
{
|
||||
PasswordIsDefined = true;
|
||||
Password = password;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
STDMETHODIMP CExtractCallbackImp::CryptoGetTextPassword(BSTR *password)
|
||||
{
|
||||
if (!PasswordIsDefined)
|
||||
{
|
||||
CPasswordDialog dialog;
|
||||
|
||||
if (dialog.Create(ParentWindow) == IDCANCEL)
|
||||
return E_ABORT;
|
||||
|
||||
Password = dialog.Password;
|
||||
PasswordIsDefined = true;
|
||||
}
|
||||
CMyComBSTR tempName(Password);
|
||||
*password = tempName.Detach();
|
||||
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
|
||||
// IExtractCallBack3
|
||||
STDMETHODIMP CExtractCallbackImp::AskWrite(
|
||||
const wchar_t *srcPath, Int32 srcIsFolder,
|
||||
const FILETIME *srcTime, const UInt64 *srcSize,
|
||||
const wchar_t *destPath,
|
||||
BSTR *destPathResult,
|
||||
Int32 *writeAnswer)
|
||||
{
|
||||
UString destPathResultTemp = destPath;
|
||||
/*
|
||||
{
|
||||
CMyComBSTR destPathResultBSTR = destPath;
|
||||
*destPathResult = destPathResultBSTR.Detach();
|
||||
}
|
||||
*/
|
||||
*destPathResult = 0;
|
||||
*writeAnswer = BoolToInt(false);
|
||||
|
||||
UString destPathSpec = destPath;
|
||||
UString destPathSys = destPathSpec;
|
||||
bool srcIsFolderSpec = IntToBool(srcIsFolder);
|
||||
CFileInfoW destFileInfo;
|
||||
if (FindFile(destPathSys, destFileInfo))
|
||||
{
|
||||
if (srcIsFolderSpec)
|
||||
{
|
||||
if (!destFileInfo.IsDirectory())
|
||||
{
|
||||
UString message = UString(L"can not replace file \'")
|
||||
+ destPathSpec +
|
||||
UString(L"\' with folder with same name");
|
||||
RINOK(MessageError(message));
|
||||
return E_ABORT;
|
||||
}
|
||||
*writeAnswer = BoolToInt(false);
|
||||
return S_OK;
|
||||
}
|
||||
if (destFileInfo.IsDirectory())
|
||||
{
|
||||
UString message = UString(L"can not replace folder \'")
|
||||
+ destPathSpec +
|
||||
UString(L"\' with file with same name");
|
||||
RINOK(MessageError(message));
|
||||
return E_FAIL;
|
||||
}
|
||||
|
||||
switch(OverwriteMode)
|
||||
{
|
||||
case NExtract::NOverwriteMode::kSkipExisting:
|
||||
return S_OK;
|
||||
case NExtract::NOverwriteMode::kAskBefore:
|
||||
{
|
||||
Int32 overwiteResult;
|
||||
RINOK(AskOverwrite(
|
||||
destPathSpec,
|
||||
&destFileInfo.LastWriteTime, &destFileInfo.Size,
|
||||
srcPath,
|
||||
srcTime, srcSize,
|
||||
&overwiteResult));
|
||||
switch(overwiteResult)
|
||||
{
|
||||
case NOverwriteAnswer::kCancel:
|
||||
return E_ABORT;
|
||||
case NOverwriteAnswer::kNo:
|
||||
return S_OK;
|
||||
case NOverwriteAnswer::kNoToAll:
|
||||
OverwriteMode = NExtract::NOverwriteMode::kSkipExisting;
|
||||
return S_OK;
|
||||
case NOverwriteAnswer::kYesToAll:
|
||||
OverwriteMode = NExtract::NOverwriteMode::kWithoutPrompt;
|
||||
break;
|
||||
case NOverwriteAnswer::kYes:
|
||||
break;
|
||||
case NOverwriteAnswer::kAutoRename:
|
||||
OverwriteMode = NExtract::NOverwriteMode::kAutoRename;
|
||||
break;
|
||||
default:
|
||||
throw 20413;
|
||||
}
|
||||
}
|
||||
}
|
||||
if (OverwriteMode == NExtract::NOverwriteMode::kAutoRename)
|
||||
{
|
||||
if (!AutoRenamePath(destPathSys))
|
||||
{
|
||||
UString message = UString(L"can not create name of file ")
|
||||
+ destPathSys;
|
||||
RINOK(MessageError(message));
|
||||
return E_ABORT;
|
||||
}
|
||||
destPathResultTemp = destPathSys;
|
||||
}
|
||||
else
|
||||
if (!NFile::NDirectory::DeleteFileAlways(destPathSys))
|
||||
{
|
||||
UString message = UString(L"can not delete output file ")
|
||||
+ destPathSys;
|
||||
RINOK(MessageError(message));
|
||||
return E_ABORT;
|
||||
}
|
||||
}
|
||||
CMyComBSTR destPathResultBSTR = destPathResultTemp;
|
||||
*destPathResult = destPathResultBSTR.Detach();
|
||||
*writeAnswer = BoolToInt(true);
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user