@push.rocks/smartproxy 16.0.2 → 16.0.3

package/ts/proxies/smart-proxy/port-manager.ts

@@ -0,0 +1,195 @@
+import * as plugins from '../../plugins.js';
+import type { ISmartProxyOptions } from './models/interfaces.js';
+import { RouteConnectionHandler } from './route-connection-handler.js';
+
+/**
+ * PortManager handles the dynamic creation and removal of port listeners
+ * 
+ * This class provides methods to add and remove listening ports at runtime,
+ * allowing SmartProxy to adapt to configuration changes without requiring
+ * a full restart.
+ */
+export class PortManager {
+  private servers: Map<number, plugins.net.Server> = new Map();
+  private settings: ISmartProxyOptions;
+  private routeConnectionHandler: RouteConnectionHandler;
+  private isShuttingDown: boolean = false;
+
+  /**
+   * Create a new PortManager
+   * 
+   * @param settings The SmartProxy settings
+   * @param routeConnectionHandler The handler for new connections
+   */
+  constructor(
+    settings: ISmartProxyOptions,
+    routeConnectionHandler: RouteConnectionHandler
+  ) {
+    this.settings = settings;
+    this.routeConnectionHandler = routeConnectionHandler;
+  }
+
+  /**
+   * Start listening on a specific port
+   * 
+   * @param port The port number to listen on
+   * @returns Promise that resolves when the server is listening or rejects on error
+   */
+  public async addPort(port: number): Promise<void> {
+    // Check if we're already listening on this port
+    if (this.servers.has(port)) {
+      console.log(`PortManager: Already listening on port ${port}`);
+      return;
+    }
+
+    // Create a server for this port
+    const server = plugins.net.createServer((socket) => {
+      // Check if shutting down
+      if (this.isShuttingDown) {
+        socket.end();
+        socket.destroy();
+        return;
+      }
+      
+      // Delegate to route connection handler
+      this.routeConnectionHandler.handleConnection(socket);
+    }).on('error', (err: Error) => {
+      console.log(`Server Error on port ${port}: ${err.message}`);
+    });
+    
+    // Start listening on the port
+    return new Promise<void>((resolve, reject) => {
+      server.listen(port, () => {
+        const isNetworkProxyPort = this.settings.useNetworkProxy?.includes(port);
+        console.log(
+          `SmartProxy -> OK: Now listening on port ${port}${
+            isNetworkProxyPort ? ' (NetworkProxy forwarding enabled)' : ''
+          }`
+        );
+        
+        // Store the server reference
+        this.servers.set(port, server);
+        resolve();
+      }).on('error', (err) => {
+        console.log(`Failed to listen on port ${port}: ${err.message}`);
+        reject(err);
+      });
+    });
+  }
+
+  /**
+   * Stop listening on a specific port
+   * 
+   * @param port The port to stop listening on
+   * @returns Promise that resolves when the server is closed
+   */
+  public async removePort(port: number): Promise<void> {
+    // Get the server for this port
+    const server = this.servers.get(port);
+    if (!server) {
+      console.log(`PortManager: Not listening on port ${port}`);
+      return;
+    }
+
+    // Close the server
+    return new Promise<void>((resolve) => {
+      server.close((err) => {
+        if (err) {
+          console.log(`Error closing server on port ${port}: ${err.message}`);
+        } else {
+          console.log(`SmartProxy -> Stopped listening on port ${port}`);
+        }
+        
+        // Remove the server reference
+        this.servers.delete(port);
+        resolve();
+      });
+    });
+  }
+
+  /**
+   * Add multiple ports at once
+   * 
+   * @param ports Array of ports to add
+   * @returns Promise that resolves when all servers are listening
+   */
+  public async addPorts(ports: number[]): Promise<void> {
+    const uniquePorts = [...new Set(ports)];
+    await Promise.all(uniquePorts.map(port => this.addPort(port)));
+  }
+
+  /**
+   * Remove multiple ports at once
+   * 
+   * @param ports Array of ports to remove
+   * @returns Promise that resolves when all servers are closed
+   */
+  public async removePorts(ports: number[]): Promise<void> {
+    const uniquePorts = [...new Set(ports)];
+    await Promise.all(uniquePorts.map(port => this.removePort(port)));
+  }
+
+  /**
+   * Update listening ports to match the provided list
+   * 
+   * This will add any ports that aren't currently listening,
+   * and remove any ports that are no longer needed.
+   * 
+   * @param ports Array of ports that should be listening
+   * @returns Promise that resolves when all operations are complete
+   */
+  public async updatePorts(ports: number[]): Promise<void> {
+    const targetPorts = new Set(ports);
+    const currentPorts = new Set(this.servers.keys());
+    
+    // Find ports to add and remove
+    const portsToAdd = ports.filter(port => !currentPorts.has(port));
+    const portsToRemove = Array.from(currentPorts).filter(port => !targetPorts.has(port));
+    
+    // Log the changes
+    if (portsToAdd.length > 0) {
+      console.log(`PortManager: Adding new listeners for ports: ${portsToAdd.join(', ')}`);
+    }
+    
+    if (portsToRemove.length > 0) {
+      console.log(`PortManager: Removing listeners for ports: ${portsToRemove.join(', ')}`);
+    }
+    
+    // Add and remove ports
+    await this.removePorts(portsToRemove);
+    await this.addPorts(portsToAdd);
+  }
+
+  /**
+   * Get all ports that are currently listening
+   * 
+   * @returns Array of port numbers
+   */
+  public getListeningPorts(): number[] {
+    return Array.from(this.servers.keys());
+  }
+
+  /**
+   * Mark the port manager as shutting down
+   */
+  public setShuttingDown(isShuttingDown: boolean): void {
+    this.isShuttingDown = isShuttingDown;
+  }
+
+  /**
+   * Close all listening servers
+   * 
+   * @returns Promise that resolves when all servers are closed
+   */
+  public async closeAll(): Promise<void> {
+    const allPorts = Array.from(this.servers.keys());
+    await this.removePorts(allPorts);
+  }
+
+  /**
+   * Get all server instances (for testing or debugging)
+   */
+  public getServers(): Map<number, plugins.net.Server> {
+    return new Map(this.servers);
+  }
+}

package/ts/proxies/smart-proxy/route-connection-handler.ts

@@ -8,7 +8,8 @@ import {
 } from './models/interfaces.js';
 import type {
   IRouteConfig,
-  IRouteAction
+  IRouteAction,
+  IRouteContext
 } from './models/route-types.js';
 import { ConnectionManager } from './connection-manager.js';
 import { SecurityManager } from './security-manager.js';
@@ -24,6 +25,9 @@ import type { ForwardingHandler } from '../../forwarding/handlers/base-handler.j
 export class RouteConnectionHandler {
   private settings: ISmartProxyOptions;
 
+  // Cache for route contexts to avoid recreation
+  private routeContextCache: Map<string, IRouteContext> = new Map();
+
   constructor(
     settings: ISmartProxyOptions,
     private connectionManager: ConnectionManager,
@@ -36,6 +40,47 @@ export class RouteConnectionHandler {
     this.settings = settings;
   }
 
+  /**
+   * Create a route context object for port and host mapping functions
+   */
+  private createRouteContext(options: {
+    connectionId: string;
+    port: number;
+    domain?: string;
+    clientIp: string;
+    serverIp: string;
+    isTls: boolean;
+    tlsVersion?: string;
+    routeName?: string;
+    routeId?: string;
+    path?: string;
+    query?: string;
+    headers?: Record<string, string>;
+  }): IRouteContext {
+    return {
+      // Connection information
+      port: options.port,
+      domain: options.domain,
+      clientIp: options.clientIp,
+      serverIp: options.serverIp,
+      path: options.path,
+      query: options.query,
+      headers: options.headers,
+
+      // TLS information
+      isTls: options.isTls,
+      tlsVersion: options.tlsVersion,
+
+      // Route information
+      routeName: options.routeName,
+      routeId: options.routeId,
+
+      // Additional properties
+      timestamp: Date.now(),
+      connectionId: options.connectionId
+    };
+  }
+
   /**
    * Handle a new incoming connection
    */
@@ -325,7 +370,7 @@ export class RouteConnectionHandler {
   ): void {
     const connectionId = record.id;
     const action = route.action;
-    
+
     // We should have a target configuration for forwarding
     if (!action.target) {
       console.log(`[${connectionId}] Forward action missing target configuration`);
@@ -333,24 +378,82 @@ export class RouteConnectionHandler {
       this.connectionManager.cleanupConnection(record, 'missing_target');
       return;
     }
-    
+
+    // Create the routing context for this connection
+    const routeContext = this.createRouteContext({
+      connectionId: record.id,
+      port: record.localPort,
+      domain: record.lockedDomain,
+      clientIp: record.remoteIP,
+      serverIp: socket.localAddress || '',
+      isTls: record.isTLS || false,
+      tlsVersion: record.tlsVersion,
+      routeName: route.name,
+      routeId: route.id
+    });
+
+    // Cache the context for potential reuse
+    this.routeContextCache.set(connectionId, routeContext);
+
+    // Determine host using function or static value
+    let targetHost: string | string[];
+    if (typeof action.target.host === 'function') {
+      try {
+        targetHost = action.target.host(routeContext);
+        if (this.settings.enableDetailedLogging) {
+          console.log(`[${connectionId}] Dynamic host resolved to: ${Array.isArray(targetHost) ? targetHost.join(', ') : targetHost}`);
+        }
+      } catch (err) {
+        console.log(`[${connectionId}] Error in host mapping function: ${err}`);
+        socket.end();
+        this.connectionManager.cleanupConnection(record, 'host_mapping_error');
+        return;
+      }
+    } else {
+      targetHost = action.target.host;
+    }
+
+    // If an array of hosts, select one randomly for load balancing
+    const selectedHost = Array.isArray(targetHost)
+      ? targetHost[Math.floor(Math.random() * targetHost.length)]
+      : targetHost;
+
+    // Determine port using function or static value
+    let targetPort: number;
+    if (typeof action.target.port === 'function') {
+      try {
+        targetPort = action.target.port(routeContext);
+        if (this.settings.enableDetailedLogging) {
+          console.log(`[${connectionId}] Dynamic port mapping: ${record.localPort} -> ${targetPort}`);
+        }
+        // Store the resolved target port in the context for potential future use
+        routeContext.targetPort = targetPort;
+      } catch (err) {
+        console.log(`[${connectionId}] Error in port mapping function: ${err}`);
+        socket.end();
+        this.connectionManager.cleanupConnection(record, 'port_mapping_error');
+        return;
+      }
+    } else if (action.target.preservePort) {
+      // Use incoming port if preservePort is true
+      targetPort = record.localPort;
+    } else {
+      // Use static port from configuration
+      targetPort = action.target.port;
+    }
+
+    // Store the resolved host in the context
+    routeContext.targetHost = selectedHost;
+
     // Determine if this needs TLS handling
     if (action.tls) {
       switch (action.tls.mode) {
         case 'passthrough':
           // For TLS passthrough, just forward directly
           if (this.settings.enableDetailedLogging) {
-            console.log(`[${connectionId}] Using TLS passthrough to ${action.target.host}`);
+            console.log(`[${connectionId}] Using TLS passthrough to ${selectedHost}:${targetPort}`);
           }
           
-          // Allow for array of hosts
-          const targetHost = Array.isArray(action.target.host) 
-            ? action.target.host[Math.floor(Math.random() * action.target.host.length)]
-            : action.target.host;
-          
-          // Determine target port - either target port or preserve incoming port
-          const targetPort = action.target.preservePort ? record.localPort : action.target.port;
-          
           return this.setupDirectConnection(
             socket,
             record,
@@ -358,7 +461,7 @@ export class RouteConnectionHandler {
             record.lockedDomain,
             initialChunk,
             undefined,
-            targetHost,
+            selectedHost,
             targetPort
           );
           
@@ -402,14 +505,36 @@ export class RouteConnectionHandler {
         console.log(`[${connectionId}] Using basic forwarding to ${action.target.host}:${action.target.port}`);
       }
       
-      // Allow for array of hosts
-      const targetHost = Array.isArray(action.target.host) 
-        ? action.target.host[Math.floor(Math.random() * action.target.host.length)]
-        : action.target.host;
-      
-      // Determine target port - either target port or preserve incoming port
-      const targetPort = action.target.preservePort ? record.localPort : action.target.port;
-      
+      // Get the appropriate host value
+      let targetHost: string;
+
+      if (typeof action.target.host === 'function') {
+        // For function-based host, use the same routeContext created earlier
+        const hostResult = action.target.host(routeContext);
+        targetHost = Array.isArray(hostResult)
+          ? hostResult[Math.floor(Math.random() * hostResult.length)]
+          : hostResult;
+      } else {
+        // For static host value
+        targetHost = Array.isArray(action.target.host)
+          ? action.target.host[Math.floor(Math.random() * action.target.host.length)]
+          : action.target.host;
+      }
+
+      // Determine port - either function-based, static, or preserve incoming port
+      let targetPort: number;
+      if (typeof action.target.port === 'function') {
+        targetPort = action.target.port(routeContext);
+      } else if (action.target.preservePort) {
+        targetPort = record.localPort;
+      } else {
+        targetPort = action.target.port;
+      }
+
+      // Update the connection record and context with resolved values
+      record.targetHost = targetHost;
+      record.targetPort = targetPort;
+
       return this.setupDirectConnection(
         socket,
         record,
@@ -552,13 +677,23 @@ export class RouteConnectionHandler {
 
     // Determine target host and port if not provided
     const finalTargetHost = targetHost ||
+      record.targetHost ||
       (this.settings.defaults?.target?.host || 'localhost');
 
     // Determine target port
     const finalTargetPort = targetPort ||
+      record.targetPort ||
       (overridePort !== undefined ? overridePort :
        (this.settings.defaults?.target?.port || 443));
 
+    // Update record with final target information
+    record.targetHost = finalTargetHost;
+    record.targetPort = finalTargetPort;
+
+    if (this.settings.enableDetailedLogging) {
+      console.log(`[${connectionId}] Setting up direct connection to ${finalTargetHost}:${finalTargetPort}`);
+    }
+
     // Setup connection options
     const connectionOptions: plugins.net.NetConnectOpts = {
       host: finalTargetHost,

package/ts/proxies/smart-proxy/route-manager.ts

@@ -58,36 +58,88 @@ export class RouteManager extends plugins.EventEmitter {
   
   /**
    * Rebuild the port mapping for fast lookups
+   * Also logs information about the ports being listened on
    */
   private rebuildPortMap(): void {
     this.portMap.clear();
-    
+    this.portRangeCache.clear(); // Clear cache when rebuilding
+
+    // Track ports for logging
+    const portToRoutesMap = new Map<number, string[]>();
+
     for (const route of this.routes) {
       const ports = this.expandPortRange(route.match.ports);
-      
+
+      // Skip if no ports were found
+      if (ports.length === 0) {
+        console.warn(`Route ${route.name || 'unnamed'} has no valid ports to listen on`);
+        continue;
+      }
+
       for (const port of ports) {
+        // Add to portMap for routing
         if (!this.portMap.has(port)) {
           this.portMap.set(port, []);
         }
         this.portMap.get(port)!.push(route);
+
+        // Add to tracking for logging
+        if (!portToRoutesMap.has(port)) {
+          portToRoutesMap.set(port, []);
+        }
+        portToRoutesMap.get(port)!.push(route.name || 'unnamed');
+      }
+    }
+
+    // Log summary of ports and routes
+    const totalPorts = this.portMap.size;
+    const totalRoutes = this.routes.length;
+    console.log(`Route manager configured with ${totalRoutes} routes across ${totalPorts} ports`);
+
+    // Log port details if detailed logging is enabled
+    const enableDetailedLogging = this.options.enableDetailedLogging;
+    if (enableDetailedLogging) {
+      for (const [port, routes] of this.portMap.entries()) {
+        console.log(`Port ${port}: ${routes.length} routes (${portToRoutesMap.get(port)!.join(', ')})`);
       }
     }
   }
   
   /**
    * Expand a port range specification into an array of individual ports
+   * Uses caching to improve performance for frequently used port ranges
+   *
+   * @public - Made public to allow external code to interpret port ranges
    */
-  private expandPortRange(portRange: TPortRange): number[] {
+  public expandPortRange(portRange: TPortRange): number[] {
+    // For simple number, return immediately
     if (typeof portRange === 'number') {
       return [portRange];
     }
-    
+
+    // Create a cache key for this port range
+    const cacheKey = JSON.stringify(portRange);
+
+    // Check if we have a cached result
+    if (this.portRangeCache.has(cacheKey)) {
+      return this.portRangeCache.get(cacheKey)!;
+    }
+
+    // Process the port range
+    let result: number[] = [];
+
     if (Array.isArray(portRange)) {
       // Handle array of port objects or numbers
-      return portRange.flatMap(item => {
+      result = portRange.flatMap(item => {
         if (typeof item === 'number') {
           return [item];
         } else if (typeof item === 'object' && 'from' in item && 'to' in item) {
+          // Handle port range object - check valid range
+          if (item.from > item.to) {
+            console.warn(`Invalid port range: from (${item.from}) > to (${item.to})`);
+            return [];
+          }
+
           // Handle port range object
           const ports: number[] = [];
           for (let p = item.from; p <= item.to; p++) {
@@ -98,14 +150,24 @@ export class RouteManager extends plugins.EventEmitter {
         return [];
       });
     }
-    
-    return [];
+
+    // Cache the result
+    this.portRangeCache.set(cacheKey, result);
+
+    return result;
   }
   
+  /**
+   * Memoization cache for expanded port ranges
+   */
+  private portRangeCache: Map<string, number[]> = new Map();
+
   /**
    * Get all ports that should be listened on
+   * This method automatically infers all required ports from route configurations
    */
   public getListeningPorts(): number[] {
+    // Return the unique set of ports from all routes
     return Array.from(this.portMap.keys());
   }
   
@@ -182,21 +244,36 @@ export class RouteManager extends plugins.EventEmitter {
    * Match an IP against a pattern
    */
   private matchIpPattern(pattern: string, ip: string): boolean {
-    // Handle exact match
-    if (pattern === ip) {
+    // Normalize IPv6-mapped IPv4 addresses
+    const normalizedIp = ip.startsWith('::ffff:') ? ip.substring(7) : ip;
+    const normalizedPattern = pattern.startsWith('::ffff:') ? pattern.substring(7) : pattern;
+    
+    // Handle exact match with normalized addresses
+    if (pattern === ip || normalizedPattern === normalizedIp || 
+        pattern === normalizedIp || normalizedPattern === ip) {
       return true;
     }
     
     // Handle CIDR notation (e.g., 192.168.1.0/24)
     if (pattern.includes('/')) {
-      return this.matchIpCidr(pattern, ip);
+      return this.matchIpCidr(pattern, normalizedIp) || 
+             (normalizedPattern !== pattern && this.matchIpCidr(normalizedPattern, normalizedIp));
     }
     
     // Handle glob pattern (e.g., 192.168.1.*)
     if (pattern.includes('*')) {
       const regexPattern = pattern.replace(/\./g, '\\.').replace(/\*/g, '.*');
       const regex = new RegExp(`^${regexPattern}$`);
-      return regex.test(ip);
+      if (regex.test(ip) || regex.test(normalizedIp)) {
+        return true;
+      }
+      
+      // If pattern was normalized, also test with normalized pattern
+      if (normalizedPattern !== pattern) {
+        const normalizedRegexPattern = normalizedPattern.replace(/\./g, '\\.').replace(/\*/g, '.*');
+        const normalizedRegex = new RegExp(`^${normalizedRegexPattern}$`);
+        return normalizedRegex.test(ip) || normalizedRegex.test(normalizedIp);
+      }
     }
     
     return false;
@@ -212,9 +289,13 @@ export class RouteManager extends plugins.EventEmitter {
       const [subnet, bits] = cidr.split('/');
       const mask = parseInt(bits, 10);
       
+      // Normalize IPv6-mapped IPv4 addresses
+      const normalizedIp = ip.startsWith('::ffff:') ? ip.substring(7) : ip;
+      const normalizedSubnet = subnet.startsWith('::ffff:') ? subnet.substring(7) : subnet;
+      
       // Convert IP addresses to numeric values
-      const ipNum = this.ipToNumber(ip);
-      const subnetNum = this.ipToNumber(subnet);
+      const ipNum = this.ipToNumber(normalizedIp);
+      const subnetNum = this.ipToNumber(normalizedSubnet);
       
       // Calculate subnet mask
       const maskNum = ~(2 ** (32 - mask) - 1);
@@ -231,7 +312,10 @@ export class RouteManager extends plugins.EventEmitter {
    * Convert an IP address to a numeric value
    */
   private ipToNumber(ip: string): number {
-    const parts = ip.split('.').map(part => parseInt(part, 10));
+    // Normalize IPv6-mapped IPv4 addresses
+    const normalizedIp = ip.startsWith('::ffff:') ? ip.substring(7) : ip;
+    
+    const parts = normalizedIp.split('.').map(part => parseInt(part, 10));
     return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3];
   }