components/policy/content/safe_sites_navigation_throttle.cc

// Copyright 2020 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 "components/policy/content/safe_sites_navigation_throttle.h"

#include "base/bind.h"
#include "base/strings/string_piece.h"
#include "components/policy/content/safe_search_service.h"
#include "components/policy/core/browser/url_util.h"
#include "content/public/browser/navigation_handle.h"
#include "url/gurl.h"

SafeSitesNavigationThrottle::SafeSitesNavigationThrottle(
    content::NavigationHandle* navigation_handle,
    content::BrowserContext* context)
    : SafeSitesNavigationThrottle(
          navigation_handle,
          context,
          base::BindRepeating(&SafeSitesNavigationThrottle::OnDeferredResult,
                              base::Unretained(this))) {}

SafeSitesNavigationThrottle::SafeSitesNavigationThrottle(
    content::NavigationHandle* navigation_handle,
    content::BrowserContext* context,
    DeferredResultCallback deferred_result_callback)
    : NavigationThrottle(navigation_handle),
      safe_seach_service_(SafeSearchFactory::GetForBrowserContext(context)),
      deferred_result_callback_(std::move(deferred_result_callback)) {}

// Use of Unretained for is safe because it is called synchronously from this
// object.
SafeSitesNavigationThrottle::SafeSitesNavigationThrottle(
    content::NavigationHandle* navigation_handle,
    content::BrowserContext* context,
    base::StringPiece safe_sites_error_page_content)
    : NavigationThrottle(navigation_handle),
      safe_seach_service_(SafeSearchFactory::GetForBrowserContext(context)),
      deferred_result_callback_(
          base::BindRepeating(&SafeSitesNavigationThrottle::OnDeferredResult,
                              base::Unretained(this))),
      safe_sites_error_page_content_(
          safe_sites_error_page_content.as_string()) {}

SafeSitesNavigationThrottle::~SafeSitesNavigationThrottle() = default;

content::NavigationThrottle::ThrottleCheckResult
SafeSitesNavigationThrottle::WillStartRequest() {
  const GURL& url = navigation_handle()->GetURL();

  // Ignore blob scheme because we may use it to deliver navigation responses
  // to the renderer process.
  if (url.SchemeIs(url::kBlobScheme))
    return PROCEED;

  // Safe Sites filter applies to top-level HTTP[S] requests.
  if (!url.SchemeIsHTTPOrHTTPS())
    return PROCEED;

  GURL effective_url = policy::url_util::GetEmbeddedURL(url);
  if (!effective_url.is_valid())
    effective_url = url;

  bool synchronous = safe_seach_service_->CheckSafeSearchURL(
      effective_url,
      base::BindOnce(&SafeSitesNavigationThrottle::CheckSafeSearchCallback,
                     weak_ptr_factory_.GetWeakPtr()));
  if (!synchronous) {
    deferred_ = true;
    return DEFER;
  }

  if (should_cancel_)
    return CreateCancelResult();
  return PROCEED;
}

content::NavigationThrottle::ThrottleCheckResult
SafeSitesNavigationThrottle::WillRedirectRequest() {
  return WillStartRequest();
}

const char* SafeSitesNavigationThrottle::GetNameForLogging() {
  return "SafeSitesNavigationThrottle";
}

void SafeSitesNavigationThrottle::CheckSafeSearchCallback(bool is_safe) {
  if (!deferred_) {
    should_cancel_ = !is_safe;
    return;
  }

  deferred_ = false;
  deferred_result_callback_.Run(is_safe, CreateCancelResult());
}

void SafeSitesNavigationThrottle::OnDeferredResult(
    bool is_safe,
    ThrottleCheckResult cancel_result) {
  if (is_safe) {
    Resume();
  } else {
    CancelDeferredNavigation(cancel_result);
  }
}

content::NavigationThrottle::ThrottleCheckResult
SafeSitesNavigationThrottle::CreateCancelResult() const {
  return ThrottleCheckResult(CANCEL, net::ERR_BLOCKED_BY_ADMINISTRATOR,
                             safe_sites_error_page_content_);
}