Source code
Revision control
Copy as Markdown
Other Tools
/* -*- js-indent-level: 2; indent-tabs-mode: nil -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
import { AppConstants } from "resource://gre/modules/AppConstants.sys.mjs";
import { XPCOMUtils } from "resource://gre/modules/XPCOMUtils.sys.mjs";
const lazy = {};
XPCOMUtils.defineLazyServiceGetters(lazy, {
WinTaskSvc: [
"@mozilla.org/win-task-scheduler-service;1",
"nsIWinTaskSchedulerService",
],
XreDirProvider: [
"@mozilla.org/xre/directory-provider;1",
"nsIXREDirProvider",
],
});
/**
* Task generation and management for Windows, using Task Scheduler 2.0 (taskschd).
*
* Implements the API exposed in TaskScheduler.sys.mjs
* Not intended for external use, this is in a separate module to ship the code only
* on Windows, and to expose for testing.
*/
export var WinImpl = {
registerTask(id, command, intervalSeconds, options) {
// The folder might not yet exist.
this._createFolderIfNonexistent();
const xml = this._formatTaskDefinitionXML(
command,
intervalSeconds,
options
);
const updateExisting = true;
lazy.WinTaskSvc.registerTask(
this._taskFolderName(),
this._formatTaskName(id, options),
xml,
updateExisting
);
},
deleteTask(id, options) {
lazy.WinTaskSvc.deleteTask(
this._taskFolderName(),
this._formatTaskName(id, options)
);
},
/**
* Delete all tasks created by this installation.
*
* The Windows Default Browser Agent task is special: it's
* registered by the installer and might run as a different user and
* require permissions to delete. We ignore it and leave it for the
* uninstaller to remove.
*/
deleteAllTasks() {
const taskFolderName = this._taskFolderName();
let allTasks;
try {
allTasks = lazy.WinTaskSvc.getFolderTasks(taskFolderName);
} catch (ex) {
if (ex.result == Cr.NS_ERROR_FILE_NOT_FOUND) {
// Folder doesn't exist, nothing to delete.
return;
}
throw ex;
}
const tasksToDelete = allTasks.filter(name => this._matchAppTaskName(name));
let numberDeleted = 0;
let lastFailedTaskName;
// We need `MOZ_APP_DISPLAYNAME` since that's what the WDBA (written in C++) uses.
const defaultBrowserAgentTaskName =
AppConstants.MOZ_APP_DISPLAYNAME_DO_NOT_USE +
" Default Browser Agent " +
lazy.XreDirProvider.getInstallHash();
for (const taskName of tasksToDelete) {
if (taskName == defaultBrowserAgentTaskName) {
// Skip the Windows Default Browser Agent task.
continue;
}
try {
lazy.WinTaskSvc.deleteTask(taskFolderName, taskName);
numberDeleted += 1;
} catch (e) {
lastFailedTaskName = taskName;
}
}
if (lastFailedTaskName) {
// There's no standard way to chain exceptions, so instead try again,
// which should fail and throw again. It's possible this isn't idempotent
// but we're expecting failures to be due to permission errors, which are
// likely to be static.
lazy.WinTaskSvc.deleteTask(taskFolderName, lastFailedTaskName);
}
if (allTasks.length == numberDeleted) {
// Deleted every task, remove the folder.
this._deleteFolderIfEmpty();
}
},
taskExists(id, options) {
const taskFolderName = this._taskFolderName();
let allTasks;
try {
allTasks = lazy.WinTaskSvc.getFolderTasks(taskFolderName);
} catch (ex) {
if (ex.result == Cr.NS_ERROR_FILE_NOT_FOUND) {
// Folder doesn't exist, so neither do tasks within it.
return false;
}
throw ex;
}
return allTasks.includes(this._formatTaskName(id, options));
},
_formatTaskDefinitionXML(command, intervalSeconds, options) {
const startTime = new Date(Date.now() + intervalSeconds * 1000);
// Fill in the constant parts of the task, and those that don't require escaping.
const docBase = `<Task xmlns="${xmlns}">
<Triggers>
<TimeTrigger>
<StartBoundary>${startTime.toISOString()}</StartBoundary>
<Repetition>
<Interval>PT${intervalSeconds}S</Interval>
</Repetition>
</TimeTrigger>
</Triggers>
<Actions>
<Exec />
</Actions>
<Settings>
<StartWhenAvailable>true</StartWhenAvailable>
<MultipleInstancesPolicy>IgnoreNew</MultipleInstancesPolicy>
</Settings>
<RegistrationInfo>
<Author />
</RegistrationInfo>
</Task>`;
const doc = new DOMParser().parseFromString(docBase, "text/xml");
const execAction = doc.querySelector("Actions Exec");
const settings = doc.querySelector("Settings");
const commandNode = doc.createElementNS(xmlns, "Command");
commandNode.textContent = command;
execAction.appendChild(commandNode);
if (options?.args) {
const args = doc.createElementNS(xmlns, "Arguments");
args.textContent = options.args.map(this._quoteString).join(" ");
execAction.appendChild(args);
}
if (options?.workingDirectory) {
const workingDirectory = doc.createElementNS(xmlns, "WorkingDirectory");
workingDirectory.textContent = options.workingDirectory;
execAction.appendChild(workingDirectory);
}
if (options?.disabled) {
const enabled = doc.createElementNS(xmlns, "Enabled");
enabled.textContent = "false";
settings.appendChild(enabled);
}
if (options?.executionTimeoutSec && options.executionTimeoutSec > 0) {
const timeout = doc.createElementNS(xmlns, "ExecutionTimeLimit");
timeout.textContent = `PT${options.executionTimeoutSec}S`;
settings.appendChild(timeout);
}
// Other settings to consider for the future:
// Idle
// Battery
doc.querySelector("RegistrationInfo Author").textContent =
Services.appinfo.vendor;
if (options?.description) {
const registrationInfo = doc.querySelector("RegistrationInfo");
const description = doc.createElementNS(xmlns, "Description");
description.textContent = options.description;
registrationInfo.appendChild(description);
}
const serializer = new XMLSerializer();
return serializer.serializeToString(doc);
},
_createFolderIfNonexistent() {
const { parentName, subName } = this._taskFolderNameParts();
try {
lazy.WinTaskSvc.createFolder(parentName, subName);
} catch (e) {
if (e.result != Cr.NS_ERROR_FILE_ALREADY_EXISTS) {
throw e;
}
}
},
_deleteFolderIfEmpty() {
const { parentName, subName } = this._taskFolderNameParts();
try {
lazy.WinTaskSvc.deleteFolder(parentName, subName);
} catch (e) {
// Missed one somehow, possibly a subfolder?
if (e.result != Cr.NS_ERROR_FILE_DIR_NOT_EMPTY) {
throw e;
}
}
},
/**
* Quotes a string for use as a single command argument, using Windows quoting
* conventions.
*
* copied from quoteString() in toolkit/modules/subproces/subprocess_worker_win.js
*
*
*
* @param {string} str
* The argument string to quote.
* @returns {string}
*/
_quoteString(str) {
if (!/[\s"]/.test(str)) {
return str;
}
let escaped = str.replace(/(\\*)("|$)/g, (m0, m1, m2) => {
if (m2) {
m2 = `\\${m2}`;
}
return `${m1}${m1}${m2}`;
});
return `"${escaped}"`;
},
_taskFolderName() {
return `\\${Services.appinfo.vendor}`;
},
_taskFolderNameParts() {
return {
parentName: "\\",
subName: Services.appinfo.vendor,
};
},
/**
* Formats a given task id according to one of two formats.
*
* @param id
* A string representing the identifier of the task to format
*
* @param {Object} options
* Optional, as are all of its properties:
* {
* options.nameVersion
* Specifies whether to search for tasks using nameVersion 1
* which is `${taskID} ${installHash}` or nameVersion 2 which is
* `${taskID} ${currentUserSid} ${installHash}`. Defaults to nameVersion 2.
* }
*
* @return
* Formatted task name.
*/
_formatTaskName(id, options) {
const installHash = lazy.XreDirProvider.getInstallHash();
if (options?.nameVersion == 1) {
return `${id} ${installHash}`;
}
const currentUserSid = lazy.WinTaskSvc.getCurrentUserSid();
return `${id} ${currentUserSid} ${installHash}`;
},
_matchAppTaskName(name) {
const installHash = lazy.XreDirProvider.getInstallHash();
return name.endsWith(` ${installHash}`);
},
_updateTaskNameFormat(id) {
const taskFolderName = this._taskFolderName();
const allTasks = lazy.WinTaskSvc.getFolderTasks(taskFolderName);
const taskNameV1 = this._formatTaskName(id, { nameVersion: 1 });
const taskNameV2 = this._formatTaskName(id, { nameVersion: 2 });
if (allTasks.includes(taskNameV1)) {
const taskXML = lazy.WinTaskSvc.getTaskXML(taskFolderName, taskNameV1);
lazy.WinTaskSvc.registerTask(taskFolderName, taskNameV2, taskXML, true);
lazy.WinTaskSvc.deleteTask(taskFolderName, taskNameV1);
}
},
};