forked from mirrors/gecko-dev
09d88e5fd2
This is no longer necessary as the Quantum DOM project is no longer happening, and removing support simplifies various components inside of IPDL. As some code used the support to get a `nsISerialEventTarget` for an actor's worker thread, that method was replaced with a method which instead pulls the nsISerialEventTarget from the MessageChannel and should work on all actors. Differential Revision: https://phabricator.services.mozilla.com/D135411
187 lines
5.2 KiB
HTML
187 lines
5.2 KiB
HTML
<!DOCTYPE HTML>
|
|
<html>
|
|
<head>
|
|
<title>Test the URI Classifier Matched Info (bug 1288633) </title>
|
|
<script src="/tests/SimpleTest/SimpleTest.js"></script>
|
|
<script type="text/javascript" src="classifierHelper.js"></script>
|
|
<link rel="stylesheet" type="text/css" href="/tests/SimpleTest/test.css" />
|
|
</head>
|
|
|
|
<body>
|
|
<p id="display"></p>
|
|
<div id="content" style="display: none">
|
|
</div>
|
|
<pre id="test">
|
|
|
|
<script class="testbody" type="application/javascript">
|
|
var Cc = SpecialPowers.Cc;
|
|
var Ci = SpecialPowers.Ci;
|
|
var Cr = SpecialPowers.Cr;
|
|
|
|
var inputDatas = [
|
|
{ url: "malware.mochi.test/",
|
|
db: "mochi-block-simple",
|
|
},
|
|
{ url: "malware1.mochi.test/",
|
|
db: "mochi1-block-simple",
|
|
},
|
|
{ url: "malware1.mochi.test/",
|
|
db: "mochi1-malware-simple",
|
|
provider: "mozilla",
|
|
},
|
|
{ url: "malware2.mochi.test/",
|
|
db: "mochi2-unwanted-simple",
|
|
provider: "mozilla",
|
|
},
|
|
{ url: "malware2.mochi.test/",
|
|
db: "mochi2-malware-simple",
|
|
provider: "mozilla",
|
|
},
|
|
{ url: "malware3.mochi.test/",
|
|
db: "mochig3-malware-simple",
|
|
provider: "google",
|
|
},
|
|
{ url: "malware3.mochi.test/",
|
|
db: "mochim3-malware-simple",
|
|
provider: "mozilla",
|
|
},
|
|
];
|
|
|
|
function hash(str) {
|
|
function bytesFromString(str1) {
|
|
let converter =
|
|
Cc["@mozilla.org/intl/scriptableunicodeconverter"]
|
|
.createInstance(Ci.nsIScriptableUnicodeConverter);
|
|
converter.charset = "UTF-8";
|
|
return converter.convertToByteArray(str1);
|
|
}
|
|
|
|
let hasher = Cc["@mozilla.org/security/hash;1"]
|
|
.createInstance(Ci.nsICryptoHash);
|
|
|
|
let data = bytesFromString(str);
|
|
hasher.init(hasher.SHA256);
|
|
hasher.update(data, data.length);
|
|
|
|
return hasher.finish(false);
|
|
}
|
|
|
|
var testDatas = [
|
|
// Match empty provider
|
|
{ url: "http://malware.mochi.test",
|
|
expect: { error: Cr.NS_ERROR_BLOCKED_URI,
|
|
table: "mochi-block-simple",
|
|
provider: "",
|
|
fullhash: (function() {
|
|
return hash("malware.mochi.test/");
|
|
})(),
|
|
},
|
|
},
|
|
// Match multiple tables, only one has valid provider
|
|
{ url: "http://malware1.mochi.test",
|
|
expect: { error: Cr.NS_ERROR_MALWARE_URI,
|
|
table: "mochi1-malware-simple",
|
|
provider: "mozilla",
|
|
fullhash: (function() {
|
|
return hash("malware1.mochi.test/");
|
|
})(),
|
|
},
|
|
},
|
|
// Match multiple tables, handle order
|
|
{ url: "http://malware2.mochi.test",
|
|
expect: { error: Cr.NS_ERROR_MALWARE_URI,
|
|
table: "mochi2-malware-simple",
|
|
provider: "mozilla",
|
|
fullhash: (function() {
|
|
return hash("malware2.mochi.test/");
|
|
})(),
|
|
},
|
|
},
|
|
// Match multiple tables, handle order
|
|
{ url: "http://malware3.mochi.test",
|
|
expect: { error: Cr.NS_ERROR_MALWARE_URI,
|
|
table: "mochig3-malware-simple",
|
|
provider: "google",
|
|
fullhash: (function() {
|
|
return hash("malware3.mochi.test/");
|
|
})(),
|
|
},
|
|
},
|
|
|
|
];
|
|
|
|
SimpleTest.waitForExplicitFinish();
|
|
|
|
function setupTestData(datas) {
|
|
let prefValues = {};
|
|
for (let data of datas) {
|
|
if (!data.provider) {
|
|
continue;
|
|
}
|
|
let providerPref = "browser.safebrowsing.provider." + data.provider + ".lists";
|
|
let prefValue;
|
|
if (!prefValues[providerPref]) {
|
|
prefValue = data.db;
|
|
} else {
|
|
prefValue = prefValues[providerPref] + "," + data.db;
|
|
}
|
|
prefValues[providerPref] = prefValue;
|
|
}
|
|
|
|
// Convert map to array
|
|
let prefArray = [];
|
|
for (var pref in prefValues) {
|
|
prefArray.push([pref, prefValues[pref]]);
|
|
}
|
|
|
|
let activeTablePref = "urlclassifier.malwareTable";
|
|
let activeTable = SpecialPowers.getCharPref(activeTablePref);
|
|
for (let data of datas) {
|
|
activeTable += "," + data.db;
|
|
}
|
|
prefArray.push([activeTablePref, activeTable]);
|
|
|
|
return SpecialPowers.pushPrefEnv({set: prefArray});
|
|
}
|
|
|
|
function runTest() {
|
|
return new Promise(resolve => {
|
|
function runNextTest() {
|
|
if (!testDatas.length) {
|
|
resolve();
|
|
return;
|
|
}
|
|
let test = testDatas.shift();
|
|
let uri = SpecialPowers.Services.io.newURI(test.url);
|
|
let prin = SpecialPowers.Services.scriptSecurityManager.createContentPrincipal(uri, {});
|
|
SpecialPowers.doUrlClassify(prin, function(errorCode, table, provider, fullhash) {
|
|
is(errorCode, test.expect.error, `Test url ${test.url} correct error`);
|
|
is(table, test.expect.table, `Test url ${test.url} correct table`);
|
|
is(provider, test.expect.provider, `Test url ${test.url} correct provider`);
|
|
is(fullhash, btoa(test.expect.fullhash), `Test url ${test.url} correct full hash`);
|
|
runNextTest();
|
|
});
|
|
}
|
|
runNextTest();
|
|
});
|
|
}
|
|
|
|
SpecialPowers.pushPrefEnv(
|
|
{"set": [["browser.safebrowsing.malware.enabled", true]]},
|
|
function() {
|
|
classifierHelper.waitForInit()
|
|
.then(() => setupTestData(inputDatas))
|
|
.then(() => classifierHelper.addUrlToDB(inputDatas))
|
|
.then(runTest)
|
|
.then(function() {
|
|
SimpleTest.finish();
|
|
}).catch(function(e) {
|
|
ok(false, "Some tests failed with error " + e);
|
|
SimpleTest.finish();
|
|
});
|
|
});
|
|
|
|
</script>
|
|
</pre>
|
|
</body>
|
|
</html>
|