OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef EXTENSIONS_RENDERER_SCRIPT_INJECTION_MANAGER_H_ | 5 #ifndef EXTENSIONS_RENDERER_SCRIPT_INJECTION_MANAGER_H_ |
6 #define EXTENSIONS_RENDERER_SCRIPT_INJECTION_MANAGER_H_ | 6 #define EXTENSIONS_RENDERER_SCRIPT_INJECTION_MANAGER_H_ |
7 | 7 |
8 #include <stdint.h> | 8 #include <stdint.h> |
9 | 9 |
10 #include <map> | 10 #include <map> |
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
46 | 46 |
47 private: | 47 private: |
48 // A RenderFrameObserver implementation which watches the various render | 48 // A RenderFrameObserver implementation which watches the various render |
49 // frames in order to notify the ScriptInjectionManager of different | 49 // frames in order to notify the ScriptInjectionManager of different |
50 // document load states and IPCs. | 50 // document load states and IPCs. |
51 class RFOHelper; | 51 class RFOHelper; |
52 | 52 |
53 using FrameStatusMap = | 53 using FrameStatusMap = |
54 std::map<content::RenderFrame*, UserScript::RunLocation>; | 54 std::map<content::RenderFrame*, UserScript::RunLocation>; |
55 | 55 |
56 using ScriptInjectionVector = std::vector<scoped_ptr<ScriptInjection>>; | 56 using ScriptInjectionVector = std::vector<std::unique_ptr<ScriptInjection>>; |
57 | 57 |
58 // Notifies that an injection has been finished. | 58 // Notifies that an injection has been finished. |
59 void OnInjectionFinished(ScriptInjection* injection); | 59 void OnInjectionFinished(ScriptInjection* injection); |
60 | 60 |
61 // UserScriptSetManager::Observer implementation. | 61 // UserScriptSetManager::Observer implementation. |
62 void OnUserScriptsUpdated(const std::set<HostID>& changed_hosts, | 62 void OnUserScriptsUpdated(const std::set<HostID>& changed_hosts, |
63 const std::vector<UserScript*>& scripts) override; | 63 const std::vector<UserScript*>& scripts) override; |
64 | 64 |
65 // Notifies that an RFOHelper should be removed. | 65 // Notifies that an RFOHelper should be removed. |
66 void RemoveObserver(RFOHelper* helper); | 66 void RemoveObserver(RFOHelper* helper); |
67 | 67 |
68 // Invalidate any pending tasks associated with |frame|. | 68 // Invalidate any pending tasks associated with |frame|. |
69 void InvalidateForFrame(content::RenderFrame* frame); | 69 void InvalidateForFrame(content::RenderFrame* frame); |
70 | 70 |
71 // Starts the process to inject appropriate scripts into |frame|. | 71 // Starts the process to inject appropriate scripts into |frame|. |
72 void StartInjectScripts(content::RenderFrame* frame, | 72 void StartInjectScripts(content::RenderFrame* frame, |
73 UserScript::RunLocation run_location); | 73 UserScript::RunLocation run_location); |
74 | 74 |
75 // Actually injects the scripts into |frame|. | 75 // Actually injects the scripts into |frame|. |
76 void InjectScripts(content::RenderFrame* frame, | 76 void InjectScripts(content::RenderFrame* frame, |
77 UserScript::RunLocation run_location); | 77 UserScript::RunLocation run_location); |
78 | 78 |
79 // Try to inject and store injection if it has not finished. | 79 // Try to inject and store injection if it has not finished. |
80 void TryToInject(scoped_ptr<ScriptInjection> injection, | 80 void TryToInject(std::unique_ptr<ScriptInjection> injection, |
81 UserScript::RunLocation run_location, | 81 UserScript::RunLocation run_location, |
82 ScriptsRunInfo* scripts_run_info); | 82 ScriptsRunInfo* scripts_run_info); |
83 | 83 |
84 // Handle the ExecuteCode extension message. | 84 // Handle the ExecuteCode extension message. |
85 void HandleExecuteCode(const ExtensionMsg_ExecuteCode_Params& params, | 85 void HandleExecuteCode(const ExtensionMsg_ExecuteCode_Params& params, |
86 content::RenderFrame* render_frame); | 86 content::RenderFrame* render_frame); |
87 | 87 |
88 // Handle the ExecuteDeclarativeScript extension message. | 88 // Handle the ExecuteDeclarativeScript extension message. |
89 void HandleExecuteDeclarativeScript(content::RenderFrame* web_frame, | 89 void HandleExecuteDeclarativeScript(content::RenderFrame* web_frame, |
90 int tab_id, | 90 int tab_id, |
91 const ExtensionId& extension_id, | 91 const ExtensionId& extension_id, |
92 int script_id, | 92 int script_id, |
93 const GURL& url); | 93 const GURL& url); |
94 | 94 |
95 // Handle the GrantInjectionPermission extension message. | 95 // Handle the GrantInjectionPermission extension message. |
96 void HandlePermitScriptInjection(int64_t request_id); | 96 void HandlePermitScriptInjection(int64_t request_id); |
97 | 97 |
98 // The map of active web frames to their corresponding statuses. The | 98 // The map of active web frames to their corresponding statuses. The |
99 // RunLocation of the frame corresponds to the last location that has ran. | 99 // RunLocation of the frame corresponds to the last location that has ran. |
100 FrameStatusMap frame_statuses_; | 100 FrameStatusMap frame_statuses_; |
101 | 101 |
102 // The frames currently being injected into, so long as that frame is valid. | 102 // The frames currently being injected into, so long as that frame is valid. |
103 std::set<content::RenderFrame*> active_injection_frames_; | 103 std::set<content::RenderFrame*> active_injection_frames_; |
104 | 104 |
105 // The collection of RFOHelpers. | 105 // The collection of RFOHelpers. |
106 std::vector<scoped_ptr<RFOHelper>> rfo_helpers_; | 106 std::vector<std::unique_ptr<RFOHelper>> rfo_helpers_; |
107 | 107 |
108 // The set of UserScripts associated with extensions. Owned by the Dispatcher. | 108 // The set of UserScripts associated with extensions. Owned by the Dispatcher. |
109 UserScriptSetManager* user_script_set_manager_; | 109 UserScriptSetManager* user_script_set_manager_; |
110 | 110 |
111 // Pending injections which are waiting for either the proper run location or | 111 // Pending injections which are waiting for either the proper run location or |
112 // user consent. | 112 // user consent. |
113 ScriptInjectionVector pending_injections_; | 113 ScriptInjectionVector pending_injections_; |
114 | 114 |
115 // Running injections which are waiting for async callbacks from blink. | 115 // Running injections which are waiting for async callbacks from blink. |
116 ScriptInjectionVector running_injections_; | 116 ScriptInjectionVector running_injections_; |
117 | 117 |
118 ScopedObserver<UserScriptSetManager, UserScriptSetManager::Observer> | 118 ScopedObserver<UserScriptSetManager, UserScriptSetManager::Observer> |
119 user_script_set_manager_observer_; | 119 user_script_set_manager_observer_; |
120 | 120 |
121 DISALLOW_COPY_AND_ASSIGN(ScriptInjectionManager); | 121 DISALLOW_COPY_AND_ASSIGN(ScriptInjectionManager); |
122 }; | 122 }; |
123 | 123 |
124 } // namespace extensions | 124 } // namespace extensions |
125 | 125 |
126 #endif // EXTENSIONS_RENDERER_SCRIPT_INJECTION_MANAGER_H_ | 126 #endif // EXTENSIONS_RENDERER_SCRIPT_INJECTION_MANAGER_H_ |
OLD | NEW |