Source/WebCore/history/PageCache.cpp - WebKit - Git at Google

 /*
  * Copyright (C) 2007, 2014, 2015 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.
  *
  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``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 INC. OR
  * 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 "PageCache.h"

 #include "ApplicationCacheHost.h"
 #include "BackForwardController.h"
 #include "CachedPage.h"
 #include "DOMWindow.h"
 #include "DeviceMotionController.h"
 #include "DeviceOrientationController.h"
 #include "DiagnosticLoggingClient.h"
 #include "DiagnosticLoggingKeys.h"
 #include "Document.h"
 #include "DocumentLoader.h"
 #include "FocusController.h"
 #include "Frame.h"
 #include "FrameLoader.h"
 #include "FrameLoaderClient.h"
 #include "FrameView.h"
 #include "HistoryController.h"
 #include "IgnoreOpensDuringUnloadCountIncrementer.h"
 #include "Logging.h"
 #include "Page.h"
 #include "ScriptDisallowedScope.h"
 #include "Settings.h"
 #include "SubframeLoader.h"
 #include <pal/Logging.h>
 #include <wtf/MemoryPressureHandler.h>
 #include <wtf/NeverDestroyed.h>
 #include <wtf/SetForScope.h>
 #include <wtf/text/CString.h>
 #include <wtf/text/StringConcatenate.h>

 namespace WebCore {

 #define PCLOG(...) LOG(BackForwardCache, "%*s%s", indentLevel*4, "", makeString(__VA_ARGS__).utf8().data())

 static inline void logPageCacheFailureDiagnosticMessage(DiagnosticLoggingClient& client, const String& reason)
 {
     client.logDiagnosticMessage(DiagnosticLoggingKeys::pageCacheFailureKey(), reason, ShouldSample::No);
 }

 static inline void logPageCacheFailureDiagnosticMessage(Page* page, const String& reason)
 {
     if (!page)
         return;

     logPageCacheFailureDiagnosticMessage(page->diagnosticLoggingClient(), reason);
 }

 static bool canCacheFrame(Frame& frame, DiagnosticLoggingClient& diagnosticLoggingClient, unsigned indentLevel)
 {
     PCLOG("+---");
     FrameLoader& frameLoader = frame.loader();

     // Prevent page caching if a subframe is still in provisional load stage.
     // We only do this check for subframes because the main frame is reused when navigating to a new page.
     if (!frame.isMainFrame() && frameLoader.state() == FrameStateProvisional) {
         PCLOG("   -Frame is in provisional load stage");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::provisionalLoadKey());
         return false;
     }

     if (frame.isMainFrame() && frameLoader.stateMachine().isDisplayingInitialEmptyDocument()) {
         PCLOG("   -MainFrame is displaying initial empty document");
         return false;
     }

     if (!frame.document()) {
         PCLOG("   -Frame has no document");
         return false;
     }

     if (!frame.document()->frame()) {
         PCLOG("   -Document is detached from frame");
         ASSERT_NOT_REACHED();
         return false;
     }

     DocumentLoader* documentLoader = frameLoader.documentLoader();
     if (!documentLoader) {
         PCLOG("   -There is no DocumentLoader object");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::noDocumentLoaderKey());
         return false;
     }

     URL currentURL = documentLoader->url();
     URL newURL = frameLoader.provisionalDocumentLoader() ? frameLoader.provisionalDocumentLoader()->url() : URL();
     if (!newURL.isEmpty())
         PCLOG(" Determining if frame can be cached navigating from (", currentURL.string(), ") to (", newURL.string(), "):");
     else
         PCLOG(" Determining if subframe with URL (", currentURL.string(), ") can be cached:");

     bool isCacheable = true;
     // Do not cache error pages (these can be recognized as pages with substitute data or unreachable URLs).
     if (documentLoader->substituteData().isValid() && !documentLoader->substituteData().failingURL().isEmpty()) {
         PCLOG("   -Frame is an error page");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::isErrorPageKey());
         isCacheable = false;
     }
     if (frameLoader.subframeLoader().containsPlugins() && !frame.page()->settings().pageCacheSupportsPlugins()) {
         PCLOG("   -Frame contains plugins");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::hasPluginsKey());
         isCacheable = false;
     }
     if (frame.isMainFrame() && !frameLoader.history().currentItem()) {
         PCLOG("   -Main frame has no current history item");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::noCurrentHistoryItemKey());
         isCacheable = false;
     }
     if (frameLoader.quickRedirectComing()) {
         PCLOG("   -Quick redirect is coming");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::quirkRedirectComingKey());
         isCacheable = false;
     }
     if (documentLoader->isLoading()) {
         PCLOG("   -DocumentLoader is still loading");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::isLoadingKey());
         isCacheable = false;
     }
     if (documentLoader->isStopping()) {
         PCLOG("   -DocumentLoader is in the middle of stopping");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::documentLoaderStoppingKey());
         isCacheable = false;
     }

     Vector<ActiveDOMObject*> unsuspendableObjects;
     if (frame.document() && !frame.document()->canSuspendActiveDOMObjectsForDocumentSuspension(&unsuspendableObjects)) {
         PCLOG("   -The document cannot suspend its active DOM Objects");
         for (auto* activeDOMObject : unsuspendableObjects) {
             PCLOG("    - Unsuspendable: ", activeDOMObject->activeDOMObjectName());
             diagnosticLoggingClient.logDiagnosticMessage(DiagnosticLoggingKeys::unsuspendableDOMObjectKey(), activeDOMObject->activeDOMObjectName(), ShouldSample::No);
             UNUSED_PARAM(activeDOMObject);
         }
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::cannotSuspendActiveDOMObjectsKey());
         isCacheable = false;
     }

     // FIXME: We should investigating caching frames that have an associated
     // application cache. <rdar://problem/5917899> tracks that work.
     if (!documentLoader->applicationCacheHost().canCacheInPageCache()) {
         PCLOG("   -The DocumentLoader uses an application cache");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::applicationCacheKey());
         isCacheable = false;
     }
     if (!frameLoader.client().canCachePage()) {
         PCLOG("   -The client says this frame cannot be cached");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::deniedByClientKey());
         isCacheable = false;
     }

     for (Frame* child = frame.tree().firstChild(); child; child = child->tree().nextSibling()) {
         if (!canCacheFrame(*child, diagnosticLoggingClient, indentLevel + 1))
             isCacheable = false;
     }

     PCLOG(isCacheable ? " Frame CAN be cached" : " Frame CANNOT be cached");
     PCLOG("+---");

     return isCacheable;
 }

 static bool canCachePage(Page& page)
 {
     RELEASE_ASSERT(!page.isRestoringCachedPage());

     unsigned indentLevel = 0;
     PCLOG("--------\n Determining if page can be cached:");

     DiagnosticLoggingClient& diagnosticLoggingClient = page.diagnosticLoggingClient();
     bool isCacheable = canCacheFrame(page.mainFrame(), diagnosticLoggingClient, indentLevel + 1);

     if (!page.settings().usesPageCache() || page.isResourceCachingDisabled()) {
         PCLOG("   -Page settings says b/f cache disabled");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::isDisabledKey());
         isCacheable = false;
     }
 #if ENABLE(DEVICE_ORIENTATION) && !PLATFORM(IOS_FAMILY)
     if (DeviceMotionController::isActiveAt(&page)) {
         PCLOG("   -Page is using DeviceMotion");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::deviceMotionKey());
         isCacheable = false;
     }
     if (DeviceOrientationController::isActiveAt(&page)) {
         PCLOG("   -Page is using DeviceOrientation");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::deviceOrientationKey());
         isCacheable = false;
     }
 #endif

     FrameLoadType loadType = page.mainFrame().loader().loadType();
     switch (loadType) {
     case FrameLoadType::Reload:
         // No point writing to the cache on a reload, since we will just write over it again when we leave that page.
         PCLOG("   -Load type is: Reload");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::reloadKey());
         isCacheable = false;
         break;
     case FrameLoadType::Same: // user loads same URL again (but not reload button)
         // No point writing to the cache on a same load, since we will just write over it again when we leave that page.
         PCLOG("   -Load type is: Same");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::sameLoadKey());
         isCacheable = false;
         break;
     case FrameLoadType::RedirectWithLockedBackForwardList:
         // Don't write to the cache if in the middle of a redirect, since we will want to store the final page we end up on.
         PCLOG("   -Load type is: RedirectWithLockedBackForwardList");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::redirectKey());
         isCacheable = false;
         break;
     case FrameLoadType::Replace:
         // No point writing to the cache on a replace, since we will just write over it again when we leave that page.
         PCLOG("   -Load type is: Replace");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::replaceKey());
         isCacheable = false;
         break;
     case FrameLoadType::ReloadFromOrigin: {
         // No point writing to the cache on a reload, since we will just write over it again when we leave that page.
         PCLOG("   -Load type is: ReloadFromOrigin");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::reloadFromOriginKey());
         isCacheable = false;
         break;
     }
     case FrameLoadType::ReloadExpiredOnly: {
         // No point writing to the cache on a reload, since we will just write over it again when we leave that page.
         PCLOG("   -Load type is: ReloadRevalidatingExpired");
         logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::reloadRevalidatingExpiredKey());
         isCacheable = false;
         break;
     }
     case FrameLoadType::Standard:
     case FrameLoadType::Back:
     case FrameLoadType::Forward:
     case FrameLoadType::IndexedBackForward: // a multi-item hop in the backforward list
         // Cacheable.
         break;
     }

     if (isCacheable)
         PCLOG(" Page CAN be cached\n--------");
     else
         PCLOG(" Page CANNOT be cached\n--------");

     diagnosticLoggingClient.logDiagnosticMessageWithResult(DiagnosticLoggingKeys::pageCacheKey(), DiagnosticLoggingKeys::canCacheKey(), isCacheable ? DiagnosticLoggingResultPass : DiagnosticLoggingResultFail, ShouldSample::No);
     return isCacheable;
 }

 PageCache& PageCache::singleton()
 {
     static NeverDestroyed<PageCache> globalPageCache;
     return globalPageCache;
 }

 PageCache::PageCache()
 {
     static std::once_flag onceFlag;
     std::call_once(onceFlag, [] {
         PAL::registerNotifyCallback("com.apple.WebKit.showPageCache", [] {
             PageCache::singleton().dump();
         });
     });
 }

 void PageCache::dump() const
 {
     WTFLogAlways("\nPage Cache:");
     for (auto& item : m_items) {
         CachedPage& cachedPage = *item->m_cachedPage;
         WTFLogAlways("  Page %p, document %p %s", &cachedPage.page(), cachedPage.document(), cachedPage.document() ? cachedPage.document()->url().string().utf8().data() : "");
     }
 }

 bool PageCache::canCache(Page& page) const
 {
     if (!m_maxSize) {
         logPageCacheFailureDiagnosticMessage(&page, DiagnosticLoggingKeys::isDisabledKey());
         return false;
     }

     if (MemoryPressureHandler::singleton().isUnderMemoryPressure()) {
         logPageCacheFailureDiagnosticMessage(&page, DiagnosticLoggingKeys::underMemoryPressureKey());
         return false;
     }

     return canCachePage(page);
 }

 void PageCache::pruneToSizeNow(unsigned size, PruningReason pruningReason)
 {
     SetForScope<unsigned> change(m_maxSize, size);
     prune(pruningReason);
 }

 void PageCache::setMaxSize(unsigned maxSize)
 {
     m_maxSize = maxSize;
     prune(PruningReason::None);
 }

 unsigned PageCache::frameCount() const
 {
     unsigned frameCount = m_items.size();
     for (auto& item : m_items) {
         ASSERT(item->m_cachedPage);
         frameCount += item->m_cachedPage->cachedMainFrame()->descendantFrameCount();
     }

     return frameCount;
 }

 void PageCache::markPagesForDeviceOrPageScaleChanged(Page& page)
 {
     for (auto& item : m_items) {
         CachedPage& cachedPage = *item->m_cachedPage;
         if (&page.mainFrame() == &cachedPage.cachedMainFrame()->view()->frame())
             cachedPage.markForDeviceOrPageScaleChanged();
     }
 }

 void PageCache::markPagesForContentsSizeChanged(Page& page)
 {
     for (auto& item : m_items) {
         CachedPage& cachedPage = *item->m_cachedPage;
         if (&page.mainFrame() == &cachedPage.cachedMainFrame()->view()->frame())
             cachedPage.markForContentsSizeChanged();
     }
 }

 #if ENABLE(VIDEO_TRACK)
 void PageCache::markPagesForCaptionPreferencesChanged()
 {
     for (auto& item : m_items) {
         ASSERT(item->m_cachedPage);
         item->m_cachedPage->markForCaptionPreferencesChanged();
     }
 }
 #endif

 static String pruningReasonToDiagnosticLoggingKey(PruningReason pruningReason)
 {
     switch (pruningReason) {
     case PruningReason::MemoryPressure:
         return DiagnosticLoggingKeys::prunedDueToMemoryPressureKey();
     case PruningReason::ProcessSuspended:
         return DiagnosticLoggingKeys::prunedDueToProcessSuspended();
     case PruningReason::ReachedMaxSize:
         return DiagnosticLoggingKeys::prunedDueToMaxSizeReached();
     case PruningReason::None:
         break;
     }
     ASSERT_NOT_REACHED();
     return emptyString();
 }

 static void setPageCacheState(Page& page, Document::PageCacheState pageCacheState)
 {
     for (Frame* frame = &page.mainFrame(); frame; frame = frame->tree().traverseNext()) {
         if (auto* document = frame->document())
             document->setPageCacheState(pageCacheState);
     }
 }

 // When entering page cache, tear down the render tree before setting the in-cache flag.
 // This maintains the invariant that render trees are never present in the page cache.
 // Note that destruction happens bottom-up so that the main frame's tree dies last.
 static void destroyRenderTree(Frame& mainFrame)
 {
     for (Frame* frame = mainFrame.tree().traversePrevious(CanWrap::Yes); frame; frame = frame->tree().traversePrevious(CanWrap::No)) {
         if (!frame->document())
             continue;
         auto& document = *frame->document();
         if (document.hasLivingRenderTree())
             document.destroyRenderTree();
     }
 }

 static void firePageHideEventRecursively(Frame& frame)
 {
     auto* document = frame.document();
     if (!document)
         return;

     // stopLoading() will fire the pagehide event in each subframe and the HTML specification states
     // that the parent document's ignore-opens-during-unload counter should be incremented while the
     // pagehide event is being fired in its subframes:
     // https://html.spec.whatwg.org/multipage/browsers.html#unload-a-document
     IgnoreOpensDuringUnloadCountIncrementer ignoreOpensDuringUnloadCountIncrementer(document);

     frame.loader().stopLoading(UnloadEventPolicyUnloadAndPageHide);

     for (RefPtr<Frame> child = frame.tree().firstChild(); child; child = child->tree().nextSibling())
         firePageHideEventRecursively(*child);
 }

 bool PageCache::addIfCacheable(HistoryItem& item, Page* page)
 {
     if (item.isInPageCache())
         return false;

     if (!page)
         return false;

     page->mainFrame().loader().stopForPageCache();

     if (!canCache(*page))
         return false;

     ASSERT_WITH_MESSAGE(!page->isUtilityPage(), "Utility pages such as SVGImage pages should never go into PageCache");

     setPageCacheState(*page, Document::AboutToEnterPageCache);

     // Focus the main frame, defocusing a focused subframe (if we have one). We do this here,
     // before the page enters the page cache, while we still can dispatch DOM blur/focus events.
     if (page->focusController().focusedFrame())
         page->focusController().setFocusedFrame(&page->mainFrame());

     // Fire the pagehide event in all frames.
     firePageHideEventRecursively(page->mainFrame());

     destroyRenderTree(page->mainFrame());

     // Stop all loads again before checking if we can still cache the page after firing the pagehide
     // event, since the page may have started ping loads in its pagehide event handler.
     page->mainFrame().loader().stopForPageCache();

     // Check that the page is still page-cacheable after firing the pagehide event. The JS event handlers
     // could have altered the page in a way that could prevent caching.
     if (!canCache(*page)) {
         setPageCacheState(*page, Document::NotInPageCache);
         return false;
     }

     setPageCacheState(*page, Document::InPageCache);

     {
         // Make sure we don't fire any JS events in this scope.
         ScriptDisallowedScope::InMainThread scriptDisallowedScope;

         item.setCachedPage(makeUnique<CachedPage>(*page));
         item.m_pruningReason = PruningReason::None;
         m_items.add(&item);
     }
     prune(PruningReason::ReachedMaxSize);

     RELEASE_LOG(BackForwardCache, "PageCache::addIfCacheable item: %s, size: %u / %u", item.identifier().string().utf8().data(), pageCount(), maxSize());

     return true;
 }

 std::unique_ptr<CachedPage> PageCache::take(HistoryItem& item, Page* page)
 {
     if (!item.m_cachedPage) {
         if (item.m_pruningReason != PruningReason::None)
             logPageCacheFailureDiagnosticMessage(page, pruningReasonToDiagnosticLoggingKey(item.m_pruningReason));
         return nullptr;
     }

     m_items.remove(&item);
     std::unique_ptr<CachedPage> cachedPage = item.takeCachedPage();

     RELEASE_LOG(BackForwardCache, "PageCache::take item: %s, size: %u / %u", item.identifier().string().utf8().data(), pageCount(), maxSize());

     if (cachedPage->hasExpired() || (page && page->isResourceCachingDisabled())) {
         LOG(BackForwardCache, "Not restoring page for %s from back/forward cache because cache entry has expired", item.url().string().ascii().data());
         logPageCacheFailureDiagnosticMessage(page, DiagnosticLoggingKeys::expiredKey());
         return nullptr;
     }

     return cachedPage;
 }

 void PageCache::removeAllItemsForPage(Page& page)
 {
 #if !ASSERT_DISABLED
     ASSERT_WITH_MESSAGE(!m_isInRemoveAllItemsForPage, "We should not reenter this method");
     SetForScope<bool> inRemoveAllItemsForPageScope { m_isInRemoveAllItemsForPage, true };
 #endif

     for (auto it = m_items.begin(); it != m_items.end();) {
         // Increment iterator first so it stays valid after the removal.
         auto current = it;
         ++it;
         if (&(*current)->m_cachedPage->page() == &page) {
             RELEASE_LOG(BackForwardCache, "PageCache::removeAllItemsForPage removing item: %s, size: %u / %u", (*current)->identifier().string().utf8().data(), pageCount() - 1, maxSize());
             (*current)->setCachedPage(nullptr);
             m_items.remove(current);
         }
     }
 }

 CachedPage* PageCache::get(HistoryItem& item, Page* page)
 {
     CachedPage* cachedPage = item.m_cachedPage.get();
     if (!cachedPage) {
         if (item.m_pruningReason != PruningReason::None)
             logPageCacheFailureDiagnosticMessage(page, pruningReasonToDiagnosticLoggingKey(item.m_pruningReason));
         return nullptr;
     }

     if (cachedPage->hasExpired() || (page && page->isResourceCachingDisabled())) {
         LOG(BackForwardCache, "Not restoring page for %s from back/forward cache because cache entry has expired", item.url().string().ascii().data());
         logPageCacheFailureDiagnosticMessage(page, DiagnosticLoggingKeys::expiredKey());
         remove(item);
         return nullptr;
     }
     return cachedPage;
 }

 void PageCache::remove(HistoryItem& item)
 {
     // Safely ignore attempts to remove items not in the cache.
     if (!item.m_cachedPage)
         return;

     m_items.remove(&item);
     item.setCachedPage(nullptr);

     RELEASE_LOG(BackForwardCache, "PageCache::remove item: %s, size: %u / %u", item.identifier().string().utf8().data(), pageCount(), maxSize());

 }

 void PageCache::prune(PruningReason pruningReason)
 {
     while (pageCount() > maxSize()) {
         auto oldestItem = m_items.takeFirst();
         oldestItem->setCachedPage(nullptr);
         oldestItem->m_pruningReason = pruningReason;
         RELEASE_LOG(BackForwardCache, "PageCache::prune removing item: %s, size: %u / %u", oldestItem->identifier().string().utf8().data(), pageCount(), maxSize());
     }
 }

 } // namespace WebCore
	/*
	* Copyright (C) 2007, 2014, 2015 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.
	*
	* THIS SOFTWARE IS PROVIDED BY APPLE INC. ``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 INC. OR
	* 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 "PageCache.h"

	#include "ApplicationCacheHost.h"
	#include "BackForwardController.h"
	#include "CachedPage.h"
	#include "DOMWindow.h"
	#include "DeviceMotionController.h"
	#include "DeviceOrientationController.h"
	#include "DiagnosticLoggingClient.h"
	#include "DiagnosticLoggingKeys.h"
	#include "Document.h"
	#include "DocumentLoader.h"
	#include "FocusController.h"
	#include "Frame.h"
	#include "FrameLoader.h"
	#include "FrameLoaderClient.h"
	#include "FrameView.h"
	#include "HistoryController.h"
	#include "IgnoreOpensDuringUnloadCountIncrementer.h"
	#include "Logging.h"
	#include "Page.h"
	#include "ScriptDisallowedScope.h"
	#include "Settings.h"
	#include "SubframeLoader.h"
	#include <pal/Logging.h>
	#include <wtf/MemoryPressureHandler.h>
	#include <wtf/NeverDestroyed.h>
	#include <wtf/SetForScope.h>
	#include <wtf/text/CString.h>
	#include <wtf/text/StringConcatenate.h>

	namespace WebCore {

	#define PCLOG(...) LOG(BackForwardCache, "%s%s", indentLevel4, "", makeString(__VA_ARGS__).utf8().data())

	static inline void logPageCacheFailureDiagnosticMessage(DiagnosticLoggingClient& client, const String& reason)
	{
	client.logDiagnosticMessage(DiagnosticLoggingKeys::pageCacheFailureKey(), reason, ShouldSample::No);
	}

	static inline void logPageCacheFailureDiagnosticMessage(Page* page, const String& reason)
	{
	if (!page)
	return;

	logPageCacheFailureDiagnosticMessage(page->diagnosticLoggingClient(), reason);
	}

	static bool canCacheFrame(Frame& frame, DiagnosticLoggingClient& diagnosticLoggingClient, unsigned indentLevel)
	{
	PCLOG("+---");
	FrameLoader& frameLoader = frame.loader();

	// Prevent page caching if a subframe is still in provisional load stage.
	// We only do this check for subframes because the main frame is reused when navigating to a new page.
	if (!frame.isMainFrame() && frameLoader.state() == FrameStateProvisional) {
	PCLOG(" -Frame is in provisional load stage");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::provisionalLoadKey());
	return false;
	}

	if (frame.isMainFrame() && frameLoader.stateMachine().isDisplayingInitialEmptyDocument()) {
	PCLOG(" -MainFrame is displaying initial empty document");
	return false;
	}

	if (!frame.document()) {
	PCLOG(" -Frame has no document");
	return false;
	}

	if (!frame.document()->frame()) {
	PCLOG(" -Document is detached from frame");
	ASSERT_NOT_REACHED();
	return false;
	}

	DocumentLoader* documentLoader = frameLoader.documentLoader();
	if (!documentLoader) {
	PCLOG(" -There is no DocumentLoader object");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::noDocumentLoaderKey());
	return false;
	}

	URL currentURL = documentLoader->url();
	URL newURL = frameLoader.provisionalDocumentLoader() ? frameLoader.provisionalDocumentLoader()->url() : URL();
	if (!newURL.isEmpty())
	PCLOG(" Determining if frame can be cached navigating from (", currentURL.string(), ") to (", newURL.string(), "):");
	else
	PCLOG(" Determining if subframe with URL (", currentURL.string(), ") can be cached:");

	bool isCacheable = true;
	// Do not cache error pages (these can be recognized as pages with substitute data or unreachable URLs).
	if (documentLoader->substituteData().isValid() && !documentLoader->substituteData().failingURL().isEmpty()) {
	PCLOG(" -Frame is an error page");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::isErrorPageKey());
	isCacheable = false;
	}
	if (frameLoader.subframeLoader().containsPlugins() && !frame.page()->settings().pageCacheSupportsPlugins()) {
	PCLOG(" -Frame contains plugins");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::hasPluginsKey());
	isCacheable = false;
	}
	if (frame.isMainFrame() && !frameLoader.history().currentItem()) {
	PCLOG(" -Main frame has no current history item");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::noCurrentHistoryItemKey());
	isCacheable = false;
	}
	if (frameLoader.quickRedirectComing()) {
	PCLOG(" -Quick redirect is coming");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::quirkRedirectComingKey());
	isCacheable = false;
	}
	if (documentLoader->isLoading()) {
	PCLOG(" -DocumentLoader is still loading");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::isLoadingKey());
	isCacheable = false;
	}
	if (documentLoader->isStopping()) {
	PCLOG(" -DocumentLoader is in the middle of stopping");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::documentLoaderStoppingKey());
	isCacheable = false;
	}

	Vector<ActiveDOMObject*> unsuspendableObjects;
	if (frame.document() && !frame.document()->canSuspendActiveDOMObjectsForDocumentSuspension(&unsuspendableObjects)) {
	PCLOG(" -The document cannot suspend its active DOM Objects");
	for (auto* activeDOMObject : unsuspendableObjects) {
	PCLOG(" - Unsuspendable: ", activeDOMObject->activeDOMObjectName());
	diagnosticLoggingClient.logDiagnosticMessage(DiagnosticLoggingKeys::unsuspendableDOMObjectKey(), activeDOMObject->activeDOMObjectName(), ShouldSample::No);
	UNUSED_PARAM(activeDOMObject);
	}
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::cannotSuspendActiveDOMObjectsKey());
	isCacheable = false;
	}

	// FIXME: We should investigating caching frames that have an associated
	// application cache. <rdar://problem/5917899> tracks that work.
	if (!documentLoader->applicationCacheHost().canCacheInPageCache()) {
	PCLOG(" -The DocumentLoader uses an application cache");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::applicationCacheKey());
	isCacheable = false;
	}
	if (!frameLoader.client().canCachePage()) {
	PCLOG(" -The client says this frame cannot be cached");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::deniedByClientKey());
	isCacheable = false;
	}

	for (Frame* child = frame.tree().firstChild(); child; child = child->tree().nextSibling()) {
	if (!canCacheFrame(*child, diagnosticLoggingClient, indentLevel + 1))
	isCacheable = false;
	}

	PCLOG(isCacheable ? " Frame CAN be cached" : " Frame CANNOT be cached");
	PCLOG("+---");

	return isCacheable;
	}

	static bool canCachePage(Page& page)
	{
	RELEASE_ASSERT(!page.isRestoringCachedPage());

	unsigned indentLevel = 0;
	PCLOG("--------\n Determining if page can be cached:");

	DiagnosticLoggingClient& diagnosticLoggingClient = page.diagnosticLoggingClient();
	bool isCacheable = canCacheFrame(page.mainFrame(), diagnosticLoggingClient, indentLevel + 1);

	if (!page.settings().usesPageCache() \|\| page.isResourceCachingDisabled()) {
	PCLOG(" -Page settings says b/f cache disabled");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::isDisabledKey());
	isCacheable = false;
	}
	#if ENABLE(DEVICE_ORIENTATION) && !PLATFORM(IOS_FAMILY)
	if (DeviceMotionController::isActiveAt(&page)) {
	PCLOG(" -Page is using DeviceMotion");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::deviceMotionKey());
	isCacheable = false;
	}
	if (DeviceOrientationController::isActiveAt(&page)) {
	PCLOG(" -Page is using DeviceOrientation");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::deviceOrientationKey());
	isCacheable = false;
	}
	#endif

	FrameLoadType loadType = page.mainFrame().loader().loadType();
	switch (loadType) {
	case FrameLoadType::Reload:
	// No point writing to the cache on a reload, since we will just write over it again when we leave that page.
	PCLOG(" -Load type is: Reload");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::reloadKey());
	isCacheable = false;
	break;
	case FrameLoadType::Same: // user loads same URL again (but not reload button)
	// No point writing to the cache on a same load, since we will just write over it again when we leave that page.
	PCLOG(" -Load type is: Same");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::sameLoadKey());
	isCacheable = false;
	break;
	case FrameLoadType::RedirectWithLockedBackForwardList:
	// Don't write to the cache if in the middle of a redirect, since we will want to store the final page we end up on.
	PCLOG(" -Load type is: RedirectWithLockedBackForwardList");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::redirectKey());
	isCacheable = false;
	break;
	case FrameLoadType::Replace:
	// No point writing to the cache on a replace, since we will just write over it again when we leave that page.
	PCLOG(" -Load type is: Replace");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::replaceKey());
	isCacheable = false;
	break;
	case FrameLoadType::ReloadFromOrigin: {
	// No point writing to the cache on a reload, since we will just write over it again when we leave that page.
	PCLOG(" -Load type is: ReloadFromOrigin");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::reloadFromOriginKey());
	isCacheable = false;
	break;
	}
	case FrameLoadType::ReloadExpiredOnly: {
	// No point writing to the cache on a reload, since we will just write over it again when we leave that page.
	PCLOG(" -Load type is: ReloadRevalidatingExpired");
	logPageCacheFailureDiagnosticMessage(diagnosticLoggingClient, DiagnosticLoggingKeys::reloadRevalidatingExpiredKey());
	isCacheable = false;
	break;
	}
	case FrameLoadType::Standard:
	case FrameLoadType::Back:
	case FrameLoadType::Forward:
	case FrameLoadType::IndexedBackForward: // a multi-item hop in the backforward list
	// Cacheable.
	break;
	}

	if (isCacheable)
	PCLOG(" Page CAN be cached\n--------");
	else
	PCLOG(" Page CANNOT be cached\n--------");

	diagnosticLoggingClient.logDiagnosticMessageWithResult(DiagnosticLoggingKeys::pageCacheKey(), DiagnosticLoggingKeys::canCacheKey(), isCacheable ? DiagnosticLoggingResultPass : DiagnosticLoggingResultFail, ShouldSample::No);
	return isCacheable;
	}

	PageCache& PageCache::singleton()
	{
	static NeverDestroyed<PageCache> globalPageCache;
	return globalPageCache;
	}

	PageCache::PageCache()
	{
	static std::once_flag onceFlag;
	std::call_once(onceFlag, [] {
	PAL::registerNotifyCallback("com.apple.WebKit.showPageCache", [] {
	PageCache::singleton().dump();
	});
	});
	}

	void PageCache::dump() const
	{
	WTFLogAlways("\nPage Cache:");
	for (auto& item : m_items) {
	CachedPage& cachedPage = *item->m_cachedPage;
	WTFLogAlways(" Page %p, document %p %s", &cachedPage.page(), cachedPage.document(), cachedPage.document() ? cachedPage.document()->url().string().utf8().data() : "");
	}
	}

	bool PageCache::canCache(Page& page) const
	{
	if (!m_maxSize) {
	logPageCacheFailureDiagnosticMessage(&page, DiagnosticLoggingKeys::isDisabledKey());
	return false;
	}

	if (MemoryPressureHandler::singleton().isUnderMemoryPressure()) {
	logPageCacheFailureDiagnosticMessage(&page, DiagnosticLoggingKeys::underMemoryPressureKey());
	return false;
	}

	return canCachePage(page);
	}

	void PageCache::pruneToSizeNow(unsigned size, PruningReason pruningReason)
	{
	SetForScope<unsigned> change(m_maxSize, size);
	prune(pruningReason);
	}

	void PageCache::setMaxSize(unsigned maxSize)
	{
	m_maxSize = maxSize;
	prune(PruningReason::None);
	}

	unsigned PageCache::frameCount() const
	{
	unsigned frameCount = m_items.size();
	for (auto& item : m_items) {
	ASSERT(item->m_cachedPage);
	frameCount += item->m_cachedPage->cachedMainFrame()->descendantFrameCount();
	}

	return frameCount;
	}

	void PageCache::markPagesForDeviceOrPageScaleChanged(Page& page)
	{
	for (auto& item : m_items) {
	CachedPage& cachedPage = *item->m_cachedPage;
	if (&page.mainFrame() == &cachedPage.cachedMainFrame()->view()->frame())
	cachedPage.markForDeviceOrPageScaleChanged();
	}
	}

	void PageCache::markPagesForContentsSizeChanged(Page& page)
	{
	for (auto& item : m_items) {
	CachedPage& cachedPage = *item->m_cachedPage;
	if (&page.mainFrame() == &cachedPage.cachedMainFrame()->view()->frame())
	cachedPage.markForContentsSizeChanged();
	}
	}

	#if ENABLE(VIDEO_TRACK)
	void PageCache::markPagesForCaptionPreferencesChanged()
	{
	for (auto& item : m_items) {
	ASSERT(item->m_cachedPage);
	item->m_cachedPage->markForCaptionPreferencesChanged();
	}
	}
	#endif

	static String pruningReasonToDiagnosticLoggingKey(PruningReason pruningReason)
	{
	switch (pruningReason) {
	case PruningReason::MemoryPressure:
	return DiagnosticLoggingKeys::prunedDueToMemoryPressureKey();
	case PruningReason::ProcessSuspended:
	return DiagnosticLoggingKeys::prunedDueToProcessSuspended();
	case PruningReason::ReachedMaxSize:
	return DiagnosticLoggingKeys::prunedDueToMaxSizeReached();
	case PruningReason::None:
	break;
	}
	ASSERT_NOT_REACHED();
	return emptyString();
	}

	static void setPageCacheState(Page& page, Document::PageCacheState pageCacheState)
	{
	for (Frame* frame = &page.mainFrame(); frame; frame = frame->tree().traverseNext()) {
	if (auto* document = frame->document())
	document->setPageCacheState(pageCacheState);
	}
	}

	// When entering page cache, tear down the render tree before setting the in-cache flag.
	// This maintains the invariant that render trees are never present in the page cache.
	// Note that destruction happens bottom-up so that the main frame's tree dies last.
	static void destroyRenderTree(Frame& mainFrame)
	{
	for (Frame* frame = mainFrame.tree().traversePrevious(CanWrap::Yes); frame; frame = frame->tree().traversePrevious(CanWrap::No)) {
	if (!frame->document())
	continue;
	auto& document = *frame->document();
	if (document.hasLivingRenderTree())
	document.destroyRenderTree();
	}
	}

	static void firePageHideEventRecursively(Frame& frame)
	{
	auto* document = frame.document();
	if (!document)
	return;

	// stopLoading() will fire the pagehide event in each subframe and the HTML specification states
	// that the parent document's ignore-opens-during-unload counter should be incremented while the
	// pagehide event is being fired in its subframes:
	// https://html.spec.whatwg.org/multipage/browsers.html#unload-a-document
	IgnoreOpensDuringUnloadCountIncrementer ignoreOpensDuringUnloadCountIncrementer(document);

	frame.loader().stopLoading(UnloadEventPolicyUnloadAndPageHide);

	for (RefPtr<Frame> child = frame.tree().firstChild(); child; child = child->tree().nextSibling())
	firePageHideEventRecursively(*child);
	}

	bool PageCache::addIfCacheable(HistoryItem& item, Page* page)
	{
	if (item.isInPageCache())
	return false;

	if (!page)
	return false;

	page->mainFrame().loader().stopForPageCache();

	if (!canCache(*page))
	return false;

	ASSERT_WITH_MESSAGE(!page->isUtilityPage(), "Utility pages such as SVGImage pages should never go into PageCache");

	setPageCacheState(*page, Document::AboutToEnterPageCache);

	// Focus the main frame, defocusing a focused subframe (if we have one). We do this here,
	// before the page enters the page cache, while we still can dispatch DOM blur/focus events.
	if (page->focusController().focusedFrame())
	page->focusController().setFocusedFrame(&page->mainFrame());

	// Fire the pagehide event in all frames.
	firePageHideEventRecursively(page->mainFrame());

	destroyRenderTree(page->mainFrame());

	// Stop all loads again before checking if we can still cache the page after firing the pagehide
	// event, since the page may have started ping loads in its pagehide event handler.
	page->mainFrame().loader().stopForPageCache();

	// Check that the page is still page-cacheable after firing the pagehide event. The JS event handlers
	// could have altered the page in a way that could prevent caching.
	if (!canCache(*page)) {
	setPageCacheState(*page, Document::NotInPageCache);
	return false;
	}

	setPageCacheState(*page, Document::InPageCache);

	{
	// Make sure we don't fire any JS events in this scope.
	ScriptDisallowedScope::InMainThread scriptDisallowedScope;

	item.setCachedPage(makeUnique<CachedPage>(*page));
	item.m_pruningReason = PruningReason::None;
	m_items.add(&item);
	}
	prune(PruningReason::ReachedMaxSize);

	RELEASE_LOG(BackForwardCache, "PageCache::addIfCacheable item: %s, size: %u / %u", item.identifier().string().utf8().data(), pageCount(), maxSize());

	return true;
	}

	std::unique_ptr<CachedPage> PageCache::take(HistoryItem& item, Page* page)
	{
	if (!item.m_cachedPage) {
	if (item.m_pruningReason != PruningReason::None)
	logPageCacheFailureDiagnosticMessage(page, pruningReasonToDiagnosticLoggingKey(item.m_pruningReason));
	return nullptr;
	}

	m_items.remove(&item);
	std::unique_ptr<CachedPage> cachedPage = item.takeCachedPage();

	RELEASE_LOG(BackForwardCache, "PageCache::take item: %s, size: %u / %u", item.identifier().string().utf8().data(), pageCount(), maxSize());

	if (cachedPage->hasExpired() \|\| (page && page->isResourceCachingDisabled())) {
	LOG(BackForwardCache, "Not restoring page for %s from back/forward cache because cache entry has expired", item.url().string().ascii().data());
	logPageCacheFailureDiagnosticMessage(page, DiagnosticLoggingKeys::expiredKey());
	return nullptr;
	}

	return cachedPage;
	}

	void PageCache::removeAllItemsForPage(Page& page)
	{
	#if !ASSERT_DISABLED
	ASSERT_WITH_MESSAGE(!m_isInRemoveAllItemsForPage, "We should not reenter this method");
	SetForScope<bool> inRemoveAllItemsForPageScope { m_isInRemoveAllItemsForPage, true };
	#endif

	for (auto it = m_items.begin(); it != m_items.end();) {
	// Increment iterator first so it stays valid after the removal.
	auto current = it;
	++it;
	if (&(*current)->m_cachedPage->page() == &page) {
	RELEASE_LOG(BackForwardCache, "PageCache::removeAllItemsForPage removing item: %s, size: %u / %u", (*current)->identifier().string().utf8().data(), pageCount() - 1, maxSize());
	(*current)->setCachedPage(nullptr);
	m_items.remove(current);
	}
	}
	}

	CachedPage* PageCache::get(HistoryItem& item, Page* page)
	{
	CachedPage* cachedPage = item.m_cachedPage.get();
	if (!cachedPage) {
	if (item.m_pruningReason != PruningReason::None)
	logPageCacheFailureDiagnosticMessage(page, pruningReasonToDiagnosticLoggingKey(item.m_pruningReason));
	return nullptr;
	}

	if (cachedPage->hasExpired() \|\| (page && page->isResourceCachingDisabled())) {
	LOG(BackForwardCache, "Not restoring page for %s from back/forward cache because cache entry has expired", item.url().string().ascii().data());
	logPageCacheFailureDiagnosticMessage(page, DiagnosticLoggingKeys::expiredKey());
	remove(item);
	return nullptr;
	}
	return cachedPage;
	}

	void PageCache::remove(HistoryItem& item)
	{
	// Safely ignore attempts to remove items not in the cache.
	if (!item.m_cachedPage)
	return;

	m_items.remove(&item);
	item.setCachedPage(nullptr);

	RELEASE_LOG(BackForwardCache, "PageCache::remove item: %s, size: %u / %u", item.identifier().string().utf8().data(), pageCount(), maxSize());

	}

	void PageCache::prune(PruningReason pruningReason)
	{
	while (pageCount() > maxSize()) {
	auto oldestItem = m_items.takeFirst();
	oldestItem->setCachedPage(nullptr);
	oldestItem->m_pruningReason = pruningReason;
	RELEASE_LOG(BackForwardCache, "PageCache::prune removing item: %s, size: %u / %u", oldestItem->identifier().string().utf8().data(), pageCount(), maxSize());
	}
	}

	} // namespace WebCore