1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef UI_BASE_DRAGDROP_OS_EXCHANGE_DATA_PROVIDER_WIN_H_
#define UI_BASE_DRAGDROP_OS_EXCHANGE_DATA_PROVIDER_WIN_H_
#include <windows.h>
#undef LoadBitmap // Conflicts with ui/ API name.
#include <objidl.h>
#include <shlobj.h>
#include <stddef.h>
#include <wrl/client.h>
#include <memory>
#include <string>
#include <string_view>
#include <utility>
#include <vector>
#include "base/component_export.h"
#include "base/containers/span.h"
#include "ui/base/dragdrop/os_exchange_data.h"
#include "ui/base/dragdrop/os_exchange_data_provider.h"
#include "ui/gfx/geometry/vector2d.h"
#include "ui/gfx/image/image_skia.h"
namespace ui {
class DataObjectImpl : public DownloadFileObserver,
public IDataObject,
public IDataObjectAsyncCapability {
public:
DataObjectImpl();
// Accessors.
void set_in_drag_loop(bool in_drag_loop) { in_drag_loop_ = in_drag_loop; }
// Number of known formats.
size_t size() const { return contents_.size(); }
// DownloadFileObserver implementation:
void OnDownloadCompleted(const base::FilePath& file_path) override;
void OnDownloadAborted() override;
// IDataObject implementation:
HRESULT __stdcall GetData(FORMATETC* format_etc, STGMEDIUM* medium) override;
HRESULT __stdcall GetDataHere(FORMATETC* format_etc,
STGMEDIUM* medium) override;
HRESULT __stdcall QueryGetData(FORMATETC* format_etc) override;
HRESULT __stdcall GetCanonicalFormatEtc(FORMATETC* format_etc,
FORMATETC* result) override;
HRESULT __stdcall SetData(FORMATETC* format_etc,
STGMEDIUM* medium,
BOOL should_release) override;
HRESULT __stdcall EnumFormatEtc(DWORD direction,
IEnumFORMATETC** enumerator) override;
HRESULT __stdcall DAdvise(FORMATETC* format_etc,
DWORD advf,
IAdviseSink* sink,
DWORD* connection) override;
HRESULT __stdcall DUnadvise(DWORD connection) override;
HRESULT __stdcall EnumDAdvise(IEnumSTATDATA** enumerator) override;
// IDataObjectAsyncCapability implementation:
HRESULT __stdcall EndOperation(HRESULT result,
IBindCtx* reserved,
DWORD effects) override;
HRESULT __stdcall GetAsyncMode(BOOL* is_op_async) override;
HRESULT __stdcall InOperation(BOOL* in_async_op) override;
HRESULT __stdcall SetAsyncMode(BOOL do_op_async) override;
HRESULT __stdcall StartOperation(IBindCtx* reserved) override;
// IUnknown implementation:
HRESULT __stdcall QueryInterface(const IID& iid, void** object) override;
ULONG __stdcall AddRef() override;
ULONG __stdcall Release() override;
private:
// FormatEtcEnumerator only likes us for our StoredDataMap typedef.
friend class FormatEtcEnumerator;
friend class OSExchangeDataProviderWin;
~DataObjectImpl() override;
void StopDownloads();
// Removes from contents_ the first data that matches |format|.
void RemoveData(const FORMATETC& format);
// Our internal representation of stored data & type info.
struct StoredDataInfo {
public:
FORMATETC format_etc;
STGMEDIUM medium;
std::unique_ptr<DownloadFileProvider> downloader;
~StoredDataInfo();
StoredDataInfo(const StoredDataInfo&) = delete;
StoredDataInfo& operator=(const StoredDataInfo&) = delete;
// Takes ownership of and nullifies `medium` to approximate moving from
// STGMEDIUM.
static std::unique_ptr<StoredDataInfo> TakeStorageMedium(
const FORMATETC& format_etc,
STGMEDIUM& medium);
private:
// STGMEDIUM is just a POD, it does not guarantee `medium` is no longer be
// used after calling this constructor while the ownership of `medium` is
// passed.
StoredDataInfo(const FORMATETC& format_etc, const STGMEDIUM& medium);
};
typedef std::vector<std::unique_ptr<StoredDataInfo>> StoredData;
StoredData contents_;
Microsoft::WRL::ComPtr<IDataObject> source_object_;
bool is_aborting_;
bool in_drag_loop_;
bool in_async_mode_;
bool async_operation_started_;
};
class COMPONENT_EXPORT(UI_BASE) OSExchangeDataProviderWin
: public OSExchangeDataProvider {
public:
static DataObjectImpl* GetDataObjectImpl(const OSExchangeData& data);
static IDataObject* GetIDataObject(const OSExchangeData& data);
explicit OSExchangeDataProviderWin(IDataObject* source);
OSExchangeDataProviderWin();
OSExchangeDataProviderWin(const OSExchangeDataProviderWin&) = delete;
OSExchangeDataProviderWin& operator=(const OSExchangeDataProviderWin&) =
delete;
~OSExchangeDataProviderWin() override;
IDataObject* data_object() const { return data_.get(); }
IDataObjectAsyncCapability* async_operation() const { return data_.get(); }
// OSExchangeDataProvider methods.
std::unique_ptr<OSExchangeDataProvider> Clone() const override;
void MarkRendererTaintedFromOrigin(const url::Origin& origin) override;
bool IsRendererTainted() const override;
std::optional<url::Origin> GetRendererTaintedOrigin() const override;
void MarkAsFromPrivileged() override;
bool IsFromPrivileged() const override;
void SetString(std::u16string_view data) override;
void SetURL(const GURL& url, std::u16string_view title) override;
void SetFilename(const base::FilePath& path) override;
void SetFilenames(const std::vector<FileInfo>& filenames) override;
// Test only method for adding virtual file content to the data store. The
// first value in the pair is the file display name, the second is a string
// providing the file content.
void SetVirtualFileContentsForTesting(
const std::vector<std::pair<base::FilePath, std::string>>&
filenames_and_contents,
DWORD tymed) override;
void SetPickledData(const ClipboardFormatType& format,
const base::Pickle& data) override;
void SetFileContents(const base::FilePath& filename,
const std::string& file_contents) override;
void SetHtml(const std::u16string& html, const GURL& base_url) override;
std::optional<std::u16string> GetString() const override;
std::optional<UrlInfo> GetURLAndTitle(
FilenameToURLPolicy policy) const override;
std::optional<std::vector<GURL>> GetURLs(
FilenameToURLPolicy policy) const override;
std::optional<std::vector<FileInfo>> GetFilenames() const override;
bool HasVirtualFilenames() const override;
std::optional<std::vector<FileInfo>> GetVirtualFilenames() const override;
void GetVirtualFilesAsTempFiles(
base::OnceCallback<
void(const std::vector<std::pair<base::FilePath, base::FilePath>>&)>
callback) const override;
std::optional<base::Pickle> GetPickledData(
const ClipboardFormatType& format) const override;
std::optional<FileContentsInfo> GetFileContents() const override;
std::optional<HtmlInfo> GetHtml() const override;
bool HasString() const override;
bool HasURL(FilenameToURLPolicy policy) const override;
bool HasFile() const override;
bool HasFileContents() const override;
bool HasHtml() const override;
bool HasCustomFormat(const ClipboardFormatType& format) const override;
void SetDownloadFileInfo(DownloadFileInfo* download_info) override;
void SetDragImage(const gfx::ImageSkia& image_skia,
const gfx::Vector2d& cursor_offset) override;
gfx::ImageSkia GetDragImage() const override;
gfx::Vector2d GetDragImageOffset() const override;
void SetSource(std::unique_ptr<DataTransferEndpoint> data_source) override;
DataTransferEndpoint* GetSource() const override;
private:
// Returns true if `GetPlainTextURL()` would return a GURL and false
// otherwise.
bool HasPlainTextURL() const;
// Returns a GURL if text is present and that text is a valid URL, and if
// `IsRendererTainted()` is true, that the URL has an HTTP or HTTPS scheme;
// otherwise, returns `std::nullopt`.
std::optional<GURL> GetPlainTextURL() const;
void SetVirtualFileContentAtIndexForTesting(base::span<const uint8_t> data,
DWORD tymed,
LONG index);
scoped_refptr<DataObjectImpl> data_;
Microsoft::WRL::ComPtr<IDataObject> source_object_;
};
} // namespace ui
#endif // UI_BASE_DRAGDROP_OS_EXCHANGE_DATA_PROVIDER_WIN_H_
|
