src/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/extensions/extension_system.h"
  10 #include "chrome/browser/profiles/profile.h"
  11 #include "chrome/browser/sessions/session_id.h"
  12 #include "chrome/common/extensions/extension.h"
  13 #include "chrome/common/extensions/extension_messages.h"
  14 #include "chrome/common/extensions/permissions/permissions_data.h"
  15 #include "content/public/browser/navigation_details.h"
  16 #include "content/public/browser/navigation_entry.h"
  17 #include "content/public/browser/notification_service.h"
  18 #include "content/public/browser/notification_source.h"
  19 #include "content/public/browser/render_process_host.h"
  20 #include "content/public/browser/web_contents.h"
  21 #include "extensions/common/permissions/permission_set.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, new_hosts, URLPatternSet());
  63     PermissionsData::UpdateTabSpecificPermissions(extension,
  64                                                   tab_id_,
  65                                                   new_permissions);
  66     const content::NavigationEntry* navigation_entry =
  67         web_contents()->GetController().GetVisibleEntry();
  68     if (navigation_entry) {
  69       Send(new ExtensionMsg_UpdateTabSpecificPermissions(
  70           navigation_entry->GetPageID(),
  71           tab_id_,
  72           extension->id(),
  73           new_hosts));
  74     }
  75   }
  76 }
  77
  78 void ActiveTabPermissionGranter::DidNavigateMainFrame(
  79     const content::LoadCommittedDetails& details,
  80     const content::FrameNavigateParams& params) {
  81   if (details.is_in_page)
  82     return;
  83   DCHECK(details.is_main_frame);  // important: sub-frames don't get granted!
  84   ClearActiveExtensionsAndNotify();
  85 }
  86
  87 void ActiveTabPermissionGranter::WebContentsDestroyed(
  88     content::WebContents* web_contents) {
  89   ClearActiveExtensionsAndNotify();
  90 }
  91
  92 void ActiveTabPermissionGranter::Observe(
  93     int type,
  94     const content::NotificationSource& source,
  95     const content::NotificationDetails& details) {
  96   DCHECK_EQ(type, chrome::NOTIFICATION_EXTENSION_UNLOADED);
  97   const Extension* extension =
  98       content::Details<UnloadedExtensionInfo>(details)->extension;
  99   // Note: don't need to clear the permissions (nor tell the renderer about it)
 100   // because it's being unloaded anyway.
 101   granted_extensions_.Remove(extension->id());
 102 }
 103
 104 void ActiveTabPermissionGranter::ClearActiveExtensionsAndNotify() {
 105   if (granted_extensions_.is_empty())
 106     return;
 107
 108   std::vector<std::string> extension_ids;
 109
 110   for (ExtensionSet::const_iterator it = granted_extensions_.begin();
 111        it != granted_extensions_.end(); ++it) {
 112     PermissionsData::ClearTabSpecificPermissions(it->get(), tab_id_);
 113     extension_ids.push_back((*it)->id());
 114   }
 115
 116   Send(new ExtensionMsg_ClearTabSpecificPermissions(tab_id_, extension_ids));
 117   granted_extensions_.Clear();
 118 }
 119
 120 }  // namespace extensions