xref: /aosp_15_r20/external/lzma/CPP/7zip/Archive/Udf/UdfHandler.cpp (revision f6dc9357d832569d4d1f5d24eacdb3935a1ae8e6)
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