2011-06-12 Adam Barth <abarth@webkit.org>
[WebKit-https.git] / Source / WebCore / loader / MainResourceLoader.h
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
72         bool isLoadingMultipartContent() const { return m_loadingMultipartContent; }
73
74     private:
75         MainResourceLoader(Frame*);
76
77         virtual void willCancel(const ResourceError&);
78         virtual void didCancel(const ResourceError&);
79
80         bool loadNow(ResourceRequest&);
81
82         void handleEmptyLoad(const KURL&, bool forURLScheme);
83         void handleDataLoadSoon(const ResourceRequest& r);
84
85         void startDataLoadTimer();
86         void handleDataLoad(ResourceRequest&);
87
88         void receivedError(const ResourceError&);
89         ResourceError interruptForPolicyChangeError() const;
90         void stopLoadingForPolicyChange();
91         bool isPostOrRedirectAfterPost(const ResourceRequest& newRequest, const ResourceResponse& redirectResponse);
92
93         static void callContinueAfterNavigationPolicy(void*, const ResourceRequest&, PassRefPtr<FormState>, bool shouldContinue);
94         void continueAfterNavigationPolicy(const ResourceRequest&, bool shouldContinue);
95
96         static void callContinueAfterContentPolicy(void*, PolicyAction);
97         void continueAfterContentPolicy(PolicyAction);
98         void continueAfterContentPolicy(PolicyAction, const ResourceResponse&);
99         
100 #if PLATFORM(QT)
101         void substituteMIMETypeFromPluginDatabase(const ResourceResponse&);
102 #endif
103
104         ResourceRequest m_initialRequest;
105         SubstituteData m_substituteData;
106
107         MainResourceLoaderTimer m_dataLoadTimer;
108
109         bool m_loadingMultipartContent;
110         bool m_waitingForContentPolicy;
111         double m_timeOfLastDataReceived;
112     };
113
114 }
115
116 #endif