1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/toolkit/mozapps/extensions/test/browser/browser_hotfix.js Wed Dec 31 06:09:35 2014 +0100 1.3 @@ -0,0 +1,190 @@ 1.4 +/* Any copyright is dedicated to the Public Domain. 1.5 + * http://creativecommons.org/publicdomain/zero/1.0/ 1.6 + */ 1.7 + 1.8 +const PREF_EM_HOTFIX_ID = "extensions.hotfix.id"; 1.9 +const PREF_EM_HOTFIX_LASTVERSION = "extensions.hotfix.lastVersion"; 1.10 +const PREF_EM_HOTFIX_URL = "extensions.hotfix.url"; 1.11 +const PREF_EM_HOTFIX_CERTS = "extensions.hotfix.certs."; 1.12 +const PREF_EM_CERT_CHECKATTRIBUTES = "extensions.hotfix.cert.checkAttributes"; 1.13 + 1.14 +const PREF_INSTALL_REQUIREBUILTINCERTS = "extensions.install.requireBuiltInCerts"; 1.15 +const PREF_UPDATE_REQUIREBUILTINCERTS = "extensions.update.requireBuiltInCerts"; 1.16 + 1.17 +const PREF_APP_UPDATE_ENABLED = "app.update.enabled"; 1.18 +const PREF_APP_UPDATE_URL = "app.update.url"; 1.19 + 1.20 +const HOTFIX_ID = "hotfix@tests.mozilla.org"; 1.21 + 1.22 +var gNextTest; 1.23 + 1.24 +var SuccessfulInstallListener = { 1.25 + onDownloadCancelled: function(aInstall) { 1.26 + ok(false, "Should not have seen the download cancelled"); 1.27 + is(aInstall.addon.id, HOTFIX_ID, "Should have seen the right add-on"); 1.28 + 1.29 + AddonManager.removeInstallListener(this); 1.30 + gNextTest(); 1.31 + }, 1.32 + 1.33 + onInstallEnded: function(aInstall) { 1.34 + ok(true, "Should have seen the install complete"); 1.35 + is(aInstall.addon.id, HOTFIX_ID, "Should have installed the right add-on"); 1.36 + 1.37 + AddonManager.removeInstallListener(this); 1.38 + aInstall.addon.uninstall(); 1.39 + Services.prefs.clearUserPref(PREF_EM_HOTFIX_LASTVERSION); 1.40 + gNextTest(); 1.41 + } 1.42 +} 1.43 + 1.44 +var FailedInstallListener = { 1.45 + onDownloadCancelled: function(aInstall) { 1.46 + ok(true, "Should have seen the download cancelled"); 1.47 + is(aInstall.addon.id, HOTFIX_ID, "Should have seen the right add-on"); 1.48 + 1.49 + AddonManager.removeInstallListener(this); 1.50 + gNextTest(); 1.51 + }, 1.52 + 1.53 + onInstallEnded: function(aInstall) { 1.54 + ok(false, "Should not have seen the install complete"); 1.55 + is(aInstall.addon.id, HOTFIX_ID, "Should have installed the right add-on"); 1.56 + 1.57 + AddonManager.removeInstallListener(this); 1.58 + aInstall.addon.uninstall(); 1.59 + Services.prefs.clearUserPref(PREF_EM_HOTFIX_LASTVERSION); 1.60 + gNextTest(); 1.61 + } 1.62 +} 1.63 + 1.64 +function test() { 1.65 + waitForExplicitFinish(); 1.66 + 1.67 + var oldAusUrl = Services.prefs.getDefaultBranch(null).getCharPref(PREF_APP_UPDATE_URL); 1.68 + Services.prefs.getDefaultBranch(null).setCharPref(PREF_APP_UPDATE_URL, TESTROOT + "ausdummy.xml"); 1.69 + Services.prefs.setBoolPref(PREF_APP_UPDATE_ENABLED, true); 1.70 + Services.prefs.setBoolPref(PREF_INSTALL_REQUIREBUILTINCERTS, false); 1.71 + Services.prefs.setBoolPref(PREF_UPDATE_REQUIREBUILTINCERTS, false); 1.72 + Services.prefs.setCharPref(PREF_EM_HOTFIX_ID, HOTFIX_ID); 1.73 + var oldURL = Services.prefs.getCharPref(PREF_EM_HOTFIX_URL); 1.74 + Services.prefs.setCharPref(PREF_EM_HOTFIX_URL, TESTROOT + "signed_hotfix.rdf"); 1.75 + 1.76 + registerCleanupFunction(function() { 1.77 + Services.prefs.setBoolPref(PREF_APP_UPDATE_ENABLED, false); 1.78 + Services.prefs.getDefaultBranch(null).setCharPref(PREF_APP_UPDATE_URL, oldAusUrl); 1.79 + Services.prefs.clearUserPref(PREF_EM_HOTFIX_ID); 1.80 + Services.prefs.setCharPref(PREF_EM_HOTFIX_URL, oldURL); 1.81 + Services.prefs.clearUserPref(PREF_INSTALL_REQUIREBUILTINCERTS); 1.82 + Services.prefs.clearUserPref(PREF_UPDATE_REQUIREBUILTINCERTS); 1.83 + 1.84 + Services.prefs.clearUserPref(PREF_EM_CERT_CHECKATTRIBUTES); 1.85 + var prefs = Services.prefs.getChildList(PREF_EM_HOTFIX_CERTS); 1.86 + prefs.forEach(Services.prefs.clearUserPref); 1.87 + }); 1.88 + 1.89 + run_next_test(); 1.90 +} 1.91 + 1.92 +function end_test() { 1.93 + finish(); 1.94 +} 1.95 + 1.96 +add_test(function check_no_cert_checks() { 1.97 + Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, false); 1.98 + AddonManager.addInstallListener(SuccessfulInstallListener); 1.99 + 1.100 + gNextTest = run_next_test; 1.101 + 1.102 + AddonManagerPrivate.backgroundUpdateCheck(); 1.103 +}); 1.104 + 1.105 +add_test(function check_wrong_cert_fingerprint() { 1.106 + Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, true); 1.107 + Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint", "foo"); 1.108 + 1.109 + AddonManager.addInstallListener(FailedInstallListener); 1.110 + 1.111 + gNextTest = function() { 1.112 + Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint"); 1.113 + 1.114 + run_next_test(); 1.115 + }; 1.116 + 1.117 + AddonManagerPrivate.backgroundUpdateCheck(); 1.118 +}); 1.119 + 1.120 +add_test(function check_right_cert_fingerprint() { 1.121 + Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, true); 1.122 + Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint", "3E:B9:4E:07:12:FE:3C:01:41:46:13:46:FC:84:52:1A:8C:BE:1D:A2"); 1.123 + 1.124 + AddonManager.addInstallListener(SuccessfulInstallListener); 1.125 + 1.126 + gNextTest = function() { 1.127 + Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint"); 1.128 + 1.129 + run_next_test(); 1.130 + }; 1.131 + 1.132 + AddonManagerPrivate.backgroundUpdateCheck(); 1.133 +}); 1.134 + 1.135 +add_test(function check_multi_cert_fingerprint_1() { 1.136 + Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, true); 1.137 + Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint", "3E:B9:4E:07:12:FE:3C:01:41:46:13:46:FC:84:52:1A:8C:BE:1D:A2"); 1.138 + Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "2.sha1Fingerprint", "foo"); 1.139 + 1.140 + AddonManager.addInstallListener(SuccessfulInstallListener); 1.141 + 1.142 + gNextTest = function() { 1.143 + Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint"); 1.144 + Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "2.sha1Fingerprint"); 1.145 + 1.146 + run_next_test(); 1.147 + }; 1.148 + 1.149 + AddonManagerPrivate.backgroundUpdateCheck(); 1.150 +}); 1.151 + 1.152 +add_test(function check_multi_cert_fingerprint_2() { 1.153 + Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, true); 1.154 + Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint", "foo"); 1.155 + Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "2.sha1Fingerprint", "3E:B9:4E:07:12:FE:3C:01:41:46:13:46:FC:84:52:1A:8C:BE:1D:A2"); 1.156 + 1.157 + AddonManager.addInstallListener(SuccessfulInstallListener); 1.158 + 1.159 + gNextTest = function() { 1.160 + Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint"); 1.161 + Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "2.sha1Fingerprint"); 1.162 + 1.163 + run_next_test(); 1.164 + }; 1.165 + 1.166 + AddonManagerPrivate.backgroundUpdateCheck(); 1.167 +}); 1.168 + 1.169 +add_test(function check_no_cert_no_checks() { 1.170 + Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, false); 1.171 + Services.prefs.setCharPref(PREF_EM_HOTFIX_URL, TESTROOT + "unsigned_hotfix.rdf"); 1.172 + 1.173 + AddonManager.addInstallListener(SuccessfulInstallListener); 1.174 + 1.175 + gNextTest = run_next_test; 1.176 + 1.177 + AddonManagerPrivate.backgroundUpdateCheck(); 1.178 +}); 1.179 + 1.180 +add_test(function check_no_cert_cert_fingerprint_check() { 1.181 + Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, true); 1.182 + Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint", "3E:B9:4E:07:12:FE:3C:01:41:46:13:46:FC:84:52:1A:8C:BE:1D:A2"); 1.183 + 1.184 + AddonManager.addInstallListener(FailedInstallListener); 1.185 + 1.186 + gNextTest = function() { 1.187 + Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint"); 1.188 + 1.189 + run_next_test(); 1.190 + }; 1.191 + 1.192 + AddonManagerPrivate.backgroundUpdateCheck(); 1.193 +});