| 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 |