1/*
2 * Copyright (C) 2007, 2014 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 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 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#ifndef WCDataObject_h
27#define WCDataObject_h
28
29#include "DragData.h"
30#include <ShlObj.h>
31#include <objidl.h>
32#include <wtf/Forward.h>
33#include <wtf/Vector.h>
34
35namespace WebCore {
36
37struct StgMediumDeleter {
38    void operator()(STGMEDIUM* medium)
39    {
40        ::ReleaseStgMedium(medium);
41    }
42};
43
44class WCDataObject : public IDataObject {
45public:
46    void CopyMedium(STGMEDIUM* pMedDest, STGMEDIUM* pMedSrc, FORMATETC* pFmtSrc);
47
48    //IUnknown
49    virtual HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void** ppvObject);
50    virtual ULONG STDMETHODCALLTYPE AddRef(void);
51    virtual ULONG STDMETHODCALLTYPE Release(void);
52
53    //IDataObject
54    virtual HRESULT STDMETHODCALLTYPE GetData(FORMATETC* pformatIn, STGMEDIUM* pmedium);
55    virtual HRESULT STDMETHODCALLTYPE GetDataHere(FORMATETC* pformat, STGMEDIUM* pmedium);
56    virtual HRESULT STDMETHODCALLTYPE QueryGetData(FORMATETC* pformat);
57    virtual HRESULT STDMETHODCALLTYPE GetCanonicalFormatEtc(FORMATETC* pformatectIn,FORMATETC* pformatOut);
58    virtual HRESULT STDMETHODCALLTYPE SetData(FORMATETC* pformat, STGMEDIUM*pmedium, BOOL release);
59    virtual HRESULT STDMETHODCALLTYPE EnumFormatEtc(DWORD dwDirection, IEnumFORMATETC** ppenumFormatEtc);
60    virtual HRESULT STDMETHODCALLTYPE DAdvise(FORMATETC*, DWORD, IAdviseSink*, DWORD*);
61    virtual HRESULT STDMETHODCALLTYPE DUnadvise(DWORD);
62    virtual HRESULT STDMETHODCALLTYPE EnumDAdvise(IEnumSTATDATA**);
63
64    void clearData(CLIPFORMAT);
65
66    static HRESULT createInstance(WCDataObject**);
67    static HRESULT createInstance(WCDataObject**, const DragDataMap&);
68private:
69    WCDataObject();
70    long m_ref;
71    Vector<std::unique_ptr<FORMATETC>> m_formats;
72    Vector<std::unique_ptr<STGMEDIUM, StgMediumDeleter>> m_medium;
73};
74
75}
76
77#endif //!WCDataObject_h
78