Home | History | Annotate | Download | only in win 
      1 /*
      2  * Copyright (C) 2007 Apple Inc.  All rights reserved.
      3  *
      4  * Redistribution and use in source and binary forms, with or without
      5  * modification, are permitted provided that the following conditions
      6  * are met:
      7  * 1. Redistributions of source code must retain the above copyright
      8  *    notice, this list of conditions and the following disclaimer.
      9  * 2. Redistributions in binary form must reproduce the above copyright
     10  *    notice, this list of conditions and the following disclaimer in the
     11  *    documentation and/or other materials provided with the distribution.
     12  *
     13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
     14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
     17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
     18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
     19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
     20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
     21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     24  */
     25 
     26 #include "config.h"
     27 #include "WCDataObject.h"
     28 
     29 #include "ClipboardUtilitiesWin.h"
     30 #include "DragData.h"
     31 #include "PlatformString.h"
     32 
     33 namespace WebCore {
     34 
     35 class WCEnumFormatEtc : public IEnumFORMATETC
     36 {
     37 public:
     38     WCEnumFormatEtc(const Vector<FORMATETC>& formats);
     39     WCEnumFormatEtc(const Vector<FORMATETC*>& formats);
     40 
     41     //IUnknown members
     42     STDMETHOD(QueryInterface)(REFIID, void**);
     43     STDMETHOD_(ULONG, AddRef)(void);
     44     STDMETHOD_(ULONG, Release)(void);
     45 
     46     //IEnumFORMATETC members
     47     STDMETHOD(Next)(ULONG, LPFORMATETC, ULONG*);
     48     STDMETHOD(Skip)(ULONG);
     49     STDMETHOD(Reset)(void);
     50     STDMETHOD(Clone)(IEnumFORMATETC**);
     51 
     52 private:
     53     long m_ref;
     54     Vector<FORMATETC>  m_formats;
     55     size_t m_current;
     56 };
     57 
     58 
     59 
     60 WCEnumFormatEtc::WCEnumFormatEtc(const Vector<FORMATETC>& formats)
     61 : m_ref(1)
     62 , m_current(0)
     63 {
     64     for(size_t i = 0; i < formats.size(); ++i)
     65         m_formats.append(formats[i]);
     66 }
     67 
     68 WCEnumFormatEtc::WCEnumFormatEtc(const Vector<FORMATETC*>& formats)
     69 : m_ref(1)
     70 , m_current(0)
     71 {
     72     for(size_t i = 0; i < formats.size(); ++i)
     73         m_formats.append(*formats[i]);
     74 }
     75 
     76 STDMETHODIMP  WCEnumFormatEtc::QueryInterface(REFIID riid, void** ppvObject)
     77 {
     78     *ppvObject = 0;
     79     if (IsEqualIID(riid, IID_IUnknown) || IsEqualIID(riid, IID_IEnumFORMATETC)) {
     80         *ppvObject = this;
     81         AddRef();
     82         return S_OK;
     83     }
     84 
     85     return E_NOINTERFACE;
     86 }
     87 
     88 STDMETHODIMP_(ULONG) WCEnumFormatEtc::AddRef(void)
     89 {
     90     return InterlockedIncrement(&m_ref);
     91 }
     92 
     93 STDMETHODIMP_(ULONG) WCEnumFormatEtc::Release(void)
     94 {
     95     long c = InterlockedDecrement(&m_ref);
     96     if (c == 0)
     97         delete this;
     98     return c;
     99 }
    100 
    101 STDMETHODIMP WCEnumFormatEtc::Next(ULONG celt, LPFORMATETC lpFormatEtc, ULONG* pceltFetched)
    102 {
    103     if(pceltFetched != 0)
    104         *pceltFetched=0;
    105 
    106     ULONG cReturn = celt;
    107 
    108     if(celt <= 0 || lpFormatEtc == 0 || m_current >= m_formats.size())
    109         return S_FALSE;
    110 
    111     if(pceltFetched == 0 && celt != 1) // pceltFetched can be 0 only for 1 item request
    112         return S_FALSE;
    113 
    114     while (m_current < m_formats.size() && cReturn > 0) {
    115         *lpFormatEtc++ = m_formats[m_current++];
    116         --cReturn;
    117     }
    118     if (pceltFetched != 0)
    119         *pceltFetched = celt - cReturn;
    120 
    121     return (cReturn == 0) ? S_OK : S_FALSE;
    122 }
    123 
    124 STDMETHODIMP WCEnumFormatEtc::Skip(ULONG celt)
    125 {
    126     if((m_current + int(celt)) >= m_formats.size())
    127         return S_FALSE;
    128     m_current += celt;
    129     return S_OK;
    130 }
    131 
    132 STDMETHODIMP WCEnumFormatEtc::Reset(void)
    133 {
    134     m_current = 0;
    135     return S_OK;
    136 }
    137 
    138 STDMETHODIMP WCEnumFormatEtc::Clone(IEnumFORMATETC** ppCloneEnumFormatEtc)
    139 {
    140     if(!ppCloneEnumFormatEtc)
    141         return E_POINTER;
    142 
    143     WCEnumFormatEtc *newEnum = new WCEnumFormatEtc(m_formats);
    144     if(!newEnum)
    145         return E_OUTOFMEMORY;
    146 
    147     newEnum->AddRef();
    148     newEnum->m_current = m_current;
    149     *ppCloneEnumFormatEtc = newEnum;
    150     return S_OK;
    151 }
    152 
    153 
    154 
    155 //////////////////////////////////////////////////////////////////////////
    156 
    157 HRESULT WCDataObject::createInstance(WCDataObject** result)
    158 {
    159     if (!result)
    160         return E_POINTER;
    161     *result = new WCDataObject();
    162     return S_OK;
    163 }
    164 
    165 HRESULT WCDataObject::createInstance(WCDataObject** result, const DragDataMap& dataMap)
    166 {
    167     if (!result)
    168         return E_POINTER;
    169     *result = new WCDataObject;
    170 
    171     for (DragDataMap::const_iterator it = dataMap.begin(); it != dataMap.end(); ++it)
    172         setClipboardData(*result, it->first, it->second);
    173     return S_OK;
    174 }
    175 
    176 WCDataObject::WCDataObject()
    177 : m_ref(1)
    178 {
    179 }
    180 
    181 WCDataObject::~WCDataObject()
    182 {
    183     for(size_t i = 0; i < m_medium.size(); ++i) {
    184         ReleaseStgMedium(m_medium[i]);
    185         delete m_medium[i];
    186     }
    187     WTF::deleteAllValues(m_formats);
    188 }
    189 
    190 STDMETHODIMP WCDataObject::QueryInterface(REFIID riid,void** ppvObject)
    191 {
    192     *ppvObject = 0;
    193     if (IsEqualIID(riid, IID_IUnknown) ||
    194         IsEqualIID(riid, IID_IDataObject)) {
    195         *ppvObject=this;
    196     }
    197     if (*ppvObject) {
    198         AddRef();
    199         return S_OK;
    200     }
    201     return E_NOINTERFACE;
    202 }
    203 
    204 STDMETHODIMP_(ULONG) WCDataObject::AddRef( void)
    205 {
    206     return InterlockedIncrement(&m_ref);
    207 }
    208 
    209 STDMETHODIMP_(ULONG) WCDataObject::Release( void)
    210 {
    211     long c = InterlockedDecrement(&m_ref);
    212     if (c == 0)
    213         delete this;
    214     return c;
    215 }
    216 
    217 STDMETHODIMP WCDataObject::GetData(FORMATETC* pformatetcIn, STGMEDIUM* pmedium)
    218 {
    219     if(!pformatetcIn || !pmedium)
    220         return E_POINTER;
    221     pmedium->hGlobal = 0;
    222 
    223     for(size_t i=0; i < m_formats.size(); ++i) {
    224         if(/*pformatetcIn->tymed & m_formats[i]->tymed &&*/     // tymed can be 0 (TYMED_NULL) - but it can have a medium that contains an pUnkForRelease
    225             pformatetcIn->lindex == m_formats[i]->lindex &&
    226             pformatetcIn->dwAspect == m_formats[i]->dwAspect &&
    227             pformatetcIn->cfFormat == m_formats[i]->cfFormat) {
    228             CopyMedium(pmedium, m_medium[i], m_formats[i]);
    229             return S_OK;
    230         }
    231     }
    232     return DV_E_FORMATETC;
    233 }
    234 
    235 STDMETHODIMP WCDataObject::GetDataHere(FORMATETC*, STGMEDIUM*)
    236 {
    237     return E_NOTIMPL;
    238 }
    239 
    240 STDMETHODIMP WCDataObject::QueryGetData(FORMATETC* pformatetc)
    241 {
    242     if(!pformatetc)
    243         return E_POINTER;
    244 
    245     if (!(DVASPECT_CONTENT & pformatetc->dwAspect))
    246         return (DV_E_DVASPECT);
    247     HRESULT  hr = DV_E_TYMED;
    248     for(size_t i = 0; i < m_formats.size(); ++i) {
    249         if(pformatetc->tymed & m_formats[i]->tymed) {
    250             if(pformatetc->cfFormat == m_formats[i]->cfFormat)
    251                 return S_OK;
    252             else
    253                 hr = DV_E_CLIPFORMAT;
    254         }
    255         else
    256             hr = DV_E_TYMED;
    257     }
    258     return hr;
    259 }
    260 
    261 STDMETHODIMP WCDataObject::GetCanonicalFormatEtc(FORMATETC*, FORMATETC*)
    262 {
    263     return DATA_S_SAMEFORMATETC;
    264 }
    265 
    266 STDMETHODIMP WCDataObject::SetData(FORMATETC* pformatetc, STGMEDIUM* pmedium, BOOL fRelease)
    267 {
    268     if(!pformatetc || !pmedium)
    269         return E_POINTER;
    270 
    271     FORMATETC* fetc=new FORMATETC;
    272     if (!fetc)
    273         return E_OUTOFMEMORY;
    274 
    275     STGMEDIUM* pStgMed = new STGMEDIUM;
    276 
    277     if(!pStgMed) {
    278         delete fetc;
    279         return E_OUTOFMEMORY;
    280     }
    281 
    282     ZeroMemory(fetc,sizeof(FORMATETC));
    283     ZeroMemory(pStgMed,sizeof(STGMEDIUM));
    284 
    285     *fetc = *pformatetc;
    286     m_formats.append(fetc);
    287 
    288     if(fRelease)
    289         *pStgMed = *pmedium;
    290     else
    291         CopyMedium(pStgMed, pmedium, pformatetc);
    292     m_medium.append(pStgMed);
    293 
    294     return S_OK;
    295 }
    296 
    297 void WCDataObject::CopyMedium(STGMEDIUM* pMedDest, STGMEDIUM* pMedSrc, FORMATETC* pFmtSrc)
    298 {
    299     switch(pMedSrc->tymed)
    300     {
    301 #if !OS(WINCE)
    302     case TYMED_HGLOBAL:
    303         pMedDest->hGlobal = (HGLOBAL)OleDuplicateData(pMedSrc->hGlobal,pFmtSrc->cfFormat, 0);
    304         break;
    305     case TYMED_GDI:
    306         pMedDest->hBitmap = (HBITMAP)OleDuplicateData(pMedSrc->hBitmap,pFmtSrc->cfFormat, 0);
    307         break;
    308     case TYMED_MFPICT:
    309         pMedDest->hMetaFilePict = (HMETAFILEPICT)OleDuplicateData(pMedSrc->hMetaFilePict,pFmtSrc->cfFormat, 0);
    310         break;
    311     case TYMED_ENHMF:
    312         pMedDest->hEnhMetaFile = (HENHMETAFILE)OleDuplicateData(pMedSrc->hEnhMetaFile,pFmtSrc->cfFormat, 0);
    313         break;
    314     case TYMED_FILE:
    315         pMedSrc->lpszFileName = (LPOLESTR)OleDuplicateData(pMedSrc->lpszFileName,pFmtSrc->cfFormat, 0);
    316         break;
    317 #endif
    318     case TYMED_ISTREAM:
    319         pMedDest->pstm = pMedSrc->pstm;
    320         pMedSrc->pstm->AddRef();
    321         break;
    322     case TYMED_ISTORAGE:
    323         pMedDest->pstg = pMedSrc->pstg;
    324         pMedSrc->pstg->AddRef();
    325         break;
    326     default:
    327         break;
    328     }
    329     pMedDest->tymed = pMedSrc->tymed;
    330     pMedDest->pUnkForRelease = 0;
    331     if(pMedSrc->pUnkForRelease != 0) {
    332         pMedDest->pUnkForRelease = pMedSrc->pUnkForRelease;
    333         pMedSrc->pUnkForRelease->AddRef();
    334     }
    335 }
    336 STDMETHODIMP WCDataObject::EnumFormatEtc(DWORD dwDirection, IEnumFORMATETC** ppenumFormatEtc)
    337 {
    338     if(!ppenumFormatEtc)
    339         return E_POINTER;
    340 
    341     *ppenumFormatEtc=0;
    342     switch (dwDirection)
    343     {
    344     case DATADIR_GET:
    345         *ppenumFormatEtc= new WCEnumFormatEtc(m_formats);
    346         if(!(*ppenumFormatEtc))
    347             return E_OUTOFMEMORY;
    348         break;
    349 
    350     case DATADIR_SET:
    351     default:
    352         return E_NOTIMPL;
    353         break;
    354     }
    355 
    356     return S_OK;
    357 }
    358 
    359 STDMETHODIMP WCDataObject::DAdvise(FORMATETC*, DWORD, IAdviseSink*,DWORD*)
    360 {
    361     return OLE_E_ADVISENOTSUPPORTED;
    362 }
    363 
    364 STDMETHODIMP WCDataObject::DUnadvise(DWORD)
    365 {
    366     return E_NOTIMPL;
    367 }
    368 
    369 HRESULT STDMETHODCALLTYPE WCDataObject::EnumDAdvise(IEnumSTATDATA**)
    370 {
    371     return OLE_E_ADVISENOTSUPPORTED;
    372 }
    373 
    374 void WCDataObject::clearData(CLIPFORMAT format)
    375 {
    376     size_t ptr = 0;
    377     while (ptr < m_formats.size()) {
    378         if (m_formats[ptr]->cfFormat == format) {
    379             FORMATETC* current = m_formats[ptr];
    380             m_formats[ptr] = m_formats[m_formats.size() - 1];
    381             m_formats[m_formats.size() - 1] = 0;
    382             m_formats.removeLast();
    383             delete current;
    384             STGMEDIUM* medium = m_medium[ptr];
    385             m_medium[ptr] = m_medium[m_medium.size() - 1];
    386             m_medium[m_medium.size() - 1] = 0;
    387             m_medium.removeLast();
    388             ReleaseStgMedium(medium);
    389             delete medium;
    390             continue;
    391         }
    392         ptr++;
    393     }
    394 }
    395 
    396 }
    397