feat(PortProxy): Enhanced PortProxy with custom target host and improved testing

This commit is contained in:
2025-02-21 17:01:02 +00:00
parent 21e9d0fd0d
commit 4328d4365f
4 changed files with 50 additions and 16 deletions

View File

@ -3,6 +3,6 @@
*/
export const commitinfo = {
name: '@push.rocks/smartproxy',
version: '3.3.1',
version: '3.4.0',
description: 'a proxy for handling high workloads of proxying'
}

View File

@ -6,23 +6,27 @@ export interface DomainConfig {
allowedIPs: string[]; // glob patterns for IPs allowed to access this domain
}
export interface ProxySettings {
export interface ProxySettings extends plugins.tls.TlsOptions {
// Port configuration
fromPort: number;
toPort: number;
toHost?: string; // Target host to proxy to, defaults to 'localhost'
// Domain and security settings
domains: DomainConfig[];
sniEnabled?: boolean;
tlsOptions?: plugins.tls.TlsOptions;
defaultAllowedIPs?: string[]; // Optional default IP patterns if no matching domain found
}
export class PortProxy {
netServer: plugins.net.Server | plugins.tls.Server;
fromPort: number;
toPort: number;
settings: ProxySettings;
constructor(fromPortArg: number, toPortArg: number, settings: plugins.tls.TlsOptions & ProxySettings) {
this.fromPort = fromPortArg;
this.toPort = toPortArg;
this.settings = settings;
constructor(settings: ProxySettings) {
this.settings = {
...settings,
toHost: settings.toHost || 'localhost'
};
}
public async start() {
@ -44,7 +48,9 @@ export class PortProxy {
return this.settings.domains.find(config => plugins.minimatch(serverName, config.domain));
};
const server = this.settings.sniEnabled ? plugins.tls.createServer(this.settings.tlsOptions || {}) : plugins.net.createServer();
const server = this.settings.sniEnabled
? plugins.tls.createServer(this.settings)
: plugins.net.createServer();
this.netServer = server.on('connection', (from: plugins.net.Socket) => {
const remoteIP = from.remoteAddress || '';
@ -74,8 +80,8 @@ export class PortProxy {
}
const to = plugins.net.createConnection({
host: 'localhost',
port: this.toPort,
host: this.settings.toHost!,
port: this.settings.toPort,
});
from.setTimeout(120000);
from.pipe(to);
@ -105,8 +111,8 @@ export class PortProxy {
cleanUpSockets(from, to);
});
})
.listen(this.fromPort);
console.log(`PortProxy -> OK: Now listening on port ${this.fromPort}`);
.listen(this.settings.fromPort);
console.log(`PortProxy -> OK: Now listening on port ${this.settings.fromPort}`);
}
public async stop() {