| OLD | NEW |
| (Empty) |
| 1 // Copyright 2016 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 "components/image_fetcher/image_data_fetcher.h" | |
| 6 | |
| 7 #include "net/base/load_flags.h" | |
| 8 #include "net/http/http_response_headers.h" | |
| 9 #include "net/http/http_status_code.h" | |
| 10 #include "net/url_request/url_fetcher.h" | |
| 11 #include "net/url_request/url_request_context_getter.h" | |
| 12 #include "net/url_request/url_request_status.h" | |
| 13 #include "url/gurl.h" | |
| 14 | |
| 15 using data_use_measurement::DataUseUserData; | |
| 16 | |
| 17 namespace image_fetcher { | |
| 18 | |
| 19 // An active image URL fetcher request. The struct contains the related requests | |
| 20 // state. | |
| 21 struct ImageDataFetcher::ImageDataFetcherRequest { | |
| 22 ImageDataFetcherRequest(const ImageDataFetcherCallback& callback, | |
| 23 std::unique_ptr<net::URLFetcher> url_fetcher) | |
| 24 : callback(callback), | |
| 25 url_fetcher(std::move(url_fetcher)) {} | |
| 26 | |
| 27 ~ImageDataFetcherRequest() {} | |
| 28 | |
| 29 // The callback to run after the image data was fetched. The callback will | |
| 30 // be run even if the image data could not be fetched successfully. | |
| 31 ImageDataFetcherCallback callback; | |
| 32 | |
| 33 std::unique_ptr<net::URLFetcher> url_fetcher; | |
| 34 }; | |
| 35 | |
| 36 ImageDataFetcher::ImageDataFetcher( | |
| 37 net::URLRequestContextGetter* url_request_context_getter) | |
| 38 : url_request_context_getter_(url_request_context_getter), | |
| 39 data_use_service_name_(DataUseUserData::IMAGE_FETCHER_UNTAGGED), | |
| 40 next_url_fetcher_id_(0) {} | |
| 41 | |
| 42 ImageDataFetcher::~ImageDataFetcher() {} | |
| 43 | |
| 44 void ImageDataFetcher::SetDataUseServiceName( | |
| 45 DataUseServiceName data_use_service_name) { | |
| 46 data_use_service_name_ = data_use_service_name; | |
| 47 } | |
| 48 | |
| 49 void ImageDataFetcher::FetchImageData( | |
| 50 const GURL& image_url, | |
| 51 const ImageDataFetcherCallback& callback) { | |
| 52 FetchImageData( | |
| 53 image_url, callback, /*referrer=*/std::string(), | |
| 54 net::URLRequest::CLEAR_REFERRER_ON_TRANSITION_FROM_SECURE_TO_INSECURE); | |
| 55 } | |
| 56 | |
| 57 void ImageDataFetcher::FetchImageData( | |
| 58 const GURL& image_url, | |
| 59 const ImageDataFetcherCallback& callback, | |
| 60 const std::string& referrer, | |
| 61 net::URLRequest::ReferrerPolicy referrer_policy) { | |
| 62 std::unique_ptr<net::URLFetcher> url_fetcher = net::URLFetcher::Create( | |
| 63 next_url_fetcher_id_++, image_url, net::URLFetcher::GET, this); | |
| 64 | |
| 65 DataUseUserData::AttachToFetcher(url_fetcher.get(), data_use_service_name_); | |
| 66 | |
| 67 std::unique_ptr<ImageDataFetcherRequest> request( | |
| 68 new ImageDataFetcherRequest(callback, std::move(url_fetcher))); | |
| 69 request->url_fetcher->SetRequestContext(url_request_context_getter_.get()); | |
| 70 request->url_fetcher->SetReferrer(referrer); | |
| 71 request->url_fetcher->SetReferrerPolicy(referrer_policy); | |
| 72 request->url_fetcher->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | | |
| 73 net::LOAD_DO_NOT_SAVE_COOKIES | | |
| 74 net::LOAD_DO_NOT_SEND_AUTH_DATA); | |
| 75 request->url_fetcher->Start(); | |
| 76 | |
| 77 pending_requests_[request->url_fetcher.get()] = std::move(request); | |
| 78 } | |
| 79 | |
| 80 void ImageDataFetcher::OnURLFetchComplete(const net::URLFetcher* source) { | |
| 81 auto request_iter = pending_requests_.find(source); | |
| 82 DCHECK(request_iter != pending_requests_.end()); | |
| 83 | |
| 84 bool success = source->GetStatus().status() == net::URLRequestStatus::SUCCESS; | |
| 85 | |
| 86 RequestMetadata metadata; | |
| 87 if (success && source->GetResponseHeaders()) { | |
| 88 source->GetResponseHeaders()->GetMimeType(&metadata.mime_type); | |
| 89 metadata.http_response_code = source->GetResponseHeaders()->response_code(); | |
| 90 success &= (metadata.http_response_code == net::HTTP_OK); | |
| 91 } | |
| 92 metadata.from_http_cache = source->WasCached(); | |
| 93 | |
| 94 std::string image_data; | |
| 95 if (success) { | |
| 96 source->GetResponseAsString(&image_data); | |
| 97 } | |
| 98 request_iter->second->callback.Run(image_data, metadata); | |
| 99 | |
| 100 // Remove the finished request. | |
| 101 pending_requests_.erase(request_iter); | |
| 102 } | |
| 103 | |
| 104 } // namespace image_fetcher | |
| OLD | NEW |