mirror of
https://github.com/Xevion/easy7zip.git
synced 2025-12-10 20:07:05 -06:00
4.20
This commit is contained in:
Igor Pavlov
committed by
Kornel Lesiński
Kornel Lesiński
parent
8c1b5c7b7e
commit
3c510ba80b
@@ -7,88 +7,148 @@
|
||||
#include "Common/NewHandler.h"
|
||||
#include "Common/StringConvert.h"
|
||||
#include "Common/CommandLineParser.h"
|
||||
#include "Common/Exception.h"
|
||||
|
||||
#include "Windows/COM.h"
|
||||
#include "Windows/FileMapping.h"
|
||||
#include "Windows/FileDir.h"
|
||||
#include "Windows/Synchronization.h"
|
||||
#include "Windows/Error.h"
|
||||
#include "Windows/FileName.h"
|
||||
|
||||
#include "../../IStream.h"
|
||||
#include "../../IPassword.h"
|
||||
|
||||
// #include "../../Compress/Interface/CompressInterface.h"
|
||||
// #include "../../FileManager/FolderInterface.h"
|
||||
#include "../../FileManager/StringUtils.h"
|
||||
|
||||
#include "../Common/ExitCode.h"
|
||||
#include "../Common/ArchiveCommandLine.h"
|
||||
|
||||
#include "../Resource/Extract/resource.h"
|
||||
#include "../Agent/Agent.h"
|
||||
// #include "../Common/FolderArchiveInterface.h"
|
||||
#include "../Explorer/MyMessages.h"
|
||||
|
||||
#include "Test.h"
|
||||
#include "Extract.h"
|
||||
#include "Compress.h"
|
||||
#include "ExtractGUI.h"
|
||||
#include "UpdateGUI.h"
|
||||
|
||||
using namespace NWindows;
|
||||
using namespace NCommandLineParser;
|
||||
|
||||
HINSTANCE g_hInstance;
|
||||
|
||||
static const int kNumSwitches = 5;
|
||||
static const wchar_t *kExceptionErrorMessage = L"Error:";
|
||||
static const wchar_t *kUserBreak = L"Break signaled";
|
||||
|
||||
namespace NKey {
|
||||
enum Enum
|
||||
static const wchar_t *kMemoryExceptionMessage = L"ERROR: Can't allocate required memory!";
|
||||
static const wchar_t *kUnknownExceptionMessage = L"Unknown Error";
|
||||
static const wchar_t *kInternalExceptionMessage = L"Internal Error #";
|
||||
|
||||
static const wchar_t *kIncorrectCommandMessage = L"Incorrect command";
|
||||
|
||||
static void ErrorMessage(const wchar_t *message)
|
||||
{
|
||||
// kHelp1 = 0,
|
||||
// kHelp2,
|
||||
// kDisablePercents,
|
||||
kArchiveType,
|
||||
// kYes,
|
||||
// kPassword,
|
||||
// kProperty,
|
||||
kOutputDir,
|
||||
// kWorkingDir,
|
||||
kInclude,
|
||||
// kExclude,
|
||||
// kUpdate,
|
||||
// kRecursed,
|
||||
// kSfx,
|
||||
// kOverwrite,
|
||||
kEmail,
|
||||
kShowDialog
|
||||
// kMap
|
||||
};
|
||||
|
||||
MessageBoxW(0, message, L"7-Zip GUI", MB_ICONERROR);
|
||||
}
|
||||
|
||||
static const int kSomeCludePostStringMinSize = 2; // at least <@|!><N>ame must be
|
||||
static const int kSomeCludeAfterRecursedPostStringMinSize = 2; // at least <@|!><N>ame must be
|
||||
static const wchar_t *kRecursedPostCharSet = L"0-";
|
||||
static const wchar_t *kOverwritePostCharSet = L"asut";
|
||||
int Main2()
|
||||
{
|
||||
/*
|
||||
TCHAR t[512];
|
||||
GetCurrentDirectory(512, t);
|
||||
ErrorMessage(t);
|
||||
return 0;
|
||||
*/
|
||||
|
||||
static const CSwitchForm kSwitchForms[kNumSwitches] =
|
||||
UStringVector commandStrings;
|
||||
NCommandLineParser::SplitCommandLine(GetCommandLineW(), commandStrings);
|
||||
if(commandStrings.Size() <= 1)
|
||||
{
|
||||
// { L"?", NSwitchType::kSimple, false },
|
||||
// { L"H", NSwitchType::kSimple, false },
|
||||
// { L"BD", NSwitchType::kSimple, false },
|
||||
{ L"T", NSwitchType::kUnLimitedPostString, false, 1 },
|
||||
// { L"Y", NSwitchType::kSimple, false },
|
||||
// { L"P", NSwitchType::kUnLimitedPostString, false, 0 },
|
||||
// { L"M", NSwitchType::kUnLimitedPostString, true, 1 },
|
||||
{ L"O", NSwitchType::kUnLimitedPostString, false, 1 },
|
||||
// { L"W", NSwitchType::kUnLimitedPostString, false, 0 },
|
||||
{ L"I", NSwitchType::kUnLimitedPostString, true, kSomeCludePostStringMinSize},
|
||||
// { L"X", NSwitchType::kUnLimitedPostString, true, kSomeCludePostStringMinSize},
|
||||
// { L"U", NSwitchType::kUnLimitedPostString, true, 1},
|
||||
// { L"R", NSwitchType::kPostChar, false, 0, 0, kRecursedPostCharSet },
|
||||
// { L"SFX", NSwitchType::kUnLimitedPostString, false, 0 },
|
||||
// { L"AO", NSwitchType::kPostChar, false, 1, 1, kOverwritePostCharSet},
|
||||
{ L"SEML", NSwitchType::kSimple, false },
|
||||
{ L"AD", NSwitchType::kSimple, false }
|
||||
// { L"MAP=", NSwitchType::kUnLimitedPostString, false, 1 }
|
||||
};
|
||||
MessageBoxW(0, L"Specify command", L"7-Zip", 0);
|
||||
return 0;
|
||||
}
|
||||
commandStrings.Delete(0);
|
||||
|
||||
CArchiveCommandLineOptions options;
|
||||
CArchiveCommandLineParser parser;
|
||||
|
||||
parser.Parse1(commandStrings, options);
|
||||
parser.Parse2(options);
|
||||
|
||||
bool isExtractGroupCommand = options.Command.IsFromExtractGroup();
|
||||
|
||||
if (isExtractGroupCommand)
|
||||
{
|
||||
CExtractCallbackImp *ecs = new CExtractCallbackImp;
|
||||
CMyComPtr<IFolderArchiveExtractCallback> extractCallback = ecs;
|
||||
ecs->PasswordIsDefined = options.PasswordEnabled;
|
||||
ecs->Password = options.Password;
|
||||
ecs->Init();
|
||||
|
||||
COpenCallbackGUI openCallback;
|
||||
openCallback.PasswordIsDefined = options.PasswordEnabled;
|
||||
openCallback.Password = options.Password;
|
||||
|
||||
CExtractOptions eo;
|
||||
eo.StdOutMode = options.StdOutMode;
|
||||
eo.OutputDir = options.OutputDir;
|
||||
eo.YesToAll = options.YesToAll;
|
||||
eo.OverwriteMode = options.OverwriteMode;
|
||||
eo.PathMode = options.Command.GetPathMode();
|
||||
eo.TestMode = options.Command.IsTestMode();
|
||||
|
||||
HRESULT result = ExtractGUI(
|
||||
options.ArchivePathsSorted,
|
||||
options.ArchivePathsFullSorted,
|
||||
options.WildcardCensor.Pairs.Front().Head,
|
||||
eo, options.ShowDialog, &openCallback, ecs);
|
||||
if (result == S_FALSE)
|
||||
{
|
||||
MyMessageBox(IDS_OPEN_IS_NOT_SUPORTED_ARCHIVE, 0x02000604);
|
||||
return NExitCode::kFatalError;
|
||||
}
|
||||
else if (result != S_OK)
|
||||
throw CSystemException(result);
|
||||
if (ecs->Messages.Size() > 0)
|
||||
return NExitCode::kFatalError;
|
||||
}
|
||||
else if (options.Command.IsFromUpdateGroup())
|
||||
{
|
||||
bool passwordIsDefined =
|
||||
options.PasswordEnabled && !options.Password.IsEmpty();
|
||||
|
||||
COpenCallbackGUI openCallback;
|
||||
openCallback.PasswordIsDefined = passwordIsDefined;
|
||||
openCallback.Password = options.Password;
|
||||
|
||||
CUpdateCallbackGUI callback;
|
||||
// callback.EnablePercents = options.EnablePercents;
|
||||
callback.PasswordIsDefined = passwordIsDefined;
|
||||
callback.AskPassword = options.PasswordEnabled && options.Password.IsEmpty();
|
||||
callback.Password = options.Password;
|
||||
// callback.StdOutMode = options.UpdateOptions.StdOutMode;
|
||||
callback.Init();
|
||||
|
||||
CUpdateErrorInfo errorInfo;
|
||||
|
||||
HRESULT result = UpdateGUI(
|
||||
options.WildcardCensor, options.UpdateOptions,
|
||||
options.ShowDialog,
|
||||
errorInfo, &openCallback, &callback);
|
||||
|
||||
if (result != S_OK)
|
||||
{
|
||||
if (!errorInfo.Message.IsEmpty())
|
||||
ErrorMessage(errorInfo.Message);
|
||||
throw CSystemException(result);
|
||||
}
|
||||
if (callback.FailedFiles.Size() > 0)
|
||||
return NExitCode::kWarning;
|
||||
}
|
||||
else
|
||||
{
|
||||
ErrorMessage(L"Use correct command");
|
||||
return 0;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
static bool inline IsItWindowsNT()
|
||||
{
|
||||
@@ -99,263 +159,76 @@ static bool inline IsItWindowsNT()
|
||||
return (versionInfo.dwPlatformId == VER_PLATFORM_WIN32_NT);
|
||||
}
|
||||
|
||||
static void MyMessageBoxError(const char *message)
|
||||
{
|
||||
::MessageBoxA(0, message, "Error", 0);
|
||||
}
|
||||
|
||||
static wchar_t *kIncorrectCommandMessage = L"Incorrect command";
|
||||
|
||||
static void ErrorMessage(const wchar_t *message)
|
||||
{
|
||||
MessageBoxW(0, message, L"7-Zip GUI", MB_ICONERROR);
|
||||
}
|
||||
|
||||
static bool ParseIncludeMap(const UString &switchParam, UStringVector &fileNames)
|
||||
{
|
||||
int splitPos = switchParam.Find(L':');
|
||||
if (splitPos < 0)
|
||||
{
|
||||
ErrorMessage(L"Bad switch");
|
||||
return false;
|
||||
}
|
||||
UString mappingName = switchParam.Left(splitPos);
|
||||
|
||||
UString switchParam2 = switchParam.Mid(splitPos + 1);
|
||||
splitPos = switchParam2.Find(L':');
|
||||
if (splitPos < 0)
|
||||
{
|
||||
ErrorMessage(L"Bad switch");
|
||||
return false;
|
||||
}
|
||||
|
||||
UString mappingSize = switchParam2.Left(splitPos);
|
||||
UString eventName = switchParam2.Mid(splitPos + 1);
|
||||
|
||||
wchar_t *endptr;
|
||||
UINT32 dataSize = wcstoul(mappingSize, &endptr, 10);
|
||||
|
||||
{
|
||||
CFileMapping fileMapping;
|
||||
if (!fileMapping.Open(FILE_MAP_READ, false,
|
||||
GetSystemString(mappingName)))
|
||||
{
|
||||
// ShowLastErrorMessage(0);
|
||||
ErrorMessage(L"Can not open mapping");
|
||||
return false;
|
||||
}
|
||||
LPVOID data = fileMapping.MapViewOfFile(FILE_MAP_READ, 0, dataSize);
|
||||
if (data == NULL)
|
||||
{
|
||||
ErrorMessage(L"MapViewOfFile error");
|
||||
return false;
|
||||
}
|
||||
try
|
||||
{
|
||||
const wchar_t *curData = (const wchar_t *)data;
|
||||
if (*curData != 0)
|
||||
{
|
||||
ErrorMessage(L"Incorrect mapping data");
|
||||
return false;
|
||||
}
|
||||
UINT32 numChars = dataSize /2;
|
||||
UString name;
|
||||
|
||||
for (int i = 1; i < numChars; i++)
|
||||
{
|
||||
wchar_t c = curData[i];
|
||||
if (c == L'\0')
|
||||
{
|
||||
fileNames.Add(name);
|
||||
name.Empty();
|
||||
}
|
||||
else
|
||||
name += c;
|
||||
}
|
||||
if (!name.IsEmpty())
|
||||
{
|
||||
ErrorMessage(L"data error");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
catch(...)
|
||||
{
|
||||
UnmapViewOfFile(data);
|
||||
throw;
|
||||
}
|
||||
UnmapViewOfFile(data);
|
||||
}
|
||||
|
||||
{
|
||||
NSynchronization::CEvent event;
|
||||
event.Open(EVENT_MODIFY_STATE, false, GetSystemString(eventName));
|
||||
event.Set();
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
static bool ParseIncludeSwitches(CParser &parser, UStringVector &fileNames)
|
||||
{
|
||||
if (parser[NKey::kInclude].ThereIs)
|
||||
{
|
||||
for (int i = 0; i < parser[NKey::kInclude].PostStrings.Size(); i++)
|
||||
{
|
||||
UString switchParam = parser[NKey::kInclude].PostStrings[i];
|
||||
if (switchParam.Length() < 1)
|
||||
return false;
|
||||
if (switchParam[0] == L'#')
|
||||
{
|
||||
if (!ParseIncludeMap(switchParam.Mid(1), fileNames))
|
||||
return false;
|
||||
}
|
||||
else if (switchParam[0] == L'!')
|
||||
{
|
||||
fileNames.Add(switchParam.Mid(1));
|
||||
}
|
||||
else
|
||||
{
|
||||
ErrorMessage(L"Incorrect command");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
const UStringVector &nonSwitchStrings = parser.NonSwitchStrings;
|
||||
for (int i = 2; i < nonSwitchStrings.Size(); i++)
|
||||
{
|
||||
// ErrorMessage(nonSwitchStrings[i]);
|
||||
fileNames.Add(nonSwitchStrings[i]);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
int APIENTRY WinMain(HINSTANCE hInstance,
|
||||
HINSTANCE hPrevInstance,
|
||||
LPSTR lpCmdLine,
|
||||
int nCmdShow)
|
||||
int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
|
||||
{
|
||||
g_hInstance = hInstance;
|
||||
InitCommonControls();
|
||||
|
||||
ReloadLang();
|
||||
|
||||
|
||||
#ifdef UNICODE
|
||||
if (!IsItWindowsNT())
|
||||
{
|
||||
// g_StdOut << "This program requires Windows NT/2000/XP";
|
||||
return 0;
|
||||
MyMessageBox(L"This program requires Windows NT/2000/XP/2003");
|
||||
return NExitCode::kFatalError;
|
||||
}
|
||||
#endif
|
||||
// setlocale(LC_COLLATE, ".ACP");
|
||||
int result = 0;
|
||||
try
|
||||
{
|
||||
UStringVector commandStrings;
|
||||
SplitCommandLine(GetCommandLineW(), commandStrings);
|
||||
if (commandStrings.Size() > 0)
|
||||
commandStrings.Delete(0);
|
||||
CParser parser(kNumSwitches);
|
||||
try
|
||||
{
|
||||
parser.ParseStrings(kSwitchForms, commandStrings);
|
||||
}
|
||||
catch(...)
|
||||
{
|
||||
MyMessageBox(kIncorrectCommandMessage);
|
||||
return 1;
|
||||
}
|
||||
|
||||
const UStringVector &nonSwitchStrings = parser.NonSwitchStrings;
|
||||
int numNonSwitchStrings = nonSwitchStrings.Size();
|
||||
|
||||
if(numNonSwitchStrings < 1)
|
||||
{
|
||||
MyMessageBox(kIncorrectCommandMessage);
|
||||
return 1;
|
||||
}
|
||||
|
||||
UString command = nonSwitchStrings[0];
|
||||
|
||||
if (command == L"t")
|
||||
{
|
||||
if(numNonSwitchStrings < 2)
|
||||
{
|
||||
MyMessageBox(kIncorrectCommandMessage);
|
||||
return 1;
|
||||
}
|
||||
UString archiveName = nonSwitchStrings[1];
|
||||
HRESULT result = TestArchive(0, archiveName);
|
||||
if (result == S_FALSE)
|
||||
MyMessageBox(IDS_OPEN_IS_NOT_SUPORTED_ARCHIVE, 0x02000604);
|
||||
else if (result != S_OK)
|
||||
ShowErrorMessage(0, result);
|
||||
}
|
||||
else if (command == L"x")
|
||||
{
|
||||
if(numNonSwitchStrings < 2)
|
||||
{
|
||||
MyMessageBox(kIncorrectCommandMessage);
|
||||
return 1;
|
||||
}
|
||||
UString archiveName = nonSwitchStrings[1];
|
||||
|
||||
UString outputDir;
|
||||
bool outputDirDefined = parser[NKey::kOutputDir].ThereIs;
|
||||
if(outputDirDefined)
|
||||
outputDir = parser[NKey::kOutputDir].PostStrings[0];
|
||||
else
|
||||
NFile::NDirectory::MyGetCurrentDirectory(outputDir);
|
||||
NFile::NName::NormalizeDirPathPrefix(outputDir);
|
||||
HRESULT result = ExtractArchive(0, archiveName,
|
||||
false, parser[NKey::kShowDialog].ThereIs, outputDir);
|
||||
if (result == S_FALSE)
|
||||
MyMessageBox(IDS_OPEN_IS_NOT_SUPORTED_ARCHIVE, 0x02000604);
|
||||
else if (result != S_OK)
|
||||
ShowErrorMessage(0, result);
|
||||
}
|
||||
else if (command == L"a")
|
||||
{
|
||||
if(numNonSwitchStrings < 2)
|
||||
{
|
||||
MyMessageBox(kIncorrectCommandMessage);
|
||||
return 1;
|
||||
}
|
||||
const UString &archiveName = nonSwitchStrings[1];
|
||||
|
||||
UString mapString, emailString;
|
||||
bool emailMode = parser[NKey::kEmail].ThereIs;
|
||||
UStringVector fileNames;
|
||||
if (!ParseIncludeSwitches(parser, fileNames))
|
||||
return 1;
|
||||
if (fileNames.Size() == 0)
|
||||
{
|
||||
ErrorMessage(L"Incorrect command: No files");
|
||||
return 1;
|
||||
}
|
||||
UString archiveType = L"7z";;
|
||||
if (parser[NKey::kArchiveType].ThereIs)
|
||||
archiveType = parser[NKey::kArchiveType].PostStrings.Front();
|
||||
HRESULT result = CompressArchive(archiveName, fileNames,
|
||||
archiveType, emailMode, parser[NKey::kShowDialog].ThereIs);
|
||||
// if (result != S_OK)
|
||||
// ShowErrorMessage(result);
|
||||
}
|
||||
else
|
||||
{
|
||||
ErrorMessage(L"Use correct command");
|
||||
return 0;
|
||||
}
|
||||
return 0;
|
||||
return Main2();
|
||||
}
|
||||
catch(const CNewException &)
|
||||
{
|
||||
// MyMessageBoxError(kMemoryExceptionMessage);
|
||||
return 1;
|
||||
MyMessageBox(kMemoryExceptionMessage);
|
||||
return (NExitCode::kMemoryError);
|
||||
}
|
||||
catch(const CSystemException &systemError)
|
||||
{
|
||||
if (systemError.ErrorCode == E_OUTOFMEMORY)
|
||||
{
|
||||
MyMessageBox(kMemoryExceptionMessage);
|
||||
return (NExitCode::kMemoryError);
|
||||
}
|
||||
if (systemError.ErrorCode == E_ABORT)
|
||||
{
|
||||
// MyMessageBox(kUserBreak);
|
||||
return (NExitCode::kUserBreak);
|
||||
}
|
||||
UString message;
|
||||
NError::MyFormatMessage(systemError.ErrorCode, message);
|
||||
MyMessageBox(message);
|
||||
return (NExitCode::kFatalError);
|
||||
}
|
||||
/*
|
||||
catch(NExitCode::EEnum &exitCode)
|
||||
{
|
||||
g_StdErr << kInternalExceptionMessage << exitCode << endl;
|
||||
return (exitCode);
|
||||
}
|
||||
*/
|
||||
catch(const UString &s)
|
||||
{
|
||||
MyMessageBox(s);
|
||||
return (NExitCode::kFatalError);
|
||||
}
|
||||
catch(const char *s)
|
||||
{
|
||||
MyMessageBox(GetUnicodeString(s));
|
||||
return (NExitCode::kFatalError);
|
||||
}
|
||||
/*
|
||||
catch(int t)
|
||||
{
|
||||
g_StdErr << kInternalExceptionMessage << t << endl;
|
||||
return (NExitCode::kFatalError);
|
||||
}
|
||||
*/
|
||||
catch(...)
|
||||
{
|
||||
// g_StdOut << kUnknownExceptionMessage;
|
||||
return 2;
|
||||
MyMessageBox(kUnknownExceptionMessage);
|
||||
return (NExitCode::kFatalError);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user