You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
1.7 KiB
94 lines
1.7 KiB
10 months ago
|
#include "OutStreamWrapper.h"
|
||
|
|
|
||
|
|
|
||
|
|
namespace SevenZip
|
||
|
|
{
|
||
|
|
namespace intl
|
||
|
|
{
|
||
|
|
|
||
|
|
OutStreamWrapper::OutStreamWrapper( const CMyComPtr< IStream >& baseStream )
|
||
|
|
: m_refCount( 0 )
|
||
|
|
, m_baseStream( baseStream )
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
OutStreamWrapper::~OutStreamWrapper()
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE OutStreamWrapper::QueryInterface( REFIID iid, void** ppvObject )
|
||
|
|
{
|
||
|
|
if ( iid == __uuidof( IUnknown ) )
|
||
|
|
{
|
||
|
|
*ppvObject = static_cast< IUnknown* >( this );
|
||
|
|
AddRef();
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( iid == IID_ISequentialOutStream )
|
||
|
|
{
|
||
|
|
*ppvObject = static_cast< ISequentialOutStream* >( this );
|
||
|
|
AddRef();
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( iid == IID_IOutStream )
|
||
|
|
{
|
||
|
|
*ppvObject = static_cast< IOutStream* >( this );
|
||
|
|
AddRef();
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
return E_NOINTERFACE;
|
||
|
|
}
|
||
|
|
|
||
|
|
ULONG STDMETHODCALLTYPE OutStreamWrapper::AddRef()
|
||
|
|
{
|
||
|
|
return static_cast< ULONG >( InterlockedIncrement( &m_refCount ) );
|
||
|
|
}
|
||
|
|
|
||
|
|
ULONG STDMETHODCALLTYPE OutStreamWrapper::Release()
|
||
|
|
{
|
||
|
|
ULONG res = static_cast< ULONG >( InterlockedDecrement( &m_refCount ) );
|
||
|
|
if ( res == 0 )
|
||
|
|
{
|
||
|
|
delete this;
|
||
|
|
}
|
||
|
|
return res;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP OutStreamWrapper::Write( const void* data, UInt32 size, UInt32* processedSize )
|
||
|
|
{
|
||
|
|
ULONG written = 0;
|
||
|
|
HRESULT hr = m_baseStream->Write( data, size, &written );
|
||
|
|
if ( processedSize != NULL )
|
||
|
|
{
|
||
|
|
*processedSize = written;
|
||
|
|
}
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP OutStreamWrapper::Seek( Int64 offset, UInt32 seekOrigin, UInt64* newPosition )
|
||
|
|
{
|
||
|
|
LARGE_INTEGER move;
|
||
|
|
ULARGE_INTEGER newPos;
|
||
|
|
|
||
|
|
move.QuadPart = offset;
|
||
|
|
HRESULT hr = m_baseStream->Seek( move, seekOrigin, &newPos );
|
||
|
|
if ( newPosition != NULL )
|
||
|
|
{
|
||
|
|
*newPosition = newPos.QuadPart;
|
||
|
|
}
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP OutStreamWrapper::SetSize( UInt64 newSize )
|
||
|
|
{
|
||
|
|
ULARGE_INTEGER size;
|
||
|
|
size.QuadPart = newSize;
|
||
|
|
return m_baseStream->SetSize( size );
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|