Home | History | Annotate | Download | only in win 
      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 "WebResource.h"
     29 
     30 #include "MarshallingHelpers.h"
     31 #include "MemoryStream.h"
     32 #include <WebCore/BString.h>
     33 
     34 using namespace WebCore;
     35 
     36 // WebResource ---------------------------------------------------------------------
     37 
     38 WebResource::WebResource(IStream* data, const WebCore::KURL& url, const WTF::String& mimeType, const WTF::String& textEncodingName, const WTF::String& frameName)
     39     : m_refCount(0)
     40     , m_data(data)
     41     , m_url(url)
     42     , m_mimeType(mimeType)
     43     , m_textEncodingName(textEncodingName)
     44     , m_frameName(frameName)
     45 {
     46     gClassCount++;
     47     gClassNameCount.add("WebResource");
     48 }
     49 
     50 WebResource::~WebResource()
     51 {
     52     gClassCount--;
     53     gClassNameCount.remove("WebResource");
     54 }
     55 
     56 WebResource* WebResource::createInstance(PassRefPtr<WebCore::SharedBuffer> data, const WebCore::ResourceResponse& response)
     57 {
     58     COMPtr<MemoryStream> memoryStream = MemoryStream::createInstance(data);
     59 
     60     WebResource* instance = new WebResource(memoryStream.get(), response.url(), response.mimeType(), response.textEncodingName(), String());
     61     instance->AddRef();
     62     return instance;
     63 }
     64 
     65 // IUnknown -------------------------------------------------------------------
     66 
     67 HRESULT STDMETHODCALLTYPE WebResource::QueryInterface(REFIID riid, void** ppvObject)
     68 {
     69     *ppvObject = 0;
     70     if (IsEqualGUID(riid, IID_IUnknown))
     71         *ppvObject = static_cast<IUnknown*>(this);
     72     else if (IsEqualGUID(riid, IID_IWebResource))
     73         *ppvObject = static_cast<IWebResource*>(this);
     74     else
     75         return E_NOINTERFACE;
     76 
     77     AddRef();
     78     return S_OK;
     79 }
     80 
     81 ULONG STDMETHODCALLTYPE WebResource::AddRef(void)
     82 {
     83     return ++m_refCount;
     84 }
     85 
     86 ULONG STDMETHODCALLTYPE WebResource::Release(void)
     87 {
     88     ULONG newRef = --m_refCount;
     89     if (!newRef)
     90         delete(this);
     91 
     92     return newRef;
     93 }
     94 
     95 // WebResource ------------------------------------------------------------------
     96 
     97 HRESULT STDMETHODCALLTYPE WebResource::initWithData(
     98     /* [in] */ IStream *data,
     99     /* [in] */ BSTR url,
    100     /* [in] */ BSTR mimeType,
    101     /* [in] */ BSTR textEncodingName,
    102     /* [in] */ BSTR frameName)
    103 {
    104     m_data = data;
    105     m_url = MarshallingHelpers::BSTRToKURL(url);
    106     m_mimeType = String(mimeType);
    107     m_textEncodingName = String(textEncodingName);
    108     m_frameName = String(frameName);
    109 
    110     return S_OK;
    111 }
    112 
    113 
    114 HRESULT STDMETHODCALLTYPE WebResource::data(
    115     /* [retval][out] */ IStream **data)
    116 {
    117     return m_data.copyRefTo(data);
    118 }
    119 
    120 HRESULT STDMETHODCALLTYPE WebResource::URL(
    121     /* [retval][out] */ BSTR *url)
    122 {
    123     if (!url) {
    124         ASSERT_NOT_REACHED();
    125         return E_POINTER;
    126     }
    127 
    128     *url = BString(String(m_url.string())).release();
    129     return S_OK;
    130 }
    131 
    132 HRESULT STDMETHODCALLTYPE WebResource::MIMEType(
    133     /* [retval][out] */ BSTR *mime)
    134 {
    135     if (!mime) {
    136         ASSERT_NOT_REACHED();
    137         return E_POINTER;
    138     }
    139 
    140     *mime = BString(m_mimeType).release();
    141     return S_OK;
    142 }
    143 
    144 HRESULT STDMETHODCALLTYPE WebResource::textEncodingName(
    145     /* [retval][out] */ BSTR *encodingName)
    146 {
    147     if (!encodingName) {
    148         ASSERT_NOT_REACHED();
    149         return E_POINTER;
    150     }
    151 
    152     *encodingName = BString(m_textEncodingName).release();
    153     return S_OK;
    154 }
    155 
    156 HRESULT STDMETHODCALLTYPE WebResource::frameName(
    157     /* [retval][out] */ BSTR *name)
    158 {
    159     if (!name) {
    160         ASSERT_NOT_REACHED();
    161         return E_POINTER;
    162     }
    163 
    164     *name = BString(m_frameName).release();
    165     return S_OK;
    166 }
    167