OLD | NEW |
| (Empty) |
1 // Copyright 2015 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 "chrome/browser/extensions/extension_user_script_loader.h" | |
6 | |
7 #include <set> | |
8 #include <string> | |
9 | |
10 #include "base/bind.h" | |
11 #include "base/bind_helpers.h" | |
12 #include "base/files/file_path.h" | |
13 #include "base/files/file_util.h" | |
14 #include "base/version.h" | |
15 #include "chrome/browser/profiles/profile.h" | |
16 #include "content/public/browser/browser_thread.h" | |
17 #include "content/public/browser/notification_service.h" | |
18 #include "content/public/browser/render_process_host.h" | |
19 #include "extensions/browser/component_extension_resource_manager.h" | |
20 #include "extensions/browser/content_verifier.h" | |
21 #include "extensions/browser/extension_registry.h" | |
22 #include "extensions/browser/extension_system.h" | |
23 #include "extensions/browser/extensions_browser_client.h" | |
24 #include "extensions/common/file_util.h" | |
25 #include "extensions/common/manifest_handlers/default_locale_handler.h" | |
26 #include "extensions/common/message_bundle.h" | |
27 #include "extensions/common/one_shot_event.h" | |
28 #include "ui/base/resource/resource_bundle.h" | |
29 | |
30 namespace extensions { | |
31 | |
32 namespace { | |
33 | |
34 // Verifies file contents as they are read. | |
35 void VerifyContent(const scoped_refptr<ContentVerifier>& verifier, | |
36 const std::string& extension_id, | |
37 const base::FilePath& extension_root, | |
38 const base::FilePath& relative_path, | |
39 const std::string& content) { | |
40 DCHECK_CURRENTLY_ON(content::BrowserThread::IO); | |
41 scoped_refptr<ContentVerifyJob> job( | |
42 verifier->CreateJobFor(extension_id, extension_root, relative_path)); | |
43 if (job.get()) { | |
44 job->Start(); | |
45 job->BytesRead(content.size(), content.data()); | |
46 job->DoneReading(); | |
47 } | |
48 } | |
49 | |
50 // Loads user scripts from the extension who owns these scripts. | |
51 bool ExtensionLoadScriptContent( | |
52 const HostID& host_id, | |
53 UserScript::File* script_file, | |
54 const UserScriptLoader::SubstitutionMap* localization_messages, | |
55 const scoped_refptr<ContentVerifier>& verifier) { | |
56 DCHECK(script_file); | |
57 std::string content; | |
58 const base::FilePath& path = ExtensionResource::GetFilePath( | |
59 script_file->extension_root(), script_file->relative_path(), | |
60 ExtensionResource::SYMLINKS_MUST_RESOLVE_WITHIN_ROOT); | |
61 if (path.empty()) { | |
62 int resource_id = 0; | |
63 if (ExtensionsBrowserClient::Get() | |
64 ->GetComponentExtensionResourceManager() | |
65 ->IsComponentExtensionResource(script_file->extension_root(), | |
66 script_file->relative_path(), | |
67 &resource_id)) { | |
68 const ResourceBundle& rb = ResourceBundle::GetSharedInstance(); | |
69 content = rb.GetRawDataResource(resource_id).as_string(); | |
70 } else { | |
71 LOG(WARNING) << "Failed to get file path to " | |
72 << script_file->relative_path().value() << " from " | |
73 << script_file->extension_root().value(); | |
74 return false; | |
75 } | |
76 } else { | |
77 if (!base::ReadFileToString(path, &content)) { | |
78 LOG(WARNING) << "Failed to load user script file: " << path.value(); | |
79 return false; | |
80 } | |
81 if (verifier.get()) { | |
82 content::BrowserThread::PostTask( | |
83 content::BrowserThread::IO, FROM_HERE, | |
84 base::Bind(&VerifyContent, verifier, host_id.id(), | |
85 script_file->extension_root(), | |
86 script_file->relative_path(), content)); | |
87 } | |
88 } | |
89 | |
90 // Localize the content. | |
91 if (localization_messages) { | |
92 std::string error; | |
93 MessageBundle::ReplaceMessagesWithExternalDictionary(*localization_messages, | |
94 &content, &error); | |
95 if (!error.empty()) | |
96 LOG(WARNING) << "Failed to replace messages in script: " << error; | |
97 } | |
98 | |
99 // Remove BOM from the content. | |
100 std::string::size_type index = content.find(base::kUtf8ByteOrderMark); | |
101 if (index == 0) | |
102 script_file->set_content(content.substr(strlen(base::kUtf8ByteOrderMark))); | |
103 else | |
104 script_file->set_content(content); | |
105 | |
106 return true; | |
107 } | |
108 | |
109 } // namespace | |
110 | |
111 ExtensionUserScriptLoader::ExtensionUserScriptLoader( | |
112 Profile* profile, | |
113 const HostID& host_id, | |
114 bool listen_for_extension_system_loaded) | |
115 : UserScriptLoader(profile, | |
116 host_id, | |
117 ExtensionSystem::Get(profile)->content_verifier()), | |
118 extension_registry_observer_(this), | |
119 weak_factory_(this) { | |
120 extension_registry_observer_.Add(ExtensionRegistry::Get(profile)); | |
121 if (listen_for_extension_system_loaded) { | |
122 ExtensionSystem::Get(profile)->ready().Post( | |
123 FROM_HERE, | |
124 base::Bind(&ExtensionUserScriptLoader::OnExtensionSystemReady, | |
125 weak_factory_.GetWeakPtr())); | |
126 } else { | |
127 SetReady(true); | |
128 } | |
129 } | |
130 | |
131 ExtensionUserScriptLoader::~ExtensionUserScriptLoader() { | |
132 } | |
133 | |
134 void ExtensionUserScriptLoader::UpdateHostsInfo( | |
135 const std::set<HostID>& changed_hosts) { | |
136 ExtensionRegistry* registry = ExtensionRegistry::Get(profile()); | |
137 for (const HostID& host_id : changed_hosts) { | |
138 const Extension* extension = | |
139 registry->GetExtensionById(host_id.id(), ExtensionRegistry::ENABLED); | |
140 // |changed_hosts_| may include hosts that have been removed, | |
141 // which leads to the above lookup failing. In this case, just continue. | |
142 if (!extension) | |
143 continue; | |
144 AddHostInfo(host_id, ExtensionSet::ExtensionPathAndDefaultLocale( | |
145 extension->path(), | |
146 LocaleInfo::GetDefaultLocale(extension))); | |
147 } | |
148 } | |
149 | |
150 UserScriptLoader::LoadUserScriptsContentFunction | |
151 ExtensionUserScriptLoader::GetLoadUserScriptsFunction() { | |
152 return base::Bind(&ExtensionLoadScriptContent); | |
153 } | |
154 | |
155 void ExtensionUserScriptLoader::OnExtensionUnloaded( | |
156 content::BrowserContext* browser_context, | |
157 const Extension* extension, | |
158 UnloadedExtensionInfo::Reason reason) { | |
159 RemoveHostInfo(HostID(HostID::EXTENSIONS, extension->id())); | |
160 } | |
161 | |
162 void ExtensionUserScriptLoader::OnExtensionSystemReady() { | |
163 SetReady(true); | |
164 } | |
165 | |
166 } // namespace extensions | |
OLD | NEW |