WebCore/platform/network/ResourceHandle.cpp

   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