1 // Copyright (c) 2012 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/pack_extension_job.h"
6
7 #include <memory>
8
9 #include "base/bind.h"
10 #include "base/strings/sys_string_conversions.h"
11 #include "base/strings/utf_string_conversions.h"
12 #include "base/threading/sequenced_task_runner_handle.h"
13 #include "chrome/grit/generated_resources.h"
14 #include "content/public/browser/browser_thread.h"
15 #include "extensions/browser/extension_creator.h"
16 #include "extensions/browser/extension_file_task_runner.h"
17 #include "extensions/common/constants.h"
18 #include "ui/base/l10n/l10n_util.h"
19
20 using content::BrowserThread;
21
22 namespace extensions {
23
PackExtensionJob(Client * client,const base::FilePath & root_directory,const base::FilePath & key_file,int run_flags)24 PackExtensionJob::PackExtensionJob(Client* client,
25 const base::FilePath& root_directory,
26 const base::FilePath& key_file,
27 int run_flags)
28 : client_(client),
29 key_file_(key_file),
30 run_flags_(run_flags | ExtensionCreator::kRequireModernManifestVersion) {
31 root_directory_ = root_directory.StripTrailingSeparators();
32 }
33
~PackExtensionJob()34 PackExtensionJob::~PackExtensionJob() {}
35
Start()36 void PackExtensionJob::Start() {
37 if (run_mode_ == RunMode::ASYNCHRONOUS) {
38 scoped_refptr<base::SequencedTaskRunner> task_runner =
39 base::SequencedTaskRunnerHandle::Get();
40 GetExtensionFileTaskRunner()->PostTask(
41 FROM_HERE,
42 base::BindOnce(&PackExtensionJob::Run,
43 // See class level comments for why Unretained is safe.
44 base::Unretained(this), std::move(task_runner)));
45 } else {
46 DCHECK_EQ(RunMode::SYNCHRONOUS, run_mode_);
47 Run(nullptr);
48 }
49 }
50
Run(scoped_refptr<base::SequencedTaskRunner> async_reply_task_runner)51 void PackExtensionJob::Run(
52 scoped_refptr<base::SequencedTaskRunner> async_reply_task_runner) {
53 DCHECK_EQ(!!async_reply_task_runner, run_mode_ == RunMode::ASYNCHRONOUS)
54 << "Provide task runner iff we are running in asynchronous mode.";
55 auto crx_file_out = std::make_unique<base::FilePath>(
56 root_directory_.AddExtension(kExtensionFileExtension));
57
58 auto key_file_out = std::make_unique<base::FilePath>();
59 if (key_file_.empty())
60 *key_file_out = root_directory_.AddExtension(kExtensionKeyFileExtension);
61
62 // TODO(aa): Need to internationalize the errors that ExtensionCreator
63 // returns. See bug 20734.
64 ExtensionCreator creator;
65 if (creator.Run(root_directory_, *crx_file_out, key_file_, *key_file_out,
66 run_flags_)) {
67 if (run_mode_ == RunMode::ASYNCHRONOUS) {
68 async_reply_task_runner->PostTask(
69 FROM_HERE,
70 base::BindOnce(&PackExtensionJob::ReportSuccessOnClientSequence,
71 // See class level comments for why Unretained is safe.
72 base::Unretained(this), std::move(crx_file_out),
73 std::move(key_file_out)));
74
75 } else {
76 ReportSuccessOnClientSequence(std::move(crx_file_out),
77 std::move(key_file_out));
78 }
79 } else {
80 if (run_mode_ == RunMode::ASYNCHRONOUS) {
81 async_reply_task_runner->PostTask(
82 FROM_HERE,
83 base::BindOnce(&PackExtensionJob::ReportFailureOnClientSequence,
84 // See class level comments for why Unretained is safe.
85 base::Unretained(this), creator.error_message(),
86 creator.error_type()));
87 } else {
88 DCHECK_EQ(RunMode::SYNCHRONOUS, run_mode_);
89 ReportFailureOnClientSequence(creator.error_message(),
90 creator.error_type());
91 }
92 }
93 }
94
ReportSuccessOnClientSequence(std::unique_ptr<base::FilePath> crx_file_out,std::unique_ptr<base::FilePath> key_file_out)95 void PackExtensionJob::ReportSuccessOnClientSequence(
96 std::unique_ptr<base::FilePath> crx_file_out,
97 std::unique_ptr<base::FilePath> key_file_out) {
98 DCHECK(client_);
99 DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
100 client_->OnPackSuccess(*crx_file_out, *key_file_out);
101 }
102
ReportFailureOnClientSequence(const std::string & error,ExtensionCreator::ErrorType error_type)103 void PackExtensionJob::ReportFailureOnClientSequence(
104 const std::string& error,
105 ExtensionCreator::ErrorType error_type) {
106 DCHECK(client_);
107 DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
108 client_->OnPackFailure(error, error_type);
109 }
110
111 // static
StandardSuccessMessage(const base::FilePath & crx_file,const base::FilePath & key_file)112 base::string16 PackExtensionJob::StandardSuccessMessage(
113 const base::FilePath& crx_file,
114 const base::FilePath& key_file) {
115 base::string16 crx_file_string = crx_file.LossyDisplayName();
116 base::string16 key_file_string = key_file.LossyDisplayName();
117 if (key_file_string.empty()) {
118 return l10n_util::GetStringFUTF16(
119 IDS_EXTENSION_PACK_DIALOG_SUCCESS_BODY_UPDATE,
120 crx_file_string);
121 } else {
122 return l10n_util::GetStringFUTF16(
123 IDS_EXTENSION_PACK_DIALOG_SUCCESS_BODY_NEW,
124 crx_file_string,
125 key_file_string);
126 }
127 }
128
129 } // namespace extensions
130