Home | History | Annotate | Download | only in extensions 
      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 
     49 bool Send(IPC::Message* message) {
     50   return content::UtilityThread::Get()->Send(message);
     51 }
     52 
     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 
     62 ExtensionsHandler::ExtensionsHandler() {
     63 }
     64 
     65 ExtensionsHandler::~ExtensionsHandler() {
     66 }
     67 
     68 // static
     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 
     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 
     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 
    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 
    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 
    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 
    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 
    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 
    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)
    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)
    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)
    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 
    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 
    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)
    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