oox/source/helper/binaryoutputstream.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  *
   9  * This file incorporates work covered by the following license notice:
  10  *
  11  *   Licensed to the Apache Software Foundation (ASF) under one or more
  12  *   contributor license agreements. See the NOTICE file distributed
  13  *   with this work for additional information regarding copyright
  14  *   ownership. The ASF licenses this file to you under the Apache
  15  *   License, Version 2.0 (the "License"); you may not use this file
  16  *   except in compliance with the License. You may obtain a copy of
  17  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
  18  */
  19
  20 #include "oox/helper/binaryoutputstream.hxx"
  21
  22 #include <com/sun/star/io/XOutputStream.hpp>
  23 #include <com/sun/star/io/XSeekable.hpp>
  24 #include <osl/diagnose.h>
  25 #include <string.h>
  26
  27 namespace oox {
  28
  29 // ============================================================================
  30
  31 using namespace ::com::sun::star::io;
  32 using namespace ::com::sun::star::uno;
  33
  34 namespace {
  35
  36 const sal_Int32 OUTPUTSTREAM_BUFFERSIZE     = 0x8000;
  37
  38 } // namespace
  39
  40 // ============================================================================
  41
  42 BinaryXOutputStream::BinaryXOutputStream( const Reference< XOutputStream >& rxOutStrm, bool bAutoClose ) :
  43     BinaryStreamBase( Reference< XSeekable >( rxOutStrm, UNO_QUERY ).is() ),
  44     BinaryXSeekableStream( Reference< XSeekable >( rxOutStrm, UNO_QUERY ) ),
  45     maBuffer( OUTPUTSTREAM_BUFFERSIZE ),
  46     mxOutStrm( rxOutStrm ),
  47     mbAutoClose( bAutoClose && rxOutStrm.is() )
  48 {
  49     mbEof = !mxOutStrm.is();
  50 }
  51
  52 BinaryXOutputStream::~BinaryXOutputStream()
  53 {
  54     close();
  55 }
  56
  57 void BinaryXOutputStream::close()
  58 {
  59     OSL_ENSURE( !mbAutoClose || mxOutStrm.is(), "BinaryXOutputStream::close - invalid call" );
  60     if( mxOutStrm.is() ) try
  61     {
  62         mxOutStrm->flush();
  63         mxOutStrm->closeOutput();
  64     }
  65     catch( Exception& )
  66     {
  67         OSL_FAIL( "BinaryXOutputStream::close - closing output stream failed" );
  68     }
  69     mxOutStrm.clear();
  70     mbAutoClose = false;
  71     BinaryXSeekableStream::close();
  72 }
  73
  74 void BinaryXOutputStream::writeData( const StreamDataSequence& rData, size_t /*nAtomSize*/ )
  75 {
  76     if( mxOutStrm.is() ) try
  77     {
  78         mxOutStrm->writeBytes( rData );
  79     }
  80     catch( Exception& )
  81     {
  82         OSL_FAIL( "BinaryXOutputStream::writeData - stream read error" );
  83     }
  84 }
  85
  86 void BinaryXOutputStream::writeMemory( const void* pMem, sal_Int32 nBytes, size_t nAtomSize )
  87 {
  88     if( mxOutStrm.is() && (nBytes > 0) )
  89     {
  90         sal_Int32 nBufferSize = getLimitedValue< sal_Int32, sal_Int32 >( nBytes, 0, (OUTPUTSTREAM_BUFFERSIZE / nAtomSize) * nAtomSize );
  91         const sal_uInt8* pnMem = reinterpret_cast< const sal_uInt8* >( pMem );
  92         while( nBytes > 0 )
  93         {
  94             sal_Int32 nWriteSize = getLimitedValue< sal_Int32, sal_Int32 >( nBytes, 0, nBufferSize );
  95             maBuffer.realloc( nWriteSize );
  96             memcpy( maBuffer.getArray(), pnMem, static_cast< size_t >( nWriteSize ) );
  97             writeData( maBuffer, nAtomSize );
  98             pnMem += nWriteSize;
  99             nBytes -= nWriteSize;
 100         }
 101     }
 102 }
 103
 104 void
 105 BinaryOutputStream::writeCharArrayUC( const OUString& rString, rtl_TextEncoding eTextEnc, bool bAllowNulChars )
 106 {
 107     OString sBuf( OUStringToOString( rString, eTextEnc ) );
 108     if( !bAllowNulChars )
 109         sBuf = sBuf.replace( '\0', '?' );
 110     writeMemory( static_cast< const void* >( sBuf.getStr() ), sBuf.getLength() );
 111 }
 112
 113 void
 114 BinaryOutputStream::writeUnicodeArray( const OUString& rString, bool bAllowNulChars )
 115 {
 116     OUString sBuf( rString );
 117     if( !bAllowNulChars )
 118         sBuf = sBuf.replace( '\0', '?' );
 119 #ifdef OSL_BIGENDIAN
 120     // need a non-const buffer for swapping byte order
 121     sal_Unicode notConst[sBuf.getLength()];
 122     memcpy( notConst, sBuf.getStr(), sizeof(sal_Unicode)*sBuf.getLength() );
 123     writeArray( notConst, sBuf.getLength() );
 124 #else
 125     writeArray( sBuf.getStr(), sBuf.getLength() );
 126 #endif
 127 }
 128
 129 void
 130 BinaryOutputStream::writeCompressedUnicodeArray( const OUString& rString, bool bCompressed, bool bAllowNulChars )
 131 {
 132     if ( bCompressed )
 133          // ISO-8859-1 maps all byte values 0xHH to the same Unicode code point U+00HH
 134         writeCharArrayUC( rString, RTL_TEXTENCODING_ISO_8859_1, bAllowNulChars );
 135     else
 136         writeUnicodeArray( rString, bAllowNulChars );
 137 }
 138
 139 // ============================================================================
 140
 141 SequenceOutputStream::SequenceOutputStream( StreamDataSequence& rData ) :
 142     BinaryStreamBase( true ),
 143     SequenceSeekableStream( rData )
 144 {
 145 }
 146
 147 void SequenceOutputStream::writeData( const StreamDataSequence& rData, size_t nAtomSize )
 148 {
 149     if( mpData && rData.hasElements() )
 150         writeMemory( rData.getConstArray(), rData.getLength(), nAtomSize );
 151 }
 152
 153 void SequenceOutputStream::writeMemory( const void* pMem, sal_Int32 nBytes, size_t /*nAtomSize*/ )
 154 {
 155     if( mpData && (nBytes > 0) )
 156     {
 157         if( mpData->getLength() - mnPos < nBytes )
 158             const_cast< StreamDataSequence* >( mpData )->realloc( mnPos + nBytes );
 159         memcpy( const_cast< StreamDataSequence* >( mpData )->getArray() + mnPos, pMem, static_cast< size_t >( nBytes ) );
 160         mnPos += nBytes;
 161     }
 162 }
 163
 164 // ============================================================================
 165
 166 } // namespace oox
 167
 168 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */