blob: 447a3efdba1e04e3d6336e7c99fa5fe43b372b2f [file] [log] [blame]
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <objidl.h>
#include <shlobj.h>
#include <stddef.h>
#include <wrl/client.h>
#include <utility>
#include <memory>
#include <string>
#include <vector>
#include "base/component_export.h"
#include "base/macros.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 {
// 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;
// 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 {
FORMATETC format_etc;
std::unique_ptr<DownloadFileProvider> downloader;
StoredDataInfo(const StoredDataInfo&) = delete;
StoredDataInfo& operator=(const StoredDataInfo&) = delete;
// Takes ownership of and nullifies `medium` to approximate moving from
static std::unique_ptr<StoredDataInfo> TakeStorageMedium(
const FORMATETC& format_etc,
STGMEDIUM& medium);
// 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 {
// Returns true if source has plain text that is a valid url.
static bool HasPlainTextURL(IDataObject* source);
// Returns true if source has plain text that is a valid URL and sets url to
// that url.
static bool GetPlainTextURL(IDataObject* source, GURL* url);
static DataObjectImpl* GetDataObjectImpl(const OSExchangeData& data);
static IDataObject* GetIDataObject(const OSExchangeData& data);
explicit OSExchangeDataProviderWin(IDataObject* source);
~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 MarkOriginatedFromRenderer() override;
bool DidOriginateFromRenderer() const override;
void SetString(const base::string16& data) override;
void SetURL(const GURL& url, const base::string16& 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>>&
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 base::string16& html, const GURL& base_url) override;
bool GetString(base::string16* data) const override;
bool GetURLAndTitle(FilenameToURLPolicy policy,
GURL* url,
base::string16* title) const override;
bool GetFilename(base::FilePath* path) const override;
bool GetFilenames(std::vector<FileInfo>* filenames) const override;
bool HasVirtualFilenames() const override;
bool GetVirtualFilenames(std::vector<FileInfo>* filenames) const override;
bool GetVirtualFilesAsTempFiles(
void(const std::vector<std::pair<base::FilePath, base::FilePath>>&)>
callback) const override;
bool GetPickledData(const ClipboardFormatType& format,
base::Pickle* data) const override;
bool GetFileContents(base::FilePath* filename,
std::string* file_contents) const override;
bool GetHtml(base::string16* html, GURL* base_url) 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;
void SetVirtualFileContentAtIndexForTesting(base::span<const uint8_t> data,
DWORD tymed,
size_t index);
scoped_refptr<DataObjectImpl> data_;
Microsoft::WRL::ComPtr<IDataObject> source_object_;
} // namespace ui