| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/extensions/extension_info_map.h" | 5 #include "chrome/browser/extensions/extension_info_map.h" |
| 6 | 6 |
| 7 #include "chrome/common/extensions/extension.h" | 7 #include "chrome/common/extensions/extension.h" |
| 8 #include "chrome/common/extensions/extension_set.h" | 8 #include "chrome/common/extensions/extension_set.h" |
| 9 #include "chrome/common/url_constants.h" | 9 #include "chrome/common/url_constants.h" |
| 10 #include "content/public/browser/browser_thread.h" | 10 #include "content/public/browser/browser_thread.h" |
| (...skipping 89 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 100 return false; | 100 return false; |
| 101 } | 101 } |
| 102 | 102 |
| 103 bool ExtensionInfoMap::CanCrossIncognito(const Extension* extension) { | 103 bool ExtensionInfoMap::CanCrossIncognito(const Extension* extension) { |
| 104 // This is duplicated from ExtensionService :(. | 104 // This is duplicated from ExtensionService :(. |
| 105 return IsIncognitoEnabled(extension->id()) && | 105 return IsIncognitoEnabled(extension->id()) && |
| 106 !extension->incognito_split_mode(); | 106 !extension->incognito_split_mode(); |
| 107 } | 107 } |
| 108 | 108 |
| 109 void ExtensionInfoMap::RegisterExtensionProcess(const std::string& extension_id, | 109 void ExtensionInfoMap::RegisterExtensionProcess(const std::string& extension_id, |
| 110 int process_id) { | 110 int process_id, |
| 111 if (!process_map_.Insert(extension_id, process_id)) { | 111 int site_instance_id) { |
| 112 if (!process_map_.Insert(extension_id, process_id, site_instance_id)) { |
| 112 NOTREACHED() << "Duplicate extension process registration for: " | 113 NOTREACHED() << "Duplicate extension process registration for: " |
| 113 << extension_id << "," << process_id << "."; | 114 << extension_id << "," << process_id << "."; |
| 114 } | 115 } |
| 115 } | 116 } |
| 116 | 117 |
| 117 void ExtensionInfoMap::UnregisterExtensionProcess( | 118 void ExtensionInfoMap::UnregisterExtensionProcess( |
| 118 const std::string& extension_id, | 119 const std::string& extension_id, |
| 119 int process_id) { | 120 int process_id, |
| 120 if (!process_map_.Remove(extension_id, process_id)) { | 121 int site_instance_id) { |
| 122 if (!process_map_.Remove(extension_id, process_id, site_instance_id)) { |
| 121 NOTREACHED() << "Unknown extension process registration for: " | 123 NOTREACHED() << "Unknown extension process registration for: " |
| 122 << extension_id << "," << process_id << "."; | 124 << extension_id << "," << process_id << "."; |
| 123 } | 125 } |
| 124 } | 126 } |
| 125 | 127 |
| 126 void ExtensionInfoMap::UnregisterAllExtensionsInProcess(int process_id) { | 128 void ExtensionInfoMap::UnregisterAllExtensionsInProcess(int process_id) { |
| 127 process_map_.Remove(process_id); | 129 process_map_.RemoveAllFromProcess(process_id); |
| 128 } | 130 } |
| 129 | 131 |
| 130 bool ExtensionInfoMap::SecurityOriginHasAPIPermission( | 132 bool ExtensionInfoMap::SecurityOriginHasAPIPermission( |
| 131 const GURL& origin, int process_id, | 133 const GURL& origin, int process_id, |
| 132 ExtensionAPIPermission::ID permission) const { | 134 ExtensionAPIPermission::ID permission) const { |
| 133 if (origin.SchemeIs(chrome::kExtensionScheme)) { | 135 if (origin.SchemeIs(chrome::kExtensionScheme)) { |
| 134 const std::string& id = origin.host(); | 136 const std::string& id = origin.host(); |
| 135 return extensions_.GetByID(id)->HasAPIPermission(permission) && | 137 return extensions_.GetByID(id)->HasAPIPermission(permission) && |
| 136 process_map_.Contains(id, process_id); | 138 process_map_.Contains(id, process_id); |
| 137 } | 139 } |
| 138 | 140 |
| 139 ExtensionSet::ExtensionMap::const_iterator i = extensions_.begin(); | 141 ExtensionSet::ExtensionMap::const_iterator i = extensions_.begin(); |
| 140 for (; i != extensions_.end(); ++i) { | 142 for (; i != extensions_.end(); ++i) { |
| 141 if (i->second->web_extent().MatchesSecurityOrigin(origin) && | 143 if (i->second->web_extent().MatchesSecurityOrigin(origin) && |
| 142 process_map_.Contains(i->first, process_id) && | 144 process_map_.Contains(i->first, process_id) && |
| 143 i->second->HasAPIPermission(permission)) { | 145 i->second->HasAPIPermission(permission)) { |
| 144 return true; | 146 return true; |
| 145 } | 147 } |
| 146 } | 148 } |
| 147 return false; | 149 return false; |
| 148 } | 150 } |
| OLD | NEW |