yujunjun/playwright
1
0
Fork 0
mirror of https://github.com/microsoft/playwright.git synced 2025-06-26 21:40:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: reduce file reads in client-certificates internal TLS server (#31865)

Browse Source
This commit is contained in:
Max Schmitt 2024-07-25 22:36:25 +02:00 committed by GitHub
parent a41cebc1c9
commit a02ed38e60
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
packages/playwright-core/src/server/socksClientCertificatesInterceptor.ts
View File

@ -27,6 +27,16 @@ import { SocksProxy } from '../common/socksProxy';
import type * as channels from '@protocol/channels'; import type * as channels from '@protocol/channels';
import { debugLogger } from '../utils/debugLogger'; import { debugLogger } from '../utils/debugLogger';
let dummyServerTlsOptions: tls.TlsOptions | undefined = undefined;
function loadDummyServerCertsIfNeeded() {
if (dummyServerTlsOptions)
return;
dummyServerTlsOptions = {
key: fs.readFileSync(path.join(__dirname, '../../bin/socks-certs/key.pem')),
cert: fs.readFileSync(path.join(__dirname, '../../bin/socks-certs/cert.pem')),
};
}
class ALPNCache { class ALPNCache {
private _cache = new Map<string, ManualPromise<string>>(); private _cache = new Map<string, ManualPromise<string>>();
@ -122,8 +132,7 @@ class SocksProxyConnection {
this.socksProxy.alpnCache.get(rewriteToLocalhostIfNeeded(this.host), this.port, alpnProtocolChosenByServer => { this.socksProxy.alpnCache.get(rewriteToLocalhostIfNeeded(this.host), this.port, alpnProtocolChosenByServer => {
debugLogger.log('client-certificates', `Proxy->Target ${this.host}:${this.port} chooses ALPN ${alpnProtocolChosenByServer}`); debugLogger.log('client-certificates', `Proxy->Target ${this.host}:${this.port} chooses ALPN ${alpnProtocolChosenByServer}`);
const dummyServer = tls.createServer({ const dummyServer = tls.createServer({
key: fs.readFileSync(path.join(__dirname, '../../bin/socks-certs/key.pem')), ...dummyServerTlsOptions,
cert: fs.readFileSync(path.join(__dirname, '../../bin/socks-certs/cert.pem')),
ALPNProtocols: alpnProtocolChosenByServer === 'h2' ? ['h2', 'http/1.1'] : ['http/1.1'], ALPNProtocols: alpnProtocolChosenByServer === 'h2' ? ['h2', 'http/1.1'] : ['http/1.1'],
}); });
this.internal?.on('close', () => dummyServer.close()); this.internal?.on('close', () => dummyServer.close());
@ -211,6 +220,7 @@ export class ClientCertificatesProxy {
this._connections.get(payload.uid)?.onClose(); this._connections.get(payload.uid)?.onClose();
this._connections.delete(payload.uid); this._connections.delete(payload.uid);
}); });
loadDummyServerCertsIfNeeded();
} }
public async listen(): Promise<string> { public async listen(): Promise<string> {