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
3 // found in the LICENSE file.
5 #include "chrome/browser/favicon/favicon_handler.h"
7 #include "build/build_config.h"
12 #include "base/bind.h"
13 #include "base/bind_helpers.h"
14 #include "base/memory/ref_counted_memory.h"
15 #include "chrome/browser/bookmarks/bookmark_service.h"
16 #include "chrome/browser/favicon/favicon_service_factory.h"
17 #include "chrome/browser/favicon/favicon_util.h"
18 #include "chrome/browser/history/select_favicon_frames.h"
19 #include "chrome/browser/profiles/profile.h"
20 #include "content/public/browser/favicon_status.h"
21 #include "content/public/browser/navigation_entry.h"
22 #include "skia/ext/image_operations.h"
23 #include "ui/gfx/codec/png_codec.h"
24 #include "ui/gfx/image/image.h"
25 #include "ui/gfx/image/image_skia.h"
26 #include "ui/gfx/image/image_util.h"
28 using content::FaviconURL
;
29 using content::NavigationEntry
;
33 // Size (along each axis) of a touch icon. This currently corresponds to
34 // the apple touch icon for iPad.
35 const int kTouchIconSize
= 144;
37 // Returns chrome::IconType the given icon_type corresponds to.
38 chrome::IconType
ToHistoryIconType(FaviconURL::IconType icon_type
) {
40 case FaviconURL::FAVICON
:
41 return chrome::FAVICON
;
42 case FaviconURL::TOUCH_ICON
:
43 return chrome::TOUCH_ICON
;
44 case FaviconURL::TOUCH_PRECOMPOSED_ICON
:
45 return chrome::TOUCH_PRECOMPOSED_ICON
;
46 case FaviconURL::INVALID_ICON
:
47 return chrome::INVALID_ICON
;
50 return chrome::INVALID_ICON
;
53 // Get the maximal icon size in pixels for a icon of type |icon_type| for the
55 int GetMaximalIconSize(chrome::IconType icon_type
) {
58 #if defined(OS_ANDROID)
61 return gfx::ImageSkia::GetMaxSupportedScale() * gfx::kFaviconSize
;
63 case chrome::TOUCH_ICON
:
64 case chrome::TOUCH_PRECOMPOSED_ICON
:
65 return kTouchIconSize
;
66 case chrome::INVALID_ICON
:
73 bool DoUrlAndIconMatch(const FaviconURL
& favicon_url
,
75 chrome::IconType icon_type
) {
76 return favicon_url
.icon_url
== url
&&
77 favicon_url
.icon_type
== static_cast<FaviconURL::IconType
>(icon_type
);
80 // Returns true if all of the icon URLs and icon types in |bitmap_results| are
81 // identical and if they match the icon URL and icon type in |favicon_url|.
82 // Returns false if |bitmap_results| is empty.
83 bool DoUrlsAndIconsMatch(
84 const FaviconURL
& favicon_url
,
85 const std::vector
<chrome::FaviconBitmapResult
>& bitmap_results
) {
86 if (bitmap_results
.empty())
89 chrome::IconType icon_type
= ToHistoryIconType(favicon_url
.icon_type
);
91 for (size_t i
= 0; i
< bitmap_results
.size(); ++i
) {
92 if (favicon_url
.icon_url
!= bitmap_results
[i
].icon_url
||
93 icon_type
!= bitmap_results
[i
].icon_type
) {
100 std::string
UrlWithoutFragment(const GURL
& gurl
) {
101 GURL::Replacements replacements
;
102 replacements
.ClearRef();
103 return gurl
.ReplaceComponents(replacements
).spec();
106 bool UrlMatches(const GURL
& gurl_a
, const GURL
& gurl_b
) {
107 return UrlWithoutFragment(gurl_a
) == UrlWithoutFragment(gurl_b
);
110 // Return true if |bitmap_result| is expired.
111 bool IsExpired(const chrome::FaviconBitmapResult
& bitmap_result
) {
112 return bitmap_result
.expired
;
115 // Return true if |bitmap_result| is valid.
116 bool IsValid(const chrome::FaviconBitmapResult
& bitmap_result
) {
117 return bitmap_result
.is_valid();
120 // Returns true if at least one of the bitmaps in |bitmap_results| is expired or
121 // if |bitmap_results| is missing favicons for |desired_size_in_dip| and one of
122 // the scale factors in FaviconUtil::GetFaviconScaleFactors().
123 bool HasExpiredOrIncompleteResult(
124 int desired_size_in_dip
,
125 const std::vector
<chrome::FaviconBitmapResult
>& bitmap_results
) {
126 // Check if at least one of the bitmaps is expired.
127 std::vector
<chrome::FaviconBitmapResult
>::const_iterator it
=
128 std::find_if(bitmap_results
.begin(), bitmap_results
.end(), IsExpired
);
129 if (it
!= bitmap_results
.end())
132 // Any favicon size is good if the desired size is 0.
133 if (desired_size_in_dip
== 0)
136 // Check if the favicon for at least one of the scale factors is missing.
137 // |bitmap_results| should always be complete for data inserted by
138 // FaviconHandler as the FaviconHandler stores favicons resized to all
139 // of FaviconUtil::GetFaviconScaleFactors() into the history backend.
140 // Examples of when |bitmap_results| can be incomplete:
141 // - Favicons inserted into the history backend by sync.
142 // - Favicons for imported bookmarks.
143 std::vector
<gfx::Size
> favicon_sizes
;
144 for (size_t i
= 0; i
< bitmap_results
.size(); ++i
)
145 favicon_sizes
.push_back(bitmap_results
[i
].pixel_size
);
147 std::vector
<ui::ScaleFactor
> scale_factors
=
148 FaviconUtil::GetFaviconScaleFactors();
149 for (size_t i
= 0; i
< scale_factors
.size(); ++i
) {
150 int edge_size_in_pixel
= floor(
151 desired_size_in_dip
* ui::GetImageScale(scale_factors
[i
]));
152 std::vector
<gfx::Size
>::iterator it
= std::find(favicon_sizes
.begin(),
153 favicon_sizes
.end(), gfx::Size(edge_size_in_pixel
, edge_size_in_pixel
));
154 if (it
== favicon_sizes
.end())
160 // Returns true if at least one of |bitmap_results| is valid.
162 const std::vector
<chrome::FaviconBitmapResult
>& bitmap_results
) {
163 std::vector
<chrome::FaviconBitmapResult
>::const_iterator it
=
164 std::find_if(bitmap_results
.begin(), bitmap_results
.end(), IsValid
);
165 return it
!= bitmap_results
.end();
170 ////////////////////////////////////////////////////////////////////////////////
172 FaviconHandler::DownloadRequest::DownloadRequest()
173 : icon_type(chrome::INVALID_ICON
) {
176 FaviconHandler::DownloadRequest::~DownloadRequest() {
179 FaviconHandler::DownloadRequest::DownloadRequest(
181 const GURL
& image_url
,
182 chrome::IconType icon_type
)
184 image_url(image_url
),
185 icon_type(icon_type
) {
188 ////////////////////////////////////////////////////////////////////////////////
190 FaviconHandler::FaviconCandidate::FaviconCandidate()
192 icon_type(chrome::INVALID_ICON
) {
195 FaviconHandler::FaviconCandidate::~FaviconCandidate() {
198 FaviconHandler::FaviconCandidate::FaviconCandidate(
200 const GURL
& image_url
,
201 const gfx::Image
& image
,
203 chrome::IconType icon_type
)
205 image_url(image_url
),
208 icon_type(icon_type
) {
211 ////////////////////////////////////////////////////////////////////////////////
213 FaviconHandler::FaviconHandler(Profile
* profile
,
214 FaviconHandlerDelegate
* delegate
,
216 : got_favicon_from_history_(false),
217 favicon_expired_or_incomplete_(false),
218 icon_types_(icon_type
== FAVICON
? chrome::FAVICON
:
219 chrome::TOUCH_ICON
| chrome::TOUCH_PRECOMPOSED_ICON
),
221 delegate_(delegate
) {
226 FaviconHandler::~FaviconHandler() {
229 void FaviconHandler::FetchFavicon(const GURL
& url
) {
230 cancelable_task_tracker_
.TryCancelAll();
234 favicon_expired_or_incomplete_
= got_favicon_from_history_
= false;
237 // Request the favicon from the history service. In parallel to this the
238 // renderer is going to notify us (well WebContents) when the favicon url is
240 if (GetFaviconService()) {
244 base::Bind(&FaviconHandler::OnFaviconDataForInitialURL
,
245 base::Unretained(this)),
246 &cancelable_task_tracker_
);
250 FaviconService
* FaviconHandler::GetFaviconService() {
251 return FaviconServiceFactory::GetForProfile(
252 profile_
, Profile::EXPLICIT_ACCESS
);
255 bool FaviconHandler::UpdateFaviconCandidate(const GURL
& url
,
256 const GURL
& image_url
,
257 const gfx::Image
& image
,
259 chrome::IconType icon_type
) {
260 bool update_candidate
= false;
261 bool exact_match
= score
== 1;
262 if (preferred_icon_size() == 0) {
263 // No preferred size, use this icon.
264 update_candidate
= true;
266 } else if (favicon_candidate_
.icon_type
== chrome::INVALID_ICON
) {
267 // No current candidate, use this.
268 update_candidate
= true;
271 // Exact match, use this.
272 update_candidate
= true;
274 // Compare against current candidate.
275 if (score
> favicon_candidate_
.score
)
276 update_candidate
= true;
279 if (update_candidate
) {
280 favicon_candidate_
= FaviconCandidate(
281 url
, image_url
, image
, score
, icon_type
);
286 void FaviconHandler::SetFavicon(
288 const GURL
& icon_url
,
289 const gfx::Image
& image
,
290 chrome::IconType icon_type
) {
291 if (GetFaviconService() && ShouldSaveFavicon(url
))
292 SetHistoryFavicons(url
, icon_url
, icon_type
, image
);
294 if (UrlMatches(url
, url_
) && icon_type
== chrome::FAVICON
) {
295 NavigationEntry
* entry
= GetEntry();
297 UpdateFavicon(entry
, icon_url
, image
);
301 void FaviconHandler::UpdateFavicon(NavigationEntry
* entry
,
302 const std::vector
<chrome::FaviconBitmapResult
>& favicon_bitmap_results
) {
303 gfx::Image resized_image
= FaviconUtil::SelectFaviconFramesFromPNGs(
304 favicon_bitmap_results
,
305 FaviconUtil::GetFaviconScaleFactors(),
306 preferred_icon_size());
307 // The history service sends back results for a single icon URL, so it does
308 // not matter which result we get the |icon_url| from.
309 const GURL icon_url
= favicon_bitmap_results
.empty() ?
310 GURL() : favicon_bitmap_results
[0].icon_url
;
311 UpdateFavicon(entry
, icon_url
, resized_image
);
314 void FaviconHandler::UpdateFavicon(NavigationEntry
* entry
,
315 const GURL
& icon_url
,
316 const gfx::Image
& image
) {
317 // No matter what happens, we need to mark the favicon as being set.
318 entry
->GetFavicon().valid
= true;
320 bool icon_url_changed
= (entry
->GetFavicon().url
!= icon_url
);
321 entry
->GetFavicon().url
= icon_url
;
326 gfx::Image image_with_adjusted_colorspace
= image
;
327 FaviconUtil::SetFaviconColorSpace(&image_with_adjusted_colorspace
);
329 entry
->GetFavicon().image
= image_with_adjusted_colorspace
;
330 NotifyFaviconUpdated(icon_url_changed
);
333 void FaviconHandler::OnUpdateFaviconURL(
335 const std::vector
<FaviconURL
>& candidates
) {
338 favicon_candidate_
= FaviconCandidate();
339 for (std::vector
<FaviconURL
>::const_iterator i
= candidates
.begin();
340 i
!= candidates
.end(); ++i
) {
341 if (!i
->icon_url
.is_empty() && (i
->icon_type
& icon_types_
))
342 image_urls_
.push_back(*i
);
345 // TODO(davemoore) Should clear on empty url. Currently we ignore it.
346 // This appears to be what FF does as well.
347 if (image_urls_
.empty())
350 if (!GetFaviconService())
356 void FaviconHandler::ProcessCurrentUrl() {
357 DCHECK(!image_urls_
.empty());
359 NavigationEntry
* entry
= GetEntry();
364 if (current_candidate()->icon_type
== FaviconURL::FAVICON
) {
365 if (!favicon_expired_or_incomplete_
&& entry
->GetFavicon().valid
&&
366 DoUrlAndIconMatch(*current_candidate(), entry
->GetFavicon().url
,
369 } else if (!favicon_expired_or_incomplete_
&& got_favicon_from_history_
&&
370 HasValidResult(history_results_
) &&
371 DoUrlsAndIconsMatch(*current_candidate(), history_results_
)) {
375 if (got_favicon_from_history_
)
376 DownloadFaviconOrAskHistory(entry
->GetURL(), current_candidate()->icon_url
,
377 ToHistoryIconType(current_candidate()->icon_type
));
380 void FaviconHandler::OnDidDownloadFavicon(
382 const GURL
& image_url
,
383 const std::vector
<SkBitmap
>& bitmaps
,
384 const std::vector
<gfx::Size
>& original_bitmap_sizes
) {
385 DownloadRequests::iterator i
= download_requests_
.find(id
);
386 if (i
== download_requests_
.end()) {
387 // Currently WebContents notifies us of ANY downloads so that it is
388 // possible to get here.
392 if (current_candidate() &&
393 DoUrlAndIconMatch(*current_candidate(), image_url
, i
->second
.icon_type
)) {
395 std::vector
<ui::ScaleFactor
> scale_factors
=
396 FaviconUtil::GetFaviconScaleFactors();
397 gfx::Image
image(SelectFaviconFrames(bitmaps
,
398 original_bitmap_sizes
,
400 preferred_icon_size(),
403 // The downloaded icon is still valid when there is no FaviconURL update
404 // during the downloading.
405 bool request_next_icon
= true;
406 if (!bitmaps
.empty()) {
407 request_next_icon
= !UpdateFaviconCandidate(
408 i
->second
.url
, image_url
, image
, score
, i
->second
.icon_type
);
410 if (request_next_icon
&& GetEntry() && image_urls_
.size() > 1) {
411 // Remove the first member of image_urls_ and process the remaining.
412 image_urls_
.pop_front();
414 } else if (favicon_candidate_
.icon_type
!= chrome::INVALID_ICON
) {
415 // No more icons to request, set the favicon from the candidate.
416 SetFavicon(favicon_candidate_
.url
,
417 favicon_candidate_
.image_url
,
418 favicon_candidate_
.image
,
419 favicon_candidate_
.icon_type
);
422 favicon_candidate_
= FaviconCandidate();
425 download_requests_
.erase(i
);
428 NavigationEntry
* FaviconHandler::GetEntry() {
429 NavigationEntry
* entry
= delegate_
->GetActiveEntry();
430 if (entry
&& UrlMatches(entry
->GetURL(), url_
))
433 // If the URL has changed out from under us (as will happen with redirects)
438 int FaviconHandler::DownloadFavicon(const GURL
& image_url
,
439 int max_bitmap_size
) {
440 if (!image_url
.is_valid()) {
444 return delegate_
->StartDownload(image_url
, max_bitmap_size
);
447 void FaviconHandler::UpdateFaviconMappingAndFetch(
448 const GURL
& page_url
,
449 const GURL
& icon_url
,
450 chrome::IconType icon_type
,
451 const FaviconService::FaviconResultsCallback
& callback
,
452 CancelableTaskTracker
* tracker
) {
453 // TODO(pkotwicz): pass in all of |image_urls_| to
454 // UpdateFaviconMappingsAndFetch().
455 std::vector
<GURL
> icon_urls
;
456 icon_urls
.push_back(icon_url
);
457 GetFaviconService()->UpdateFaviconMappingsAndFetch(
458 page_url
, icon_urls
, icon_type
, preferred_icon_size(), callback
, tracker
);
461 void FaviconHandler::GetFavicon(
462 const GURL
& icon_url
,
463 chrome::IconType icon_type
,
464 const FaviconService::FaviconResultsCallback
& callback
,
465 CancelableTaskTracker
* tracker
) {
466 GetFaviconService()->GetFavicon(
467 icon_url
, icon_type
, preferred_icon_size(), callback
, tracker
);
470 void FaviconHandler::GetFaviconForURL(
471 const GURL
& page_url
,
473 const FaviconService::FaviconResultsCallback
& callback
,
474 CancelableTaskTracker
* tracker
) {
475 GetFaviconService()->GetFaviconForURL(
476 FaviconService::FaviconForURLParams(page_url
, icon_types
,
477 preferred_icon_size()),
482 void FaviconHandler::SetHistoryFavicons(const GURL
& page_url
,
483 const GURL
& icon_url
,
484 chrome::IconType icon_type
,
485 const gfx::Image
& image
) {
486 GetFaviconService()->SetFavicons(page_url
, icon_url
, icon_type
, image
);
489 bool FaviconHandler::ShouldSaveFavicon(const GURL
& url
) {
490 if (!profile_
->IsOffTheRecord())
493 // Otherwise store the favicon if the page is bookmarked.
494 BookmarkService
* bookmark_service
=
495 BookmarkService::FromBrowserContext(profile_
);
496 return bookmark_service
&& bookmark_service
->IsBookmarked(url
);
499 void FaviconHandler::NotifyFaviconUpdated(bool icon_url_changed
) {
500 delegate_
->NotifyFaviconUpdated(icon_url_changed
);
503 void FaviconHandler::OnFaviconDataForInitialURL(
504 const std::vector
<chrome::FaviconBitmapResult
>& favicon_bitmap_results
) {
505 NavigationEntry
* entry
= GetEntry();
509 got_favicon_from_history_
= true;
510 history_results_
= favicon_bitmap_results
;
512 bool has_results
= !favicon_bitmap_results
.empty();
513 favicon_expired_or_incomplete_
= has_results
&& HasExpiredOrIncompleteResult(
514 preferred_icon_size(), favicon_bitmap_results
);
516 if (has_results
&& icon_types_
== chrome::FAVICON
&&
517 !entry
->GetFavicon().valid
&&
518 (!current_candidate() ||
519 DoUrlsAndIconsMatch(*current_candidate(), favicon_bitmap_results
))) {
520 if (HasValidResult(favicon_bitmap_results
)) {
521 // The db knows the favicon (although it may be out of date) and the entry
522 // doesn't have an icon. Set the favicon now, and if the favicon turns out
523 // to be expired (or the wrong url) we'll fetch later on. This way the
524 // user doesn't see a flash of the default favicon.
525 UpdateFavicon(entry
, favicon_bitmap_results
);
527 // If |favicon_bitmap_results| does not have any valid results, treat the
528 // favicon as if it's expired.
529 // TODO(pkotwicz): Do something better.
530 favicon_expired_or_incomplete_
= true;
534 if (has_results
&& !favicon_expired_or_incomplete_
) {
535 if (current_candidate() &&
536 !DoUrlsAndIconsMatch(*current_candidate(), favicon_bitmap_results
)) {
537 // Mapping in the database is wrong. DownloadFavIconOrAskHistory will
538 // update the mapping for this url and download the favicon if we don't
540 DownloadFaviconOrAskHistory(entry
->GetURL(),
541 current_candidate()->icon_url
,
542 static_cast<chrome::IconType
>(current_candidate()->icon_type
));
544 } else if (current_candidate()) {
545 // We know the official url for the favicon, by either don't have the
546 // favicon or its expired. Continue on to DownloadFaviconOrAskHistory to
547 // either download or check history again.
548 DownloadFaviconOrAskHistory(entry
->GetURL(), current_candidate()->icon_url
,
549 ToHistoryIconType(current_candidate()->icon_type
));
551 // else we haven't got the icon url. When we get it we'll ask the
552 // renderer to download the icon.
555 void FaviconHandler::DownloadFaviconOrAskHistory(
556 const GURL
& page_url
,
557 const GURL
& icon_url
,
558 chrome::IconType icon_type
) {
559 if (favicon_expired_or_incomplete_
) {
560 // We have the mapping, but the favicon is out of date. Download it now.
561 ScheduleDownload(page_url
, icon_url
, icon_type
);
562 } else if (GetFaviconService()) {
563 // We don't know the favicon, but we may have previously downloaded the
564 // favicon for another page that shares the same favicon. Ask for the
565 // favicon given the favicon URL.
566 if (profile_
->IsOffTheRecord()) {
569 base::Bind(&FaviconHandler::OnFaviconData
, base::Unretained(this)),
570 &cancelable_task_tracker_
);
572 // Ask the history service for the icon. This does two things:
573 // 1. Attempts to fetch the favicon data from the database.
574 // 2. If the favicon exists in the database, this updates the database to
575 // include the mapping between the page url and the favicon url.
576 // This is asynchronous. The history service will call back when done.
577 // Issue the request and associate the current page ID with it.
578 UpdateFaviconMappingAndFetch(
579 page_url
, icon_url
, icon_type
,
580 base::Bind(&FaviconHandler::OnFaviconData
, base::Unretained(this)),
581 &cancelable_task_tracker_
);
586 void FaviconHandler::OnFaviconData(
587 const std::vector
<chrome::FaviconBitmapResult
>& favicon_bitmap_results
) {
588 NavigationEntry
* entry
= GetEntry();
592 bool has_results
= !favicon_bitmap_results
.empty();
593 bool has_expired_or_incomplete_result
= HasExpiredOrIncompleteResult(
594 preferred_icon_size(), favicon_bitmap_results
);
596 if (has_results
&& icon_types_
== chrome::FAVICON
) {
597 if (HasValidResult(favicon_bitmap_results
)) {
598 // There is a favicon, set it now. If expired we'll download the current
599 // one again, but at least the user will get some icon instead of the
600 // default and most likely the current one is fine anyway.
601 UpdateFavicon(entry
, favicon_bitmap_results
);
603 if (has_expired_or_incomplete_result
) {
604 // The favicon is out of date. Request the current one.
605 ScheduleDownload(entry
->GetURL(), entry
->GetFavicon().url
,
608 } else if (current_candidate() &&
609 (!has_results
|| has_expired_or_incomplete_result
||
610 !(DoUrlsAndIconsMatch(*current_candidate(), favicon_bitmap_results
)))) {
611 // We don't know the favicon, it is out of date or its type is not same as
612 // one got from page. Request the current one.
613 ScheduleDownload(entry
->GetURL(), current_candidate()->icon_url
,
614 ToHistoryIconType(current_candidate()->icon_type
));
616 history_results_
= favicon_bitmap_results
;
619 int FaviconHandler::ScheduleDownload(
621 const GURL
& image_url
,
622 chrome::IconType icon_type
) {
623 // A max bitmap size is specified to avoid receiving huge bitmaps in
624 // OnDidDownloadFavicon(). See FaviconHandlerDelegate::StartDownload()
625 // for more details about the max bitmap size.
626 const int download_id
= DownloadFavicon(image_url
,
627 GetMaximalIconSize(icon_type
));
629 // Download ids should be unique.
630 DCHECK(download_requests_
.find(download_id
) == download_requests_
.end());
631 download_requests_
[download_id
] =
632 DownloadRequest(url
, image_url
, icon_type
);