extensions/utility/utility_handler.cc

   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 "extensions/utility/utility_handler.h"
   6
   7 #include "base/command_line.h"
   8 #include "base/files/file_path.h"
   9 #include "base/files/file_util.h"
  10 #include "base/i18n/rtl.h"
  11 #include "content/public/utility/utility_thread.h"
  12 #include "extensions/common/constants.h"
  13 #include "extensions/common/extension.h"
  14 #include "extensions/common/extension_l10n_util.h"
  15 #include "extensions/common/extension_utility_messages.h"
  16 #include "extensions/common/extensions_client.h"
  17 #include "extensions/common/manifest.h"
  18 #include "extensions/common/update_manifest.h"
  19 #include "extensions/strings/grit/extensions_strings.h"
  20 #include "extensions/utility/unpacker.h"
  21 #include "ipc/ipc_message.h"
  22 #include "ipc/ipc_message_macros.h"
  23 #include "third_party/zlib/google/zip.h"
  24 #include "ui/base/l10n/l10n_util.h"
  25 #include "ui/base/ui_base_switches.h"
  26
  27 namespace extensions {
  28
  29 namespace {
  30
  31 bool Send(IPC::Message* message) {
  32   return content::UtilityThread::Get()->Send(message);
  33 }
  34
  35 void ReleaseProcessIfNeeded() {
  36   content::UtilityThread::Get()->ReleaseProcessIfNeeded();
  37 }
  38
  39 const char kExtensionHandlerUnzipError[] =
  40     "Could not unzip extension for install.";
  41
  42 }  // namespace
  43
  44 UtilityHandler::UtilityHandler() {
  45 }
  46
  47 UtilityHandler::~UtilityHandler() {
  48 }
  49
  50 // static
  51 void UtilityHandler::UtilityThreadStarted() {
  52   base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
  53   std::string lang = command_line->GetSwitchValueASCII(switches::kLang);
  54   if (!lang.empty())
  55     extension_l10n_util::SetProcessLocale(lang);
  56 }
  57
  58 bool UtilityHandler::OnMessageReceived(const IPC::Message& message) {
  59   bool handled = true;
  60   IPC_BEGIN_MESSAGE_MAP(UtilityHandler, message)
  61     IPC_MESSAGE_HANDLER(ExtensionUtilityMsg_ParseUpdateManifest,
  62                         OnParseUpdateManifest)
  63     IPC_MESSAGE_HANDLER(ExtensionUtilityMsg_UnzipToDir, OnUnzipToDir)
  64     IPC_MESSAGE_HANDLER(ExtensionUtilityMsg_UnpackExtension, OnUnpackExtension)
  65     IPC_MESSAGE_UNHANDLED(handled = false)
  66   IPC_END_MESSAGE_MAP()
  67   return handled;
  68 }
  69
  70 void UtilityHandler::OnParseUpdateManifest(const std::string& xml) {
  71   UpdateManifest manifest;
  72   if (!manifest.Parse(xml)) {
  73     Send(new ExtensionUtilityHostMsg_ParseUpdateManifest_Failed(
  74         manifest.errors()));
  75   } else {
  76     Send(new ExtensionUtilityHostMsg_ParseUpdateManifest_Succeeded(
  77         manifest.results()));
  78   }
  79   ReleaseProcessIfNeeded();
  80 }
  81
  82 void UtilityHandler::OnUnzipToDir(const base::FilePath& zip_path,
  83                                   const base::FilePath& dir) {
  84   if (!zip::Unzip(zip_path, dir)) {
  85     Send(new ExtensionUtilityHostMsg_UnzipToDir_Failed(
  86         std::string(kExtensionHandlerUnzipError)));
  87   } else {
  88     Send(new ExtensionUtilityHostMsg_UnzipToDir_Succeeded(dir));
  89   }
  90
  91   ReleaseProcessIfNeeded();
  92 }
  93
  94 void UtilityHandler::OnUnpackExtension(const base::FilePath& directory_path,
  95                                        const std::string& extension_id,
  96                                        int location,
  97                                        int creation_flags) {
  98   CHECK_GT(location, Manifest::INVALID_LOCATION);
  99   CHECK_LT(location, Manifest::NUM_LOCATIONS);
 100   DCHECK(ExtensionsClient::Get());
 101   content::UtilityThread::Get()->EnsureBlinkInitialized();
 102   Unpacker unpacker(directory_path.DirName(), directory_path, extension_id,
 103                     static_cast<Manifest::Location>(location), creation_flags);
 104   if (unpacker.Run()) {
 105     Send(new ExtensionUtilityHostMsg_UnpackExtension_Succeeded(
 106         *unpacker.parsed_manifest()));
 107   } else {
 108     Send(new ExtensionUtilityHostMsg_UnpackExtension_Failed(
 109         unpacker.error_message()));
 110   }
 111   ReleaseProcessIfNeeded();
 112 }
 113
 114
 115 }  // namespace extensions