michael@0: var ScriptableUnicodeConverter =
michael@0:   Components.Constructor("@mozilla.org/intl/scriptableunicodeconverter",
michael@0:                          "nsIScriptableUnicodeConverter");
michael@0: var CryptoHash =
michael@0:   Components.Constructor("@mozilla.org/security/hash;1",
michael@0:                          "nsICryptoHash",
michael@0:                          "initWithString");
michael@0: 
michael@0: var messages = [
michael@0:   "The quick brown fox jumps over the lazy dog",
michael@0:   ""
michael@0: ];
michael@0: var hashes = {
michael@0:   md2: [
michael@0:     "03d85a0d629d2c442e987525319fc471",
michael@0:     "8350e5a3e24c153df2275c9f80692773"
michael@0:   ],
michael@0:   md5: [
michael@0:     "9e107d9d372bb6826bd81d3542a419d6",
michael@0:     "d41d8cd98f00b204e9800998ecf8427e"
michael@0:   ],
michael@0:   sha1: [
michael@0:     "2fd4e1c67a2d28fced849ee1bb76e7391b93eb12",
michael@0:     "da39a3ee5e6b4b0d3255bfef95601890afd80709"
michael@0:   ],
michael@0:   sha256: [
michael@0:     "d7a8fbb307d7809469ca9abcb0082e4f8d5651e46d3cdb762d02d0bf37c9e592",
michael@0:     "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855"
michael@0:   ],
michael@0:   sha384: [
michael@0:    "ca737f1014a48f4c0b6dd43cb177b0afd9e5169367544c494011e3317dbf9a509cb1e5dc1e85a941bbee3d7f2afbc9b1",
michael@0:    "38b060a751ac96384cd9327eb1b1e36a21fdb71114be07434c0cc7bf63f6e1da274edebfe76f65fbd51ad2f14898b95b"
michael@0:   ],   
michael@0:   sha512: [
michael@0:     "07e547d9586f6a73f73fbac0435ed76951218fb7d0c8d788a309d785436bbb642e93a252a954f23912547d1e8a3b5ed6e1bfd7097821233fa0538f3db854fee6",
michael@0:     "cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e"
michael@0:   ]
michael@0: };
michael@0: 
michael@0: function hexdigest(data) {
michael@0:   /*
michael@0:    * Coment taken from bug 383390:
michael@0:    *  
michael@0:    * First, |data| is the final string value produced by the cryptohash.  |for (i in
michael@0:    * data)| uses the Mozilla JS extension that iterating over a string iterates over
michael@0:    * its character indexes, so that's 0..length-1 over the hash string.
michael@0:    * 
michael@0:    * Returning to the left, the |charCodeAt| gets the value of the character at that
michael@0:    * index in the string.
michael@0:    * 
michael@0:    * |slice(-2)| is equivalent to |slice(length of this string - 2)| as a convenient
michael@0:    * way of wrapping around to the other end of a string without doing the actual
michael@0:    * calculation.  When provided with only one argument, slice selects to the end of
michael@0:    * the string, so this chomps off the last two characters of the string.
michael@0:    * 
michael@0:    * The last-two-characters part clarifies the |"0" +| -- if the Unicode value is
michael@0:    * <10, we have a single-character hex string when we want one that's two
michael@0:    * characters, and unconditionally prepending a "0" solves the problem.
michael@0:    * 
michael@0:    * The array comprehension just creates an array whose elements are these
michael@0:    * two-character strings.
michael@0:    */
michael@0:   return [("0" + data.charCodeAt(i).toString(16)).slice(-2) for (i in data)].join("");
michael@0: }
michael@0: 
michael@0: function doHash(algo, value, cmp) {
michael@0:   var converter = new ScriptableUnicodeConverter();
michael@0:   var hash = new CryptoHash(algo);
michael@0: 
michael@0:   converter.charset = 'utf8';
michael@0:   value = converter.convertToByteArray(value);
michael@0:   hash.update(value, value.length);
michael@0:   var hash1 = hexdigest(hash.finish(false));
michael@0:   if (cmp != hash1) {
michael@0:     do_throw("Hash mismatch!\n" +
michael@0:              "  Expected: " + cmp + "\n" +
michael@0:              "  Actual: " + hash1 + "\n" +
michael@0:              "  Algo: " + algo);
michael@0:   }                                                                                                                                                                                                                                  
michael@0: 
michael@0:   hash.initWithString(algo);
michael@0:   hash.update(value, value.length);
michael@0:   var hash2 = hexdigest(hash.finish(false));
michael@0:   if (cmp != hash2) {
michael@0:     do_throw("Hash mismatch after crypto hash re-init!\n" +
michael@0:              "  Expected: " + cmp + "\n" +
michael@0:              "  Actual: " + hash2 + "\n" +
michael@0:              "  Algo: " + algo);
michael@0:   }
michael@0: }
michael@0: 
michael@0: function doHashStream(algo, value, cmp) {
michael@0:   var converter = new ScriptableUnicodeConverter();
michael@0:   var hash = new CryptoHash(algo);
michael@0: 
michael@0:   converter.charset = 'utf8';
michael@0:   var stream = converter.convertToInputStream(value);
michael@0:   hash.updateFromStream(stream, stream.available());
michael@0:   hash = hexdigest(hash.finish(false));
michael@0:   if (cmp != hash) {
michael@0:     do_throw("Hash mismatch!\n" +
michael@0:              "  Expected: " + cmp + "\n" +
michael@0:              "  Actual: " + hash + "\n" +
michael@0:              "  Algo: " + algo);
michael@0:   }
michael@0: }
michael@0: 
michael@0: function run_test() {
michael@0:   for (algo in hashes) {
michael@0:     hashes[algo].forEach(
michael@0:       function(e, i) {
michael@0:         doHash(algo, messages[i], e);
michael@0:         
michael@0:         if (messages[i].length) {
michael@0:           // this test doesn't work for empty string/stream
michael@0:           doHashStream(algo, messages[i], e);
michael@0:         }
michael@0:       }
michael@0:     );
michael@0:   }
michael@0: }