chrome/browser/ui/sad_tab_helper.cc

// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/sad_tab_helper.h"

#include "build/build_config.h"
#include "chrome/browser/lifetime/browser_shutdown.h"
#include "chrome/browser/ui/sad_tab.h"
#include "content/common/content_navigation_policy.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/web_contents.h"

namespace {

SadTabKind SadTabKindFromTerminationStatus(base::TerminationStatus status) {
  switch (status) {
#if BUILDFLAG(IS_CHROMEOS)
    case base::TERMINATION_STATUS_PROCESS_WAS_KILLED_BY_OOM:
      return SAD_TAB_KIND_KILLED_BY_OOM;
#endif
    case base::TERMINATION_STATUS_PROCESS_WAS_KILLED:
    case base::TERMINATION_STATUS_LAUNCH_FAILED:
      return SAD_TAB_KIND_KILLED;
    case base::TERMINATION_STATUS_OOM:
      return SAD_TAB_KIND_OOM;
    default:
      return SAD_TAB_KIND_CRASHED;
  }
}

}  // namespace

SadTabHelper::~SadTabHelper() {}

SadTabHelper::SadTabHelper(content::WebContents* web_contents)
    : content::WebContentsObserver(web_contents),
      content::WebContentsUserData<SadTabHelper>(*web_contents) {}

void SadTabHelper::ReinstallInWebView() {
  if (sad_tab_)
    sad_tab_->ReinstallInWebView();
}

void SadTabHelper::RenderFrameCreated(
    content::RenderFrameHost* render_frame_host) {
  if (content::ShouldSkipEarlyCommitPendingForCrashedFrame())
    sad_tab_.reset();
}

void SadTabHelper::RenderViewReady() {
  if (!content::ShouldSkipEarlyCommitPendingForCrashedFrame())
    sad_tab_.reset();
}

void SadTabHelper::DidFinishNavigation(
    content::NavigationHandle* navigation_handle) {
  if (!content::ShouldSkipEarlyCommitPendingForCrashedFrame())
    return;
  // If the navigation did not commit and we went back to the crashed frame,
  // reinstall the sad tab, if needed.
  if (!sad_tab_ && !navigation_handle->HasCommitted() &&
      web_contents()->IsCrashed()) {
    InstallSadTab(web_contents()->GetCrashedStatus());
  }
}

void SadTabHelper::PrimaryMainFrameRenderProcessGone(
    base::TerminationStatus status) {
  // Only show the sad tab if we're not in browser shutdown, so that WebContents
  // objects that are not in a browser (e.g., HTML dialogs) and thus are
  // visible do not flash a sad tab page.
  if (browser_shutdown::HasShutdownStarted())
    return;

  if (sad_tab_)
    return;

  if (SadTab::ShouldShow(status))
    InstallSadTab(status);
}

void SadTabHelper::InstallSadTab(base::TerminationStatus status) {
  sad_tab_.reset(
      SadTab::Create(web_contents(), SadTabKindFromTerminationStatus(status)));
}

WEB_CONTENTS_USER_DATA_KEY_IMPL(SadTabHelper);