src/chrome/browser/managed_mode/managed_mode_resource_throttle.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/managed_mode/managed_mode_resource_throttle.h"
   6
   7 #include "base/bind.h"
   8 #include "base/lazy_instance.h"
   9 #include "chrome/browser/managed_mode/managed_mode_interstitial.h"
  10 #include "chrome/browser/managed_mode/managed_mode_navigation_observer.h"
  11 #include "chrome/browser/managed_mode/managed_mode_url_filter.h"
  12 #include "content/public/browser/browser_thread.h"
  13 #include "content/public/browser/resource_controller.h"
  14 #include "content/public/browser/resource_request_info.h"
  15 #include "net/url_request/url_request.h"
  16
  17 using content::BrowserThread;
  18
  19 ManagedModeResourceThrottle::ManagedModeResourceThrottle(
  20     const net::URLRequest* request,
  21     bool is_main_frame,
  22     const ManagedModeURLFilter* url_filter)
  23     : weak_ptr_factory_(this),
  24       request_(request),
  25       is_main_frame_(is_main_frame),
  26       url_filter_(url_filter) {}
  27
  28 ManagedModeResourceThrottle::~ManagedModeResourceThrottle() {}
  29
  30 void ManagedModeResourceThrottle::ShowInterstitialIfNeeded(bool is_redirect,
  31                                                            const GURL& url,
  32                                                            bool* defer) {
  33   // Only treat main frame requests for now (ignoring subresources).
  34   if (!is_main_frame_)
  35     return;
  36
  37   if (url_filter_->GetFilteringBehaviorForURL(url) !=
  38       ManagedModeURLFilter::BLOCK) {
  39     return;
  40   }
  41
  42   *defer = true;
  43   const content::ResourceRequestInfo* info =
  44       content::ResourceRequestInfo::ForRequest(request_);
  45   BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
  46       base::Bind(&ManagedModeNavigationObserver::OnRequestBlocked,
  47                  info->GetChildID(), info->GetRouteID(), url,
  48                  base::Bind(&ManagedModeResourceThrottle::OnInterstitialResult,
  49                             weak_ptr_factory_.GetWeakPtr())));
  50 }
  51
  52 void ManagedModeResourceThrottle::WillStartRequest(bool* defer) {
  53   ShowInterstitialIfNeeded(false, request_->url(), defer);
  54 }
  55
  56 void ManagedModeResourceThrottle::WillRedirectRequest(const GURL& new_url,
  57                                                       bool* defer) {
  58   ShowInterstitialIfNeeded(true, new_url, defer);
  59 }
  60
  61 void ManagedModeResourceThrottle::OnInterstitialResult(bool continue_request) {
  62   if (continue_request)
  63     controller()->Resume();
  64   else
  65     controller()->Cancel();
  66 }