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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
|
// Copyright 2014 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.
#ifndef EXTENSIONS_BROWSER_USER_SCRIPT_LOADER_H_
#define EXTENSIONS_BROWSER_USER_SCRIPT_LOADER_H_
#include <map>
#include <set>
#include "base/compiler_specific.h"
#include "base/memory/scoped_ptr.h"
#include "base/memory/shared_memory.h"
#include "base/memory/weak_ptr.h"
#include "base/scoped_observer.h"
#include "content/public/browser/notification_observer.h"
#include "content/public/browser/notification_registrar.h"
#include "extensions/common/host_id.h"
#include "extensions/common/user_script.h"
namespace base {
class SharedMemory;
}
namespace content {
class BrowserContext;
class RenderProcessHost;
}
namespace extensions {
class ContentVerifier;
// Manages one "logical unit" of user scripts in shared memory by constructing a
// new shared memory region when the set of scripts changes. Also notifies
// renderers of new shared memory region when new renderers appear, or when
// script reloading completes. Script loading lives on the UI thread. Instances
// of this class are embedded within classes with names ending in
// UserScriptMaster. These "master" classes implement the strategy for which
// scripts to load/unload on this logical unit of scripts.
class UserScriptLoader : public content::NotificationObserver {
public:
using PathAndDefaultLocale = std::pair<base::FilePath, std::string>;
using HostsInfo = std::map<HostID, PathAndDefaultLocale>;
using SubstitutionMap = std::map<std::string, std::string>;
using LoadUserScriptsContentFunction =
base::Callback<bool(const HostID&,
UserScript::File*,
const SubstitutionMap*,
const scoped_refptr<ContentVerifier>&)>;
// Parses the includes out of |script| and returns them in |includes|.
static bool ParseMetadataHeader(const base::StringPiece& script_text,
UserScript* script);
UserScriptLoader(content::BrowserContext* browser_context,
const HostID& host_id,
const scoped_refptr<ContentVerifier>& content_verifier);
~UserScriptLoader() override;
// A wrapper around the method to load user scripts, which is normally run on
// the file thread. Exposed only for tests.
void LoadScriptsForTest(UserScriptList* user_scripts);
// Add |scripts| to the set of scripts managed by this loader.
void AddScripts(const std::set<UserScript>& scripts);
// Remove |scripts| from the set of scripts managed by this loader.
void RemoveScripts(const std::set<UserScript>& scripts);
// Clears the set of scripts managed by this loader.
void ClearScripts();
// Initiates procedure to start loading scripts on the file thread.
void StartLoad();
// Returns true if we have any scripts ready.
bool scripts_ready() const { return shared_memory_.get() != NULL; }
protected:
// Updates |hosts_info_| to contain info for each element of
// |changed_hosts_|.
virtual void UpdateHostsInfo(const std::set<HostID>& changed_hosts) = 0;
// Returns a function pointer of a static funcion to load user scripts.
// Derived classes can specify their ways to load scripts in the static
// function they return.
// Note: It has to be safe to call multiple times.
virtual LoadUserScriptsContentFunction GetLoadUserScriptsFunction() = 0;
// Adds the |host_id, location| to the |hosts_info_| map.
// Only inserts the entry to the map when the given host_id doesn't
// exists.
void AddHostInfo(const HostID& host_id, const PathAndDefaultLocale& location);
// Removes the entries with the given host_id from the |hosts_info_| map.
void RemoveHostInfo(const HostID& host_id);
// Sets the flag if the initial set of hosts has finished loading; if it's
// set to be true, calls AttempLoad() to bootstrap.
void SetReady(bool ready);
content::BrowserContext* browser_context() const { return browser_context_; }
const HostID& host_id() const { return host_id_; }
private:
// content::NotificationObserver implementation.
void Observe(int type,
const content::NotificationSource& source,
const content::NotificationDetails& details) override;
// Returns whether or not it is possible that calls to AddScripts(),
// RemoveScripts(), and/or ClearScripts() have caused any real change in the
// set of scripts to be loaded.
bool ScriptsMayHaveChanged() const;
// Attempts to initiate a load.
void AttemptLoad();
// Called once we have finished loading the scripts on the file thread.
void OnScriptsLoaded(scoped_ptr<UserScriptList> user_scripts,
scoped_ptr<base::SharedMemory> shared_memory);
// Sends the renderer process a new set of user scripts. If
// |changed_hosts| is not empty, this signals that only the scripts from
// those hosts should be updated. Otherwise, all hosts will be
// updated.
void SendUpdate(content::RenderProcessHost* process,
base::SharedMemory* shared_memory,
const std::set<HostID>& changed_hosts);
bool is_loading() const {
// Ownership of |user_scripts_| is passed to the file thread when loading.
return user_scripts_.get() == NULL;
}
// Manages our notification registrations.
content::NotificationRegistrar registrar_;
// Contains the scripts that were found the last time scripts were updated.
scoped_ptr<base::SharedMemory> shared_memory_;
// List of scripts from currently-installed extensions we should load.
scoped_ptr<UserScriptList> user_scripts_;
// Maps host info needed for localization to a host ID.
HostsInfo hosts_info_;
// The mutually-exclusive sets of scripts that were added or removed since the
// last script load.
std::set<UserScript> added_scripts_;
std::set<UserScript> removed_scripts_;
// Indicates whether the the collection of scripts should be cleared before
// additions and removals on the next script load.
bool clear_scripts_;
// The IDs of the extensions which changed in the last update sent to the
// renderer.
std::set<HostID> changed_hosts_;
// If the initial set of hosts has finished loading.
bool ready_;
// If list of user scripts is modified while we're loading it, we note
// that we're currently mid-load and then start over again once the load
// finishes. This boolean tracks whether another load is pending.
bool pending_load_;
// Whether or not we are currently loading.
bool is_loading_;
// The browser_context for which the scripts managed here are installed.
content::BrowserContext* browser_context_;
// ID of the host that owns these scripts, if any. This is only set to a
// non-empty value for declarative user script shared memory regions.
HostID host_id_;
// Manages content verification of the loaded user scripts.
scoped_refptr<ContentVerifier> content_verifier_;
base::WeakPtrFactory<UserScriptLoader> weak_factory_;
DISALLOW_COPY_AND_ASSIGN(UserScriptLoader);
};
} // namespace extensions
#endif // EXTENSIONS_BROWSER_USER_SCRIPT_LOADER_H_
|
