#include "config.h"
#include "WCDataObject.h"
#include "ClipboardUtilitiesWin.h"
#include "DragData.h"
#include <wtf/text/WTFString.h>
namespace WebCore {
class WCEnumFormatEtc : public IEnumFORMATETC
{
public:
WCEnumFormatEtc(const Vector<FORMATETC>& formats);
WCEnumFormatEtc(const Vector<std::unique_ptr<FORMATETC>>& formats);
STDMETHOD(QueryInterface)(REFIID, void**);
STDMETHOD_(ULONG, AddRef)(void);
STDMETHOD_(ULONG, Release)(void);
STDMETHOD(Next)(ULONG, LPFORMATETC, ULONG*);
STDMETHOD(Skip)(ULONG);
STDMETHOD(Reset)(void);
STDMETHOD(Clone)(IEnumFORMATETC**);
private:
long m_ref;
Vector<FORMATETC> m_formats;
size_t m_current;
};
WCEnumFormatEtc::WCEnumFormatEtc(const Vector<FORMATETC>& formats)
: m_ref(1)
, m_current(0)
{
for(size_t i = 0; i < formats.size(); ++i)
m_formats.append(formats[i]);
}
WCEnumFormatEtc::WCEnumFormatEtc(const Vector<std::unique_ptr<FORMATETC>>& formats)
: m_ref(1)
, m_current(0)
{
for (auto& format : formats)
m_formats.append(*format);
}
STDMETHODIMP WCEnumFormatEtc::QueryInterface(REFIID riid, void** ppvObject)
{
*ppvObject = 0;
if (IsEqualIID(riid, IID_IUnknown) || IsEqualIID(riid, IID_IEnumFORMATETC)) {
*ppvObject = this;
AddRef();
return S_OK;
}
return E_NOINTERFACE;
}
STDMETHODIMP_(ULONG) WCEnumFormatEtc::AddRef(void)
{
return InterlockedIncrement(&m_ref);
}
STDMETHODIMP_(ULONG) WCEnumFormatEtc::Release(void)
{
long c = InterlockedDecrement(&m_ref);
if (c == 0)
delete this;
return c;
}
STDMETHODIMP WCEnumFormatEtc::Next(ULONG celt, LPFORMATETC lpFormatEtc, ULONG* pceltFetched)
{
if(pceltFetched != 0)
*pceltFetched=0;
ULONG cReturn = celt;
if(celt <= 0 || lpFormatEtc == 0 || m_current >= m_formats.size())
return S_FALSE;
if(pceltFetched == 0 && celt != 1) return S_FALSE;
while (m_current < m_formats.size() && cReturn > 0) {
*lpFormatEtc++ = m_formats[m_current++];
--cReturn;
}
if (pceltFetched != 0)
*pceltFetched = celt - cReturn;
return (cReturn == 0) ? S_OK : S_FALSE;
}
STDMETHODIMP WCEnumFormatEtc::Skip(ULONG celt)
{
if((m_current + int(celt)) >= m_formats.size())
return S_FALSE;
m_current += celt;
return S_OK;
}
STDMETHODIMP WCEnumFormatEtc::Reset(void)
{
m_current = 0;
return S_OK;
}
STDMETHODIMP WCEnumFormatEtc::Clone(IEnumFORMATETC** ppCloneEnumFormatEtc)
{
if(!ppCloneEnumFormatEtc)
return E_POINTER;
WCEnumFormatEtc *newEnum = new WCEnumFormatEtc(m_formats);
if(!newEnum)
return E_OUTOFMEMORY;
newEnum->AddRef();
newEnum->m_current = m_current;
*ppCloneEnumFormatEtc = newEnum;
return S_OK;
}
HRESULT WCDataObject::createInstance(WCDataObject** result)
{
if (!result)
return E_POINTER;
*result = new WCDataObject();
return S_OK;
}
HRESULT WCDataObject::createInstance(WCDataObject** result, const DragDataMap& dataMap)
{
if (!result)
return E_POINTER;
*result = new WCDataObject;
for (DragDataMap::const_iterator it = dataMap.begin(); it != dataMap.end(); ++it)
setClipboardData(*result, it->key, it->value);
return S_OK;
}
WCDataObject::WCDataObject()
: m_ref(1)
{
}
STDMETHODIMP WCDataObject::QueryInterface(REFIID riid,void** ppvObject)
{
*ppvObject = 0;
if (IsEqualIID(riid, IID_IUnknown) ||
IsEqualIID(riid, IID_IDataObject)) {
*ppvObject=this;
}
if (*ppvObject) {
AddRef();
return S_OK;
}
return E_NOINTERFACE;
}
STDMETHODIMP_(ULONG) WCDataObject::AddRef( void)
{
return InterlockedIncrement(&m_ref);
}
STDMETHODIMP_(ULONG) WCDataObject::Release( void)
{
long c = InterlockedDecrement(&m_ref);
if (c == 0)
delete this;
return c;
}
STDMETHODIMP WCDataObject::GetData(FORMATETC* pformatetcIn, STGMEDIUM* pmedium)
{
if(!pformatetcIn || !pmedium)
return E_POINTER;
pmedium->hGlobal = 0;
for (size_t i = 0; i < m_formats.size(); ++i) {
if( pformatetcIn->lindex == m_formats[i]->lindex &&
pformatetcIn->dwAspect == m_formats[i]->dwAspect &&
pformatetcIn->cfFormat == m_formats[i]->cfFormat) {
CopyMedium(pmedium, m_medium[i].get(), m_formats[i].get());
return S_OK;
}
}
return DV_E_FORMATETC;
}
STDMETHODIMP WCDataObject::GetDataHere(FORMATETC*, STGMEDIUM*)
{
return E_NOTIMPL;
}
STDMETHODIMP WCDataObject::QueryGetData(FORMATETC* pformatetc)
{
if(!pformatetc)
return E_POINTER;
if (!(DVASPECT_CONTENT & pformatetc->dwAspect))
return (DV_E_DVASPECT);
HRESULT hr = DV_E_TYMED;
for (auto& format : m_formats) {
if (pformatetc->tymed & format->tymed) {
if (pformatetc->cfFormat == format->cfFormat)
return S_OK;
hr = DV_E_CLIPFORMAT;
}
else
hr = DV_E_TYMED;
}
return hr;
}
STDMETHODIMP WCDataObject::GetCanonicalFormatEtc(FORMATETC*, FORMATETC*)
{
return DATA_S_SAMEFORMATETC;
}
STDMETHODIMP WCDataObject::SetData(FORMATETC* pformatetc, STGMEDIUM* pmedium, BOOL fRelease)
{
if(!pformatetc || !pmedium)
return E_POINTER;
auto fetc = std::make_unique<FORMATETC>();
std::unique_ptr<STGMEDIUM, StgMediumDeleter> pStgMed(new STGMEDIUM);
ZeroMemory(fetc.get(), sizeof(FORMATETC));
ZeroMemory(pStgMed.get(), sizeof(STGMEDIUM));
*fetc = *pformatetc;
m_formats.append(WTF::move(fetc));
if(fRelease)
*pStgMed = *pmedium;
else
CopyMedium(pStgMed.get(), pmedium, pformatetc);
m_medium.append(WTF::move(pStgMed));
return S_OK;
}
void WCDataObject::CopyMedium(STGMEDIUM* pMedDest, STGMEDIUM* pMedSrc, FORMATETC* pFmtSrc)
{
switch(pMedSrc->tymed)
{
#if !OS(WINCE)
case TYMED_HGLOBAL:
pMedDest->hGlobal = (HGLOBAL)OleDuplicateData(pMedSrc->hGlobal,pFmtSrc->cfFormat, 0);
break;
case TYMED_GDI:
pMedDest->hBitmap = (HBITMAP)OleDuplicateData(pMedSrc->hBitmap,pFmtSrc->cfFormat, 0);
break;
case TYMED_MFPICT:
pMedDest->hMetaFilePict = (HMETAFILEPICT)OleDuplicateData(pMedSrc->hMetaFilePict,pFmtSrc->cfFormat, 0);
break;
case TYMED_ENHMF:
pMedDest->hEnhMetaFile = (HENHMETAFILE)OleDuplicateData(pMedSrc->hEnhMetaFile,pFmtSrc->cfFormat, 0);
break;
case TYMED_FILE:
pMedSrc->lpszFileName = (LPOLESTR)OleDuplicateData(pMedSrc->lpszFileName,pFmtSrc->cfFormat, 0);
break;
#endif
case TYMED_ISTREAM:
pMedDest->pstm = pMedSrc->pstm;
pMedSrc->pstm->AddRef();
break;
case TYMED_ISTORAGE:
pMedDest->pstg = pMedSrc->pstg;
pMedSrc->pstg->AddRef();
break;
default:
break;
}
pMedDest->tymed = pMedSrc->tymed;
pMedDest->pUnkForRelease = 0;
if(pMedSrc->pUnkForRelease != 0) {
pMedDest->pUnkForRelease = pMedSrc->pUnkForRelease;
pMedSrc->pUnkForRelease->AddRef();
}
}
STDMETHODIMP WCDataObject::EnumFormatEtc(DWORD dwDirection, IEnumFORMATETC** ppenumFormatEtc)
{
if(!ppenumFormatEtc)
return E_POINTER;
*ppenumFormatEtc=0;
switch (dwDirection)
{
case DATADIR_GET:
*ppenumFormatEtc= new WCEnumFormatEtc(m_formats);
if(!(*ppenumFormatEtc))
return E_OUTOFMEMORY;
break;
case DATADIR_SET:
default:
return E_NOTIMPL;
break;
}
return S_OK;
}
STDMETHODIMP WCDataObject::DAdvise(FORMATETC*, DWORD, IAdviseSink*,DWORD*)
{
return OLE_E_ADVISENOTSUPPORTED;
}
STDMETHODIMP WCDataObject::DUnadvise(DWORD)
{
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE WCDataObject::EnumDAdvise(IEnumSTATDATA**)
{
return OLE_E_ADVISENOTSUPPORTED;
}
void WCDataObject::clearData(CLIPFORMAT format)
{
size_t ptr = 0;
while (ptr < m_formats.size()) {
if (m_formats[ptr]->cfFormat == format) {
m_formats[ptr] = m_formats.takeLast();
m_medium[ptr] = m_medium.takeLast();
continue;
}
ptr++;
}
}
}