[WIP] WebRestrictions

components/web_restriction/content_resolver_web_restriction_provider.h

+// 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.
+
+#ifndef COMPONENTS_WEB_RESTRICTION_CONTENT_RESOLVER_WEB_RESTRICTION_PROVIDER_H_
+#define COMPONENTS_WEB_RESTRICTION_CONTENT_RESOLVER_WEB_RESTRICTION_PROVIDER_H_
+
+#include <jni.h>
+#include <list>
+#include <map>
+
+#include "base/android/jni_android.h"
+#include "base/callback.h"
+#include "base/macros.h"
+#include "base/memory/ref_counted.h"
+#include "base/task_runner.h"
+#include "components/web_restriction/web_restriction_provider.h"
+
+namespace web_restriction {
+
+class ContentResolverWebRestrictionProvider;
+
+namespace android {
+
+// A wrapper to the callback class to facilitate getting a callback from java
+// into C++. Objects of this class delete itself only when they are called back
+// so we must ensure that happens even for error cases.
+class SelfDeletingCallback {
+ public:
+  SelfDeletingCallback(const GURL& url,
+                       const base::Callback<void(bool)>& callback,
+                       const scoped_refptr<base::TaskRunner>& callback_runner,
+                       ContentResolverWebRestrictionProvider* provider);
+
+  void RequestSuccess(JNIEnv* env,
+                      const base::android::JavaParamRef<jobject>& obj,
+                      jboolean request_success);
+  void ShouldProceed(JNIEnv* env,
+                     const base::android::JavaParamRef<jobject>& obj,
+                     jboolean should_proceed,
+                     const base::android::JavaParamRef<jstring>& error_page);
+
+ private:
+  GURL url_;
+  base::Callback<void(bool)> callback_;
+  scoped_refptr<base::TaskRunner> callback_runner_;
+  ContentResolverWebRestrictionProvider* provider_;
+
+  // Only the callback can delete itself. We must ensure it is indeed
+  // called back.
+  ~SelfDeletingCallback();
+  DISALLOW_COPY_AND_ASSIGN(SelfDeletingCallback);
+};
+
+}  // namespace android
+
+class ContentResolverWebRestrictionProvider : public WebRestrictionProvider {
+ public:
+  ContentResolverWebRestrictionProvider();
+  ~ContentResolverWebRestrictionProvider() override;
+
+  // Register JNI methods.
+  static bool Register(JNIEnv* env);
+
+  // Verify the content provider and Setup.
+  void Initialize(const std::string& content_provider_authority);
+
+  // WebRestrictionProvider:
+  UrlAccess ShouldProceed(bool is_main_frame,
+                          const GURL& url,
+                          const base::Callback<void(bool)>& callback) override;
+
+  bool SupportsRequest() const override;
+
+  bool GetErrorHtml(const GURL& url, std::string* error_html) const override;
+
+  void RequestPermission(const GURL& url,
+                         const base::Callback<void(bool)>& callback) override;
+
+ private:
+  friend class android::SelfDeletingCallback;  // For updating the cache.
+
+  bool initialized_;
+  // Setup during Initialize.
+  bool supports_request_;
+  base::android::ScopedJavaGlobalRef<jobject> java_provider_;
+
+  std::map<GURL, std::string> error_page_cache_;
+  std::map<GURL, bool> url_access_cache_;
+  std::list<GURL> recent_urls_;
+
+  DISALLOW_COPY_AND_ASSIGN(ContentResolverWebRestrictionProvider);
+};
+
+}  // namespace web_restriction
+
+#endif  // COMPONENTS_WEB_RESTRICTION_CONTENT_RESOLVER_WEB_RESTRICTION_PROVIDER_H_