--- /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 <shlobj.h>
+#include <tchar.h>
+
+#pragma warning(push, 0)
+#include <WebCore/BString.h>
+#pragma warning(pop)
+
+using namespace WebCore;
+
+
+// DefaultDownloadDelegate ----------------------------------------------------------------
+
+DefaultDownloadDelegate::DefaultDownloadDelegate()
+    : m_refCount(0)
+{
+    gClassCount++;
+}
+
+DefaultDownloadDelegate::~DefaultDownloadDelegate()
+{
+    gClassCount--;
+    HashSet<IWebDownload*>::iterator i = m_downloads.begin();
+    for (;i != m_downloads.end(); ++i)
+        (*i)->Release();
+}
+
+DefaultDownloadDelegate* DefaultDownloadDelegate::sharedInstance()
+{
+    static COMPtr<DefaultDownloadDelegate> 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<IUnknown*>(this);
+    else if (IsEqualGUID(riid, IID_IWebDownloadDelegate))
+        *ppvObject = static_cast<IWebDownloadDelegate*>(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);
+    }
+}