rlm@1: // 7zDecode.cpp
rlm@1: 
rlm@1: #include "StdAfx.h"
rlm@1: 
rlm@1: #include "../../Common/LimitedStreams.h"
rlm@1: #include "../../Common/LockedStream.h"
rlm@1: #include "../../Common/ProgressUtils.h"
rlm@1: #include "../../Common/StreamObjects.h"
rlm@1: 
rlm@1: #include "7zDecode.h"
rlm@1: 
rlm@1: namespace NArchive {
rlm@1: namespace N7z {
rlm@1: 
rlm@1: static void ConvertFolderItemInfoToBindInfo(const CFolder &folder,
rlm@1:     CBindInfoEx &bindInfo)
rlm@1: {
rlm@1:   bindInfo.Clear();
rlm@1:   int i;
rlm@1:   for (i = 0; i < folder.BindPairs.Size(); i++)
rlm@1:   {
rlm@1:     NCoderMixer::CBindPair bindPair;
rlm@1:     bindPair.InIndex = (UInt32)folder.BindPairs[i].InIndex;
rlm@1:     bindPair.OutIndex = (UInt32)folder.BindPairs[i].OutIndex;
rlm@1:     bindInfo.BindPairs.Add(bindPair);
rlm@1:   }
rlm@1:   UInt32 outStreamIndex = 0;
rlm@1:   for (i = 0; i < folder.Coders.Size(); i++)
rlm@1:   {
rlm@1:     NCoderMixer::CCoderStreamsInfo coderStreamsInfo;
rlm@1:     const CCoderInfo &coderInfo = folder.Coders[i];
rlm@1:     coderStreamsInfo.NumInStreams = (UInt32)coderInfo.NumInStreams;
rlm@1:     coderStreamsInfo.NumOutStreams = (UInt32)coderInfo.NumOutStreams;
rlm@1:     bindInfo.Coders.Add(coderStreamsInfo);
rlm@1:     bindInfo.CoderMethodIDs.Add(coderInfo.MethodID);
rlm@1:     for (UInt32 j = 0; j < coderStreamsInfo.NumOutStreams; j++, outStreamIndex++)
rlm@1:       if (folder.FindBindPairForOutStream(outStreamIndex) < 0)
rlm@1:         bindInfo.OutStreams.Add(outStreamIndex);
rlm@1:   }
rlm@1:   for (i = 0; i < folder.PackStreams.Size(); i++)
rlm@1:     bindInfo.InStreams.Add((UInt32)folder.PackStreams[i]);
rlm@1: }
rlm@1: 
rlm@1: static bool AreCodersEqual(const NCoderMixer::CCoderStreamsInfo &a1,
rlm@1:     const NCoderMixer::CCoderStreamsInfo &a2)
rlm@1: {
rlm@1:   return (a1.NumInStreams == a2.NumInStreams) &&
rlm@1:     (a1.NumOutStreams == a2.NumOutStreams);
rlm@1: }
rlm@1: 
rlm@1: static bool AreBindPairsEqual(const NCoderMixer::CBindPair &a1, const NCoderMixer::CBindPair &a2)
rlm@1: {
rlm@1:   return (a1.InIndex == a2.InIndex) &&
rlm@1:     (a1.OutIndex == a2.OutIndex);
rlm@1: }
rlm@1: 
rlm@1: static bool AreBindInfoExEqual(const CBindInfoEx &a1, const CBindInfoEx &a2)
rlm@1: {
rlm@1:   if (a1.Coders.Size() != a2.Coders.Size())
rlm@1:     return false;
rlm@1:   int i;
rlm@1:   for (i = 0; i < a1.Coders.Size(); i++)
rlm@1:     if (!AreCodersEqual(a1.Coders[i], a2.Coders[i]))
rlm@1:       return false;
rlm@1:   if (a1.BindPairs.Size() != a2.BindPairs.Size())
rlm@1:     return false;
rlm@1:   for (i = 0; i < a1.BindPairs.Size(); i++)
rlm@1:     if (!AreBindPairsEqual(a1.BindPairs[i], a2.BindPairs[i]))
rlm@1:       return false;
rlm@1:   for (i = 0; i < a1.CoderMethodIDs.Size(); i++)
rlm@1:     if (a1.CoderMethodIDs[i] != a2.CoderMethodIDs[i])
rlm@1:       return false;
rlm@1:   if (a1.InStreams.Size() != a2.InStreams.Size())
rlm@1:     return false;
rlm@1:   if (a1.OutStreams.Size() != a2.OutStreams.Size())
rlm@1:     return false;
rlm@1:   return true;
rlm@1: }
rlm@1: 
rlm@1: CDecoder::CDecoder(bool multiThread)
rlm@1: {
rlm@1:   #ifndef _ST_MODE
rlm@1:   multiThread = true;
rlm@1:   #endif
rlm@1:   _multiThread = multiThread;
rlm@1:   _bindInfoExPrevIsDefined = false;
rlm@1: }
rlm@1: 
rlm@1: HRESULT CDecoder::Decode(
rlm@1:     DECL_EXTERNAL_CODECS_LOC_VARS
rlm@1:     IInStream *inStream,
rlm@1:     UInt64 startPos,
rlm@1:     const UInt64 *packSizes,
rlm@1:     const CFolder &folderInfo,
rlm@1:     ISequentialOutStream *outStream,
rlm@1:     ICompressProgressInfo *compressProgress
rlm@1:     #ifndef _NO_CRYPTO
rlm@1:     , ICryptoGetTextPassword *getTextPassword, bool &passwordIsDefined
rlm@1:     #endif
rlm@1:     #ifdef COMPRESS_MT
rlm@1:     , bool mtMode, UInt32 numThreads
rlm@1:     #endif
rlm@1:     )
rlm@1: {
rlm@1:   if (!folderInfo.CheckStructure())
rlm@1:     return E_NOTIMPL;
rlm@1:   #ifndef _NO_CRYPTO
rlm@1:   passwordIsDefined = false;
rlm@1:   #endif
rlm@1:   CObjectVector< CMyComPtr<ISequentialInStream> > inStreams;
rlm@1:   
rlm@1:   CLockedInStream lockedInStream;
rlm@1:   lockedInStream.Init(inStream);
rlm@1:   
rlm@1:   for (int j = 0; j < folderInfo.PackStreams.Size(); j++)
rlm@1:   {
rlm@1:     CLockedSequentialInStreamImp *lockedStreamImpSpec = new
rlm@1:         CLockedSequentialInStreamImp;
rlm@1:     CMyComPtr<ISequentialInStream> lockedStreamImp = lockedStreamImpSpec;
rlm@1:     lockedStreamImpSpec->Init(&lockedInStream, startPos);
rlm@1:     startPos += packSizes[j];
rlm@1:     
rlm@1:     CLimitedSequentialInStream *streamSpec = new
rlm@1:         CLimitedSequentialInStream;
rlm@1:     CMyComPtr<ISequentialInStream> inStream = streamSpec;
rlm@1:     streamSpec->SetStream(lockedStreamImp);
rlm@1:     streamSpec->Init(packSizes[j]);
rlm@1:     inStreams.Add(inStream);
rlm@1:   }
rlm@1:   
rlm@1:   int numCoders = folderInfo.Coders.Size();
rlm@1:   
rlm@1:   CBindInfoEx bindInfo;
rlm@1:   ConvertFolderItemInfoToBindInfo(folderInfo, bindInfo);
rlm@1:   bool createNewCoders;
rlm@1:   if (!_bindInfoExPrevIsDefined)
rlm@1:     createNewCoders = true;
rlm@1:   else
rlm@1:     createNewCoders = !AreBindInfoExEqual(bindInfo, _bindInfoExPrev);
rlm@1:   if (createNewCoders)
rlm@1:   {
rlm@1:     int i;
rlm@1:     _decoders.Clear();
rlm@1:     // _decoders2.Clear();
rlm@1:     
rlm@1:     _mixerCoder.Release();
rlm@1: 
rlm@1:     if (_multiThread)
rlm@1:     {
rlm@1:       _mixerCoderMTSpec = new NCoderMixer::CCoderMixer2MT;
rlm@1:       _mixerCoder = _mixerCoderMTSpec;
rlm@1:       _mixerCoderCommon = _mixerCoderMTSpec;
rlm@1:     }
rlm@1:     else
rlm@1:     {
rlm@1:       #ifdef _ST_MODE
rlm@1:       _mixerCoderSTSpec = new NCoderMixer::CCoderMixer2ST;
rlm@1:       _mixerCoder = _mixerCoderSTSpec;
rlm@1:       _mixerCoderCommon = _mixerCoderSTSpec;
rlm@1:       #endif
rlm@1:     }
rlm@1:     RINOK(_mixerCoderCommon->SetBindInfo(bindInfo));
rlm@1:     
rlm@1:     for (i = 0; i < numCoders; i++)
rlm@1:     {
rlm@1:       const CCoderInfo &coderInfo = folderInfo.Coders[i];
rlm@1: 
rlm@1:   
rlm@1:       CMyComPtr<ICompressCoder> decoder;
rlm@1:       CMyComPtr<ICompressCoder2> decoder2;
rlm@1:       RINOK(CreateCoder(
rlm@1:           EXTERNAL_CODECS_LOC_VARS
rlm@1:           coderInfo.MethodID, decoder, decoder2, false));
rlm@1:       CMyComPtr<IUnknown> decoderUnknown;
rlm@1:       if (coderInfo.IsSimpleCoder())
rlm@1:       {
rlm@1:         if (decoder == 0)
rlm@1:           return E_NOTIMPL;
rlm@1: 
rlm@1:         decoderUnknown = (IUnknown *)decoder;
rlm@1:         
rlm@1:         if (_multiThread)
rlm@1:           _mixerCoderMTSpec->AddCoder(decoder);
rlm@1:         #ifdef _ST_MODE
rlm@1:         else
rlm@1:           _mixerCoderSTSpec->AddCoder(decoder, false);
rlm@1:         #endif
rlm@1:       }
rlm@1:       else
rlm@1:       {
rlm@1:         if (decoder2 == 0)
rlm@1:           return E_NOTIMPL;
rlm@1:         decoderUnknown = (IUnknown *)decoder2;
rlm@1:         if (_multiThread)
rlm@1:           _mixerCoderMTSpec->AddCoder2(decoder2);
rlm@1:         #ifdef _ST_MODE
rlm@1:         else
rlm@1:           _mixerCoderSTSpec->AddCoder2(decoder2, false);
rlm@1:         #endif
rlm@1:       }
rlm@1:       _decoders.Add(decoderUnknown);
rlm@1:       #ifdef EXTERNAL_CODECS
rlm@1:       CMyComPtr<ISetCompressCodecsInfo> setCompressCodecsInfo;
rlm@1:       decoderUnknown.QueryInterface(IID_ISetCompressCodecsInfo, (void **)&setCompressCodecsInfo);
rlm@1:       if (setCompressCodecsInfo)
rlm@1:       {
rlm@1:         RINOK(setCompressCodecsInfo->SetCompressCodecsInfo(codecsInfo));
rlm@1:       }
rlm@1:       #endif
rlm@1:     }
rlm@1:     _bindInfoExPrev = bindInfo;
rlm@1:     _bindInfoExPrevIsDefined = true;
rlm@1:   }
rlm@1:   int i;
rlm@1:   _mixerCoderCommon->ReInit();
rlm@1:   
rlm@1:   UInt32 packStreamIndex = 0, unpackStreamIndex = 0;
rlm@1:   UInt32 coderIndex = 0;
rlm@1:   // UInt32 coder2Index = 0;
rlm@1:   
rlm@1:   for (i = 0; i < numCoders; i++)
rlm@1:   {
rlm@1:     const CCoderInfo &coderInfo = folderInfo.Coders[i];
rlm@1:     CMyComPtr<IUnknown> &decoder = _decoders[coderIndex];
rlm@1:     
rlm@1:     {
rlm@1:       CMyComPtr<ICompressSetDecoderProperties2> setDecoderProperties;
rlm@1:       decoder.QueryInterface(IID_ICompressSetDecoderProperties2, &setDecoderProperties);
rlm@1:       if (setDecoderProperties)
rlm@1:       {
rlm@1:         const CByteBuffer &props = coderInfo.Props;
rlm@1:         size_t size = props.GetCapacity();
rlm@1:         if (size > 0xFFFFFFFF)
rlm@1:           return E_NOTIMPL;
rlm@1:         if (size > 0)
rlm@1:         {
rlm@1:           RINOK(setDecoderProperties->SetDecoderProperties2((const Byte *)props, (UInt32)size));
rlm@1:         }
rlm@1:       }
rlm@1:     }
rlm@1: 
rlm@1:     #ifdef COMPRESS_MT
rlm@1:     if (mtMode)
rlm@1:     {
rlm@1:       CMyComPtr<ICompressSetCoderMt> setCoderMt;
rlm@1:       decoder.QueryInterface(IID_ICompressSetCoderMt, &setCoderMt);
rlm@1:       if (setCoderMt)
rlm@1:       {
rlm@1:         RINOK(setCoderMt->SetNumberOfThreads(numThreads));
rlm@1:       }
rlm@1:     }
rlm@1:     #endif
rlm@1: 
rlm@1:     #ifndef _NO_CRYPTO
rlm@1:     {
rlm@1:       CMyComPtr<ICryptoSetPassword> cryptoSetPassword;
rlm@1:       decoder.QueryInterface(IID_ICryptoSetPassword, &cryptoSetPassword);
rlm@1:       if (cryptoSetPassword)
rlm@1:       {
rlm@1:         if (getTextPassword == 0)
rlm@1:           return E_FAIL;
rlm@1:         CMyComBSTR passwordBSTR;
rlm@1:         RINOK(getTextPassword->CryptoGetTextPassword(&passwordBSTR));
rlm@1:         CByteBuffer buffer;
rlm@1:         passwordIsDefined = true;
rlm@1:         const UString password(passwordBSTR);
rlm@1:         const UInt32 sizeInBytes = password.Length() * 2;
rlm@1:         buffer.SetCapacity(sizeInBytes);
rlm@1:         for (int i = 0; i < password.Length(); i++)
rlm@1:         {
rlm@1:           wchar_t c = password[i];
rlm@1:           ((Byte *)buffer)[i * 2] = (Byte)c;
rlm@1:           ((Byte *)buffer)[i * 2 + 1] = (Byte)(c >> 8);
rlm@1:         }
rlm@1:         RINOK(cryptoSetPassword->CryptoSetPassword((const Byte *)buffer, sizeInBytes));
rlm@1:       }
rlm@1:     }
rlm@1:     #endif
rlm@1: 
rlm@1:     coderIndex++;
rlm@1:     
rlm@1:     UInt32 numInStreams = (UInt32)coderInfo.NumInStreams;
rlm@1:     UInt32 numOutStreams = (UInt32)coderInfo.NumOutStreams;
rlm@1:     CRecordVector<const UInt64 *> packSizesPointers;
rlm@1:     CRecordVector<const UInt64 *> unpackSizesPointers;
rlm@1:     packSizesPointers.Reserve(numInStreams);
rlm@1:     unpackSizesPointers.Reserve(numOutStreams);
rlm@1:     UInt32 j;
rlm@1:     for (j = 0; j < numOutStreams; j++, unpackStreamIndex++)
rlm@1:       unpackSizesPointers.Add(&folderInfo.UnpackSizes[unpackStreamIndex]);
rlm@1:     
rlm@1:     for (j = 0; j < numInStreams; j++, packStreamIndex++)
rlm@1:     {
rlm@1:       int bindPairIndex = folderInfo.FindBindPairForInStream(packStreamIndex);
rlm@1:       if (bindPairIndex >= 0)
rlm@1:         packSizesPointers.Add(
rlm@1:         &folderInfo.UnpackSizes[(UInt32)folderInfo.BindPairs[bindPairIndex].OutIndex]);
rlm@1:       else
rlm@1:       {
rlm@1:         int index = folderInfo.FindPackStreamArrayIndex(packStreamIndex);
rlm@1:         if (index < 0)
rlm@1:           return E_FAIL;
rlm@1:         packSizesPointers.Add(&packSizes[index]);
rlm@1:       }
rlm@1:     }
rlm@1:     
rlm@1:     _mixerCoderCommon->SetCoderInfo(i,
rlm@1:         &packSizesPointers.Front(),
rlm@1:         &unpackSizesPointers.Front());
rlm@1:   }
rlm@1:   UInt32 mainCoder, temp;
rlm@1:   bindInfo.FindOutStream(bindInfo.OutStreams[0], mainCoder, temp);
rlm@1: 
rlm@1:   if (_multiThread)
rlm@1:     _mixerCoderMTSpec->SetProgressCoderIndex(mainCoder);
rlm@1:   /*
rlm@1:   else
rlm@1:     _mixerCoderSTSpec->SetProgressCoderIndex(mainCoder);;
rlm@1:   */
rlm@1:   
rlm@1:   if (numCoders == 0)
rlm@1:     return 0;
rlm@1:   CRecordVector<ISequentialInStream *> inStreamPointers;
rlm@1:   inStreamPointers.Reserve(inStreams.Size());
rlm@1:   for (i = 0; i < inStreams.Size(); i++)
rlm@1:     inStreamPointers.Add(inStreams[i]);
rlm@1:   ISequentialOutStream *outStreamPointer = outStream;
rlm@1:   return _mixerCoder->Code(&inStreamPointers.Front(), NULL,
rlm@1:     inStreams.Size(), &outStreamPointer, NULL, 1, compressProgress);
rlm@1: }
rlm@1: 
rlm@1: }}