Home | History | Annotate | Download | only in network 
      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