Home | History | Annotate | Download | only in win 
      1 /*
      2  * Copyright (C) 2008 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 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 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 "WebIconFetcher.h"
     29 
     30 #include "MemoryStream.h"
     31 #include <WebCore/COMPtr.h>
     32 #include <WebCore/SharedBuffer.h>
     33 #include <wtf/PassRefPtr.h>
     34 
     35 using namespace WebCore;
     36 
     37 class WebIconFetcherClient : public IconFetcherClient {
     38 public:
     39     WebIconFetcherClient(IWebIconFetcherDelegate* delegate)
     40         : m_fetcher(0)
     41         , m_delegate(delegate)
     42     {
     43     }
     44 
     45     virtual void finishedFetchingIcon(PassRefPtr<SharedBuffer> iconData)
     46     {
     47         COMPtr<MemoryStream> memoryStream;
     48         if (iconData)
     49             memoryStream = MemoryStream::createInstance(iconData);
     50 
     51         m_delegate->finishedLoadingIcon(m_fetcher, memoryStream.get());
     52 
     53         delete this;
     54     }
     55 
     56    void setFetcher(WebIconFetcher *fetcher) { m_fetcher = fetcher; }
     57 
     58 private:
     59     WebIconFetcher* m_fetcher;
     60     COMPtr<IWebIconFetcherDelegate> m_delegate;
     61 };
     62 
     63 // WebIconFetcher -------------------------------------------------------------------
     64 
     65 WebIconFetcher* WebIconFetcher::fetchApplicationIcon(Frame* frame, IWebIconFetcherDelegate* delegate)
     66 {
     67     WebIconFetcherClient* client = new WebIconFetcherClient(delegate);
     68 
     69     RefPtr<IconFetcher> fetcher = IconFetcher::create(frame, client);
     70 
     71     if (!fetcher)
     72         return 0;
     73 
     74     COMPtr<WebIconFetcher> iconFetcher = new WebIconFetcher(fetcher.release());
     75     client->setFetcher(iconFetcher.get());
     76 
     77     return iconFetcher.releaseRef();
     78 }
     79 
     80 WebIconFetcher::WebIconFetcher(PassRefPtr<IconFetcher> iconFetcher)
     81     : m_refCount(0)
     82     , m_iconFetcher(iconFetcher)
     83 {
     84     gClassCount++;
     85 }
     86 
     87 WebIconFetcher::~WebIconFetcher()
     88 {
     89     gClassCount--;
     90 }
     91 
     92 // IUnknown -------------------------------------------------------------------
     93 
     94 HRESULT STDMETHODCALLTYPE WebIconFetcher::QueryInterface(REFIID riid, void** ppvObject)
     95 {
     96     *ppvObject = 0;
     97     if (IsEqualGUID(riid, IID_IUnknown))
     98         *ppvObject = static_cast<IUnknown*>(this);
     99     else if (IsEqualGUID(riid, IID_IWebIconFetcher))
    100         *ppvObject = static_cast<IWebIconFetcher*>(this);
    101     else
    102         return E_NOINTERFACE;
    103 
    104     AddRef();
    105     return S_OK;
    106 }
    107 
    108 ULONG STDMETHODCALLTYPE WebIconFetcher::AddRef()
    109 {
    110     return ++m_refCount;
    111 }
    112 
    113 ULONG STDMETHODCALLTYPE WebIconFetcher::Release()
    114 {
    115     ULONG newRef = --m_refCount;
    116     if (!newRef)
    117         delete this;
    118 
    119     return newRef;
    120 }
    121 
    122 // IUnknown -------------------------------------------------------------------
    123 
    124 HRESULT STDMETHODCALLTYPE WebIconFetcher::cancel()
    125 {
    126     m_iconFetcher->cancel();
    127 
    128     return S_OK;
    129 }
    130