Allow tabCapture API to be granted for chrome:// pages.

chrome/browser/extensions/active_tab_permission_granter.cc

 // Copyright (c) 2012 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 "chrome/browser/extensions/active_tab_permission_granter.h"
 
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_system.h"
 #include "chrome/browser/profiles/profile.h"
@@ skipping 19 matching lines @@
     content::WebContents* web_contents, int tab_id, Profile* profile)
     : WebContentsObserver(web_contents), tab_id_(tab_id) {
   registrar_.Add(this,
                  chrome::NOTIFICATION_EXTENSION_UNLOADED,
                  content::Source<Profile>(profile));
 }
 
 ActiveTabPermissionGranter::~ActiveTabPermissionGranter() {}
 
 void ActiveTabPermissionGranter::GrantIfRequested(const Extension* extension) {
-  if (!extension->HasAPIPermission(extensions::APIPermission::kActiveTab) &&
-      !extension->HasAPIPermission(extensions::APIPermission::kTabCapture)) {
+  if (granted_extensions_.Contains(extension->id()))
     return;
+
+  APIPermissionSet new_apis;
+  URLPatternSet new_hosts;
+
+  if (extension->HasAPIPermission(APIPermission::kActiveTab)) {
+    URLPattern pattern(UserScript::ValidUserScriptSchemes());
+    // Pattern parsing could fail if this is an unsupported URL e.g. chrome://.
+    if (pattern.Parse(web_contents()->GetURL().spec()) ==
+            URLPattern::PARSE_SUCCESS) {
+      new_hosts.AddPattern(pattern);
+      new_apis.insert(APIPermission::kTab);
+      granted_extensions_.Insert(extension);
+    }
   }
 
-  if (IsGranted(extension))
-    return;
+  if (extension->HasAPIPermission(APIPermission::kTabCapture))
+    new_apis.insert(APIPermission::kTabCaptureForTab);
 
-  URLPattern pattern(UserScript::ValidUserScriptSchemes());
-  if (pattern.Parse(web_contents()->GetURL().spec()) !=
-          URLPattern::PARSE_SUCCESS) {
-    // Pattern parsing could fail if this is an unsupported URL e.g. chrome://.
-    return;
+  if (!new_apis.empty()) {
+    scoped_refptr<const PermissionSet> new_permissions =
+        new PermissionSet(new_apis, new_hosts, URLPatternSet());
+    PermissionsData::UpdateTabSpecificPermissions(extension,
+                                                  tab_id_,
+                                                  new_permissions);
+    Send(new ExtensionMsg_UpdateTabSpecificPermissions(GetPageID(),
+                                                       tab_id_,
+                                                       extension->id(),
+                                                       new_hosts));
   }
-
-  APIPermissionSet new_apis;
-  new_apis.insert(APIPermission::kTab);
-  URLPatternSet new_hosts;
-  new_hosts.AddPattern(pattern);
-  scoped_refptr<const PermissionSet> new_permissions =
-      new PermissionSet(new_apis, new_hosts, URLPatternSet());
-
-  PermissionsData::UpdateTabSpecificPermissions(extension,
-                                                tab_id_,
-                                                new_permissions);
-  granted_extensions_.Insert(extension);
-  Send(new ExtensionMsg_UpdateTabSpecificPermissions(GetPageID(),
-                                                     tab_id_,
-                                                     extension->id(),
-                                                     new_hosts));
-}
-
-bool ActiveTabPermissionGranter::IsGranted(const Extension* extension) {
-  return granted_extensions_.Contains(extension->id());
 }
 
 void ActiveTabPermissionGranter::DidNavigateMainFrame(
     const content::LoadCommittedDetails& details,
     const content::FrameNavigateParams& params) {
   if (details.is_in_page)
     return;
   DCHECK(details.is_main_frame);  // important: sub-frames don't get granted!
   ClearActiveExtensionsAndNotify();
 }
@@ skipping 29 matching lines @@
 
   Send(new ExtensionMsg_ClearTabSpecificPermissions(tab_id_, extension_ids));
   granted_extensions_.Clear();
 }
 
 int32 ActiveTabPermissionGranter::GetPageID() {
   return web_contents()->GetController().GetVisibleEntry()->GetPageID();
 }
 
 }  // namespace extensions