ios/chrome/browser/safe_browsing/safe_browsing_client_impl.mm

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#import "ios/chrome/browser/safe_browsing/safe_browsing_client_impl.h"

#import "base/check.h"
#import "base/memory/weak_ptr.h"
#import "components/safe_browsing/core/browser/realtime/url_lookup_service.h"
#import "components/security_interstitials/core/unsafe_resource.h"
#import "ios/chrome/browser/application_context/application_context.h"
#import "ios/chrome/browser/prerender/prerender_service.h"
#import "ios/web/public/web_state.h"

#if !defined(__has_feature) || !__has_feature(objc_arc)
#error "This file requires ARC support."
#endif

SafeBrowsingClientImpl::SafeBrowsingClientImpl(
    safe_browsing::RealTimeUrlLookupService* lookup_service,
    PrerenderService* prerender_service)
    : lookup_service_(lookup_service), prerender_service_(prerender_service) {}

SafeBrowsingClientImpl::~SafeBrowsingClientImpl() = default;

base::WeakPtr<SafeBrowsingClient> SafeBrowsingClientImpl::AsWeakPtr() {
  return weak_factory_.GetWeakPtr();
}

SafeBrowsingService* SafeBrowsingClientImpl::GetSafeBrowsingService() {
  return GetApplicationContext()->GetSafeBrowsingService();
}

safe_browsing::RealTimeUrlLookupService*
SafeBrowsingClientImpl::GetRealTimeUrlLookupService() {
  return lookup_service_;
}

bool SafeBrowsingClientImpl::ShouldBlockUnsafeResource(
    const security_interstitials::UnsafeResource& resource) const {
  // Send do-not-proceed signal if the WebState is for a prerender tab.
  web::WebState* web_state = resource.weak_web_state.get();
  return prerender_service_ &&
         prerender_service_->IsWebStatePrerendered(web_state);
}

void SafeBrowsingClientImpl::OnMainFrameUrlQueryCancellationDecided(
    web::WebState* web_state,
    const GURL& url) {
  // When a prendered page is unsafe, cancel the prerender.
  if (prerender_service_ &&
      prerender_service_->IsWebStatePrerendered(web_state)) {
    prerender_service_->CancelPrerender();
  }
}

bool SafeBrowsingClientImpl::OnSubFrameUrlQueryCancellationDecided(
    web::WebState* web_state,
    const GURL& url) {
  // When a subframe in a prerendered page is unsafe, cancel the prerender.
  if (prerender_service_ &&
      prerender_service_->IsWebStatePrerendered(web_state)) {
    prerender_service_->CancelPrerender();
    return false;
  }
  return true;
}