diff src/win32/7zip/7z/CPP/7zip/Common/StreamUtils.cpp @ 1:f9f4f1b99eed

importing src directory
author Robert McIntyre <rlm@mit.edu>
date Sat, 03 Mar 2012 10:31:27 -0600
parents
children
line wrap: on
line diff
     1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/src/win32/7zip/7z/CPP/7zip/Common/StreamUtils.cpp	Sat Mar 03 10:31:27 2012 -0600
     1.3 @@ -0,0 +1,56 @@
     1.4 +// StreamUtils.cpp
     1.5 +
     1.6 +#include "StdAfx.h"
     1.7 +
     1.8 +#include "StreamUtils.h"
     1.9 +
    1.10 +static const UInt32 kBlockSize = ((UInt32)1 << 31);
    1.11 +
    1.12 +HRESULT ReadStream(ISequentialInStream *stream, void *data, size_t *processedSize)
    1.13 +{
    1.14 +  size_t size = *processedSize;
    1.15 +  *processedSize = 0;
    1.16 +  while (size != 0)
    1.17 +  {
    1.18 +    UInt32 curSize = (size < kBlockSize) ? (UInt32)size : kBlockSize;
    1.19 +    UInt32 processedSizeLoc;
    1.20 +    HRESULT res = stream->Read(data, curSize, &processedSizeLoc);
    1.21 +    *processedSize += processedSizeLoc;
    1.22 +    data = (void *)((Byte *)data + processedSizeLoc);
    1.23 +    size -= processedSizeLoc;
    1.24 +    RINOK(res);
    1.25 +    if (processedSizeLoc == 0)
    1.26 +      return S_OK;
    1.27 +  }
    1.28 +  return S_OK;
    1.29 +}
    1.30 +
    1.31 +HRESULT ReadStream_FALSE(ISequentialInStream *stream, void *data, size_t size)
    1.32 +{
    1.33 +  size_t processedSize = size;
    1.34 +  RINOK(ReadStream(stream, data, &processedSize));
    1.35 +  return (size == processedSize) ? S_OK : S_FALSE;
    1.36 +}
    1.37 +
    1.38 +HRESULT ReadStream_FAIL(ISequentialInStream *stream, void *data, size_t size)
    1.39 +{
    1.40 +  size_t processedSize = size;
    1.41 +  RINOK(ReadStream(stream, data, &processedSize));
    1.42 +  return (size == processedSize) ? S_OK : E_FAIL;
    1.43 +}
    1.44 +
    1.45 +HRESULT WriteStream(ISequentialOutStream *stream, const void *data, size_t size)
    1.46 +{
    1.47 +  while (size != 0)
    1.48 +  {
    1.49 +    UInt32 curSize = (size < kBlockSize) ? (UInt32)size : kBlockSize;
    1.50 +    UInt32 processedSizeLoc;
    1.51 +    HRESULT res = stream->Write(data, curSize, &processedSizeLoc);
    1.52 +    data = (const void *)((const Byte *)data + processedSizeLoc);
    1.53 +    size -= processedSizeLoc;
    1.54 +    RINOK(res);
    1.55 +    if (processedSizeLoc == 0)
    1.56 +      return E_FAIL;
    1.57 +  }
    1.58 +  return S_OK;
    1.59 +}