OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/extensions/webstore_install_helper.h" | 5 #include "chrome/browser/extensions/webstore_install_helper.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/memory/weak_ptr.h" |
10 #include "base/thread_task_runner_handle.h" | 11 #include "base/thread_task_runner_handle.h" |
11 #include "base/values.h" | 12 #include "base/values.h" |
12 #include "chrome/browser/bitmap_fetcher/bitmap_fetcher.h" | 13 #include "chrome/browser/bitmap_fetcher/bitmap_fetcher.h" |
13 #include "chrome/common/chrome_utility_messages.h" | |
14 #include "chrome/common/extensions/chrome_utility_extensions_messages.h" | 14 #include "chrome/common/extensions/chrome_utility_extensions_messages.h" |
15 #include "chrome/grit/generated_resources.h" | 15 #include "chrome/grit/generated_resources.h" |
| 16 #include "components/safe_json_parser/safe_json_parser.h" |
16 #include "content/public/browser/browser_thread.h" | 17 #include "content/public/browser/browser_thread.h" |
17 #include "content/public/browser/utility_process_host.h" | |
18 #include "net/base/load_flags.h" | 18 #include "net/base/load_flags.h" |
19 #include "net/url_request/url_request.h" | 19 #include "net/url_request/url_request.h" |
20 #include "ui/base/l10n/l10n_util.h" | 20 #include "ui/base/l10n/l10n_util.h" |
21 | 21 |
22 using content::BrowserThread; | 22 using content::BrowserThread; |
23 using content::UtilityProcessHost; | |
24 | 23 |
25 namespace { | 24 namespace { |
26 | 25 |
27 const char kImageDecodeError[] = "Image decode failed"; | 26 const char kImageDecodeError[] = "Image decode failed"; |
28 | 27 |
29 } // namespace | 28 } // namespace |
30 | 29 |
31 namespace extensions { | 30 namespace extensions { |
32 | 31 |
33 WebstoreInstallHelper::WebstoreInstallHelper( | 32 WebstoreInstallHelper::WebstoreInstallHelper( |
(...skipping 23 matching lines...) Expand all Loading... |
57 // No existing |icon_fetcher_| to avoid unbalanced AddRef(). | 56 // No existing |icon_fetcher_| to avoid unbalanced AddRef(). |
58 CHECK(!icon_fetcher_.get()); | 57 CHECK(!icon_fetcher_.get()); |
59 AddRef(); // Balanced in OnFetchComplete(). | 58 AddRef(); // Balanced in OnFetchComplete(). |
60 icon_fetcher_.reset(new chrome::BitmapFetcher(icon_url_, this)); | 59 icon_fetcher_.reset(new chrome::BitmapFetcher(icon_url_, this)); |
61 icon_fetcher_->Start( | 60 icon_fetcher_->Start( |
62 context_getter_, std::string(), | 61 context_getter_, std::string(), |
63 net::URLRequest::CLEAR_REFERRER_ON_TRANSITION_FROM_SECURE_TO_INSECURE, | 62 net::URLRequest::CLEAR_REFERRER_ON_TRANSITION_FROM_SECURE_TO_INSECURE, |
64 net::LOAD_DO_NOT_SAVE_COOKIES | net::LOAD_DO_NOT_SEND_COOKIES); | 63 net::LOAD_DO_NOT_SAVE_COOKIES | net::LOAD_DO_NOT_SEND_COOKIES); |
65 } | 64 } |
66 | 65 |
67 BrowserThread::PostTask( | 66 base::WeakPtrFactory<WebstoreInstallHelper> weak_factory(this); |
68 BrowserThread::IO, | 67 scoped_refptr<safe_json_parser::SafeJsonParser> parser = |
69 FROM_HERE, | 68 new safe_json_parser::SafeJsonParser( |
70 base::Bind(&WebstoreInstallHelper::StartWorkOnIOThread, this)); | 69 manifest_, base::Bind(&WebstoreInstallHelper::OnJSONParseSucceeded, |
71 } | 70 weak_factory.GetWeakPtr()), |
72 | 71 base::Bind(&WebstoreInstallHelper::OnJSONParseFailed, |
73 void WebstoreInstallHelper::StartWorkOnIOThread() { | 72 weak_factory.GetWeakPtr())); |
74 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 73 parser->Start(); |
75 utility_host_ = UtilityProcessHost::Create( | |
76 this, base::ThreadTaskRunnerHandle::Get().get())->AsWeakPtr(); | |
77 utility_host_->SetName(l10n_util::GetStringUTF16( | |
78 IDS_UTILITY_PROCESS_JSON_PARSER_NAME)); | |
79 utility_host_->StartBatchMode(); | |
80 | |
81 utility_host_->Send(new ChromeUtilityMsg_ParseJSON(manifest_)); | |
82 } | |
83 | |
84 bool WebstoreInstallHelper::OnMessageReceived(const IPC::Message& message) { | |
85 bool handled = true; | |
86 IPC_BEGIN_MESSAGE_MAP(WebstoreInstallHelper, message) | |
87 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_ParseJSON_Succeeded, | |
88 OnJSONParseSucceeded) | |
89 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_ParseJSON_Failed, | |
90 OnJSONParseFailed) | |
91 IPC_MESSAGE_UNHANDLED(handled = false) | |
92 IPC_END_MESSAGE_MAP() | |
93 return handled; | |
94 } | 74 } |
95 | 75 |
96 void WebstoreInstallHelper::OnFetchComplete(const GURL& url, | 76 void WebstoreInstallHelper::OnFetchComplete(const GURL& url, |
97 const SkBitmap* image) { | 77 const SkBitmap* image) { |
98 CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 78 CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
99 // OnFetchComplete should only be called as icon_fetcher_ delegate to avoid | 79 // OnFetchComplete should only be called as icon_fetcher_ delegate to avoid |
100 // unbalanced Release(). | 80 // unbalanced Release(). |
101 CHECK(icon_fetcher_.get()); | 81 CHECK(icon_fetcher_.get()); |
102 | 82 |
103 if (image) | 83 if (image) |
104 icon_ = *image; | 84 icon_ = *image; |
105 icon_decode_complete_ = true; | 85 icon_decode_complete_ = true; |
106 if (icon_.empty()) { | 86 if (icon_.empty()) { |
107 error_ = kImageDecodeError; | 87 error_ = kImageDecodeError; |
108 parse_error_ = Delegate::ICON_ERROR; | 88 parse_error_ = Delegate::ICON_ERROR; |
109 } | 89 } |
110 icon_fetcher_.reset(); | 90 icon_fetcher_.reset(); |
111 BrowserThread::PostTask( | 91 BrowserThread::PostTask( |
112 BrowserThread::IO, | 92 BrowserThread::IO, |
113 FROM_HERE, | 93 FROM_HERE, |
114 base::Bind(&WebstoreInstallHelper::ReportResultsIfComplete, this)); | 94 base::Bind(&WebstoreInstallHelper::ReportResultsIfComplete, this)); |
115 Release(); // Balanced in Start(). | 95 Release(); // Balanced in Start(). |
116 } | 96 } |
117 | 97 |
118 void WebstoreInstallHelper::OnJSONParseSucceeded( | 98 void WebstoreInstallHelper::OnJSONParseSucceeded( |
119 const base::ListValue& wrapper) { | 99 scoped_ptr<base::Value> wrapper) { |
120 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 100 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
121 manifest_parse_complete_ = true; | 101 manifest_parse_complete_ = true; |
122 const base::Value* value = NULL; | 102 const base::ListValue* wrapper_list = NULL; |
123 CHECK(wrapper.Get(0, &value)); | 103 if (!wrapper->GetAsList(&wrapper_list)) { |
124 if (value->IsType(base::Value::TYPE_DICTIONARY)) { | 104 parse_error_ = Delegate::MANIFEST_ERROR; |
125 parsed_manifest_.reset( | |
126 static_cast<const base::DictionaryValue*>(value)->DeepCopy()); | |
127 } else { | 105 } else { |
128 parse_error_ = Delegate::MANIFEST_ERROR; | 106 const base::Value* value = NULL; |
| 107 CHECK(wrapper_list->Get(0, &value)); |
| 108 if (value->IsType(base::Value::TYPE_DICTIONARY)) { |
| 109 parsed_manifest_.reset( |
| 110 static_cast<const base::DictionaryValue*>(value)->DeepCopy()); |
| 111 } else { |
| 112 parse_error_ = Delegate::MANIFEST_ERROR; |
| 113 } |
129 } | 114 } |
130 ReportResultsIfComplete(); | 115 ReportResultsIfComplete(); |
131 } | 116 } |
132 | 117 |
133 void WebstoreInstallHelper::OnJSONParseFailed( | 118 void WebstoreInstallHelper::OnJSONParseFailed( |
134 const std::string& error_message) { | 119 const std::string& error_message) { |
135 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 120 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
136 manifest_parse_complete_ = true; | 121 manifest_parse_complete_ = true; |
137 error_ = error_message; | 122 error_ = error_message; |
138 parse_error_ = Delegate::MANIFEST_ERROR; | 123 parse_error_ = Delegate::MANIFEST_ERROR; |
139 ReportResultsIfComplete(); | 124 ReportResultsIfComplete(); |
140 } | 125 } |
141 | 126 |
142 void WebstoreInstallHelper::ReportResultsIfComplete() { | 127 void WebstoreInstallHelper::ReportResultsIfComplete() { |
143 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 128 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
144 | 129 |
145 if (!icon_decode_complete_ || !manifest_parse_complete_) | 130 if (!icon_decode_complete_ || !manifest_parse_complete_) |
146 return; | 131 return; |
147 | 132 |
148 // The utility_host_ will take care of deleting itself after this call. | |
149 if (utility_host_.get()) { | |
150 utility_host_->EndBatchMode(); | |
151 utility_host_.reset(); | |
152 } | |
153 | |
154 BrowserThread::PostTask( | 133 BrowserThread::PostTask( |
155 BrowserThread::UI, | 134 BrowserThread::UI, |
156 FROM_HERE, | 135 FROM_HERE, |
157 base::Bind(&WebstoreInstallHelper::ReportResultFromUIThread, this)); | 136 base::Bind(&WebstoreInstallHelper::ReportResultFromUIThread, this)); |
158 } | 137 } |
159 | 138 |
160 void WebstoreInstallHelper::ReportResultFromUIThread() { | 139 void WebstoreInstallHelper::ReportResultFromUIThread() { |
161 CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 140 CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
162 if (error_.empty() && parsed_manifest_) | 141 if (error_.empty() && parsed_manifest_) |
163 delegate_->OnWebstoreParseSuccess(id_, icon_, parsed_manifest_.release()); | 142 delegate_->OnWebstoreParseSuccess(id_, icon_, parsed_manifest_.release()); |
164 else | 143 else |
165 delegate_->OnWebstoreParseFailure(id_, parse_error_, error_); | 144 delegate_->OnWebstoreParseFailure(id_, parse_error_, error_); |
166 } | 145 } |
167 | 146 |
168 } // namespace extensions | 147 } // namespace extensions |
OLD | NEW |