netwerk/test/unit/test_redirect_veto.js

   1 "use strict";
   2
   3 const { HttpServer } = ChromeUtils.importESModule(
   4   "resource://testing-common/httpd.sys.mjs"
   5 );
   6
   7 ChromeUtils.defineLazyGetter(this, "URL", function () {
   8   return "http://localhost:" + httpServer.identity.primaryPort;
   9 });
  10
  11 var httpServer = null;
  12 // Need to randomize, because apparently no one clears our cache
  13 var randomPath = "/redirect/" + Math.random();
  14
  15 ChromeUtils.defineLazyGetter(this, "randomURI", function () {
  16   return URL + randomPath;
  17 });
  18
  19 function make_channel(url) {
  20   return NetUtil.newChannel({ uri: url, loadUsingSystemPrincipal: true });
  21 }
  22
  23 const responseBody = "response body";
  24
  25 function redirectHandler(metadata, response) {
  26   response.setStatusLine(metadata.httpVersion, 301, "Moved");
  27   response.setHeader("Location", URL + "/content", false);
  28 }
  29
  30 function contentHandler(metadata, response) {
  31   response.setHeader("Content-Type", "text/plain");
  32   response.bodyOutputStream.write(responseBody, responseBody.length);
  33 }
  34
  35 let ChannelEventSink2 = {
  36   _classDescription: "WebRequest channel event sink",
  37   _classID: Components.ID("115062f8-92f1-11e5-8b7f-08001110f7ec"),
  38   _contractID: "@mozilla.org/webrequest/channel-event-sink;1",
  39
  40   QueryInterface: ChromeUtils.generateQI(["nsIChannelEventSink", "nsIFactory"]),
  41
  42   init() {
  43     Components.manager
  44       .QueryInterface(Ci.nsIComponentRegistrar)
  45       .registerFactory(
  46         this._classID,
  47         this._classDescription,
  48         this._contractID,
  49         this
  50       );
  51   },
  52
  53   register() {
  54     Services.catMan.addCategoryEntry(
  55       "net-channel-event-sinks",
  56       this._contractID,
  57       this._contractID,
  58       false,
  59       true
  60     );
  61   },
  62
  63   unregister() {
  64     Services.catMan.deleteCategoryEntry(
  65       "net-channel-event-sinks",
  66       this._contractID,
  67       false
  68     );
  69   },
  70
  71   // nsIChannelEventSink implementation
  72   asyncOnChannelRedirect(oldChannel, newChannel, flags, redirectCallback) {
  73     // Abort the redirection
  74     redirectCallback.onRedirectVerifyCallback(Cr.NS_ERROR_NO_INTERFACE);
  75   },
  76
  77   // nsIFactory implementation
  78   createInstance(iid) {
  79     return this.QueryInterface(iid);
  80   },
  81 };
  82
  83 add_task(async function test_redirect_veto() {
  84   httpServer = new HttpServer();
  85   httpServer.registerPathHandler(randomPath, redirectHandler);
  86   httpServer.registerPathHandler("/content", contentHandler);
  87   httpServer.start(-1);
  88
  89   ChannelEventSink2.init();
  90   ChannelEventSink2.register();
  91
  92   let chan = make_channel(randomURI);
  93   let [req, buff] = await new Promise(resolve =>
  94     chan.asyncOpen(
  95       new ChannelListener((aReq, aBuff) => resolve([aReq, aBuff], null))
  96     )
  97   );
  98   Assert.equal(buff, "");
  99   Assert.equal(req.status, Cr.NS_OK);
 100   await httpServer.stop();
 101   ChannelEventSink2.unregister();
 102 });