Mercurial > vba-linux

annotate src/win32/7zip/7z/CPP/7zip/Archive/GZip/GZipHandler.cpp @ 1:f9f4f1b99eed

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
importing src directory
author Robert McIntyre <rlm@mit.edu>
date Sat, 03 Mar 2012 10:31:27 -0600
parents
children
rev   line source
rlm@1 1 // GZipHandler.cpp
rlm@1 2
rlm@1 3 #include "StdAfx.h"
rlm@1 4
rlm@1 5 #include "GZipHandler.h"
rlm@1 6
rlm@1 7 #include "Common/Defs.h"
rlm@1 8 #include "Common/StringConvert.h"
rlm@1 9 #include "Common/ComTry.h"
rlm@1 10 #include "Windows/PropVariant.h"
rlm@1 11 #include "Windows/Time.h"
rlm@1 12
rlm@1 13 #include "../../ICoder.h"
rlm@1 14 #include "../../Common/ProgressUtils.h"
rlm@1 15 #include "../../Common/CreateCoder.h"
rlm@1 16 #include "../Common/OutStreamWithCRC.h"
rlm@1 17
rlm@1 18 using namespace NWindows;
rlm@1 19
rlm@1 20 namespace NArchive {
rlm@1 21 namespace NGZip {
rlm@1 22
rlm@1 23 static const CMethodId kMethodId_Deflate = 0x040108;
rlm@1 24
rlm@1 25 const wchar_t *kHostOS[] =
rlm@1 26 {
rlm@1 27 L"FAT",
rlm@1 28 L"AMIGA",
rlm@1 29 L"VMS",
rlm@1 30 L"Unix",
rlm@1 31 L"VM_CMS",
rlm@1 32 L"Atari", // what if it's a minix filesystem? [cjh]
rlm@1 33 L"HPFS", // filesystem used by OS/2 (and NT 3.x)
rlm@1 34 L"Mac",
rlm@1 35 L"Z_System",
rlm@1 36 L"CPM",
rlm@1 37 L"TOPS20", // pkzip 2.50 NTFS
rlm@1 38 L"NTFS", // filesystem used by Windows NT
rlm@1 39 L"QDOS ", // SMS/QDOS
rlm@1 40 L"Acorn", // Archimedes Acorn RISC OS
rlm@1 41 L"VFAT", // filesystem used by Windows 95, NT
rlm@1 42 L"MVS",
rlm@1 43 L"BeOS", // hybrid POSIX/database filesystem
rlm@1 44 // BeBOX or PowerMac
rlm@1 45 L"Tandem",
rlm@1 46 L"THEOS"
rlm@1 47 };
rlm@1 48
rlm@1 49 static const int kNumHostOSes = sizeof(kHostOS) / sizeof(kHostOS[0]);
rlm@1 50
rlm@1 51 static const wchar_t *kUnknownOS = L"Unknown";
rlm@1 52
rlm@1 53 /*
rlm@1 54 enum // PropID
rlm@1 55 {
rlm@1 56 kpidExtraIsPresent = kpidUserDefined,
rlm@1 57 kpidExtraFlags,
rlm@1 58 kpidIsText
rlm@1 59 };
rlm@1 60 */
rlm@1 61
rlm@1 62 STATPROPSTG kProps[] =
rlm@1 63 {
rlm@1 64 { NULL, kpidPath, VT_BSTR},
rlm@1 65 { NULL, kpidSize, VT_UI8},
rlm@1 66 { NULL, kpidPackSize, VT_UI8},
rlm@1 67 { NULL, kpidMTime, VT_FILETIME},
rlm@1 68 // { NULL, kpidMethod, VT_UI1},
rlm@1 69 { NULL, kpidHostOS, VT_BSTR},
rlm@1 70 { NULL, kpidCRC, VT_UI4}
rlm@1 71 // { L"Extra", kpidExtraIsPresent, VT_BOOL}
rlm@1 72 // { L"Extra flags", kpidExtraFlags, VT_UI1},
rlm@1 73 // { L"Is Text", kpidIsText, VT_BOOL},
rlm@1 74 };
rlm@1 75
rlm@1 76 IMP_IInArchive_Props
rlm@1 77 IMP_IInArchive_ArcProps_NO
rlm@1 78
rlm@1 79 STDMETHODIMP CHandler::GetNumberOfItems(UInt32 *numItems)
rlm@1 80 {
rlm@1 81 *numItems = 1;
rlm@1 82 return S_OK;
rlm@1 83 }
rlm@1 84
rlm@1 85 STDMETHODIMP CHandler::GetProperty(UInt32 /* index */, PROPID propID, PROPVARIANT *value)
rlm@1 86 {
rlm@1 87 COM_TRY_BEGIN
rlm@1 88 NWindows::NCOM::CPropVariant prop;
rlm@1 89 switch(propID)
rlm@1 90 {
rlm@1 91 case kpidPath:
rlm@1 92 if (m_Item.NameIsPresent())
rlm@1 93 prop = MultiByteToUnicodeString(m_Item.Name, CP_ACP);
rlm@1 94 break;
rlm@1 95 case kpidMTime:
rlm@1 96 {
rlm@1 97 FILETIME utcTime;
rlm@1 98 if (m_Item.Time != 0)
rlm@1 99 {
rlm@1 100 NTime::UnixTimeToFileTime((UInt32)m_Item.Time, utcTime);
rlm@1 101 prop = utcTime;
rlm@1 102 }
rlm@1 103 else
rlm@1 104 {
rlm@1 105 // utcTime.dwLowDateTime = utcTime.dwHighDateTime = 0;
rlm@1 106 // prop = utcTime;
rlm@1 107 }
rlm@1 108 break;
rlm@1 109 }
rlm@1 110 case kpidSize: prop = UInt64(m_Item.UnPackSize32); break;
rlm@1 111 case kpidPackSize: prop = m_PackSize; break;
rlm@1 112 case kpidCommented: prop = m_Item.CommentIsPresent(); break;
rlm@1 113 case kpidHostOS:
rlm@1 114 prop = (m_Item.HostOS < kNumHostOSes) ?
rlm@1 115 kHostOS[m_Item.HostOS] : kUnknownOS;
rlm@1 116 break;
rlm@1 117 case kpidMethod: prop = m_Item.CompressionMethod; break;
rlm@1 118 case kpidCRC: prop = m_Item.FileCRC; break;
rlm@1 119 /*
rlm@1 120 case kpidExtraFlags: prop = m_Item.ExtraFlags; break;
rlm@1 121 case kpidIsText: prop = m_Item.IsText(); break;
rlm@1 122 case kpidExtraIsPresent: prop = m_Item.ExtraFieldIsPresent(); break;
rlm@1 123 */
rlm@1 124 }
rlm@1 125 prop.Detach(value);
rlm@1 126 return S_OK;
rlm@1 127 COM_TRY_END
rlm@1 128 }
rlm@1 129
rlm@1 130 STDMETHODIMP CHandler::Open(IInStream *inStream,
rlm@1 131 const UInt64 * /* maxCheckStartPosition */,
rlm@1 132 IArchiveOpenCallback * /* openArchiveCallback */)
rlm@1 133 {
rlm@1 134 COM_TRY_BEGIN
rlm@1 135 try
rlm@1 136 {
rlm@1 137 CInArchive archive;
rlm@1 138 RINOK(inStream->Seek(0, STREAM_SEEK_CUR, &m_StreamStartPosition));
rlm@1 139 RINOK(archive.ReadHeader(inStream, m_Item));
rlm@1 140 m_DataOffset = archive.GetOffset();
rlm@1 141 UInt64 newPosition;
rlm@1 142 RINOK(inStream->Seek(-8, STREAM_SEEK_END, &newPosition));
rlm@1 143 m_PackSize = newPosition - (m_StreamStartPosition + m_DataOffset);
rlm@1 144 if (archive.ReadPostHeader(inStream, m_Item) != S_OK)
rlm@1 145 return S_FALSE;
rlm@1 146 m_Stream = inStream;
rlm@1 147 }
rlm@1 148 catch(...)
rlm@1 149 {
rlm@1 150 return S_FALSE;
rlm@1 151 }
rlm@1 152 return S_OK;
rlm@1 153 COM_TRY_END
rlm@1 154 }
rlm@1 155
rlm@1 156 STDMETHODIMP CHandler::Close()
rlm@1 157 {
rlm@1 158 m_Stream.Release();
rlm@1 159 return S_OK;
rlm@1 160 }
rlm@1 161
rlm@1 162 STDMETHODIMP CHandler::Extract(const UInt32* indices, UInt32 numItems,
rlm@1 163 Int32 _aTestMode, IArchiveExtractCallback *extractCallback)
rlm@1 164 {
rlm@1 165 COM_TRY_BEGIN
rlm@1 166 bool allFilesMode = (numItems == UInt32(-1));
rlm@1 167 if (!allFilesMode)
rlm@1 168 {
rlm@1 169 if (numItems == 0)
rlm@1 170 return S_OK;
rlm@1 171 if (numItems != 1)
rlm@1 172 return E_INVALIDARG;
rlm@1 173 if (indices[0] != 0)
rlm@1 174 return E_INVALIDARG;
rlm@1 175 }
rlm@1 176
rlm@1 177 bool testMode = (_aTestMode != 0);
rlm@1 178
rlm@1 179 extractCallback->SetTotal(m_PackSize);
rlm@1 180
rlm@1 181 UInt64 currentTotalPacked = 0;
rlm@1 182
rlm@1 183 RINOK(extractCallback->SetCompleted(&currentTotalPacked));
rlm@1 184 CMyComPtr<ISequentialOutStream> realOutStream;
rlm@1 185 Int32 askMode;
rlm@1 186 askMode = testMode ? NArchive::NExtract::NAskMode::kTest :
rlm@1 187 NArchive::NExtract::NAskMode::kExtract;
rlm@1 188
rlm@1 189 RINOK(extractCallback->GetStream(0, &realOutStream, askMode));
rlm@1 190
rlm@1 191 if(!testMode && !realOutStream)
rlm@1 192 return S_OK;
rlm@1 193
rlm@1 194 extractCallback->PrepareOperation(askMode);
rlm@1 195
rlm@1 196 COutStreamWithCRC *outStreamSpec = new COutStreamWithCRC;
rlm@1 197 CMyComPtr<ISequentialOutStream> outStream(outStreamSpec);
rlm@1 198 outStreamSpec->SetStream(realOutStream);
rlm@1 199 outStreamSpec->Init();
rlm@1 200 realOutStream.Release();
rlm@1 201
rlm@1 202 CLocalProgress *lps = new CLocalProgress;
rlm@1 203 CMyComPtr<ICompressProgressInfo> progress = lps;
rlm@1 204 lps->Init(extractCallback, true);
rlm@1 205
rlm@1 206 CMyComPtr<ICompressCoder> deflateDecoder;
rlm@1 207 bool firstItem = true;
rlm@1 208 RINOK(m_Stream->Seek(m_StreamStartPosition, STREAM_SEEK_SET, NULL));
rlm@1 209 Int32 opRes;
rlm@1 210 for (;;)
rlm@1 211 {
rlm@1 212 lps->InSize = currentTotalPacked;
rlm@1 213 lps->OutSize = outStreamSpec->GetSize();
rlm@1 214
rlm@1 215 CInArchive archive;
rlm@1 216 CItem item;
rlm@1 217 HRESULT result = archive.ReadHeader(m_Stream, item);
rlm@1 218 if (result != S_OK)
rlm@1 219 {
rlm@1 220 if (firstItem)
rlm@1 221 return E_FAIL;
rlm@1 222 opRes = NArchive::NExtract::NOperationResult::kOK;
rlm@1 223 break;
rlm@1 224 }
rlm@1 225 firstItem = false;
rlm@1 226
rlm@1 227 UInt64 dataStartPos;
rlm@1 228 RINOK(m_Stream->Seek(0, STREAM_SEEK_CUR, &dataStartPos));
rlm@1 229
rlm@1 230 outStreamSpec->InitCRC();
rlm@1 231
rlm@1 232 if (item.CompressionMethod != NFileHeader::NCompressionMethod::kDeflate)
rlm@1 233 {
rlm@1 234 opRes = NArchive::NExtract::NOperationResult::kUnSupportedMethod;
rlm@1 235 break;
rlm@1 236 }
rlm@1 237
rlm@1 238 if (!deflateDecoder)
rlm@1 239 {
rlm@1 240 RINOK(CreateCoder(
rlm@1 241 EXTERNAL_CODECS_VARS
rlm@1 242 kMethodId_Deflate, deflateDecoder, false));
rlm@1 243 if (!deflateDecoder)
rlm@1 244 {
rlm@1 245 opRes = NArchive::NExtract::NOperationResult::kUnSupportedMethod;
rlm@1 246 break;
rlm@1 247 }
rlm@1 248 }
rlm@1 249 result = deflateDecoder->Code(m_Stream, outStream, NULL, NULL, progress);
rlm@1 250 if (result != S_OK)
rlm@1 251 {
rlm@1 252 if (result != S_FALSE)
rlm@1 253 return result;
rlm@1 254 opRes = NArchive::NExtract::NOperationResult::kDataError;
rlm@1 255 break;
rlm@1 256 }
rlm@1 257
rlm@1 258 CMyComPtr<ICompressGetInStreamProcessedSize> getInStreamProcessedSize;
rlm@1 259 RINOK(deflateDecoder.QueryInterface(IID_ICompressGetInStreamProcessedSize,
rlm@1 260 &getInStreamProcessedSize));
rlm@1 261 UInt64 packSize;
rlm@1 262 RINOK(getInStreamProcessedSize->GetInStreamProcessedSize(&packSize));
rlm@1 263 UInt64 pos;
rlm@1 264 RINOK(m_Stream->Seek(dataStartPos + packSize, STREAM_SEEK_SET, &pos));
rlm@1 265
rlm@1 266 currentTotalPacked = pos - m_StreamStartPosition;
rlm@1 267
rlm@1 268 CItem postItem;
rlm@1 269 if (archive.ReadPostHeader(m_Stream, postItem) != S_OK)
rlm@1 270 return E_FAIL;
rlm@1 271 if((outStreamSpec->GetCRC() != postItem.FileCRC))
rlm@1 272 {
rlm@1 273 opRes = NArchive::NExtract::NOperationResult::kCRCError;
rlm@1 274 break;
rlm@1 275 }
rlm@1 276 }
rlm@1 277 outStream.Release();
rlm@1 278 return extractCallback->SetOperationResult(opRes);
rlm@1 279 COM_TRY_END
rlm@1 280 }
rlm@1 281
rlm@1 282 IMPL_ISetCompressCodecsInfo
rlm@1 283
rlm@1 284 }}