0075ec618e1ec99e72917c217eb7f58734f56fbe
[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 "CachedRawResource.h"
33 #include "CachedResourceHandle.h"
34 #include "FrameLoaderTypes.h"
35 #include "ResourceLoader.h"
36 #include "SubstituteData.h"
37 #include <wtf/Forward.h>
38
39 #if HAVE(RUNLOOP_TIMER)
40 #include <wtf/RunLoopTimer.h>
41 #else
42 #include "Timer.h"
43 #endif
44
45 namespace WebCore {
46
47 class ResourceRequest;
48
49 class MainResourceLoader : public RefCounted<MainResourceLoader>, public CachedRawResourceClient {
50     WTF_MAKE_FAST_ALLOCATED;
51 public:
52     static PassRefPtr<MainResourceLoader> create(DocumentLoader*);
53     virtual ~MainResourceLoader();
54
55     void load(const ResourceRequest&);
56     void cancel();
57     void cancel(const ResourceError&);
58     ResourceLoader* loader() const;
59     PassRefPtr<ResourceBuffer> resourceData();
60
61     void setDefersLoading(bool);
62     void setDataBufferingPolicy(DataBufferingPolicy);
63
64 #if HAVE(RUNLOOP_TIMER)
65     typedef RunLoopTimer<MainResourceLoader> MainResourceLoaderTimer;
66 #else
67     typedef Timer<MainResourceLoader> MainResourceLoaderTimer;
68 #endif
69
70     CachedRawResource* cachedMainResource() { return m_resource.get(); }
71     unsigned long identifierForLoadWithoutResourceLoader() const { return m_identifierForLoadWithoutResourceLoader; }
72     void clearIdentifierForLoadWithoutResourceLoader() { m_identifierForLoadWithoutResourceLoader = 0; }
73
74     unsigned long identifier() const;
75
76     void reportMemoryUsage(MemoryObjectInfo*) const;
77
78     void takeIdentifierFromResourceLoader() { m_identifierForLoadWithoutResourceLoader = identifier(); }
79     void handleSubstituteDataLoadSoon(const ResourceRequest&);
80     void clearResource();
81
82 private:
83     explicit MainResourceLoader(DocumentLoader*);
84
85     virtual void redirectReceived(CachedResource*, ResourceRequest&, const ResourceResponse&) OVERRIDE;
86     virtual void responseReceived(CachedResource*, const ResourceResponse&) OVERRIDE;
87     virtual void dataReceived(CachedResource*, const char* data, int dataLength) OVERRIDE;
88     virtual void notifyFinished(CachedResource*) OVERRIDE;
89
90     void handleSubstituteDataLoadNow(MainResourceLoaderTimer*);
91
92     void startDataLoadTimer();
93
94     void receivedError(const ResourceError&);
95     
96 #if PLATFORM(QT)
97     void substituteMIMETypeFromPluginDatabase(const ResourceResponse&);
98 #endif
99
100     FrameLoader* frameLoader() const;
101     DocumentLoader* documentLoader() const { return m_documentLoader.get(); }
102
103     const ResourceRequest& request() const;
104
105     bool defersLoading() const;
106
107     CachedResourceHandle<CachedRawResource> m_resource;
108
109     ResourceRequest m_initialRequest;
110
111     MainResourceLoaderTimer m_dataLoadTimer;
112     RefPtr<DocumentLoader> m_documentLoader;
113
114     unsigned long m_identifierForLoadWithoutResourceLoader;
115 };
116
117 }
118
119 #endif