diff options
Diffstat (limited to 'app/os_exchange_data_win.cc')
-rw-r--r-- | app/os_exchange_data_win.cc | 706 |
1 files changed, 706 insertions, 0 deletions
diff --git a/app/os_exchange_data_win.cc b/app/os_exchange_data_win.cc new file mode 100644 index 0000000..c9e2311 --- /dev/null +++ b/app/os_exchange_data_win.cc @@ -0,0 +1,706 @@ +// Copyright (c) 2006-2008 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 <shlobj.h> + +#include "app/os_exchange_data.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/generated_resources.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(OSExchangeData::StoredData::const_iterator begin, + OSExchangeData::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( + OSExchangeData::StoredData::const_iterator start, + OSExchangeData::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; +} + +/////////////////////////////////////////////////////////////////////////////// +// OSExchangeData, public: + +// static +bool OSExchangeData::HasPlainTextURL(IDataObject* source) { + std::wstring plain_text; + return (ClipboardUtil::GetPlainText(source, &plain_text) && + !plain_text.empty() && GURL(plain_text).is_valid()); +} + +// static +bool OSExchangeData::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; +} + +OSExchangeData::OSExchangeData() + : ref_count_(0) { +} + +OSExchangeData::OSExchangeData(IDataObject* source) + : ref_count_(0) { + source_object_ = source; +} + +OSExchangeData::~OSExchangeData() { + STLDeleteContainerPointers(contents_.begin(), contents_.end()); +} + +void OSExchangeData::SetString(const std::wstring& data) { + STGMEDIUM* storage = GetStorageForWString(data); + contents_.push_back(new StoredDataInfo(CF_UNICODETEXT, storage)); + + // Also add plain text. + storage = GetStorageForString(WideToUTF8(data)); + contents_.push_back(new StoredDataInfo(CF_TEXT, storage)); +} + +void OSExchangeData::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); + contents_.push_back(new 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())); + contents_.push_back(new StoredDataInfo( + ClipboardUtil::GetUrlWFormat()->cfFormat, storage)); + storage = GetStorageForString(url.spec()); + contents_.push_back(new 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())); + contents_.push_back(new StoredDataInfo(CF_UNICODETEXT, storage)); + storage = GetStorageForString(url.spec()); + contents_.push_back(new StoredDataInfo(CF_TEXT, storage)); +} + +void OSExchangeData::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 + StoredDataInfo* info = new StoredDataInfo(CF_HDROP, storage); + contents_.push_back(info); +} + +void OSExchangeData::SetPickledData(CLIPFORMAT format, const Pickle& data) { + STGMEDIUM* storage = GetStorageForString( + std::string(static_cast<const char *>(data.data()), + static_cast<size_t>(data.size()))); + contents_.push_back(new StoredDataInfo(format, storage)); +} + +void OSExchangeData::SetFileContents(const std::wstring& filename, + const std::string& file_contents) { + // Add CFSTR_FILEDESCRIPTOR + STGMEDIUM* storage = GetStorageForFileDescriptor(filename); + contents_.push_back(new StoredDataInfo( + ClipboardUtil::GetFileDescriptorFormat()->cfFormat, storage)); + + // Add CFSTR_FILECONTENTS + storage = GetStorageForBytes(file_contents.data(), file_contents.length()); + contents_.push_back(new StoredDataInfo( + ClipboardUtil::GetFileContentFormatZero()->cfFormat, storage)); +} + +void OSExchangeData::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()); + contents_.push_back(new StoredDataInfo( + ClipboardUtil::GetHtmlFormat()->cfFormat, storage)); + + STGMEDIUM* storage_plain = GetStorageForBytes(utf8_html.c_str(), + utf8_html.size()); + contents_.push_back(new StoredDataInfo( + ClipboardUtil::GetTextHtmlFormat()->cfFormat, storage_plain)); +} + +bool OSExchangeData::GetString(std::wstring* data) const { + return ClipboardUtil::GetPlainText(source_object_, data); +} + +bool OSExchangeData::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 OSExchangeData::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 OSExchangeData::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 OSExchangeData::GetFileContents(std::wstring* filename, + std::string* file_contents) const { + return ClipboardUtil::GetFileContents(source_object_, filename, + file_contents); +} + +bool OSExchangeData::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 OSExchangeData::HasString() const { + return ClipboardUtil::HasPlainText(source_object_); +} + +bool OSExchangeData::HasURL() const { + return (ClipboardUtil::HasUrl(source_object_) || + HasPlainTextURL(source_object_)); +} + +bool OSExchangeData::HasFile() const { + return ClipboardUtil::HasFilenames(source_object_); +} + +bool OSExchangeData::HasFormat(CLIPFORMAT format) const { + FORMATETC format_etc = + { format, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL }; + return (source_object_->QueryGetData(&format_etc) == S_OK); +} + +/////////////////////////////////////////////////////////////////////////////// +// OSExchangeData, 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(); +} + +HRESULT OSExchangeData::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 OSExchangeData::GetDataHere(FORMATETC* format_etc, STGMEDIUM* medium) { + return DATA_E_FORMATETC; +} + +HRESULT OSExchangeData::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 OSExchangeData::GetCanonicalFormatEtc( + FORMATETC* format_etc, FORMATETC* result) { + format_etc->ptd = NULL; + return E_NOTIMPL; +} + +HRESULT OSExchangeData::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); + } + + StoredDataInfo* info = + new 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 OSExchangeData::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 OSExchangeData::DAdvise( + FORMATETC* format_etc, DWORD advf, IAdviseSink* sink, DWORD* connection) { + return OLE_E_ADVISENOTSUPPORTED; +} + +HRESULT OSExchangeData::DUnadvise(DWORD connection) { + return OLE_E_ADVISENOTSUPPORTED; +} + +HRESULT OSExchangeData::EnumDAdvise(IEnumSTATDATA** enumerator) { + return OLE_E_ADVISENOTSUPPORTED; +} + +/////////////////////////////////////////////////////////////////////////////// +// OSExchangeData, IUnknown implementation: + +HRESULT OSExchangeData::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 OSExchangeData::AddRef() { + return InterlockedIncrement(&ref_count_); +} + +ULONG OSExchangeData::Release() { + if (InterlockedDecrement(&ref_count_) == 0) { + ULONG copied_refcnt = ref_count_; + delete this; + return copied_refcnt; + } + return ref_count_; +} + +/////////////////////////////////////////////////////////////////////////////// +// OSExchangeData, 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_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; +} |