1 /* 2 * Copyright (C) 2005, 2006, 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 * 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of 14 * its contributors may be used to endorse or promote products derived 15 * from this software without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY 18 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 20 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY 21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 22 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 23 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 24 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29 #ifndef MainResourceLoader_h 30 #define MainResourceLoader_h 31 32 #include "FrameLoaderTypes.h" 33 #include "ResourceLoader.h" 34 #include "SubstituteData.h" 35 #include <wtf/Forward.h> 36 37 #if HAVE(RUNLOOP_TIMER) 38 #include "RunLoopTimer.h" 39 #else 40 #include "Timer.h" 41 #endif 42 43 namespace WebCore { 44 45 class FormState; 46 class ResourceRequest; 47 48 class MainResourceLoader : public ResourceLoader { 49 public: 50 static PassRefPtr<MainResourceLoader> create(Frame*); 51 virtual ~MainResourceLoader(); 52 53 virtual bool load(const ResourceRequest&, const SubstituteData&); 54 virtual void addData(const char*, int, bool allAtOnce); 55 56 virtual void setDefersLoading(bool); 57 58 virtual void willSendRequest(ResourceRequest&, const ResourceResponse& redirectResponse); 59 virtual void didReceiveResponse(const ResourceResponse&); 60 virtual void didReceiveData(const char*, int, long long encodedDataLength, bool allAtOnce); 61 virtual void didFinishLoading(double finishTime); 62 virtual void didFail(const ResourceError&); 63 64 #if HAVE(RUNLOOP_TIMER) 65 typedef RunLoopTimer<MainResourceLoader> MainResourceLoaderTimer; 66 #else 67 typedef Timer<MainResourceLoader> MainResourceLoaderTimer; 68 #endif 69 70 void handleDataLoadNow(MainResourceLoaderTimer*); 71 isLoadingMultipartContent()72 bool isLoadingMultipartContent() const { return m_loadingMultipartContent; } 73 74 private: 75 MainResourceLoader(Frame*); 76 77 virtual void didCancel(const ResourceError&); 78 79 bool loadNow(ResourceRequest&); 80 81 void handleEmptyLoad(const KURL&, bool forURLScheme); 82 void handleDataLoadSoon(const ResourceRequest& r); 83 84 void startDataLoadTimer(); 85 void handleDataLoad(ResourceRequest&); 86 87 void receivedError(const ResourceError&); 88 ResourceError interruptionForPolicyChangeError() const; 89 void stopLoadingForPolicyChange(); 90 bool isPostOrRedirectAfterPost(const ResourceRequest& newRequest, const ResourceResponse& redirectResponse); 91 92 static void callContinueAfterNavigationPolicy(void*, const ResourceRequest&, PassRefPtr<FormState>, bool shouldContinue); 93 void continueAfterNavigationPolicy(const ResourceRequest&, bool shouldContinue); 94 95 static void callContinueAfterContentPolicy(void*, PolicyAction); 96 void continueAfterContentPolicy(PolicyAction); 97 void continueAfterContentPolicy(PolicyAction, const ResourceResponse&); 98 99 #if PLATFORM(QT) 100 void substituteMIMETypeFromPluginDatabase(const ResourceResponse&); 101 #endif 102 103 ResourceRequest m_initialRequest; 104 SubstituteData m_substituteData; 105 106 MainResourceLoaderTimer m_dataLoadTimer; 107 108 bool m_loadingMultipartContent; 109 bool m_waitingForContentPolicy; 110 double m_timeOfLastDataReceived; 111 }; 112 113 } 114 115 #endif 116