Bug 977177 - Invalidate the page-icon image cache when necessary. r=adw draft
authorMarco Bonardo <mbonardo@mozilla.com>
Fri, 31 Mar 2017 17:03:25 +0200
changeset 561157 64266310cf5c025f7c055fb7ca141bbe3e82c9fe
parent 561156 d3d59b35ab9a7f2be7c32bdd859b88bf62d1d59b
child 623901 656a0ace32568e5026cd2935aa31b001efe377af
push id53651
push usermak77@bonardo.net
push dateWed, 12 Apr 2017 09:15:32 +0000
reviewersadw
bugs977177
milestone55.0a1
Bug 977177 - Invalidate the page-icon image cache when necessary. r=adw Remove page-icon entries when the icon changes, so that we start showing the new content. Also adds an SVG page-icon test. MozReview-Commit-ID: 10MIOvwbQ20
toolkit/components/places/nsFaviconService.cpp
toolkit/components/places/tests/favicons/test_page-icon_protocol.js
--- a/toolkit/components/places/nsFaviconService.cpp
+++ b/toolkit/components/places/nsFaviconService.cpp
@@ -27,16 +27,17 @@
 #include "nsIClassInfoImpl.h"
 #include "mozilla/ArrayUtils.h"
 #include "mozilla/LoadInfo.h"
 #include "mozilla/Preferences.h"
 #include "nsILoadInfo.h"
 #include "nsIContentPolicy.h"
 #include "nsContentUtils.h"
 #include "NullPrincipal.h"
+#include "imgICache.h"
 
 #define MAX_FAILED_FAVICONS 256
 #define FAVICON_CACHE_REDUCE_COUNT 64
 
 #define UNASSOCIATED_FAVICONS_LENGTH 32
 
 // When replaceFaviconData is called, we store the icons in an in-memory cache
 // instead of in storage. Icons in the cache are expired according to this
@@ -255,16 +256,36 @@ nsFaviconService::GetDefaultFavicon(nsIU
 void
 nsFaviconService::SendFaviconNotifications(nsIURI* aPageURI,
                                            nsIURI* aFaviconURI,
                                            const nsACString& aGUID)
 {
   nsAutoCString faviconSpec;
   nsNavHistory* history = nsNavHistory::GetHistoryService();
   if (history && NS_SUCCEEDED(aFaviconURI->GetSpec(faviconSpec))) {
+    // Invalide page-icon image cache, since the icon is about to change.
+    nsCString spec;
+    nsresult rv = aPageURI->GetSpec(spec);
+    MOZ_ASSERT(NS_SUCCEEDED(rv));
+    if (NS_SUCCEEDED(rv)) {
+      nsCString pageIconSpec("page-icon:");
+      pageIconSpec.Append(spec);
+      nsCOMPtr<nsIURI> pageIconURI;
+      rv = NS_NewURI(getter_AddRefs(pageIconURI), pageIconSpec);
+      MOZ_ASSERT(NS_SUCCEEDED(rv));
+      if (NS_SUCCEEDED(rv)) {
+        nsCOMPtr<imgICache> imgCache;
+        rv = GetImgTools()->GetImgCacheForDocument(nullptr, getter_AddRefs(imgCache));
+        MOZ_ASSERT(NS_SUCCEEDED(rv));
+        if (NS_SUCCEEDED(rv)) {
+          Unused << imgCache->RemoveEntry(pageIconURI, nullptr);
+        }
+      }
+    }
+
     history->SendPageChangedNotification(aPageURI,
                                          nsINavHistoryObserver::ATTRIBUTE_FAVICON,
                                          NS_ConvertUTF8toUTF16(faviconSpec),
                                          aGUID);
   }
 }
 
 NS_IMETHODIMP
--- a/toolkit/components/places/tests/favicons/test_page-icon_protocol.js
+++ b/toolkit/components/places/tests/favicons/test_page-icon_protocol.js
@@ -1,16 +1,16 @@
-const ICON_DATA = "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAAAAAA6fptVAAAACklEQVQI12NgAAAAAgAB4iG8MwAAAABJRU5ErkJggg==";
+const ICON_DATAURL = "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAAAAAA6fptVAAAACklEQVQI12NgAAAAAgAB4iG8MwAAAABJRU5ErkJggg==";
 const TEST_URI = NetUtil.newURI("http://mozilla.org/");
 const ICON_URI = NetUtil.newURI("http://mozilla.org/favicon.ico");
 
 function fetchIconForSpec(spec) {
  return new Promise((resolve, reject) => {
     NetUtil.asyncFetch({
-      uri: NetUtil.newURI("page-icon:" + TEST_URI.spec),
+      uri: NetUtil.newURI(spec),
       loadUsingSystemPrincipal: true,
       contentPolicyType: Ci.nsIContentPolicy.TYPE_INTERNAL_IMAGE_FAVICON
     }, (input, status, request) => {
        if (!Components.isSuccessCode(status)) {
         reject(new Error("unable to load icon"));
         return;
       }
 
@@ -25,48 +25,60 @@ function fetchIconForSpec(spec) {
     });
   });
 }
 
 var gDefaultFavicon;
 var gFavicon;
 
 add_task(function* setup() {
-  yield PlacesTestUtils.addVisits({ uri: TEST_URI });
+  yield PlacesTestUtils.addVisits(TEST_URI);
 
   PlacesUtils.favicons.replaceFaviconDataFromDataURL(
-    ICON_URI, ICON_DATA, (Date.now() + 8640000) * 1000,
+    ICON_URI, ICON_DATAURL, (Date.now() + 8640000) * 1000,
     Services.scriptSecurityManager.getSystemPrincipal());
 
   yield new Promise(resolve => {
     PlacesUtils.favicons.setAndFetchFaviconForPage(
       TEST_URI, ICON_URI, false,
       PlacesUtils.favicons.FAVICON_LOAD_NON_PRIVATE,
       resolve, Services.scriptSecurityManager.getSystemPrincipal());
   });
 
-  gDefaultFavicon = yield fetchIconForSpec(PlacesUtils.favicons.defaultFavicon);
-  gFavicon = yield fetchIconForSpec(ICON_DATA);
+  gDefaultFavicon = yield fetchIconForSpec(PlacesUtils.favicons.defaultFavicon.spec);
+  gFavicon = yield fetchIconForSpec(ICON_DATAURL);
 });
 
 add_task(function* known_url() {
-  let {data, contentType} = yield fetchIconForSpec(TEST_URI.spec);
+  let {data, contentType} = yield fetchIconForSpec("page-icon:" + TEST_URI.spec);
   Assert.equal(contentType, gFavicon.contentType);
   Assert.deepEqual(data, gFavicon.data, "Got the favicon data");
 });
 
 add_task(function* unknown_url() {
-  let {data, contentType} = yield fetchIconForSpec("http://www.moz.org/");
+  let {data, contentType} = yield fetchIconForSpec("page-icon:http://www.moz.org/");
   Assert.equal(contentType, gDefaultFavicon.contentType);
   Assert.deepEqual(data, gDefaultFavicon.data, "Got the default favicon data");
 });
 
 add_task(function* invalid_url() {
-  let {data, contentType} = yield fetchIconForSpec("test");
+  let {data, contentType} = yield fetchIconForSpec("page-icon:test");
   Assert.equal(contentType, gDefaultFavicon.contentType);
   Assert.ok(data == gDefaultFavicon.data, "Got the default favicon data");
 });
 
 add_task(function* subpage_url_fallback() {
-  let {data, contentType} = yield fetchIconForSpec("http://mozilla.org/missing");
+  let {data, contentType} = yield fetchIconForSpec("page-icon:http://mozilla.org/missing");
   Assert.equal(contentType, gFavicon.contentType);
   Assert.deepEqual(data, gFavicon.data, "Got the root favicon data");
 });
+
+add_task(function* svg_icon() {
+  let faviconURI = NetUtil.newURI("http://places.test/favicon.svg");
+  PlacesUtils.favicons.replaceFaviconDataFromDataURL(
+    faviconURI, SMALLSVG_DATA_URI.spec, 0, Services.scriptSecurityManager.getSystemPrincipal());
+  yield setFaviconForPage(TEST_URI, faviconURI);
+  let svgIcon = yield fetchIconForSpec(SMALLSVG_DATA_URI.spec);
+  do_print(svgIcon.contentType)
+  let pageIcon = yield fetchIconForSpec("page-icon:" + TEST_URI.spec);
+  Assert.equal(svgIcon.contentType, pageIcon.contentType);
+  Assert.deepEqual(svgIcon.data, pageIcon.data, "Got the root favicon data");
+});