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