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 #include "FrameLoaderTypes.h" 30 #include "ResourceLoader.h" 31 #include "SubstituteData.h" 32 #include <wtf/Forward.h> 33 34 #if HAVE(RUNLOOP_TIMER) 35 #include "RunLoopTimer.h" 36 #else 37 #include "Timer.h" 38 #endif 39 40 namespace WebCore { 41 42 class FormState; 43 class ResourceRequest; 44 45 class MainResourceLoader : public ResourceLoader { 46 public: 47 static PassRefPtr<MainResourceLoader> create(Frame*); 48 virtual ~MainResourceLoader(); 49 50 virtual bool load(const ResourceRequest&, const SubstituteData&); 51 virtual void addData(const char*, int, bool allAtOnce); 52 53 virtual void setDefersLoading(bool); 54 55 virtual void willSendRequest(ResourceRequest&, const ResourceResponse& redirectResponse); 56 virtual void didReceiveResponse(const ResourceResponse&); 57 virtual void didReceiveData(const char*, int, long long lengthReceived, bool allAtOnce); 58 virtual void didFinishLoading(); 59 virtual void didFail(const ResourceError&); 60 61 #if HAVE(RUNLOOP_TIMER) 62 typedef RunLoopTimer<MainResourceLoader> MainResourceLoaderTimer; 63 #else 64 typedef Timer<MainResourceLoader> MainResourceLoaderTimer; 65 #endif 66 67 void handleDataLoadNow(MainResourceLoaderTimer*); 68 isLoadingMultipartContent()69 bool isLoadingMultipartContent() const { return m_loadingMultipartContent; } 70 71 private: 72 MainResourceLoader(Frame*); 73 74 virtual void didCancel(const ResourceError&); 75 76 bool loadNow(ResourceRequest&); 77 78 void handleEmptyLoad(const KURL&, bool forURLScheme); 79 void handleDataLoadSoon(ResourceRequest& r); 80 81 void startDataLoadTimer(); 82 void handleDataLoad(ResourceRequest&); 83 84 void receivedError(const ResourceError&); 85 ResourceError interruptionForPolicyChangeError() const; 86 void stopLoadingForPolicyChange(); 87 bool isPostOrRedirectAfterPost(const ResourceRequest& newRequest, const ResourceResponse& redirectResponse); 88 89 static void callContinueAfterNavigationPolicy(void*, const ResourceRequest&, PassRefPtr<FormState>, bool shouldContinue); 90 void continueAfterNavigationPolicy(const ResourceRequest&, bool shouldContinue); 91 92 static void callContinueAfterContentPolicy(void*, PolicyAction); 93 void continueAfterContentPolicy(PolicyAction); 94 void continueAfterContentPolicy(PolicyAction, const ResourceResponse&); 95 96 #if PLATFORM(QT) 97 void substituteMIMETypeFromPluginDatabase(const ResourceResponse&); 98 #endif 99 100 ResourceRequest m_initialRequest; 101 SubstituteData m_substituteData; 102 103 MainResourceLoaderTimer m_dataLoadTimer; 104 105 bool m_loadingMultipartContent; 106 bool m_waitingForContentPolicy; 107 }; 108 109 } 110