[WIP] WebRestrictions

components/web_restriction/web_restriction_resource_throttle.cc

+// Copyright 2015 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/web_restriction/web_restriction_resource_throttle.h"
+
+#include "base/bind.h"
+#include "components/web_restriction/web_restriction_provider.h"
+#include "content/public/browser/resource_controller.h"
+#include "net/base/net_errors.h"
+#include "net/url_request/redirect_info.h"
+#include "net/url_request/url_request.h"
+
+namespace web_restriction {
+
+WebRestrictionResourceThrottle::WebRestrictionResourceThrottle(
+    WebRestrictionProvider* provider,
+    const GURL& request_url,
+    bool is_main_frame)
+    : provider_(provider),
+      request_url_(request_url),
+      is_main_frame_(is_main_frame),
+      weak_ptr_factory_(this) {}
+
+WebRestrictionResourceThrottle::~WebRestrictionResourceThrottle() {}
+
+void WebRestrictionResourceThrottle::WillStartRequest(bool* defer) {
+  *defer = ShouldDefer(request_url_);
+}
+
+void WebRestrictionResourceThrottle::WillRedirectRequest(
+    const net::RedirectInfo& redirect_info,
+    bool* defer) {
+  *defer = ShouldDefer(redirect_info.new_url);
+}
+
+const char* WebRestrictionResourceThrottle::GetNameForLogging() const {
+  return "WebRestrictionResourceThrottle";
+}
+
+bool WebRestrictionResourceThrottle::ShouldDefer(const GURL& url) {
+  // For requests to function correctly, we need to allow subresources.
+  if (provider_->SupportsRequest() && !is_main_frame_)
+    return false;
+  UrlAccess access = provider_->ShouldProceed(
+      is_main_frame_, url,
+      base::Bind(&WebRestrictionResourceThrottle::OnCheckResult,
+                 weak_ptr_factory_.GetWeakPtr()));
+  if (access == DISALLOW)
+    OnCheckResult(false);
+  return access == PENDING;
+}
+
+void WebRestrictionResourceThrottle::OnCheckResult(const bool should_proceed) {
+  if (should_proceed) {
+    controller()->Resume();
+  } else {
+    controller()->CancelWithError(net::ERR_BLOCKED_BY_ADMINISTRATOR);
+  }
+}
+
+}  // namespace web_restriction