WebCore/loader/DocumentThreadableLoader.cpp

   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 #include "config.h"
  32 #include "DocumentThreadableLoader.h"
  33
  34 #include "AuthenticationChallenge.h"
  35 #include "Document.h"
  36 #include "DocumentThreadableLoader.h"
  37 #include "Frame.h"
  38 #include "FrameLoader.h"
  39 #include "ResourceRequest.h"
  40 #include "SecurityOrigin.h"
  41 #include "SubresourceLoader.h"
  42 #include "ThreadableLoaderClient.h"
  43
  44 namespace WebCore {
  45
  46 void DocumentThreadableLoader::loadResourceSynchronously(Document* document, const ResourceRequest& request, ThreadableLoaderClient& client)
  47 {
  48     bool sameOriginRequest = document->securityOrigin()->canRequest(request.url());
  49
  50     Vector<char> data;
  51     ResourceError error;
  52     ResourceResponse response;
  53     unsigned long identifier = std::numeric_limits<unsigned long>::max();
  54     if (document->frame())
  55         identifier = document->frame()->loader()->loadResourceSynchronously(request, error, response, data);
  56
  57     // No exception for file:/// resources, see <rdar://problem/4962298>.
  58     // Also, if we have an HTTP response, then it wasn't a network error in fact.
  59     if (!error.isNull() && !request.url().isLocalFile() && response.httpStatusCode() <= 0) {
  60         client.didFail(error);
  61         return;
  62     }
  63
  64     // FIXME: This check along with the one in willSendRequest is specific to xhr and
  65     // should be made more generic.
  66     if (sameOriginRequest && !document->securityOrigin()->canRequest(response.url())) {
  67         client.didFailRedirectCheck();
  68         return;
  69     }
  70
  71     client.didReceiveResponse(response);
  72
  73     const char* bytes = static_cast<const char*>(data.data());
  74     int len = static_cast<int>(data.size());
  75     client.didReceiveData(bytes, len);
  76
  77     client.didFinishLoading(identifier);
  78 }
  79
  80 PassRefPtr<DocumentThreadableLoader> DocumentThreadableLoader::create(Document* document, ThreadableLoaderClient* client, const ResourceRequest& request, LoadCallbacks callbacksSetting, ContentSniff contentSniff)
  81 {
  82     ASSERT(document);
  83     RefPtr<DocumentThreadableLoader> loader = adoptRef(new DocumentThreadableLoader(document, client, request, callbacksSetting, contentSniff));
  84     if (!loader->m_loader)
  85         loader = 0;
  86     return loader.release();
  87 }
  88
  89 DocumentThreadableLoader::DocumentThreadableLoader(Document* document, ThreadableLoaderClient* client, const ResourceRequest& request, LoadCallbacks callbacksSetting, ContentSniff contentSniff)
  90     : m_client(client)
  91     , m_document(document)
  92 {
  93     ASSERT(document);
  94     ASSERT(client);
  95     m_loader = SubresourceLoader::create(document->frame(), this, request, false, callbacksSetting == SendLoadCallbacks, contentSniff == SniffContent);
  96 }
  97
  98 DocumentThreadableLoader::~DocumentThreadableLoader()
  99 {
 100     if (m_loader)
 101         m_loader->clearClient();
 102 }
 103
 104 void DocumentThreadableLoader::cancel()
 105 {
 106     if (!m_loader)
 107         return;
 108
 109     m_loader->cancel();
 110     m_loader->clearClient();
 111     m_loader = 0;
 112     m_client = 0;
 113 }
 114
 115 void DocumentThreadableLoader::willSendRequest(SubresourceLoader*, ResourceRequest& request, const ResourceResponse&)
 116 {
 117     ASSERT(m_client);
 118
 119     // FIXME: This needs to be fixed to follow the redirect correctly even for cross-domain requests.
 120     if (!m_document->securityOrigin()->canRequest(request.url())) {
 121         RefPtr<DocumentThreadableLoader> protect(this);
 122         m_client->didFailRedirectCheck();
 123         cancel();
 124     }
 125 }
 126
 127 void DocumentThreadableLoader::didSendData(SubresourceLoader*, unsigned long long bytesSent, unsigned long long totalBytesToBeSent)
 128 {
 129     ASSERT(m_client);
 130     m_client->didSendData(bytesSent, totalBytesToBeSent);
 131 }
 132
 133 void DocumentThreadableLoader::didReceiveResponse(SubresourceLoader*, const ResourceResponse& response)
 134 {
 135     ASSERT(m_client);
 136     m_client->didReceiveResponse(response);
 137 }
 138
 139 void DocumentThreadableLoader::didReceiveData(SubresourceLoader*, const char* data, int lengthReceived)
 140 {
 141     ASSERT(m_client);
 142     m_client->didReceiveData(data, lengthReceived);
 143 }
 144
 145 void DocumentThreadableLoader::didFinishLoading(SubresourceLoader* loader)
 146 {
 147     ASSERT(loader);
 148     ASSERT(m_client);
 149     m_client->didFinishLoading(loader->identifier());
 150 }
 151
 152 void DocumentThreadableLoader::didFail(SubresourceLoader*, const ResourceError& error)
 153 {
 154     ASSERT(m_client);
 155     m_client->didFail(error);
 156 }
 157
 158 void DocumentThreadableLoader::receivedCancellation(SubresourceLoader*, const AuthenticationChallenge& challenge)
 159 {
 160     ASSERT(m_client);
 161     m_client->didReceiveAuthenticationCancellation(challenge.failureResponse());
 162 }
 163
 164 } // namespace WebCore