1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/src/win32/7zip/7z/CPP/7zip/Archive/7z/7zFolderInStream.cpp	Sat Mar 03 10:31:27 2012 -0600
     1.3 @@ -0,0 +1,130 @@
     1.4 +// 7zFolderInStream.cpp
     1.5 +
     1.6 +#include "StdAfx.h"
     1.7 +
     1.8 +#include "7zFolderInStream.h"
     1.9 +
    1.10 +namespace NArchive {
    1.11 +namespace N7z {
    1.12 +
    1.13 +CFolderInStream::CFolderInStream()
    1.14 +{
    1.15 +  _inStreamWithHashSpec = new CSequentialInStreamWithCRC;
    1.16 +  _inStreamWithHash = _inStreamWithHashSpec;
    1.17 +}
    1.18 +
    1.19 +void CFolderInStream::Init(IArchiveUpdateCallback *updateCallback,
    1.20 +    const UInt32 *fileIndices, UInt32 numFiles)
    1.21 +{
    1.22 +  _updateCallback = updateCallback;
    1.23 +  _numFiles = numFiles;
    1.24 +  _fileIndex = 0;
    1.25 +  _fileIndices = fileIndices;
    1.26 +  Processed.Clear();
    1.27 +  CRCs.Clear();
    1.28 +  Sizes.Clear();
    1.29 +  _fileIsOpen = false;
    1.30 +  _currentSizeIsDefined = false;
    1.31 +}
    1.32 +
    1.33 +HRESULT CFolderInStream::OpenStream()
    1.34 +{
    1.35 +  _filePos = 0;
    1.36 +  while (_fileIndex < _numFiles)
    1.37 +  {
    1.38 +    _currentSizeIsDefined = false;
    1.39 +    CMyComPtr<ISequentialInStream> stream;
    1.40 +    HRESULT result = _updateCallback->GetStream(_fileIndices[_fileIndex], &stream);
    1.41 +    if (result != S_OK && result != S_FALSE)
    1.42 +      return result;
    1.43 +    _fileIndex++;
    1.44 +    _inStreamWithHashSpec->SetStream(stream);
    1.45 +    _inStreamWithHashSpec->Init();
    1.46 +    if (!stream)
    1.47 +    {
    1.48 +      RINOK(_updateCallback->SetOperationResult(NArchive::NUpdate::NOperationResult::kOK));
    1.49 +      Sizes.Add(0);
    1.50 +      Processed.Add(result == S_OK);
    1.51 +      AddDigest();
    1.52 +      continue;
    1.53 +    }
    1.54 +    CMyComPtr<IStreamGetSize> streamGetSize;
    1.55 +    if (stream.QueryInterface(IID_IStreamGetSize, &streamGetSize) == S_OK)
    1.56 +    {
    1.57 +      if(streamGetSize)
    1.58 +      {
    1.59 +        _currentSizeIsDefined = true;
    1.60 +        RINOK(streamGetSize->GetSize(&_currentSize));
    1.61 +      }
    1.62 +    }
    1.63 +
    1.64 +    _fileIsOpen = true;
    1.65 +    return S_OK;
    1.66 +  }
    1.67 +  return S_OK;
    1.68 +}
    1.69 +
    1.70 +void CFolderInStream::AddDigest()
    1.71 +{
    1.72 +  CRCs.Add(_inStreamWithHashSpec->GetCRC());
    1.73 +}
    1.74 +
    1.75 +HRESULT CFolderInStream::CloseStream()
    1.76 +{
    1.77 +  RINOK(_updateCallback->SetOperationResult(NArchive::NUpdate::NOperationResult::kOK));
    1.78 +  _inStreamWithHashSpec->ReleaseStream();
    1.79 +  _fileIsOpen = false;
    1.80 +  Processed.Add(true);
    1.81 +  Sizes.Add(_filePos);
    1.82 +  AddDigest();
    1.83 +  return S_OK;
    1.84 +}
    1.85 +
    1.86 +STDMETHODIMP CFolderInStream::Read(void *data, UInt32 size, UInt32 *processedSize)
    1.87 +{
    1.88 +  UInt32 realProcessedSize = 0;
    1.89 +  while ((_fileIndex < _numFiles || _fileIsOpen) && size > 0)
    1.90 +  {
    1.91 +    if (_fileIsOpen)
    1.92 +    {
    1.93 +      UInt32 localProcessedSize;
    1.94 +      RINOK(_inStreamWithHash->Read(
    1.95 +          ((Byte *)data) + realProcessedSize, size, &localProcessedSize));
    1.96 +      if (localProcessedSize == 0)
    1.97 +      {
    1.98 +        RINOK(CloseStream());
    1.99 +        continue;
   1.100 +      }
   1.101 +      realProcessedSize += localProcessedSize;
   1.102 +      _filePos += localProcessedSize;
   1.103 +      size -= localProcessedSize;
   1.104 +      break;
   1.105 +    }
   1.106 +    else
   1.107 +    {
   1.108 +      RINOK(OpenStream());
   1.109 +    }
   1.110 +  }
   1.111 +  if (processedSize != 0)
   1.112 +    *processedSize = realProcessedSize;
   1.113 +  return S_OK;
   1.114 +}
   1.115 +
   1.116 +STDMETHODIMP CFolderInStream::GetSubStreamSize(UInt64 subStream, UInt64 *value)
   1.117 +{
   1.118 +  *value = 0;
   1.119 +  int subStreamIndex = (int)subStream;
   1.120 +  if (subStreamIndex < 0 || subStream > Sizes.Size())
   1.121 +    return E_FAIL;
   1.122 +  if (subStreamIndex < Sizes.Size())
   1.123 +  {
   1.124 +    *value= Sizes[subStreamIndex];
   1.125 +    return S_OK;
   1.126 +  }
   1.127 +  if (!_currentSizeIsDefined)
   1.128 +    return S_FALSE;
   1.129 +  *value = _currentSize;
   1.130 +  return S_OK;
   1.131 +}
   1.132 +
   1.133 +}}