1 // 7zExtract.cpp
2 
3 #include "StdAfx.h"
4 
5 #include "../../../Common/ComTry.h"
6 
7 #include "../../Common/ProgressUtils.h"
8 
9 #include "7zDecode.h"
10 // #include "7z1Decode.h"
11 #include "7zFolderOutStream.h"
12 #include "7zHandler.h"
13 
14 namespace NArchive {
15 namespace N7z {
16 
17 struct CExtractFolderInfo
18 {
19   #ifdef _7Z_VOL
20   int VolumeIndex;
21   #endif
22   CNum FileIndex;
23   CNum FolderIndex;
24   CBoolVector ExtractStatuses;
25   UInt64 UnpackSize;
CExtractFolderInfoNArchive::N7z::CExtractFolderInfo26   CExtractFolderInfo(
27     #ifdef _7Z_VOL
28     int volumeIndex,
29     #endif
30     CNum fileIndex, CNum folderIndex):
31     #ifdef _7Z_VOL
32     VolumeIndex(volumeIndex),
33     #endif
34     FileIndex(fileIndex),
35     FolderIndex(folderIndex),
36     UnpackSize(0)
37   {
38     if (fileIndex != kNumNoIndex)
39     {
40       ExtractStatuses.Reserve(1);
41       ExtractStatuses.Add(true);
42     }
43   };
44 };
45 
Extract(const UInt32 * indices,UInt32 numItems,Int32 testModeSpec,IArchiveExtractCallback * extractCallbackSpec)46 STDMETHODIMP CHandler::Extract(const UInt32 *indices, UInt32 numItems,
47     Int32 testModeSpec, IArchiveExtractCallback *extractCallbackSpec)
48 {
49   COM_TRY_BEGIN
50   bool testMode = (testModeSpec != 0);
51   CMyComPtr<IArchiveExtractCallback> extractCallback = extractCallbackSpec;
52   UInt64 importantTotalUnpacked = 0;
53 
54   bool allFilesMode = (numItems == (UInt32)-1);
55   if (allFilesMode)
56     numItems =
57     #ifdef _7Z_VOL
58     _refs.Size();
59     #else
60     _db.Files.Size();
61     #endif
62 
63   if(numItems == 0)
64     return S_OK;
65 
66   /*
67   if(_volumes.Size() != 1)
68     return E_FAIL;
69   const CVolume &volume = _volumes.Front();
70   const CArchiveDatabaseEx &_db = volume.Database;
71   IInStream *_inStream = volume.Stream;
72   */
73 
74   CObjectVector<CExtractFolderInfo> extractFolderInfoVector;
75   for (UInt32 ii = 0; ii < numItems; ii++)
76   {
77     // UInt32 fileIndex = allFilesMode ? indexIndex : indices[indexIndex];
78     UInt32 ref2Index = allFilesMode ? ii : indices[ii];
79     // const CRef2 &ref2 = _refs[ref2Index];
80 
81     // for (UInt32 ri = 0; ri < ref2.Refs.Size(); ri++)
82     {
83       #ifdef _7Z_VOL
84       // const CRef &ref = ref2.Refs[ri];
85       const CRef &ref = _refs[ref2Index];
86 
87       int volumeIndex = ref.VolumeIndex;
88       const CVolume &volume = _volumes[volumeIndex];
89       const CArchiveDatabaseEx &db = volume.Database;
90       UInt32 fileIndex = ref.ItemIndex;
91       #else
92       const CArchiveDatabaseEx &db = _db;
93       UInt32 fileIndex = ref2Index;
94       #endif
95 
96       CNum folderIndex = db.FileIndexToFolderIndexMap[fileIndex];
97       if (folderIndex == kNumNoIndex)
98       {
99         extractFolderInfoVector.Add(CExtractFolderInfo(
100             #ifdef _7Z_VOL
101             volumeIndex,
102             #endif
103             fileIndex, kNumNoIndex));
104         continue;
105       }
106       if (extractFolderInfoVector.IsEmpty() ||
107         folderIndex != extractFolderInfoVector.Back().FolderIndex
108         #ifdef _7Z_VOL
109         || volumeIndex != extractFolderInfoVector.Back().VolumeIndex
110         #endif
111         )
112       {
113         extractFolderInfoVector.Add(CExtractFolderInfo(
114             #ifdef _7Z_VOL
115             volumeIndex,
116             #endif
117             kNumNoIndex, folderIndex));
118         const CFolder &folderInfo = db.Folders[folderIndex];
119         UInt64 unpackSize = folderInfo.GetUnpackSize();
120         importantTotalUnpacked += unpackSize;
121         extractFolderInfoVector.Back().UnpackSize = unpackSize;
122       }
123 
124       CExtractFolderInfo &efi = extractFolderInfoVector.Back();
125 
126       // const CFolderInfo &folderInfo = m_dam_Folders[folderIndex];
127       CNum startIndex = db.FolderStartFileIndex[folderIndex];
128       for (CNum index = efi.ExtractStatuses.Size();
129           index <= fileIndex - startIndex; index++)
130       {
131         // UInt64 unpackSize = _db.Files[startIndex + index].UnpackSize;
132         // Count partial_folder_size
133         // efi.UnpackSize += unpackSize;
134         // importantTotalUnpacked += unpackSize;
135         efi.ExtractStatuses.Add(index == fileIndex - startIndex);
136       }
137     }
138   }
139 
140   RINOK(extractCallback->SetTotal(importantTotalUnpacked));
141 
142   CDecoder decoder(
143     #ifdef _ST_MODE
144     false
145     #else
146     true
147     #endif
148     );
149   // CDecoder1 decoder;
150 
151   UInt64 totalPacked = 0;
152   UInt64 totalUnpacked = 0;
153   UInt64 curPacked, curUnpacked;
154 
155   CLocalProgress *lps = new CLocalProgress;
156   CMyComPtr<ICompressProgressInfo> progress = lps;
157   lps->Init(extractCallback, false);
158 
159   for (int i = 0;; i++, totalUnpacked += curUnpacked, totalPacked += curPacked)
160   {
161     lps->OutSize = totalUnpacked;
162     lps->InSize = totalPacked;
163     RINOK(lps->SetCur());
164 
165     if (i >= extractFolderInfoVector.Size())
166       break;
167 
168     const CExtractFolderInfo &efi = extractFolderInfoVector[i];
169     curUnpacked = efi.UnpackSize;
170     curPacked = 0;
171 
172     CFolderOutStream *folderOutStream = new CFolderOutStream;
173     CMyComPtr<ISequentialOutStream> outStream(folderOutStream);
174 
175     #ifdef _7Z_VOL
176     const CVolume &volume = _volumes[efi.VolumeIndex];
177     const CArchiveDatabaseEx &db = volume.Database;
178     #else
179     const CArchiveDatabaseEx &db = _db;
180     #endif
181 
182     CNum startIndex;
183     if (efi.FileIndex != kNumNoIndex)
184       startIndex = efi.FileIndex;
185     else
186       startIndex = db.FolderStartFileIndex[efi.FolderIndex];
187 
188     HRESULT result = folderOutStream->Init(&db,
189         #ifdef _7Z_VOL
190         volume.StartRef2Index,
191         #else
192         0,
193         #endif
194         startIndex,
195         &efi.ExtractStatuses, extractCallback, testMode, _crcSize != 0);
196 
197     RINOK(result);
198 
199     if (efi.FileIndex != kNumNoIndex)
200       continue;
201 
202     CNum folderIndex = efi.FolderIndex;
203     const CFolder &folderInfo = db.Folders[folderIndex];
204 
205     curPacked = _db.GetFolderFullPackSize(folderIndex);
206 
207     CNum packStreamIndex = db.FolderStartPackStreamIndex[folderIndex];
208     UInt64 folderStartPackPos = db.GetFolderStreamPos(folderIndex, 0);
209 
210     #ifndef _NO_CRYPTO
211     CMyComPtr<ICryptoGetTextPassword> getTextPassword;
212     if (extractCallback)
213       extractCallback.QueryInterface(IID_ICryptoGetTextPassword, &getTextPassword);
214     #endif
215 
216     try
217     {
218       #ifndef _NO_CRYPTO
219       bool passwordIsDefined;
220       #endif
221 
222       HRESULT result = decoder.Decode(
223           EXTERNAL_CODECS_VARS
224           #ifdef _7Z_VOL
225           volume.Stream,
226           #else
227           _inStream,
228           #endif
229           folderStartPackPos,
230           &db.PackSizes[packStreamIndex],
231           folderInfo,
232           outStream,
233           progress
234           #ifndef _NO_CRYPTO
235           , getTextPassword, passwordIsDefined
236           #endif
237           #if !defined(_7ZIP_ST) && !defined(_SFX)
238           , true, _numThreads
239           #endif
240           );
241 
242       if (result == S_FALSE)
243       {
244         RINOK(folderOutStream->FlushCorrupted(NExtract::NOperationResult::kDataError));
245         continue;
246       }
247       if (result == E_NOTIMPL)
248       {
249         RINOK(folderOutStream->FlushCorrupted(NExtract::NOperationResult::kUnSupportedMethod));
250         continue;
251       }
252       if (result != S_OK)
253         return result;
254       if (folderOutStream->WasWritingFinished() != S_OK)
255       {
256         RINOK(folderOutStream->FlushCorrupted(NExtract::NOperationResult::kDataError));
257         continue;
258       }
259     }
260     catch(...)
261     {
262       RINOK(folderOutStream->FlushCorrupted(NExtract::NOperationResult::kDataError));
263       continue;
264     }
265   }
266   return S_OK;
267   COM_TRY_END
268 }
269 
270 }}
271