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/browser/extensions/zipfile_installer.h" 6 7#include "base/files/file_util.h" 8#include "base/path_service.h" 9#include "chrome/browser/extensions/extension_error_reporter.h" 10#include "chrome/browser/extensions/extension_service.h" 11#include "chrome/browser/extensions/unpacked_installer.h" 12#include "chrome/browser/profiles/profile.h" 13#include "chrome/common/extensions/chrome_utility_extensions_messages.h" 14#include "content/public/browser/browser_thread.h" 15#include "content/public/browser/utility_process_host.h" 16 17using content::BrowserThread; 18using content::UtilityProcessHost; 19 20namespace { 21 22const char kExtensionHandlerTempDirError[] = 23 "Could not create temporary directory for zipped extension."; 24 25} // namespace 26 27namespace extensions { 28 29ZipFileInstaller::ZipFileInstaller(ExtensionService* extension_service) 30 : be_noisy_on_failure_(true), 31 extension_service_weak_(extension_service->AsWeakPtr()) { 32} 33 34void ZipFileInstaller::LoadFromZipFile(const base::FilePath& path) { 35 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); 36 zip_path_ = path; 37 BrowserThread::PostTask(BrowserThread::FILE, 38 FROM_HERE, 39 base::Bind(&ZipFileInstaller::PrepareTempDir, this)); 40} 41 42void ZipFileInstaller::PrepareTempDir() { 43 CHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 44 base::FilePath temp_dir; 45 PathService::Get(base::DIR_TEMP, &temp_dir); 46 base::FilePath new_temp_dir; 47 if (!base::CreateTemporaryDirInDir( 48 temp_dir, 49 zip_path_.RemoveExtension().BaseName().value() + 50 FILE_PATH_LITERAL("_"), 51 &new_temp_dir)) { 52 OnUnzipFailed(std::string(kExtensionHandlerTempDirError)); 53 return; 54 } 55 BrowserThread::PostTask( 56 BrowserThread::IO, 57 FROM_HERE, 58 base::Bind(&ZipFileInstaller::StartWorkOnIOThread, this, new_temp_dir)); 59} 60 61ZipFileInstaller::~ZipFileInstaller() { 62} 63 64// static 65scoped_refptr<ZipFileInstaller> ZipFileInstaller::Create( 66 ExtensionService* extension_service) { 67 DCHECK(extension_service); 68 return scoped_refptr<ZipFileInstaller>( 69 new ZipFileInstaller(extension_service)); 70} 71 72void ZipFileInstaller::StartWorkOnIOThread(const base::FilePath& temp_dir) { 73 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 74 UtilityProcessHost* host = 75 UtilityProcessHost::Create(this, base::MessageLoopProxy::current().get()); 76 host->SetExposedDir(temp_dir); 77 host->Send(new ChromeUtilityMsg_UnzipToDir(zip_path_, temp_dir)); 78} 79 80void ZipFileInstaller::ReportSuccessOnUIThread( 81 const base::FilePath& unzipped_path) { 82 CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); 83 if (extension_service_weak_.get()) 84 UnpackedInstaller::Create(extension_service_weak_.get()) 85 ->Load(unzipped_path); 86} 87 88void ZipFileInstaller::ReportErrorOnUIThread(const std::string& error) { 89 CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); 90 if (extension_service_weak_.get()) { 91 ExtensionErrorReporter::GetInstance()->ReportLoadError( 92 zip_path_, 93 error, 94 extension_service_weak_->profile(), 95 be_noisy_on_failure_); 96 } 97} 98 99void ZipFileInstaller::OnUnzipSucceeded(const base::FilePath& unzipped_path) { 100 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 101 BrowserThread::PostTask( 102 BrowserThread::UI, 103 FROM_HERE, 104 base::Bind( 105 &ZipFileInstaller::ReportSuccessOnUIThread, this, unzipped_path)); 106} 107 108void ZipFileInstaller::OnUnzipFailed(const std::string& error) { 109 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 110 BrowserThread::PostTask( 111 BrowserThread::UI, 112 FROM_HERE, 113 base::Bind(&ZipFileInstaller::ReportErrorOnUIThread, this, error)); 114} 115 116bool ZipFileInstaller::OnMessageReceived(const IPC::Message& message) { 117 bool handled = true; 118 IPC_BEGIN_MESSAGE_MAP(ZipFileInstaller, message) 119 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_UnzipToDir_Succeeded, 120 OnUnzipSucceeded) 121 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_UnzipToDir_Failed, OnUnzipFailed) 122 IPC_MESSAGE_UNHANDLED(handled = false) 123 IPC_END_MESSAGE_MAP() 124 return handled; 125} 126 127} // namespace extensions 128