RevelaNetwork/Revela
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

protect WebSocket methods

Browse source
This commit is contained in:
David Reed 2022-11-19 18:18:49 -05:00
parent ff8708438e
commit ad0de820ef
No known key found for this signature in database
GPG key ID: 2211691D8A1EE72F
1 changed files with 30 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
src/uv.handler.js
View file

@ -1143,6 +1143,36 @@ function __uvHook(window, config = {}, bare = '/bare/') {
eventTarget(MockWebSocket.prototype, 'message');
eventTarget(MockWebSocket.prototype, 'error');
for (const hook of [
'url',
'protocol',
'extensions',
'readyState',
'binaryType',
]) {
const officialDesc = Object.getOwnPropertyDescriptor(
WebSocket.prototype,
hook
);
const customDesc = Object.getOwnPropertyDescriptor(
MockWebSocket.prototype,
hook
);
if (customDesc?.get && officialDesc?.get)
client.emit('wrap', customDesc.get, officialDesc.get);
if (customDesc?.set && officialDesc?.set)
client.emit('wrap', customDesc.get, officialDesc.get);
}
client.emit('wrap', WebSocket.prototype.send, MockWebSocket.prototype.send);
client.emit(
'wrap',
WebSocket.prototype.close,
MockWebSocket.prototype.close
);
client.override(
window,
'WebSocket',