Refactor PluginPermissionHelper as WebViewPermissionHelper

chrome/browser/guest_view/web_view/web_view_permission_helper.h

+// Copyright 2014 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 CHROME_BROWSER_GUEST_VIEW_WEB_VIEW_WEB_VIEW_PERMISSION_HELPER_H_
+#define CHROME_BROWSER_GUEST_VIEW_WEB_VIEW_WEB_VIEW_PERMISSION_HELPER_H_
+
+#include "base/memory/weak_ptr.h"
+#include "base/metrics/user_metrics_action.h"
+#include "chrome/browser/guest_view/guest_view_constants.h"
+#include "chrome/browser/guest_view/web_view/web_view_permission_types.h"
+#include "content/public/browser/web_contents_observer.h"
+#include "content/public/browser/web_contents_user_data.h"
+#include "content/public/common/media_stream_request.h"
+
+using base::UserMetricsAction;
+
+class WebViewGuest;
+
+class WebViewPermissionHelper
+    : public content::WebContentsUserData<WebViewPermissionHelper>,
+      public content::WebContentsObserver {
+ public:
+  explicit WebViewPermissionHelper(content::WebContents* web_contents);
+  virtual ~WebViewPermissionHelper();
+  typedef base::Callback<
+      void(bool /* allow */, const std::string& /* user_input */)>
+      PermissionResponseCallback;
+
+  // A map to store the callback for a request keyed by the request's id.
+  struct PermissionResponseInfo {
+    PermissionResponseCallback callback;
+    WebViewPermissionType permission_type;
+    bool allowed_by_default;
+    PermissionResponseInfo();
+    PermissionResponseInfo(const PermissionResponseCallback& callback,
+                           WebViewPermissionType permission_type,
+                           bool allowed_by_default);
+    ~PermissionResponseInfo();
+  };
+
+  typedef std::map<int, PermissionResponseInfo> RequestMap;
+
+  int RequestPermission(WebViewPermissionType permission_type,
+                        const base::DictionaryValue& request_info,
+                        const PermissionResponseCallback& callback,
+                        bool allowed_by_default);
+
+  void RequestMediaAccessPermission(
+      content::WebContents* source,
+      const content::MediaStreamRequest& request,
+      const content::MediaResponseCallback& callback);
+  void CanDownload(content::RenderViewHost* render_view_host,
+                   const GURL& url,
+                   const std::string& request_method,
+                   const base::Callback<void(bool)>& callback);
+  void RequestPointerLockPermission(bool user_gesture,
+                                    bool last_unlocked_by_target,
+                                    const base::Callback<void(bool)>& callback);
+
+  // Requests Geolocation Permission from the embedder.
+  void RequestGeolocationPermission(int bridge_id,
+                                    const GURL& requesting_frame,
+                                    bool user_gesture,
+                                    const base::Callback<void(bool)>& callback);
+  void CancelGeolocationPermissionRequest(int bridge_id);
+
+  void RequestFileSystemPermission(const GURL& url,
+                                   bool allowed_by_default,
+                                   const base::Callback<void(bool)>& callback);
+
+  // Called when file system access is requested by the guest content using the
+  // asynchronous HTML5 file system API. The request is plumbed through the
+  // <webview> permission request API. The request will be:
+  // - Allowed if the embedder explicitly allowed it.
+  // - Denied if the embedder explicitly denied.
+  // - Determined by the guest's content settings if the embedder does not
+  // perform an explicit action.
+  // If access was blocked due to the page's content settings,
+  // |blocked_by_policy| should be true, and this function should invoke
+  // OnContentBlocked.
+  void FileSystemAccessedAsync(int render_process_id,
+                               int render_frame_id,
+                               int request_id,
+                               const GURL& url,
+                               bool blocked_by_policy);
+
+  // Called when file system access is requested by the guest content using the
+  // synchronous HTML5 file system API in a worker thread or shared worker. The
+  // request is plumbed through the <webview> permission request API. The
+  // request will be:
+  // - Allowed if the embedder explicitly allowed it.
+  // - Denied if the embedder explicitly denied.
+  // - Determined by the guest's content settings if the embedder does not
+  // perform an explicit action.
+  // If access was blocked due to the page's content settings,
+  // |blocked_by_policy| should be true, and this function should invoke
+  // OnContentBlocked.
+  void FileSystemAccessedSync(int render_process_id,
+                              int render_frame_id,
+                              const GURL& url,
+                              bool blocked_by_policy,
+                              IPC::Message* reply_msg);
+
+  enum PermissionResponseAction { DENY, ALLOW, DEFAULT };
+
+  enum SetPermissionResult {
+    SET_PERMISSION_INVALID,
+    SET_PERMISSION_ALLOWED,
+    SET_PERMISSION_DENIED
+  };
+
+  // Responds to the permission request |request_id| with |action| and
+  // |user_input|. Returns whether there was a pending request for the provided
+  // |request_id|.
+  SetPermissionResult SetPermission(int request_id,
+                                    PermissionResponseAction action,
+                                    const std::string& user_input);
+
+ private:
+#if defined(ENABLE_PLUGINS)
+  friend class content::WebContentsUserData<WebViewPermissionHelper>;
+
+  // content::WebContentsObserver implementation.
+  virtual bool OnMessageReceived(
+      const IPC::Message& message,
+      content::RenderFrameHost* render_frame_host) OVERRIDE;
+  virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
+
+  // Message handlers:
+  void OnBlockedUnauthorizedPlugin(const base::string16& name,
+                                   const std::string& identifier);
+  void OnCouldNotLoadPlugin(const base::FilePath& plugin_path);
+  void OnBlockedOutdatedPlugin(int placeholder_id,
+                               const std::string& identifier);
+  void OnNPAPINotSupported(const std::string& identifier);
+  void OnOpenAboutPlugins();
+#if defined(ENABLE_PLUGIN_INSTALLATION)
+  void OnFindMissingPlugin(int placeholder_id, const std::string& mime_type);
+
+  void OnRemovePluginPlaceholderHost(int placeholder_id);
+#endif  // defined(ENABLE_PLUGIN_INSTALLATION)
+
+  void OnPermissionResponse(const std::string& identifier,
+                            bool allow,
+                            const std::string& user_input);
+#endif  // defiend(ENABLE_PLUGINS)
+
+  static void RecordUserInitiatedUMA(const PermissionResponseInfo& info,
+                                     bool allow);

[Fady Samuel, 2014/06/20 22:54:33]

Space

[Xi Han, 2014/06/24 13:55:33]

Done.

+  void OnWebViewGeolocationPermissionResponse(
+      int bridge_id,
+      bool user_gesture,
+      const base::Callback<void(bool)>& callback,
+      bool allow,
+      const std::string& user_input);
+
+  void OnWebViewFileSystemPermissionResponse(
+      const base::Callback<void(bool)>& callback,
+      bool allow,
+      const std::string& user_input);
+
+  void OnWebViewMediaPermissionResponse(
+      const content::MediaStreamRequest& request,
+      const content::MediaResponseCallback& callback,
+      bool allow,
+      const std::string& user_input);
+
+  void OnWebViewDownloadPermissionResponse(
+      const base::Callback<void(bool)>& callback,
+      bool allow,
+      const std::string& user_input);
+
+  void OnWebViewPointerLockPermissionResponse(
+      const base::Callback<void(bool)>& callback,
+      bool allow,
+      const std::string& user_input);
+
+  // Bridge IDs correspond to a geolocation request. This method will remove
+  // the bookkeeping for a particular geolocation request associated with the
+  // provided |bridge_id|. It returns the request ID of the geolocation request.
+  int RemoveBridgeID(int bridge_id);
+
+  void FileSystemAccessedAsyncResponse(int render_process_id,
+                                       int render_frame_id,
+                                       int request_id,
+                                       const GURL& url,
+                                       bool allowed);
+
+  void FileSystemAccessedSyncResponse(int render_process_id,
+                                      int render_frame_id,
+                                      const GURL& url,
+                                      IPC::Message* reply_msg,
+                                      bool allowed);
+
+  base::WeakPtrFactory<WebViewPermissionHelper> weak_factory_;
+
+  // A counter to generate a unique request id for a permission request.
+  // We only need the ids to be unique for a given WebViewGuest.
+  int next_permission_request_id_;
+
+  WebViewPermissionHelper::RequestMap pending_permission_requests_;
+
+  std::map<int, int> bridge_id_to_request_id_map_;
+
+  WebViewGuest* const web_view_guest_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebViewPermissionHelper);
+};
+
+#endif  // CHROME_BROWSER_GUEST_VIEW_WEB_VIEW_WEB_VIEW_PERMISSION_HELPER_H_