oox/source/helper/binaryinputstream.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/binaryinputstream.hxx"
  21
  22 #include <com/sun/star/io/XInputStream.hpp>
  23 #include <com/sun/star/io/XSeekable.hpp>
  24 #include <string.h>
  25 #include <algorithm>
  26 #include <vector>
  27 #include <rtl/strbuf.hxx>
  28 #include <rtl/ustrbuf.hxx>
  29 #include <osl/diagnose.h>
  30 #include "oox/helper/binaryoutputstream.hxx"
  31
  32 namespace oox {
  33
  34 using namespace ::com::sun::star::io;
  35 using namespace ::com::sun::star::uno;
  36
  37 namespace {
  38
  39 const sal_Int32 INPUTSTREAM_BUFFERSIZE      = 0x8000;
  40
  41 } // namespace
  42
  43 OUString BinaryInputStream::readNulUnicodeArray()
  44 {
  45     OUStringBuffer aBuffer;
  46     for (;;)
  47     {
  48       sal_uInt16 nChar = readuInt16();
  49       if ( mbEof || (nChar == 0) ) break;
  50       aBuffer.append( static_cast< sal_Unicode >( nChar ) );
  51     }
  52     return aBuffer.makeStringAndClear();
  53 }
  54
  55 OString BinaryInputStream::readCharArray( sal_Int32 nChars )
  56 {
  57     if( nChars <= 0 )
  58         return OString();
  59
  60     ::std::vector< sal_uInt8 > aBuffer;
  61     sal_Int32 nCharsRead = readArray( aBuffer, nChars );
  62     if( nCharsRead <= 0 )
  63         return OString();
  64
  65     aBuffer.resize( static_cast< size_t >( nCharsRead ) );
  66     // NUL characters are replaced by question marks.
  67     ::std::replace( aBuffer.begin(), aBuffer.end(), '\0', '?' );
  68
  69     return OString(reinterpret_cast<sal_Char*>(aBuffer.data()), nCharsRead);
  70 }
  71
  72 OUString BinaryInputStream::readCharArrayUC( sal_Int32 nChars, rtl_TextEncoding eTextEnc )
  73 {
  74     return OStringToOUString( readCharArray( nChars ), eTextEnc );
  75 }
  76
  77 OUString BinaryInputStream::readUnicodeArray( sal_Int32 nChars, bool bAllowNulChars )
  78 {
  79     if( nChars <= 0 )
  80         return OUString();
  81
  82     ::std::vector< sal_uInt16 > aBuffer;
  83     sal_Int32 nCharsRead = readArray( aBuffer, nChars );
  84     if( nCharsRead <= 0 )
  85         return OUString();
  86
  87     aBuffer.resize( static_cast< size_t >( nCharsRead ) );
  88     if( !bAllowNulChars )
  89         ::std::replace( aBuffer.begin(), aBuffer.begin() + nCharsRead, '\0', '?' );
  90
  91     OUStringBuffer aStringBuffer;
  92     aStringBuffer.ensureCapacity( nCharsRead );
  93     for( ::std::vector< sal_uInt16 >::iterator aIt = aBuffer.begin(), aEnd = aBuffer.end(); aIt != aEnd; ++aIt )
  94         aStringBuffer.append( static_cast< sal_Unicode >( *aIt ) );
  95     return aStringBuffer.makeStringAndClear();
  96 }
  97
  98 OUString BinaryInputStream::readCompressedUnicodeArray( sal_Int32 nChars, bool bCompressed )
  99 {
 100     return bCompressed ?
 101          // ISO-8859-1 maps all byte values 0xHH to the same Unicode code point U+00HH
 102         readCharArrayUC( nChars, RTL_TEXTENCODING_ISO_8859_1 ) :
 103         readUnicodeArray( nChars );
 104 }
 105
 106 void BinaryInputStream::copyToStream( BinaryOutputStream& rOutStrm )
 107 {
 108     sal_Int64 nBytes = SAL_MAX_INT64;
 109     sal_Int32 nBufferSize = INPUTSTREAM_BUFFERSIZE;
 110     StreamDataSequence aBuffer( nBufferSize );
 111     while( nBytes > 0 )
 112     {
 113         sal_Int32 nReadSize = getLimitedValue< sal_Int32, sal_Int64 >( nBytes, 0, nBufferSize );
 114         sal_Int32 nBytesRead = readData( aBuffer, nReadSize );
 115         rOutStrm.writeData( aBuffer );
 116         if( nReadSize == nBytesRead )
 117             nBytes -= nReadSize;
 118         else
 119             nBytes = 0;
 120     }
 121 }
 122
 123 BinaryXInputStream::BinaryXInputStream( const Reference< XInputStream >& rxInStrm, bool bAutoClose ) :
 124     BinaryStreamBase( Reference< XSeekable >( rxInStrm, UNO_QUERY ).is() ),
 125     BinaryXSeekableStream( Reference< XSeekable >( rxInStrm, UNO_QUERY ) ),
 126     maBuffer( INPUTSTREAM_BUFFERSIZE ),
 127     mxInStrm( rxInStrm ),
 128     mbAutoClose( bAutoClose && rxInStrm.is() )
 129 {
 130     mbEof = !mxInStrm.is();
 131 }
 132
 133 BinaryXInputStream::~BinaryXInputStream()
 134 {
 135     close();
 136 }
 137
 138 void BinaryXInputStream::close()
 139 {
 140     OSL_ENSURE( !mbAutoClose || mxInStrm.is(), "BinaryXInputStream::close - invalid call" );
 141     if( mxInStrm.is() ) try
 142     {
 143         mxInStrm->closeInput();
 144     }
 145     catch( Exception& )
 146     {
 147         OSL_FAIL( "BinaryXInputStream::close - closing input stream failed" );
 148     }
 149     mxInStrm.clear();
 150     mbAutoClose = false;
 151     BinaryXSeekableStream::close();
 152 }
 153
 154 sal_Int32 BinaryXInputStream::readData( StreamDataSequence& orData, sal_Int32 nBytes, size_t /*nAtomSize*/ )
 155 {
 156     sal_Int32 nRet = 0;
 157     if( !mbEof && (nBytes > 0) ) try
 158     {
 159         nRet = mxInStrm->readBytes( orData, nBytes );
 160         mbEof = nRet != nBytes;
 161     }
 162     catch( Exception& )
 163     {
 164         mbEof = true;
 165     }
 166     return nRet;
 167 }
 168
 169 sal_Int32 BinaryXInputStream::readMemory( void* opMem, sal_Int32 nBytes, size_t nAtomSize )
 170 {
 171     sal_Int32 nRet = 0;
 172     if( !mbEof && (nBytes > 0) )
 173     {
 174         sal_Int32 nBufferSize = getLimitedValue< sal_Int32, sal_Int32 >( nBytes, 0, INPUTSTREAM_BUFFERSIZE );
 175         sal_uInt8* opnMem = static_cast< sal_uInt8* >( opMem );
 176         while( !mbEof && (nBytes > 0) )
 177         {
 178             sal_Int32 nReadSize = getLimitedValue< sal_Int32, sal_Int32 >( nBytes, 0, nBufferSize );
 179             sal_Int32 nBytesRead = readData( maBuffer, nReadSize, nAtomSize );
 180             if( nBytesRead > 0 )
 181                 memcpy( opnMem, maBuffer.getConstArray(), static_cast< size_t >( nBytesRead ) );
 182             opnMem += nBytesRead;
 183             nBytes -= nBytesRead;
 184             nRet += nBytesRead;
 185         }
 186     }
 187     return nRet;
 188 }
 189
 190 void BinaryXInputStream::skip( sal_Int32 nBytes, size_t /*nAtomSize*/ )
 191 {
 192     if( !mbEof ) try
 193     {
 194         mxInStrm->skipBytes( nBytes );
 195     }
 196     catch( Exception& )
 197     {
 198         mbEof = true;
 199     }
 200 }
 201
 202 SequenceInputStream::SequenceInputStream( const StreamDataSequence& rData ) :
 203     BinaryStreamBase( true ),
 204     SequenceSeekableStream( rData )
 205 {
 206 }
 207
 208 sal_Int32 SequenceInputStream::readData( StreamDataSequence& orData, sal_Int32 nBytes, size_t /*nAtomSize*/ )
 209 {
 210     sal_Int32 nReadBytes = 0;
 211     if( !mbEof )
 212     {
 213         nReadBytes = getMaxBytes( nBytes );
 214         orData.realloc( nReadBytes );
 215         if( nReadBytes > 0 )
 216             memcpy( orData.getArray(), mpData->getConstArray() + mnPos, static_cast< size_t >( nReadBytes ) );
 217         mnPos += nReadBytes;
 218         mbEof = nReadBytes < nBytes;
 219     }
 220     return nReadBytes;
 221 }
 222
 223 sal_Int32 SequenceInputStream::readMemory( void* opMem, sal_Int32 nBytes, size_t /*nAtomSize*/ )
 224 {
 225     sal_Int32 nReadBytes = 0;
 226     if( !mbEof )
 227     {
 228         nReadBytes = getMaxBytes( nBytes );
 229         if( nReadBytes > 0 )
 230             memcpy( opMem, mpData->getConstArray() + mnPos, static_cast< size_t >( nReadBytes ) );
 231         mnPos += nReadBytes;
 232         mbEof = nReadBytes < nBytes;
 233     }
 234     return nReadBytes;
 235 }
 236
 237 void SequenceInputStream::skip( sal_Int32 nBytes, size_t /*nAtomSize*/ )
 238 {
 239     if( !mbEof )
 240     {
 241         sal_Int32 nSkipBytes = getMaxBytes( nBytes );
 242         mnPos += nSkipBytes;
 243         mbEof = nSkipBytes < nBytes;
 244     }
 245 }
 246
 247 RelativeInputStream::RelativeInputStream( BinaryInputStream& rInStrm, sal_Int64 nSize ) :
 248     BinaryStreamBase( rInStrm.isSeekable() ),
 249     mpInStrm( &rInStrm ),
 250     mnStartPos( rInStrm.tell() ),
 251     mnRelPos( 0 )
 252 {
 253     sal_Int64 nRemaining = rInStrm.getRemaining();
 254     mnSize = (nRemaining >= 0) ? ::std::min( nSize, nRemaining ) : nSize;
 255     mbEof = mbEof || rInStrm.isEof() || (mnSize < 0);
 256 }
 257
 258 sal_Int64 RelativeInputStream::size() const
 259 {
 260     return mpInStrm ? mnSize : -1;
 261 }
 262
 263 sal_Int64 RelativeInputStream::tell() const
 264 {
 265     return mpInStrm ? mnRelPos : -1;
 266 }
 267
 268 void RelativeInputStream::seek( sal_Int64 nPos )
 269 {
 270     if( mpInStrm && isSeekable() && (mnStartPos >= 0) )
 271     {
 272         mnRelPos = getLimitedValue< sal_Int64, sal_Int64 >( nPos, 0, mnSize );
 273         mpInStrm->seek( mnStartPos + mnRelPos );
 274         mbEof = (mnRelPos != nPos) || mpInStrm->isEof();
 275     }
 276 }
 277
 278 void RelativeInputStream::close()
 279 {
 280     mpInStrm = nullptr;
 281     mbEof = true;
 282 }
 283
 284 sal_Int32 RelativeInputStream::readData( StreamDataSequence& orData, sal_Int32 nBytes, size_t nAtomSize )
 285 {
 286     sal_Int32 nReadBytes = 0;
 287     if( !mbEof )
 288     {
 289         sal_Int32 nMaxBytes = getMaxBytes( nBytes );
 290         nReadBytes = mpInStrm->readData( orData, nMaxBytes, nAtomSize );
 291         mnRelPos += nReadBytes;
 292         mbEof = (nMaxBytes < nBytes) || mpInStrm->isEof();
 293     }
 294     return nReadBytes;
 295 }
 296
 297 sal_Int32 RelativeInputStream::readMemory( void* opMem, sal_Int32 nBytes, size_t nAtomSize )
 298 {
 299     sal_Int32 nReadBytes = 0;
 300     if( !mbEof )
 301     {
 302         sal_Int32 nMaxBytes = getMaxBytes( nBytes );
 303         nReadBytes = mpInStrm->readMemory( opMem, nMaxBytes, nAtomSize );
 304         mnRelPos += nReadBytes;
 305         mbEof = (nMaxBytes < nBytes) || mpInStrm->isEof();
 306     }
 307     return nReadBytes;
 308 }
 309
 310 void RelativeInputStream::skip( sal_Int32 nBytes, size_t nAtomSize )
 311 {
 312     if( !mbEof )
 313     {
 314         sal_Int32 nSkipBytes = getMaxBytes( nBytes );
 315         mpInStrm->skip( nSkipBytes, nAtomSize );
 316         mnRelPos += nSkipBytes;
 317         mbEof = nSkipBytes < nBytes;
 318     }
 319 }
 320
 321 } // namespace oox
 322
 323 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */