Index: app/os_exchange_data_provider_win.cc |
=================================================================== |
--- app/os_exchange_data_provider_win.cc (revision 71050) |
+++ app/os_exchange_data_provider_win.cc (working copy) |
@@ -1,923 +0,0 @@ |
-// Copyright (c) 2010 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/file_path.h" |
-#include "base/i18n/file_util_icu.h" |
-#include "base/logging.h" |
-#include "base/pickle.h" |
-#include "base/scoped_handle.h" |
-#include "base/stl_util-inl.h" |
-#include "base/utf_string_conversions.h" |
-#include "base/win/scoped_hglobal.h" |
-#include "googleurl/src/gurl.h" |
-#include "grit/app_strings.h" |
-#include "net/base/net_util.h" |
-#include "ui/base/clipboard/clipboard_util_win.h" |
- |
-using ui::ClipboardUtil; |
- |
-// 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 new STGMEDIUM object to hold a file. |
-static STGMEDIUM* GetStorageForFileName(const std::wstring& full_path); |
-// 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_|. |
- size_t cursor_; |
- |
- LONG ref_count_; |
- |
- DISALLOW_COPY_AND_ASSIGN(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|. |
- ULONG index = 0; |
- while (cursor_ < contents_.size() && index < count) { |
- CloneFormatEtc(contents_[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_ <= 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 |
-DataObjectImpl* OSExchangeDataProviderWin::GetDataObjectImpl( |
- const OSExchangeData& data) { |
- return static_cast<const OSExchangeDataProviderWin*>(&data.provider())-> |
- data_.get(); |
-} |
- |
-// static |
-IDataObject* OSExchangeDataProviderWin::GetIDataObject( |
- const OSExchangeData& data) { |
- return static_cast<const OSExchangeDataProviderWin*>(&data.provider())-> |
- data_object(); |
-} |
- |
-// static |
-IAsyncOperation* OSExchangeDataProviderWin::GetIAsyncOperation( |
- const OSExchangeData& data) { |
- return static_cast<const OSExchangeDataProviderWin*>(&data.provider())-> |
- async_operation(); |
-} |
- |
-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): add CF_HTML. |
- // http://code.google.com/p/chromium/issues/detail?id=6767 |
- |
- // 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) { |
- STGMEDIUM* storage = GetStorageForFileName(full_path); |
- 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(), 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, true); |
- 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) { |
- base::win::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); |
-} |
- |
-void OSExchangeDataProviderWin::SetDownloadFileInfo( |
- const OSExchangeData::DownloadFileInfo& download) { |
- // If the filename is not provided, set stoarge to NULL to indicate that |
- // the delay rendering will be used. |
- STGMEDIUM* storage = NULL; |
- if (!download.filename.empty()) |
- storage = GetStorageForFileName(download.filename.value()); |
- |
- // Add CF_HDROP. |
- DataObjectImpl::StoredDataInfo* info = new DataObjectImpl::StoredDataInfo( |
- ClipboardUtil::GetCFHDropFormat()->cfFormat, storage); |
- info->downloader = download.downloader; |
- data_->contents_.push_back(info); |
-} |
- |
-/////////////////////////////////////////////////////////////////////////////// |
-// 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() |
- : is_aborting_(false), |
- in_async_mode_(false), |
- async_operation_started_(false), |
- observer_(NULL) { |
-} |
- |
-DataObjectImpl::~DataObjectImpl() { |
- StopDownloads(); |
- STLDeleteContainerPointers(contents_.begin(), contents_.end()); |
- if (observer_) |
- observer_->OnDataObjectDisposed(); |
-} |
- |
-void DataObjectImpl::StopDownloads() { |
- for (StoredData::iterator iter = contents_.begin(); |
- iter != contents_.end(); ++iter) { |
- if ((*iter)->downloader.get()) { |
- (*iter)->downloader->Stop(); |
- (*iter)->downloader = 0; |
- } |
- } |
-} |
- |
-void DataObjectImpl::OnDownloadCompleted(const FilePath& file_path) { |
- CLIPFORMAT hdrop_format = ClipboardUtil::GetCFHDropFormat()->cfFormat; |
- DataObjectImpl::StoredData::iterator iter = contents_.begin(); |
- for (; iter != contents_.end(); ++iter) { |
- if ((*iter)->format_etc.cfFormat == hdrop_format) { |
- // Release the old storage. |
- if ((*iter)->owns_medium) { |
- ReleaseStgMedium((*iter)->medium); |
- delete (*iter)->medium; |
- } |
- |
- // Update the storage. |
- (*iter)->owns_medium = true; |
- (*iter)->medium = GetStorageForFileName(file_path.value()); |
- |
- break; |
- } |
- } |
- DCHECK(iter != contents_.end()); |
-} |
- |
-void DataObjectImpl::OnDownloadAborted() { |
-} |
- |
-HRESULT DataObjectImpl::GetData(FORMATETC* format_etc, STGMEDIUM* medium) { |
- if (is_aborting_) |
- return DV_E_FORMATETC; |
- |
- StoredData::iterator iter = contents_.begin(); |
- while (iter != contents_.end()) { |
- if ((*iter)->format_etc.cfFormat == format_etc->cfFormat && |
- (*iter)->format_etc.lindex == format_etc->lindex && |
- ((*iter)->format_etc.tymed & format_etc->tymed)) { |
- // If medium is NULL, delay-rendering will be used. |
- if ((*iter)->medium) { |
- DuplicateMedium((*iter)->format_etc.cfFormat, (*iter)->medium, medium); |
- } else { |
- // Check if the left button is down. |
- bool is_left_button_down = (GetKeyState(VK_LBUTTON) & 0x8000) != 0; |
- |
- bool wait_for_data = false; |
- if ((*iter)->in_delay_rendering) { |
- // Make sure the left button is up. Sometimes the drop target, like |
- // Shell, might be too aggresive in calling GetData when the left |
- // button is not released. |
- if (is_left_button_down) |
- return DV_E_FORMATETC; |
- |
- // In async mode, we do not want to start waiting for the data before |
- // the async operation is started. This is because we want to postpone |
- // until Shell kicks off a background thread to do the work so that |
- // we do not block the UI thread. |
- if (!in_async_mode_ || async_operation_started_) |
- wait_for_data = true; |
- } else { |
- // If the left button is up and the target has not requested the data |
- // yet, it probably means that the target does not support delay- |
- // rendering. So instead, we wait for the data. |
- if (is_left_button_down) { |
- (*iter)->in_delay_rendering = true; |
- memset(medium, 0, sizeof(STGMEDIUM)); |
- } else { |
- wait_for_data = true; |
- } |
- } |
- |
- if (!wait_for_data) |
- return DV_E_FORMATETC; |
- |
- // Notify the observer we start waiting for the data. This gives |
- // an observer a chance to end the drag and drop. |
- if (observer_) |
- observer_->OnWaitForData(); |
- |
- // Now we can start the download. |
- if ((*iter)->downloader.get()) { |
- if (!(*iter)->downloader->Start(this)) { |
- is_aborting_ = true; |
- return DV_E_FORMATETC; |
- } |
- } |
- |
- // The stored data should have been updated with the final version. |
- // So we just need to call this function again to retrieve it. |
- return GetData(format_etc, 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, IAsyncOperation implementation: |
- |
-HRESULT DataObjectImpl::EndOperation( |
- HRESULT result, IBindCtx* reserved, DWORD effects) { |
- async_operation_started_ = false; |
- return S_OK; |
-} |
- |
-HRESULT DataObjectImpl::GetAsyncMode(BOOL* is_op_async) { |
- *is_op_async = in_async_mode_ ? TRUE : FALSE; |
- return S_OK; |
-} |
- |
-HRESULT DataObjectImpl::InOperation(BOOL* in_async_op) { |
- *in_async_op = async_operation_started_ ? TRUE : FALSE; |
- return S_OK; |
-} |
- |
-HRESULT DataObjectImpl::SetAsyncMode(BOOL do_op_async) { |
- in_async_mode_ = (do_op_async == TRUE); |
- return S_OK; |
-} |
- |
-HRESULT DataObjectImpl::StartOperation(IBindCtx* reserved) { |
- async_operation_started_ = true; |
- return S_OK; |
-} |
- |
-/////////////////////////////////////////////////////////////////////////////// |
-// DataObjectImpl, IUnknown implementation: |
- |
-HRESULT DataObjectImpl::QueryInterface(const IID& iid, void** object) { |
- if (!object) |
- return E_POINTER; |
- if (IsEqualIID(iid, IID_IDataObject) || IsEqualIID(iid, IID_IUnknown)) { |
- *object = static_cast<IDataObject*>(this); |
- } else if (in_async_mode_ && IsEqualIID(iid, IID_IAsyncOperation)) { |
- *object = static_cast<IAsyncOperation*>(this); |
- } else { |
- *object = NULL; |
- return E_NOINTERFACE; |
- } |
- AddRef(); |
- return S_OK; |
-} |
- |
-ULONG DataObjectImpl::AddRef() { |
- base::RefCountedThreadSafe<DownloadFileObserver>::AddRef(); |
- return 0; |
-} |
- |
-ULONG DataObjectImpl::Release() { |
- base::RefCountedThreadSafe<DownloadFileObserver>::Release(); |
- return 0; |
-} |
- |
-/////////////////////////////////////////////////////////////////////////////// |
-// DataObjectImpl, private: |
- |
-static STGMEDIUM* GetStorageForBytes(const char* data, size_t bytes) { |
- HANDLE handle = GlobalAlloc(GPTR, static_cast<int>(bytes)); |
- base::win::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(), FilePath()).ToWStringHack(); |
- } else { |
- // Nothing else can be done, just use a default. |
- *validated = |
- l10n_util::GetStringUTF16(IDS_APP_UNTITLED_SHORTCUT_FILE_NAME); |
- } |
- } else { |
- *validated = title; |
- file_util::ReplaceIllegalCharactersInPath(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* GetStorageForFileName(const std::wstring& full_path) { |
- const size_t kDropSize = sizeof(DROPFILES); |
- const size_t kTotalBytes = |
- kDropSize + (full_path.length() + 2) * sizeof(wchar_t); |
- HANDLE hdata = GlobalAlloc(GMEM_MOVEABLE, kTotalBytes); |
- |
- base::win::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*>( |
- reinterpret_cast<BYTE*>(drop_files) + kDropSize); |
- 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 |
- |
- STGMEDIUM* storage = new STGMEDIUM; |
- storage->tymed = TYMED_HGLOBAL; |
- storage->hGlobal = drop_files; |
- storage->pUnkForRelease = NULL; |
- return storage; |
-} |
- |
-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()); |
-} |