@push.rocks/smartproxy 16.0.2 → 16.0.3

package/ts/proxies/network-proxy/certificate-manager.ts

@@ -8,6 +8,7 @@ import { CertificateEvents } from '../../certificate/events/certificate-events.j
 import { buildPort80Handler } from '../../certificate/acme/acme-factory.js';
 import { subscribeToPort80Handler } from '../../core/utils/event-utils.js';
 import type { IDomainOptions } from '../../certificate/models/certificate-types.js';
+import type { IRouteConfig } from '../smart-proxy/models/route-types.js';
 
 /**
  * Manages SSL certificates for NetworkProxy including ACME integration
@@ -91,7 +92,7 @@ export class CertificateManager {
   public setExternalPort80Handler(handler: Port80Handler): void {
     if (this.port80Handler && !this.externalPort80Handler) {
       this.logger.warn('Replacing existing internal Port80Handler with external handler');
-      
+
       // Clean up existing handler if needed
       if (this.port80Handler !== handler) {
         // Unregister event handlers to avoid memory leaks
@@ -101,11 +102,11 @@ export class CertificateManager {
         this.port80Handler.removeAllListeners(CertificateEvents.CERTIFICATE_EXPIRING);
       }
     }
-    
+
     // Set the external handler
     this.port80Handler = handler;
     this.externalPort80Handler = true;
-    
+
     // Subscribe to Port80Handler events
     subscribeToPort80Handler(this.port80Handler, {
       onCertificateIssued: this.handleCertificateIssued.bind(this),
@@ -115,17 +116,40 @@ export class CertificateManager {
         this.logger.info(`Certificate for ${data.domain} expires in ${data.daysRemaining} days`);
       }
     });
-    
+
     this.logger.info('External Port80Handler connected to CertificateManager');
-    
+
     // Register domains with Port80Handler if we have any certificates cached
     if (this.certificateCache.size > 0) {
       const domains = Array.from(this.certificateCache.keys())
         .filter(domain => !domain.includes('*')); // Skip wildcard domains
-      
+
       this.registerDomainsWithPort80Handler(domains);
     }
   }
+
+  /**
+   * Update route configurations managed by this certificate manager
+   * This method is called when route configurations change
+   *
+   * @param routes Array of route configurations
+   */
+  public updateRouteConfigs(routes: IRouteConfig[]): void {
+    if (!this.port80Handler) {
+      this.logger.warn('Cannot update routes - Port80Handler is not initialized');
+      return;
+    }
+
+    // Register domains from routes with Port80Handler
+    this.registerRoutesWithPort80Handler(routes);
+
+    // Process individual routes for certificate requirements
+    for (const route of routes) {
+      this.processRouteForCertificates(route);
+    }
+
+    this.logger.info(`Updated certificate management for ${routes.length} routes`);
+  }
   
   /**
    * Handle newly issued or renewed certificates from Port80Handler
@@ -317,20 +341,21 @@ export class CertificateManager {
   
   /**
    * Registers domains with Port80Handler for ACME certificate management
+   * @param domains String array of domains to register
    */
   public registerDomainsWithPort80Handler(domains: string[]): void {
     if (!this.port80Handler) {
       this.logger.warn('Port80Handler is not initialized');
       return;
     }
-    
+
     for (const domain of domains) {
       // Skip wildcard domains - can't get certs for these with HTTP-01 validation
       if (domain.includes('*')) {
         this.logger.info(`Skipping wildcard domain for ACME: ${domain}`);
         continue;
       }
-      
+
       // Skip domains already with certificates if configured to do so
       if (this.options.acme?.skipConfiguredCerts) {
         const cachedCert = this.certificateCache.get(domain);
@@ -339,18 +364,97 @@ export class CertificateManager {
           continue;
         }
       }
-      
+
       // Register the domain for certificate issuance with new domain options format
       const domainOptions: IDomainOptions = {
         domainName: domain,
         sslRedirect: true,
         acmeMaintenance: true
       };
-      
+
       this.port80Handler.addDomain(domainOptions);
       this.logger.info(`Registered domain for ACME certificate issuance: ${domain}`);
     }
   }
+
+  /**
+   * Extract domains from route configurations and register with Port80Handler
+   * This method enables direct integration with route-based configuration
+   *
+   * @param routes Array of route configurations
+   */
+  public registerRoutesWithPort80Handler(routes: IRouteConfig[]): void {
+    if (!this.port80Handler) {
+      this.logger.warn('Port80Handler is not initialized');
+      return;
+    }
+
+    // Extract domains from route configurations
+    const domains: Set<string> = new Set();
+
+    for (const route of routes) {
+      // Skip disabled routes
+      if (route.enabled === false) {
+        continue;
+      }
+
+      // Skip routes without HTTPS termination
+      if (route.action.type !== 'forward' || route.action.tls?.mode !== 'terminate') {
+        continue;
+      }
+
+      // Extract domains from match criteria
+      if (route.match.domains) {
+        if (typeof route.match.domains === 'string') {
+          domains.add(route.match.domains);
+        } else if (Array.isArray(route.match.domains)) {
+          for (const domain of route.match.domains) {
+            domains.add(domain);
+          }
+        }
+      }
+    }
+
+    // Register extracted domains
+    this.registerDomainsWithPort80Handler(Array.from(domains));
+  }
+
+  /**
+   * Process a route config to determine if it requires automatic certificate provisioning
+   * @param route Route configuration to process
+   */
+  public processRouteForCertificates(route: IRouteConfig): void {
+    // Skip disabled routes
+    if (route.enabled === false) {
+      return;
+    }
+
+    // Skip routes without HTTPS termination or auto certificate
+    if (route.action.type !== 'forward' ||
+        route.action.tls?.mode !== 'terminate' ||
+        route.action.tls?.certificate !== 'auto') {
+      return;
+    }
+
+    // Extract domains from match criteria
+    const domains: string[] = [];
+    if (route.match.domains) {
+      if (typeof route.match.domains === 'string') {
+        domains.push(route.match.domains);
+      } else if (Array.isArray(route.match.domains)) {
+        domains.push(...route.match.domains);
+      }
+    }
+
+    // Request certificates for the domains
+    for (const domain of domains) {
+      if (!domain.includes('*')) { // Skip wildcard domains
+        this.requestCertificate(domain).catch(err => {
+          this.logger.error(`Error requesting certificate for domain ${domain}:`, err);
+        });
+      }
+    }
+  }
   
   /**
    * Initialize internal Port80Handler

package/ts/proxies/network-proxy/context-creator.ts

@@ -0,0 +1,145 @@
+import * as plugins from '../../plugins.js';
+import '../../core/models/socket-augmentation.js';
+import type { IRouteContext, IHttpRouteContext, IHttp2RouteContext } from '../../core/models/route-context.js';
+
+/**
+ * Context creator for NetworkProxy
+ * Creates route contexts for matching and function evaluation
+ */
+export class ContextCreator {
+  /**
+   * Create a route context from HTTP request information
+   */
+  public createHttpRouteContext(req: any, options: {
+    tlsVersion?: string;
+    connectionId: string;
+    clientIp: string;
+    serverIp: string;
+  }): IHttpRouteContext {
+    // Parse headers
+    const headers: Record<string, string> = {};
+    for (const [key, value] of Object.entries(req.headers)) {
+      if (typeof value === 'string') {
+        headers[key.toLowerCase()] = value;
+      } else if (Array.isArray(value) && value.length > 0) {
+        headers[key.toLowerCase()] = value[0];
+      }
+    }
+
+    // Parse domain from Host header
+    const domain = headers['host']?.split(':')[0] || '';
+
+    // Parse URL
+    const url = new URL(`http://${domain}${req.url || '/'}`);
+
+    return {
+      // Connection basics
+      port: req.socket.localPort || 0,
+      domain,
+      clientIp: options.clientIp,
+      serverIp: options.serverIp,
+
+      // HTTP specifics
+      path: url.pathname,
+      query: url.search ? url.search.substring(1) : '',
+      headers,
+
+      // TLS information
+      isTls: !!req.socket.encrypted,
+      tlsVersion: options.tlsVersion,
+
+      // Request objects
+      req,
+
+      // Metadata
+      timestamp: Date.now(),
+      connectionId: options.connectionId
+    };
+  }
+
+  /**
+   * Create a route context from HTTP/2 stream and headers
+   */
+  public createHttp2RouteContext(
+    stream: plugins.http2.ServerHttp2Stream,
+    headers: plugins.http2.IncomingHttpHeaders,
+    options: {
+      connectionId: string;
+      clientIp: string;
+      serverIp: string;
+    }
+  ): IHttp2RouteContext {
+    // Parse headers, excluding HTTP/2 pseudo-headers
+    const processedHeaders: Record<string, string> = {};
+    for (const [key, value] of Object.entries(headers)) {
+      if (!key.startsWith(':') && typeof value === 'string') {
+        processedHeaders[key.toLowerCase()] = value;
+      }
+    }
+
+    // Get domain from :authority pseudo-header
+    const authority = headers[':authority'] as string || '';
+    const domain = authority.split(':')[0];
+
+    // Get path from :path pseudo-header
+    const path = headers[':path'] as string || '/';
+
+    // Parse the path to extract query string
+    const pathParts = path.split('?');
+    const pathname = pathParts[0];
+    const query = pathParts.length > 1 ? pathParts[1] : '';
+
+    // Get the socket from the session
+    const socket = (stream.session as any)?.socket;
+
+    return {
+      // Connection basics
+      port: socket?.localPort || 0,
+      domain,
+      clientIp: options.clientIp,
+      serverIp: options.serverIp,
+
+      // HTTP specifics
+      path: pathname,
+      query,
+      headers: processedHeaders,
+
+      // HTTP/2 specific properties
+      method: headers[':method'] as string,
+      stream,
+
+      // TLS information - HTTP/2 is always on TLS in browsers
+      isTls: true,
+      tlsVersion: socket?.getTLSVersion?.() || 'TLSv1.3',
+
+      // Metadata
+      timestamp: Date.now(),
+      connectionId: options.connectionId
+    };
+  }
+
+  /**
+   * Create a basic route context from socket information
+   */
+  public createSocketRouteContext(socket: plugins.net.Socket, options: {
+    domain?: string;
+    tlsVersion?: string;
+    connectionId: string;
+  }): IRouteContext {
+    return {
+      // Connection basics
+      port: socket.localPort || 0,
+      domain: options.domain,
+      clientIp: socket.remoteAddress?.replace('::ffff:', '') || '0.0.0.0',
+      serverIp: socket.localAddress?.replace('::ffff:', '') || '0.0.0.0',
+
+      // TLS information
+      isTls: options.tlsVersion !== undefined,
+      tlsVersion: options.tlsVersion,
+
+      // Metadata
+      timestamp: Date.now(),
+      connectionId: options.connectionId
+    };
+  }
+}

package/ts/proxies/network-proxy/function-cache.ts

@@ -0,0 +1,259 @@
+import type { IRouteContext } from '../../core/models/route-context.js';
+import type { ILogger } from './models/types.js';
+
+/**
+ * Interface for cached function result
+ */
+interface ICachedResult<T> {
+  value: T;
+  expiry: number;
+  hash: string;
+}
+
+/**
+ * Function cache for NetworkProxy function-based targets
+ * 
+ * This cache improves performance for function-based targets by storing
+ * the results of function evaluations and reusing them for similar contexts.
+ */
+export class FunctionCache {
+  // Cache storage
+  private hostCache: Map<string, ICachedResult<string | string[]>> = new Map();
+  private portCache: Map<string, ICachedResult<number>> = new Map();
+  
+  // Maximum number of entries to store in each cache
+  private maxCacheSize: number;
+  
+  // Default TTL for cache entries in milliseconds (default: 5 seconds)
+  private defaultTtl: number;
+  
+  // Logger
+  private logger: ILogger;
+  
+  /**
+   * Creates a new function cache
+   * 
+   * @param logger Logger for debug output
+   * @param options Cache options
+   */
+  constructor(
+    logger: ILogger,
+    options: {
+      maxCacheSize?: number;
+      defaultTtl?: number;
+    } = {}
+  ) {
+    this.logger = logger;
+    this.maxCacheSize = options.maxCacheSize || 1000;
+    this.defaultTtl = options.defaultTtl || 5000; // 5 seconds default
+    
+    // Start the cache cleanup timer
+    setInterval(() => this.cleanupCache(), 30000); // Cleanup every 30 seconds
+  }
+  
+  /**
+   * Compute a hash for a context object
+   * This is used to identify similar contexts for caching
+   * 
+   * @param context The route context to hash
+   * @param functionId Identifier for the function (usually route name or ID)
+   * @returns A string hash
+   */
+  private computeContextHash(context: IRouteContext, functionId: string): string {
+    // Extract relevant properties for the hash
+    const hashBase = {
+      functionId,
+      port: context.port,
+      domain: context.domain,
+      clientIp: context.clientIp,
+      path: context.path,
+      query: context.query,
+      isTls: context.isTls,
+      tlsVersion: context.tlsVersion
+    };
+    
+    // Generate a hash string
+    return JSON.stringify(hashBase);
+  }
+  
+  /**
+   * Get cached host result for a function and context
+   * 
+   * @param context Route context
+   * @param functionId Identifier for the function
+   * @returns Cached host value or undefined if not found
+   */
+  public getCachedHost(context: IRouteContext, functionId: string): string | string[] | undefined {
+    const hash = this.computeContextHash(context, functionId);
+    const cached = this.hostCache.get(hash);
+    
+    // Return if no cached value or expired
+    if (!cached || cached.expiry < Date.now()) {
+      if (cached) {
+        // If expired, remove from cache
+        this.hostCache.delete(hash);
+        this.logger.debug(`Cache miss (expired) for host function: ${functionId}`);
+      } else {
+        this.logger.debug(`Cache miss for host function: ${functionId}`);
+      }
+      return undefined;
+    }
+    
+    this.logger.debug(`Cache hit for host function: ${functionId}`);
+    return cached.value;
+  }
+  
+  /**
+   * Get cached port result for a function and context
+   * 
+   * @param context Route context
+   * @param functionId Identifier for the function
+   * @returns Cached port value or undefined if not found
+   */
+  public getCachedPort(context: IRouteContext, functionId: string): number | undefined {
+    const hash = this.computeContextHash(context, functionId);
+    const cached = this.portCache.get(hash);
+    
+    // Return if no cached value or expired
+    if (!cached || cached.expiry < Date.now()) {
+      if (cached) {
+        // If expired, remove from cache
+        this.portCache.delete(hash);
+        this.logger.debug(`Cache miss (expired) for port function: ${functionId}`);
+      } else {
+        this.logger.debug(`Cache miss for port function: ${functionId}`);
+      }
+      return undefined;
+    }
+    
+    this.logger.debug(`Cache hit for port function: ${functionId}`);
+    return cached.value;
+  }
+  
+  /**
+   * Store a host function result in the cache
+   * 
+   * @param context Route context
+   * @param functionId Identifier for the function
+   * @param value Host value to cache
+   * @param ttl Optional TTL in milliseconds
+   */
+  public cacheHost(
+    context: IRouteContext,
+    functionId: string,
+    value: string | string[],
+    ttl?: number
+  ): void {
+    const hash = this.computeContextHash(context, functionId);
+    const expiry = Date.now() + (ttl || this.defaultTtl);
+    
+    // Check if we need to prune the cache before adding
+    if (this.hostCache.size >= this.maxCacheSize) {
+      this.pruneOldestEntries(this.hostCache);
+    }
+    
+    // Store the result
+    this.hostCache.set(hash, { value, expiry, hash });
+    this.logger.debug(`Cached host function result for: ${functionId}`);
+  }
+  
+  /**
+   * Store a port function result in the cache
+   * 
+   * @param context Route context
+   * @param functionId Identifier for the function
+   * @param value Port value to cache
+   * @param ttl Optional TTL in milliseconds
+   */
+  public cachePort(
+    context: IRouteContext,
+    functionId: string,
+    value: number,
+    ttl?: number
+  ): void {
+    const hash = this.computeContextHash(context, functionId);
+    const expiry = Date.now() + (ttl || this.defaultTtl);
+    
+    // Check if we need to prune the cache before adding
+    if (this.portCache.size >= this.maxCacheSize) {
+      this.pruneOldestEntries(this.portCache);
+    }
+    
+    // Store the result
+    this.portCache.set(hash, { value, expiry, hash });
+    this.logger.debug(`Cached port function result for: ${functionId}`);
+  }
+  
+  /**
+   * Remove expired entries from the cache
+   */
+  private cleanupCache(): void {
+    const now = Date.now();
+    let expiredCount = 0;
+    
+    // Clean up host cache
+    for (const [hash, cached] of this.hostCache.entries()) {
+      if (cached.expiry < now) {
+        this.hostCache.delete(hash);
+        expiredCount++;
+      }
+    }
+    
+    // Clean up port cache
+    for (const [hash, cached] of this.portCache.entries()) {
+      if (cached.expiry < now) {
+        this.portCache.delete(hash);
+        expiredCount++;
+      }
+    }
+    
+    if (expiredCount > 0) {
+      this.logger.debug(`Cleaned up ${expiredCount} expired cache entries`);
+    }
+  }
+  
+  /**
+   * Prune oldest entries from a cache map
+   * Used when the cache exceeds the maximum size
+   * 
+   * @param cache The cache map to prune
+   */
+  private pruneOldestEntries<T>(cache: Map<string, ICachedResult<T>>): void {
+    // Find the oldest entries
+    const now = Date.now();
+    const itemsToRemove = Math.floor(this.maxCacheSize * 0.2); // Remove 20% of the cache
+    
+    // Convert to array for sorting
+    const entries = Array.from(cache.entries());
+    
+    // Sort by expiry (oldest first)
+    entries.sort((a, b) => a[1].expiry - b[1].expiry);
+    
+    // Remove oldest entries
+    const toRemove = entries.slice(0, itemsToRemove);
+    for (const [hash] of toRemove) {
+      cache.delete(hash);
+    }
+    
+    this.logger.debug(`Pruned ${toRemove.length} oldest cache entries`);
+  }
+  
+  /**
+   * Get current cache stats
+   */
+  public getStats(): { hostCacheSize: number; portCacheSize: number } {
+    return {
+      hostCacheSize: this.hostCache.size,
+      portCacheSize: this.portCache.size
+    };
+  }
+  
+  /**
+   * Clear all cached entries
+   */
+  public clearCache(): void {
+    this.hostCache.clear();
+    this.portCache.clear();
+    this.logger.info('Function cache cleared');
+  }
+}