dtrans/source/win32/dtobj/APNDataObject.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 "APNDataObject.hxx"
  21 #include <osl/diagnose.h>
  22
  23 #include <systools/win32/comtools.hxx>
  24 #ifdef __MINGW32__
  25 #if defined __uuidof
  26 #undef __uuidof
  27 #endif
  28 #define __uuidof(I) IID_##I
  29 #endif
  30
  31 #define FREE_HGLOB_ON_RELEASE   TRUE
  32 #define KEEP_HGLOB_ON_RELEASE   FALSE
  33
  34 // ctor
  35
  36 CAPNDataObject::CAPNDataObject( IDataObjectPtr rIDataObject ) :
  37     m_rIDataObjectOrg( rIDataObject ),
  38     m_hGlobal( NULL ),
  39     m_nRefCnt( 0 )
  40 {
  41
  42     OSL_ENSURE( m_rIDataObjectOrg.get( ), "constructing CAPNDataObject with empty data object" );
  43
  44     // we marshal the IDataObject interface pointer here so
  45     // that it can be unmarshaled multiple times when this
  46     // class will be used from another apartment
  47     IStreamPtr pStm;
  48     HRESULT hr = CreateStreamOnHGlobal( 0, KEEP_HGLOB_ON_RELEASE, &pStm );
  49
  50     OSL_ENSURE( E_INVALIDARG != hr, "invalid args passed to CreateStreamOnHGlobal" );
  51
  52     if ( SUCCEEDED( hr ) )
  53     {
  54         HRESULT hr_marshal = CoMarshalInterface(
  55             pStm.get(),
  56             __uuidof(IDataObject),
  57             static_cast<LPUNKNOWN>(m_rIDataObjectOrg.get()),
  58             MSHCTX_LOCAL,
  59             NULL,
  60             MSHLFLAGS_TABLEWEAK );
  61
  62         OSL_ENSURE( CO_E_NOTINITIALIZED != hr_marshal, "COM is not initialized" );
  63
  64         // marshalling may fail if COM is not initialized
  65         // for the calling thread which is a program time
  66         // error or because of stream errors which are runtime
  67         // errors for instance E_OUTOFMEMORY etc.
  68
  69         hr = GetHGlobalFromStream(pStm.get(), &m_hGlobal );
  70
  71         OSL_ENSURE( E_INVALIDARG != hr, "invalid stream passed to GetHGlobalFromStream" );
  72
  73         // if the marshalling failed we free the
  74         // global memory again and set m_hGlobal
  75         // to a defined value
  76         if (FAILED(hr_marshal))
  77         {
  78             OSL_FAIL("marshalling failed");
  79
  80             #if OSL_DEBUG_LEVEL > 0
  81             HGLOBAL hGlobal =
  82             #endif
  83                 GlobalFree(m_hGlobal);
  84             OSL_ENSURE(NULL == hGlobal, "GlobalFree failed");
  85             m_hGlobal = NULL;
  86         }
  87     }
  88 }
  89
  90 CAPNDataObject::~CAPNDataObject( )
  91 {
  92     if (m_hGlobal)
  93     {
  94         IStreamPtr pStm;
  95         HRESULT  hr = CreateStreamOnHGlobal(m_hGlobal, FREE_HGLOB_ON_RELEASE, &pStm);
  96
  97         OSL_ENSURE( E_INVALIDARG != hr, "invalid args passed to CreateStreamOnHGlobal" );
  98
  99         if (SUCCEEDED(hr))
 100         {
 101             hr = CoReleaseMarshalData(pStm.get());
 102             OSL_ENSURE(SUCCEEDED(hr), "CoReleaseMarshalData failed");
 103         }
 104     }
 105 }
 106
 107 // IUnknown->QueryInterface
 108
 109 STDMETHODIMP CAPNDataObject::QueryInterface( REFIID iid, LPVOID* ppvObject )
 110 {
 111     OSL_ASSERT( NULL != ppvObject );
 112
 113     if ( NULL == ppvObject )
 114         return E_INVALIDARG;
 115
 116     HRESULT hr = E_NOINTERFACE;
 117     *ppvObject = NULL;
 118
 119     if ( ( __uuidof( IUnknown ) == iid ) || ( __uuidof( IDataObject ) == iid ) )
 120     {
 121         *ppvObject = static_cast< IUnknown* >( this );
 122         ( (LPUNKNOWN)*ppvObject )->AddRef( );
 123         hr = S_OK;
 124     }
 125
 126     return hr;
 127 }
 128
 129 // IUnknown->AddRef
 130
 131 STDMETHODIMP_(ULONG) CAPNDataObject::AddRef( )
 132 {
 133     return static_cast< ULONG >( InterlockedIncrement( &m_nRefCnt ) );
 134 }
 135
 136 // IUnknown->Release
 137
 138 STDMETHODIMP_(ULONG) CAPNDataObject::Release( )
 139 {
 140     // we need a helper variable because it's not allowed to access
 141     // a member variable after an object is destroyed
 142     ULONG nRefCnt = static_cast< ULONG >( InterlockedDecrement( &m_nRefCnt ) );
 143
 144     if ( 0 == nRefCnt )
 145         delete this;
 146
 147     return nRefCnt;
 148 }
 149
 150 // IDataObject->GetData
 151
 152 STDMETHODIMP CAPNDataObject::GetData( LPFORMATETC pFormatetc, LPSTGMEDIUM pmedium )
 153 {
 154     HRESULT hr = m_rIDataObjectOrg->GetData( pFormatetc, pmedium );
 155
 156     if (RPC_E_WRONG_THREAD == hr)
 157     {
 158         IDataObjectPtr pIDOTmp;
 159         hr = MarshalIDataObjectIntoCurrentApartment(&pIDOTmp);
 160
 161         if (SUCCEEDED(hr))
 162             hr = pIDOTmp->GetData(pFormatetc, pmedium);
 163     }
 164     return hr;
 165 }
 166
 167 // IDataObject->EnumFormatEtc
 168
 169 STDMETHODIMP CAPNDataObject::EnumFormatEtc( DWORD dwDirection, IEnumFORMATETC** ppenumFormatetc )
 170 {
 171     HRESULT hr = m_rIDataObjectOrg->EnumFormatEtc(dwDirection, ppenumFormatetc);
 172
 173     if (RPC_E_WRONG_THREAD == hr)
 174     {
 175         IDataObjectPtr pIDOTmp;
 176         hr = MarshalIDataObjectIntoCurrentApartment(&pIDOTmp);
 177
 178         if (SUCCEEDED(hr))
 179             hr = pIDOTmp->EnumFormatEtc(dwDirection, ppenumFormatetc);
 180     }
 181     return hr;
 182 }
 183
 184 // IDataObject->QueryGetData
 185
 186 STDMETHODIMP CAPNDataObject::QueryGetData( LPFORMATETC pFormatetc )
 187 {
 188     HRESULT hr = m_rIDataObjectOrg->QueryGetData( pFormatetc );
 189
 190     if (RPC_E_WRONG_THREAD == hr)
 191     {
 192         IDataObjectPtr pIDOTmp;
 193         hr = MarshalIDataObjectIntoCurrentApartment( &pIDOTmp );
 194
 195         if (SUCCEEDED(hr))
 196             hr = pIDOTmp->QueryGetData(pFormatetc);
 197     }
 198     return hr;
 199 }
 200
 201 // IDataObject->GetDataHere
 202
 203 STDMETHODIMP CAPNDataObject::GetDataHere( LPFORMATETC pFormatetc, LPSTGMEDIUM pmedium )
 204 {
 205     HRESULT hr = m_rIDataObjectOrg->GetDataHere(pFormatetc, pmedium);
 206
 207     if (RPC_E_WRONG_THREAD == hr)
 208     {
 209         IDataObjectPtr pIDOTmp;
 210         hr = MarshalIDataObjectIntoCurrentApartment(&pIDOTmp);
 211
 212         if (SUCCEEDED(hr))
 213             hr = pIDOTmp->GetDataHere(pFormatetc, pmedium);
 214     }
 215     return hr;
 216 }
 217
 218 // IDataObject->GetCanonicalFormatEtc
 219
 220 STDMETHODIMP CAPNDataObject::GetCanonicalFormatEtc(LPFORMATETC pFormatectIn, LPFORMATETC pFormatetcOut)
 221 {
 222     HRESULT hr = m_rIDataObjectOrg->GetCanonicalFormatEtc( pFormatectIn, pFormatetcOut );
 223
 224     if (RPC_E_WRONG_THREAD == hr)
 225     {
 226         IDataObjectPtr pIDOTmp;
 227         hr = MarshalIDataObjectIntoCurrentApartment(&pIDOTmp);
 228
 229         if (SUCCEEDED(hr))
 230             hr = pIDOTmp->GetCanonicalFormatEtc(pFormatectIn, pFormatetcOut);
 231     }
 232     return hr;
 233 }
 234
 235 // IDataObject->SetData
 236
 237 STDMETHODIMP CAPNDataObject::SetData( LPFORMATETC pFormatetc, LPSTGMEDIUM pmedium, BOOL fRelease )
 238 {
 239     HRESULT hr = m_rIDataObjectOrg->SetData( pFormatetc, pmedium, fRelease );
 240
 241     if (RPC_E_WRONG_THREAD == hr)
 242     {
 243         IDataObjectPtr pIDOTmp;
 244         hr = MarshalIDataObjectIntoCurrentApartment(&pIDOTmp);
 245
 246         if (SUCCEEDED(hr))
 247             hr = pIDOTmp->SetData(pFormatetc, pmedium, fRelease);
 248     }
 249     return hr;
 250 }
 251
 252 // IDataObject->DAdvise
 253
 254 STDMETHODIMP CAPNDataObject::DAdvise( LPFORMATETC pFormatetc, DWORD advf, LPADVISESINK pAdvSink, DWORD * pdwConnection )
 255 {
 256     HRESULT hr = m_rIDataObjectOrg->DAdvise(pFormatetc, advf, pAdvSink, pdwConnection);
 257
 258     if (RPC_E_WRONG_THREAD == hr)
 259     {
 260         IDataObjectPtr pIDOTmp;
 261         hr = MarshalIDataObjectIntoCurrentApartment(&pIDOTmp);
 262
 263         if (SUCCEEDED(hr))
 264             hr = pIDOTmp->DAdvise(pFormatetc, advf, pAdvSink, pdwConnection);
 265     }
 266     return hr;
 267 }
 268
 269 // IDataObject->DUnadvise
 270
 271 STDMETHODIMP CAPNDataObject::DUnadvise( DWORD dwConnection )
 272 {
 273     HRESULT hr = m_rIDataObjectOrg->DUnadvise( dwConnection );
 274
 275     if (RPC_E_WRONG_THREAD == hr)
 276     {
 277         IDataObjectPtr pIDOTmp;
 278         hr = MarshalIDataObjectIntoCurrentApartment(&pIDOTmp);
 279
 280         if (SUCCEEDED(hr))
 281             hr = pIDOTmp->DUnadvise(dwConnection);
 282     }
 283     return hr;
 284 }
 285
 286 // IDataObject->EnumDAdvise
 287
 288 STDMETHODIMP CAPNDataObject::EnumDAdvise( LPENUMSTATDATA * ppenumAdvise )
 289 {
 290     HRESULT hr = m_rIDataObjectOrg->EnumDAdvise(ppenumAdvise);
 291
 292     if (RPC_E_WRONG_THREAD == hr)
 293     {
 294         IDataObjectPtr pIDOTmp;
 295         hr = MarshalIDataObjectIntoCurrentApartment(&pIDOTmp);
 296
 297         if (SUCCEEDED(hr))
 298             hr = pIDOTmp->EnumDAdvise(ppenumAdvise);
 299     }
 300     return hr;
 301 }
 302
 303 // for our convenience
 304
 305 CAPNDataObject::operator IDataObject*( )
 306 {
 307     return static_cast< IDataObject* >( this );
 308 }
 309
 310 // helper function
 311
 312 HRESULT CAPNDataObject::MarshalIDataObjectIntoCurrentApartment( IDataObject** ppIDataObj )
 313 {
 314     OSL_ASSERT(NULL != ppIDataObj);
 315
 316     *ppIDataObj = NULL;
 317     HRESULT hr = E_FAIL;
 318
 319     if (m_hGlobal)
 320     {
 321         IStreamPtr pStm;
 322         hr = CreateStreamOnHGlobal(m_hGlobal, KEEP_HGLOB_ON_RELEASE, &pStm);
 323
 324         OSL_ENSURE(E_INVALIDARG != hr, "CreateStreamOnHGlobal with invalid args called");
 325
 326         if (SUCCEEDED(hr))
 327         {
 328             hr = CoUnmarshalInterface(pStm.get(), __uuidof(IDataObject), (void**)ppIDataObj);
 329             OSL_ENSURE(CO_E_NOTINITIALIZED != hr, "COM is not initialized");
 330         }
 331     }
 332     return hr;
 333 }
 334
 335 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */