michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: const Ci = Components.interfaces;
michael@0: const Cu = Components.utils;
michael@0: const Cc = Components.classes;
michael@0: const CC = Components.Constructor;
michael@0: 
michael@0: var BinaryOutputStream = CC("@mozilla.org/binaryoutputstream;1",
michael@0:                             "nsIBinaryOutputStream",
michael@0:                             "setOutputStream");
michael@0: 
michael@0: Cu.import("resource://testing-common/httpd.js");
michael@0: 
michael@0: var httpserver = new HttpServer();
michael@0: 
michael@0: var testRan = 0;
michael@0: 
michael@0: // The tests files we want to test, and the type we should have after sniffing.
michael@0: const tests = [
michael@0:   // Real webm and mkv files truncated to 512 bytes.
michael@0:   { path: "data/file.webm", expected: "video/webm" },
michael@0:   { path: "data/file.mkv", expected: "application/octet-stream" },
michael@0:   // MP3 files with and without id3 headers truncated to 512 bytes.
michael@0:   // NB these have 208/209 byte frames, but mp3 can require up to
michael@0:   // 1445 bytes to detect with our method.
michael@0:   { path: "data/id3tags.mp3", expected: "audio/mpeg" },
michael@0:   { path: "data/notags.mp3", expected: "audio/mpeg" },
michael@0:   // MPEG-2 mp3 files.
michael@0:   { path: "data/detodos.mp3", expected: "audio/mpeg" },
michael@0:   // Padding bit flipped in the first header: sniffing should fail.
michael@0:   { path: "data/notags-bad.mp3", expected: "application/octet-stream" },
michael@0:   // Garbage before header: sniffing should fail.
michael@0:   { path: "data/notags-scan.mp3", expected: "application/octet-stream" },
michael@0:   // VBR from the layer III test patterns. We can't sniff this.
michael@0:   { path: "data/he_free.mp3", expected: "application/octet-stream" },
michael@0:   // Make sure we reject mp2, which has a similar header.
michael@0:   { path: "data/fl10.mp2", expected: "application/octet-stream" },
michael@0:   // Truncated ff installer regression test for bug 875769.
michael@0:   { path: "data/ff-inst.exe", expected: "application/octet-stream" },
michael@0: ];
michael@0: 
michael@0: // A basic listener that reads checks the if we sniffed properly.
michael@0: var listener = {
michael@0:   onStartRequest: function(request, context) {
michael@0:     do_print("Sniffing " + tests[testRan].path);
michael@0:     do_check_eq(request.QueryInterface(Ci.nsIChannel).contentType, tests[testRan].expected);
michael@0:   },
michael@0: 
michael@0:   onDataAvailable: function(request, context, stream, offset, count) {
michael@0:     try {
michael@0:       var bis = Components.classes["@mozilla.org/binaryinputstream;1"]
michael@0:                           .createInstance(Components.interfaces.nsIBinaryInputStream);
michael@0:       bis.setInputStream(stream);
michael@0:       var array = bis.readByteArray(bis.available());
michael@0:     } catch (ex) {
michael@0:       do_throw("Error in onDataAvailable: " + ex);
michael@0:     }
michael@0:   },
michael@0: 
michael@0:   onStopRequest: function(request, context, status) {
michael@0:     testRan++;
michael@0:     runNext();
michael@0:   }
michael@0: };
michael@0: 
michael@0: function setupChannel(url) {
michael@0:   var ios = Components.classes["@mozilla.org/network/io-service;1"].
michael@0:                        getService(Ci.nsIIOService);
michael@0:   var chan = ios.newChannel("http://localhost:" +
michael@0:                            httpserver.identity.primaryPort + url, "", null);
michael@0:   var httpChan = chan.QueryInterface(Components.interfaces.nsIHttpChannel);
michael@0:   return httpChan;
michael@0: }
michael@0: 
michael@0: function runNext() {
michael@0:   if (testRan == tests.length) {
michael@0:     do_test_finished();
michael@0:     return;
michael@0:   }
michael@0:   var channel = setupChannel("/");
michael@0:   channel.asyncOpen(listener, channel, null);
michael@0: }
michael@0: 
michael@0: function getFileContents(aFile) {
michael@0:   const PR_RDONLY = 0x01;
michael@0:   var fileStream = Cc["@mozilla.org/network/file-input-stream;1"]
michael@0:                       .createInstance(Ci.nsIFileInputStream);
michael@0:   fileStream.init(aFile, 1, -1, null);
michael@0:   var bis = Components.classes["@mozilla.org/binaryinputstream;1"]
michael@0:                       .createInstance(Components.interfaces.nsIBinaryInputStream);
michael@0:   bis.setInputStream(fileStream);
michael@0: 
michael@0:   var data = bis.readByteArray(bis.available());
michael@0: 
michael@0:   return data;
michael@0: }
michael@0: 
michael@0: function handler(metadata, response) {
michael@0:   response.setStatusLine(metadata.httpVersion, 200, "OK");
michael@0:   // Send an empty Content-Type, so we are guaranteed to sniff.
michael@0:   response.setHeader("Content-Type", "", false);
michael@0:   var body = getFileContents(do_get_file(tests[testRan].path));
michael@0:   var bos = new BinaryOutputStream(response.bodyOutputStream);
michael@0:   bos.writeByteArray(body, body.length);
michael@0: }
michael@0: 
michael@0: function run_test() {
michael@0:   // We use a custom handler so we can change the header to force sniffing.
michael@0:   httpserver.registerPathHandler("/", handler);
michael@0:   httpserver.start(-1);
michael@0:   do_test_pending();
michael@0:   try {
michael@0:     runNext();
michael@0:   } catch (e) {
michael@0:     print("ERROR - " + e + "\n");
michael@0:   }
michael@0: }