The Tor Browser: other-licenses/7zstub/src/7zip/UI/Common/ArchiverInfo.cpp@6474c204b198

Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.

     1 // ArchiverInfo.cpp

     3 #include "StdAfx.h"

     5 #include "ArchiverInfo.h"

     7 #ifndef EXCLUDE_COM

     9 #include "Common/StringConvert.h"

    10 #include "Windows/FileFind.h"

    11 #include "Windows/FileName.h"

    12 #include "Windows/DLL.h"

    13 #ifdef _WIN32

    14 #include "Windows/Registry.h"

    15 #endif

    16 #include "Windows/PropVariant.h"

    17 #include "../../Archive/IArchive.h"

    19 using namespace NWindows;

    20 using namespace NFile;

    22 #endif

    24 extern HINSTANCE g_hInstance;

    26 #ifndef EXCLUDE_COM

    28 static void SplitString(const UString &srcString, UStringVector &destStrings)

    29 {

    30   destStrings.Clear();

    31   UString string;

    32   int len = srcString.Length();

    33   if (len == 0)

    34     return;

    35   for (int i = 0; i < len; i++)

    36   {

    37     wchar_t c = srcString[i];

    38     if (c == L' ')

    39     {

    40       if (!string.IsEmpty())

    41       {

    42         destStrings.Add(string);

    43         string.Empty();

    44       }

    45     }

    46     else

    47       string += c;

    48   }

    49   if (!string.IsEmpty())

    50     destStrings.Add(string);

    51 }

    53 typedef UInt32 (WINAPI * GetHandlerPropertyFunc)(

    54     PROPID propID, PROPVARIANT *value);

    56 static UString GetModuleFolderPrefix()

    57 {

    58   UString path;

    59   NDLL::MyGetModuleFileName(g_hInstance, path);

    60   int pos = path.ReverseFind(WCHAR_PATH_SEPARATOR);

    61   return path.Left(pos + 1);

    62 }

    64 static wchar_t *kFormatFolderName = L"Formats";

    66 #ifdef _WIN32

    67 static LPCTSTR kRegistryPath = TEXT("Software\\7-zip");

    68 static LPCWSTR kProgramPathValue = L"Path";

    69 static bool ReadPathFromRegistry(HKEY baseKey, UString &path)

    70 {

    71   NRegistry::CKey key;

    72   if(key.Open(baseKey, kRegistryPath, KEY_READ) == ERROR_SUCCESS)

    73     if (key.QueryValue(kProgramPathValue, path) == ERROR_SUCCESS)

    74     {

    75       NName::NormalizeDirPathPrefix(path);

    76       return true;

    77     }

    78   return false;

    79 }

    80 #endif

    82 static UString GetBaseFolderPrefixFromRegistry()

    83 {

    84   UString moduleFolderPrefix = GetModuleFolderPrefix();

    85   NFind::CFileInfoW fileInfo;

    86   if (NFind::FindFile(moduleFolderPrefix + kFormatFolderName, fileInfo))

    87     if (fileInfo.IsDirectory())

    88       return moduleFolderPrefix;

    89   UString path;

    90   #ifdef _WIN32

    91   if(ReadPathFromRegistry(HKEY_CURRENT_USER, path))

    92     return path;

    93   if(ReadPathFromRegistry(HKEY_LOCAL_MACHINE, path))

    94     return path;

    95   #endif

    96   return moduleFolderPrefix;

    97 }

    99 typedef UInt32 (WINAPI *CreateObjectPointer)(

   100     const GUID *clsID,

   101     const GUID *interfaceID,

   102     void **outObject);

   104 #endif

   106 #ifndef _SFX

   107 static void SetBuffer(CByteBuffer &bb, const Byte *data, int size)

   108 {

   109   bb.SetCapacity(size);

   110   memmove((Byte *)bb, data, size);

   111 }

   112 #endif

   114 void ReadArchiverInfoList(CObjectVector<CArchiverInfo> &archivers)

   115 {

   116   archivers.Clear();

   118   #ifdef EXCLUDE_COM

   120   #ifdef FORMAT_7Z

   121   {

   122     CArchiverInfo item;

   123     item.UpdateEnabled = true;

   124     item.Name = L"7z";

   125     item.Extensions.Add(CArchiverExtInfo(L"7z"));

   126     #ifndef _SFX

   127     const unsigned char kSig[] = {'7' , 'z', 0xBC, 0xAF, 0x27, 0x1C};

   128     SetBuffer(item.StartSignature, kSig, 6);

   129     #endif

   130     archivers.Add(item);

   131   }

   132   #endif

   134   #ifdef FORMAT_BZIP2

   135   {

   136     CArchiverInfo item;

   137     item.UpdateEnabled = true;

   138     item.KeepName = true;

   139     item.Name = L"BZip2";

   140     item.Extensions.Add(CArchiverExtInfo(L"bz2"));

   141     item.Extensions.Add(CArchiverExtInfo(L"tbz2", L".tar"));

   142     #ifndef _SFX

   143     const unsigned char sig[] = {'B' , 'Z', 'h' };

   144     SetBuffer(item.StartSignature, sig, 3);

   145     #endif

   146     archivers.Add(item);

   147   }

   148   #endif

   150   #ifdef FORMAT_GZIP

   151   {

   152     CArchiverInfo item;

   153     item.UpdateEnabled = true;

   154     item.Name = L"GZip";

   155     item.Extensions.Add(CArchiverExtInfo(L"gz"));

   156     item.Extensions.Add(CArchiverExtInfo(L"tgz", L".tar"));

   157     #ifndef _SFX

   158     const unsigned char sig[] = { 0x1F, 0x8B };

   159     SetBuffer(item.StartSignature, sig, 2);

   160     #endif

   161     archivers.Add(item);

   162   }

   163   #endif

   165   #ifdef FORMAT_SPLIT

   166   {

   167     CArchiverInfo item;

   168     item.UpdateEnabled = false;

   169     item.KeepName = true;

   170     item.Name = L"Split";

   171     item.Extensions.Add(CArchiverExtInfo(L"001"));

   172     archivers.Add(item);

   173   }

   174   #endif

   176   #ifdef FORMAT_TAR

   177   {

   178     CArchiverInfo item;

   179     item.UpdateEnabled = true;

   180     item.Name = L"Tar";

   181     item.Extensions.Add(CArchiverExtInfo(L"tar"));

   182     archivers.Add(item);

   183   }

   184   #endif

   186   #ifdef FORMAT_ZIP

   187   {

   188     CArchiverInfo item;

   189     item.UpdateEnabled = true;

   190     item.Name = L"Zip";

   191     item.Extensions.Add(CArchiverExtInfo(L"zip"));

   192     #ifndef _SFX

   193     const unsigned char sig[] = { 0x50, 0x4B, 0x03, 0x04 };

   194     SetBuffer(item.StartSignature, sig, 4);

   195     #endif

   196     archivers.Add(item);

   197   }

   198   #endif

   200   #ifdef FORMAT_CPIO

   201   {

   202     CArchiverInfo item;

   203     item.Name = L"Cpio";

   204     item.Extensions.Add(CArchiverExtInfo(L"cpio"));

   205     archivers.Add(item);

   206   }

   207   #endif

   209   #ifdef FORMAT_RPM

   210   {

   211     CArchiverInfo item;

   212     item.Name = L"Rpm";

   213     item.Extensions.Add(CArchiverExtInfo(L"rpm", L".cpio.gz"));

   214     archivers.Add(item);

   215   }

   216   #endif

   218   #ifdef FORMAT_ARJ

   219   {

   220     CArchiverInfo item;

   221     item.Name = L"Arj";

   222     item.Extensions.Add(CArchiverExtInfo(L"arj"));

   223     #ifndef _SFX

   224     const unsigned char sig[] = { 0x60, 0xEA };

   225     SetBuffer(item.StartSignature, sig, 2);

   226     #endif

   227     archivers.Add(item);

   228   }

   229   #endif

   231   #ifdef FORMAT_Z

   232   {

   233     CArchiverInfo item;

   234     item.Name = L"Z";

   235     item.Extensions.Add(CArchiverExtInfo(L"Z"));

   236     #ifndef _SFX

   237     const unsigned char sig[] = { 0x1F, 0x9D };

   238     SetBuffer(item.StartSignature, sig, 2);

   239     #endif

   240     archivers.Add(item);

   241   }

   242   #endif

   244   #else

   246   UString folderPath = GetBaseFolderPrefixFromRegistry() +

   247       (UString)kFormatFolderName + (UString)WSTRING_PATH_SEPARATOR;

   248   NFind::CEnumeratorW enumerator(folderPath + L"*");

   249   NFind::CFileInfoW fileInfo;

   250   while (enumerator.Next(fileInfo))

   251   {

   252     if (fileInfo.IsDirectory())

   253       continue;

   254     UString filePath = folderPath + fileInfo.Name;

   255     {

   256       NDLL::CLibrary library;

   257       if (!library.LoadEx(filePath, LOAD_LIBRARY_AS_DATAFILE))

   258         continue;

   259     }

   261     NDLL::CLibrary library;

   262     if (!library.Load(filePath))

   263       continue;

   264     GetHandlerPropertyFunc getHandlerProperty = (GetHandlerPropertyFunc)

   265         library.GetProcAddress("GetHandlerProperty");

   266     if (getHandlerProperty == NULL)

   267       continue;

   269     CArchiverInfo item;

   270     item.FilePath = filePath;

   272     NWindows::NCOM::CPropVariant prop;

   273     if (getHandlerProperty(NArchive::kName, &prop) != S_OK)

   274       continue;

   275     if (prop.vt != VT_BSTR)

   276       continue;

   277     item.Name = prop.bstrVal;

   278     prop.Clear();

   280     if (getHandlerProperty(NArchive::kClassID, &prop) != S_OK)

   281       continue;

   282     if (prop.vt != VT_BSTR)

   283       continue;

   284     item.ClassID = *(const GUID *)prop.bstrVal;

   285     prop.Clear();

   287     if (getHandlerProperty(NArchive::kExtension, &prop) != S_OK)

   288       continue;

   289     if (prop.vt != VT_BSTR)

   290       continue;

   292     UString ext  = prop.bstrVal;

   293     UString addExt;

   295     prop.Clear();

   297     if (getHandlerProperty(NArchive::kAddExtension, &prop) != S_OK)

   298       continue;

   299     if (prop.vt == VT_BSTR)

   300     {

   301       addExt = prop.bstrVal;

   302     }

   303     else if (prop.vt != VT_EMPTY)

   304       continue;

   305     prop.Clear();

   307     UStringVector exts, addExts;

   308     SplitString(ext, exts);

   309     SplitString(addExt, addExts);

   311     prop.Clear();

   312     for (int i = 0; i < exts.Size(); i++)

   313     {

   314       CArchiverExtInfo extInfo;

   315       extInfo.Ext = exts[i];

   316       if (addExts.Size() > 0)

   317         extInfo.AddExt = addExts[i];

   318       if (extInfo.AddExt == L"*")

   319         extInfo.AddExt.Empty();

   320       item.Extensions.Add(extInfo);

   321     }

   323     if (getHandlerProperty(NArchive::kUpdate, &prop) == S_OK)

   324       if (prop.vt == VT_BOOL)

   325         item.UpdateEnabled = VARIANT_BOOLToBool(prop.boolVal);

   326     prop.Clear();

   328     if (item.UpdateEnabled)

   329     {

   330       if (getHandlerProperty(NArchive::kKeepName, &prop) == S_OK)

   331         if (prop.vt == VT_BOOL)

   332           item.KeepName = VARIANT_BOOLToBool(prop.boolVal);

   333       prop.Clear();

   334     }

   336     if (getHandlerProperty(NArchive::kStartSignature, &prop) == S_OK)

   337     {

   338       if (prop.vt == VT_BSTR)

   339       {

   340         UINT len = ::SysStringByteLen(prop.bstrVal);

   341         item.StartSignature.SetCapacity(len);

   342         memmove(item.StartSignature, prop.bstrVal, len);

   343       }

   344     }

   345     prop.Clear();

   347     if (getHandlerProperty(NArchive::kAssociate, &prop) == S_OK)

   348       if (prop.vt == VT_BOOL)

   349         item.Associate = VARIANT_BOOLToBool(prop.boolVal);

   350     prop.Clear();

   353     archivers.Add(item);

   354   }

   356   #endif

   357 }

The Tor Browser / file revision

other-licenses/7zstub/src/7zip/UI/Common/ArchiverInfo.cpp@6474c204b198

other-licenses/7zstub/src/7zip/UI/Common/ArchiverInfo.cpp