Index: components/favicon/core/favicon_handler.cc |
diff --git a/components/favicon/core/favicon_handler.cc b/components/favicon/core/favicon_handler.cc |
index 738fa28cd48797d373e57c5d96ccd3101ceec7b4..c60c503dda29677dbe7e523b8789e22e58886cd9 100644 |
--- a/components/favicon/core/favicon_handler.cc |
+++ b/components/favicon/core/favicon_handler.cc |
@@ -227,7 +227,9 @@ void FaviconHandler::FetchFavicon(const GURL& url) { |
initial_history_result_expired_or_incomplete_ = false; |
redownload_icons_ = false; |
got_favicon_from_history_ = false; |
- download_request_.Cancel(); |
+ manifest_download_request_.Cancel(); |
+ image_download_request_.Cancel(); |
+ manifest_url_.reset(); |
candidates_.clear(); |
notification_icon_url_ = GURL(); |
notification_icon_type_ = favicon_base::INVALID_ICON; |
@@ -303,23 +305,104 @@ void FaviconHandler::NotifyFaviconUpdated(const GURL& icon_url, |
if (image.IsEmpty()) |
return; |
+ // If a manifest is being processed, it's URL overrides the icon URL. |
+ const GURL& notification_icon_url = manifest_url_.value_or(icon_url); |
+ |
gfx::Image image_with_adjusted_colorspace = image; |
favicon_base::SetFaviconColorSpace(&image_with_adjusted_colorspace); |
- delegate_->OnFaviconUpdated(url_, handler_type_, icon_url, |
- icon_url != notification_icon_url_, |
+ delegate_->OnFaviconUpdated(url_, handler_type_, notification_icon_url, |
+ notification_icon_url != notification_icon_url_, |
image_with_adjusted_colorspace); |
- notification_icon_url_ = icon_url; |
+ notification_icon_url_ = notification_icon_url; |
notification_icon_type_ = icon_type; |
} |
-void FaviconHandler::OnUpdateFaviconURL( |
+void FaviconHandler::OnUpdateCandidates( |
const GURL& page_url, |
- const std::vector<FaviconURL>& candidates) { |
+ const std::vector<FaviconURL>& candidates, |
+ const base::Optional<GURL>& manifest_url) { |
+ DCHECK_EQ(page_url, url_); |
pkotwicz
2017/04/24 14:42:15
Can you please double check that the DCHECK_EQ() i
|
if (page_url != url_) |
return; |
+ // |candidates| or |manifest_url| could have been modified via Javascript. |
+ if (manifest_url_ && manifest_url_ == manifest_url) |
+ return; |
+ |
+ manifest_url_ = manifest_url; |
pkotwicz
2017/04/24 14:42:15
Would this code work correctly if the manifest URL
mastiz
2017/04/27 13:54:50
Fixed this by splitting into two CancelableTaskTra
|
+ non_manifest_original_candidates_ = candidates; |
+ |
+ // Check if the manifest previously returned a 404 (or a 200 but contained no |
+ // icons), and ignore the manifest URL if that's the case. |
+ if (manifest_url_ && service_->WasUnableToDownloadFavicon(*manifest_url_)) { |
+ DVLOG(1) << "Skip failed Manifest: " << *manifest_url_; |
+ manifest_url_.reset(); |
+ } |
+ |
+ if (manifest_url_) { |
+ // See if there is a cached favicon for the manifest. |
+ GetFaviconAndUpdateMappingsUnlessIncognito( |
+ /*icon_url=*/*manifest_url_, favicon_base::FAVICON, |
+ base::Bind(&FaviconHandler::OnFaviconDataForManifestFromFaviconService, |
+ base::Unretained(this))); |
+ } else { |
+ // If no manifest available, proceed with the regular candidates only. |
+ OnGotFinalIconURLCandidates(candidates); |
+ } |
+} |
+ |
+void FaviconHandler::OnFaviconDataForManifestFromFaviconService( |
+ const std::vector<favicon_base::FaviconRawBitmapResult>& |
+ favicon_bitmap_results) { |
+ bool has_valid_result = HasValidResult(favicon_bitmap_results); |
+ bool has_expired_or_incomplete_result = |
+ !has_valid_result || HasExpiredOrIncompleteResult(preferred_icon_size(), |
+ favicon_bitmap_results); |
+ |
+ if (has_valid_result) { |
+ // There is a valid favicon. Notify any observers. It is useful to notify |
+ // the observers even if the favicon is expired or incomplete (incorrect |
+ // size) because temporarily showing the user an expired favicon or |
+ // streched favicon is preferable to showing the user the default favicon. |
+ NotifyFaviconUpdated(favicon_bitmap_results); |
+ } |
+ |
+ if (has_expired_or_incomplete_result) { |
+ manifest_download_request_.Reset(base::Bind( |
+ &FaviconHandler::OnDidDownloadManifest, base::Unretained(this))); |
+ delegate_->DownloadManifest(*manifest_url_, |
+ manifest_download_request_.callback()); |
+ } |
+} |
+ |
+void FaviconHandler::OnDidDownloadManifest( |
+ int status_code, |
+ const std::vector<FaviconURL>& candidates) { |
+ // Mark manifest download as finished. |
+ manifest_download_request_.Cancel(); |
+ |
+ if (!candidates.empty()) { |
+ OnGotFinalIconURLCandidates(candidates); |
+ return; |
+ } |
+ |
+ // If either the downloading of the manifest failed, OR the manifest contains |
+ // no icons, proceed with the list of icons listed in the HTML. |
+ DVLOG(1) << "Could not fetch Manifest icons from " << *manifest_url_ |
+ << ", falling back to inlined ones, which are " |
+ << non_manifest_original_candidates_.size(); |
+ |
+ if (status_code == 404 || status_code == 200) |
+ service_->UnableToDownloadFavicon(*manifest_url_); |
+ |
+ manifest_url_.reset(); |
+ OnGotFinalIconURLCandidates(non_manifest_original_candidates_); |
+} |
+ |
+void FaviconHandler::OnGotFinalIconURLCandidates( |
+ const std::vector<FaviconURL>& candidates) { |
std::vector<FaviconCandidate> sorted_candidates; |
const std::vector<int> desired_pixel_sizes = |
GetDesiredPixelSizes(handler_type_); |
@@ -339,7 +422,7 @@ void FaviconHandler::OnUpdateFaviconURL( |
return; |
} |
- download_request_.Cancel(); |
+ image_download_request_.Cancel(); |
candidates_ = std::move(sorted_candidates); |
num_download_requests_ = 0; |
current_candidate_index_ = 0u; |
@@ -386,7 +469,7 @@ void FaviconHandler::OnDidDownloadFavicon( |
const std::vector<SkBitmap>& bitmaps, |
const std::vector<gfx::Size>& original_bitmap_sizes) { |
// Mark download as finished. |
- download_request_.Cancel(); |
+ image_download_request_.Cancel(); |
if (bitmaps.empty() && http_status_code == 404) { |
DVLOG(1) << "Failed to Download Favicon:" << image_url; |
@@ -436,9 +519,10 @@ void FaviconHandler::OnDidDownloadFavicon( |
RecordDownloadAttemptsForHandlerType(handler_type_, num_download_requests_); |
// We have either found the ideal candidate or run out of candidates. |
if (best_favicon_.candidate.icon_type != favicon_base::INVALID_ICON) { |
- // No more icons to request, set the favicon from the candidate. |
- SetFavicon(best_favicon_.candidate.icon_url, best_favicon_.image, |
- best_favicon_.candidate.icon_type); |
+ // No more icons to request, set the favicon from the candidate. Note that |
+ // manifest URLs override icon URLs if available. |
+ SetFavicon(manifest_url_.value_or(best_favicon_.candidate.icon_url), |
+ best_favicon_.image, best_favicon_.candidate.icon_type); |
} |
// Clear download related state. |
current_candidate_index_ = candidates_.size(); |
@@ -455,7 +539,8 @@ const std::vector<GURL> FaviconHandler::GetIconURLs() const { |
} |
bool FaviconHandler::HasPendingTasksForTest() { |
- return !download_request_.IsCancelled() || |
+ return !manifest_download_request_.IsCancelled() || |
+ !image_download_request_.IsCancelled() || |
cancelable_task_tracker_.HasTrackedTasks(); |
} |
@@ -479,15 +564,20 @@ void FaviconHandler::OnFaviconDataForInitialURLFromFaviconService( |
redownload_icons_ = initial_history_result_expired_or_incomplete_ && |
!favicon_bitmap_results.empty(); |
- if (has_valid_result && (!current_candidate() || |
- DoUrlsAndIconsMatch(current_candidate()->icon_url, |
- current_candidate()->icon_type, |
- favicon_bitmap_results))) { |
- // The db knows the favicon (although it may be out of date) and the entry |
- // doesn't have an icon. Set the favicon now, and if the favicon turns out |
- // to be expired (or the wrong url) we'll fetch later on. This way the |
- // user doesn't see a flash of the default favicon. |
- NotifyFaviconUpdated(favicon_bitmap_results); |
+ if (has_valid_result) { |
+ if (!current_candidate() || |
+ DoUrlsAndIconsMatch(current_candidate()->icon_url, |
+ current_candidate()->icon_type, |
+ favicon_bitmap_results) || |
+ (manifest_url_ && |
+ DoUrlsAndIconsMatch(*manifest_url_, favicon_base::FAVICON, |
+ favicon_bitmap_results))) { |
+ // The db knows the favicon (although it may be out of date) and the entry |
+ // doesn't have an icon. Set the favicon now, and if the favicon turns out |
+ // to be expired (or the wrong url) we'll fetch later on. This way the |
+ // user doesn't see a flash of the default favicon. |
+ NotifyFaviconUpdated(favicon_bitmap_results); |
+ } |
} |
if (current_candidate()) |
@@ -495,34 +585,42 @@ void FaviconHandler::OnFaviconDataForInitialURLFromFaviconService( |
} |
void FaviconHandler::DownloadCurrentCandidateOrAskFaviconService() { |
- GURL icon_url = current_candidate()->icon_url; |
- favicon_base::IconType icon_type = current_candidate()->icon_type; |
- |
- if (redownload_icons_) { |
+ const GURL icon_url = current_candidate()->icon_url; |
+ const favicon_base::IconType icon_type = current_candidate()->icon_type; |
+ // If the icons listed in a manifest are being processed, skip the cache |
+ // lookup for |icon_url| since the manifest's URL is used for caching anyway, |
+ // and this lookup has happened earlier. |
+ if (redownload_icons_ || manifest_url_) { |
// We have the mapping, but the favicon is out of date. Download it now. |
- ScheduleDownload(icon_url, icon_type); |
+ ScheduleFaviconDownload(icon_url, icon_type); |
} else { |
- // We don't know the favicon, but we may have previously downloaded the |
- // favicon for another page that shares the same favicon. Ask for the |
- // favicon given the favicon URL. |
- if (delegate_->IsOffTheRecord()) { |
- service_->GetFavicon( |
- icon_url, icon_type, preferred_icon_size(), |
- base::Bind(&FaviconHandler::OnFaviconData, base::Unretained(this)), |
- &cancelable_task_tracker_); |
- } else { |
- // Ask the history service for the icon. This does two things: |
- // 1. Attempts to fetch the favicon data from the database. |
- // 2. If the favicon exists in the database, this updates the database to |
- // include the mapping between the page url and the favicon url. |
- // This is asynchronous. The history service will call back when done. |
- // TODO(pkotwicz): pass in all of |image_urls_| to |
- // UpdateFaviconMappingsAndFetch(). |
- service_->UpdateFaviconMappingsAndFetch( |
- url_, {icon_url}, icon_type, preferred_icon_size(), |
- base::Bind(&FaviconHandler::OnFaviconData, base::Unretained(this)), |
- &cancelable_task_tracker_); |
- } |
+ GetFaviconAndUpdateMappingsUnlessIncognito( |
+ icon_url, icon_type, |
+ base::Bind(&FaviconHandler::OnFaviconData, base::Unretained(this))); |
+ } |
+} |
+ |
+void FaviconHandler::GetFaviconAndUpdateMappingsUnlessIncognito( |
+ const GURL& icon_url, |
+ favicon_base::IconType icon_type, |
+ const favicon_base::FaviconResultsCallback& callback) { |
+ // We don't know the favicon, but we may have previously downloaded the |
+ // favicon for another page that shares the same favicon. Ask for the |
+ // favicon given the favicon URL. |
+ if (delegate_->IsOffTheRecord()) { |
+ service_->GetFavicon(icon_url, icon_type, preferred_icon_size(), callback, |
+ &cancelable_task_tracker_); |
+ } else { |
+ // Ask the history service for the icon. This does two things: |
+ // 1. Attempts to fetch the favicon data from the database. |
+ // 2. If the favicon exists in the database, this updates the database to |
+ // include the mapping between the page url and the favicon url. |
+ // This is asynchronous. The history service will call back when done. |
+ // TODO(pkotwicz): pass in all of |image_urls_| to |
+ // UpdateFaviconMappingsAndFetch(). |
+ service_->UpdateFaviconMappingsAndFetch(url_, {icon_url}, icon_type, |
+ preferred_icon_size(), callback, |
+ &cancelable_task_tracker_); |
} |
} |
@@ -551,18 +649,19 @@ void FaviconHandler::OnFaviconData(const std::vector< |
} |
if (has_expired_or_incomplete_result) { |
- ScheduleDownload(current_candidate()->icon_url, |
- current_candidate()->icon_type); |
+ ScheduleFaviconDownload(current_candidate()->icon_url, |
+ current_candidate()->icon_type); |
} else if (num_download_requests_ > 0) { |
RecordDownloadAttemptsForHandlerType(handler_type_, num_download_requests_); |
} |
} |
-void FaviconHandler::ScheduleDownload(const GURL& image_url, |
- favicon_base::IconType icon_type) { |
+void FaviconHandler::ScheduleFaviconDownload(const GURL& image_url, |
+ favicon_base::IconType icon_type) { |
DCHECK(image_url.is_valid()); |
// Note that CancelableCallback starts cancelled. |
- DCHECK(download_request_.IsCancelled()) << "More than one ongoing download"; |
+ DCHECK(image_download_request_.IsCancelled()) |
+ << "More than one ongoing download"; |
if (service_->WasUnableToDownloadFavicon(image_url)) { |
DVLOG(1) << "Skip Failed FavIcon: " << image_url; |
RecordDownloadOutcome(DownloadOutcome::SKIPPED); |
@@ -571,14 +670,15 @@ void FaviconHandler::ScheduleDownload(const GURL& image_url, |
return; |
} |
++num_download_requests_; |
- download_request_.Reset(base::Bind(&FaviconHandler::OnDidDownloadFavicon, |
- base::Unretained(this), icon_type)); |
+ image_download_request_.Reset( |
+ base::Bind(&FaviconHandler::OnDidDownloadFavicon, base::Unretained(this), |
+ icon_type)); |
// A max bitmap size is specified to avoid receiving huge bitmaps in |
// OnDidDownloadFavicon(). See FaviconDriver::StartDownload() |
// for more details about the max bitmap size. |
const int download_id = |
delegate_->DownloadImage(image_url, GetMaximalIconSize(handler_type_), |
- download_request_.callback()); |
+ image_download_request_.callback()); |
DCHECK_NE(download_id, 0); |
} |