/* Any copyright is dedicated to the Public Domain. * http://creativecommons.org/publicdomain/zero/1.0/ */ "use strict"; const { CookieXPCShellUtils } = ChromeUtils.importESModule( "resource://testing-common/CookieXPCShellUtils.sys.mjs" ); // Small red image. const IMG_BYTES = atob( "iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAYAAACNbyblAAAAHElEQVQI12" + "P4//8/w38GIAXDIBKE0DHxgljNBAAO9TXL0Y4OHwAAAABJRU5ErkJggg==" ); let gHints = 0; CookieXPCShellUtils.init(this); function countMatchingCacheEntries(cacheEntries, domain, path) { return cacheEntries .map(entry => entry.uri.asciiSpec) .filter(spec => spec.includes(domain)) .filter(spec => spec.includes(path)).length; } async function checkCache(originAttributes) { const loadContextInfo = Services.loadContextInfo.custom( false, originAttributes ); const data = await new Promise(resolve => { let cacheEntries = []; let cacheVisitor = { onCacheStorageInfo(num, consumption) {}, onCacheEntryInfo(uri, idEnhance) { cacheEntries.push({ uri, idEnhance }); }, onCacheEntryVisitCompleted() { resolve(cacheEntries); }, QueryInterface: ChromeUtils.generateQI(["nsICacheStorageVisitor"]), }; // Visiting the disk cache also visits memory storage so we do not // need to use Services.cache2.memoryCacheStorage() here. let storage = Services.cache2.diskCacheStorage(loadContextInfo); storage.asyncVisitStorage(cacheVisitor, true); }); let foundEntryCount = countMatchingCacheEntries( data, "example.org", "image.png" ); ok( foundEntryCount > 0, `Cache entries expected for image.png and OA=${originAttributes}` ); } add_task(async () => { do_get_profile(); Services.prefs.setBoolPref("network.prefetch-next", true); Services.prefs.setIntPref("network.cookie.cookieBehavior", 0); const server = CookieXPCShellUtils.createServer({ hosts: ["example.org", "foo.com"], }); server.registerPathHandler("/image.png", (metadata, response) => { gHints++; response.setHeader("Cache-Control", "max-age=10000", false); response.setStatusLine(metadata.httpVersion, 200, "OK"); response.setHeader("Content-Type", "image/png", false); response.write(IMG_BYTES); }); server.registerPathHandler("/prefetch", (metadata, response) => { response.setStatusLine(metadata.httpVersion, 200, "OK"); response.setHeader("Content-Type", "text/html", false); var body = ``; response.bodyOutputStream.write(body, body.length); }); const tests = [ { // 2 hints because we have 2 different top-level origins, loading the // same resource. This will end up creating 2 separate cache entries. hints: 2, originAttributes: { partitionKey: "(http,example.org)" }, prefValue: true, }, { // 1 hint because, with network-state isolation, the cache entry will be // reused for the second loading, even if the top-level origins are // different. hints: 1, originAttributes: {}, prefValue: false, }, ]; for (let test of tests) { await new Promise(resolve => Services.clearData.deleteData(Ci.nsIClearDataService.CLEAR_ALL, resolve) ); info("Reset the counter"); gHints = 0; info("Enabling network state partitioning"); Services.prefs.setBoolPref( "privacy.partition.network_state", test.prefValue ); let complete = new Promise(resolve => { server.registerPathHandler("/done", (metadata, response) => { response.setHeader("Cache-Control", "max-age=10000", false); response.setStatusLine(metadata.httpVersion, 200, "OK"); response.setHeader("Content-Type", "text/html", false); var body = "OK"; response.bodyOutputStream.write(body, body.length); resolve(); }); }); info("Let's load a page with origin A"); let contentPage = await CookieXPCShellUtils.loadContentPage( "http://example.org/prefetch" ); await complete; await checkCache(test.originAttributes); await contentPage.close(); complete = new Promise(resolve => { server.registerPathHandler("/done", (metadata, response) => { response.setHeader("Cache-Control", "max-age=10000", false); response.setStatusLine(metadata.httpVersion, 200, "OK"); response.setHeader("Content-Type", "text/html", false); var body = "OK"; response.bodyOutputStream.write(body, body.length); resolve(); }); }); info("Let's load a page with origin B"); contentPage = await CookieXPCShellUtils.loadContentPage( "http://foo.com/prefetch" ); await complete; await checkCache(test.originAttributes); await contentPage.close(); Assert.equal( gHints, test.hints, "We have the current number of requests with pref " + test.prefValue ); } });