1/*
2 * Copyright (C) 2006, 2007, 2008, 2009 Apple Inc. All rights reserved.
3 * Copyright (C) 2008 Nokia Corporation and/or its subsidiary(-ies)
4 * Copyright (C) 2008, 2009 Torch Mobile Inc. All rights reserved. (http://www.torchmobile.com/)
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1.  Redistributions of source code must retain the above copyright
11 *     notice, this list of conditions and the following disclaimer.
12 * 2.  Redistributions in binary form must reproduce the above copyright
13 *     notice, this list of conditions and the following disclaimer in the
14 *     documentation and/or other materials provided with the distribution.
15 * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
16 *     its contributors may be used to endorse or promote products derived
17 *     from this software without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
20 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
21 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
22 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
23 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
24 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
25 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
26 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
28 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30
31#include "config.h"
32#include "ResourceLoadNotifier.h"
33
34#include "DocumentLoader.h"
35#include "Frame.h"
36#include "FrameLoader.h"
37#include "FrameLoaderClient.h"
38#include "InspectorInstrumentation.h"
39#include "Page.h"
40#include "ProgressTracker.h"
41#include "ResourceLoader.h"
42
43namespace WebCore {
44
45ResourceLoadNotifier::ResourceLoadNotifier(Frame* frame)
46    : m_frame(frame)
47{
48}
49
50void ResourceLoadNotifier::didReceiveAuthenticationChallenge(ResourceLoader* loader, const AuthenticationChallenge& currentWebChallenge)
51{
52    m_frame->loader()->client()->dispatchDidReceiveAuthenticationChallenge(loader->documentLoader(), loader->identifier(), currentWebChallenge);
53}
54
55void ResourceLoadNotifier::didCancelAuthenticationChallenge(ResourceLoader* loader, const AuthenticationChallenge& currentWebChallenge)
56{
57    m_frame->loader()->client()->dispatchDidCancelAuthenticationChallenge(loader->documentLoader(), loader->identifier(), currentWebChallenge);
58}
59
60void ResourceLoadNotifier::willSendRequest(ResourceLoader* loader, ResourceRequest& clientRequest, const ResourceResponse& redirectResponse)
61{
62    m_frame->loader()->applyUserAgent(clientRequest);
63
64    dispatchWillSendRequest(loader->documentLoader(), loader->identifier(), clientRequest, redirectResponse);
65}
66
67void ResourceLoadNotifier::didReceiveResponse(ResourceLoader* loader, const ResourceResponse& r)
68{
69    loader->documentLoader()->addResponse(r);
70
71    if (Page* page = m_frame->page())
72        page->progress()->incrementProgress(loader->identifier(), r);
73
74    dispatchDidReceiveResponse(loader->documentLoader(), loader->identifier(), r);
75}
76
77void ResourceLoadNotifier::didReceiveData(ResourceLoader* loader, const char* data, int dataLength, int encodedDataLength)
78{
79    if (Page* page = m_frame->page())
80        page->progress()->incrementProgress(loader->identifier(), data, dataLength);
81
82    dispatchDidReceiveContentLength(loader->documentLoader(), loader->identifier(), dataLength, encodedDataLength);
83}
84
85void ResourceLoadNotifier::didFinishLoad(ResourceLoader* loader, double finishTime)
86{
87    if (Page* page = m_frame->page())
88        page->progress()->completeProgress(loader->identifier());
89    dispatchDidFinishLoading(loader->documentLoader(), loader->identifier(), finishTime);
90}
91
92void ResourceLoadNotifier::didFailToLoad(ResourceLoader* loader, const ResourceError& error)
93{
94    if (Page* page = m_frame->page())
95        page->progress()->completeProgress(loader->identifier());
96
97    if (!error.isNull())
98        m_frame->loader()->client()->dispatchDidFailLoading(loader->documentLoader(), loader->identifier(), error);
99
100    InspectorInstrumentation::didFailLoading(m_frame, loader->identifier(), error);
101}
102
103void ResourceLoadNotifier::assignIdentifierToInitialRequest(unsigned long identifier, DocumentLoader* loader, const ResourceRequest& request)
104{
105    m_frame->loader()->client()->assignIdentifierToInitialRequest(identifier, loader, request);
106}
107
108void ResourceLoadNotifier::dispatchWillSendRequest(DocumentLoader* loader, unsigned long identifier, ResourceRequest& request, const ResourceResponse& redirectResponse)
109{
110    StringImpl* oldRequestURL = request.url().string().impl();
111    m_frame->loader()->documentLoader()->didTellClientAboutLoad(request.url());
112
113    m_frame->loader()->client()->dispatchWillSendRequest(loader, identifier, request, redirectResponse);
114
115    // If the URL changed, then we want to put that new URL in the "did tell client" set too.
116    if (!request.isNull() && oldRequestURL != request.url().string().impl())
117        m_frame->loader()->documentLoader()->didTellClientAboutLoad(request.url());
118
119    InspectorInstrumentation::willSendRequest(m_frame, identifier, loader, request, redirectResponse);
120
121    // Report WebTiming for all frames.
122    if (loader && !request.isNull() && request.url() == loader->requestURL())
123        request.setReportLoadTiming(true);
124}
125
126void ResourceLoadNotifier::dispatchDidReceiveResponse(DocumentLoader* loader, unsigned long identifier, const ResourceResponse& r)
127{
128    InspectorInstrumentationCookie cookie = InspectorInstrumentation::willReceiveResourceResponse(m_frame, identifier, r);
129    m_frame->loader()->client()->dispatchDidReceiveResponse(loader, identifier, r);
130    InspectorInstrumentation::didReceiveResourceResponse(cookie, identifier, loader, r);
131}
132
133void ResourceLoadNotifier::dispatchDidReceiveContentLength(DocumentLoader* loader, unsigned long identifier, int dataLength, int encodedDataLength)
134{
135    m_frame->loader()->client()->dispatchDidReceiveContentLength(loader, identifier, dataLength);
136
137    InspectorInstrumentation::didReceiveContentLength(m_frame, identifier, dataLength, encodedDataLength);
138}
139
140void ResourceLoadNotifier::dispatchDidFinishLoading(DocumentLoader* loader, unsigned long identifier, double finishTime)
141{
142    m_frame->loader()->client()->dispatchDidFinishLoading(loader, identifier);
143
144    InspectorInstrumentation::didFinishLoading(m_frame, identifier, finishTime);
145}
146
147void ResourceLoadNotifier::dispatchTransferLoadingResourceFromPage(unsigned long identifier, DocumentLoader* loader, const ResourceRequest& request, Page* oldPage)
148{
149    ASSERT(oldPage != m_frame->page());
150    m_frame->loader()->client()->transferLoadingResourceFromPage(identifier, loader, request, oldPage);
151
152    oldPage->progress()->completeProgress(identifier);
153}
154
155void ResourceLoadNotifier::sendRemainingDelegateMessages(DocumentLoader* loader, unsigned long identifier, const ResourceResponse& response, int dataLength, int encodedDataLength, const ResourceError& error)
156{
157    if (!response.isNull())
158        dispatchDidReceiveResponse(loader, identifier, response);
159
160    if (dataLength > 0)
161        dispatchDidReceiveContentLength(loader, identifier, dataLength, encodedDataLength);
162
163    if (error.isNull())
164        dispatchDidFinishLoading(loader, identifier, 0);
165    else
166        m_frame->loader()->client()->dispatchDidFailLoading(loader, identifier, error);
167}
168
169} // namespace WebCore
170