dtrans/source/win32/dtobj/DTransHelper.cxx

   1 /*************************************************************************
   2  *
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * Copyright 2008 by Sun Microsystems, Inc.
   6  *
   7  * OpenOffice.org - a multi-platform office productivity suite
   8  *
   9  * $RCSfile: DTransHelper.cxx,v $
  10  * $Revision: 1.14 $
  11  *
  12  * This file is part of OpenOffice.org.
  13  *
  14  * OpenOffice.org is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU Lesser General Public License version 3
  16  * only, as published by the Free Software Foundation.
  17  *
  18  * OpenOffice.org is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU Lesser General Public License version 3 for more details
  22  * (a copy is included in the LICENSE file that accompanied this code).
  23  *
  24  * You should have received a copy of the GNU Lesser General Public License
  25  * version 3 along with OpenOffice.org.  If not, see
  26  * <http://www.openoffice.org/license.html>
  27  * for a copy of the LGPLv3 License.
  28  *
  29  ************************************************************************/
  30
  31 // MARKER(update_precomp.py): autogen include statement, do not remove
  32 #include "precompiled_dtrans.hxx"
  33
  34 //------------------------------------------------------------------------
  35 // includes
  36 //------------------------------------------------------------------------
  37 #include <rtl/ustring.h>
  38 #include <osl/diagnose.h>
  39 #include "DTransHelper.hxx"
  40
  41 //------------------------------------------------------------------------
  42 // implementation
  43 //------------------------------------------------------------------------
  44
  45 CStgTransferHelper::CStgTransferHelper( sal_Bool bAutoInit,
  46                                         HGLOBAL hGlob,
  47                                         sal_Bool bDelStgOnRelease ) :
  48     m_lpStream( NULL ),
  49     m_bDelStgOnRelease( bDelStgOnRelease )
  50 {
  51     if ( bAutoInit )
  52         init( hGlob, m_bDelStgOnRelease );
  53 }
  54
  55 //------------------------------------------------------------------------
  56 // dtor
  57 //------------------------------------------------------------------------
  58
  59
  60 CStgTransferHelper::~CStgTransferHelper( )
  61 {
  62     if ( m_lpStream )
  63         m_lpStream->Release( );
  64 }
  65
  66 //------------------------------------------------------------------------
  67 // TransferData into the
  68 //------------------------------------------------------------------------
  69
  70 void SAL_CALL CStgTransferHelper::write( const void* lpData, ULONG cb, ULONG* cbWritten )
  71 {
  72     HRESULT hr = E_FAIL;
  73
  74     if ( m_lpStream )
  75         hr = m_lpStream->Write( lpData, cb, cbWritten );
  76
  77     if ( FAILED( hr ) )
  78         throw CStgTransferException( hr );
  79
  80 #if OSL_DEBUG_LEVEL > 0
  81     HGLOBAL hGlob;
  82     hr = GetHGlobalFromStream( m_lpStream, &hGlob );
  83     OSL_ASSERT( SUCCEEDED( hr ) );
  84
  85     /*DWORD dwSize =*/ GlobalSize( hGlob );
  86     /*LPVOID lpdbgData =*/ GlobalLock( hGlob );
  87     GlobalUnlock( hGlob );
  88 #endif
  89 }
  90
  91 //------------------------------------------------------------------------
  92 // read
  93 //------------------------------------------------------------------------
  94
  95 void SAL_CALL CStgTransferHelper::read( LPVOID pv, ULONG cb, ULONG* pcbRead )
  96 {
  97     HRESULT hr = E_FAIL;
  98
  99     if ( m_lpStream )
 100         hr = m_lpStream->Read( pv, cb , pcbRead );
 101
 102     if ( FAILED( hr ) )
 103         throw CStgTransferException( hr );
 104 }
 105
 106 //------------------------------------------------------------------------
 107 // GetHGlobal
 108 //------------------------------------------------------------------------
 109
 110 HGLOBAL SAL_CALL CStgTransferHelper::getHGlobal( ) const
 111 {
 112     OSL_ASSERT( m_lpStream );
 113
 114     HGLOBAL hGlob = NULL;
 115
 116     if ( m_lpStream )
 117     {
 118         HRESULT hr = GetHGlobalFromStream( m_lpStream, &hGlob );
 119         if ( FAILED( hr ) )
 120             hGlob = NULL;
 121     }
 122
 123     return hGlob;
 124 }
 125
 126 //------------------------------------------------------------------------
 127 // getIStream
 128 //------------------------------------------------------------------------
 129
 130 void SAL_CALL CStgTransferHelper::getIStream( LPSTREAM* ppStream )
 131 {
 132     OSL_ASSERT( ppStream );
 133     *ppStream = m_lpStream;
 134     if ( *ppStream )
 135         static_cast< LPUNKNOWN >( *ppStream )->AddRef( );
 136 }
 137
 138 //------------------------------------------------------------------------
 139 // Init
 140 //------------------------------------------------------------------------
 141
 142 void SAL_CALL CStgTransferHelper::init( SIZE_T newSize,
 143                                         sal_uInt32 uiFlags,
 144                                         sal_Bool bDelStgOnRelease )
 145 {
 146     cleanup( );
 147
 148     m_bDelStgOnRelease      = bDelStgOnRelease;
 149
 150     HGLOBAL hGlob = GlobalAlloc( uiFlags, newSize );
 151     if ( NULL == hGlob )
 152         throw CStgTransferException( STG_E_MEDIUMFULL );
 153
 154     HRESULT hr = CreateStreamOnHGlobal( hGlob, m_bDelStgOnRelease, &m_lpStream );
 155     if ( FAILED( hr ) )
 156     {
 157         GlobalFree( hGlob );
 158         m_lpStream = NULL;
 159         throw CStgTransferException( hr );
 160     }
 161
 162 #if OSL_DEBUG_LEVEL > 0
 163     STATSTG statstg;
 164     hr = m_lpStream->Stat( &statstg, STATFLAG_DEFAULT );
 165     OSL_ASSERT( SUCCEEDED( hr ) );
 166 #endif
 167 }
 168
 169 //------------------------------------------------------------------------
 170 // Init
 171 //------------------------------------------------------------------------
 172
 173 void SAL_CALL CStgTransferHelper::init( HGLOBAL hGlob,
 174                                          sal_Bool bDelStgOnRelease )
 175 {
 176     cleanup( );
 177
 178     m_bDelStgOnRelease      = bDelStgOnRelease;
 179
 180     HRESULT hr = CreateStreamOnHGlobal( hGlob, m_bDelStgOnRelease, &m_lpStream );
 181     if ( FAILED( hr ) )
 182         throw CStgTransferException( hr );
 183 }
 184
 185 //------------------------------------------------------------------------
 186 // free the global memory and invalidate the stream pointer
 187 //------------------------------------------------------------------------
 188
 189 void SAL_CALL CStgTransferHelper::cleanup( )
 190 {
 191     if ( m_lpStream && !m_bDelStgOnRelease )
 192     {
 193         HGLOBAL hGlob;
 194         GetHGlobalFromStream( m_lpStream, &hGlob );
 195         GlobalFree( hGlob );
 196     }
 197
 198     if ( m_lpStream )
 199     {
 200         m_lpStream->Release( );
 201         m_lpStream = NULL;
 202     }
 203 }
 204
 205 //------------------------------------------------------------------------
 206 // return the size of memory we point to
 207 //------------------------------------------------------------------------
 208
 209 sal_uInt32 SAL_CALL CStgTransferHelper::memSize( CLIPFORMAT cf ) const
 210 {
 211     DWORD dwSize = 0;
 212
 213     if ( NULL != m_lpStream )
 214     {
 215         HGLOBAL hGlob;
 216         GetHGlobalFromStream( m_lpStream, &hGlob );
 217
 218         if ( CF_TEXT == cf || RegisterClipboardFormat( "HTML Format" ) == cf )
 219         {
 220             sal_Char* pText = static_cast< sal_Char* >( GlobalLock( hGlob ) );
 221             if ( pText )
 222             {
 223                 dwSize = strlen(pText) + 1; // strlen + trailing '\0'
 224                 GlobalUnlock( hGlob );
 225             }
 226         }
 227         else if ( CF_UNICODETEXT == cf )
 228         {
 229             sal_Unicode* pText = static_cast< sal_Unicode* >( GlobalLock( hGlob ) );
 230             if ( pText )
 231             {
 232                 dwSize = rtl_ustr_getLength( pText ) * sizeof( sal_Unicode );
 233                 GlobalUnlock( hGlob );
 234             }
 235         }
 236         else
 237             dwSize = GlobalSize( hGlob );
 238     }
 239
 240     return dwSize;
 241 }
 242