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