mirror of
https://github.com/holub/mame
synced 2025-06-07 13:23:50 +03:00
102 lines
2.6 KiB
C++
102 lines
2.6 KiB
C++
// 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;
|
|
}
|