1/*
2 * Copyright (C) 2009 Google 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 are
6 * met:
7 *
8 *     * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 *     * Redistributions in binary form must reproduce the above
11 * copyright notice, this list of conditions and the following disclaimer
12 * in the documentation and/or other materials provided with the
13 * distribution.
14 *     * Neither the name of Google Inc. nor the names of its
15 * contributors may be used to endorse or promote products derived from
16 * this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30
31#ifndef WebDataSourceImpl_h
32#define WebDataSourceImpl_h
33
34#include "DocumentLoader.h"
35#include "KURL.h"
36#include "WebDataSource.h"
37#include "WebPluginLoadObserver.h"
38#include "WrappedResourceRequest.h"
39#include "WrappedResourceResponse.h"
40#include <wtf/OwnPtr.h>
41#include <wtf/PassOwnPtr.h>
42#include <wtf/Vector.h>
43
44namespace WebKit {
45
46class WebPluginLoadObserver;
47
48class WebDataSourceImpl : public WebCore::DocumentLoader, public WebDataSource {
49public:
50    static PassRefPtr<WebDataSourceImpl> create(const WebCore::ResourceRequest&,
51                                                const WebCore::SubstituteData&);
52
53    static WebDataSourceImpl* fromDocumentLoader(WebCore::DocumentLoader* loader)
54    {
55        return static_cast<WebDataSourceImpl*>(loader);
56    }
57
58    // WebDataSource methods:
59    virtual const WebURLRequest& originalRequest() const;
60    virtual const WebURLRequest& request() const;
61    virtual const WebURLResponse& response() const;
62    virtual bool hasUnreachableURL() const;
63    virtual WebURL unreachableURL() const;
64    virtual void redirectChain(WebVector<WebURL>&) const;
65    virtual WebString pageTitle() const;
66    virtual WebNavigationType navigationType() const;
67    virtual double triggeringEventTime() const;
68    virtual ExtraData* extraData() const;
69    virtual void setExtraData(ExtraData*);
70    virtual WebApplicationCacheHost* applicationCacheHost();
71    virtual void setDeferMainResourceDataLoad(bool);
72
73    static WebNavigationType toWebNavigationType(WebCore::NavigationType type);
74
75    bool hasRedirectChain() const { return !m_redirectChain.isEmpty(); }
76    const WebCore::KURL& endOfRedirectChain() const;
77    void clearRedirectChain();
78    void appendRedirect(const WebCore::KURL& url);
79
80    PassOwnPtr<WebPluginLoadObserver> releasePluginLoadObserver() { return m_pluginLoadObserver.release(); }
81    static void setNextPluginLoadObserver(PassOwnPtr<WebPluginLoadObserver>);
82
83private:
84    WebDataSourceImpl(const WebCore::ResourceRequest&, const WebCore::SubstituteData&);
85    ~WebDataSourceImpl();
86
87    // Mutable because the const getters will magically sync these to the
88    // latest version from WebKit.
89    mutable WrappedResourceRequest m_originalRequestWrapper;
90    mutable WrappedResourceRequest m_requestWrapper;
91    mutable WrappedResourceResponse m_responseWrapper;
92
93    // Lists all intermediate URLs that have redirected for the current provisional load.
94    // See WebFrameLoaderClient::dispatchDidReceiveServerRedirectForProvisionalLoad for a
95    // description of who modifies this when to keep it up to date.
96    Vector<WebCore::KURL> m_redirectChain;
97
98    OwnPtr<ExtraData> m_extraData;
99    OwnPtr<WebPluginLoadObserver> m_pluginLoadObserver;
100
101    static WebPluginLoadObserver* m_nextPluginLoadObserver;
102};
103
104} // namespace WebKit
105
106#endif  // WebDataSourceImpl_h
107