dtrans/source/win32/dtobj/DTransHelper.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 <rtl/ustring.h>
  21 #include <osl/diagnose.h>
  22 #include "DTransHelper.hxx"
  23
  24 //------------------------------------------------------------------------
  25 // implementation
  26 //------------------------------------------------------------------------
  27
  28 CStgTransferHelper::CStgTransferHelper( sal_Bool bAutoInit,
  29                                         HGLOBAL hGlob,
  30                                         sal_Bool bDelStgOnRelease ) :
  31     m_lpStream( NULL ),
  32     m_bDelStgOnRelease( bDelStgOnRelease )
  33 {
  34     if ( bAutoInit )
  35         init( hGlob, m_bDelStgOnRelease );
  36 }
  37
  38 //------------------------------------------------------------------------
  39 // dtor
  40 //------------------------------------------------------------------------
  41
  42
  43 CStgTransferHelper::~CStgTransferHelper( )
  44 {
  45     if ( m_lpStream )
  46         m_lpStream->Release( );
  47 }
  48
  49 //------------------------------------------------------------------------
  50 // TransferData into the
  51 //------------------------------------------------------------------------
  52
  53 void SAL_CALL CStgTransferHelper::write( const void* lpData, ULONG cb, ULONG* cbWritten )
  54 {
  55     HRESULT hr = E_FAIL;
  56
  57     if ( m_lpStream )
  58         hr = m_lpStream->Write( lpData, cb, cbWritten );
  59
  60     if ( FAILED( hr ) )
  61         throw CStgTransferException( hr );
  62
  63 #if OSL_DEBUG_LEVEL > 0
  64     HGLOBAL hGlob;
  65     hr = GetHGlobalFromStream( m_lpStream, &hGlob );
  66     OSL_ASSERT( SUCCEEDED( hr ) );
  67
  68     /*DWORD dwSize =*/ GlobalSize( hGlob );
  69     /*LPVOID lpdbgData =*/ GlobalLock( hGlob );
  70     GlobalUnlock( hGlob );
  71 #endif
  72 }
  73
  74 //------------------------------------------------------------------------
  75 // read
  76 //------------------------------------------------------------------------
  77
  78 void SAL_CALL CStgTransferHelper::read( LPVOID pv, ULONG cb, ULONG* pcbRead )
  79 {
  80     HRESULT hr = E_FAIL;
  81
  82     if ( m_lpStream )
  83         hr = m_lpStream->Read( pv, cb , pcbRead );
  84
  85     if ( FAILED( hr ) )
  86         throw CStgTransferException( hr );
  87 }
  88
  89 //------------------------------------------------------------------------
  90 // GetHGlobal
  91 //------------------------------------------------------------------------
  92
  93 HGLOBAL SAL_CALL CStgTransferHelper::getHGlobal( ) const
  94 {
  95     OSL_ASSERT( m_lpStream );
  96
  97     HGLOBAL hGlob = NULL;
  98
  99     if ( m_lpStream )
 100     {
 101         HRESULT hr = GetHGlobalFromStream( m_lpStream, &hGlob );
 102         if ( FAILED( hr ) )
 103             hGlob = NULL;
 104     }
 105
 106     return hGlob;
 107 }
 108
 109 //------------------------------------------------------------------------
 110 // getIStream
 111 //------------------------------------------------------------------------
 112
 113 void SAL_CALL CStgTransferHelper::getIStream( LPSTREAM* ppStream )
 114 {
 115     OSL_ASSERT( ppStream );
 116     *ppStream = m_lpStream;
 117     if ( *ppStream )
 118         static_cast< LPUNKNOWN >( *ppStream )->AddRef( );
 119 }
 120
 121 //------------------------------------------------------------------------
 122 // Init
 123 //------------------------------------------------------------------------
 124
 125 void SAL_CALL CStgTransferHelper::init( SIZE_T newSize,
 126                                         sal_uInt32 uiFlags,
 127                                         sal_Bool bDelStgOnRelease )
 128 {
 129     cleanup( );
 130
 131     m_bDelStgOnRelease      = bDelStgOnRelease;
 132
 133     HGLOBAL hGlob = GlobalAlloc( uiFlags, newSize );
 134     if ( NULL == hGlob )
 135         throw CStgTransferException( STG_E_MEDIUMFULL );
 136
 137     HRESULT hr = CreateStreamOnHGlobal( hGlob, m_bDelStgOnRelease, &m_lpStream );
 138     if ( FAILED( hr ) )
 139     {
 140         GlobalFree( hGlob );
 141         m_lpStream = NULL;
 142         throw CStgTransferException( hr );
 143     }
 144
 145 #if OSL_DEBUG_LEVEL > 0
 146     STATSTG statstg;
 147     hr = m_lpStream->Stat( &statstg, STATFLAG_DEFAULT );
 148     OSL_ASSERT( SUCCEEDED( hr ) );
 149 #endif
 150 }
 151
 152 //------------------------------------------------------------------------
 153 // Init
 154 //------------------------------------------------------------------------
 155
 156 void SAL_CALL CStgTransferHelper::init( HGLOBAL hGlob,
 157                                          sal_Bool bDelStgOnRelease )
 158 {
 159     cleanup( );
 160
 161     m_bDelStgOnRelease      = bDelStgOnRelease;
 162
 163     HRESULT hr = CreateStreamOnHGlobal( hGlob, m_bDelStgOnRelease, &m_lpStream );
 164     if ( FAILED( hr ) )
 165         throw CStgTransferException( hr );
 166 }
 167
 168 //------------------------------------------------------------------------
 169 // free the global memory and invalidate the stream pointer
 170 //------------------------------------------------------------------------
 171
 172 void SAL_CALL CStgTransferHelper::cleanup( )
 173 {
 174     if ( m_lpStream && !m_bDelStgOnRelease )
 175     {
 176         HGLOBAL hGlob;
 177         GetHGlobalFromStream( m_lpStream, &hGlob );
 178         GlobalFree( hGlob );
 179     }
 180
 181     if ( m_lpStream )
 182     {
 183         m_lpStream->Release( );
 184         m_lpStream = NULL;
 185     }
 186 }
 187
 188 //------------------------------------------------------------------------
 189 // return the size of memory we point to
 190 //------------------------------------------------------------------------
 191
 192 sal_uInt32 SAL_CALL CStgTransferHelper::memSize( CLIPFORMAT cf ) const
 193 {
 194     DWORD dwSize = 0;
 195
 196     if ( NULL != m_lpStream )
 197     {
 198         HGLOBAL hGlob;
 199         GetHGlobalFromStream( m_lpStream, &hGlob );
 200
 201         if ( CF_TEXT == cf || RegisterClipboardFormat( "HTML Format" ) == cf )
 202         {
 203             sal_Char* pText = static_cast< sal_Char* >( GlobalLock( hGlob ) );
 204             if ( pText )
 205             {
 206                 dwSize = strlen(pText) + 1; // strlen + trailing '\0'
 207                 GlobalUnlock( hGlob );
 208             }
 209         }
 210         else if ( CF_UNICODETEXT == cf )
 211         {
 212             sal_Unicode* pText = static_cast< sal_Unicode* >( GlobalLock( hGlob ) );
 213             if ( pText )
 214             {
 215                 dwSize = rtl_ustr_getLength( pText ) * sizeof( sal_Unicode );
 216                 GlobalUnlock( hGlob );
 217             }
 218         }
 219         else
 220             dwSize = GlobalSize( hGlob );
 221     }
 222
 223     return dwSize;
 224 }
 225
 226 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */