sal/workben/clipboardwben/testcopy/XTDataObject.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 <osl/diagnose.h>
  21
  22 #include "XTDataObject.hxx"
  23
  24 #include <windows.h>
  25 #include <ole2.h>
  26 #include <memory>
  27
  28 //------------------------------------------------------------------------
  29 // namespace directives
  30 //------------------------------------------------------------------------
  31
  32
  33 //============================================================================
  34 // OTWrapperDataObject
  35 //============================================================================
  36
  37 //------------------------------------------------------------------------
  38 // ctor
  39 //------------------------------------------------------------------------
  40
  41 CXTDataObject::CXTDataObject( LONG nRefCntInitVal ) :
  42     m_nRefCnt( nRefCntInitVal )
  43 {
  44 }
  45
  46 //------------------------------------------------------------------------
  47 // dtor
  48 //------------------------------------------------------------------------
  49
  50 CXTDataObject::~CXTDataObject( )
  51 {
  52 }
  53
  54 //------------------------------------------------------------------------
  55 // IUnknown->QueryInterface
  56 //------------------------------------------------------------------------
  57
  58 STDMETHODIMP CXTDataObject::QueryInterface( REFIID iid, LPVOID* ppvObject )
  59 {
  60     OSL_ASSERT( NULL != ppvObject );
  61
  62     if ( NULL == ppvObject )
  63         return E_INVALIDARG;
  64
  65     HRESULT hr = E_NOINTERFACE;
  66
  67     *ppvObject = NULL;
  68
  69     if ( ( __uuidof( IUnknown ) == iid ) || ( __uuidof( IDataObject ) == iid ) )
  70     {
  71         *ppvObject = static_cast< IUnknown* >( this );
  72         ( (LPUNKNOWN)*ppvObject )->AddRef( );
  73         hr = S_OK;
  74     }
  75
  76     return hr;
  77 }
  78
  79 //------------------------------------------------------------------------
  80 // IUnknown->AddRef
  81 //------------------------------------------------------------------------
  82
  83 STDMETHODIMP_(ULONG) CXTDataObject::AddRef( )
  84 {
  85     return static_cast< ULONG >( InterlockedIncrement( &m_nRefCnt ) );
  86 }
  87
  88 //------------------------------------------------------------------------
  89 // IUnknown->Release
  90 //------------------------------------------------------------------------
  91
  92 STDMETHODIMP_(ULONG) CXTDataObject::Release( )
  93 {
  94     // we need a helper variable because it's
  95     // not allowed to access a member variable
  96     // after an object is destroyed
  97     ULONG nRefCnt = static_cast< ULONG >( InterlockedDecrement( &m_nRefCnt ) );
  98
  99     if ( 0 == nRefCnt )
 100     {
 101         delete this;
 102     }
 103
 104     return nRefCnt;
 105 }
 106
 107 //------------------------------------------------------------------------
 108 // IDataObject->GetData
 109 // warning: 'goto' ahead (to easy error handling without using exceptions)
 110 //------------------------------------------------------------------------
 111
 112 STDMETHODIMP CXTDataObject::GetData(LPFORMATETC pFormatetc, LPSTGMEDIUM pmedium )
 113 {
 114     OSL_ASSERT( ( NULL != pFormatetc ) &&
 115                 ( !IsBadReadPtr( (LPVOID)pFormatetc, sizeof( FORMATETC ) ) ) );
 116     OSL_ASSERT( ( NULL != pmedium ) &&
 117                 ( !IsBadWritePtr( (LPVOID)pmedium, sizeof( STGMEDIUM ) ) ) );
 118
 119     if ( ( NULL == pFormatetc ) || ( NULL == pmedium ) )
 120         return E_INVALIDARG;
 121
 122     HRESULT hr = E_FAIL;
 123
 124     if ( CF_TEXT == pFormatetc->cfFormat )
 125     {
 126         char     buff[] = "Hello World, How are you!";
 127         LPSTREAM lpStream;
 128
 129         hr = CreateStreamOnHGlobal( NULL, FALSE, &lpStream );
 130         if ( SUCCEEDED( hr ) )
 131         {
 132             hr = lpStream->Write( buff, sizeof( buff ) * sizeof( char ), NULL );
 133             if ( SUCCEEDED( hr ) )
 134             {
 135                 HGLOBAL hGlob;
 136
 137                 GetHGlobalFromStream( lpStream, &hGlob );
 138
 139                 pmedium->tymed          = TYMED_HGLOBAL;
 140                 pmedium->hGlobal        = hGlob;
 141                 pmedium->pUnkForRelease = NULL;
 142             }
 143             lpStream->Release( );
 144             hr = S_OK;
 145         }
 146         else
 147         {
 148             pmedium->tymed = TYMED_NULL;
 149         }
 150     }
 151     else if ( CF_UNICODETEXT == pFormatetc->cfFormat )
 152     {
 153         WCHAR     buff[] = L"Hello World, How are you!";
 154         LPSTREAM lpStream;
 155
 156         hr = CreateStreamOnHGlobal( NULL, FALSE, &lpStream );
 157         if ( SUCCEEDED( hr ) )
 158         {
 159             hr = lpStream->Write( buff, sizeof( buff ) * sizeof( WCHAR ), NULL );
 160             if ( SUCCEEDED( hr ) )
 161             {
 162                 HGLOBAL hGlob;
 163
 164                 GetHGlobalFromStream( lpStream, &hGlob );
 165
 166                 pmedium->tymed          = TYMED_HGLOBAL;
 167                 pmedium->hGlobal        = hGlob;
 168                 pmedium->pUnkForRelease = NULL;
 169             }
 170             lpStream->Release( );
 171             hr = S_OK;
 172         }
 173         else
 174         {
 175             pmedium->tymed = TYMED_NULL;
 176         }
 177     }
 178
 179     return hr;
 180 }
 181
 182 //------------------------------------------------------------------------
 183 // IDataObject->EnumFormatEtc
 184 //------------------------------------------------------------------------
 185
 186 STDMETHODIMP CXTDataObject::EnumFormatEtc( DWORD dwDirection, IEnumFORMATETC** ppenumFormatetc )
 187 {
 188     if ( ( NULL == ppenumFormatetc ) || ( DATADIR_SET == dwDirection ) )
 189         return E_INVALIDARG;
 190
 191     *ppenumFormatetc = NULL;
 192
 193     HRESULT hr = E_FAIL;
 194
 195     if ( DATADIR_GET == dwDirection )
 196     {
 197         *ppenumFormatetc = new CEnumFormatEtc( this );
 198         static_cast< LPUNKNOWN >( *ppenumFormatetc )->AddRef( );
 199         hr = S_OK;
 200     }
 201
 202     return hr;
 203 }
 204
 205 //------------------------------------------------------------------------
 206 // IDataObject->QueryGetData
 207 //------------------------------------------------------------------------
 208
 209 STDMETHODIMP CXTDataObject::QueryGetData( LPFORMATETC pFormatetc )
 210 {
 211     return E_NOTIMPL;
 212 }
 213
 214 //------------------------------------------------------------------------
 215 // IDataObject->GetDataHere
 216 //------------------------------------------------------------------------
 217
 218 STDMETHODIMP CXTDataObject::GetDataHere( LPFORMATETC, LPSTGMEDIUM )
 219 {
 220     return E_NOTIMPL;
 221 }
 222
 223 //------------------------------------------------------------------------
 224 // IDataObject->GetCanonicalFormatEtc
 225 //------------------------------------------------------------------------
 226
 227 STDMETHODIMP CXTDataObject::GetCanonicalFormatEtc( LPFORMATETC, LPFORMATETC )
 228 {
 229     return E_NOTIMPL;
 230 }
 231
 232 //------------------------------------------------------------------------
 233 // IDataObject->SetData
 234 //------------------------------------------------------------------------
 235
 236 STDMETHODIMP CXTDataObject::SetData( LPFORMATETC, LPSTGMEDIUM, BOOL )
 237 {
 238     return E_NOTIMPL;
 239 }
 240
 241 //------------------------------------------------------------------------
 242 // IDataObject->DAdvise
 243 //------------------------------------------------------------------------
 244
 245 STDMETHODIMP CXTDataObject::DAdvise( LPFORMATETC, DWORD, LPADVISESINK, DWORD * )
 246 {
 247     return E_NOTIMPL;
 248 }
 249
 250 //------------------------------------------------------------------------
 251 // IDataObject->DUnadvise
 252 //------------------------------------------------------------------------
 253
 254 STDMETHODIMP CXTDataObject::DUnadvise( DWORD )
 255 {
 256     return E_NOTIMPL;
 257 }
 258
 259 //------------------------------------------------------------------------
 260 // IDataObject->EnumDAdvise
 261 //------------------------------------------------------------------------
 262
 263 STDMETHODIMP CXTDataObject::EnumDAdvise( LPENUMSTATDATA * )
 264 {
 265     return E_NOTIMPL;
 266 }
 267
 268 //------------------------------------------------------------------------
 269 // for our convenience
 270 //------------------------------------------------------------------------
 271
 272 CXTDataObject::operator IDataObject*( )
 273 {
 274     return static_cast< IDataObject* >( this );
 275 }
 276
 277
 278 //============================================================================
 279 // CEnumFormatEtc
 280 //============================================================================
 281
 282
 283 //----------------------------------------------------------------------------
 284 // ctor
 285 //----------------------------------------------------------------------------
 286
 287 CEnumFormatEtc::CEnumFormatEtc( LPUNKNOWN pUnkDataObj ) :
 288     m_nRefCnt( 0 ),
 289     m_pUnkDataObj( pUnkDataObj ),
 290     m_nCurrentPos( 0 )
 291 {
 292     m_cfFormats[0] = CF_UNICODETEXT;
 293     m_cfFormats[1] = CF_TEXT;
 294 }
 295
 296 //----------------------------------------------------------------------------
 297 // dtor
 298 //----------------------------------------------------------------------------
 299
 300 CEnumFormatEtc::~CEnumFormatEtc( )
 301 {
 302 }
 303
 304 //----------------------------------------------------------------------------
 305 // IUnknown->QueryInterface
 306 //----------------------------------------------------------------------------
 307
 308 STDMETHODIMP CEnumFormatEtc::QueryInterface( REFIID iid, LPVOID* ppvObject )
 309 {
 310     if ( NULL == ppvObject )
 311         return E_INVALIDARG;
 312
 313     HRESULT hr = E_NOINTERFACE;
 314
 315     *ppvObject = NULL;
 316
 317     if ( ( __uuidof( IUnknown ) == iid ) || ( __uuidof( IEnumFORMATETC ) == iid ) )
 318     {
 319         *ppvObject = static_cast< IUnknown* >( this );
 320         static_cast< LPUNKNOWN >( *ppvObject )->AddRef( );
 321         hr = S_OK;
 322     }
 323
 324     return hr;
 325 }
 326
 327 //----------------------------------------------------------------------------
 328 // IUnknown->AddRef
 329 //----------------------------------------------------------------------------
 330
 331 STDMETHODIMP_(ULONG) CEnumFormatEtc::AddRef( )
 332 {
 333     // keep the dataobject alive
 334     m_pUnkDataObj->AddRef( );
 335     return InterlockedIncrement( &m_nRefCnt );
 336 }
 337
 338 //----------------------------------------------------------------------------
 339 // IUnknown->Release
 340 //----------------------------------------------------------------------------
 341
 342 STDMETHODIMP_(ULONG) CEnumFormatEtc::Release( )
 343 {
 344     // release the outer dataobject
 345     m_pUnkDataObj->Release( );
 346
 347     // we need a helper variable because it's
 348     // not allowed to access a member variable
 349     // after an object is destroyed
 350     ULONG nRefCnt = InterlockedDecrement( &m_nRefCnt );
 351     if ( 0 == nRefCnt )
 352         delete this;
 353
 354     return nRefCnt;
 355 }
 356
 357 //----------------------------------------------------------------------------
 358 // IEnumFORMATETC->Next
 359 //----------------------------------------------------------------------------
 360
 361 STDMETHODIMP CEnumFormatEtc::Next( ULONG celt, LPFORMATETC rgelt, ULONG* pceltFetched )
 362 {
 363     OSL_ASSERT( ( ( celt > 0 ) && ( NULL != rgelt ) ) ||
 364                 ( ( 0 == celt ) && ( NULL == rgelt ) ) );
 365
 366     if ( ( 0 != celt ) && ( NULL == rgelt ) )
 367         return E_INVALIDARG;
 368
 369     ULONG   ulFetched = 0;
 370     ULONG   ulToFetch = celt;
 371     HRESULT hr        = S_FALSE;
 372
 373     while( ( m_nCurrentPos < sizeof( m_cfFormats ) ) && ( ulToFetch > 0 ) )
 374     {
 375         OSL_ASSERT( !IsBadWritePtr( (LPVOID)rgelt, sizeof( FORMATETC ) ) );
 376
 377         rgelt->cfFormat = m_cfFormats[m_nCurrentPos];
 378         rgelt->ptd      = NULL;
 379         rgelt->dwAspect = DVASPECT_CONTENT;
 380         rgelt->lindex   = -1;
 381         rgelt->tymed    = TYMED_HGLOBAL;
 382
 383         ++m_nCurrentPos;
 384         ++rgelt;
 385         --ulToFetch;
 386         ++ulFetched;
 387     }
 388
 389     if ( ulFetched == celt )
 390         hr = S_OK;
 391
 392     if ( NULL != pceltFetched )
 393     {
 394         OSL_ASSERT( !IsBadWritePtr( (LPVOID)pceltFetched, sizeof( ULONG ) ) );
 395         *pceltFetched = ulFetched;
 396     }
 397
 398     return hr;
 399 }
 400
 401 //----------------------------------------------------------------------------
 402 // IEnumFORMATETC->Skip
 403 //----------------------------------------------------------------------------
 404
 405 STDMETHODIMP CEnumFormatEtc::Skip( ULONG celt )
 406 {
 407     HRESULT hr = S_FALSE;
 408
 409     if ( ( m_nCurrentPos + celt ) < sizeof( m_cfFormats ) )
 410     {
 411         m_nCurrentPos += celt;
 412         hr = S_OK;
 413     }
 414
 415     return hr;
 416 }
 417
 418 //----------------------------------------------------------------------------
 419 // IEnumFORMATETC->Reset
 420 //----------------------------------------------------------------------------
 421
 422 STDMETHODIMP CEnumFormatEtc::Reset( )
 423 {
 424     m_nCurrentPos = 0;
 425     return S_OK;
 426 }
 427
 428 //----------------------------------------------------------------------------
 429 // IEnumFORMATETC->Clone
 430 //----------------------------------------------------------------------------
 431
 432 STDMETHODIMP CEnumFormatEtc::Clone( IEnumFORMATETC** ppenum )
 433 {
 434     OSL_ASSERT( NULL != ppenum );
 435
 436     if ( NULL == ppenum )
 437         return E_INVALIDARG;
 438
 439     HRESULT hr = E_FAIL;
 440
 441     *ppenum = NULL;
 442
 443     CEnumFormatEtc* pCEnumFEtc = new CEnumFormatEtc( m_pUnkDataObj );
 444     if ( NULL != pCEnumFEtc )
 445     {
 446         pCEnumFEtc->m_nCurrentPos = m_nCurrentPos;
 447         *ppenum = static_cast< IEnumFORMATETC* >( pCEnumFEtc );
 448         static_cast< LPUNKNOWN >( *ppenum )->AddRef( );
 449         hr = NOERROR;
 450     }
 451
 452     return hr;
 453 }
 454
 455 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */