Filip/playwright
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

properly handle errors

Browse source
This commit is contained in:
Simon Knott 2024-12-16 12:28:18 +01:00
parent 2e019eb746
commit 41aeb0ee56
No known key found for this signature in database
GPG key ID: 8CEDC00028084AEC
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/third_party/proxy/index.ts vendored
View file

@ -3,6 +3,7 @@ import * as net from 'net';
import * as url from 'url'; import * as url from 'url';
import * as http from 'http'; import * as http from 'http';
import * as os from 'os'; import * as os from 'os';
import { pipeline } from 'stream/promises';
const pkg = { version: '1.0.0' } const pkg = { version: '1.0.0' }
@ -475,13 +476,19 @@ function onupgrade(req: http.IncomingMessage, socket: net.Socket, head: Buffer)
localAddress: this.localAddress, localAddress: this.localAddress,
}); });
proxyReq.on('upgrade', function (proxyRes, proxySocket, proxyHead) { proxyReq.on('upgrade', async function (proxyRes, proxySocket, proxyHead) {
const header = ['HTTP/1.1 101 Switching Protocols']; const header = ['HTTP/1.1 101 Switching Protocols'];
for (const [key, value] of Object.entries(proxyRes.headersDistinct)) for (const [key, value] of Object.entries(proxyRes.headersDistinct))
header.push(`${key}: ${value}`); header.push(`${key}: ${value}`);
socket.write(header.join('\r\n') + '\r\n\r\n'); socket.write(header.join('\r\n') + '\r\n\r\n');
if (proxyHead && proxyHead.length) proxySocket.unshift(proxyHead); if (proxyHead && proxyHead.length) proxySocket.unshift(proxyHead);
proxySocket.pipe(socket).pipe(proxySocket);
try {
await pipeline(proxySocket, socket, proxySocket);
} catch (error) {
if (error.code !== "ECONNRESET")
throw error;
}
}); });
proxyReq.end(head); proxyReq.end(head);