1*f6dc9357SAndroid Build Coastguard Worker // UdfHandler.cpp
2*f6dc9357SAndroid Build Coastguard Worker
3*f6dc9357SAndroid Build Coastguard Worker #include "StdAfx.h"
4*f6dc9357SAndroid Build Coastguard Worker
5*f6dc9357SAndroid Build Coastguard Worker #include "../../../Common/ComTry.h"
6*f6dc9357SAndroid Build Coastguard Worker
7*f6dc9357SAndroid Build Coastguard Worker #include "../../../Windows/PropVariant.h"
8*f6dc9357SAndroid Build Coastguard Worker #include "../../../Windows/TimeUtils.h"
9*f6dc9357SAndroid Build Coastguard Worker
10*f6dc9357SAndroid Build Coastguard Worker #include "../../Common/LimitedStreams.h"
11*f6dc9357SAndroid Build Coastguard Worker #include "../../Common/ProgressUtils.h"
12*f6dc9357SAndroid Build Coastguard Worker #include "../../Common/RegisterArc.h"
13*f6dc9357SAndroid Build Coastguard Worker #include "../../Common/StreamObjects.h"
14*f6dc9357SAndroid Build Coastguard Worker
15*f6dc9357SAndroid Build Coastguard Worker #include "../../Compress/CopyCoder.h"
16*f6dc9357SAndroid Build Coastguard Worker
17*f6dc9357SAndroid Build Coastguard Worker #include "UdfHandler.h"
18*f6dc9357SAndroid Build Coastguard Worker
19*f6dc9357SAndroid Build Coastguard Worker namespace NArchive {
20*f6dc9357SAndroid Build Coastguard Worker namespace NUdf {
21*f6dc9357SAndroid Build Coastguard Worker
UdfTimeToFileTime(const CTime & t,NWindows::NCOM::CPropVariant & prop)22*f6dc9357SAndroid Build Coastguard Worker static void UdfTimeToFileTime(const CTime &t, NWindows::NCOM::CPropVariant &prop)
23*f6dc9357SAndroid Build Coastguard Worker {
24*f6dc9357SAndroid Build Coastguard Worker UInt64 numSecs;
25*f6dc9357SAndroid Build Coastguard Worker const Byte *d = t.Data;
26*f6dc9357SAndroid Build Coastguard Worker if (!NWindows::NTime::GetSecondsSince1601(t.GetYear(), d[4], d[5], d[6], d[7], d[8], numSecs))
27*f6dc9357SAndroid Build Coastguard Worker return;
28*f6dc9357SAndroid Build Coastguard Worker if (t.IsLocal())
29*f6dc9357SAndroid Build Coastguard Worker numSecs = (UInt64)((Int64)numSecs - (Int64)((Int32)t.GetMinutesOffset() * 60));
30*f6dc9357SAndroid Build Coastguard Worker const UInt32 m0 = d[9];
31*f6dc9357SAndroid Build Coastguard Worker const UInt32 m1 = d[10];
32*f6dc9357SAndroid Build Coastguard Worker const UInt32 m2 = d[11];
33*f6dc9357SAndroid Build Coastguard Worker unsigned numDigits = 0;
34*f6dc9357SAndroid Build Coastguard Worker UInt64 v = numSecs * 10000000;
35*f6dc9357SAndroid Build Coastguard Worker if (m0 < 100 && m1 < 100 && m2 < 100)
36*f6dc9357SAndroid Build Coastguard Worker {
37*f6dc9357SAndroid Build Coastguard Worker v += m0 * 100000 + m1 * 1000 + m2 * 10;
38*f6dc9357SAndroid Build Coastguard Worker numDigits = 6;
39*f6dc9357SAndroid Build Coastguard Worker }
40*f6dc9357SAndroid Build Coastguard Worker prop.SetAsTimeFrom_Ft64_Prec(v, k_PropVar_TimePrec_Base + numDigits);
41*f6dc9357SAndroid Build Coastguard Worker }
42*f6dc9357SAndroid Build Coastguard Worker
43*f6dc9357SAndroid Build Coastguard Worker static const Byte kProps[] =
44*f6dc9357SAndroid Build Coastguard Worker {
45*f6dc9357SAndroid Build Coastguard Worker kpidPath,
46*f6dc9357SAndroid Build Coastguard Worker kpidIsDir,
47*f6dc9357SAndroid Build Coastguard Worker kpidSize,
48*f6dc9357SAndroid Build Coastguard Worker kpidPackSize,
49*f6dc9357SAndroid Build Coastguard Worker kpidMTime,
50*f6dc9357SAndroid Build Coastguard Worker kpidATime,
51*f6dc9357SAndroid Build Coastguard Worker kpidCTime,
52*f6dc9357SAndroid Build Coastguard Worker kpidChangeTime,
53*f6dc9357SAndroid Build Coastguard Worker // kpidUserId,
54*f6dc9357SAndroid Build Coastguard Worker // kpidGroupId,
55*f6dc9357SAndroid Build Coastguard Worker // kpidPosixAttrib,
56*f6dc9357SAndroid Build Coastguard Worker kpidLinks
57*f6dc9357SAndroid Build Coastguard Worker };
58*f6dc9357SAndroid Build Coastguard Worker
59*f6dc9357SAndroid Build Coastguard Worker static const Byte kArcProps[] =
60*f6dc9357SAndroid Build Coastguard Worker {
61*f6dc9357SAndroid Build Coastguard Worker kpidUnpackVer,
62*f6dc9357SAndroid Build Coastguard Worker kpidClusterSize,
63*f6dc9357SAndroid Build Coastguard Worker kpidSectorSize,
64*f6dc9357SAndroid Build Coastguard Worker kpidCTime,
65*f6dc9357SAndroid Build Coastguard Worker kpidMTime,
66*f6dc9357SAndroid Build Coastguard Worker kpidComment
67*f6dc9357SAndroid Build Coastguard Worker };
68*f6dc9357SAndroid Build Coastguard Worker
69*f6dc9357SAndroid Build Coastguard Worker IMP_IInArchive_Props
70*f6dc9357SAndroid Build Coastguard Worker IMP_IInArchive_ArcProps
71*f6dc9357SAndroid Build Coastguard Worker
Z7_COM7F_IMF(CHandler::GetArchiveProperty (PROPID propID,PROPVARIANT * value))72*f6dc9357SAndroid Build Coastguard Worker Z7_COM7F_IMF(CHandler::GetArchiveProperty(PROPID propID, PROPVARIANT *value))
73*f6dc9357SAndroid Build Coastguard Worker {
74*f6dc9357SAndroid Build Coastguard Worker COM_TRY_BEGIN
75*f6dc9357SAndroid Build Coastguard Worker NWindows::NCOM::CPropVariant prop;
76*f6dc9357SAndroid Build Coastguard Worker switch (propID)
77*f6dc9357SAndroid Build Coastguard Worker {
78*f6dc9357SAndroid Build Coastguard Worker case kpidPhySize: prop = _archive.PhySize; break;
79*f6dc9357SAndroid Build Coastguard Worker
80*f6dc9357SAndroid Build Coastguard Worker case kpidUnpackVer:
81*f6dc9357SAndroid Build Coastguard Worker {
82*f6dc9357SAndroid Build Coastguard Worker if (_archive.LogVols.Size() == 1)
83*f6dc9357SAndroid Build Coastguard Worker {
84*f6dc9357SAndroid Build Coastguard Worker UString s;
85*f6dc9357SAndroid Build Coastguard Worker const CLogVol &vol = _archive.LogVols[0];
86*f6dc9357SAndroid Build Coastguard Worker vol.DomainId.AddUdfVersionTo(s);
87*f6dc9357SAndroid Build Coastguard Worker if (!s.IsEmpty())
88*f6dc9357SAndroid Build Coastguard Worker prop = s;
89*f6dc9357SAndroid Build Coastguard Worker }
90*f6dc9357SAndroid Build Coastguard Worker break;
91*f6dc9357SAndroid Build Coastguard Worker }
92*f6dc9357SAndroid Build Coastguard Worker case kpidComment:
93*f6dc9357SAndroid Build Coastguard Worker {
94*f6dc9357SAndroid Build Coastguard Worker UString comment = _archive.GetComment();
95*f6dc9357SAndroid Build Coastguard Worker if (!comment.IsEmpty())
96*f6dc9357SAndroid Build Coastguard Worker prop = comment;
97*f6dc9357SAndroid Build Coastguard Worker break;
98*f6dc9357SAndroid Build Coastguard Worker }
99*f6dc9357SAndroid Build Coastguard Worker
100*f6dc9357SAndroid Build Coastguard Worker case kpidClusterSize:
101*f6dc9357SAndroid Build Coastguard Worker if (_archive.LogVols.Size() > 0)
102*f6dc9357SAndroid Build Coastguard Worker {
103*f6dc9357SAndroid Build Coastguard Worker UInt32 blockSize = _archive.LogVols[0].BlockSize;
104*f6dc9357SAndroid Build Coastguard Worker unsigned i;
105*f6dc9357SAndroid Build Coastguard Worker for (i = 1; i < _archive.LogVols.Size(); i++)
106*f6dc9357SAndroid Build Coastguard Worker if (_archive.LogVols[i].BlockSize != blockSize)
107*f6dc9357SAndroid Build Coastguard Worker break;
108*f6dc9357SAndroid Build Coastguard Worker if (i == _archive.LogVols.Size())
109*f6dc9357SAndroid Build Coastguard Worker prop = blockSize;
110*f6dc9357SAndroid Build Coastguard Worker }
111*f6dc9357SAndroid Build Coastguard Worker break;
112*f6dc9357SAndroid Build Coastguard Worker
113*f6dc9357SAndroid Build Coastguard Worker case kpidSectorSize: prop = ((UInt32)1 << _archive.SecLogSize); break;
114*f6dc9357SAndroid Build Coastguard Worker
115*f6dc9357SAndroid Build Coastguard Worker case kpidCTime:
116*f6dc9357SAndroid Build Coastguard Worker if (_archive.LogVols.Size() == 1)
117*f6dc9357SAndroid Build Coastguard Worker {
118*f6dc9357SAndroid Build Coastguard Worker const CLogVol &vol = _archive.LogVols[0];
119*f6dc9357SAndroid Build Coastguard Worker if (vol.FileSets.Size() >= 1)
120*f6dc9357SAndroid Build Coastguard Worker UdfTimeToFileTime(vol.FileSets[0].RecordingTime, prop);
121*f6dc9357SAndroid Build Coastguard Worker }
122*f6dc9357SAndroid Build Coastguard Worker break;
123*f6dc9357SAndroid Build Coastguard Worker case kpidMTime:
124*f6dc9357SAndroid Build Coastguard Worker if (_archive.PrimeVols.Size() == 1)
125*f6dc9357SAndroid Build Coastguard Worker {
126*f6dc9357SAndroid Build Coastguard Worker const CPrimeVol &pv = _archive.PrimeVols[0];
127*f6dc9357SAndroid Build Coastguard Worker UdfTimeToFileTime(pv.RecordingTime, prop);
128*f6dc9357SAndroid Build Coastguard Worker }
129*f6dc9357SAndroid Build Coastguard Worker break;
130*f6dc9357SAndroid Build Coastguard Worker
131*f6dc9357SAndroid Build Coastguard Worker case kpidErrorFlags:
132*f6dc9357SAndroid Build Coastguard Worker {
133*f6dc9357SAndroid Build Coastguard Worker UInt32 v = 0;
134*f6dc9357SAndroid Build Coastguard Worker if (!_archive.IsArc) v |= kpv_ErrorFlags_IsNotArc;
135*f6dc9357SAndroid Build Coastguard Worker if (_archive.Unsupported) v |= kpv_ErrorFlags_UnsupportedFeature;
136*f6dc9357SAndroid Build Coastguard Worker if (_archive.UnexpectedEnd) v |= kpv_ErrorFlags_UnexpectedEnd;
137*f6dc9357SAndroid Build Coastguard Worker if (_archive.NoEndAnchor) v |= kpv_ErrorFlags_HeadersError;
138*f6dc9357SAndroid Build Coastguard Worker prop = v;
139*f6dc9357SAndroid Build Coastguard Worker break;
140*f6dc9357SAndroid Build Coastguard Worker }
141*f6dc9357SAndroid Build Coastguard Worker }
142*f6dc9357SAndroid Build Coastguard Worker prop.Detach(value);
143*f6dc9357SAndroid Build Coastguard Worker return S_OK;
144*f6dc9357SAndroid Build Coastguard Worker COM_TRY_END
145*f6dc9357SAndroid Build Coastguard Worker }
146*f6dc9357SAndroid Build Coastguard Worker
147*f6dc9357SAndroid Build Coastguard Worker class CProgressImp Z7_final: public CProgressVirt
148*f6dc9357SAndroid Build Coastguard Worker {
149*f6dc9357SAndroid Build Coastguard Worker CMyComPtr<IArchiveOpenCallback> _callback;
150*f6dc9357SAndroid Build Coastguard Worker UInt64 _numFiles;
151*f6dc9357SAndroid Build Coastguard Worker UInt64 _numBytes;
152*f6dc9357SAndroid Build Coastguard Worker public:
153*f6dc9357SAndroid Build Coastguard Worker HRESULT SetTotal(UInt64 numBytes) Z7_override;
154*f6dc9357SAndroid Build Coastguard Worker HRESULT SetCompleted(UInt64 numFiles, UInt64 numBytes) Z7_override;
155*f6dc9357SAndroid Build Coastguard Worker HRESULT SetCompleted() Z7_override;
CProgressImp(IArchiveOpenCallback * callback)156*f6dc9357SAndroid Build Coastguard Worker CProgressImp(IArchiveOpenCallback *callback): _callback(callback), _numFiles(0), _numBytes(0) {}
157*f6dc9357SAndroid Build Coastguard Worker };
158*f6dc9357SAndroid Build Coastguard Worker
SetTotal(UInt64 numBytes)159*f6dc9357SAndroid Build Coastguard Worker HRESULT CProgressImp::SetTotal(UInt64 numBytes)
160*f6dc9357SAndroid Build Coastguard Worker {
161*f6dc9357SAndroid Build Coastguard Worker if (_callback)
162*f6dc9357SAndroid Build Coastguard Worker return _callback->SetTotal(NULL, &numBytes);
163*f6dc9357SAndroid Build Coastguard Worker return S_OK;
164*f6dc9357SAndroid Build Coastguard Worker }
165*f6dc9357SAndroid Build Coastguard Worker
SetCompleted(UInt64 numFiles,UInt64 numBytes)166*f6dc9357SAndroid Build Coastguard Worker HRESULT CProgressImp::SetCompleted(UInt64 numFiles, UInt64 numBytes)
167*f6dc9357SAndroid Build Coastguard Worker {
168*f6dc9357SAndroid Build Coastguard Worker _numFiles = numFiles;
169*f6dc9357SAndroid Build Coastguard Worker _numBytes = numBytes;
170*f6dc9357SAndroid Build Coastguard Worker return SetCompleted();
171*f6dc9357SAndroid Build Coastguard Worker }
172*f6dc9357SAndroid Build Coastguard Worker
SetCompleted()173*f6dc9357SAndroid Build Coastguard Worker HRESULT CProgressImp::SetCompleted()
174*f6dc9357SAndroid Build Coastguard Worker {
175*f6dc9357SAndroid Build Coastguard Worker if (_callback)
176*f6dc9357SAndroid Build Coastguard Worker return _callback->SetCompleted(&_numFiles, &_numBytes);
177*f6dc9357SAndroid Build Coastguard Worker return S_OK;
178*f6dc9357SAndroid Build Coastguard Worker }
179*f6dc9357SAndroid Build Coastguard Worker
Z7_COM7F_IMF(CHandler::Open (IInStream * stream,const UInt64 *,IArchiveOpenCallback * callback))180*f6dc9357SAndroid Build Coastguard Worker Z7_COM7F_IMF(CHandler::Open(IInStream *stream, const UInt64 *, IArchiveOpenCallback *callback))
181*f6dc9357SAndroid Build Coastguard Worker {
182*f6dc9357SAndroid Build Coastguard Worker COM_TRY_BEGIN
183*f6dc9357SAndroid Build Coastguard Worker {
184*f6dc9357SAndroid Build Coastguard Worker Close();
185*f6dc9357SAndroid Build Coastguard Worker CProgressImp progressImp(callback);
186*f6dc9357SAndroid Build Coastguard Worker RINOK(_archive.Open(stream, &progressImp))
187*f6dc9357SAndroid Build Coastguard Worker bool showVolName = (_archive.LogVols.Size() > 1);
188*f6dc9357SAndroid Build Coastguard Worker FOR_VECTOR (volIndex, _archive.LogVols)
189*f6dc9357SAndroid Build Coastguard Worker {
190*f6dc9357SAndroid Build Coastguard Worker const CLogVol &vol = _archive.LogVols[volIndex];
191*f6dc9357SAndroid Build Coastguard Worker bool showFileSetName = (vol.FileSets.Size() > 1);
192*f6dc9357SAndroid Build Coastguard Worker // showFileSetName = true; // for debug
193*f6dc9357SAndroid Build Coastguard Worker FOR_VECTOR (fsIndex, vol.FileSets)
194*f6dc9357SAndroid Build Coastguard Worker {
195*f6dc9357SAndroid Build Coastguard Worker const CFileSet &fs = vol.FileSets[fsIndex];
196*f6dc9357SAndroid Build Coastguard Worker for (unsigned i = ((showVolName || showFileSetName) ? 0 : 1); i < fs.Refs.Size(); i++)
197*f6dc9357SAndroid Build Coastguard Worker {
198*f6dc9357SAndroid Build Coastguard Worker CRef2 ref2;
199*f6dc9357SAndroid Build Coastguard Worker ref2.Vol = volIndex;
200*f6dc9357SAndroid Build Coastguard Worker ref2.Fs = fsIndex;
201*f6dc9357SAndroid Build Coastguard Worker ref2.Ref = i;
202*f6dc9357SAndroid Build Coastguard Worker _refs2.Add(ref2);
203*f6dc9357SAndroid Build Coastguard Worker }
204*f6dc9357SAndroid Build Coastguard Worker }
205*f6dc9357SAndroid Build Coastguard Worker }
206*f6dc9357SAndroid Build Coastguard Worker _inStream = stream;
207*f6dc9357SAndroid Build Coastguard Worker }
208*f6dc9357SAndroid Build Coastguard Worker return S_OK;
209*f6dc9357SAndroid Build Coastguard Worker COM_TRY_END
210*f6dc9357SAndroid Build Coastguard Worker }
211*f6dc9357SAndroid Build Coastguard Worker
Z7_COM7F_IMF(CHandler::Close ())212*f6dc9357SAndroid Build Coastguard Worker Z7_COM7F_IMF(CHandler::Close())
213*f6dc9357SAndroid Build Coastguard Worker {
214*f6dc9357SAndroid Build Coastguard Worker _inStream.Release();
215*f6dc9357SAndroid Build Coastguard Worker _archive.Clear();
216*f6dc9357SAndroid Build Coastguard Worker _refs2.Clear();
217*f6dc9357SAndroid Build Coastguard Worker return S_OK;
218*f6dc9357SAndroid Build Coastguard Worker }
219*f6dc9357SAndroid Build Coastguard Worker
Z7_COM7F_IMF(CHandler::GetNumberOfItems (UInt32 * numItems))220*f6dc9357SAndroid Build Coastguard Worker Z7_COM7F_IMF(CHandler::GetNumberOfItems(UInt32 *numItems))
221*f6dc9357SAndroid Build Coastguard Worker {
222*f6dc9357SAndroid Build Coastguard Worker *numItems = _refs2.Size();
223*f6dc9357SAndroid Build Coastguard Worker return S_OK;
224*f6dc9357SAndroid Build Coastguard Worker }
225*f6dc9357SAndroid Build Coastguard Worker
Z7_COM7F_IMF(CHandler::GetProperty (UInt32 index,PROPID propID,PROPVARIANT * value))226*f6dc9357SAndroid Build Coastguard Worker Z7_COM7F_IMF(CHandler::GetProperty(UInt32 index, PROPID propID, PROPVARIANT *value))
227*f6dc9357SAndroid Build Coastguard Worker {
228*f6dc9357SAndroid Build Coastguard Worker COM_TRY_BEGIN
229*f6dc9357SAndroid Build Coastguard Worker NWindows::NCOM::CPropVariant prop;
230*f6dc9357SAndroid Build Coastguard Worker {
231*f6dc9357SAndroid Build Coastguard Worker const CRef2 &ref2 = _refs2[index];
232*f6dc9357SAndroid Build Coastguard Worker const CLogVol &vol = _archive.LogVols[ref2.Vol];
233*f6dc9357SAndroid Build Coastguard Worker const CRef &ref = vol.FileSets[ref2.Fs].Refs[ref2.Ref];
234*f6dc9357SAndroid Build Coastguard Worker const CFile &file = _archive.Files[ref.FileIndex];
235*f6dc9357SAndroid Build Coastguard Worker const CItem &item = _archive.Items[file.ItemIndex];
236*f6dc9357SAndroid Build Coastguard Worker switch (propID)
237*f6dc9357SAndroid Build Coastguard Worker {
238*f6dc9357SAndroid Build Coastguard Worker case kpidPath: prop = _archive.GetItemPath(ref2.Vol, ref2.Fs, ref2.Ref,
239*f6dc9357SAndroid Build Coastguard Worker _archive.LogVols.Size() > 1, vol.FileSets.Size() > 1); break;
240*f6dc9357SAndroid Build Coastguard Worker case kpidIsDir: prop = item.IsDir(); break;
241*f6dc9357SAndroid Build Coastguard Worker case kpidSize: if (!item.IsDir()) prop = (UInt64)item.Size; break;
242*f6dc9357SAndroid Build Coastguard Worker case kpidPackSize: if (!item.IsDir()) prop = (UInt64)item.NumLogBlockRecorded * vol.BlockSize; break;
243*f6dc9357SAndroid Build Coastguard Worker case kpidMTime: UdfTimeToFileTime(item.MTime, prop); break;
244*f6dc9357SAndroid Build Coastguard Worker case kpidATime: UdfTimeToFileTime(item.ATime, prop); break;
245*f6dc9357SAndroid Build Coastguard Worker case kpidCTime:
246*f6dc9357SAndroid Build Coastguard Worker if (item.IsExtended)
247*f6dc9357SAndroid Build Coastguard Worker UdfTimeToFileTime(item.CreateTime, prop);
248*f6dc9357SAndroid Build Coastguard Worker break;
249*f6dc9357SAndroid Build Coastguard Worker case kpidChangeTime: UdfTimeToFileTime(item.AttribTime, prop); break;
250*f6dc9357SAndroid Build Coastguard Worker // case kpidUserId: prop = item.Uid; break;
251*f6dc9357SAndroid Build Coastguard Worker // case kpidGroupId: prop = item.Gid; break;
252*f6dc9357SAndroid Build Coastguard Worker // case kpidPosixAttrib: prop = (UInt32)item.Permissions; break;
253*f6dc9357SAndroid Build Coastguard Worker case kpidLinks: prop = (UInt32)item.FileLinkCount; break;
254*f6dc9357SAndroid Build Coastguard Worker }
255*f6dc9357SAndroid Build Coastguard Worker }
256*f6dc9357SAndroid Build Coastguard Worker prop.Detach(value);
257*f6dc9357SAndroid Build Coastguard Worker return S_OK;
258*f6dc9357SAndroid Build Coastguard Worker COM_TRY_END
259*f6dc9357SAndroid Build Coastguard Worker }
260*f6dc9357SAndroid Build Coastguard Worker
Z7_COM7F_IMF(CHandler::GetStream (UInt32 index,ISequentialInStream ** stream))261*f6dc9357SAndroid Build Coastguard Worker Z7_COM7F_IMF(CHandler::GetStream(UInt32 index, ISequentialInStream **stream))
262*f6dc9357SAndroid Build Coastguard Worker {
263*f6dc9357SAndroid Build Coastguard Worker *stream = NULL;
264*f6dc9357SAndroid Build Coastguard Worker
265*f6dc9357SAndroid Build Coastguard Worker const CRef2 &ref2 = _refs2[index];
266*f6dc9357SAndroid Build Coastguard Worker const CLogVol &vol = _archive.LogVols[ref2.Vol];
267*f6dc9357SAndroid Build Coastguard Worker const CRef &ref = vol.FileSets[ref2.Fs].Refs[ref2.Ref];
268*f6dc9357SAndroid Build Coastguard Worker const CFile &file = _archive.Files[ref.FileIndex];
269*f6dc9357SAndroid Build Coastguard Worker const CItem &item = _archive.Items[file.ItemIndex];
270*f6dc9357SAndroid Build Coastguard Worker UInt64 size = item.Size;
271*f6dc9357SAndroid Build Coastguard Worker
272*f6dc9357SAndroid Build Coastguard Worker if (!item.IsRecAndAlloc() || !item.CheckChunkSizes() || ! _archive.CheckItemExtents(ref2.Vol, item))
273*f6dc9357SAndroid Build Coastguard Worker return E_NOTIMPL;
274*f6dc9357SAndroid Build Coastguard Worker
275*f6dc9357SAndroid Build Coastguard Worker if (item.IsInline)
276*f6dc9357SAndroid Build Coastguard Worker {
277*f6dc9357SAndroid Build Coastguard Worker Create_BufInStream_WithNewBuffer(item.InlineData, stream);
278*f6dc9357SAndroid Build Coastguard Worker return S_OK;
279*f6dc9357SAndroid Build Coastguard Worker }
280*f6dc9357SAndroid Build Coastguard Worker
281*f6dc9357SAndroid Build Coastguard Worker CExtentsStream *extentStreamSpec = new CExtentsStream();
282*f6dc9357SAndroid Build Coastguard Worker CMyComPtr<ISequentialInStream> extentStream = extentStreamSpec;
283*f6dc9357SAndroid Build Coastguard Worker
284*f6dc9357SAndroid Build Coastguard Worker extentStreamSpec->Stream = _inStream;
285*f6dc9357SAndroid Build Coastguard Worker
286*f6dc9357SAndroid Build Coastguard Worker UInt64 virtOffset = 0;
287*f6dc9357SAndroid Build Coastguard Worker FOR_VECTOR (extentIndex, item.Extents)
288*f6dc9357SAndroid Build Coastguard Worker {
289*f6dc9357SAndroid Build Coastguard Worker const CMyExtent &extent = item.Extents[extentIndex];
290*f6dc9357SAndroid Build Coastguard Worker UInt32 len = extent.GetLen();
291*f6dc9357SAndroid Build Coastguard Worker if (len == 0)
292*f6dc9357SAndroid Build Coastguard Worker continue;
293*f6dc9357SAndroid Build Coastguard Worker if (size < len)
294*f6dc9357SAndroid Build Coastguard Worker return S_FALSE;
295*f6dc9357SAndroid Build Coastguard Worker
296*f6dc9357SAndroid Build Coastguard Worker const unsigned partitionIndex = vol.PartitionMaps[extent.PartitionRef].PartitionIndex;
297*f6dc9357SAndroid Build Coastguard Worker UInt32 logBlockNumber = extent.Pos;
298*f6dc9357SAndroid Build Coastguard Worker const CPartition &partition = _archive.Partitions[partitionIndex];
299*f6dc9357SAndroid Build Coastguard Worker UInt64 offset = ((UInt64)partition.Pos << _archive.SecLogSize) +
300*f6dc9357SAndroid Build Coastguard Worker (UInt64)logBlockNumber * vol.BlockSize;
301*f6dc9357SAndroid Build Coastguard Worker
302*f6dc9357SAndroid Build Coastguard Worker CSeekExtent se;
303*f6dc9357SAndroid Build Coastguard Worker se.Phy = offset;
304*f6dc9357SAndroid Build Coastguard Worker se.Virt = virtOffset;
305*f6dc9357SAndroid Build Coastguard Worker virtOffset += len;
306*f6dc9357SAndroid Build Coastguard Worker extentStreamSpec->Extents.Add(se);
307*f6dc9357SAndroid Build Coastguard Worker
308*f6dc9357SAndroid Build Coastguard Worker size -= len;
309*f6dc9357SAndroid Build Coastguard Worker }
310*f6dc9357SAndroid Build Coastguard Worker if (size != 0)
311*f6dc9357SAndroid Build Coastguard Worker return S_FALSE;
312*f6dc9357SAndroid Build Coastguard Worker CSeekExtent se;
313*f6dc9357SAndroid Build Coastguard Worker se.Phy = 0;
314*f6dc9357SAndroid Build Coastguard Worker se.Virt = virtOffset;
315*f6dc9357SAndroid Build Coastguard Worker extentStreamSpec->Extents.Add(se);
316*f6dc9357SAndroid Build Coastguard Worker extentStreamSpec->Init();
317*f6dc9357SAndroid Build Coastguard Worker *stream = extentStream.Detach();
318*f6dc9357SAndroid Build Coastguard Worker return S_OK;
319*f6dc9357SAndroid Build Coastguard Worker }
320*f6dc9357SAndroid Build Coastguard Worker
Z7_COM7F_IMF(CHandler::Extract (const UInt32 * indices,UInt32 numItems,Int32 testMode,IArchiveExtractCallback * extractCallback))321*f6dc9357SAndroid Build Coastguard Worker Z7_COM7F_IMF(CHandler::Extract(const UInt32 *indices, UInt32 numItems,
322*f6dc9357SAndroid Build Coastguard Worker Int32 testMode, IArchiveExtractCallback *extractCallback))
323*f6dc9357SAndroid Build Coastguard Worker {
324*f6dc9357SAndroid Build Coastguard Worker COM_TRY_BEGIN
325*f6dc9357SAndroid Build Coastguard Worker const bool allFilesMode = (numItems == (UInt32)(Int32)-1);
326*f6dc9357SAndroid Build Coastguard Worker if (allFilesMode)
327*f6dc9357SAndroid Build Coastguard Worker numItems = _refs2.Size();
328*f6dc9357SAndroid Build Coastguard Worker if (numItems == 0)
329*f6dc9357SAndroid Build Coastguard Worker return S_OK;
330*f6dc9357SAndroid Build Coastguard Worker UInt64 totalSize = 0;
331*f6dc9357SAndroid Build Coastguard Worker UInt32 i;
332*f6dc9357SAndroid Build Coastguard Worker
333*f6dc9357SAndroid Build Coastguard Worker for (i = 0; i < numItems; i++)
334*f6dc9357SAndroid Build Coastguard Worker {
335*f6dc9357SAndroid Build Coastguard Worker const UInt32 index = (allFilesMode ? i : indices[i]);
336*f6dc9357SAndroid Build Coastguard Worker const CRef2 &ref2 = _refs2[index];
337*f6dc9357SAndroid Build Coastguard Worker const CRef &ref = _archive.LogVols[ref2.Vol].FileSets[ref2.Fs].Refs[ref2.Ref];
338*f6dc9357SAndroid Build Coastguard Worker const CFile &file = _archive.Files[ref.FileIndex];
339*f6dc9357SAndroid Build Coastguard Worker const CItem &item = _archive.Items[file.ItemIndex];
340*f6dc9357SAndroid Build Coastguard Worker if (!item.IsDir())
341*f6dc9357SAndroid Build Coastguard Worker totalSize += item.Size;
342*f6dc9357SAndroid Build Coastguard Worker }
343*f6dc9357SAndroid Build Coastguard Worker RINOK(extractCallback->SetTotal(totalSize))
344*f6dc9357SAndroid Build Coastguard Worker
345*f6dc9357SAndroid Build Coastguard Worker UInt64 currentTotalSize = 0;
346*f6dc9357SAndroid Build Coastguard Worker
347*f6dc9357SAndroid Build Coastguard Worker CMyComPtr2_Create<ICompressProgressInfo, CLocalProgress> lps;
348*f6dc9357SAndroid Build Coastguard Worker lps->Init(extractCallback, false);
349*f6dc9357SAndroid Build Coastguard Worker CMyComPtr2_Create<ICompressCoder, NCompress::CCopyCoder> copyCoder;
350*f6dc9357SAndroid Build Coastguard Worker CMyComPtr2_Create<ISequentialOutStream, CLimitedSequentialOutStream> outStream;
351*f6dc9357SAndroid Build Coastguard Worker
352*f6dc9357SAndroid Build Coastguard Worker for (i = 0;; i++)
353*f6dc9357SAndroid Build Coastguard Worker {
354*f6dc9357SAndroid Build Coastguard Worker lps->InSize = lps->OutSize = currentTotalSize;
355*f6dc9357SAndroid Build Coastguard Worker RINOK(lps->SetCur())
356*f6dc9357SAndroid Build Coastguard Worker if (i >= numItems)
357*f6dc9357SAndroid Build Coastguard Worker break;
358*f6dc9357SAndroid Build Coastguard Worker CMyComPtr<ISequentialOutStream> realOutStream;
359*f6dc9357SAndroid Build Coastguard Worker const Int32 askMode = testMode ?
360*f6dc9357SAndroid Build Coastguard Worker NExtract::NAskMode::kTest :
361*f6dc9357SAndroid Build Coastguard Worker NExtract::NAskMode::kExtract;
362*f6dc9357SAndroid Build Coastguard Worker const UInt32 index = allFilesMode ? i : indices[i];
363*f6dc9357SAndroid Build Coastguard Worker
364*f6dc9357SAndroid Build Coastguard Worker RINOK(extractCallback->GetStream(index, &realOutStream, askMode))
365*f6dc9357SAndroid Build Coastguard Worker
366*f6dc9357SAndroid Build Coastguard Worker const CRef2 &ref2 = _refs2[index];
367*f6dc9357SAndroid Build Coastguard Worker const CRef &ref = _archive.LogVols[ref2.Vol].FileSets[ref2.Fs].Refs[ref2.Ref];
368*f6dc9357SAndroid Build Coastguard Worker const CFile &file = _archive.Files[ref.FileIndex];
369*f6dc9357SAndroid Build Coastguard Worker const CItem &item = _archive.Items[file.ItemIndex];
370*f6dc9357SAndroid Build Coastguard Worker
371*f6dc9357SAndroid Build Coastguard Worker if (item.IsDir())
372*f6dc9357SAndroid Build Coastguard Worker {
373*f6dc9357SAndroid Build Coastguard Worker RINOK(extractCallback->PrepareOperation(askMode))
374*f6dc9357SAndroid Build Coastguard Worker RINOK(extractCallback->SetOperationResult(NExtract::NOperationResult::kOK))
375*f6dc9357SAndroid Build Coastguard Worker continue;
376*f6dc9357SAndroid Build Coastguard Worker }
377*f6dc9357SAndroid Build Coastguard Worker currentTotalSize += item.Size;
378*f6dc9357SAndroid Build Coastguard Worker
379*f6dc9357SAndroid Build Coastguard Worker if (!testMode && !realOutStream)
380*f6dc9357SAndroid Build Coastguard Worker continue;
381*f6dc9357SAndroid Build Coastguard Worker
382*f6dc9357SAndroid Build Coastguard Worker RINOK(extractCallback->PrepareOperation(askMode))
383*f6dc9357SAndroid Build Coastguard Worker outStream->SetStream(realOutStream);
384*f6dc9357SAndroid Build Coastguard Worker realOutStream.Release();
385*f6dc9357SAndroid Build Coastguard Worker outStream->Init(item.Size);
386*f6dc9357SAndroid Build Coastguard Worker Int32 opRes;
387*f6dc9357SAndroid Build Coastguard Worker {
388*f6dc9357SAndroid Build Coastguard Worker CMyComPtr<ISequentialInStream> udfInStream;
389*f6dc9357SAndroid Build Coastguard Worker const HRESULT res = GetStream(index, &udfInStream);
390*f6dc9357SAndroid Build Coastguard Worker if (res == E_NOTIMPL)
391*f6dc9357SAndroid Build Coastguard Worker opRes = NExtract::NOperationResult::kUnsupportedMethod;
392*f6dc9357SAndroid Build Coastguard Worker else if (res != S_OK)
393*f6dc9357SAndroid Build Coastguard Worker opRes = NExtract::NOperationResult::kDataError;
394*f6dc9357SAndroid Build Coastguard Worker else
395*f6dc9357SAndroid Build Coastguard Worker {
396*f6dc9357SAndroid Build Coastguard Worker RINOK(copyCoder.Interface()->Code(udfInStream, outStream, NULL, NULL, lps))
397*f6dc9357SAndroid Build Coastguard Worker opRes = outStream->IsFinishedOK() ?
398*f6dc9357SAndroid Build Coastguard Worker NExtract::NOperationResult::kOK:
399*f6dc9357SAndroid Build Coastguard Worker NExtract::NOperationResult::kDataError;
400*f6dc9357SAndroid Build Coastguard Worker }
401*f6dc9357SAndroid Build Coastguard Worker }
402*f6dc9357SAndroid Build Coastguard Worker outStream->ReleaseStream();
403*f6dc9357SAndroid Build Coastguard Worker RINOK(extractCallback->SetOperationResult(opRes))
404*f6dc9357SAndroid Build Coastguard Worker }
405*f6dc9357SAndroid Build Coastguard Worker return S_OK;
406*f6dc9357SAndroid Build Coastguard Worker COM_TRY_END
407*f6dc9357SAndroid Build Coastguard Worker }
408*f6dc9357SAndroid Build Coastguard Worker
409*f6dc9357SAndroid Build Coastguard Worker static const UInt32 kIsoStartPos = 0x8000;
410*f6dc9357SAndroid Build Coastguard Worker
411*f6dc9357SAndroid Build Coastguard Worker // 5, { 0, 'N', 'S', 'R', '0' },
412*f6dc9357SAndroid Build Coastguard Worker
413*f6dc9357SAndroid Build Coastguard Worker static const Byte k_Signature[] =
414*f6dc9357SAndroid Build Coastguard Worker {
415*f6dc9357SAndroid Build Coastguard Worker 8, 0, 'B', 'E', 'A', '0', '1', 1, 0,
416*f6dc9357SAndroid Build Coastguard Worker 6, 1, 'C', 'D', '0', '0', '1'
417*f6dc9357SAndroid Build Coastguard Worker };
418*f6dc9357SAndroid Build Coastguard Worker
419*f6dc9357SAndroid Build Coastguard Worker REGISTER_ARC_I(
420*f6dc9357SAndroid Build Coastguard Worker "Udf", "udf iso img", NULL, 0xE0,
421*f6dc9357SAndroid Build Coastguard Worker k_Signature,
422*f6dc9357SAndroid Build Coastguard Worker kIsoStartPos,
423*f6dc9357SAndroid Build Coastguard Worker NArcInfoFlags::kMultiSignature |
424*f6dc9357SAndroid Build Coastguard Worker NArcInfoFlags::kStartOpen,
425*f6dc9357SAndroid Build Coastguard Worker IsArc_Udf)
426*f6dc9357SAndroid Build Coastguard Worker
427*f6dc9357SAndroid Build Coastguard Worker }}
428