1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// WebViewRendererState manages state data for WebView guest renderer processes.
//
// This class's data can be accessed via its methods from both the UI and IO
// threads, and uses locks to mediate this access. When making changes to this
// class, ensure that you avoid introducing any reentrant code in the methods,
// and that you always acquire the locks in the order `web_view_info_map_lock_`
// -> `web_view_partition_id_map_lock_` (if both are needed in one method).
#ifndef EXTENSIONS_BROWSER_GUEST_VIEW_WEB_VIEW_WEB_VIEW_RENDERER_STATE_H_
#define EXTENSIONS_BROWSER_GUEST_VIEW_WEB_VIEW_WEB_VIEW_RENDERER_STATE_H_
#include <map>
#include <set>
#include <string>
#include <utility>
#include "base/memory/singleton.h"
#include "content/public/browser/child_process_id.h"
#include "content/public/browser/global_routing_id.h"
namespace extensions {
class WebViewGuest;
class WebViewRendererState {
public:
struct WebViewInfo {
content::ChildProcessId embedder_process_id;
int instance_id;
int rules_registry_id;
std::string partition_id;
std::string owner_host;
std::set<std::string> content_script_ids;
WebViewInfo();
WebViewInfo(const WebViewInfo& other);
~WebViewInfo();
};
WebViewRendererState(const WebViewRendererState&) = delete;
WebViewRendererState& operator=(const WebViewRendererState&) = delete;
static WebViewRendererState* GetInstance();
// Looks up the information for the embedder WebView for a RenderViewHost,
// given its process and view ID. Returns true and writes the information to
// `web_view_info` if found, otherwise returns false.
bool GetInfo(int guest_process_id,
int guest_routing_id,
WebViewInfo* web_view_info) const;
// Looks up the information for the owner of a WebView guest process, given
// its process ID. Returns true and writes the info to `owner_process_id` and
// `owner_host` if found, otherwise returns false.
bool GetOwnerInfo(int guest_process_id,
int* owner_process_id,
std::string* owner_host) const;
// Looks up the partition ID for a WebView guest process, given its
// process ID. Returns true and writes the partition ID to `partition_id` if
// found, otherwise returns false.
bool GetPartitionID(int guest_process_id, std::string* partition_id) const;
// Returns true if the renderer with process ID `render_process_id` is a
// WebView guest process.
bool IsGuest(int render_process_id) const;
void AddContentScriptIDs(int embedder_process_id,
int view_instance_id,
const std::set<std::string>& script_ids);
void RemoveContentScriptIDs(int embedder_process_id,
int view_instance_id,
const std::set<std::string>& script_ids);
// Returns the number of WebView guest instances tracked by this object.
size_t guest_count_for_testing() { return web_view_info_map_.size(); }
private:
friend class WebViewGuest;
friend struct base::DefaultSingletonTraits<WebViewRendererState>;
using GlobalRenderFrameHostId = content::GlobalRenderFrameHostId;
using WebViewInfoMap = std::map<GlobalRenderFrameHostId, WebViewInfo>;
struct WebViewPartitionInfo {
int web_view_frame_count;
std::string partition_id;
WebViewPartitionInfo() {}
WebViewPartitionInfo(int count, const std::string& partition)
: web_view_frame_count(count), partition_id(partition) {}
};
using WebViewPartitionIDMap = std::map<int, WebViewPartitionInfo>;
WebViewRendererState();
~WebViewRendererState();
// Adds/removes a WebView guest render process to/from the set.
void AddGuest(int render_process_host_id, int routing_id,
const WebViewInfo& web_view_info);
void RemoveGuest(int render_process_host_id, int routing_id);
// Maps WebView guest frames (identified by process ID and RenderFrameHost
// routing ID pairs) to WebViewInfo.
WebViewInfoMap web_view_info_map_;
// Maps each WebView guest renderer process ID to its partition ID.
WebViewPartitionIDMap web_view_partition_id_map_;
// Locks are used to mediate access to the maps above from both the UI and IO
// threads.
mutable base::Lock web_view_info_map_lock_;
mutable base::Lock web_view_partition_id_map_lock_;
};
} // namespace extensions
#endif // EXTENSIONS_BROWSER_GUEST_VIEW_WEB_VIEW_WEB_VIEW_RENDERER_STATE_H_
|
