diff -r 000000000000 -r dd21522fd290 webengine/osswebengine/WebKit/win/DefaultDownloadDelegate.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/webengine/osswebengine/WebKit/win/DefaultDownloadDelegate.cpp Mon Mar 30 12:54:55 2009 +0300 @@ -0,0 +1,243 @@ +/* + * 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. + */ +#include "config.h" +#include "WebKitDLL.h" +#include "DefaultDownloadDelegate.h" + +#include "MarshallingHelpers.h" +#include "WebKit.h" +#include "WebKitLogging.h" +#include "WebMutableURLRequest.h" + +#include +#include + +#pragma warning(push, 0) +#include +#pragma warning(pop) + +using namespace WebCore; + + +// DefaultDownloadDelegate ---------------------------------------------------------------- + +DefaultDownloadDelegate::DefaultDownloadDelegate() + : m_refCount(0) +{ + gClassCount++; +} + +DefaultDownloadDelegate::~DefaultDownloadDelegate() +{ + gClassCount--; + HashSet::iterator i = m_downloads.begin(); + for (;i != m_downloads.end(); ++i) + (*i)->Release(); +} + +DefaultDownloadDelegate* DefaultDownloadDelegate::sharedInstance() +{ + static COMPtr shared; + if (!shared) + shared.adoptRef(DefaultDownloadDelegate::createInstance()); + return shared.get(); +} + +DefaultDownloadDelegate* DefaultDownloadDelegate::createInstance() +{ + DefaultDownloadDelegate* instance = new DefaultDownloadDelegate(); + instance->AddRef(); + return instance; +} + +// IUnknown ------------------------------------------------------------------- + +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::QueryInterface(REFIID riid, void** ppvObject) +{ + *ppvObject = 0; + if (IsEqualGUID(riid, IID_IUnknown)) + *ppvObject = static_cast(this); + else if (IsEqualGUID(riid, IID_IWebDownloadDelegate)) + *ppvObject = static_cast(this); + else + return E_NOINTERFACE; + + AddRef(); + return S_OK; +} + +ULONG STDMETHODCALLTYPE DefaultDownloadDelegate::AddRef() +{ + return ++m_refCount; +} + +ULONG STDMETHODCALLTYPE DefaultDownloadDelegate::Release() +{ + ULONG newRef = --m_refCount; + if (!newRef) + delete(this); + + return newRef; +} + +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::decideDestinationWithSuggestedFilename(IWebDownload *download, BSTR filename) +{ + LOG(Download, "DefaultDownloadDelegate %p - decideDestinationWithSuggestedFilename %s", download, String(filename, SysStringLen(filename)).ascii().data()); + + TCHAR pathChars[MAX_PATH]; + if (FAILED(SHGetFolderPath(0, CSIDL_DESKTOPDIRECTORY | CSIDL_FLAG_CREATE, 0, 0, pathChars))) { + if (FAILED(download->setDestination(filename, true))) { + LOG_ERROR("Failed to set destination on file"); + return E_FAIL; + } + return S_OK; + } + + size_t fullLength = _tcslen(pathChars) + SysStringLen(filename) + 2; + BSTR full = SysAllocStringLen(0, (UINT)fullLength); + if (!full) + return E_OUTOFMEMORY; + + _tcscpy_s(full, fullLength, pathChars); + _tcscat_s(full, fullLength, _T("\\")); + _tcscat_s(full, fullLength, filename); + BString fullPath; + fullPath.adoptBSTR(full); + +#ifndef NDEBUG + String debug((BSTR)fullPath, SysStringLen(BSTR(fullPath))); + LOG(Download, "Setting path to %s", debug.ascii().data()); +#endif + + if (FAILED(download->setDestination(fullPath, true))) { + LOG_ERROR("Failed to set destination on file"); + return E_FAIL; + } + return S_OK; +} +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::didCancelAuthenticationChallenge(IWebDownload* download, IWebURLAuthenticationChallenge* challenge) +{ + LOG(Download, "DefaultDownloadDelegate %p - didCancelAuthenticationChallenge %p", download, challenge); + download = 0; + challenge = 0; + return S_OK; +} +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::didCreateDestination(IWebDownload* download, BSTR destination) +{ + LOG(Download, "DefaultDownloadDelegate %p - didCreateDestination %s", download, String(destination, SysStringLen(destination)).ascii().data()); + download = 0; + destination = 0; + return S_OK; +} + +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::didReceiveAuthenticationChallenge(IWebDownload* download, IWebURLAuthenticationChallenge* challenge) +{ + LOG(Download, "DefaultDownloadDelegate %p - didReceiveAuthenticationChallenge %p", download, challenge); + download = 0; + challenge = 0; + return S_OK; +} + +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::didReceiveDataOfLength(IWebDownload* download, unsigned length) +{ + LOG(Download, "DefaultDownloadDelegate %p - didReceiveDataOfLength %i", download, length); + download = 0; + length = 0; + return S_OK; +} + +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::didReceiveResponse(IWebDownload* download, IWebURLResponse* response) +{ + LOG(Download, "DefaultDownloadDelegate %p - didReceiveResponse %p", download, response); + download = 0; + response = 0; + return S_OK; +} + +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::shouldDecodeSourceDataOfMIMEType(IWebDownload* download, BSTR encodingType, BOOL* shouldDecode) +{ + LOG(Download, "DefaultDownloadDelegate %p - shouldDecodeSourceDataOfMIMEType %s", download, String(encodingType, SysStringLen(encodingType)).ascii().data()); + download = 0; + encodingType = 0; + *shouldDecode = false; + return S_OK; +} + +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::willResumeWithResponse(IWebDownload* download, IWebURLResponse* response, long long fromByte) +{ + LOG(Download, "DefaultDownloadDelegate %p - willResumeWithResponse %p, %q", download, response, fromByte); + download = 0; + response = 0; + fromByte = 0; + return S_OK; +} + +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::willSendRequest(IWebDownload* download, IWebMutableURLRequest* request, + IWebURLResponse* redirectResponse, IWebMutableURLRequest** finalRequest) +{ + LOG(Download, "DefaultDownloadDelegate %p - willSendRequest %p %p", download, request, redirectResponse); + download = 0; + redirectResponse = 0; + *finalRequest = request; + return S_OK; +} + +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::didBegin(IWebDownload* download) +{ + LOG(Download, "DefaultDownloadDelegate %p - didBegin", download); + registerDownload(download); + return S_OK; +} + +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::didFinish(IWebDownload* download) +{ + LOG(Download, "DefaultDownloadDelegate %p - didFinish", download); + unregisterDownload(download); + return S_OK; +} + +HRESULT STDMETHODCALLTYPE DefaultDownloadDelegate::didFailWithError(IWebDownload* download, IWebError* error) +{ + LOG(Download, "DefaultDownloadDelegate %p - didFailWithError %p", download, error); + unregisterDownload(download); + error = 0; + return S_OK; +} + +void DefaultDownloadDelegate::registerDownload(IWebDownload* download) +{ + if (m_downloads.contains(download)) + return; + download->AddRef(); + m_downloads.add(download); +} + +void DefaultDownloadDelegate::unregisterDownload(IWebDownload* download) +{ + if (m_downloads.contains(download)) { + download->Release(); + m_downloads.remove(download); + } +}