Source/WebKit/win/WebDownload.cpp

   1 /*
   2  * Copyright (C) 2007 Apple Inc.  All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
  14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
  17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include "config.h"
  27 #include "WebKitDLL.h"
  28 #include "WebDownload.h"
  29
  30 #include "DefaultDownloadDelegate.h"
  31 #include "MarshallingHelpers.h"
  32 #include "WebError.h"
  33 #include "WebKit.h"
  34 #include "WebKitLogging.h"
  35 #include "WebMutableURLRequest.h"
  36 #include "WebURLAuthenticationChallenge.h"
  37 #include "WebURLCredential.h"
  38 #include "WebURLResponse.h"
  39 #include <wtf/text/CString.h>
  40
  41 #include <WebCore/BString.h>
  42 #include <WebCore/DownloadBundle.h>
  43 #include <WebCore/NotImplemented.h>
  44 #include <WebCore/ResourceError.h>
  45 #include <WebCore/ResourceHandle.h>
  46 #include <WebCore/ResourceRequest.h>
  47 #include <WebCore/ResourceResponse.h>
  48 #include <wtf/CurrentTime.h>
  49 #include <wtf/StdLibExtras.h>
  50
  51 using namespace WebCore;
  52
  53 // WebDownload ----------------------------------------------------------------
  54
  55 WebDownload::WebDownload()
  56     : m_refCount(0)
  57 {
  58     gClassCount++;
  59     gClassNameCount.add("WebDownload");
  60 }
  61
  62 WebDownload::~WebDownload()
  63 {
  64     LOG(Download, "WebDownload - Destroying download (%p)", this);
  65     cancel();
  66     gClassCount--;
  67     gClassNameCount.remove("WebDownload");
  68 }
  69
  70 WebDownload* WebDownload::createInstance()
  71 {
  72     WebDownload* instance = new WebDownload();
  73     instance->AddRef();
  74     return instance;
  75 }
  76
  77 WebDownload* WebDownload::createInstance(ResourceHandle* handle, const ResourceRequest& request, const ResourceResponse& response, IWebDownloadDelegate* delegate)
  78 {
  79     WebDownload* instance = new WebDownload();
  80     instance->AddRef();
  81     instance->init(handle, request, response, delegate);
  82     return instance;
  83 }
  84
  85 WebDownload* WebDownload::createInstance(const KURL& url, IWebDownloadDelegate* delegate)
  86 {
  87     WebDownload* instance = new WebDownload();
  88     instance->AddRef();
  89     instance->init(url, delegate);
  90     return instance;
  91 }
  92
  93 // IUnknown -------------------------------------------------------------------
  94
  95 HRESULT STDMETHODCALLTYPE WebDownload::QueryInterface(REFIID riid, void** ppvObject)
  96 {
  97     *ppvObject = 0;
  98     if (IsEqualGUID(riid, IID_IUnknown))
  99         *ppvObject = static_cast<IWebDownload*>(this);
 100     else if (IsEqualGUID(riid, IID_IWebDownload))
 101         *ppvObject = static_cast<IWebDownload*>(this);
 102     else if (IsEqualGUID(riid, IID_IWebURLAuthenticationChallengeSender))
 103         *ppvObject = static_cast<IWebURLAuthenticationChallengeSender*>(this);
 104     else if (IsEqualGUID(riid, CLSID_WebDownload))
 105         *ppvObject = static_cast<WebDownload*>(this);
 106     else
 107         return E_NOINTERFACE;
 108
 109     AddRef();
 110     return S_OK;
 111 }
 112
 113 ULONG STDMETHODCALLTYPE WebDownload::AddRef(void)
 114 {
 115     return ++m_refCount;
 116 }
 117
 118 ULONG STDMETHODCALLTYPE WebDownload::Release(void)
 119 {
 120     ULONG newRef = --m_refCount;
 121     if (!newRef)
 122         delete(this);
 123
 124     return newRef;
 125 }
 126
 127 // IWebDownload -------------------------------------------------------------------
 128
 129 HRESULT STDMETHODCALLTYPE WebDownload::canResumeDownloadDecodedWithEncodingMIMEType(
 130         /* [in] */ BSTR,
 131         /* [out, retval] */ BOOL*)
 132 {
 133     notImplemented();
 134     return E_FAIL;
 135 }
 136
 137 HRESULT STDMETHODCALLTYPE WebDownload::bundlePathForTargetPath(
 138         /* [in] */ BSTR targetPath,
 139         /* [out, retval] */ BSTR* bundlePath)
 140 {
 141     if (!targetPath)
 142         return E_INVALIDARG;
 143
 144     String bundle(targetPath, SysStringLen(targetPath));
 145     if (bundle.isEmpty())
 146         return E_INVALIDARG;
 147
 148     if (bundle[bundle.length()-1] == '/')
 149         bundle.truncate(1);
 150
 151     bundle += DownloadBundle::fileExtension();
 152     *bundlePath = SysAllocStringLen(bundle.characters(), bundle.length());
 153     if (!*bundlePath)
 154        return E_FAIL;
 155     return S_OK;
 156 }
 157
 158 HRESULT STDMETHODCALLTYPE WebDownload::request(
 159         /* [out, retval] */ IWebURLRequest** request)
 160 {
 161     if (request) {
 162         *request = m_request.get();
 163         if (*request)
 164             (*request)->AddRef();
 165     }
 166     return S_OK;
 167 }