mirror of
https://github.com/angular/angular
synced 2026-05-24 09:28:37 +00:00
a6b2d7dd10
One common problem encountered by the devtools content script is that it accepted almost any message send over the message bus. Some websites like `auth.openai.com` were spamming the bus and DDOS the devtools app. By introducing event tagging and skipping non-devtools events we prevent DDOS of the Angular devtools content script by on forward tagged events. PR Close #62645
86 lines
2.4 KiB
TypeScript
86 lines
2.4 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright Google LLC All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by an MIT-style license that can be
|
|
* found in the LICENSE file at https://angular.dev/license
|
|
*/
|
|
|
|
/// <reference types="chrome"/>
|
|
|
|
import {Events, MessageBus, Parameters} from '../../../protocol';
|
|
|
|
interface ChromeMessage<T, K extends keyof T> {
|
|
topic: K;
|
|
args: Parameters<T[K]>;
|
|
}
|
|
|
|
type AnyEventCallback<Ev> = <E extends keyof Ev>(topic: E, args: Parameters<Ev[E]>) => void;
|
|
|
|
export class ChromeMessageBus extends MessageBus<Events> {
|
|
private _disconnected = false;
|
|
private _listeners: any[] = [];
|
|
|
|
constructor(private _port: chrome.runtime.Port) {
|
|
super();
|
|
|
|
_port.onDisconnect.addListener(() => {
|
|
// console.log('Disconnected the port');
|
|
this._disconnected = true;
|
|
});
|
|
}
|
|
|
|
onAny(cb: AnyEventCallback<Events>): () => void {
|
|
const listener = (msg: ChromeMessage<Events, keyof Events>): void => {
|
|
cb(msg.topic, msg.args);
|
|
};
|
|
this._port.onMessage.addListener(listener);
|
|
this._listeners.push(listener);
|
|
return () => {
|
|
this._listeners.splice(this._listeners.indexOf(listener), 1);
|
|
this._port.onMessage.removeListener(listener);
|
|
};
|
|
}
|
|
|
|
override on<E extends keyof Events>(topic: E, cb: Events[E]): () => void {
|
|
const listener = (msg: ChromeMessage<Events, keyof Events>): void => {
|
|
if (msg.topic === topic) {
|
|
(cb as any).apply(null, msg.args);
|
|
}
|
|
};
|
|
this._port.onMessage.addListener(listener);
|
|
this._listeners.push(listener);
|
|
return () => {
|
|
this._listeners.splice(this._listeners.indexOf(listener), 1);
|
|
this._port.onMessage.removeListener(listener);
|
|
};
|
|
}
|
|
|
|
override once<E extends keyof Events>(topic: E, cb: Events[E]): void {
|
|
const listener = (msg: ChromeMessage<Events, keyof Events>) => {
|
|
if (msg.topic === topic) {
|
|
(cb as any).apply(null, msg.args);
|
|
this._port.onMessage.removeListener(listener);
|
|
}
|
|
};
|
|
this._port.onMessage.addListener(listener);
|
|
}
|
|
|
|
override emit<E extends keyof Events>(topic: E, args?: Parameters<Events[E]>): boolean {
|
|
if (this._disconnected) {
|
|
return false;
|
|
}
|
|
this._port.postMessage({
|
|
topic,
|
|
args,
|
|
__ignore_ng_zone__: true,
|
|
__NG_DEVTOOLS_EVENT__: true,
|
|
});
|
|
return true;
|
|
}
|
|
|
|
override destroy(): void {
|
|
this._listeners.forEach((l) => window.removeEventListener('message', l));
|
|
this._listeners = [];
|
|
}
|
|
}
|