1 // ArchiveOpenCallback.h
2 
3 #ifndef __ARCHIVE_OPEN_CALLBACK_H
4 #define __ARCHIVE_OPEN_CALLBACK_H
5 
6 #include "../../../Common/MyCom.h"
7 
8 #include "../../../Windows/FileFind.h"
9 
10 #ifndef _NO_CRYPTO
11 #include "../../IPassword.h"
12 #endif
13 #include "../../Archive/IArchive.h"
14 
15 #ifdef _NO_CRYPTO
16 
17 #define INTERFACE_IOpenCallbackUI_Crypto(x)
18 
19 #else
20 
21 #define INTERFACE_IOpenCallbackUI_Crypto(x) \
22   virtual HRESULT Open_CryptoGetTextPassword(BSTR *password) x; \
23   virtual HRESULT Open_GetPasswordIfAny(bool &passwordIsDefined, UString &password) x; \
24   virtual bool Open_WasPasswordAsked() x; \
25   virtual void Open_ClearPasswordWasAskedFlag() x; \
26 
27 #endif
28 
29 #define INTERFACE_IOpenCallbackUI(x) \
30   virtual HRESULT Open_CheckBreak() x; \
31   virtual HRESULT Open_SetTotal(const UInt64 *files, const UInt64 *bytes) x; \
32   virtual HRESULT Open_SetCompleted(const UInt64 *files, const UInt64 *bytes) x; \
33   INTERFACE_IOpenCallbackUI_Crypto(x)
34 
35 struct IOpenCallbackUI
36 {
37   INTERFACE_IOpenCallbackUI(=0)
38 };
39 
40 class COpenCallbackImp:
41   public IArchiveOpenCallback,
42   public IArchiveOpenVolumeCallback,
43   public IArchiveOpenSetSubArchiveName,
44   #ifndef _NO_CRYPTO
45   public ICryptoGetTextPassword,
46   #endif
47   public CMyUnknownImp
48 {
49 public:
50   #ifndef _NO_CRYPTO
51   MY_UNKNOWN_IMP3(
52       IArchiveOpenVolumeCallback,
53       ICryptoGetTextPassword,
54       IArchiveOpenSetSubArchiveName
55       )
56   #else
57   MY_UNKNOWN_IMP2(
58       IArchiveOpenVolumeCallback,
59       IArchiveOpenSetSubArchiveName
60       )
61   #endif
62 
63   INTERFACE_IArchiveOpenCallback(;)
64   INTERFACE_IArchiveOpenVolumeCallback(;)
65 
66   #ifndef _NO_CRYPTO
67   STDMETHOD(CryptoGetTextPassword)(BSTR *password);
68   #endif
69 
STDMETHOD(SetSubArchiveName (const wchar_t * name))70   STDMETHOD(SetSubArchiveName(const wchar_t *name))
71   {
72     _subArchiveMode = true;
73     _subArchiveName = name;
74     // TotalSize = 0;
75     return S_OK;
76   }
77 
78 private:
79   FString _folderPrefix;
80   NWindows::NFile::NFind::CFileInfo _fileInfo;
81   bool _subArchiveMode;
82   UString _subArchiveName;
83 
84 public:
85   UStringVector FileNames;
86   CBoolVector FileNames_WasUsed;
87   CRecordVector<UInt64> FileSizes;
88 
89   IOpenCallbackUI *Callback;
90   CMyComPtr<IArchiveOpenCallback> ReOpenCallback;
91   // UInt64 TotalSize;
92 
COpenCallbackImp()93   COpenCallbackImp(): Callback(NULL) {}
Init(const FString & folderPrefix,const FString & fileName)94   void Init(const FString &folderPrefix, const FString &fileName)
95   {
96     _folderPrefix = folderPrefix;
97     if (!_fileInfo.Find(_folderPrefix + fileName))
98       throw 20121118;
99     FileNames.Clear();
100     FileNames_WasUsed.Clear();
101     FileSizes.Clear();
102     _subArchiveMode = false;
103     // TotalSize = 0;
104   }
SetSecondFileInfo(CFSTR newName)105   bool SetSecondFileInfo(CFSTR newName)
106   {
107     return _fileInfo.Find(newName) && !_fileInfo.IsDir();
108   }
109 };
110 
111 #endif
112