michael@0: var ScriptableUnicodeConverter =
michael@0:   Components.Constructor("@mozilla.org/intl/scriptableunicodeconverter",
michael@0:                          "nsIScriptableUnicodeConverter"); 
michael@0:                          
michael@0: function getHMAC(data, key, alg)
michael@0: {
michael@0:   var converter = new ScriptableUnicodeConverter(); 
michael@0:   converter.charset = 'utf8'; 
michael@0:   var dataarray = converter.convertToByteArray(data);
michael@0:   
michael@0:   var keyObject = Components.classes["@mozilla.org/security/keyobjectfactory;1"]
michael@0:     .getService(Components.interfaces.nsIKeyObjectFactory)
michael@0:       .keyFromString(Components.interfaces.nsIKeyObject.HMAC, key);
michael@0:   
michael@0:   var cryptoHMAC = Components.classes["@mozilla.org/security/hmac;1"]
michael@0:     .createInstance(Components.interfaces.nsICryptoHMAC);
michael@0:     
michael@0:   cryptoHMAC.init(alg, keyObject);
michael@0:   cryptoHMAC.update(dataarray, dataarray.length);
michael@0:   var digest1 = cryptoHMAC.finish(false);
michael@0:   
michael@0:   cryptoHMAC.reset();
michael@0:   cryptoHMAC.update(dataarray, dataarray.length);
michael@0:   var digest2 = cryptoHMAC.finish(false);
michael@0:   
michael@0:   do_check_eq(digest1, digest2);
michael@0:   
michael@0:   return digest1;
michael@0: }
michael@0: 
michael@0: function testHMAC(alg) {
michael@0:   const key1 = 'MyKey_ABCDEFGHIJKLMN';
michael@0:   const key2 = 'MyKey_01234567890123';
michael@0:   
michael@0:   const dataA = "Secret message";
michael@0:   const dataB = "Secres message";
michael@0:   
michael@0:   var diegest1a = getHMAC(key1, dataA, alg);
michael@0:   var diegest2 = getHMAC(key2, dataA, alg);
michael@0:   var diegest1b = getHMAC(key1, dataA, alg);
michael@0:   
michael@0:   do_check_eq(diegest1a, diegest1b);
michael@0:   do_check_neq(diegest1a, diegest2);
michael@0:   
michael@0:   var diegest1 = getHMAC(key1, dataA, alg);
michael@0:   diegest2 = getHMAC(key1, dataB, alg);
michael@0:   
michael@0:   do_check_neq(diegest1, diegest2);
michael@0:   
michael@0:   return diegest1;
michael@0: }
michael@0: 
michael@0: function hexdigest(data) {
michael@0:   return [("0" + data.charCodeAt(i).toString(16)).slice(-2) for (i in data)].join("");
michael@0: }
michael@0: 
michael@0: function testVectors() {
michael@0:   // These are test vectors taken from RFC 4231, section 4.3. (Test Case 2)
michael@0:   
michael@0:   const keyTestVector = "Jefe";
michael@0:   const dataTestVector = "what do ya want for nothing?";
michael@0: 
michael@0:   var diegest;
michael@0:   /*
michael@0:   Bug 356713    
michael@0:   diegest = hexdigest(getHMAC(dataTestVector, keyTestVector, 
michael@0:           Components.interfaces.nsICryptoHMAC.SHA224));
michael@0:   do_check_eq(diegest, "a30e01098bc6dbbf45690f3a7e9e6d0f8bbea2a39e6148008fd05e44");
michael@0:   */
michael@0: 
michael@0:   diegest = hexdigest(getHMAC(dataTestVector, keyTestVector, 
michael@0:           Components.interfaces.nsICryptoHMAC.SHA256));
michael@0:   do_check_eq(diegest, "5bdcc146bf60754e6a042426089575c75a003f089d2739839dec58b964ec3843"); 
michael@0: 
michael@0:   diegest = hexdigest(getHMAC(dataTestVector, keyTestVector, 
michael@0:           Components.interfaces.nsICryptoHMAC.SHA384));
michael@0:   do_check_eq(diegest, "af45d2e376484031617f78d2b58a6b1b9c7ef464f5a01b47e42ec3736322445e8e2240ca5e69e2c78b3239ecfab21649"); 
michael@0: 
michael@0:   diegest = hexdigest(getHMAC(dataTestVector, keyTestVector, 
michael@0:           Components.interfaces.nsICryptoHMAC.SHA512));
michael@0:   do_check_eq(diegest, "164b7a7bfcf819e2e395fbe73b56e0a387bd64222e831fd610270cd7ea2505549758bf75c05a994a6d034f65f8f0e6fdcaeab1a34d4a6b4b636e070a38bce737"); 
michael@0: }
michael@0: 
michael@0: function run_test() {
michael@0:   testVectors();
michael@0: 
michael@0:   testHMAC(Components.interfaces.nsICryptoHMAC.SHA1);
michael@0:   testHMAC(Components.interfaces.nsICryptoHMAC.SHA512);
michael@0:   testHMAC(Components.interfaces.nsICryptoHMAC.MD5);
michael@0: }