1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/src/win32/7zip/7z/CPP/7zip/Archive/Lzma/LzmaHandler.cpp	Sat Mar 03 10:31:27 2012 -0600
     1.3 @@ -0,0 +1,243 @@
     1.4 +// LzmaHandler.cpp
     1.5 +
     1.6 +#include "StdAfx.h"
     1.7 +
     1.8 +#include "LzmaHandler.h"
     1.9 +
    1.10 +#include "Common/Defs.h"
    1.11 +#include "Common/StringConvert.h"
    1.12 +#include "Common/ComTry.h"
    1.13 +#include "Common/IntToString.h"
    1.14 +
    1.15 +#include "Windows/PropVariant.h"
    1.16 +
    1.17 +#include "../../Common/ProgressUtils.h"
    1.18 +#include "../../Common/StreamUtils.h"
    1.19 +#include "../Common/DummyOutStream.h"
    1.20 +
    1.21 +#include "LzmaFiltersDecode.h"
    1.22 +
    1.23 +namespace NArchive {
    1.24 +namespace NLzma {
    1.25 +
    1.26 +STATPROPSTG kProps[] =
    1.27 +{
    1.28 +  { NULL, kpidSize, VT_UI8},
    1.29 +  { NULL, kpidPackSize, VT_UI8},
    1.30 +  { NULL, kpidMethod, VT_UI1}
    1.31 +};
    1.32 +
    1.33 +IMP_IInArchive_Props
    1.34 +IMP_IInArchive_ArcProps_NO
    1.35 +
    1.36 +STDMETHODIMP CHandler::GetNumberOfItems(UInt32 *numItems)
    1.37 +{
    1.38 +  *numItems = 1;
    1.39 +  return S_OK;
    1.40 +}
    1.41 +
    1.42 +static void ConvertUInt32ToString(UInt32 value, wchar_t *s)
    1.43 +{
    1.44 +  ConvertUInt64ToString(value, s + MyStringLen(s));
    1.45 +}
    1.46 +
    1.47 +static void DictSizeToString(UInt32 value, wchar_t *s)
    1.48 +{
    1.49 +  for (int i = 0; i <= 31; i++)
    1.50 +    if ((UInt32(1) << i) == value)
    1.51 +    {
    1.52 +      ConvertUInt32ToString(i, s);
    1.53 +      return;
    1.54 +    }
    1.55 +  wchar_t c = L'b';
    1.56 +  if ((value & ((1 << 20) - 1)) == 0)
    1.57 +  {
    1.58 +    value >>= 20;
    1.59 +    c = L'm';
    1.60 +  }
    1.61 +  else if ((value & ((1 << 10) - 1)) == 0)
    1.62 +  {
    1.63 +    value >>= 10;
    1.64 +    c = L'k';
    1.65 +  }
    1.66 +  ConvertUInt32ToString(value, s);
    1.67 +  int p = MyStringLen(s);
    1.68 +  s[p++] = c;
    1.69 +  s[p++] = L'\0';
    1.70 +}
    1.71 +
    1.72 +static void MyStrCat(wchar_t *d, const wchar_t *s)
    1.73 +{
    1.74 +  MyStringCopy(d + MyStringLen(d), s);
    1.75 +}
    1.76 +
    1.77 +STDMETHODIMP CHandler::GetProperty(UInt32 index, PROPID propID,  PROPVARIANT *value)
    1.78 +{
    1.79 +  if (index != 0)
    1.80 +    return E_INVALIDARG;
    1.81 +  NWindows::NCOM::CPropVariant propVariant;
    1.82 +  switch(propID)
    1.83 +  {
    1.84 +    case kpidSize:
    1.85 +      if (m_StreamInfo.HasUnpackSize())
    1.86 +        propVariant = (UInt64)m_StreamInfo.UnpackSize;
    1.87 +      break;
    1.88 +    case kpidPackSize:
    1.89 +      propVariant = (UInt64)m_PackSize;
    1.90 +      break;
    1.91 +    case kpidMethod:
    1.92 +    {
    1.93 +      wchar_t s[64];
    1.94 +      s[0] = '\0';
    1.95 +      if (m_StreamInfo.IsThereFilter)
    1.96 +      {
    1.97 +        const wchar_t *f;
    1.98 +        if (m_StreamInfo.FilterMethod == 0)
    1.99 +          f = L"Copy";
   1.100 +        else if (m_StreamInfo.FilterMethod == 1)
   1.101 +          f = L"BCJ";
   1.102 +        else
   1.103 +          f = L"Unknown";
   1.104 +        MyStrCat(s, f);
   1.105 +        MyStrCat(s, L" ");
   1.106 +      }
   1.107 +      MyStrCat(s, L"LZMA:");
   1.108 +      DictSizeToString(m_StreamInfo.GetDicSize(), s);
   1.109 +      propVariant = s;
   1.110 +      break;
   1.111 +    }
   1.112 +  }
   1.113 +  propVariant.Detach(value);
   1.114 +  return S_OK;
   1.115 +}
   1.116 +
   1.117 +STDMETHODIMP CHandler::Open(IInStream *inStream,
   1.118 +    const UInt64 * /* maxCheckStartPosition */,
   1.119 +    IArchiveOpenCallback * /* openArchiveCallback */)
   1.120 +{
   1.121 +  {
   1.122 +    RINOK(inStream->Seek(0, STREAM_SEEK_CUR, &m_StreamStartPosition));
   1.123 +
   1.124 +    HRESULT res = ReadStreamHeader(inStream, m_StreamInfo);
   1.125 +    if (res != S_OK)
   1.126 +      return S_FALSE;
   1.127 +    
   1.128 +    Byte b;
   1.129 +    RINOK(ReadStream_FALSE(inStream, &b, 1));
   1.130 +    if (b != 0)
   1.131 +      return S_FALSE;
   1.132 +
   1.133 +    UInt64 endPos;
   1.134 +    RINOK(inStream->Seek(0, STREAM_SEEK_END, &endPos));
   1.135 +    m_PackSize = endPos - m_StreamStartPosition - m_StreamInfo.GetHeaderSize();
   1.136 +
   1.137 +    m_Stream = inStream;
   1.138 +  }
   1.139 +  return S_OK;
   1.140 +}
   1.141 +
   1.142 +STDMETHODIMP CHandler::Close()
   1.143 +{
   1.144 +  m_Stream.Release();
   1.145 +  return S_OK;
   1.146 +}
   1.147 +
   1.148 +
   1.149 +STDMETHODIMP CHandler::Extract(const UInt32* indices, UInt32 numItems,
   1.150 +    Int32 _aTestMode, IArchiveExtractCallback *extractCallback)
   1.151 +{
   1.152 +  COM_TRY_BEGIN
   1.153 +  bool allFilesMode = (numItems == UInt32(-1));
   1.154 +  if (!allFilesMode)
   1.155 +  {
   1.156 +    if (numItems == 0)
   1.157 +      return S_OK;
   1.158 +    if (numItems != 1)
   1.159 +      return E_INVALIDARG;
   1.160 +    if (indices[0] != 0)
   1.161 +      return E_INVALIDARG;
   1.162 +  }
   1.163 +
   1.164 +  bool testMode = (_aTestMode != 0);
   1.165 +
   1.166 +  RINOK(extractCallback->SetTotal(m_PackSize));
   1.167 +    
   1.168 +  UInt64 currentTotalPacked = 0;
   1.169 +
   1.170 +  CDummyOutStream *outStreamSpec = new CDummyOutStream;
   1.171 +  CMyComPtr<ISequentialOutStream> outStream(outStreamSpec);
   1.172 +
   1.173 +  {
   1.174 +    CMyComPtr<ISequentialOutStream> realOutStream;
   1.175 +    Int32 askMode = testMode ?
   1.176 +        NArchive::NExtract::NAskMode::kTest :
   1.177 +        NArchive::NExtract::NAskMode::kExtract;
   1.178 +    
   1.179 +    RINOK(extractCallback->GetStream(0, &realOutStream, askMode));
   1.180 +
   1.181 +    outStreamSpec->SetStream(realOutStream);
   1.182 +    outStreamSpec->Init();
   1.183 +    if(!testMode && !realOutStream)
   1.184 +      return S_OK;
   1.185 +    extractCallback->PrepareOperation(askMode);
   1.186 +  }
   1.187 +  
   1.188 +  CLocalProgress *lps = new CLocalProgress;
   1.189 +  CMyComPtr<ICompressProgressInfo> progress = lps;
   1.190 +  lps->Init(extractCallback, true);
   1.191 +
   1.192 +  CDecoder decoder;
   1.193 +  RINOK(m_Stream->Seek(m_StreamStartPosition, STREAM_SEEK_SET, NULL));
   1.194 +  UInt64 streamPos = m_StreamStartPosition;
   1.195 +  Int32 opRes = NArchive::NExtract::NOperationResult::kOK;
   1.196 +  bool firstItem = true;
   1.197 +  for (;;)
   1.198 +  {
   1.199 +    CHeader st;
   1.200 +    HRESULT result = ReadStreamHeader(m_Stream, st);
   1.201 +    if (result != S_OK)
   1.202 +    {
   1.203 +      if (firstItem)
   1.204 +        return E_FAIL;
   1.205 +      break;
   1.206 +    }
   1.207 +    firstItem = false;
   1.208 +
   1.209 +    lps->OutSize = outStreamSpec->GetSize();
   1.210 +    lps->InSize = currentTotalPacked;
   1.211 +    RINOK(lps->SetCur());
   1.212 +    
   1.213 +    streamPos += st.GetHeaderSize();
   1.214 +    UInt64 packProcessed;
   1.215 +
   1.216 +    {
   1.217 +      result = decoder.Code(
   1.218 +          EXTERNAL_CODECS_VARS
   1.219 +          st, m_Stream, outStream, &packProcessed, progress);
   1.220 +      if (result == E_NOTIMPL)
   1.221 +      {
   1.222 +        opRes = NArchive::NExtract::NOperationResult::kUnSupportedMethod;
   1.223 +        break;
   1.224 +      }
   1.225 +      if (result == S_FALSE)
   1.226 +      {
   1.227 +        opRes = NArchive::NExtract::NOperationResult::kDataError;
   1.228 +        break;
   1.229 +      }
   1.230 +      RINOK(result);
   1.231 +    }
   1.232 +
   1.233 +    if (packProcessed == (UInt64)(Int64)-1)
   1.234 +      break;
   1.235 +    RINOK(m_Stream->Seek(streamPos + packProcessed, STREAM_SEEK_SET, NULL));
   1.236 +    currentTotalPacked += packProcessed;
   1.237 +    streamPos += packProcessed;
   1.238 +  }
   1.239 +  outStream.Release();
   1.240 +  return extractCallback->SetOperationResult(opRes);
   1.241 +  COM_TRY_END
   1.242 +}
   1.243 +
   1.244 +IMPL_ISetCompressCodecsInfo
   1.245 +
   1.246 +}}