OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 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 | 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 "components/image_fetcher/core/image_data_fetcher.h" | 5 #include "components/image_fetcher/core/image_data_fetcher.h" |
6 | 6 |
7 #include <utility> | 7 #include <utility> |
8 | 8 |
9 #include "net/base/load_flags.h" | 9 #include "net/base/load_flags.h" |
10 #include "net/http/http_response_headers.h" | 10 #include "net/http/http_response_headers.h" |
(...skipping 23 matching lines...) Expand all Loading... |
34 ~ImageDataFetcherRequest() {} | 34 ~ImageDataFetcherRequest() {} |
35 | 35 |
36 // The callback to run after the image data was fetched. The callback will | 36 // The callback to run after the image data was fetched. The callback will |
37 // be run even if the image data could not be fetched successfully. | 37 // be run even if the image data could not be fetched successfully. |
38 ImageDataFetcherCallback callback; | 38 ImageDataFetcherCallback callback; |
39 | 39 |
40 std::unique_ptr<net::URLFetcher> url_fetcher; | 40 std::unique_ptr<net::URLFetcher> url_fetcher; |
41 }; | 41 }; |
42 | 42 |
43 ImageDataFetcher::ImageDataFetcher( | 43 ImageDataFetcher::ImageDataFetcher( |
44 net::URLRequestContextGetter* url_request_context_getter) | 44 net::URLRequestContextGetter* url_request_context_getter, |
| 45 const net::NetworkTrafficAnnotationTag& traffic_annotation) |
45 : url_request_context_getter_(url_request_context_getter), | 46 : url_request_context_getter_(url_request_context_getter), |
46 data_use_service_name_(DataUseUserData::IMAGE_FETCHER_UNTAGGED), | 47 data_use_service_name_(DataUseUserData::IMAGE_FETCHER_UNTAGGED), |
47 next_url_fetcher_id_(0) {} | 48 next_url_fetcher_id_(0), |
| 49 traffic_annotation_(traffic_annotation) {} |
48 | 50 |
49 ImageDataFetcher::~ImageDataFetcher() {} | 51 ImageDataFetcher::~ImageDataFetcher() {} |
50 | 52 |
51 void ImageDataFetcher::SetDataUseServiceName( | 53 void ImageDataFetcher::SetDataUseServiceName( |
52 DataUseServiceName data_use_service_name) { | 54 DataUseServiceName data_use_service_name) { |
53 data_use_service_name_ = data_use_service_name; | 55 data_use_service_name_ = data_use_service_name; |
54 } | 56 } |
55 | 57 |
56 void ImageDataFetcher::SetImageDownloadLimit( | 58 void ImageDataFetcher::SetImageDownloadLimit( |
57 base::Optional<int64_t> max_download_bytes) { | 59 base::Optional<int64_t> max_download_bytes) { |
58 max_download_bytes_ = max_download_bytes; | 60 max_download_bytes_ = max_download_bytes; |
59 } | 61 } |
60 | 62 |
61 void ImageDataFetcher::FetchImageData( | 63 void ImageDataFetcher::FetchImageData( |
62 const GURL& image_url, | 64 const GURL& image_url, |
63 const ImageDataFetcherCallback& callback) { | 65 const ImageDataFetcherCallback& callback) { |
64 FetchImageData( | 66 FetchImageData( |
65 image_url, callback, /*referrer=*/std::string(), | 67 image_url, callback, /*referrer=*/std::string(), |
66 net::URLRequest::CLEAR_REFERRER_ON_TRANSITION_FROM_SECURE_TO_INSECURE); | 68 net::URLRequest::CLEAR_REFERRER_ON_TRANSITION_FROM_SECURE_TO_INSECURE); |
67 } | 69 } |
68 | 70 |
69 void ImageDataFetcher::FetchImageData( | 71 void ImageDataFetcher::FetchImageData( |
70 const GURL& image_url, | 72 const GURL& image_url, |
71 const ImageDataFetcherCallback& callback, | 73 const ImageDataFetcherCallback& callback, |
72 const std::string& referrer, | 74 const std::string& referrer, |
73 net::URLRequest::ReferrerPolicy referrer_policy) { | 75 net::URLRequest::ReferrerPolicy referrer_policy) { |
74 std::unique_ptr<net::URLFetcher> url_fetcher = net::URLFetcher::Create( | 76 std::unique_ptr<net::URLFetcher> url_fetcher = |
75 next_url_fetcher_id_++, image_url, net::URLFetcher::GET, this); | 77 net::URLFetcher::Create(next_url_fetcher_id_++, image_url, |
| 78 net::URLFetcher::GET, this, traffic_annotation_); |
76 | 79 |
77 DataUseUserData::AttachToFetcher(url_fetcher.get(), data_use_service_name_); | 80 DataUseUserData::AttachToFetcher(url_fetcher.get(), data_use_service_name_); |
78 | 81 |
79 std::unique_ptr<ImageDataFetcherRequest> request( | 82 std::unique_ptr<ImageDataFetcherRequest> request( |
80 new ImageDataFetcherRequest(callback, std::move(url_fetcher))); | 83 new ImageDataFetcherRequest(callback, std::move(url_fetcher))); |
81 request->url_fetcher->SetRequestContext(url_request_context_getter_.get()); | 84 request->url_fetcher->SetRequestContext(url_request_context_getter_.get()); |
82 request->url_fetcher->SetReferrer(referrer); | 85 request->url_fetcher->SetReferrer(referrer); |
83 request->url_fetcher->SetReferrerPolicy(referrer_policy); | 86 request->url_fetcher->SetReferrerPolicy(referrer_policy); |
84 request->url_fetcher->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | | 87 request->url_fetcher->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | |
85 net::LOAD_DO_NOT_SAVE_COOKIES | | 88 net::LOAD_DO_NOT_SAVE_COOKIES | |
(...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
135 void ImageDataFetcher::FinishRequest(const net::URLFetcher* source, | 138 void ImageDataFetcher::FinishRequest(const net::URLFetcher* source, |
136 const RequestMetadata& metadata, | 139 const RequestMetadata& metadata, |
137 const std::string& image_data) { | 140 const std::string& image_data) { |
138 auto request_iter = pending_requests_.find(source); | 141 auto request_iter = pending_requests_.find(source); |
139 DCHECK(request_iter != pending_requests_.end()); | 142 DCHECK(request_iter != pending_requests_.end()); |
140 request_iter->second->callback.Run(image_data, metadata); | 143 request_iter->second->callback.Run(image_data, metadata); |
141 pending_requests_.erase(request_iter); | 144 pending_requests_.erase(request_iter); |
142 } | 145 } |
143 | 146 |
144 } // namespace image_fetcher | 147 } // namespace image_fetcher |
OLD | NEW |