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