vcs.maemo.org Git - physicsfs/blob - lzma/CPP/7zip/UI/Common/ArchiveOpenCallback.h

   1 // ArchiveOpenCallback.h
   2
   3 #ifndef __ARCHIVE_OPEN_CALLBACK_H
   4 #define __ARCHIVE_OPEN_CALLBACK_H
   5
   6 #include "Common/MyString.h"
   7 #include "Common/MyCom.h"
   8 #include "Windows/FileFind.h"
   9
  10 #ifndef _NO_CRYPTO
  11 #include "../../IPassword.h"
  12 #endif
  13 #include "../../Archive/IArchive.h"
  14
  15 struct IOpenCallbackUI
  16 {
  17   virtual HRESULT CheckBreak() = 0;
  18   virtual HRESULT SetTotal(const UInt64 *files, const UInt64 *bytes) = 0;
  19   virtual HRESULT SetCompleted(const UInt64 *files, const UInt64 *bytes) = 0;
  20   #ifndef _NO_CRYPTO
  21   virtual HRESULT CryptoGetTextPassword(BSTR *password) = 0;
  22   virtual HRESULT GetPasswordIfAny(UString &password) = 0;
  23   virtual bool WasPasswordAsked() = 0;
  24   virtual void ClearPasswordWasAskedFlag() = 0;
  25   #endif
  26 };
  27
  28 class COpenCallbackImp:
  29   public IArchiveOpenCallback,
  30   public IArchiveOpenVolumeCallback,
  31   public IArchiveOpenSetSubArchiveName,
  32   #ifndef _NO_CRYPTO
  33   public ICryptoGetTextPassword,
  34   #endif
  35   public CMyUnknownImp
  36 {
  37 public:
  38   #ifndef _NO_CRYPTO
  39   MY_UNKNOWN_IMP3(
  40       IArchiveOpenVolumeCallback,
  41       ICryptoGetTextPassword,
  42       IArchiveOpenSetSubArchiveName
  43       )
  44   #else
  45   MY_UNKNOWN_IMP2(
  46       IArchiveOpenVolumeCallback,
  47       IArchiveOpenSetSubArchiveName
  48       )
  49   #endif
  50
  51   STDMETHOD(SetTotal)(const UInt64 *files, const UInt64 *bytes);
  52   STDMETHOD(SetCompleted)(const UInt64 *files, const UInt64 *bytes);
  53
  54   // IArchiveOpenVolumeCallback
  55   STDMETHOD(GetProperty)(PROPID propID, PROPVARIANT *value);
  56   STDMETHOD(GetStream)(const wchar_t *name, IInStream **inStream);
  57
  58   #ifndef _NO_CRYPTO
  59   // ICryptoGetTextPassword
  60   STDMETHOD(CryptoGetTextPassword)(BSTR *password);
  61   #endif
  62
  63   STDMETHOD(SetSubArchiveName(const wchar_t *name))
  64   {
  65     _subArchiveMode = true;
  66     _subArchiveName = name;
  67     return  S_OK;
  68   }
  69
  70 private:
  71   UString _folderPrefix;
  72   NWindows::NFile::NFind::CFileInfoW _fileInfo;
  73   bool _subArchiveMode;
  74   UString _subArchiveName;
  75 public:
  76   UStringVector FileNames;
  77   IOpenCallbackUI *Callback;
  78   UInt64 TotalSize;
  79
  80   COpenCallbackImp(): Callback(NULL) {}
  81   void Init(const UString &folderPrefix,  const UString &fileName)
  82   {
  83     _folderPrefix = folderPrefix;
  84     if (!NWindows::NFile::NFind::FindFile(_folderPrefix + fileName, _fileInfo))
  85       throw 1;
  86     FileNames.Clear();
  87     _subArchiveMode = false;
  88     TotalSize = 0;
  89   }
  90   int FindName(const UString &name);
  91 };
  92
  93 #endif