Source code
Revision control
Copy as Markdown
// Copyright (C) 2016 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
// Qt-Security score:significant reason:default
#include "qwindowsole.h"
#include "qwindowsmimeregistry.h"
#include "qwindowscontext.h"
\
#include <QtGui/qevent.h>
#include <QtGui/qwindow.h>
#include <QtGui/qpainter.h>
#include <QtGui/qcursor.h>
#include <QtGui/qguiapplication.h>
#include <QtCore/qmimedata.h>
#include <QtCore/qdebug.h>
#include <shlobj.h>
QT_BEGIN_NAMESPACE
/*!
\class QWindowsOleDataObject
\brief OLE data container
The following methods are NOT supported for data transfer using the
clipboard or drag-drop:
\list
\li IDataObject::SetData -- return E_NOTIMPL
\li IDataObject::DAdvise -- return OLE_E_ADVISENOTSUPPORTED
\li ::DUnadvise
\li ::EnumDAdvise
\li IDataObject::GetCanonicalFormatEtc -- return E_NOTIMPL
(NOTE: must set pformatetcOut->ptd = NULL)
\endlist
\internal
*/
QWindowsOleDataObject::QWindowsOleDataObject(QMimeData *mimeData) :
data(mimeData),
CF_PERFORMEDDROPEFFECT(RegisterClipboardFormat(CFSTR_PERFORMEDDROPEFFECT))
{
qCDebug(lcQpaMime) << __FUNCTION__ << mimeData->formats();
}
QWindowsOleDataObject::~QWindowsOleDataObject() = default;
void QWindowsOleDataObject::releaseQt()
{
data = nullptr;
}
QMimeData *QWindowsOleDataObject::mimeData() const
{
return data.data();
}
DWORD QWindowsOleDataObject::reportedPerformedEffect() const
{
return performedEffect;
}
STDMETHODIMP
QWindowsOleDataObject::GetData(LPFORMATETC pformatetc, LPSTGMEDIUM pmedium) noexcept
{
HRESULT hr = ResultFromScode(DATA_E_FORMATETC);
if (data) {
const QWindowsMimeRegistry &mc = QWindowsContext::instance()->mimeConverter();
if (auto converter = mc.converterFromMime(*pformatetc, data))
if (converter->convertFromMime(*pformatetc, data, pmedium))
hr = ResultFromScode(S_OK);
}
if (QWindowsContext::verbose > 1 && lcQpaMime().isDebugEnabled())
qCDebug(lcQpaMime) <<__FUNCTION__ << *pformatetc << "returns" << Qt::hex << Qt::showbase << quint64(hr);
return hr;
}
STDMETHODIMP
QWindowsOleDataObject::GetDataHere(LPFORMATETC, LPSTGMEDIUM) noexcept
{
return ResultFromScode(DATA_E_FORMATETC);
}
STDMETHODIMP
QWindowsOleDataObject::QueryGetData(LPFORMATETC pformatetc) noexcept
{
HRESULT hr = ResultFromScode(DATA_E_FORMATETC);
if (QWindowsContext::verbose > 1)
qCDebug(lcQpaMime) << __FUNCTION__;
if (data) {
const QWindowsMimeRegistry &mc = QWindowsContext::instance()->mimeConverter();
hr = mc.converterFromMime(*pformatetc, data) ?
ResultFromScode(S_OK) : ResultFromScode(S_FALSE);
}
if (QWindowsContext::verbose > 1)
qCDebug(lcQpaMime) << __FUNCTION__ << " returns 0x" << Qt::hex << int(hr);
return hr;
}
STDMETHODIMP
QWindowsOleDataObject::GetCanonicalFormatEtc(LPFORMATETC, LPFORMATETC pformatetcOut) noexcept
{
pformatetcOut->ptd = nullptr;
return ResultFromScode(E_NOTIMPL);
}
STDMETHODIMP
QWindowsOleDataObject::SetData(LPFORMATETC pFormatetc, STGMEDIUM *pMedium, BOOL fRelease) noexcept
{
if (QWindowsContext::verbose > 1)
qCDebug(lcQpaMime) << __FUNCTION__;
HRESULT hr = ResultFromScode(E_NOTIMPL);
if (pFormatetc->cfFormat == CF_PERFORMEDDROPEFFECT && pMedium->tymed == TYMED_HGLOBAL) {
if (GlobalSize(pMedium->hGlobal) >= sizeof(DWORD)) {
if (const auto *val = static_cast<const DWORD *>(GlobalLock(pMedium->hGlobal))) {
performedEffect = *val;
GlobalUnlock(pMedium->hGlobal);
}
}
if (fRelease)
ReleaseStgMedium(pMedium);
hr = ResultFromScode(S_OK);
}
if (QWindowsContext::verbose > 1)
qCDebug(lcQpaMime) << __FUNCTION__ << " returns 0x" << Qt::hex << int(hr);
return hr;
}
STDMETHODIMP
QWindowsOleDataObject::EnumFormatEtc(DWORD dwDirection, LPENUMFORMATETC FAR* ppenumFormatEtc) noexcept
{
if (QWindowsContext::verbose > 1)
qCDebug(lcQpaMime) << __FUNCTION__ << "dwDirection=" << dwDirection;
if (!data)
return ResultFromScode(DATA_E_FORMATETC);
SCODE sc = S_OK;
QList<FORMATETC> fmtetcs;
if (dwDirection == DATADIR_GET) {
QWindowsMimeRegistry &mc = QWindowsContext::instance()->mimeConverter();
fmtetcs = mc.allFormatsForMime(data);
} else {
FORMATETC formatetc;
formatetc.cfFormat = CLIPFORMAT(CF_PERFORMEDDROPEFFECT);
formatetc.dwAspect = DVASPECT_CONTENT;
formatetc.lindex = -1;
formatetc.ptd = nullptr;
formatetc.tymed = TYMED_HGLOBAL;
fmtetcs.append(formatetc);
}
auto *enumFmtEtc = new QWindowsOleEnumFmtEtc(fmtetcs);
*ppenumFormatEtc = enumFmtEtc;
if (enumFmtEtc->isNull()) {
delete enumFmtEtc;
*ppenumFormatEtc = nullptr;
sc = E_OUTOFMEMORY;
}
return ResultFromScode(sc);
}
STDMETHODIMP
QWindowsOleDataObject::DAdvise(FORMATETC FAR*, DWORD,
LPADVISESINK, DWORD FAR*) noexcept
{
return ResultFromScode(OLE_E_ADVISENOTSUPPORTED);
}
STDMETHODIMP
QWindowsOleDataObject::DUnadvise(DWORD) noexcept
{
return ResultFromScode(OLE_E_ADVISENOTSUPPORTED);
}
STDMETHODIMP
QWindowsOleDataObject::EnumDAdvise(LPENUMSTATDATA FAR*) noexcept
{
return ResultFromScode(OLE_E_ADVISENOTSUPPORTED);
}
/*!
\class QWindowsOleEnumFmtEtc
\brief Enumerates the FORMATETC structures supported by QWindowsOleDataObject.
\internal
*/
QWindowsOleEnumFmtEtc::QWindowsOleEnumFmtEtc(const QList<FORMATETC> &fmtetcs)
{
if (QWindowsContext::verbose > 1)
qCDebug(lcQpaMime) << __FUNCTION__ << fmtetcs;
m_lpfmtetcs.reserve(fmtetcs.count());
for (int idx = 0; idx < fmtetcs.count(); ++idx) {
auto destetc = new FORMATETC();
if (copyFormatEtc(destetc, &(fmtetcs.at(idx)))) {
m_lpfmtetcs.append(destetc);
} else {
m_isNull = true;
delete destetc;
break;
}
}
}
QWindowsOleEnumFmtEtc::QWindowsOleEnumFmtEtc(const QList<LPFORMATETC> &lpfmtetcs)
{
if (QWindowsContext::verbose > 1)
qCDebug(lcQpaMime) << __FUNCTION__;
m_lpfmtetcs.reserve(lpfmtetcs.count());
for (int idx = 0; idx < lpfmtetcs.count(); ++idx) {
LPFORMATETC srcetc = lpfmtetcs.at(idx);
auto destetc = new FORMATETC();
if (copyFormatEtc(destetc, srcetc)) {
m_lpfmtetcs.append(destetc);
} else {
m_isNull = true;
delete destetc;
break;
}
}
}
QWindowsOleEnumFmtEtc::~QWindowsOleEnumFmtEtc()
{
LPMALLOC pmalloc;
if (CoGetMalloc(MEMCTX_TASK, &pmalloc) == NOERROR) {
for (int idx = 0; idx < m_lpfmtetcs.count(); ++idx) {
LPFORMATETC tmpetc = m_lpfmtetcs.at(idx);
if (tmpetc->ptd)
pmalloc->Free(tmpetc->ptd);
delete tmpetc;
}
pmalloc->Release();
}
m_lpfmtetcs.clear();
}
bool QWindowsOleEnumFmtEtc::isNull() const
{
return m_isNull;
}
// IEnumFORMATETC methods
STDMETHODIMP
QWindowsOleEnumFmtEtc::Next(ULONG celt, LPFORMATETC rgelt, ULONG FAR* pceltFetched) noexcept
{
ULONG i=0;
ULONG nOffset;
if (rgelt == nullptr)
return ResultFromScode(E_INVALIDARG);
while (i < celt) {
nOffset = m_nIndex + i;
if (nOffset < ULONG(m_lpfmtetcs.count())) {
if (!copyFormatEtc(rgelt + i, m_lpfmtetcs.at(int(nOffset)))) {
m_nIndex += i;
if (pceltFetched != nullptr)
*pceltFetched = i;
return ResultFromScode(E_OUTOFMEMORY);
}
i++;
} else {
break;
}
}
m_nIndex += i;
if (pceltFetched != nullptr)
*pceltFetched = i;
if (i != celt)
return ResultFromScode(S_FALSE);
return NOERROR;
}
STDMETHODIMP
QWindowsOleEnumFmtEtc::Skip(ULONG celt) noexcept
{
ULONG i=0;
ULONG nOffset;
while (i < celt) {
nOffset = m_nIndex + i;
if (nOffset < ULONG(m_lpfmtetcs.count())) {
i++;
} else {
break;
}
}
m_nIndex += i;
if (i != celt)
return ResultFromScode(S_FALSE);
return NOERROR;
}
STDMETHODIMP
QWindowsOleEnumFmtEtc::Reset() noexcept
{
m_nIndex = 0;
return NOERROR;
}
STDMETHODIMP
QWindowsOleEnumFmtEtc::Clone(LPENUMFORMATETC FAR* newEnum) noexcept
{
if (newEnum == nullptr)
return ResultFromScode(E_INVALIDARG);
auto *result = new QWindowsOleEnumFmtEtc(m_lpfmtetcs);
result->m_nIndex = m_nIndex;
if (result->isNull()) {
delete result;
return ResultFromScode(E_OUTOFMEMORY);
}
*newEnum = result;
return NOERROR;
}
bool QWindowsOleEnumFmtEtc::copyFormatEtc(LPFORMATETC dest, const FORMATETC *src) const
{
if (dest == nullptr || src == nullptr)
return false;
*dest = *src;
if (src->ptd) {
if (src->ptd->tdSize < sizeof(DVTARGETDEVICE)) {
dest->ptd = nullptr;
return false;
}
LPMALLOC pmalloc;
if (CoGetMalloc(MEMCTX_TASK, &pmalloc) != NOERROR) {
dest->ptd = nullptr;
return false;
}
dest->ptd = static_cast<DVTARGETDEVICE *>(pmalloc->Alloc(src->ptd->tdSize));
if (!dest->ptd) {
pmalloc->Release();
return false;
}
memcpy(dest->ptd, src->ptd, size_t(src->ptd->tdSize));
pmalloc->Release();
}
return true;
}
QT_END_NAMESPACE