Track which NativeViewIds each render process is allowed to interact with.

chrome/browser/renderer_host/renderer_security_policy.cc

 // Copyright (c) 2006-2008 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/renderer_host/renderer_security_policy.h"
 
 #include "base/file_path.h"
 #include "base/logging.h"
 #include "base/string_util.h"
 #include "chrome/common/url_constants.h"
@@ skipping 18 matching lines @@
 
   // Grant permission to upload the specified file to the web.
   void GrantUploadFile(const FilePath& file) {
     uploadable_files_.insert(file);
   }
 
   void GrantDOMUIBindings() {
     has_dom_ui_bindings_ = true;
   }
 
+  void GrantNativeViewId(gfx::NativeViewId view_id) {
+    native_view_ids_.insert(view_id);
+  }
+
   // Determine whether permission has been granted to request url.
   // Schemes that have not been granted default to being denied.
   bool CanRequestURL(const GURL& url) {
     SchemeMap::const_iterator judgment(scheme_policy_.find(url.scheme()));
 
     if (judgment == scheme_policy_.end())
       return false;  // Unmentioned schemes are disallowed.
 
     return judgment->second;
   }
 
   // Determine whether permission has been granted to upload file.
   // Files that have not been granted default to being denied.
   bool CanUploadFile(const FilePath& file) {
     return uploadable_files_.find(file) != uploadable_files_.end();
   }
 
   bool has_dom_ui_bindings() const { return has_dom_ui_bindings_; }
 
+  bool HasNativeViewId(gfx::NativeViewId view_id) {
+    return native_view_ids_.find(view_id) != native_view_ids_.end();
+  }
+
  private:
   typedef std::map<std::string, bool> SchemeMap;
   typedef std::set<FilePath> FileSet;
+  typedef std::set<gfx::NativeViewId> NativeViewIdSet;
 
   // Maps URL schemes to whether permission has been granted or revoked:
   //   |true| means the scheme has been granted.
   //   |false| means the scheme has been revoked.
   // If a scheme is not present in the map, then it has never been granted
   // or revoked.
   SchemeMap scheme_policy_;
 
   // The set of files the renderer is permited to upload to the web.
   FileSet uploadable_files_;
 
+  // The set of gfx::NativeViewIds the renderer is permitted to interact with.
+  NativeViewIdSet native_view_ids_;
+
   bool has_dom_ui_bindings_;
 
   DISALLOW_COPY_AND_ASSIGN(SecurityState);
 };
 
 RendererSecurityPolicy::RendererSecurityPolicy() {
   // We know about these schemes and believe them to be safe.
   RegisterWebSafeScheme(chrome::kHttpScheme);
   RegisterWebSafeScheme(chrome::kHttpsScheme);
   RegisterWebSafeScheme(chrome::kFtpScheme);
@@ skipping 128 matching lines @@
 
   state->second->GrantDOMUIBindings();
 
   // DOM UI bindings need the ability to request chrome-ui URLs.
   state->second->GrantScheme(chrome::kChromeUIScheme);
 
   // DOM UI pages can contain links to file:// URLs.
   state->second->GrantScheme(chrome::kFileScheme);
 }
 
+void RendererSecurityPolicy::GrantNativeViewId(int renderer_id,
+                                               gfx::NativeViewId view_id) {
+  AutoLock lock(lock_);
+
+  SecurityStateMap::iterator state = security_state_.find(renderer_id);
+  if (state == security_state_.end())
+    return;
+
+  state->second->GrantNativeViewId(view_id);
+}
+
 bool RendererSecurityPolicy::CanRequestURL(int renderer_id, const GURL& url) {
   if (!url.is_valid())
     return false;  // Can't request invalid URLs.
 
   if (IsWebSafeScheme(url.scheme()))
     return true;  // The scheme has been white-listed for every renderer.
 
   if (IsPseudoScheme(url.scheme())) {
     // There are a number of special cases for pseudo schemes.
 
@@ skipping 41 matching lines @@
 
 bool RendererSecurityPolicy::HasDOMUIBindings(int renderer_id) {
   AutoLock lock(lock_);
 
   SecurityStateMap::iterator state = security_state_.find(renderer_id);
   if (state == security_state_.end())
     return false;
 
   return state->second->has_dom_ui_bindings();
 }
+
+bool RendererSecurityPolicy::HasNativeViewId(int renderer_id,
+                                             gfx::NativeViewId view_id) {
+  AutoLock lock(lock_);
+
+  SecurityStateMap::iterator state = security_state_.find(renderer_id);
+  if (state == security_state_.end())
+    return false;
+
+  return state->second->HasNativeViewId(view_id);
+}