michael@0: Cu.import("resource://testing-common/httpd.js");
michael@0: 
michael@0: var httpserver = null;
michael@0: 
michael@0: function make_channel(url, callback, ctx) {
michael@0:   var ios = Cc["@mozilla.org/network/io-service;1"].
michael@0:             getService(Ci.nsIIOService);
michael@0:   return ios.newChannel(url, "", null);
michael@0: }
michael@0: 
michael@0: const responseBody = "response body";
michael@0: 
michael@0: function cachedHandler(metadata, response) {
michael@0:   var body = responseBody;
michael@0:   if (metadata.hasHeader("Range")) {
michael@0:     var matches = metadata.getHeader("Range").match(/^\s*bytes=(\d+)?-(\d+)?\s*$/);
michael@0:     var from = (matches[1] === undefined) ? 0 : matches[1];
michael@0:     var to = (matches[2] === undefined) ? responseBody.length - 1 : matches[2];
michael@0:     if (from >= responseBody.length) {
michael@0:       response.setStatusLine(metadata.httpVersion, 416, "Start pos too high");
michael@0:       response.setHeader("Content-Range", "*/" + responseBody.length, false);
michael@0:       return;
michael@0:     }
michael@0:     body = responseBody.slice(from, to + 1);
michael@0:     // always respond to successful range requests with 206
michael@0:     response.setStatusLine(metadata.httpVersion, 206, "Partial Content");
michael@0:     response.setHeader("Content-Range", from + "-" + to + "/" + responseBody.length, false);
michael@0:   }
michael@0: 
michael@0:   response.setHeader("Content-Type", "text/plain", false);
michael@0:   response.setHeader("ETag", "Just testing");
michael@0:   response.setHeader("Accept-Ranges", "bytes");
michael@0: 
michael@0:   response.bodyOutputStream.write(body, body.length);
michael@0: }
michael@0: 
michael@0: function Canceler(continueFn) {
michael@0:   this.continueFn = continueFn;
michael@0: }
michael@0: 
michael@0: Canceler.prototype = {
michael@0:   QueryInterface: function(iid) {
michael@0:     if (iid.equals(Ci.nsIStreamListener) ||
michael@0:         iid.equals(Ci.nsIRequestObserver) ||
michael@0:         iid.equals(Ci.nsISupports))
michael@0:       return this;
michael@0:     throw Components.results.NS_ERROR_NO_INTERFACE;
michael@0:   },
michael@0: 
michael@0:   onStartRequest: function(request, context) {
michael@0:   },
michael@0: 
michael@0:   onDataAvailable: function(request, context, stream, offset, count) {
michael@0:     request.QueryInterface(Ci.nsIChannel)
michael@0:            .cancel(Components.results.NS_BINDING_ABORTED);
michael@0:   },
michael@0: 
michael@0:   onStopRequest: function(request, context, status) {
michael@0:     do_check_eq(status, Components.results.NS_BINDING_ABORTED);
michael@0:     this.continueFn();
michael@0:   }
michael@0: };
michael@0: 
michael@0: function finish_test() {
michael@0:   httpserver.stop(do_test_finished);
michael@0: }
michael@0: 
michael@0: function start_cache_read() {
michael@0:   var chan = make_channel("http://localhost:" +
michael@0:                           httpserver.identity.primaryPort + "/cached/test.gz");
michael@0:   chan.asyncOpen(new ChannelListener(finish_test, null), null);
michael@0: }
michael@0: 
michael@0: function start_canceler() {
michael@0:   var chan = make_channel("http://localhost:" +
michael@0:                           httpserver.identity.primaryPort + "/cached/test.gz");
michael@0:   chan.asyncOpen(new Canceler(start_cache_read), null);
michael@0: }
michael@0: 
michael@0: function run_test() {
michael@0:   httpserver = new HttpServer();
michael@0:   httpserver.registerPathHandler("/cached/test.gz", cachedHandler);
michael@0:   httpserver.start(-1);
michael@0: 
michael@0:   var chan = make_channel("http://localhost:" +
michael@0:                           httpserver.identity.primaryPort + "/cached/test.gz");
michael@0:   chan.asyncOpen(new ChannelListener(start_canceler, null), null);
michael@0:   do_test_pending();
michael@0: }