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#ifndef DefaultDownloadDelegate_h 26#define DefaultDownloadDelegate_h 27 28#include "WebKit.h" 29#include <wtf/HashSet.h> 30 31#if USE(CFNETWORK) 32#include <CFNetwork/CFURLDownloadPriv.h> 33#endif 34 35class DefaultDownloadDelegate : public IWebDownloadDelegate 36{ 37public: 38 static DefaultDownloadDelegate* sharedInstance(); 39 static DefaultDownloadDelegate* createInstance(); 40private: 41 DefaultDownloadDelegate(); 42 ~DefaultDownloadDelegate(); 43public: 44 // IUnknown 45 virtual HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void** ppvObject); 46 virtual ULONG STDMETHODCALLTYPE AddRef(void); 47 virtual ULONG STDMETHODCALLTYPE Release(void); 48 49 // IWebDownloadDelegate 50 virtual HRESULT STDMETHODCALLTYPE decideDestinationWithSuggestedFilename(IWebDownload *download, BSTR filename); 51 virtual HRESULT STDMETHODCALLTYPE didCancelAuthenticationChallenge(IWebDownload* download, IWebURLAuthenticationChallenge* challenge); 52 virtual HRESULT STDMETHODCALLTYPE didCreateDestination(IWebDownload* download, BSTR destination); 53 virtual HRESULT STDMETHODCALLTYPE didFailWithError(IWebDownload* download, IWebError* error); 54 virtual HRESULT STDMETHODCALLTYPE didReceiveAuthenticationChallenge(IWebDownload* download, IWebURLAuthenticationChallenge* challenge); 55 virtual HRESULT STDMETHODCALLTYPE didReceiveDataOfLength(IWebDownload* download, unsigned length); 56 virtual HRESULT STDMETHODCALLTYPE didReceiveResponse(IWebDownload* download, IWebURLResponse* response); 57 virtual HRESULT STDMETHODCALLTYPE shouldDecodeSourceDataOfMIMEType(IWebDownload* download, BSTR encodingType, BOOL* shouldDecode); 58 virtual HRESULT STDMETHODCALLTYPE willResumeWithResponse(IWebDownload* download, IWebURLResponse* response, long long fromByte); 59 virtual HRESULT STDMETHODCALLTYPE willSendRequest(IWebDownload* download, IWebMutableURLRequest* request, IWebURLResponse* redirectResponse, IWebMutableURLRequest** finalRequest); 60 virtual HRESULT STDMETHODCALLTYPE didBegin(IWebDownload* download); 61 virtual HRESULT STDMETHODCALLTYPE didFinish(IWebDownload* download); 62 63 // DefaultDownloadDelegate 64 void registerDownload(IWebDownload*); 65 void unregisterDownload(IWebDownload*); 66protected: 67 ULONG m_refCount; 68 69 HashSet<IWebDownload*> m_downloads; 70}; 71 72#endif 73