112 lines
3.2 KiB
JavaScript
112 lines
3.2 KiB
JavaScript
/* Any copyright is dedicated to the Public Domain.
|
|
* http://creativecommons.org/publicdomain/zero/1.0/ */
|
|
|
|
/* eslint-env mozilla/frame-script */
|
|
|
|
const { classes: Cc, interfaces: Ci, results: Cr } = Components;
|
|
|
|
var dbService = Cc["@mozilla.org/url-classifier/dbservice;1"]
|
|
.getService(Ci.nsIUrlClassifierDBService);
|
|
|
|
var timer;
|
|
function setTimeout(callback, delay) {
|
|
timer = Cc["@mozilla.org/timer;1"].createInstance(Ci.nsITimer);
|
|
timer.initWithCallback({ notify: callback },
|
|
delay,
|
|
Ci.nsITimer.TYPE_ONE_SHOT);
|
|
}
|
|
|
|
function doUpdate(update) {
|
|
let listener = {
|
|
QueryInterface(iid) {
|
|
if (iid.equals(Ci.nsISupports) ||
|
|
iid.equals(Ci.nsIUrlClassifierUpdateObserver))
|
|
return this;
|
|
|
|
throw Cr.NS_ERROR_NO_INTERFACE;
|
|
},
|
|
updateUrlRequested(url) { },
|
|
streamFinished(status) { },
|
|
updateError(errorCode) {
|
|
sendAsyncMessage("updateError", errorCode);
|
|
},
|
|
updateSuccess(requestedTimeout) {
|
|
sendAsyncMessage("updateSuccess");
|
|
}
|
|
};
|
|
|
|
try {
|
|
dbService.beginUpdate(listener, "test-malware-simple,test-unwanted-simple", "");
|
|
dbService.beginStream("", "");
|
|
dbService.updateStream(update);
|
|
dbService.finishStream();
|
|
dbService.finishUpdate();
|
|
} catch (e) {
|
|
// beginUpdate may fail if there's an existing update in progress
|
|
// retry until success or testcase timeout.
|
|
setTimeout(() => { doUpdate(update); }, 1000);
|
|
}
|
|
}
|
|
|
|
function doReload() {
|
|
try {
|
|
dbService.reloadDatabase();
|
|
sendAsyncMessage("reloadSuccess");
|
|
} catch (e) {
|
|
setTimeout(() => { doReload(); }, 1000);
|
|
}
|
|
}
|
|
|
|
// SafeBrowsing.jsm is initialized after mozEntries are added. Add observer
|
|
// to receive "finished" event. For the case when this function is called
|
|
// after the event had already been notified, we lookup entries to see if
|
|
// they are already added to database.
|
|
function waitForInit() {
|
|
let observerService = Cc["@mozilla.org/observer-service;1"]
|
|
.getService(Ci.nsIObserverService);
|
|
|
|
observerService.addObserver(function() {
|
|
sendAsyncMessage("safeBrowsingInited");
|
|
}, "mozentries-update-finished");
|
|
|
|
// This url must sync with the table, url in SafeBrowsing.jsm addMozEntries
|
|
const table = "test-phish-simple";
|
|
const url = "http://itisatrap.org/firefox/its-a-trap.html";
|
|
|
|
let secMan = Cc["@mozilla.org/scriptsecuritymanager;1"]
|
|
.getService(Ci.nsIScriptSecurityManager);
|
|
let iosvc = Cc["@mozilla.org/network/io-service;1"]
|
|
.getService(Ci.nsIIOService);
|
|
|
|
let principal = secMan.createCodebasePrincipal(
|
|
iosvc.newURI(url), {});
|
|
|
|
let listener = {
|
|
QueryInterface(iid) {
|
|
if (iid.equals(Ci.nsISupports) ||
|
|
iid.equals(Ci.nsIUrlClassifierUpdateObserver))
|
|
return this;
|
|
throw Cr.NS_ERROR_NO_INTERFACE;
|
|
},
|
|
|
|
handleEvent(value) {
|
|
if (value === table) {
|
|
sendAsyncMessage("safeBrowsingInited");
|
|
}
|
|
},
|
|
};
|
|
dbService.lookup(principal, table, listener);
|
|
}
|
|
|
|
addMessageListener("doUpdate", ({ testUpdate }) => {
|
|
doUpdate(testUpdate);
|
|
});
|
|
|
|
addMessageListener("doReload", () => {
|
|
doReload();
|
|
});
|
|
|
|
addMessageListener("waitForInit", () => {
|
|
waitForInit();
|
|
});
|