netwerk/test/unit/test_retry_0rtt.js

   1 /* This Source Code Form is subject to the terms of the Mozilla Public
   2  * License, v. 2.0. If a copy of the MPL was not distributed with this
   3  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   4
   5 "use strict";
   6
   7 const { HttpServer } = ChromeUtils.importESModule(
   8   "resource://testing-common/httpd.sys.mjs"
   9 );
  10 var httpServer = null;
  11
  12 let handlerCallbacks = {};
  13
  14 function listenHandler(metadata, response) {
  15   info(metadata.path);
  16   handlerCallbacks[metadata.path] = (handlerCallbacks[metadata.path] || 0) + 1;
  17 }
  18
  19 function handlerCount(path) {
  20   return handlerCallbacks[path] || 0;
  21 }
  22
  23 ChromeUtils.importESModule("resource://gre/modules/AppConstants.sys.mjs");
  24
  25 // Bug 1805371: Tests that require FaultyServer can't currently be built
  26 // with system NSS.
  27 add_setup(
  28   {
  29     skip_if: () => AppConstants.MOZ_SYSTEM_NSS,
  30   },
  31   async () => {
  32     httpServer = new HttpServer();
  33     httpServer.registerPrefixHandler("/callback/", listenHandler);
  34     httpServer.start(-1);
  35
  36     registerCleanupFunction(async () => {
  37       await httpServer.stop();
  38     });
  39
  40     Services.env.set(
  41       "FAULTY_SERVER_CALLBACK_PORT",
  42       httpServer.identity.primaryPort
  43     );
  44     Services.env.set("MOZ_TLS_SERVER_0RTT", "1");
  45     await asyncStartTLSTestServer(
  46       "FaultyServer",
  47       "../../../security/manager/ssl/tests/unit/test_faulty_server"
  48     );
  49     let nssComponent = Cc["@mozilla.org/psm;1"].getService(Ci.nsINSSComponent);
  50     await nssComponent.asyncClearSSLExternalAndInternalSessionCache();
  51   }
  52 );
  53
  54 async function sleep(time) {
  55   return new Promise(resolve => {
  56     do_timeout(time * 1000, resolve);
  57   });
  58 }
  59
  60 function makeChan(url) {
  61   let chan = NetUtil.newChannel({
  62     uri: url,
  63     loadUsingSystemPrincipal: true,
  64   }).QueryInterface(Ci.nsIHttpChannel);
  65
  66   chan.loadFlags = Ci.nsIChannel.LOAD_INITIAL_DOCUMENT_URI;
  67   return chan;
  68 }
  69
  70 function channelOpenPromise(chan, flags) {
  71   return new Promise(resolve => {
  72     chan.asyncOpen(
  73       new ChannelListener((req, buffer) => resolve([req, buffer]), null, flags)
  74     );
  75   });
  76 }
  77
  78 add_task(
  79   {
  80     skip_if: () => AppConstants.MOZ_SYSTEM_NSS,
  81   },
  82   async function testRetry0Rtt() {
  83     var retryDomains = [
  84       "0rtt-alert-bad-mac.example.com",
  85       "0rtt-alert-protocol-version.example.com",
  86       //"0rtt-alert-unexpected.example.com", // TODO(bug 1753204): uncomment this
  87     ];
  88
  89     Services.prefs.setCharPref("network.dns.localDomains", retryDomains);
  90
  91     Services.prefs.setBoolPref("network.ssl_tokens_cache_enabled", true);
  92
  93     for (var i = 0; i < retryDomains.length; i++) {
  94       {
  95         let countOfEarlyData = handlerCount("/callback/1");
  96         let chan = makeChan(`https://${retryDomains[i]}:8443`);
  97         let [, buf] = await channelOpenPromise(chan, CL_ALLOW_UNKNOWN_CL);
  98         ok(buf);
  99         equal(
 100           handlerCount("/callback/1"),
 101           countOfEarlyData,
 102           "no early data sent"
 103         );
 104       }
 105
 106       // The server has an anti-replay mechanism that prohibits it from
 107       // accepting 0-RTT connections immediately at startup.
 108       await sleep(1);
 109
 110       {
 111         let countOfEarlyData = handlerCount("/callback/1");
 112         let chan = makeChan(`https://${retryDomains[i]}:8443`);
 113         let [, buf] = await channelOpenPromise(chan, CL_ALLOW_UNKNOWN_CL);
 114         ok(buf);
 115         equal(
 116           handlerCount("/callback/1"),
 117           countOfEarlyData + 1,
 118           "got early data"
 119         );
 120       }
 121     }
 122   }
 123 );