+
−// Main.cpp
+
−
+
−#include "StdAfx.h"
+
−
+
−#include "Common/MyInitGuid.h"
+
−
+
−#include "Common/CommandLineParser.h"
+
−#include "Common/MyException.h"
+
−#include "Common/IntToString.h"
+
−#include "Common/StdOutStream.h"
+
−#include "Common/StringConvert.h"
+
−#include "Common/StringToInt.h"
+
−
+
−#include "Windows/FileDir.h"
+
−#include "Windows/FileName.h"
+
−#include "Windows/Defs.h"
+
−#include "Windows/Error.h"
+
−#ifdef _WIN32
+
−#include "Windows/MemoryLock.h"
+
−#endif
+
−
+
−#include "../../IPassword.h"
+
−#include "../../ICoder.h"
+
−#include "../Common/UpdateAction.h"
+
−#include "../Common/Update.h"
+
−#include "../Common/Extract.h"
+
−#include "../Common/ArchiveCommandLine.h"
+
−#include "../Common/ExitCode.h"
+
−#ifdef EXTERNAL_CODECS
+
−#include "../Common/LoadCodecs.h"
+
−#endif
+
−
+
−#include "../../Compress/LZMA_Alone/LzmaBenchCon.h"
+
−
+
−#include "List.h"
+
−#include "OpenCallbackConsole.h"
+
−#include "ExtractCallbackConsole.h"
+
−#include "UpdateCallbackConsole.h"
+
−
+
−#include "../../MyVersion.h"
+
−
+
−#if defined( _WIN32) && defined( _7ZIP_LARGE_PAGES)
+
−extern "C" 
+
−{ 
+
−#include "../../../../C/Alloc.h"
+
−}
+
−#endif
+
−
+
−using namespace NWindows;
+
−using namespace NFile;
+
−using namespace NCommandLineParser;
+
−
+
−HINSTANCE g_hInstance = 0;
+
−extern CStdOutStream *g_StdStream;
+
−
+
−static const char *kCopyrightString = "\n7-Zip"
+
−#ifndef EXTERNAL_CODECS
+
−" (A)"
+
−#endif
+
−
+
−#ifdef _WIN64
+
−" [64]"
+
−#endif
+
−
+
−" " MY_VERSION_COPYRIGHT_DATE "\n";
+
−
+
−static const char *kHelpString = 
+
−    "\nUsage: 7z"
+
−#ifdef _NO_CRYPTO
+
−    "r"
+
−#else
+
−#ifndef EXTERNAL_CODECS
+
−    "a"
+
−#endif
+
−#endif
+
−    " <command> [<switches>...] <archive_name> [<file_names>...]\n"
+
−    "       [<@listfiles...>]\n"
+
−    "\n"
+
−    "<Commands>\n"
+
−    "  a: Add files to archive\n"
+
−    "  b: Benchmark\n"
+
−    "  d: Delete files from archive\n"
+
−    "  e: Extract files from archive (without using directory names)\n"
+
−    "  l: List contents of archive\n"
+
−//    "  l[a|t][f]: List contents of archive\n"
+
−//    "    a - with Additional fields\n"
+
−//    "    t - with all fields\n"
+
−//    "    f - with Full pathnames\n"
+
−    "  t: Test integrity of archive\n"
+
−    "  u: Update files to archive\n"
+
−    "  x: eXtract files with full paths\n"
+
−    "<Switches>\n"
+
−    "  -ai[r[-|0]]{@listfile|!wildcard}: Include archives\n"
+
−    "  -ax[r[-|0]]{@listfile|!wildcard}: eXclude archives\n"
+
−    "  -bd: Disable percentage indicator\n"
+
−    "  -i[r[-|0]]{@listfile|!wildcard}: Include filenames\n"
+
−    "  -m{Parameters}: set compression Method\n"
+
−    "  -o{Directory}: set Output directory\n"
+
−    "  -p{Password}: set Password\n"
+
−    "  -r[-|0]: Recurse subdirectories\n"
+
−    "  -scs{UTF-8 | WIN | DOS}: set charset for list files\n"
+
−    "  -sfx[{name}]: Create SFX archive\n"
+
−    "  -si[{name}]: read data from stdin\n"
+
−    "  -slt: show technical information for l (List) command\n"
+
−    "  -so: write data to stdout\n"
+
−    "  -ssc[-]: set sensitive case mode\n"
+
−    "  -ssw: compress shared files\n"
+
−    "  -t{Type}: Set type of archive\n"
+
−    "  -v{Size}[b|k|m|g]: Create volumes\n"
+
−    "  -u[-][p#][q#][r#][x#][y#][z#][!newArchiveName]: Update options\n"
+
−    "  -w[{path}]: assign Work directory. Empty path means a temporary directory\n"
+
−    "  -x[r[-|0]]]{@listfile|!wildcard}: eXclude filenames\n"
+
−    "  -y: assume Yes on all queries\n";
+
−
+
−// ---------------------------
+
−// exception messages
+
−
+
−static const char *kEverythingIsOk = "Everything is Ok";
+
−static const char *kUserErrorMessage  = "Incorrect command line"; // NExitCode::kUserError
+
−
+
−static const wchar_t *kDefaultSfxModule = L"7zCon.sfx";
+
−
+
−static void ShowMessageAndThrowException(CStdOutStream &s, LPCSTR message, NExitCode::EEnum code)
+
−{
+
−  s << message << endl;
+
−  throw code;
+
−}
+
−
+
−static void PrintHelpAndExit(CStdOutStream &s) // yyy
+
−{
+
−  s << kHelpString;
+
−  ShowMessageAndThrowException(s, kUserErrorMessage, NExitCode::kUserError);
+
−}
+
−
+
−#ifndef _WIN32
+
−static void GetArguments(int numArguments, const char *arguments[], UStringVector &parts)
+
−{
+
−  parts.Clear();
+
−  for(int i = 0; i < numArguments; i++)
+
−  {
+
−    UString s = MultiByteToUnicodeString(arguments[i]);
+
−    parts.Add(s);
+
−  }
+
−}
+
−#endif
+
−
+
−static void ShowCopyrightAndHelp(CStdOutStream &s, bool needHelp)
+
−{
+
−  s << kCopyrightString;
+
−  // s << "# CPUs: " << (UInt64)NWindows::NSystem::GetNumberOfProcessors() << "\n";
+
−  if (needHelp) 
+
−    s << kHelpString;
+
−}
+
−
+
−#ifdef EXTERNAL_CODECS
+
−static void PrintString(CStdOutStream &stdStream, const AString &s, int size)
+
−{
+
−  int len = s.Length();
+
−  stdStream << s;
+
−  for (int i = len; i < size; i++)
+
−    stdStream << ' ';
+
−}
+
−#endif
+
−
+
−static void PrintString(CStdOutStream &stdStream, const UString &s, int size)
+
−{
+
−  int len = s.Length();
+
−  stdStream << s;
+
−  for (int i = len; i < size; i++)
+
−    stdStream << ' ';
+
−}
+
−
+
−static inline char GetHex(Byte value)
+
−{
+
−  return (char)((value < 10) ? ('0' + value) : ('A' + (value - 10)));
+
−}
+
−
+
−int Main2(
+
−  #ifndef _WIN32  
+
−  int numArguments, const char *arguments[]
+
−  #endif
+
−)
+
−{
+
−  #ifdef _WIN32  
+
−  SetFileApisToOEM();
+
−  #endif
+
−  
+
−  UStringVector commandStrings;
+
−  #ifdef _WIN32  
+
−  NCommandLineParser::SplitCommandLine(GetCommandLineW(), commandStrings);
+
−  #else
+
−  GetArguments(numArguments, arguments, commandStrings);
+
−  #endif
+
−
+
−  if(commandStrings.Size() == 1)
+
−  {
+
−    ShowCopyrightAndHelp(g_StdOut, true);
+
−    return 0;
+
−  }
+
−  commandStrings.Delete(0);
+
−
+
−  CArchiveCommandLineOptions options;
+
−
+
−  CArchiveCommandLineParser parser;
+
−
+
−  parser.Parse1(commandStrings, options);
+
−
+
−  if(options.HelpMode)
+
−  {
+
−    ShowCopyrightAndHelp(g_StdOut, true);
+
−    return 0;
+
−  }
+
−
+
−  #if defined(_WIN32) && defined(_7ZIP_LARGE_PAGES)
+
−  if (options.LargePages)
+
−  {
+
−    SetLargePageSize();
+
−    NSecurity::EnableLockMemoryPrivilege();
+
−  }
+
−  #endif
+
−
+
−  CStdOutStream &stdStream = options.StdOutMode ? g_StdErr : g_StdOut;
+
−  g_StdStream = &stdStream;
+
−
+
−  if (options.EnableHeaders)
+
−    ShowCopyrightAndHelp(stdStream, false);
+
−
+
−  parser.Parse2(options);
+
−
+
−  CCodecs *codecs = new CCodecs;
+
−  CMyComPtr<
+
−    #ifdef EXTERNAL_CODECS
+
−    ICompressCodecsInfo
+
−    #else
+
−    IUnknown
+
−    #endif
+
−    > compressCodecsInfo = codecs;
+
−  HRESULT result = codecs->Load();
+
−  if (result != S_OK)
+
−    throw CSystemException(result);
+
−
+
−  bool isExtractGroupCommand = options.Command.IsFromExtractGroup();
+
−  if (options.Command.CommandType == NCommandType::kInfo)
+
−  {
+
−    stdStream << endl << "Formats:" << endl;
+
−    int i;
+
−    for (i = 0; i < codecs->Formats.Size(); i++)
+
−    {
+
−      const CArcInfoEx &arc = codecs->Formats[i];
+
−      #ifdef EXTERNAL_CODECS
+
−      if (arc.LibIndex >= 0)
+
−      {
+
−        char s[32];
+
−        ConvertUInt64ToString(arc.LibIndex, s);
+
−        PrintString(stdStream, s, 2);
+
−      }
+
−      else
+
−      #endif
+
−        stdStream << "  ";
+
−      stdStream << ' ';
+
−      stdStream << (char)(arc.UpdateEnabled ? 'C' : ' ');
+
−      stdStream << (char)(arc.KeepName ? 'K' : ' ');
+
−      stdStream << "  ";
+
−      PrintString(stdStream, arc.Name, 6);
+
−      stdStream << "  ";
+
−      UString s;
+
−      for (int t = 0; t < arc.Exts.Size(); t++)
+
−      {
+
−        const CArcExtInfo &ext = arc.Exts[t];
+
−        s += ext.Ext;
+
−        if (!ext.AddExt.IsEmpty())
+
−        {
+
−          s += L" (";
+
−          s += ext.AddExt;
+
−          s += L')';
+
−        }
+
−        s += L' ';
+
−      }
+
−      PrintString(stdStream, s, 14);
+
−      stdStream << "  ";
+
−      const CByteBuffer &sig = arc.StartSignature;
+
−      for (size_t j = 0; j < sig.GetCapacity(); j++)
+
−      {
+
−        Byte b = sig[j];
+
−        if (b > 0x20 && b < 0x80)
+
−        {
+
−          stdStream << (char)b;
+
−        }
+
−        else
+
−        {
+
−          stdStream << GetHex((Byte)((b >> 4) & 0xF));
+
−          stdStream << GetHex((Byte)(b & 0xF));
+
−        }
+
−        stdStream << ' ';
+
−      }
+
−      stdStream << endl;
+
−    }
+
−    stdStream << endl << "Codecs:" << endl;
+
−
+
−    #ifdef EXTERNAL_CODECS
+
−    UINT32 numMethods;
+
−    if (codecs->GetNumberOfMethods(&numMethods) == S_OK)
+
−    for (UInt32 j = 0; j < numMethods; j++)
+
−    {
+
−      int libIndex = codecs->GetCodecLibIndex(j);
+
−      if (libIndex >= 0)
+
−      {
+
−        char s[32];
+
−        ConvertUInt64ToString(libIndex, s);
+
−        PrintString(stdStream, s, 2);
+
−      }
+
−      else
+
−        stdStream << "  ";
+
−      stdStream << ' ';
+
−      stdStream << (char)(codecs->GetCodecEncoderIsAssigned(j) ? 'C' : ' ');
+
−      UInt64 id;
+
−      stdStream << "  ";
+
−      HRESULT res = codecs->GetCodecId(j, id);
+
−      if (res != S_OK)
+
−        id = (UInt64)(Int64)-1;
+
−      char s[32];
+
−      ConvertUInt64ToString(id, s, 16);
+
−      PrintString(stdStream, s, 8);
+
−      stdStream << "  ";
+
−      PrintString(stdStream, codecs->GetCodecName(j), 11);
+
−      stdStream << endl;
+
−      /*
+
−      if (res != S_OK)
+
−        throw "incorrect Codec ID";
+
−      */
+
−    }
+
−    #endif
+
−    return S_OK;
+
−  }
+
−  else if (options.Command.CommandType == NCommandType::kBenchmark)
+
−  {
+
−    if (options.Method.CompareNoCase(L"CRC") == 0)
+
−    {
+
−      HRESULT res = CrcBenchCon((FILE *)stdStream, options.NumIterations, options.NumThreads, options.DictionarySize);
+
−      if (res != S_OK)
+
−      {
+
−        if (res == S_FALSE)
+
−        {
+
−          stdStream << "\nCRC Error\n";
+
−          return NExitCode::kFatalError;
+
−        }
+
−        throw CSystemException(res);
+
−      }
+
−    }
+
−    else
+
−    {
+
−      HRESULT res = LzmaBenchCon(
+
−        #ifdef EXTERNAL_LZMA
+
−        codecs,
+
−        #endif
+
−        (FILE *)stdStream, options.NumIterations, options.NumThreads, options.DictionarySize);
+
−      if (res != S_OK)
+
−      {
+
−        if (res == S_FALSE)
+
−        {
+
−          stdStream << "\nDecoding Error\n";
+
−          return NExitCode::kFatalError;
+
−        }
+
−        throw CSystemException(res);
+
−      }
+
−    }
+
−  }
+
−  else if (isExtractGroupCommand || options.Command.CommandType == NCommandType::kList)
+
−  {
+
−    if(isExtractGroupCommand)
+
−    {
+
−      CExtractCallbackConsole *ecs = new CExtractCallbackConsole;
+
−      CMyComPtr<IFolderArchiveExtractCallback> extractCallback = ecs;
+
−
+
−      ecs->OutStream = &stdStream;
+
−      ecs->PasswordIsDefined = options.PasswordEnabled;
+
−      ecs->Password = options.Password;
+
−      ecs->Init();
+
−
+
−      COpenCallbackConsole openCallback;
+
−      openCallback.OutStream = &stdStream;
+
−      openCallback.PasswordIsDefined = options.PasswordEnabled;
+
−      openCallback.Password = options.Password;
+
−
+
−      CExtractOptions eo;
+
−      eo.StdOutMode = options.StdOutMode;
+
−      eo.PathMode = options.Command.GetPathMode();
+
−      eo.TestMode = options.Command.IsTestMode();
+
−      eo.OverwriteMode = options.OverwriteMode;
+
−      eo.OutputDir = options.OutputDir;
+
−      eo.YesToAll = options.YesToAll;
+
−      #ifdef COMPRESS_MT
+
−      eo.Properties = options.ExtractProperties;
+
−      #endif
+
−      UString errorMessage;
+
−      CDecompressStat stat;
+
−      HRESULT result = DecompressArchives(
+
−          codecs,
+
−          options.ArchivePathsSorted, 
+
−          options.ArchivePathsFullSorted,
+
−          options.WildcardCensor.Pairs.Front().Head, 
+
−          eo, &openCallback, ecs, errorMessage, stat);
+
−      if (!errorMessage.IsEmpty())
+
−      {
+
−        stdStream << endl << "Error: " << errorMessage;
+
−        if (result == S_OK)
+
−          result = E_FAIL;
+
−      }
+
−
+
−      stdStream << endl;
+
−      if (ecs->NumArchives > 1)
+
−        stdStream << "Archives: " << ecs->NumArchives << endl;
+
−      if (ecs->NumArchiveErrors != 0 || ecs->NumFileErrors != 0)
+
−      {
+
−        if (ecs->NumArchives > 1)
+
−        {
+
−          stdStream << endl;
+
−          if (ecs->NumArchiveErrors != 0)
+
−            stdStream << "Archive Errors: " << ecs->NumArchiveErrors << endl;
+
−          if (ecs->NumFileErrors != 0)
+
−            stdStream << "Sub items Errors: " << ecs->NumFileErrors << endl;
+
−        }
+
−        if (result != S_OK)
+
−          throw CSystemException(result);
+
−        return NExitCode::kFatalError;
+
−      }
+
−      if (result != S_OK)
+
−        throw CSystemException(result);
+
−      if (stat.NumFolders != 0)
+
−        stdStream << "Folders: " << stat.NumFolders << endl;
+
−      if (stat.NumFiles != 1 || stat.NumFolders != 0)
+
−          stdStream << "Files: " << stat.NumFiles << endl;
+
−      stdStream 
+
−           << "Size:       " << stat.UnpackSize << endl
+
−           << "Compressed: " << stat.PackSize << endl;
+
−    }
+
−    else
+
−    {
+
−      UInt64 numErrors = 0;
+
−      HRESULT result = ListArchives(
+
−          codecs,
+
−          options.ArchivePathsSorted, 
+
−          options.ArchivePathsFullSorted,
+
−          options.WildcardCensor.Pairs.Front().Head, 
+
−          options.EnableHeaders, 
+
−          options.TechMode,
+
−          options.PasswordEnabled, 
+
−          options.Password, numErrors);
+
−      if (numErrors > 0)
+
−      {
+
−        g_StdOut << endl << "Errors: " << numErrors;
+
−        return NExitCode::kFatalError;
+
−      }
+
−      if (result != S_OK)
+
−        throw CSystemException(result);
+
−    }
+
−  }
+
−  else if(options.Command.IsFromUpdateGroup())
+
−  {
+
−    UString workingDir;
+
−
+
−    CUpdateOptions &uo = options.UpdateOptions;
+
−    if (uo.SfxMode && uo.SfxModule.IsEmpty())
+
−      uo.SfxModule = kDefaultSfxModule;
+
−
+
−    bool passwordIsDefined = 
+
−        options.PasswordEnabled && !options.Password.IsEmpty();
+
−
+
−    COpenCallbackConsole openCallback;
+
−    openCallback.OutStream = &stdStream;
+
−    openCallback.PasswordIsDefined = passwordIsDefined;
+
−    openCallback.Password = options.Password;
+
−
+
−    CUpdateCallbackConsole callback;
+
−    callback.EnablePercents = options.EnablePercents;
+
−    callback.PasswordIsDefined = passwordIsDefined;
+
−    callback.AskPassword = options.PasswordEnabled && options.Password.IsEmpty();
+
−    callback.Password = options.Password;
+
−    callback.StdOutMode = uo.StdOutMode;
+
−    callback.Init(&stdStream);
+
−
+
−    CUpdateErrorInfo errorInfo;
+
−
+
−    if (!uo.Init(codecs, options.ArchiveName, options.ArcType))
+
−      throw "Unsupported archive type";
+
−    HRESULT result = UpdateArchive(codecs, 
+
−        options.WildcardCensor, uo, 
+
−        errorInfo, &openCallback, &callback);
+
−
+
−    int exitCode = NExitCode::kSuccess;
+
−    if (callback.CantFindFiles.Size() > 0)
+
−    {
+
−      stdStream << endl;
+
−      stdStream << "WARNINGS for files:" << endl << endl;
+
−      int numErrors = callback.CantFindFiles.Size();
+
−      for (int i = 0; i < numErrors; i++)
+
−      {
+
−        stdStream << callback.CantFindFiles[i] << " : ";
+
−        stdStream << NError::MyFormatMessageW(callback.CantFindCodes[i]) << endl;
+
−      }
+
−      stdStream << "----------------" << endl;
+
−      stdStream << "WARNING: Cannot find " << numErrors << " file";
+
−      if (numErrors > 1)
+
−        stdStream << "s";
+
−      stdStream << endl;
+
−      exitCode = NExitCode::kWarning;
+
−    }
+
−
+
−    if (result != S_OK)
+
−    {
+
−      UString message;
+
−      if (!errorInfo.Message.IsEmpty())
+
−      {
+
−        message += errorInfo.Message;
+
−        message += L"\n";
+
−      }
+
−      if (!errorInfo.FileName.IsEmpty())
+
−      {
+
−        message += errorInfo.FileName;
+
−        message += L"\n";
+
−      }
+
−      if (!errorInfo.FileName2.IsEmpty())
+
−      {
+
−        message += errorInfo.FileName2;
+
−        message += L"\n";
+
−      }
+
−      if (errorInfo.SystemError != 0)
+
−      {
+
−        message += NError::MyFormatMessageW(errorInfo.SystemError);
+
−        message += L"\n";
+
−      }
+
−      if (!message.IsEmpty())
+
−        stdStream << L"\nError:\n" << message;
+
−      throw CSystemException(result);
+
−    }
+
−    int numErrors = callback.FailedFiles.Size();
+
−    if (numErrors == 0)
+
−    {
+
−      if (callback.CantFindFiles.Size() == 0)
+
−        stdStream << kEverythingIsOk << endl;
+
−    }
+
−    else
+
−    {
+
−      stdStream << endl;
+
−      stdStream << "WARNINGS for files:" << endl << endl;
+
−      for (int i = 0; i < numErrors; i++)
+
−      {
+
−        stdStream << callback.FailedFiles[i] << " : ";
+
−        stdStream << NError::MyFormatMessageW(callback.FailedCodes[i]) << endl;
+
−      }
+
−      stdStream << "----------------" << endl;
+
−      stdStream << "WARNING: Cannot open " << numErrors << " file";
+
−      if (numErrors > 1)
+
−        stdStream << "s";
+
−      stdStream << endl;
+
−      exitCode = NExitCode::kWarning;
+
−    }
+
−    return exitCode;
+
−  }
+
−  else 
+
−    PrintHelpAndExit(stdStream);
+
−  return 0;
+
−}