extensions/browser/guest_view/web_view/web_view_renderer_state.cc

   1 // Copyright 2014 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 "content/public/browser/browser_thread.h"
   6 #include "extensions/browser/guest_view/web_view/web_view_renderer_state.h"
   7
   8 using content::BrowserThread;
   9
  10 namespace extensions {
  11
  12 // static
  13 WebViewRendererState* WebViewRendererState::GetInstance() {
  14   return Singleton<WebViewRendererState>::get();
  15 }
  16
  17 WebViewRendererState::WebViewRendererState() {
  18 }
  19
  20 WebViewRendererState::~WebViewRendererState() {
  21 }
  22
  23 bool WebViewRendererState::IsGuest(int render_process_id) {
  24   DCHECK_CURRENTLY_ON(BrowserThread::IO);
  25   return web_view_partition_id_map_.find(render_process_id) !=
  26          web_view_partition_id_map_.end();
  27 }
  28
  29 void WebViewRendererState::AddGuest(int guest_process_id,
  30                                     int guest_routing_id,
  31                                     const WebViewInfo& web_view_info) {
  32   DCHECK_CURRENTLY_ON(BrowserThread::IO);
  33   RenderId render_id(guest_process_id, guest_routing_id);
  34   bool updating =
  35       web_view_info_map_.find(render_id) != web_view_info_map_.end();
  36   web_view_info_map_[render_id] = web_view_info;
  37   if (updating)
  38     return;
  39
  40   auto iter = web_view_partition_id_map_.find(guest_process_id);
  41   if (iter != web_view_partition_id_map_.end()) {
  42     ++iter->second.web_view_count;
  43     return;
  44   }
  45   WebViewPartitionInfo partition_info(1, web_view_info.partition_id);
  46   web_view_partition_id_map_[guest_process_id] = partition_info;
  47 }
  48
  49 void WebViewRendererState::RemoveGuest(int guest_process_id,
  50                                        int guest_routing_id) {
  51   DCHECK_CURRENTLY_ON(BrowserThread::IO);
  52   RenderId render_id(guest_process_id, guest_routing_id);
  53   web_view_info_map_.erase(render_id);
  54   auto iter = web_view_partition_id_map_.find(guest_process_id);
  55   if (iter != web_view_partition_id_map_.end() &&
  56       iter->second.web_view_count > 1) {
  57     --iter->second.web_view_count;
  58     return;
  59   }
  60   web_view_partition_id_map_.erase(guest_process_id);
  61 }
  62
  63 bool WebViewRendererState::GetInfo(int guest_process_id,
  64                                    int guest_routing_id,
  65                                    WebViewInfo* web_view_info) {
  66   DCHECK_CURRENTLY_ON(BrowserThread::IO);
  67   RenderId render_id(guest_process_id, guest_routing_id);
  68   auto iter = web_view_info_map_.find(render_id);
  69   if (iter != web_view_info_map_.end()) {
  70     *web_view_info = iter->second;
  71     return true;
  72   }
  73   return false;
  74 }
  75
  76 bool WebViewRendererState::GetOwnerInfo(int guest_process_id,
  77                                         int* owner_process_id,
  78                                         std::string* owner_extension_id) const {
  79   DCHECK_CURRENTLY_ON(BrowserThread::IO);
  80   // TODO(fsamuel): Store per-process info in WebViewPartitionInfo instead of in
  81   // WebViewInfo.
  82   for (const auto& info : web_view_info_map_) {
  83     if (info.first.first == guest_process_id) {
  84       if (owner_process_id)
  85         *owner_process_id = info.second.embedder_process_id;
  86       if (owner_extension_id)
  87         *owner_extension_id = info.second.owner_extension_id;
  88       return true;
  89     }
  90   }
  91   return false;
  92 }
  93
  94 bool WebViewRendererState::GetPartitionID(int guest_process_id,
  95                                           std::string* partition_id) {
  96   DCHECK_CURRENTLY_ON(BrowserThread::IO);
  97
  98   auto iter = web_view_partition_id_map_.find(guest_process_id);
  99   if (iter != web_view_partition_id_map_.end()){
 100     *partition_id = iter->second.partition_id;
 101     return true;
 102   }
 103   return false;
 104 }
 105
 106 }  // namespace extensions