Source code
Revision control
Copy as Markdown
Other Tools
/* 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
const lazy = {};
ChromeUtils.defineESModuleGetters(lazy, {
IPPProxyManager:
"moz-src:///toolkit/components/ipprotection/IPPProxyManager.sys.mjs",
IPPProxyStates:
"moz-src:///toolkit/components/ipprotection/IPPProxyManager.sys.mjs",
IPProtectionService:
"moz-src:///toolkit/components/ipprotection/IPProtectionService.sys.mjs",
IPProtectionStates:
"moz-src:///toolkit/components/ipprotection/IPProtectionService.sys.mjs",
});
/**
* This class monitors the startup phases and registers/unregisters the channel
* filter to avoid data leak. The activation of the VPN is done by the
* IPPAutoStart and IPPAutoRestore objects above.
*
* @param {() => boolean} shouldActivate - Determines whether the filter should be activated at startup.
*/
export class IPPEarlyStartupFilter {
#autoStartAndAtStartup = false;
constructor(shouldActivate) {
this.handleEvent = this.#handleEvent.bind(this);
this.#autoStartAndAtStartup = shouldActivate();
}
init() {
if (this.#autoStartAndAtStartup) {
lazy.IPPProxyManager.createChannelFilter();
lazy.IPProtectionService.addEventListener(
"IPProtectionService:StateChanged",
this.handleEvent
);
lazy.IPPProxyManager.addEventListener(
"IPPProxyManager:StateChanged",
this.handleEvent
);
}
}
initOnStartupCompleted() {}
uninit() {
if (this.#autoStartAndAtStartup) {
this.#autoStartAndAtStartup = false;
lazy.IPPProxyManager.removeEventListener(
"IPPProxyManager:StateChanged",
this.handleEvent
);
lazy.IPProtectionService.removeEventListener(
"IPProtectionService:StateChanged",
this.handleEvent
);
}
}
#cancelChannelFilter() {
lazy.IPPProxyManager.cancelChannelFilter();
}
#handleEvent(_event) {
switch (lazy.IPProtectionService.state) {
case lazy.IPProtectionStates.UNAVAILABLE:
case lazy.IPProtectionStates.UNAUTHENTICATED:
// These states block the auto-start at startup.
this.#cancelChannelFilter();
this.uninit();
break;
default:
// Let's ignoring any other state.
break;
}
if (lazy.IPPProxyManager.state === lazy.IPPProxyStates.ACTIVE) {
// We have completed our task.
this.uninit();
}
}
}