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 using namespace ::com::sun::star::io;
  30 using namespace ::com::sun::star::uno;
  31
  32 namespace {
  33
  34 const sal_Int32 OUTPUTSTREAM_BUFFERSIZE     = 0x8000;
  35
  36 } // namespace
  37
  38 BinaryXOutputStream::BinaryXOutputStream( const Reference< XOutputStream >& rxOutStrm, bool bAutoClose ) :
  39     BinaryStreamBase( Reference< XSeekable >( rxOutStrm, UNO_QUERY ).is() ),
  40     BinaryXSeekableStream( Reference< XSeekable >( rxOutStrm, UNO_QUERY ) ),
  41     maBuffer( OUTPUTSTREAM_BUFFERSIZE ),
  42     mxOutStrm( rxOutStrm ),
  43     mbAutoClose( bAutoClose && rxOutStrm.is() )
  44 {
  45     mbEof = !mxOutStrm.is();
  46 }
  47
  48 BinaryXOutputStream::~BinaryXOutputStream()
  49 {
  50     close();
  51 }
  52
  53 void BinaryXOutputStream::close()
  54 {
  55     OSL_ENSURE( !mbAutoClose || mxOutStrm.is(), "BinaryXOutputStream::close - invalid call" );
  56     if( mxOutStrm.is() ) try
  57     {
  58         mxOutStrm->flush();
  59         if ( mbAutoClose )
  60             mxOutStrm->closeOutput();
  61     }
  62     catch( Exception& )
  63     {
  64         OSL_FAIL( "BinaryXOutputStream::close - closing output stream failed" );
  65     }
  66     mxOutStrm.clear();
  67     mbAutoClose = false;
  68     BinaryXSeekableStream::close();
  69 }
  70
  71 void BinaryXOutputStream::writeData( const StreamDataSequence& rData, size_t /*nAtomSize*/ )
  72 {
  73     if( mxOutStrm.is() ) try
  74     {
  75         mxOutStrm->writeBytes( rData );
  76     }
  77     catch( Exception& )
  78     {
  79         OSL_FAIL( "BinaryXOutputStream::writeData - stream read error" );
  80     }
  81 }
  82
  83 void BinaryXOutputStream::writeMemory( const void* pMem, sal_Int32 nBytes, size_t nAtomSize )
  84 {
  85     if( mxOutStrm.is() && (nBytes > 0) )
  86     {
  87         sal_Int32 nBufferSize = getLimitedValue< sal_Int32, sal_Int32 >( nBytes, 0, (OUTPUTSTREAM_BUFFERSIZE / nAtomSize) * nAtomSize );
  88         const sal_uInt8* pnMem = static_cast< const sal_uInt8* >( pMem );
  89         while( nBytes > 0 )
  90         {
  91             sal_Int32 nWriteSize = getLimitedValue< sal_Int32, sal_Int32 >( nBytes, 0, nBufferSize );
  92             maBuffer.realloc( nWriteSize );
  93             memcpy( maBuffer.getArray(), pnMem, static_cast< size_t >( nWriteSize ) );
  94             writeData( maBuffer, nAtomSize );
  95             pnMem += nWriteSize;
  96             nBytes -= nWriteSize;
  97         }
  98     }
  99 }
 100
 101 void
 102 BinaryOutputStream::writeCharArrayUC( const OUString& rString, rtl_TextEncoding eTextEnc, bool bAllowNulChars )
 103 {
 104     OString sBuf( OUStringToOString( rString, eTextEnc ) );
 105     if( !bAllowNulChars )
 106         sBuf = sBuf.replace( '\0', '?' );
 107     writeMemory( static_cast< const void* >( sBuf.getStr() ), sBuf.getLength() );
 108 }
 109
 110 void
 111 BinaryOutputStream::writeUnicodeArray( const OUString& rString, bool bAllowNulChars )
 112 {
 113     OUString sBuf( rString );
 114     if( !bAllowNulChars )
 115         sBuf = sBuf.replace( '\0', '?' );
 116 #ifdef OSL_BIGENDIAN
 117     // need a non-const buffer for swapping byte order
 118     sal_Unicode notConst[sBuf.getLength()];
 119     memcpy( notConst, sBuf.getStr(), sizeof(sal_Unicode)*sBuf.getLength() );
 120     writeArray( notConst, sBuf.getLength() );
 121 #else
 122     writeArray( sBuf.getStr(), sBuf.getLength() );
 123 #endif
 124 }
 125
 126 void
 127 BinaryOutputStream::writeCompressedUnicodeArray( const OUString& rString, bool bCompressed, bool bAllowNulChars )
 128 {
 129     if ( bCompressed )
 130          // ISO-8859-1 maps all byte values 0xHH to the same Unicode code point U+00HH
 131         writeCharArrayUC( rString, RTL_TEXTENCODING_ISO_8859_1, bAllowNulChars );
 132     else
 133         writeUnicodeArray( rString, bAllowNulChars );
 134 }
 135
 136 SequenceOutputStream::SequenceOutputStream( StreamDataSequence& rData ) :
 137     BinaryStreamBase( true ),
 138     SequenceSeekableStream( rData )
 139 {
 140 }
 141
 142 void SequenceOutputStream::writeData( const StreamDataSequence& rData, size_t nAtomSize )
 143 {
 144     if( mpData && rData.hasElements() )
 145         writeMemory( rData.getConstArray(), rData.getLength(), nAtomSize );
 146 }
 147
 148 void SequenceOutputStream::writeMemory( const void* pMem, sal_Int32 nBytes, size_t /*nAtomSize*/ )
 149 {
 150     if( mpData && (nBytes > 0) )
 151     {
 152         if( mpData->getLength() - mnPos < nBytes )
 153             const_cast< StreamDataSequence* >( mpData )->realloc( mnPos + nBytes );
 154         memcpy( const_cast< StreamDataSequence* >( mpData )->getArray() + mnPos, pMem, static_cast< size_t >( nBytes ) );
 155         mnPos += nBytes;
 156     }
 157 }
 158
 159 } // namespace oox
 160
 161 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */