1 /* 2 * Copyright (C) 2004, 2006, 2007, 2008 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 * 1. Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * 2. Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 13 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY 14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR 17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY 21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26 #include "config.h" 27 #include "ResourceHandle.h" 28 #include "ResourceHandleInternal.h" 29 30 #include "Logging.h" 31 #include "ResourceHandleClient.h" 32 #include "Timer.h" 33 #include <algorithm> 34 35 namespace WebCore { 36 37 static bool shouldForceContentSniffing; 38 39 ResourceHandle::ResourceHandle(const ResourceRequest& request, ResourceHandleClient* client, bool defersLoading, 40 bool shouldContentSniff, bool mightDownloadFromHandle) 41 : d(new ResourceHandleInternal(this, request, client, defersLoading, shouldContentSniff, mightDownloadFromHandle)) 42 { 43 } 44 45 PassRefPtr<ResourceHandle> ResourceHandle::create(const ResourceRequest& request, ResourceHandleClient* client, 46 Frame* frame, bool defersLoading, bool shouldContentSniff, bool mightDownloadFromHandle) 47 { 48 if (shouldContentSniff) 49 shouldContentSniff = shouldContentSniffURL(request.url()); 50 51 RefPtr<ResourceHandle> newHandle(adoptRef(new ResourceHandle(request, client, defersLoading, shouldContentSniff, mightDownloadFromHandle))); 52 53 if (!request.url().isValid()) { 54 newHandle->scheduleFailure(InvalidURLFailure); 55 return newHandle.release(); 56 } 57 58 if (!portAllowed(request.url())) { 59 newHandle->scheduleFailure(BlockedFailure); 60 return newHandle.release(); 61 } 62 63 if (newHandle->start(frame)) 64 return newHandle.release(); 65 66 return 0; 67 } 68 69 void ResourceHandle::scheduleFailure(FailureType type) 70 { 71 d->m_failureType = type; 72 d->m_failureTimer.startOneShot(0); 73 } 74 75 void ResourceHandle::fireFailure(Timer<ResourceHandle>*) 76 { 77 if (!client()) 78 return; 79 80 switch (d->m_failureType) { 81 case BlockedFailure: 82 client()->wasBlocked(this); 83 return; 84 case InvalidURLFailure: 85 client()->cannotShowURL(this); 86 return; 87 } 88 89 ASSERT_NOT_REACHED(); 90 } 91 92 ResourceHandleClient* ResourceHandle::client() const 93 { 94 return d->m_client; 95 } 96 97 void ResourceHandle::setClient(ResourceHandleClient* client) 98 { 99 d->m_client = client; 100 } 101 102 const ResourceRequest& ResourceHandle::request() const 103 { 104 return d->m_request; 105 } 106 107 void ResourceHandle::clearAuthentication() 108 { 109 #if PLATFORM(MAC) 110 d->m_currentMacChallenge = nil; 111 #endif 112 d->m_currentWebChallenge.nullify(); 113 } 114 115 bool ResourceHandle::shouldContentSniff() const 116 { 117 return d->m_shouldContentSniff; 118 } 119 120 bool ResourceHandle::shouldContentSniffURL(const KURL& url) 121 { 122 #if PLATFORM(MAC) 123 if (shouldForceContentSniffing) 124 return true; 125 #endif 126 // We shouldn't content sniff file URLs as their MIME type should be established via their extension. 127 return !url.protocolIs("file"); 128 } 129 130 void ResourceHandle::forceContentSniffing() 131 { 132 shouldForceContentSniffing = true; 133 } 134 135 } // namespace WebCore 136