blob: 92c46fee534fa2f615879046e65083be7b4daa31 [file] [log] [blame]
/**************************************************************
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*
*************************************************************/
// MARKER(update_precomp.py): autogen include statement, do not remove
#include "precompiled_package.hxx"
#include <osl/diagnose.h>
#include <comphelper/storagehelper.hxx>
#include <switchpersistencestream.hxx>
using namespace ::com::sun::star;
// ========================================================================
struct SPStreamData_Impl
{
uno::Reference< lang::XMultiServiceFactory > m_xFactory;
sal_Bool m_bInStreamBased;
// the streams below are not visible from outside so there is no need to remember position
// original stream related members
uno::Reference< io::XStream > m_xOrigStream;
uno::Reference< io::XTruncate > m_xOrigTruncate;
uno::Reference< io::XSeekable > m_xOrigSeekable;
uno::Reference< io::XInputStream > m_xOrigInStream;
uno::Reference< io::XOutputStream > m_xOrigOutStream;
sal_Bool m_bInOpen;
sal_Bool m_bOutOpen;
SPStreamData_Impl(
const uno::Reference< lang::XMultiServiceFactory >& xFactory,
sal_Bool bInStreamBased,
const uno::Reference< io::XStream >& xOrigStream,
const uno::Reference< io::XTruncate >& xOrigTruncate,
const uno::Reference< io::XSeekable >& xOrigSeekable,
const uno::Reference< io::XInputStream >& xOrigInStream,
const uno::Reference< io::XOutputStream >& xOrigOutStream,
sal_Bool bInOpen,
sal_Bool bOutOpen )
: m_xFactory( xFactory )
, m_bInStreamBased( bInStreamBased )
, m_xOrigStream( xOrigStream )
, m_xOrigTruncate( xOrigTruncate )
, m_xOrigSeekable( xOrigSeekable )
, m_xOrigInStream( xOrigInStream )
, m_xOrigOutStream( xOrigOutStream )
, m_bInOpen( bInOpen )
, m_bOutOpen( bOutOpen )
{
}
};
// ========================================================================
// ------------------------------------------------------------------------
SwitchablePersistenceStream::SwitchablePersistenceStream(
const uno::Reference< lang::XMultiServiceFactory >& xFactory,
const uno::Reference< io::XStream >& xStream )
: m_xFactory( xFactory )
, m_pStreamData( NULL )
{
SwitchPersistenceTo( xStream );
}
// ------------------------------------------------------------------------
SwitchablePersistenceStream::SwitchablePersistenceStream(
const uno::Reference< lang::XMultiServiceFactory >& xFactory,
const uno::Reference< io::XInputStream >& xInputStream )
: m_xFactory( xFactory )
, m_pStreamData( NULL )
{
SwitchPersistenceTo( xInputStream );
}
// ------------------------------------------------------------------------
SwitchablePersistenceStream::~SwitchablePersistenceStream()
{
CloseAll_Impl();
}
// ------------------------------------------------------------------------
void SwitchablePersistenceStream::SwitchPersistenceTo( const uno::Reference< io::XStream >& xStream )
{
uno::Reference< io::XTruncate > xNewTruncate( xStream, uno::UNO_QUERY_THROW );
uno::Reference< io::XSeekable > xNewSeekable( xStream, uno::UNO_QUERY_THROW );
uno::Reference< io::XInputStream > xNewInStream = xStream->getInputStream();
uno::Reference< io::XOutputStream > xNewOutStream = xStream->getOutputStream();
if ( !xNewInStream.is() || !xNewOutStream.is() )
throw uno::RuntimeException();
sal_Int64 nPos = 0;
sal_Bool bInOpen = sal_False;
sal_Bool bOutOpen = sal_False;
if ( m_pStreamData && m_pStreamData->m_xOrigSeekable.is() )
{
// check that the length is the same
if ( m_pStreamData->m_xOrigSeekable->getLength() != xNewSeekable->getLength() )
throw uno::RuntimeException();
// get the current position
nPos = m_pStreamData->m_xOrigSeekable->getPosition();
bInOpen = m_pStreamData->m_bInOpen;
bOutOpen = m_pStreamData->m_bOutOpen;
}
xNewSeekable->seek( nPos );
CloseAll_Impl();
m_pStreamData = new SPStreamData_Impl( m_xFactory, sal_False,
xStream, xNewTruncate, xNewSeekable, xNewInStream, xNewOutStream,
bInOpen, bOutOpen );
}
// ------------------------------------------------------------------------
void SwitchablePersistenceStream::SwitchPersistenceTo( const uno::Reference< io::XInputStream >& xInputStream )
{
uno::Reference< io::XStream > xNewStream;
uno::Reference< io::XTruncate > xNewTruncate;
uno::Reference< io::XSeekable > xNewSeekable( xInputStream, uno::UNO_QUERY_THROW );
uno::Reference< io::XOutputStream > xNewOutStream;
if ( !xInputStream.is() )
throw uno::RuntimeException();
sal_Int64 nPos = 0;
sal_Bool bInOpen = sal_False;
sal_Bool bOutOpen = sal_False;
if ( m_pStreamData && m_pStreamData->m_xOrigSeekable.is() )
{
// check that the length is the same
if ( m_pStreamData->m_xOrigSeekable->getLength() != xNewSeekable->getLength() )
throw uno::RuntimeException();
// get the current position
nPos = m_pStreamData->m_xOrigSeekable->getPosition();
bInOpen = m_pStreamData->m_bInOpen;
bOutOpen = m_pStreamData->m_bOutOpen;
}
xNewSeekable->seek( nPos );
CloseAll_Impl();
m_pStreamData = new SPStreamData_Impl( m_xFactory, sal_True,
xNewStream, xNewTruncate, xNewSeekable, xInputStream, xNewOutStream,
bInOpen, bOutOpen );
}
// ------------------------------------------------------------------------
void SwitchablePersistenceStream::CopyAndSwitchPersistenceTo( const uno::Reference< io::XStream >& xStream )
{
uno::Reference< io::XStream > xTargetStream = xStream;
uno::Reference< io::XSeekable > xTargetSeek;
if ( !xTargetStream.is() )
{
xTargetStream = uno::Reference < io::XStream >(
m_xFactory->createInstance( ::rtl::OUString::createFromAscii( "com.sun.star.io.TempFile" ) ),
uno::UNO_QUERY_THROW );
xTargetSeek = uno::Reference< io::XSeekable >( xTargetStream, uno::UNO_QUERY_THROW );
}
else
{
// the provided stream must be empty
xTargetSeek = uno::Reference< io::XSeekable >( xTargetStream, uno::UNO_QUERY_THROW );
if ( xTargetSeek->getLength() )
throw io::IOException();
}
uno::Reference< io::XTruncate > xTargetTruncate( xTargetStream, uno::UNO_QUERY_THROW );
uno::Reference< io::XInputStream > xTargetInStream = xTargetStream->getInputStream();
uno::Reference< io::XOutputStream > xTargetOutStream = xTargetStream->getOutputStream();
if ( !xTargetInStream.is() || !xTargetOutStream.is() )
throw uno::RuntimeException();
if ( !m_pStreamData->m_xOrigInStream.is() || !m_pStreamData->m_xOrigSeekable.is() )
throw uno::RuntimeException();
sal_Int64 nPos = m_pStreamData->m_xOrigSeekable->getPosition();
m_pStreamData->m_xOrigSeekable->seek( 0 );
::comphelper::OStorageHelper::CopyInputToOutput( m_pStreamData->m_xOrigInStream, xTargetOutStream );
xTargetOutStream->flush();
xTargetSeek->seek( nPos );
sal_Bool bInOpen = m_pStreamData->m_bInOpen;
sal_Bool bOutOpen = m_pStreamData->m_bOutOpen;
CloseAll_Impl();
m_pStreamData = new SPStreamData_Impl( m_xFactory, sal_False,
xTargetStream, xTargetTruncate, xTargetSeek, xTargetInStream, xTargetOutStream,
bInOpen, bOutOpen );
}
// ------------------------------------------------------------------------
void SwitchablePersistenceStream::CloseAll_Impl()
{
if ( m_pStreamData )
{
delete m_pStreamData;
m_pStreamData = NULL;
}
}
// com::sun::star::io::XStream
// ------------------------------------------------------------------------
uno::Reference< io::XInputStream > SAL_CALL SwitchablePersistenceStream::getInputStream( )
throw (uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( m_pStreamData )
m_pStreamData->m_bInOpen = sal_True;
return static_cast< io::XInputStream* >( this );
}
// ------------------------------------------------------------------------
uno::Reference< io::XOutputStream > SAL_CALL SwitchablePersistenceStream::getOutputStream( )
throw (uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( m_pStreamData )
m_pStreamData->m_bOutOpen = sal_True;
return static_cast< io::XOutputStream* >( this );
}
// com::sun::star::io::XInputStream
// ------------------------------------------------------------------------
::sal_Int32 SAL_CALL SwitchablePersistenceStream::readBytes( uno::Sequence< ::sal_Int8 >& aData, ::sal_Int32 nBytesToRead )
throw (io::NotConnectedException, io::BufferSizeExceededException, io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData )
throw io::NotConnectedException();
// the original stream data should be provided
if ( !m_pStreamData->m_xOrigInStream.is() )
throw uno::RuntimeException();
return m_pStreamData->m_xOrigInStream->readBytes( aData, nBytesToRead );
}
// ------------------------------------------------------------------------
::sal_Int32 SAL_CALL SwitchablePersistenceStream::readSomeBytes( uno::Sequence< ::sal_Int8 >& aData, ::sal_Int32 nMaxBytesToRead )
throw (io::NotConnectedException, io::BufferSizeExceededException, io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData )
throw io::NotConnectedException();
// the original stream data should be provided
if ( !m_pStreamData->m_xOrigInStream.is() )
throw uno::RuntimeException();
return m_pStreamData->m_xOrigInStream->readBytes( aData, nMaxBytesToRead );
}
// ------------------------------------------------------------------------
void SAL_CALL SwitchablePersistenceStream::skipBytes( ::sal_Int32 nBytesToSkip )
throw (io::NotConnectedException, io::BufferSizeExceededException, io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData )
throw io::NotConnectedException();
// the original stream data should be provided
if ( !m_pStreamData->m_xOrigInStream.is() )
throw uno::RuntimeException();
m_pStreamData->m_xOrigInStream->skipBytes( nBytesToSkip );
}
// ------------------------------------------------------------------------
::sal_Int32 SAL_CALL SwitchablePersistenceStream::available( )
throw (io::NotConnectedException, io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData )
throw io::NotConnectedException();
// the original stream data should be provided
if ( !m_pStreamData->m_xOrigInStream.is() )
throw uno::RuntimeException();
return m_pStreamData->m_xOrigInStream->available();
}
// ------------------------------------------------------------------------
void SAL_CALL SwitchablePersistenceStream::closeInput()
throw (io::NotConnectedException, io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData )
throw io::NotConnectedException();
m_pStreamData->m_bInOpen = sal_False;
if ( !m_pStreamData->m_bOutOpen )
CloseAll_Impl();
}
// com::sun::star::io::XOutputStream
// ------------------------------------------------------------------------
void SAL_CALL SwitchablePersistenceStream::writeBytes( const uno::Sequence< ::sal_Int8 >& aData )
throw (io::NotConnectedException, io::BufferSizeExceededException, io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData )
throw io::NotConnectedException();
if ( m_pStreamData->m_bInStreamBased )
throw io::IOException();
// the original stream data should be provided
if ( !m_pStreamData->m_xOrigOutStream.is() )
throw uno::RuntimeException();
m_pStreamData->m_xOrigOutStream->writeBytes( aData );
}
// ------------------------------------------------------------------------
void SAL_CALL SwitchablePersistenceStream::flush( )
throw (io::NotConnectedException, io::BufferSizeExceededException, io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData || m_pStreamData->m_bInStreamBased )
{
OSL_ENSURE( sal_False, "flush() is not acceptable!\n" );
return;
// in future throw exception, for now some code might call flush() on closed stream
// since file ucp implementation allows it
// throw io::NotConnectedException();
}
// the original stream data should be provided
if ( !m_pStreamData->m_xOrigOutStream.is() )
throw uno::RuntimeException();
m_pStreamData->m_xOrigOutStream->flush();
}
// ------------------------------------------------------------------------
void SAL_CALL SwitchablePersistenceStream::closeOutput( )
throw (io::NotConnectedException, io::BufferSizeExceededException, io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData )
throw io::NotConnectedException();
m_pStreamData->m_bOutOpen = sal_False;
if ( !m_pStreamData->m_bInOpen )
CloseAll_Impl();
}
// com::sun::star::io::XTruncate
// ------------------------------------------------------------------------
void SAL_CALL SwitchablePersistenceStream::truncate( )
throw (io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData )
throw io::NotConnectedException();
if ( m_pStreamData->m_bInStreamBased )
throw io::IOException();
// the original stream data should be provided
if ( !m_pStreamData->m_xOrigTruncate.is() )
throw uno::RuntimeException();
m_pStreamData->m_xOrigTruncate->truncate();
}
// com::sun::star::io::XSeekable
// ------------------------------------------------------------------------
void SAL_CALL SwitchablePersistenceStream::seek( ::sal_Int64 location )
throw (lang::IllegalArgumentException, io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData )
throw io::NotConnectedException();
// the original stream data should be provided
if ( !m_pStreamData->m_xOrigSeekable.is() )
throw uno::RuntimeException();
m_pStreamData->m_xOrigSeekable->seek( location );
}
// ------------------------------------------------------------------------
::sal_Int64 SAL_CALL SwitchablePersistenceStream::getPosition( )
throw (io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData )
throw io::NotConnectedException();
// the original stream data should be provided
if ( !m_pStreamData->m_xOrigSeekable.is() )
throw uno::RuntimeException();
return m_pStreamData->m_xOrigSeekable->getPosition();
}
// ------------------------------------------------------------------------
::sal_Int64 SAL_CALL SwitchablePersistenceStream::getLength( )
throw (io::IOException, uno::RuntimeException)
{
::osl::MutexGuard aGuard( m_aMutex );
if ( !m_pStreamData )
throw io::NotConnectedException();
// the original stream data should be provided
if ( !m_pStreamData->m_xOrigSeekable.is() )
throw uno::RuntimeException();
return m_pStreamData->m_xOrigSeekable->getLength();
}
// ------------------------------------------------------------------------
void SAL_CALL SwitchablePersistenceStream::waitForCompletion()
throw (::com::sun::star::io::IOException, ::com::sun::star::uno::RuntimeException)
{
if ( !m_pStreamData )
throw io::NotConnectedException();
uno::Reference< io::XAsyncOutputMonitor > asyncOutputMonitor( m_pStreamData->m_xOrigOutStream, uno::UNO_QUERY );
if ( asyncOutputMonitor.is() )
asyncOutputMonitor->waitForCompletion();
}