/* * Copyright (C) 2006, 2007 Apple Inc. All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of * its contributors may be used to endorse or promote products derived * from this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include "config.h" #include "SubresourceLoader.h" #include "Document.h" #include "DocumentLoader.h" #include "Frame.h" #include "FrameLoader.h" #include "Logging.h" #include "ResourceHandle.h" #include "ResourceRequest.h" #include "SubresourceLoaderClient.h" #include "SharedBuffer.h" namespace WebCore { #ifndef NDEBUG WTFLogChannel LogWebCoreSubresourceLoaderLeaks = { 0x00000000, "", WTFLogChannelOn }; struct SubresourceLoaderCounter { static unsigned count; ~SubresourceLoaderCounter() { if (count) LOG(WebCoreSubresourceLoaderLeaks, "LEAK: %u SubresourceLoader\n", count); } }; unsigned SubresourceLoaderCounter::count = 0; static SubresourceLoaderCounter subresourceLoaderCounter; #endif SubresourceLoader::SubresourceLoader(Frame* frame, SubresourceLoaderClient* client, bool sendResourceLoadCallbacks) : ResourceLoader(frame, sendResourceLoadCallbacks) , m_client(client) , m_loadingMultipartContent(false) { #ifndef NDEBUG ++SubresourceLoaderCounter::count; #endif m_documentLoader->addSubresourceLoader(this); } SubresourceLoader::~SubresourceLoader() { #ifndef NDEBUG --SubresourceLoaderCounter::count; #endif } bool SubresourceLoader::load(const ResourceRequest& r) { m_frame->loader()->didTellBridgeAboutLoad(r.url().url()); return ResourceLoader::load(r); } PassRefPtr SubresourceLoader::create(Frame* frame, SubresourceLoaderClient* client, const ResourceRequest& request, bool skipCanLoadCheck, bool sendResourceLoadCallbacks) { if (!frame) return 0; FrameLoader* fl = frame->loader(); if (!skipCanLoadCheck && fl->state() == FrameStateProvisional) return 0; ResourceRequest newRequest = request; if (!skipCanLoadCheck && FrameLoader::restrictAccessToLocal() && !FrameLoader::canLoad(request.url(), frame->document())) { FrameLoader::reportLocalLoadFailed(frame->page(), request.url().url()); return 0; } if (FrameLoader::shouldHideReferrer(request.url(), fl->outgoingReferrer())) newRequest.clearHTTPReferrer(); else if (!request.httpReferrer()) newRequest.setHTTPReferrer(fl->outgoingReferrer()); // Use the original request's cache policy for two reasons: // 1. For POST requests, we mutate the cache policy for the main resource, // but we do not want this to apply to subresources // 2. Delegates that modify the cache policy using willSendRequest: should // not affect any other resources. Such changes need to be done // per request. if (newRequest.isConditional()) newRequest.setCachePolicy(ReloadIgnoringCacheData); else newRequest.setCachePolicy(fl->originalRequest().cachePolicy()); fl->addExtraFieldsToRequest(newRequest, false, false); RefPtr subloader(new SubresourceLoader(frame, client, sendResourceLoadCallbacks)); if (!subloader->load(newRequest)) return 0; return subloader.release(); } void SubresourceLoader::willSendRequest(ResourceRequest& newRequest, const ResourceResponse& redirectResponse) { ResourceLoader::willSendRequest(newRequest, redirectResponse); if (!newRequest.isNull() && m_originalURL != newRequest.url() && m_client) m_client->willSendRequest(this, newRequest, redirectResponse); } void SubresourceLoader::didReceiveResponse(const ResourceResponse& r) { ASSERT(!r.isNull()); if (r.isMultipart()) m_loadingMultipartContent = true; // Reference the object in this method since the additional processing can do // anything including removing the last reference to this object; one example of this is 3266216. RefPtr protect(this); if (m_client) m_client->didReceiveResponse(this, r); // The loader can cancel a load if it receives a multipart response for a non-image if (reachedTerminalState()) return; ResourceLoader::didReceiveResponse(r); RefPtr buffer = resourceData(); if (m_loadingMultipartContent && buffer && buffer->size()) { // Since a subresource loader does not load multipart sections progressively, // deliver the previously received data to the loader all at once now. // Then clear the data to make way for the next multipart section. if (m_client) m_client->didReceiveData(this, buffer->data(), buffer->size()); clearResourceData(); // After the first multipart section is complete, signal to delegates that this load is "finished" m_documentLoader->subresourceLoaderFinishedLoadingOnePart(this); didFinishLoadingOnePart(); } } void SubresourceLoader::didReceiveData(const char* data, int length, long long lengthReceived, bool allAtOnce) { // Reference the object in this method since the additional processing can do // anything including removing the last reference to this object; one example of this is 3266216. RefPtr protect(this); ResourceLoader::didReceiveData(data, length, lengthReceived, allAtOnce); // A subresource loader does not load multipart sections progressively. // So don't deliver any data to the loader yet. if (!m_loadingMultipartContent && m_client) m_client->didReceiveData(this, data, length); } void SubresourceLoader::didFinishLoading() { if (cancelled()) return; ASSERT(!reachedTerminalState()); // Calling removeSubresourceLoader will likely result in a call to deref, so we must protect ourselves. RefPtr protect(this); if (m_client) m_client->didFinishLoading(this); m_handle = 0; if (cancelled()) return; m_documentLoader->removeSubresourceLoader(this); ResourceLoader::didFinishLoading(); } void SubresourceLoader::didFail(const ResourceError& error) { if (cancelled()) return; ASSERT(!reachedTerminalState()); // Calling removeSubresourceLoader will likely result in a call to deref, so we must protect ourselves. RefPtr protect(this); if (m_client) m_client->didFail(this, error); m_handle = 0; if (cancelled()) return; m_documentLoader->removeSubresourceLoader(this); ResourceLoader::didFail(error); } void SubresourceLoader::didCancel(const ResourceError& error) { ASSERT(!reachedTerminalState()); // Calling removeSubresourceLoader will likely result in a call to deref, so we must protect ourselves. RefPtr protect(this); if (m_client) m_client->didFail(this, error); if (cancelled()) return; m_documentLoader->removeSubresourceLoader(this); ResourceLoader::didCancel(error); } void SubresourceLoader::receivedCancellation(const AuthenticationChallenge& challenge) { ASSERT(!reachedTerminalState()); RefPtr protect(this); if (m_client) m_client->receivedCancellation(this, challenge); ResourceLoader::receivedCancellation(challenge); } }