Home | History | Annotate | Download | only in win 
      1 /*
      2  * Copyright (C) 2007 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 "WebDocumentLoader.h"
     28 
     29 #include "WebKitDLL.h"
     30 
     31 using namespace WebCore;
     32 
     33 WebDocumentLoader::WebDocumentLoader(const ResourceRequest& request, const SubstituteData& substituteData)
     34     : DocumentLoader(request, substituteData)
     35     , m_dataSource(0)
     36     , m_detachedDataSource(0)
     37 {
     38     gClassCount++;
     39     gClassNameCount.add("WebDocumentLoader");
     40 }
     41 
     42 PassRefPtr<WebDocumentLoader> WebDocumentLoader::create(const ResourceRequest& req, const SubstituteData& data)
     43 {
     44     return adoptRef(new WebDocumentLoader(req, data));
     45 }
     46 
     47 WebDocumentLoader::~WebDocumentLoader()
     48 {
     49     gClassCount--;
     50     gClassNameCount.remove("WebDocumentLoader");
     51     if (m_dataSource) {
     52         ASSERT(!m_detachedDataSource);
     53         m_dataSource->Release();
     54     }
     55 }
     56 
     57 void WebDocumentLoader::setDataSource(WebDataSource *dataSource)
     58 {
     59     ASSERT(!m_dataSource);
     60     m_dataSource = dataSource;
     61     if (m_dataSource)
     62         m_dataSource->AddRef();
     63 }
     64 
     65 WebDataSource* WebDocumentLoader::dataSource() const
     66 {
     67     return m_dataSource;
     68 }
     69 
     70 void WebDocumentLoader::detachDataSource()
     71 {
     72     // we only call detachDataSource when the WebDataSource is freed - and we won't get there if m_dataSource is not
     73     // null (because that would mean the loader still has a reference to the data source)
     74     ASSERT(!m_dataSource);
     75     m_detachedDataSource = 0;
     76 }
     77 
     78 void WebDocumentLoader::attachToFrame()
     79 {
     80     DocumentLoader::attachToFrame();
     81     if (m_detachedDataSource) {
     82         ASSERT(!m_dataSource);
     83         setDataSource(m_detachedDataSource);
     84         m_detachedDataSource = 0;
     85     }
     86 }
     87 
     88 void WebDocumentLoader::detachFromFrame()
     89 {
     90     DocumentLoader::detachFromFrame();
     91     m_detachedDataSource = m_dataSource;
     92     if (m_dataSource) {
     93         WebDataSource* dataSourceToBeReleased = m_dataSource;
     94         // It's important to null out m_dataSource before calling release on the data source.  That release can cause the data
     95         // source to be deleted - which ends up calling loader->detachDataSource() which makes the assumption that the loader no
     96         // longer holds a reference to the data source.
     97         m_dataSource = 0;
     98         dataSourceToBeReleased->Release();
     99     }
    100 }
    101