android_webview/browser/safe_browsing/aw_safe_browsing_subresource_helper.cc

// Copyright 2019 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 "android_webview/browser/safe_browsing/aw_safe_browsing_subresource_helper.h"

#include <memory>

#include "android_webview/browser/aw_browser_process.h"
#include "android_webview/browser/network_service/aw_web_resource_request.h"
#include "android_webview/browser/safe_browsing/aw_safe_browsing_blocking_page.h"
#include "android_webview/browser/safe_browsing/aw_safe_browsing_ui_manager.h"
#include "base/memory/ptr_util.h"
#include "components/security_interstitials/content/security_interstitial_tab_helper.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_contents_observer.h"
#include "net/base/net_errors.h"

namespace android_webview {

AwSafeBrowsingSubresourceHelper::~AwSafeBrowsingSubresourceHelper() {}

void AwSafeBrowsingSubresourceHelper::ReadyToCommitNavigation(
    content::NavigationHandle* navigation_handle) {
  if (navigation_handle->GetNetErrorCode() == net::ERR_BLOCKED_BY_CLIENT) {
    AwSafeBrowsingUIManager* manager =
        AwBrowserProcess::GetInstance()->GetSafeBrowsingUIManager();
    if (!manager)
      return;
    security_interstitials::UnsafeResource resource;
    if (manager->PopUnsafeResourceForURL(navigation_handle->GetURL(),
                                         &resource)) {
      std::unique_ptr<AwWebResourceRequest> request =
          std::make_unique<AwWebResourceRequest>(
              navigation_handle->GetURL().spec(),
              navigation_handle->IsPost() ? "POST" : "GET",
              navigation_handle->IsInMainFrame(),
              navigation_handle->HasUserGesture(),
              navigation_handle->GetRequestHeaders());
      request->is_renderer_initiated = navigation_handle->IsRendererInitiated();
      AwSafeBrowsingBlockingPage* blocking_page =
          AwSafeBrowsingBlockingPage::CreateBlockingPage(
              manager, navigation_handle->GetWebContents(),
              navigation_handle->GetURL(), resource, std::move(request));
      security_interstitials::SecurityInterstitialTabHelper::
          AssociateBlockingPage(navigation_handle->GetWebContents(),
                                navigation_handle->GetNavigationId(),
                                base::WrapUnique(blocking_page));
    }
  }
}

AwSafeBrowsingSubresourceHelper::AwSafeBrowsingSubresourceHelper(
    content::WebContents* web_contents)
    : WebContentsObserver(web_contents) {}

WEB_CONTENTS_USER_DATA_KEY_IMPL(AwSafeBrowsingSubresourceHelper)

}  // namespace android_webview