OLD | NEW |
| (Empty) |
1 // ExtractCallbackConsole.h | |
2 | |
3 #include "StdAfx.h" | |
4 | |
5 #include "ExtractCallbackConsole.h" | |
6 #include "UserInputUtils.h" | |
7 #include "ConsoleClose.h" | |
8 | |
9 #include "Common/Wildcard.h" | |
10 | |
11 #include "Windows/FileDir.h" | |
12 #include "Windows/FileFind.h" | |
13 #include "Windows/Time.h" | |
14 #include "Windows/Defs.h" | |
15 #include "Windows/PropVariant.h" | |
16 #include "Windows/Error.h" | |
17 #include "Windows/PropVariantConversions.h" | |
18 | |
19 #include "../../Common/FilePathAutoRename.h" | |
20 | |
21 #include "../Common/ExtractingFilePath.h" | |
22 | |
23 using namespace NWindows; | |
24 using namespace NFile; | |
25 using namespace NDirectory; | |
26 | |
27 static const char *kTestString = "Testing "; | |
28 static const char *kExtractString = "Extracting "; | |
29 static const char *kSkipString = "Skipping "; | |
30 | |
31 // static const char *kCantAutoRename = "can not create file with auto name\n"; | |
32 // static const char *kCantRenameFile = "can not rename existing file\n"; | |
33 // static const char *kCantDeleteOutputFile = "can not delete output file "; | |
34 static const char *kError = "ERROR: "; | |
35 static const char *kMemoryExceptionMessage = "Can't allocate required memory!"; | |
36 | |
37 static const char *kProcessing = "Processing archive: "; | |
38 static const char *kEverythingIsOk = "Everything is Ok"; | |
39 static const char *kNoFiles = "No files to process"; | |
40 | |
41 static const char *kUnsupportedMethod = "Unsupported Method"; | |
42 static const char *kCrcFailed = "CRC Failed"; | |
43 static const char *kCrcFailedEncrypted = "CRC Failed in encrypted file. Wrong pa
ssword?"; | |
44 static const char *kDataError = "Data Error"; | |
45 static const char *kDataErrorEncrypted = "Data Error in encrypted file. Wrong pa
ssword?"; | |
46 static const char *kUnknownError = "Unknown Error"; | |
47 | |
48 STDMETHODIMP CExtractCallbackConsole::SetTotal(UInt64) | |
49 { | |
50 if (NConsoleClose::TestBreakSignal()) | |
51 return E_ABORT; | |
52 return S_OK; | |
53 } | |
54 | |
55 STDMETHODIMP CExtractCallbackConsole::SetCompleted(const UInt64 *) | |
56 { | |
57 if (NConsoleClose::TestBreakSignal()) | |
58 return E_ABORT; | |
59 return S_OK; | |
60 } | |
61 | |
62 STDMETHODIMP CExtractCallbackConsole::AskOverwrite( | |
63 const wchar_t *existName, const FILETIME *, const UInt64 *, | |
64 const wchar_t *newName, const FILETIME *, const UInt64 *, | |
65 Int32 *answer) | |
66 { | |
67 (*OutStream) << "file " << existName << | |
68 "\nalready exists. Overwrite with " << endl; | |
69 (*OutStream) << newName; | |
70 | |
71 NUserAnswerMode::EEnum overwriteAnswer = ScanUserYesNoAllQuit(OutStream); | |
72 | |
73 switch(overwriteAnswer) | |
74 { | |
75 case NUserAnswerMode::kQuit: return E_ABORT; | |
76 case NUserAnswerMode::kNo: *answer = NOverwriteAnswer::kNo; break; | |
77 case NUserAnswerMode::kNoAll: *answer = NOverwriteAnswer::kNoToAll; break; | |
78 case NUserAnswerMode::kYesAll: *answer = NOverwriteAnswer::kYesToAll; break; | |
79 case NUserAnswerMode::kYes: *answer = NOverwriteAnswer::kYes; break; | |
80 case NUserAnswerMode::kAutoRenameAll: *answer = NOverwriteAnswer::kAutoRenam
e; break; | |
81 default: return E_FAIL; | |
82 } | |
83 return S_OK; | |
84 } | |
85 | |
86 STDMETHODIMP CExtractCallbackConsole::PrepareOperation(const wchar_t *name, bool
/* isFolder */, Int32 askExtractMode, const UInt64 *position) | |
87 { | |
88 switch (askExtractMode) | |
89 { | |
90 case NArchive::NExtract::NAskMode::kExtract: (*OutStream) << kExtractString;
break; | |
91 case NArchive::NExtract::NAskMode::kTest: (*OutStream) << kTestString; br
eak; | |
92 case NArchive::NExtract::NAskMode::kSkip: (*OutStream) << kSkipString; br
eak; | |
93 }; | |
94 (*OutStream) << name; | |
95 if (position != 0) | |
96 (*OutStream) << " <" << *position << ">"; | |
97 return S_OK; | |
98 } | |
99 | |
100 STDMETHODIMP CExtractCallbackConsole::MessageError(const wchar_t *message) | |
101 { | |
102 (*OutStream) << message << endl; | |
103 NumFileErrorsInCurrentArchive++; | |
104 NumFileErrors++; | |
105 return S_OK; | |
106 } | |
107 | |
108 STDMETHODIMP CExtractCallbackConsole::SetOperationResult(Int32 operationResult,
bool encrypted) | |
109 { | |
110 switch(operationResult) | |
111 { | |
112 case NArchive::NExtract::NOperationResult::kOK: | |
113 break; | |
114 default: | |
115 { | |
116 NumFileErrorsInCurrentArchive++; | |
117 NumFileErrors++; | |
118 (*OutStream) << " "; | |
119 switch(operationResult) | |
120 { | |
121 case NArchive::NExtract::NOperationResult::kUnSupportedMethod: | |
122 (*OutStream) << kUnsupportedMethod; | |
123 break; | |
124 case NArchive::NExtract::NOperationResult::kCRCError: | |
125 (*OutStream) << (encrypted ? kCrcFailedEncrypted: kCrcFailed); | |
126 break; | |
127 case NArchive::NExtract::NOperationResult::kDataError: | |
128 (*OutStream) << (encrypted ? kDataErrorEncrypted : kDataError); | |
129 break; | |
130 default: | |
131 (*OutStream) << kUnknownError; | |
132 } | |
133 } | |
134 } | |
135 (*OutStream) << endl; | |
136 return S_OK; | |
137 } | |
138 | |
139 #ifndef _NO_CRYPTO | |
140 | |
141 HRESULT CExtractCallbackConsole::SetPassword(const UString &password) | |
142 { | |
143 PasswordIsDefined = true; | |
144 Password = password; | |
145 return S_OK; | |
146 } | |
147 | |
148 STDMETHODIMP CExtractCallbackConsole::CryptoGetTextPassword(BSTR *password) | |
149 { | |
150 if (!PasswordIsDefined) | |
151 { | |
152 Password = GetPassword(OutStream); | |
153 PasswordIsDefined = true; | |
154 } | |
155 return StringToBstr(Password, password); | |
156 } | |
157 | |
158 #endif | |
159 | |
160 HRESULT CExtractCallbackConsole::BeforeOpen(const wchar_t *name) | |
161 { | |
162 NumArchives++; | |
163 NumFileErrorsInCurrentArchive = 0; | |
164 (*OutStream) << endl << kProcessing << name << endl; | |
165 return S_OK; | |
166 } | |
167 | |
168 HRESULT CExtractCallbackConsole::OpenResult(const wchar_t * /* name */, HRESULT
result, bool encrypted) | |
169 { | |
170 (*OutStream) << endl; | |
171 if (result != S_OK) | |
172 { | |
173 (*OutStream) << "Error: "; | |
174 if (result == S_FALSE) | |
175 { | |
176 (*OutStream) << (encrypted ? | |
177 "Can not open encrypted archive. Wrong password?" : | |
178 "Can not open file as archive"); | |
179 } | |
180 else | |
181 { | |
182 if (result == E_OUTOFMEMORY) | |
183 (*OutStream) << "Can't allocate required memory"; | |
184 else | |
185 (*OutStream) << NError::MyFormatMessage(result); | |
186 } | |
187 (*OutStream) << endl; | |
188 NumArchiveErrors++; | |
189 } | |
190 return S_OK; | |
191 } | |
192 | |
193 HRESULT CExtractCallbackConsole::ThereAreNoFiles() | |
194 { | |
195 (*OutStream) << endl << kNoFiles << endl; | |
196 return S_OK; | |
197 } | |
198 | |
199 HRESULT CExtractCallbackConsole::ExtractResult(HRESULT result) | |
200 { | |
201 if (result == S_OK) | |
202 { | |
203 (*OutStream) << endl; | |
204 if (NumFileErrorsInCurrentArchive == 0) | |
205 (*OutStream) << kEverythingIsOk << endl; | |
206 else | |
207 { | |
208 NumArchiveErrors++; | |
209 (*OutStream) << "Sub items Errors: " << NumFileErrorsInCurrentArchive << e
ndl; | |
210 } | |
211 } | |
212 if (result == S_OK) | |
213 return result; | |
214 NumArchiveErrors++; | |
215 if (result == E_ABORT || result == ERROR_DISK_FULL) | |
216 return result; | |
217 (*OutStream) << endl << kError; | |
218 if (result == E_OUTOFMEMORY) | |
219 (*OutStream) << kMemoryExceptionMessage; | |
220 else | |
221 { | |
222 UString message; | |
223 NError::MyFormatMessage(result, message); | |
224 (*OutStream) << message; | |
225 } | |
226 (*OutStream) << endl; | |
227 return S_OK; | |
228 } | |
OLD | NEW |