xref: /third_party/lzma/CPP/7zip/Common/StreamUtils.cpp

// StreamUtils.cpp

#include "StdAfx.h"

#include "../../Common/MyCom.h"

#include "StreamUtils.h"

static const UInt32 kBlockSize = ((UInt32)1 << 31);


HRESULT InStream_SeekToBegin(IInStream *stream) throw()
{
  return InStream_SeekSet(stream, 0);
}


HRESULT InStream_AtBegin_GetSize(IInStream *stream, UInt64 &sizeRes) throw()
{
#ifdef _WIN32
  {
    Z7_DECL_CMyComPtr_QI_FROM(
        IStreamGetSize,
        streamGetSize, stream)
    if (streamGetSize && streamGetSize->GetSize(&sizeRes) == S_OK)
      return S_OK;
  }
#endif
  const HRESULT hres = InStream_GetSize_SeekToEnd(stream, sizeRes);
  const HRESULT hres2 = InStream_SeekToBegin(stream);
  return hres != S_OK ? hres : hres2;
}


HRESULT InStream_GetPos_GetSize(IInStream *stream, UInt64 &curPosRes, UInt64 &sizeRes) throw()
{
  RINOK(InStream_GetPos(stream, curPosRes))
#ifdef _WIN32
  {
    Z7_DECL_CMyComPtr_QI_FROM(
        IStreamGetSize,
        streamGetSize, stream)
    if (streamGetSize && streamGetSize->GetSize(&sizeRes) == S_OK)
      return S_OK;
  }
#endif
  const HRESULT hres = InStream_GetSize_SeekToEnd(stream, sizeRes);
  const HRESULT hres2 = InStream_SeekSet(stream, curPosRes);
  return hres != S_OK ? hres : hres2;
}



HRESULT ReadStream(ISequentialInStream *stream, void *data, size_t *processedSize) throw()
{
  size_t size = *processedSize;
  *processedSize = 0;
  while (size != 0)
  {
    UInt32 curSize = (size < kBlockSize) ? (UInt32)size : kBlockSize;
    UInt32 processedSizeLoc;
    HRESULT res = stream->Read(data, curSize, &processedSizeLoc);
    *processedSize += processedSizeLoc;
    data = (void *)((Byte *)data + processedSizeLoc);
    size -= processedSizeLoc;
    RINOK(res)
    if (processedSizeLoc == 0)
      return S_OK;
  }
  return S_OK;
}

HRESULT ReadStream_FALSE(ISequentialInStream *stream, void *data, size_t size) throw()
{
  size_t processedSize = size;
  RINOK(ReadStream(stream, data, &processedSize))
  return (size == processedSize) ? S_OK : S_FALSE;
}

HRESULT ReadStream_FAIL(ISequentialInStream *stream, void *data, size_t size) throw()
{
  size_t processedSize = size;
  RINOK(ReadStream(stream, data, &processedSize))
  return (size == processedSize) ? S_OK : E_FAIL;
}

HRESULT WriteStream(ISequentialOutStream *stream, const void *data, size_t size) throw()
{
  while (size != 0)
  {
    UInt32 curSize = (size < kBlockSize) ? (UInt32)size : kBlockSize;
    UInt32 processedSizeLoc;
    HRESULT res = stream->Write(data, curSize, &processedSizeLoc);
    data = (const void *)((const Byte *)data + processedSizeLoc);
    size -= processedSizeLoc;
    RINOK(res)
    if (processedSizeLoc == 0)
      return E_FAIL;
  }
  return S_OK;
}