chrome/browser/extensions/active_tab_permission_granter.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/extensions/active_tab_permission_granter.h"
   6
   7 #include "chrome/browser/chrome_notification_types.h"
   8 #include "chrome/browser/extensions/extension_service.h"
   9 #include "chrome/browser/profiles/profile.h"
  10 #include "chrome/browser/sessions/session_id.h"
  11 #include "content/public/browser/navigation_details.h"
  12 #include "content/public/browser/navigation_entry.h"
  13 #include "content/public/browser/notification_service.h"
  14 #include "content/public/browser/notification_source.h"
  15 #include "content/public/browser/render_process_host.h"
  16 #include "content/public/browser/web_contents.h"
  17 #include "extensions/browser/extension_system.h"
  18 #include "extensions/common/extension.h"
  19 #include "extensions/common/extension_messages.h"
  20 #include "extensions/common/permissions/permission_set.h"
  21 #include "extensions/common/permissions/permissions_data.h"
  22 #include "extensions/common/user_script.h"
  23
  24 using content::RenderProcessHost;
  25 using content::WebContentsObserver;
  26
  27 namespace extensions {
  28
  29 ActiveTabPermissionGranter::ActiveTabPermissionGranter(
  30     content::WebContents* web_contents, int tab_id, Profile* profile)
  31     : WebContentsObserver(web_contents), tab_id_(tab_id) {
  32   registrar_.Add(this,
  33                  chrome::NOTIFICATION_EXTENSION_UNLOADED,
  34                  content::Source<Profile>(profile));
  35 }
  36
  37 ActiveTabPermissionGranter::~ActiveTabPermissionGranter() {}
  38
  39 void ActiveTabPermissionGranter::GrantIfRequested(const Extension* extension) {
  40   if (granted_extensions_.Contains(extension->id()))
  41     return;
  42
  43   APIPermissionSet new_apis;
  44   URLPatternSet new_hosts;
  45
  46   if (extension->HasAPIPermission(APIPermission::kActiveTab)) {
  47     URLPattern pattern(UserScript::ValidUserScriptSchemes());
  48     // Pattern parsing could fail if this is an unsupported URL e.g. chrome://.
  49     if (pattern.Parse(web_contents()->GetURL().spec()) ==
  50             URLPattern::PARSE_SUCCESS) {
  51       new_hosts.AddPattern(pattern);
  52     }
  53     new_apis.insert(APIPermission::kTab);
  54   }
  55
  56   if (extension->HasAPIPermission(APIPermission::kTabCapture))
  57     new_apis.insert(APIPermission::kTabCaptureForTab);
  58
  59   if (!new_apis.empty() || !new_hosts.is_empty()) {
  60     granted_extensions_.Insert(extension);
  61     scoped_refptr<const PermissionSet> new_permissions =
  62         new PermissionSet(new_apis, ManifestPermissionSet(),
  63                           new_hosts, URLPatternSet());
  64     PermissionsData::UpdateTabSpecificPermissions(extension,
  65                                                   tab_id_,
  66                                                   new_permissions);
  67     const content::NavigationEntry* navigation_entry =
  68         web_contents()->GetController().GetVisibleEntry();
  69     if (navigation_entry) {
  70       Send(new ExtensionMsg_UpdateTabSpecificPermissions(
  71           navigation_entry->GetPageID(),
  72           tab_id_,
  73           extension->id(),
  74           new_hosts));
  75     }
  76   }
  77 }
  78
  79 void ActiveTabPermissionGranter::DidNavigateMainFrame(
  80     const content::LoadCommittedDetails& details,
  81     const content::FrameNavigateParams& params) {
  82   if (details.is_in_page)
  83     return;
  84   DCHECK(details.is_main_frame);  // important: sub-frames don't get granted!
  85   ClearActiveExtensionsAndNotify();
  86 }
  87
  88 void ActiveTabPermissionGranter::WebContentsDestroyed(
  89     content::WebContents* web_contents) {
  90   ClearActiveExtensionsAndNotify();
  91 }
  92
  93 void ActiveTabPermissionGranter::Observe(
  94     int type,
  95     const content::NotificationSource& source,
  96     const content::NotificationDetails& details) {
  97   DCHECK_EQ(type, chrome::NOTIFICATION_EXTENSION_UNLOADED);
  98   const Extension* extension =
  99       content::Details<UnloadedExtensionInfo>(details)->extension;
 100   // Note: don't need to clear the permissions (nor tell the renderer about it)
 101   // because it's being unloaded anyway.
 102   granted_extensions_.Remove(extension->id());
 103 }
 104
 105 void ActiveTabPermissionGranter::ClearActiveExtensionsAndNotify() {
 106   if (granted_extensions_.is_empty())
 107     return;
 108
 109   std::vector<std::string> extension_ids;
 110
 111   for (ExtensionSet::const_iterator it = granted_extensions_.begin();
 112        it != granted_extensions_.end(); ++it) {
 113     PermissionsData::ClearTabSpecificPermissions(it->get(), tab_id_);
 114     extension_ids.push_back((*it)->id());
 115   }
 116
 117   Send(new ExtensionMsg_ClearTabSpecificPermissions(tab_id_, extension_ids));
 118   granted_extensions_.Clear();
 119 }
 120
 121 }  // namespace extensions