tor-browser

The Tor Browser
git clone https://git.dasho.dev/tor-browser.git
Log | Files | Refs | README | LICENSE

test_localhost_offline.js (2218B)

      1 "use strict";
      2 
      3 const { HttpServer } = ChromeUtils.importESModule(
      4  "resource://testing-common/httpd.sys.mjs"
      5 );
      6 var httpServer = null;
      7 const body = "Hello";
      8 
      9 function makeChan(url) {
     10  let chan = NetUtil.newChannel({
     11    uri: url,
     12    loadUsingSystemPrincipal: true,
     13  }).QueryInterface(Ci.nsIHttpChannel);
     14  chan.loadFlags |= Ci.nsIRequest.LOAD_BYPASS_CACHE;
     15  chan.loadFlags |= Ci.nsIRequest.INHIBIT_CACHING;
     16  return chan;
     17 }
     18 
     19 function channelOpenPromise(chan, flags) {
     20  return new Promise(resolve => {
     21    function finish(req, buffer) {
     22      resolve([req, buffer]);
     23    }
     24    chan.asyncOpen(new ChannelListener(finish, null, flags));
     25  });
     26 }
     27 
     28 function makeURL(host) {
     29  return `http://${host}:${httpServer.identity.primaryPort}/`;
     30 }
     31 
     32 add_task(async function test_localhost_offline() {
     33  Services.io.offline = true;
     34  Services.prefs.setBoolPref("network.disable-localhost-when-offline", false);
     35  let chan = makeChan(makeURL("127.0.0.1"));
     36  let [, resp] = await channelOpenPromise(chan);
     37  Assert.equal(resp, body, "Should get correct response");
     38 
     39  chan = makeChan(makeURL("localhost"));
     40  [, resp] = await channelOpenPromise(chan);
     41  Assert.equal(resp, body, "Should get response");
     42 
     43  Services.prefs.setBoolPref("network.disable-localhost-when-offline", true);
     44 
     45  chan = makeChan(makeURL("127.0.0.1"));
     46  let [req] = await channelOpenPromise(
     47    chan,
     48    CL_ALLOW_UNKNOWN_CL | CL_EXPECT_FAILURE
     49  );
     50  req.QueryInterface(Ci.nsIHttpChannel);
     51  Assert.equal(req.status, Cr.NS_ERROR_OFFLINE);
     52 
     53  chan = makeChan(makeURL("localhost"));
     54  [req] = await channelOpenPromise(
     55    chan,
     56    CL_ALLOW_UNKNOWN_CL | CL_EXPECT_FAILURE
     57  );
     58  req.QueryInterface(Ci.nsIHttpChannel);
     59  Assert.equal(req.status, Cr.NS_ERROR_OFFLINE);
     60 
     61  Services.prefs.clearUserPref("network.disable-localhost-when-offline");
     62  Services.io.offline = false;
     63 });
     64 
     65 function run_test() {
     66  httpServer = new HttpServer();
     67  httpServer.registerPathHandler("/", (request, response) => {
     68    response.seizePower();
     69    response.write("HTTP/1.1 200 OK\r\n");
     70    response.write("Content-Length: " + body.length + "\r\n");
     71    response.write("\r\n");
     72    response.write(body);
     73    response.finish();
     74  });
     75  httpServer.start(-1);
     76  run_next_test();
     77 }