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