summaryrefslogtreecommitdiffstats
path: root/app/os_exchange_data_provider_win.cc
diff options
context:
space:
mode:
authorsky@chromium.org <sky@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>2009-08-12 21:58:18 +0000
committersky@chromium.org <sky@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>2009-08-12 21:58:18 +0000
commit8c3dc79bc13ba84f418d3c135e1bf296a3e29722 (patch)
tree32eb9688d500a90eb74a7c4f8cb5a97e507dd0fc /app/os_exchange_data_provider_win.cc
parentabaccb2cb8cff8138e5ea9daf420645e5852c9eb (diff)
downloadchromium_src-8c3dc79bc13ba84f418d3c135e1bf296a3e29722.zip
chromium_src-8c3dc79bc13ba84f418d3c135e1bf296a3e29722.tar.gz
chromium_src-8c3dc79bc13ba84f418d3c135e1bf296a3e29722.tar.bz2
Refactors OSExchangeData for easier portability.
BUG=none TEST=none Review URL: http://codereview.chromium.org/164401 git-svn-id: svn://svn.chromium.org/chrome/trunk/src@23230 0039d316-1c4b-4281-b951-d872f2087c98
Diffstat (limited to 'app/os_exchange_data_provider_win.cc')
-rw-r--r--app/os_exchange_data_provider_win.cc757
1 files changed, 757 insertions, 0 deletions
diff --git a/app/os_exchange_data_provider_win.cc b/app/os_exchange_data_provider_win.cc
new file mode 100644
index 0000000..1d8bdd8
--- /dev/null
+++ b/app/os_exchange_data_provider_win.cc
@@ -0,0 +1,757 @@
+// Copyright (c) 2009 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 "app/os_exchange_data_provider_win.h"
+
+#include "app/l10n_util.h"
+#include "base/clipboard_util.h"
+#include "base/file_util.h"
+#include "base/logging.h"
+#include "base/pickle.h"
+#include "base/scoped_handle.h"
+#include "base/stl_util-inl.h"
+#include "base/string_util.h"
+#include "googleurl/src/gurl.h"
+#include "grit/app_strings.h"
+#include "net/base/net_util.h"
+
+// Creates a new STGMEDIUM object to hold the specified text. The caller
+// owns the resulting object. The "Bytes" version does not NULL terminate, the
+// string version does.
+static STGMEDIUM* GetStorageForBytes(const char* data, size_t bytes);
+static STGMEDIUM* GetStorageForWString(const std::wstring& data);
+static STGMEDIUM* GetStorageForString(const std::string& data);
+// Creates the contents of an Internet Shortcut file for the given URL.
+static void GetInternetShortcutFileContents(const GURL& url, std::string* data);
+// Creates a valid file name given a suggested title and URL.
+static void CreateValidFileNameFromTitle(const GURL& url,
+ const std::wstring& title,
+ std::wstring* validated);
+// Creates a File Descriptor for the creation of a file to the given URL and
+// returns a handle to it.
+static STGMEDIUM* GetStorageForFileDescriptor(
+ const std::wstring& valid_file_name);
+
+///////////////////////////////////////////////////////////////////////////////
+// FormatEtcEnumerator
+
+//
+// This object implements an enumeration interface. The existence of an
+// implementation of this interface is exposed to clients through
+// OSExchangeData's EnumFormatEtc method. Our implementation is nobody's
+// business but our own, so it lives in this file.
+//
+// This Windows API is truly a gem. It wants to be an enumerator but assumes
+// some sort of sequential data (why not just use an array?). See comments
+// throughout.
+//
+class FormatEtcEnumerator : public IEnumFORMATETC {
+ public:
+ FormatEtcEnumerator(DataObjectImpl::StoredData::const_iterator begin,
+ DataObjectImpl::StoredData::const_iterator end);
+ ~FormatEtcEnumerator();
+
+ // IEnumFORMATETC implementation:
+ HRESULT __stdcall Next(
+ ULONG count, FORMATETC* elements_array, ULONG* elements_fetched);
+ HRESULT __stdcall Skip(ULONG skip_count);
+ HRESULT __stdcall Reset();
+ HRESULT __stdcall Clone(IEnumFORMATETC** clone);
+
+ // IUnknown implementation:
+ HRESULT __stdcall QueryInterface(const IID& iid, void** object);
+ ULONG __stdcall AddRef();
+ ULONG __stdcall Release();
+
+ private:
+ // This can only be called from |CloneFromOther|, since it initializes the
+ // contents_ from the other enumerator's contents.
+ FormatEtcEnumerator() : ref_count_(0) {
+ }
+
+ // Clone a new FormatEtc from another instance of this enumeration.
+ static FormatEtcEnumerator* CloneFromOther(const FormatEtcEnumerator* other);
+
+ private:
+ // We are _forced_ to use a vector as our internal data model as Windows'
+ // retarded IEnumFORMATETC API assumes a deterministic ordering of elements
+ // through methods like Next and Skip. This exposes the underlying data
+ // structure to the user. Bah.
+ std::vector<FORMATETC*> contents_;
+
+ // The cursor of the active enumeration - an index into |contents_|.
+ int cursor_;
+
+ LONG ref_count_;
+
+ DISALLOW_EVIL_CONSTRUCTORS(FormatEtcEnumerator);
+};
+
+// Safely makes a copy of all of the relevant bits of a FORMATETC object.
+static void CloneFormatEtc(FORMATETC* source, FORMATETC* clone) {
+ *clone = *source;
+ if (source->ptd) {
+ source->ptd =
+ static_cast<DVTARGETDEVICE*>(CoTaskMemAlloc(sizeof(DVTARGETDEVICE)));
+ *(clone->ptd) = *(source->ptd);
+ }
+}
+
+FormatEtcEnumerator::FormatEtcEnumerator(
+ DataObjectImpl::StoredData::const_iterator start,
+ DataObjectImpl::StoredData::const_iterator end)
+ : ref_count_(0), cursor_(0) {
+ // Copy FORMATETC data from our source into ourselves.
+ while (start != end) {
+ FORMATETC* format_etc = new FORMATETC;
+ CloneFormatEtc(&(*start)->format_etc, format_etc);
+ contents_.push_back(format_etc);
+ ++start;
+ }
+}
+
+FormatEtcEnumerator::~FormatEtcEnumerator() {
+ STLDeleteContainerPointers(contents_.begin(), contents_.end());
+}
+
+STDMETHODIMP FormatEtcEnumerator::Next(
+ ULONG count, FORMATETC* elements_array, ULONG* elements_fetched) {
+ // MSDN says |elements_fetched| is allowed to be NULL if count is 1.
+ if (!elements_fetched)
+ DCHECK(count == 1);
+
+ // This method copies count elements into |elements_array|.
+ int index = 0;
+ while (cursor_ < static_cast<int>(contents_.size()) &&
+ static_cast<ULONG>(index) < count) {
+ CloneFormatEtc(contents_.at(cursor_), &elements_array[index]);
+ ++cursor_;
+ ++index;
+ }
+ // The out param is for how many we actually copied.
+ if (elements_fetched)
+ *elements_fetched = index;
+
+ // If the two don't agree, then we fail.
+ return index == count ? S_OK : S_FALSE;
+}
+
+STDMETHODIMP FormatEtcEnumerator::Skip(ULONG skip_count) {
+ cursor_ += skip_count;
+ // MSDN implies it's OK to leave the enumerator trashed.
+ // "Whatever you say, boss"
+ return cursor_ <= static_cast<int>(contents_.size()) ? S_OK : S_FALSE;
+}
+
+STDMETHODIMP FormatEtcEnumerator::Reset() {
+ cursor_ = 0;
+ return S_OK;
+}
+
+STDMETHODIMP FormatEtcEnumerator::Clone(IEnumFORMATETC** clone) {
+ // Clone the current enumerator in its exact state, including cursor.
+ FormatEtcEnumerator* e = CloneFromOther(this);
+ e->AddRef();
+ *clone = e;
+ return S_OK;
+}
+
+STDMETHODIMP FormatEtcEnumerator::QueryInterface(const IID& iid,
+ void** object) {
+ *object = NULL;
+ if (IsEqualIID(iid, IID_IUnknown) || IsEqualIID(iid, IID_IEnumFORMATETC)) {
+ *object = this;
+ } else {
+ return E_NOINTERFACE;
+ }
+ AddRef();
+ return S_OK;
+}
+
+ULONG FormatEtcEnumerator::AddRef() {
+ return InterlockedIncrement(&ref_count_);
+}
+
+ULONG FormatEtcEnumerator::Release() {
+ if (InterlockedDecrement(&ref_count_) == 0) {
+ ULONG copied_refcnt = ref_count_;
+ delete this;
+ return copied_refcnt;
+ }
+ return ref_count_;
+}
+
+// static
+FormatEtcEnumerator* FormatEtcEnumerator::CloneFromOther(
+ const FormatEtcEnumerator* other) {
+ FormatEtcEnumerator* e = new FormatEtcEnumerator;
+ // Copy FORMATETC data from our source into ourselves.
+ std::vector<FORMATETC*>::const_iterator start = other->contents_.begin();
+ while (start != other->contents_.end()) {
+ FORMATETC* format_etc = new FORMATETC;
+ CloneFormatEtc(*start, format_etc);
+ e->contents_.push_back(format_etc);
+ ++start;
+ }
+ // Carry over
+ e->cursor_ = other->cursor_;
+ return e;
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// OSExchangeDataProviderWin, public:
+
+// static
+bool OSExchangeDataProviderWin::HasPlainTextURL(IDataObject* source) {
+ std::wstring plain_text;
+ return (ClipboardUtil::GetPlainText(source, &plain_text) &&
+ !plain_text.empty() && GURL(plain_text).is_valid());
+}
+
+// static
+bool OSExchangeDataProviderWin::GetPlainTextURL(IDataObject* source,
+ GURL* url) {
+ std::wstring plain_text;
+ if (ClipboardUtil::GetPlainText(source, &plain_text) &&
+ !plain_text.empty()) {
+ GURL gurl(plain_text);
+ if (gurl.is_valid()) {
+ *url = gurl;
+ return true;
+ }
+ }
+ return false;
+}
+
+// static
+IDataObject* OSExchangeDataProviderWin::GetIDataObject(
+ const OSExchangeData& data) {
+ return static_cast<const OSExchangeDataProviderWin&>(data.provider()).
+ data_object();
+}
+
+OSExchangeDataProviderWin::OSExchangeDataProviderWin(IDataObject* source)
+ : data_(new DataObjectImpl()),
+ source_object_(source) {
+}
+
+OSExchangeDataProviderWin::OSExchangeDataProviderWin()
+ : data_(new DataObjectImpl()),
+ source_object_(data_.get()) {
+}
+
+OSExchangeDataProviderWin::~OSExchangeDataProviderWin() {
+}
+
+void OSExchangeDataProviderWin::SetString(const std::wstring& data) {
+ STGMEDIUM* storage = GetStorageForWString(data);
+ data_->contents_.push_back(
+ new DataObjectImpl::StoredDataInfo(CF_UNICODETEXT, storage));
+
+ // Also add plain text.
+ storage = GetStorageForString(WideToUTF8(data));
+ data_->contents_.push_back(
+ new DataObjectImpl::StoredDataInfo(CF_TEXT, storage));
+}
+
+void OSExchangeDataProviderWin::SetURL(const GURL& url,
+ const std::wstring& title) {
+ // NOTE WELL:
+ // Every time you change the order of the first two CLIPFORMATS that get
+ // added here, you need to update the EnumerationViaCOM test case in
+ // the _unittest.cc file to reflect the new arrangement otherwise that test
+ // will fail! It assumes an insertion order.
+
+ // Add text/x-moz-url for drags from Firefox
+ std::wstring x_moz_url_str = UTF8ToWide(url.spec());
+ x_moz_url_str += '\n';
+ x_moz_url_str += title;
+ STGMEDIUM* storage = GetStorageForWString(x_moz_url_str);
+ data_->contents_.push_back(new DataObjectImpl::StoredDataInfo(
+ ClipboardUtil::GetMozUrlFormat()->cfFormat, storage));
+
+ // Add a .URL shortcut file for dragging to Explorer.
+ std::wstring valid_file_name;
+ CreateValidFileNameFromTitle(url, title, &valid_file_name);
+ std::string shortcut_url_file_contents;
+ GetInternetShortcutFileContents(url, &shortcut_url_file_contents);
+ SetFileContents(valid_file_name, shortcut_url_file_contents);
+
+ // Add a UniformResourceLocator link for apps like IE and Word.
+ storage = GetStorageForWString(UTF8ToWide(url.spec()));
+ data_->contents_.push_back(new DataObjectImpl::StoredDataInfo(
+ ClipboardUtil::GetUrlWFormat()->cfFormat, storage));
+ storage = GetStorageForString(url.spec());
+ data_->contents_.push_back(new DataObjectImpl::StoredDataInfo(
+ ClipboardUtil::GetUrlFormat()->cfFormat, storage));
+
+ // TODO(beng): (http://b/1085501) add CF_HTML...
+
+ // Also add text representations (these should be last since they're the
+ // least preferable).
+ storage = GetStorageForWString(UTF8ToWide(url.spec()));
+ data_->contents_.push_back(
+ new DataObjectImpl::StoredDataInfo(CF_UNICODETEXT, storage));
+ storage = GetStorageForString(url.spec());
+ data_->contents_.push_back(
+ new DataObjectImpl::StoredDataInfo(CF_TEXT, storage));
+}
+
+void OSExchangeDataProviderWin::SetFilename(const std::wstring& full_path) {
+ const size_t drop_size = sizeof(DROPFILES);
+ const size_t bytes = drop_size + (full_path.length() + 2) * sizeof(wchar_t);
+ HANDLE hdata = ::GlobalAlloc(GMEM_MOVEABLE, bytes);
+ if (!hdata)
+ return;
+
+ ScopedHGlobal<DROPFILES> locked_mem(hdata);
+ DROPFILES* drop_files = locked_mem.get();
+ drop_files->pFiles = sizeof(DROPFILES);
+ drop_files->fWide = TRUE;
+ wchar_t* data = reinterpret_cast<wchar_t*>((BYTE*)(drop_files) + drop_size);
+ const size_t copy_size = (full_path.length() + 1) * sizeof(wchar_t);
+ memcpy(data, full_path.c_str(), copy_size);
+ data[full_path.length() + 1] = L'\0'; // Double NULL
+
+ // Set up the STGMEDIUM
+ STGMEDIUM* storage = new STGMEDIUM;
+ storage->tymed = TYMED_HGLOBAL;
+ storage->hGlobal = drop_files;
+ storage->pUnkForRelease = NULL;
+
+ // Set up the StoredDataInfo
+ DataObjectImpl::StoredDataInfo* info =
+ new DataObjectImpl::StoredDataInfo(CF_HDROP, storage);
+ data_->contents_.push_back(info);
+}
+
+void OSExchangeDataProviderWin::SetPickledData(CLIPFORMAT format,
+ const Pickle& data) {
+ STGMEDIUM* storage = GetStorageForString(
+ std::string(static_cast<const char *>(data.data()),
+ static_cast<size_t>(data.size())));
+ data_->contents_.push_back(
+ new DataObjectImpl::StoredDataInfo(format, storage));
+}
+
+void OSExchangeDataProviderWin::SetFileContents(
+ const std::wstring& filename,
+ const std::string& file_contents) {
+ // Add CFSTR_FILEDESCRIPTOR
+ STGMEDIUM* storage = GetStorageForFileDescriptor(filename);
+ data_->contents_.push_back(new DataObjectImpl::StoredDataInfo(
+ ClipboardUtil::GetFileDescriptorFormat()->cfFormat, storage));
+
+ // Add CFSTR_FILECONTENTS
+ storage = GetStorageForBytes(file_contents.data(), file_contents.length());
+ data_->contents_.push_back(new DataObjectImpl::StoredDataInfo(
+ ClipboardUtil::GetFileContentFormatZero()->cfFormat, storage));
+}
+
+void OSExchangeDataProviderWin::SetHtml(const std::wstring& html,
+ const GURL& base_url) {
+ // Add both MS CF_HTML and text/html format. CF_HTML should be in utf-8.
+ std::string utf8_html = WideToUTF8(html);
+ std::string url = base_url.is_valid() ? base_url.spec() : std::string();
+
+ std::string cf_html = ClipboardUtil::HtmlToCFHtml(utf8_html, url);
+ STGMEDIUM* storage = GetStorageForBytes(cf_html.c_str(), cf_html.size());
+ data_->contents_.push_back(new DataObjectImpl::StoredDataInfo(
+ ClipboardUtil::GetHtmlFormat()->cfFormat, storage));
+
+ STGMEDIUM* storage_plain = GetStorageForBytes(utf8_html.c_str(),
+ utf8_html.size());
+ data_->contents_.push_back(new DataObjectImpl::StoredDataInfo(
+ ClipboardUtil::GetTextHtmlFormat()->cfFormat, storage_plain));
+}
+
+bool OSExchangeDataProviderWin::GetString(std::wstring* data) const {
+ return ClipboardUtil::GetPlainText(source_object_, data);
+}
+
+bool OSExchangeDataProviderWin::GetURLAndTitle(GURL* url,
+ std::wstring* title) const {
+ std::wstring url_str;
+ bool success = ClipboardUtil::GetUrl(source_object_, &url_str, title);
+ if (success) {
+ GURL test_url(url_str);
+ if (test_url.is_valid()) {
+ *url = test_url;
+ return true;
+ }
+ } else if (GetPlainTextURL(source_object_, url)) {
+ title->clear();
+ return true;
+ }
+ return false;
+}
+
+bool OSExchangeDataProviderWin::GetFilename(std::wstring* full_path) const {
+ std::vector<std::wstring> filenames;
+ bool success = ClipboardUtil::GetFilenames(source_object_, &filenames);
+ if (success)
+ full_path->assign(filenames[0]);
+ return success;
+}
+
+bool OSExchangeDataProviderWin::GetPickledData(CLIPFORMAT format,
+ Pickle* data) const {
+ DCHECK(data);
+ FORMATETC format_etc =
+ { format, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
+ bool success = false;
+ STGMEDIUM medium;
+ if (SUCCEEDED(source_object_->GetData(&format_etc, &medium))) {
+ if (medium.tymed & TYMED_HGLOBAL) {
+ ScopedHGlobal<char> c_data(medium.hGlobal);
+ DCHECK(c_data.Size() > 0);
+ // Need to subtract 1 as SetPickledData adds an extra byte to the end.
+ *data = Pickle(c_data.get(), static_cast<int>(c_data.Size() - 1));
+ success = true;
+ }
+ ReleaseStgMedium(&medium);
+ }
+ return success;
+}
+
+bool OSExchangeDataProviderWin::GetFileContents(
+ std::wstring* filename,
+ std::string* file_contents) const {
+ return ClipboardUtil::GetFileContents(source_object_, filename,
+ file_contents);
+}
+
+bool OSExchangeDataProviderWin::GetHtml(std::wstring* html,
+ GURL* base_url) const {
+ std::string url;
+ bool success = ClipboardUtil::GetHtml(source_object_, html, &url);
+ if (success)
+ *base_url = GURL(url);
+ return success;
+}
+
+bool OSExchangeDataProviderWin::HasString() const {
+ return ClipboardUtil::HasPlainText(source_object_);
+}
+
+bool OSExchangeDataProviderWin::HasURL() const {
+ return (ClipboardUtil::HasUrl(source_object_) ||
+ HasPlainTextURL(source_object_));
+}
+
+bool OSExchangeDataProviderWin::HasFile() const {
+ return ClipboardUtil::HasFilenames(source_object_);
+}
+
+bool OSExchangeDataProviderWin::HasFileContents() const {
+ return ClipboardUtil::HasFileContents(source_object_);
+}
+
+bool OSExchangeDataProviderWin::HasHtml() const {
+ return ClipboardUtil::HasHtml(source_object_);
+}
+
+bool OSExchangeDataProviderWin::HasCustomFormat(CLIPFORMAT format) const {
+ FORMATETC format_etc =
+ { format, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
+ return (source_object_->QueryGetData(&format_etc) == S_OK);
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// DataObjectImpl, IDataObject implementation:
+
+// The following function, DuplicateMedium, is derived from WCDataObject.cpp
+// in the WebKit source code. This is the license information for the file:
+/*
+ * Copyright (C) 2007 Apple Inc. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+ * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+ * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
+ * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+static void DuplicateMedium(CLIPFORMAT source_clipformat,
+ STGMEDIUM* source,
+ STGMEDIUM* destination) {
+ switch (source->tymed) {
+ case TYMED_HGLOBAL:
+ destination->hGlobal =
+ static_cast<HGLOBAL>(OleDuplicateData(
+ source->hGlobal, source_clipformat, 0));
+ break;
+ case TYMED_MFPICT:
+ destination->hMetaFilePict =
+ static_cast<HMETAFILEPICT>(OleDuplicateData(
+ source->hMetaFilePict, source_clipformat, 0));
+ break;
+ case TYMED_GDI:
+ destination->hBitmap =
+ static_cast<HBITMAP>(OleDuplicateData(
+ source->hBitmap, source_clipformat, 0));
+ break;
+ case TYMED_ENHMF:
+ destination->hEnhMetaFile =
+ static_cast<HENHMETAFILE>(OleDuplicateData(
+ source->hEnhMetaFile, source_clipformat, 0));
+ break;
+ case TYMED_FILE:
+ destination->lpszFileName =
+ static_cast<LPOLESTR>(OleDuplicateData(
+ source->lpszFileName, source_clipformat, 0));
+ break;
+ case TYMED_ISTREAM:
+ destination->pstm = source->pstm;
+ destination->pstm->AddRef();
+ break;
+ case TYMED_ISTORAGE:
+ destination->pstg = source->pstg;
+ destination->pstg->AddRef();
+ break;
+ }
+
+ destination->tymed = source->tymed;
+ destination->pUnkForRelease = source->pUnkForRelease;
+ if (destination->pUnkForRelease)
+ destination->pUnkForRelease->AddRef();
+}
+
+DataObjectImpl::DataObjectImpl() : ref_count_(0) {
+ STLDeleteContainerPointers(contents_.begin(), contents_.end());
+}
+
+DataObjectImpl::~DataObjectImpl() {
+ STLDeleteContainerPointers(contents_.begin(), contents_.end());
+}
+
+HRESULT DataObjectImpl::GetData(FORMATETC* format_etc, STGMEDIUM* medium) {
+ StoredData::const_iterator iter = contents_.begin();
+ while (iter != contents_.end()) {
+ if ((*iter)->format_etc.cfFormat == format_etc->cfFormat) {
+ DuplicateMedium((*iter)->format_etc.cfFormat, (*iter)->medium, medium);
+ return S_OK;
+ }
+ ++iter;
+ }
+
+ return DV_E_FORMATETC;
+}
+
+HRESULT DataObjectImpl::GetDataHere(FORMATETC* format_etc,
+ STGMEDIUM* medium) {
+ return DATA_E_FORMATETC;
+}
+
+HRESULT DataObjectImpl::QueryGetData(FORMATETC* format_etc) {
+ StoredData::const_iterator iter = contents_.begin();
+ while (iter != contents_.end()) {
+ if ((*iter)->format_etc.cfFormat == format_etc->cfFormat)
+ return S_OK;
+ ++iter;
+ }
+ return DV_E_FORMATETC;
+}
+
+HRESULT DataObjectImpl::GetCanonicalFormatEtc(
+ FORMATETC* format_etc, FORMATETC* result) {
+ format_etc->ptd = NULL;
+ return E_NOTIMPL;
+}
+
+HRESULT DataObjectImpl::SetData(
+ FORMATETC* format_etc, STGMEDIUM* medium, BOOL should_release) {
+ STGMEDIUM* local_medium = new STGMEDIUM;
+ if (should_release) {
+ *local_medium = *medium;
+ } else {
+ DuplicateMedium(format_etc->cfFormat, medium, local_medium);
+ }
+
+ DataObjectImpl::StoredDataInfo* info =
+ new DataObjectImpl::StoredDataInfo(format_etc->cfFormat, local_medium);
+ info->medium->tymed = format_etc->tymed;
+ info->owns_medium = !!should_release;
+ contents_.push_back(info);
+
+ return S_OK;
+}
+
+HRESULT DataObjectImpl::EnumFormatEtc(
+ DWORD direction, IEnumFORMATETC** enumerator) {
+ if (direction == DATADIR_GET) {
+ FormatEtcEnumerator* e =
+ new FormatEtcEnumerator(contents_.begin(), contents_.end());
+ e->AddRef();
+ *enumerator = e;
+ return S_OK;
+ }
+ return E_NOTIMPL;
+}
+
+HRESULT DataObjectImpl::DAdvise(
+ FORMATETC* format_etc, DWORD advf, IAdviseSink* sink, DWORD* connection) {
+ return OLE_E_ADVISENOTSUPPORTED;
+}
+
+HRESULT DataObjectImpl::DUnadvise(DWORD connection) {
+ return OLE_E_ADVISENOTSUPPORTED;
+}
+
+HRESULT DataObjectImpl::EnumDAdvise(IEnumSTATDATA** enumerator) {
+ return OLE_E_ADVISENOTSUPPORTED;
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// DataObjectImpl, IUnknown implementation:
+
+HRESULT DataObjectImpl::QueryInterface(const IID& iid, void** object) {
+ *object = NULL;
+ if (IsEqualIID(iid, IID_IUnknown) || IsEqualIID(iid, IID_IDataObject)) {
+ *object = this;
+ } else {
+ return E_NOINTERFACE;
+ }
+ AddRef();
+ return S_OK;
+}
+
+ULONG DataObjectImpl::AddRef() {
+ return InterlockedIncrement(&ref_count_);
+}
+
+ULONG DataObjectImpl::Release() {
+ if (InterlockedDecrement(&ref_count_) == 0) {
+ ULONG copied_refcnt = ref_count_;
+ delete this;
+ return copied_refcnt;
+ }
+ return ref_count_;
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// DataObjectImpl, private:
+
+static STGMEDIUM* GetStorageForBytes(const char* data, size_t bytes) {
+ HANDLE handle = GlobalAlloc(GPTR, static_cast<int>(bytes));
+ ScopedHGlobal<char> scoped(handle);
+ size_t allocated = static_cast<size_t>(GlobalSize(handle));
+ memcpy(scoped.get(), data, allocated);
+
+ STGMEDIUM* storage = new STGMEDIUM;
+ storage->hGlobal = handle;
+ storage->tymed = TYMED_HGLOBAL;
+ storage->pUnkForRelease = NULL;
+ return storage;
+}
+
+template<class T>
+static HGLOBAL CopyStringToGlobalHandle(const T& payload) {
+ int bytes = static_cast<int>(payload.size() + 1) * sizeof(T::value_type);
+ HANDLE handle = GlobalAlloc(GPTR, bytes);
+ void* data = GlobalLock(handle);
+ size_t allocated = static_cast<size_t>(GlobalSize(handle));
+ memcpy(data, payload.c_str(), allocated);
+ static_cast<T::value_type*>(data)[payload.size()] = '\0';
+ GlobalUnlock(handle);
+ return handle;
+}
+
+static STGMEDIUM* GetStorageForWString(const std::wstring& data) {
+ STGMEDIUM* storage = new STGMEDIUM;
+ storage->hGlobal = CopyStringToGlobalHandle<std::wstring>(data);
+ storage->tymed = TYMED_HGLOBAL;
+ storage->pUnkForRelease = NULL;
+ return storage;
+}
+
+static STGMEDIUM* GetStorageForString(const std::string& data) {
+ STGMEDIUM* storage = new STGMEDIUM;
+ storage->hGlobal = CopyStringToGlobalHandle<std::string>(data);
+ storage->tymed = TYMED_HGLOBAL;
+ storage->pUnkForRelease = NULL;
+ return storage;
+}
+
+static void GetInternetShortcutFileContents(const GURL& url,
+ std::string* data) {
+ DCHECK(data);
+ static const std::string kInternetShortcutFileStart =
+ "[InternetShortcut]\r\nURL=";
+ static const std::string kInternetShortcutFileEnd =
+ "\r\n";
+ *data = kInternetShortcutFileStart + url.spec() + kInternetShortcutFileEnd;
+}
+
+static void CreateValidFileNameFromTitle(const GURL& url,
+ const std::wstring& title,
+ std::wstring* validated) {
+ if (title.empty()) {
+ if (url.is_valid()) {
+ *validated = net::GetSuggestedFilename(
+ url, std::string(), std::string(), std::wstring());
+ } else {
+ // Nothing else can be done, just use a default.
+ *validated = l10n_util::GetString(IDS_APP_UNTITLED_SHORTCUT_FILE_NAME);
+ }
+ } else {
+ *validated = title;
+ file_util::ReplaceIllegalCharacters(validated, '-');
+ }
+ static const wchar_t extension[] = L".url";
+ static const size_t max_length = MAX_PATH - arraysize(extension);
+ if (validated->size() > max_length)
+ validated->erase(max_length);
+ *validated += extension;
+}
+
+static STGMEDIUM* GetStorageForFileDescriptor(
+ const std::wstring& valid_file_name) {
+ DCHECK(!valid_file_name.empty() && valid_file_name.size() + 1 <= MAX_PATH);
+ HANDLE handle = GlobalAlloc(GPTR, sizeof(FILEGROUPDESCRIPTOR));
+ FILEGROUPDESCRIPTOR* descriptor =
+ reinterpret_cast<FILEGROUPDESCRIPTOR*>(GlobalLock(handle));
+
+ descriptor->cItems = 1;
+ wcscpy_s(descriptor->fgd[0].cFileName,
+ valid_file_name.size() + 1,
+ valid_file_name.c_str());
+ descriptor->fgd[0].dwFlags = FD_LINKUI;
+
+ GlobalUnlock(handle);
+
+ STGMEDIUM* storage = new STGMEDIUM;
+ storage->hGlobal = handle;
+ storage->tymed = TYMED_HGLOBAL;
+ storage->pUnkForRelease = NULL;
+ return storage;
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// OSExchangeData, public:
+
+// static
+OSExchangeData::Provider* OSExchangeData::CreateProvider() {
+ return new OSExchangeDataProviderWin();
+}
+
+// static
+OSExchangeData::CustomFormat OSExchangeData::RegisterCustomFormat(
+ const std::string& type) {
+ return RegisterClipboardFormat(ASCIIToWide(type).c_str());
+}