Chromium Code Reviews| Index: chrome/browser/extensions/api/dial/device_description_fetcher.cc |
| diff --git a/chrome/browser/extensions/api/dial/device_description_fetcher.cc b/chrome/browser/extensions/api/dial/device_description_fetcher.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..0a4b2743fc1b3c716fc487367713728f6f92475f |
| --- /dev/null |
| +++ b/chrome/browser/extensions/api/dial/device_description_fetcher.cc |
| @@ -0,0 +1,150 @@ |
| +// Copyright (c) 2017 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "base/strings/string_util.h" |
| +#include "base/strings/stringprintf.h" |
| +#include "chrome/browser/extensions/api/dial/device_description_fetcher.h" |
| +#include "chrome/browser/extensions/api/dial/dial_device_data.h" |
| +#include "chrome/browser/profiles/profile.h" |
| +#include "content/public/browser/browser_thread.h" |
| +#include "net/base/load_flags.h" |
| +#include "net/http/http_response_headers.h" |
| +#include "net/http/http_status_code.h" |
| +#include "net/http/http_util.h" |
| +#include "net/url_request/url_fetcher.h" |
| + |
| +using content::BrowserThread; |
| + |
| +constexpr char kApplicationUrlHeaderName[] = "Application-URL"; |
| +constexpr int kMaxRetries = 3; |
| +// DIAL devices are unlikely to expose uPnP functions other than DIAL, so 256kb |
| +// should be more than sufficient. |
| +constexpr int kMaxDescriptionSizeBytes = 262144; |
| + |
| +namespace extensions { |
| +namespace api { |
| +namespace dial { |
| + |
| +DeviceDescriptionFetcher::DeviceDescriptionFetcher( |
| + const GURL& device_description_url, |
| + Profile* profile, |
| + base::OnceCallback<void(const DialDeviceDescriptionData&)> success_cb, |
| + base::OnceCallback<void(const std::string&)> error_cb) |
| + : device_description_url_(device_description_url), |
| + profile_(profile), |
| + success_cb_(std::move(success_cb)), |
| + error_cb_(std::move(error_cb)) { |
| + DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| + DCHECK(profile_); |
| + DCHECK(device_description_url_.is_valid()); |
| +} |
| + |
| +DeviceDescriptionFetcher::~DeviceDescriptionFetcher() { |
| + DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| +} |
| + |
| +void DeviceDescriptionFetcher::Start() { |
| + DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| + DCHECK(!fetcher_); |
| + |
| + // DIAL returns device descriptions via GET request. |
| + fetcher_ = |
| + net::URLFetcher::Create(kURLFetcherIDForTest, device_description_url_, |
| + net::URLFetcher::GET, this); |
| + |
| + // net::LOAD_BYPASS_PROXY: Proxies almost certainly hurt more cases than they |
| + // help. |
| + // net::LOAD_DISABLE_CACHE: The request should not touch the cache. |
| + // net::LOAD_DO_NOT_{SAVE,SEND}_COOKIES: The request should not touch cookies. |
| + // net::LOAD_DO_NOT_SEND_AUTH_DATA: The request shoud not send auth data. |
|
Wez
2017/01/09 22:12:58
typo: should
mark a. foltz
2017/01/10 00:36:14
Done.
|
| + fetcher_->SetLoadFlags(net::LOAD_BYPASS_PROXY | net::LOAD_DISABLE_CACHE | |
| + net::LOAD_DO_NOT_SAVE_COOKIES | |
| + net::LOAD_DO_NOT_SEND_COOKIES | |
| + net::LOAD_DO_NOT_SEND_AUTH_DATA); |
| + |
| + // Section 5.4 of the DIAL spec prohibits redirects. |
| + fetcher_->SetStopOnRedirect(true); |
| + |
| + // Allow the fetcher to retry on 5XX responses and ERR_NETWORK_CHANGED. |
| + fetcher_->SetMaxRetriesOn5xx(kMaxRetries); |
| + fetcher_->SetAutomaticallyRetryOnNetworkChanges(kMaxRetries); |
| + |
| + fetcher_->SetRequestContext(profile_->GetRequestContext()); |
| + fetcher_->Start(); |
| +} |
| + |
| +void DeviceDescriptionFetcher::OnURLFetchComplete( |
| + const net::URLFetcher* source) { |
| + DCHECK(source); |
| + |
| + if (source->GetResponseCode() != net::HTTP_OK) { |
| + ReportError( |
| + base::StringPrintf("HTTP %d: Unable to fetch device description", |
| + source->GetResponseCode())); |
| + return; |
| + } |
| + |
| + const net::HttpResponseHeaders* headers = source->GetResponseHeaders(); |
| + |
| + // NOTE: The uPnP spec requires devices to set a Content-Type: header of |
| + // text/xml; charset="utf-8" (sec 2.11). However Chromecast (and possibly |
| + // other devices) do not comply, so specifically not checking this header. |
| + std::string app_url_header; |
| + if (!headers->GetNormalizedHeader(kApplicationUrlHeaderName, |
| + &app_url_header) || |
| + app_url_header.empty()) { |
| + ReportError("Missing or empty Application-URL:"); |
| + return; |
| + } |
| + |
| + // Section 5.4 of the DIAL spec implies that the Application URL should not |
| + // have path, query or fragment...unsure if that can be enforced. |
| + GURL app_url(app_url_header); |
| + if (!app_url.is_valid() || !app_url.SchemeIs("http") || |
| + !app_url.HostIsIPAddress() || |
| + app_url.host() != device_description_url_.host()) { |
| + ReportError(base::StringPrintf("Invalid Application-URL: %s", |
| + app_url_header.c_str())); |
| + return; |
| + } |
| + |
| + if (source->GetReceivedResponseContentLength() > kMaxDescriptionSizeBytes) { |
| + ReportError("Response too large"); |
| + return; |
| + } |
| + |
| + std::string device_description; |
| + if (!source->GetResponseAsString(&device_description) || |
| + device_description.empty()) { |
| + ReportError("Missing or empty response"); |
| + return; |
| + } |
| + |
| + if (!base::IsStringUTF8(device_description)) { |
| + ReportError("Invalid response encoding"); |
| + return; |
| + } |
| + |
| + std::move(success_cb_) |
| + .Run(DialDeviceDescriptionData(std::move(device_description), app_url)); |
| +} |
| + |
| +void DeviceDescriptionFetcher::OnURLFetchDownloadProgress( |
| + const net::URLFetcher* source, |
| + int64_t current, |
| + int64_t total, |
| + int64_t current_network_bytes) {} |
| + |
| +void DeviceDescriptionFetcher::OnURLFetchUploadProgress( |
| + const net::URLFetcher* source, |
| + int64_t current, |
| + int64_t total) {} |
| + |
| +void DeviceDescriptionFetcher::ReportError(const std::string& message) { |
| + std::move(error_cb_).Run(message); |
| +} |
| + |
| +} // namespace dial |
| +} // namespace api |
| +} // namespace extensions |