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 "chrome/utility/extensions/extensions_handler.h"
6
7 #include "base/command_line.h"
8 #include "base/path_service.h"
9 #include "chrome/common/chrome_utility_messages.h"
10 #include "chrome/common/extensions/chrome_extensions_client.h"
11 #include "chrome/common/extensions/chrome_utility_extensions_messages.h"
12 #include "chrome/common/media_galleries/metadata_types.h"
13 #include "chrome/utility/chrome_content_utility_client.h"
14 #include "chrome/utility/extensions/unpacker.h"
15 #include "chrome/utility/media_galleries/image_metadata_extractor.h"
16 #include "content/public/common/content_paths.h"
17 #include "content/public/utility/utility_thread.h"
18 #include "extensions/common/extension.h"
19 #include "extensions/common/extension_l10n_util.h"
20 #include "extensions/common/extension_utility_messages.h"
21 #include "extensions/common/manifest.h"
22 #include "extensions/common/update_manifest.h"
23 #include "media/base/media.h"
24 #include "media/base/media_file_checker.h"
25 #include "third_party/zlib/google/zip.h"
26 #include "ui/base/ui_base_switches.h"
27
28 #if defined(OS_WIN)
29 #include "chrome/common/extensions/api/networking_private/networking_private_crypto.h"
30 #include "chrome/utility/media_galleries/itunes_pref_parser_win.h"
31 #include "components/wifi/wifi_service.h"
32 #endif // defined(OS_WIN)
33
34 #if defined(OS_MACOSX)
35 #include "chrome/utility/media_galleries/iphoto_library_parser.h"
36 #endif // defined(OS_MACOSX)
37
38 #if defined(OS_WIN) || defined(OS_MACOSX)
39 #include "chrome/utility/media_galleries/iapps_xml_utils.h"
40 #include "chrome/utility/media_galleries/itunes_library_parser.h"
41 #include "chrome/utility/media_galleries/picasa_album_table_reader.h"
42 #include "chrome/utility/media_galleries/picasa_albums_indexer.h"
43 #endif // defined(OS_WIN) || defined(OS_MACOSX)
44
45 namespace extensions {
46
47 namespace {
48
Send(IPC::Message * message)49 bool Send(IPC::Message* message) {
50 return content::UtilityThread::Get()->Send(message);
51 }
52
ReleaseProcessIfNeeded()53 void ReleaseProcessIfNeeded() {
54 content::UtilityThread::Get()->ReleaseProcessIfNeeded();
55 }
56
57 const char kExtensionHandlerUnzipError[] =
58 "Could not unzip extension for install.";
59
60 } // namespace
61
ExtensionsHandler()62 ExtensionsHandler::ExtensionsHandler() {
63 }
64
~ExtensionsHandler()65 ExtensionsHandler::~ExtensionsHandler() {
66 }
67
68 // static
PreSandboxStartup()69 void ExtensionsHandler::PreSandboxStartup() {
70 // Initialize libexif for image metadata parsing.
71 metadata::ImageMetadataExtractor::InitializeLibrary();
72
73 // Load media libraries for media file validation.
74 base::FilePath media_path;
75 PathService::Get(content::DIR_MEDIA_LIBS, &media_path);
76 if (!media_path.empty())
77 media::InitializeMediaLibrary(media_path);
78 }
79
UtilityThreadStarted()80 void ExtensionsHandler::UtilityThreadStarted() {
81 base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
82 std::string lang = command_line->GetSwitchValueASCII(switches::kLang);
83 if (!lang.empty())
84 extension_l10n_util::SetProcessLocale(lang);
85 }
86
OnMessageReceived(const IPC::Message & message)87 bool ExtensionsHandler::OnMessageReceived(const IPC::Message& message) {
88 bool handled = true;
89 IPC_BEGIN_MESSAGE_MAP(ExtensionsHandler, message)
90 IPC_MESSAGE_HANDLER(ChromeUtilityMsg_UnpackExtension, OnUnpackExtension)
91 IPC_MESSAGE_HANDLER(ChromeUtilityMsg_UnzipToDir, OnUnzipToDir)
92 IPC_MESSAGE_HANDLER(ChromeUtilityMsg_DecodeImageBase64, OnDecodeImageBase64)
93 IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ParseJSON, OnParseJSON)
94 IPC_MESSAGE_HANDLER(ChromeUtilityMsg_CheckMediaFile, OnCheckMediaFile)
95 #if defined(OS_WIN)
96 IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ParseITunesPrefXml,
97 OnParseITunesPrefXml)
98 #endif // defined(OS_WIN)
99
100 #if defined(OS_MACOSX)
101 IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ParseIPhotoLibraryXmlFile,
102 OnParseIPhotoLibraryXmlFile)
103 #endif // defined(OS_MACOSX)
104
105 #if defined(OS_WIN) || defined(OS_MACOSX)
106 IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ParseITunesLibraryXmlFile,
107 OnParseITunesLibraryXmlFile)
108 IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ParsePicasaPMPDatabase,
109 OnParsePicasaPMPDatabase)
110 IPC_MESSAGE_HANDLER(ChromeUtilityMsg_IndexPicasaAlbumsContents,
111 OnIndexPicasaAlbumsContents)
112 #endif // defined(OS_WIN) || defined(OS_MACOSX)
113
114 #if defined(OS_WIN)
115 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_GetWiFiCredentials,
116 OnGetWiFiCredentials)
117 #endif // defined(OS_WIN)
118
119 IPC_MESSAGE_HANDLER(ExtensionUtilityMsg_ParseUpdateManifest,
120 OnParseUpdateManifest)
121
122 IPC_MESSAGE_UNHANDLED(handled = false)
123 IPC_END_MESSAGE_MAP()
124 return handled;
125 }
126
OnUnpackExtension(const base::FilePath & extension_path,const std::string & extension_id,int location,int creation_flags)127 void ExtensionsHandler::OnUnpackExtension(
128 const base::FilePath& extension_path,
129 const std::string& extension_id,
130 int location,
131 int creation_flags) {
132 CHECK_GT(location, Manifest::INVALID_LOCATION);
133 CHECK_LT(location, Manifest::NUM_LOCATIONS);
134 ExtensionsClient::Set(ChromeExtensionsClient::GetInstance());
135 Unpacker unpacker(extension_path,
136 extension_id,
137 static_cast<Manifest::Location>(location),
138 creation_flags);
139 if (unpacker.Run() && unpacker.DumpImagesToFile() &&
140 unpacker.DumpMessageCatalogsToFile()) {
141 Send(new ChromeUtilityHostMsg_UnpackExtension_Succeeded(
142 *unpacker.parsed_manifest()));
143 } else {
144 Send(new ChromeUtilityHostMsg_UnpackExtension_Failed(
145 unpacker.error_message()));
146 }
147
148 ReleaseProcessIfNeeded();
149 }
150
OnUnzipToDir(const base::FilePath & zip_path,const base::FilePath & dir)151 void ExtensionsHandler::OnUnzipToDir(const base::FilePath& zip_path,
152 const base::FilePath& dir) {
153 if (!zip::Unzip(zip_path, dir)) {
154 Send(new ChromeUtilityHostMsg_UnzipToDir_Failed(
155 std::string(kExtensionHandlerUnzipError)));
156 } else {
157 Send(new ChromeUtilityHostMsg_UnzipToDir_Succeeded(dir));
158 }
159
160 ReleaseProcessIfNeeded();
161 }
162
OnParseUpdateManifest(const std::string & xml)163 void ExtensionsHandler::OnParseUpdateManifest(const std::string& xml) {
164 UpdateManifest manifest;
165 if (!manifest.Parse(xml)) {
166 Send(new ExtensionUtilityHostMsg_ParseUpdateManifest_Failed(
167 manifest.errors()));
168 } else {
169 Send(new ExtensionUtilityHostMsg_ParseUpdateManifest_Succeeded(
170 manifest.results()));
171 }
172 ReleaseProcessIfNeeded();
173 }
174
OnDecodeImageBase64(const std::string & encoded_string)175 void ExtensionsHandler::OnDecodeImageBase64(
176 const std::string& encoded_string) {
177 std::string decoded_string;
178
179 if (!base::Base64Decode(encoded_string, &decoded_string)) {
180 Send(new ChromeUtilityHostMsg_DecodeImage_Failed());
181 return;
182 }
183
184 std::vector<unsigned char> decoded_vector(decoded_string.size());
185 for (size_t i = 0; i < decoded_string.size(); ++i) {
186 decoded_vector[i] = static_cast<unsigned char>(decoded_string[i]);
187 }
188
189 ChromeContentUtilityClient::DecodeImageAndSend(decoded_vector, false);
190 }
191
OnParseJSON(const std::string & json)192 void ExtensionsHandler::OnParseJSON(const std::string& json) {
193 int error_code;
194 std::string error;
195 base::Value* value = base::JSONReader::ReadAndReturnError(
196 json, base::JSON_PARSE_RFC, &error_code, &error);
197 if (value) {
198 base::ListValue wrapper;
199 wrapper.Append(value);
200 Send(new ChromeUtilityHostMsg_ParseJSON_Succeeded(wrapper));
201 } else {
202 Send(new ChromeUtilityHostMsg_ParseJSON_Failed(error));
203 }
204 ReleaseProcessIfNeeded();
205 }
206
OnCheckMediaFile(int64 milliseconds_of_decoding,const IPC::PlatformFileForTransit & media_file)207 void ExtensionsHandler::OnCheckMediaFile(
208 int64 milliseconds_of_decoding,
209 const IPC::PlatformFileForTransit& media_file) {
210 media::MediaFileChecker checker(
211 IPC::PlatformFileForTransitToFile(media_file));
212 const bool check_success = checker.Start(
213 base::TimeDelta::FromMilliseconds(milliseconds_of_decoding));
214 Send(new ChromeUtilityHostMsg_CheckMediaFile_Finished(check_success));
215 ReleaseProcessIfNeeded();
216 }
217
218 #if defined(OS_WIN)
OnParseITunesPrefXml(const std::string & itunes_xml_data)219 void ExtensionsHandler::OnParseITunesPrefXml(
220 const std::string& itunes_xml_data) {
221 base::FilePath library_path(
222 itunes::FindLibraryLocationInPrefXml(itunes_xml_data));
223 Send(new ChromeUtilityHostMsg_GotITunesDirectory(library_path));
224 ReleaseProcessIfNeeded();
225 }
226 #endif // defined(OS_WIN)
227
228 #if defined(OS_MACOSX)
OnParseIPhotoLibraryXmlFile(const IPC::PlatformFileForTransit & iphoto_library_file)229 void ExtensionsHandler::OnParseIPhotoLibraryXmlFile(
230 const IPC::PlatformFileForTransit& iphoto_library_file) {
231 iphoto::IPhotoLibraryParser parser;
232 base::File file = IPC::PlatformFileForTransitToFile(iphoto_library_file);
233 bool result = parser.Parse(iapps::ReadFileAsString(file.Pass()));
234 Send(new ChromeUtilityHostMsg_GotIPhotoLibrary(result, parser.library()));
235 ReleaseProcessIfNeeded();
236 }
237 #endif // defined(OS_MACOSX)
238
239 #if defined(OS_WIN) || defined(OS_MACOSX)
OnParseITunesLibraryXmlFile(const IPC::PlatformFileForTransit & itunes_library_file)240 void ExtensionsHandler::OnParseITunesLibraryXmlFile(
241 const IPC::PlatformFileForTransit& itunes_library_file) {
242 itunes::ITunesLibraryParser parser;
243 base::File file = IPC::PlatformFileForTransitToFile(itunes_library_file);
244 bool result = parser.Parse(iapps::ReadFileAsString(file.Pass()));
245 Send(new ChromeUtilityHostMsg_GotITunesLibrary(result, parser.library()));
246 ReleaseProcessIfNeeded();
247 }
248
OnParsePicasaPMPDatabase(const picasa::AlbumTableFilesForTransit & album_table_files)249 void ExtensionsHandler::OnParsePicasaPMPDatabase(
250 const picasa::AlbumTableFilesForTransit& album_table_files) {
251 picasa::AlbumTableFiles files;
252 files.indicator_file =
253 IPC::PlatformFileForTransitToFile(album_table_files.indicator_file);
254 files.category_file =
255 IPC::PlatformFileForTransitToFile(album_table_files.category_file);
256 files.date_file =
257 IPC::PlatformFileForTransitToFile(album_table_files.date_file);
258 files.filename_file =
259 IPC::PlatformFileForTransitToFile(album_table_files.filename_file);
260 files.name_file =
261 IPC::PlatformFileForTransitToFile(album_table_files.name_file);
262 files.token_file =
263 IPC::PlatformFileForTransitToFile(album_table_files.token_file);
264 files.uid_file =
265 IPC::PlatformFileForTransitToFile(album_table_files.uid_file);
266
267 picasa::PicasaAlbumTableReader reader(files.Pass());
268 bool parse_success = reader.Init();
269 Send(new ChromeUtilityHostMsg_ParsePicasaPMPDatabase_Finished(
270 parse_success, reader.albums(), reader.folders()));
271 ReleaseProcessIfNeeded();
272 }
273
OnIndexPicasaAlbumsContents(const picasa::AlbumUIDSet & album_uids,const std::vector<picasa::FolderINIContents> & folders_inis)274 void ExtensionsHandler::OnIndexPicasaAlbumsContents(
275 const picasa::AlbumUIDSet& album_uids,
276 const std::vector<picasa::FolderINIContents>& folders_inis) {
277 picasa::PicasaAlbumsIndexer indexer(album_uids);
278 indexer.ParseFolderINI(folders_inis);
279
280 Send(new ChromeUtilityHostMsg_IndexPicasaAlbumsContents_Finished(
281 indexer.albums_images()));
282 ReleaseProcessIfNeeded();
283 }
284 #endif // defined(OS_WIN) || defined(OS_MACOSX)
285
286 #if defined(OS_WIN)
OnGetWiFiCredentials(const std::string & network_guid)287 void ExtensionsHandler::OnGetWiFiCredentials(const std::string& network_guid) {
288 scoped_ptr<wifi::WiFiService> wifi_service(wifi::WiFiService::Create());
289 wifi_service->Initialize(NULL);
290
291 std::string key_data;
292 std::string error;
293 wifi_service->GetKeyFromSystem(network_guid, &key_data, &error);
294
295 Send(new ChromeUtilityHostMsg_GotWiFiCredentials(key_data, error.empty()));
296 }
297 #endif // defined(OS_WIN)
298
299 } // namespace extensions
300