Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(4151)

Unified Diff: chrome/browser/guest_view/guest_view_manager.cc

Issue 298913003: Do not allow GuestViewManager to (re)use an instance ID that was already removed. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: address nits Created 6 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: chrome/browser/guest_view/guest_view_manager.cc
diff --git a/chrome/browser/guest_view/guest_view_manager.cc b/chrome/browser/guest_view/guest_view_manager.cc
index 833123f4be4954d9a877688aa32d7e7293d5daf3..782ced5e1eb09d7c5f63501e950c4fc31cdf4d15 100644
--- a/chrome/browser/guest_view/guest_view_manager.cc
+++ b/chrome/browser/guest_view/guest_view_manager.cc
@@ -60,8 +60,8 @@ class GuestWebContentsObserver
};
GuestViewManager::GuestViewManager(content::BrowserContext* context)
- : current_instance_id_(0),
- context_(context) {}
+ : current_instance_id_(0), last_instance_id_removed_(0), context_(context) {
+}
GuestViewManager::~GuestViewManager() {}
@@ -190,8 +190,8 @@ bool GuestViewManager::ForEachGuest(WebContents* embedder_web_contents,
void GuestViewManager::AddGuest(int guest_instance_id,
WebContents* guest_web_contents) {
- DCHECK(guest_web_contents_by_instance_id_.find(guest_instance_id) ==
- guest_web_contents_by_instance_id_.end());
+ CHECK(!ContainsKey(guest_web_contents_by_instance_id_, guest_instance_id));
+ CHECK(CanUseGuestInstanceID(guest_instance_id));
guest_web_contents_by_instance_id_[guest_instance_id] = guest_web_contents;
// This will add the RenderProcessHost ID when we get one.
new GuestWebContentsObserver(guest_web_contents);
@@ -204,6 +204,29 @@ void GuestViewManager::RemoveGuest(int guest_instance_id) {
render_process_host_id_multiset_.erase(
it->second->GetRenderProcessHost()->GetID());
guest_web_contents_by_instance_id_.erase(it);
+
+ // All the instance IDs that lie within [0, last_instance_id_removed_]
+ // are invalid.
+ // The remaining sparse invalid IDs are kept in |removed_instance_ids_| set.
+ // The following code compacts the set by incrementing
+ // |last_instance_id_removed_|.
+ if (guest_instance_id == last_instance_id_removed_ + 1) {
+ ++last_instance_id_removed_;
+ // Compact.
+ std::set<int>::iterator iter = removed_instance_ids_.begin();
+ while (iter != removed_instance_ids_.end()) {
+ int instance_id = *iter;
+ // The sparse invalid IDs must not lie within
+ // [0, last_instance_id_removed_]
+ DCHECK(instance_id > last_instance_id_removed_);
+ if (instance_id != last_instance_id_removed_ + 1)
+ break;
+ ++last_instance_id_removed_;
+ removed_instance_ids_.erase(iter++);
+ }
+ } else {
+ removed_instance_ids_.insert(guest_instance_id);
+ }
}
void GuestViewManager::AddRenderProcessHostID(int render_process_host_id) {
@@ -237,6 +260,12 @@ bool GuestViewManager::CanEmbedderAccessInstanceIDMaybeKill(
return true;
}
+bool GuestViewManager::CanUseGuestInstanceID(int guest_instance_id) {
+ if (guest_instance_id <= last_instance_id_removed_)
+ return false;
+ return !ContainsKey(removed_instance_ids_, guest_instance_id);
+}
+
bool GuestViewManager::CanEmbedderAccessInstanceID(
int embedder_render_process_id,
int guest_instance_id) {
« no previous file with comments | « chrome/browser/guest_view/guest_view_manager.h ('k') | chrome/browser/guest_view/guest_view_manager_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698