Web restrictions component only.

components/web_restrictions/content_resolver_web_restrictions_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/sequenced_task_runner.h"
+#include "base/single_thread_task_runner.h"
+#include "components/web_restrictions/web_restrictions_provider.h"
+
+namespace web_restrictions {
+
+class ContentResolverWebRestrictionsProvider : public WebRestrictionsProvider {
+ public:
+  // An instance of this class is expected to live through the lifetime of a
+  // browser and uses raw pointers in callbacks.
+  // Any changes to the class, enable/disable/change should be done through the
+  // SetAuthority(...) method.
+  ContentResolverWebRestrictionsProvider();
+  ~ContentResolverWebRestrictionsProvider() override;
+
+  // Register JNI methods.
+  static bool Register(JNIEnv* env);
+
+  // Verify the content provider and query it for basic information like does it
+  // support handling requests. This should be called everytime the provider
+  // changes. An empty authority can be used to disable this class.
+  void SetAuthority(const std::string& content_provider_authority);
+
+  // WebRestrictionsProvider:
+  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;
+
+  void OnWebRestrictionsChanged();
+
+ private:
+  friend class SelfDeletingCallback;  // For updating the cache.
+
+  void RecordURLAccess(const GURL& url);
+
+  void UpdateCache(std::string provider_authority,
+                   GURL url,
+                   bool should_proceed,
+                   std::string error_page);
+
+  void RequestSupportKnown(std::string provider_authority,
+                           bool supports_request);
+
+  void ClearCache();
+
+  // Set up after SetAuthority().
+  bool initialized_;
+  bool supports_request_;
+  base::android::ScopedJavaGlobalRef<jobject> java_provider_;
+  std::string provider_authority_;
+
+  scoped_refptr<base::SequencedTaskRunner> background_task_runner_;
+  scoped_refptr<base::SingleThreadTaskRunner> single_thread_task_runner_;
+
+  std::map<GURL, std::string> error_page_cache_;
+  std::map<GURL, bool> url_access_cache_;
+  std::list<GURL> recent_urls_;
+
+  DISALLOW_COPY_AND_ASSIGN(ContentResolverWebRestrictionsProvider);
+};
+
+}  // namespace web_restrictions
+
+#endif  // COMPONENTS_WEB_RESTRICTION_CONTENT_RESOLVER_WEB_RESTRICTION_PROVIDER_H_