Mercurial > vba-linux

annotate src/win32/7zip/7z/CPP/7zip/Archive/BZip2/BZip2Handler.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 // BZip2Handler.cpp
rlm@1 2
rlm@1 3 #include "StdAfx.h"
rlm@1 4
rlm@1 5 #include "Common/ComTry.h"
rlm@1 6
rlm@1 7 #include "Windows/PropVariant.h"
rlm@1 8
rlm@1 9 #include "../../Common/CreateCoder.h"
rlm@1 10 #include "../../Common/ProgressUtils.h"
rlm@1 11 #include "../../Common/StreamUtils.h"
rlm@1 12
rlm@1 13 #include "../Common/DummyOutStream.h"
rlm@1 14
rlm@1 15 #include "BZip2Handler.h"
rlm@1 16
rlm@1 17 using namespace NWindows;
rlm@1 18
rlm@1 19 namespace NArchive {
rlm@1 20 namespace NBZip2 {
rlm@1 21
rlm@1 22 static const CMethodId kMethodId_BZip2 = 0x040202;
rlm@1 23
rlm@1 24 STATPROPSTG kProps[] =
rlm@1 25 {
rlm@1 26 { NULL, kpidPackSize, VT_UI8}
rlm@1 27 };
rlm@1 28
rlm@1 29 IMP_IInArchive_Props
rlm@1 30 IMP_IInArchive_ArcProps_NO
rlm@1 31
rlm@1 32 STDMETHODIMP CHandler::GetNumberOfItems(UInt32 *numItems)
rlm@1 33 {
rlm@1 34 *numItems = 1;
rlm@1 35 return S_OK;
rlm@1 36 }
rlm@1 37
rlm@1 38 STDMETHODIMP CHandler::GetProperty(UInt32 /* index */, PROPID propID, PROPVARIANT *value)
rlm@1 39 {
rlm@1 40 NWindows::NCOM::CPropVariant prop;
rlm@1 41 switch(propID)
rlm@1 42 {
rlm@1 43 case kpidPackSize: prop = _item.PackSize; break;
rlm@1 44 }
rlm@1 45 prop.Detach(value);
rlm@1 46 return S_OK;
rlm@1 47 }
rlm@1 48
rlm@1 49 STDMETHODIMP CHandler::Open(IInStream *stream,
rlm@1 50 const UInt64 * /* maxCheckStartPosition */,
rlm@1 51 IArchiveOpenCallback * /* openArchiveCallback */)
rlm@1 52 {
rlm@1 53 COM_TRY_BEGIN
rlm@1 54 try
rlm@1 55 {
rlm@1 56 RINOK(stream->Seek(0, STREAM_SEEK_CUR, &_streamStartPosition));
rlm@1 57 const int kSignatureSize = 3;
rlm@1 58 Byte buffer[kSignatureSize];
rlm@1 59 RINOK(ReadStream_FALSE(stream, buffer, kSignatureSize));
rlm@1 60 if (buffer[0] != 'B' || buffer[1] != 'Z' || buffer[2] != 'h')
rlm@1 61 return S_FALSE;
rlm@1 62
rlm@1 63 UInt64 endPosition;
rlm@1 64 RINOK(stream->Seek(0, STREAM_SEEK_END, &endPosition));
rlm@1 65 _item.PackSize = endPosition - _streamStartPosition;
rlm@1 66
rlm@1 67 _stream = stream;
rlm@1 68 }
rlm@1 69 catch(...)
rlm@1 70 {
rlm@1 71 return S_FALSE;
rlm@1 72 }
rlm@1 73 return S_OK;
rlm@1 74 COM_TRY_END
rlm@1 75 }
rlm@1 76
rlm@1 77 STDMETHODIMP CHandler::Close()
rlm@1 78 {
rlm@1 79 _stream.Release();
rlm@1 80 return S_OK;
rlm@1 81 }
rlm@1 82
rlm@1 83
rlm@1 84 STDMETHODIMP CHandler::Extract(const UInt32* indices, UInt32 numItems,
rlm@1 85 Int32 testModeSpec, IArchiveExtractCallback *extractCallback)
rlm@1 86 {
rlm@1 87 COM_TRY_BEGIN
rlm@1 88 bool allFilesMode = (numItems == UInt32(-1));
rlm@1 89 if (!allFilesMode)
rlm@1 90 {
rlm@1 91 if (numItems == 0)
rlm@1 92 return S_OK;
rlm@1 93 if (numItems != 1)
rlm@1 94 return E_INVALIDARG;
rlm@1 95 if (indices[0] != 0)
rlm@1 96 return E_INVALIDARG;
rlm@1 97 }
rlm@1 98
rlm@1 99 bool testMode = (testModeSpec != 0);
rlm@1 100
rlm@1 101 extractCallback->SetTotal(_item.PackSize);
rlm@1 102
rlm@1 103 UInt64 currentTotalPacked = 0;
rlm@1 104
rlm@1 105 RINOK(extractCallback->SetCompleted(&currentTotalPacked));
rlm@1 106
rlm@1 107 CMyComPtr<ISequentialOutStream> realOutStream;
rlm@1 108 Int32 askMode;
rlm@1 109 askMode = testMode ? NExtract::NAskMode::kTest :
rlm@1 110 NExtract::NAskMode::kExtract;
rlm@1 111
rlm@1 112 RINOK(extractCallback->GetStream(0, &realOutStream, askMode));
rlm@1 113
rlm@1 114 if(!testMode && !realOutStream)
rlm@1 115 return S_OK;
rlm@1 116
rlm@1 117
rlm@1 118 extractCallback->PrepareOperation(askMode);
rlm@1 119
rlm@1 120 CMyComPtr<ICompressCoder> decoder;
rlm@1 121 HRESULT loadResult = CreateCoder(
rlm@1 122 EXTERNAL_CODECS_VARS
rlm@1 123 kMethodId_BZip2, decoder, false);
rlm@1 124 if (loadResult != S_OK || !decoder)
rlm@1 125 {
rlm@1 126 RINOK(extractCallback->SetOperationResult(NExtract::NOperationResult::kUnSupportedMethod));
rlm@1 127 return S_OK;
rlm@1 128 }
rlm@1 129
rlm@1 130 #ifdef COMPRESS_MT
rlm@1 131 {
rlm@1 132 CMyComPtr<ICompressSetCoderMt> setCoderMt;
rlm@1 133 decoder.QueryInterface(IID_ICompressSetCoderMt, &setCoderMt);
rlm@1 134 if (setCoderMt)
rlm@1 135 {
rlm@1 136 RINOK(setCoderMt->SetNumberOfThreads(_numThreads));
rlm@1 137 }
rlm@1 138 }
rlm@1 139 #endif
rlm@1 140
rlm@1 141 CDummyOutStream *outStreamSpec = new CDummyOutStream;
rlm@1 142 CMyComPtr<ISequentialOutStream> outStream(outStreamSpec);
rlm@1 143 outStreamSpec->SetStream(realOutStream);
rlm@1 144 outStreamSpec->Init();
rlm@1 145
rlm@1 146 realOutStream.Release();
rlm@1 147
rlm@1 148 CLocalProgress *lps = new CLocalProgress;
rlm@1 149 CMyComPtr<ICompressProgressInfo> progress = lps;
rlm@1 150 lps->Init(extractCallback, true);
rlm@1 151
rlm@1 152 RINOK(_stream->Seek(_streamStartPosition, STREAM_SEEK_SET, NULL));
rlm@1 153
rlm@1 154 HRESULT result = S_OK;
rlm@1 155
rlm@1 156 bool firstItem = true;
rlm@1 157 for (;;)
rlm@1 158 {
rlm@1 159 lps->InSize = currentTotalPacked;
rlm@1 160 lps->OutSize = outStreamSpec->GetSize();
rlm@1 161
rlm@1 162 RINOK(lps->SetCur());
rlm@1 163
rlm@1 164 const int kSignatureSize = 3;
rlm@1 165 Byte buffer[kSignatureSize];
rlm@1 166 size_t processedSize = kSignatureSize;
rlm@1 167 RINOK(ReadStream(_stream, buffer, &processedSize));
rlm@1 168 if (processedSize != kSignatureSize)
rlm@1 169 {
rlm@1 170 if (firstItem)
rlm@1 171 return E_FAIL;
rlm@1 172 break;
rlm@1 173 }
rlm@1 174 if (buffer[0] != 'B' || buffer[1] != 'Z' || buffer[2] != 'h')
rlm@1 175 {
rlm@1 176 if (firstItem)
rlm@1 177 return E_FAIL;
rlm@1 178 break;
rlm@1 179 }
rlm@1 180 firstItem = false;
rlm@1 181
rlm@1 182 UInt64 dataStartPos;
rlm@1 183 RINOK(_stream->Seek((UInt64)(Int64)(-3), STREAM_SEEK_CUR, &dataStartPos));
rlm@1 184
rlm@1 185 result = decoder->Code(_stream, outStream, NULL, NULL, progress);
rlm@1 186
rlm@1 187 if (result != S_OK)
rlm@1 188 break;
rlm@1 189
rlm@1 190 CMyComPtr<ICompressGetInStreamProcessedSize> getInStreamProcessedSize;
rlm@1 191 decoder.QueryInterface(IID_ICompressGetInStreamProcessedSize, &getInStreamProcessedSize);
rlm@1 192 if (!getInStreamProcessedSize)
rlm@1 193 break;
rlm@1 194 UInt64 packSize;
rlm@1 195 RINOK(getInStreamProcessedSize->GetInStreamProcessedSize(&packSize));
rlm@1 196 UInt64 pos;
rlm@1 197 RINOK(_stream->Seek(dataStartPos + packSize, STREAM_SEEK_SET, &pos));
rlm@1 198 currentTotalPacked = pos - _streamStartPosition;
rlm@1 199 }
rlm@1 200 outStream.Release();
rlm@1 201
rlm@1 202 Int32 retResult;
rlm@1 203 if (result == S_OK)
rlm@1 204 retResult = NExtract::NOperationResult::kOK;
rlm@1 205 else if (result == S_FALSE)
rlm@1 206 retResult = NExtract::NOperationResult::kDataError;
rlm@1 207 else
rlm@1 208 return result;
rlm@1 209 return extractCallback->SetOperationResult(retResult);
rlm@1 210
rlm@1 211 COM_TRY_END
rlm@1 212 }
rlm@1 213
rlm@1 214 IMPL_ISetCompressCodecsInfo
rlm@1 215
rlm@1 216 }}