|
|
|
@ -100,14 +100,82 @@ interface IConnectionRecord {
|
|
|
|
|
lastSleepDetection?: number; // Timestamp of the last sleep detection
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Structure to track TLS session information for proper resumption handling
|
|
|
|
|
*/
|
|
|
|
|
interface ITlsSessionInfo {
|
|
|
|
|
domain: string; // The SNI domain associated with this session
|
|
|
|
|
sessionId?: Buffer; // The TLS session ID (if available)
|
|
|
|
|
ticketId?: string; // Session ticket identifier for newer TLS versions
|
|
|
|
|
ticketTimestamp: number; // When this session was recorded
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Global cache of TLS session IDs to SNI domains
|
|
|
|
|
// This ensures resumed sessions maintain their SNI binding
|
|
|
|
|
const tlsSessionCache = new Map<string, ITlsSessionInfo>();
|
|
|
|
|
|
|
|
|
|
// Reference to session cleanup timer so we can clear it
|
|
|
|
|
let tlsSessionCleanupTimer: NodeJS.Timeout | null = null;
|
|
|
|
|
|
|
|
|
|
// Start the cleanup timer for session cache
|
|
|
|
|
function startSessionCleanupTimer() {
|
|
|
|
|
// Avoid creating multiple timers
|
|
|
|
|
if (tlsSessionCleanupTimer) {
|
|
|
|
|
clearInterval(tlsSessionCleanupTimer);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Create new cleanup timer
|
|
|
|
|
tlsSessionCleanupTimer = setInterval(() => {
|
|
|
|
|
const now = Date.now();
|
|
|
|
|
const expiryTime = 24 * 60 * 60 * 1000; // 24 hours
|
|
|
|
|
|
|
|
|
|
for (const [sessionId, info] of tlsSessionCache.entries()) {
|
|
|
|
|
if (now - info.ticketTimestamp > expiryTime) {
|
|
|
|
|
tlsSessionCache.delete(sessionId);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}, 60 * 60 * 1000); // Clean up once per hour
|
|
|
|
|
|
|
|
|
|
// Make sure the interval doesn't keep the process alive
|
|
|
|
|
if (tlsSessionCleanupTimer.unref) {
|
|
|
|
|
tlsSessionCleanupTimer.unref();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Start the timer initially
|
|
|
|
|
startSessionCleanupTimer();
|
|
|
|
|
|
|
|
|
|
// Function to stop the cleanup timer (used during shutdown)
|
|
|
|
|
function stopSessionCleanupTimer() {
|
|
|
|
|
if (tlsSessionCleanupTimer) {
|
|
|
|
|
clearInterval(tlsSessionCleanupTimer);
|
|
|
|
|
tlsSessionCleanupTimer = null;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Return type for the extractSNIInfo function
|
|
|
|
|
*/
|
|
|
|
|
interface ISNIExtractResult {
|
|
|
|
|
serverName?: string; // The extracted SNI hostname
|
|
|
|
|
sessionId?: Buffer; // The TLS session ID if present
|
|
|
|
|
sessionIdKey?: string; // The hex string representation of session ID
|
|
|
|
|
sessionTicketId?: string; // Session ticket identifier for TLS 1.3+ resumption
|
|
|
|
|
hasSessionTicket?: boolean; // Whether a session ticket extension was found
|
|
|
|
|
isResumption: boolean; // Whether this appears to be a session resumption
|
|
|
|
|
resumedDomain?: string; // The domain associated with the session if resuming
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Extracts the SNI (Server Name Indication) from a TLS ClientHello packet.
|
|
|
|
|
* Enhanced for robustness and detailed logging.
|
|
|
|
|
* Also extracts and tracks TLS Session IDs for session resumption handling.
|
|
|
|
|
*
|
|
|
|
|
* @param buffer - Buffer containing the TLS ClientHello.
|
|
|
|
|
* @param enableLogging - Whether to enable detailed logging.
|
|
|
|
|
* @returns The server name if found, otherwise undefined.
|
|
|
|
|
* @returns An object containing SNI and session information, or undefined if parsing fails.
|
|
|
|
|
*/
|
|
|
|
|
function extractSNI(buffer: Buffer, enableLogging: boolean = false): string | undefined {
|
|
|
|
|
function extractSNIInfo(buffer: Buffer, enableLogging: boolean = false): ISNIExtractResult | undefined {
|
|
|
|
|
try {
|
|
|
|
|
// Check if buffer is too small for TLS
|
|
|
|
|
if (buffer.length < 5) {
|
|
|
|
@ -153,9 +221,38 @@ function extractSNI(buffer: Buffer, enableLogging: boolean = false): string | un
|
|
|
|
|
|
|
|
|
|
offset += 2 + 32; // Skip client version and random
|
|
|
|
|
|
|
|
|
|
// Session ID
|
|
|
|
|
// Extract Session ID for session resumption tracking
|
|
|
|
|
const sessionIDLength = buffer.readUInt8(offset);
|
|
|
|
|
if (enableLogging) console.log(`Session ID Length: ${sessionIDLength}`);
|
|
|
|
|
|
|
|
|
|
// If there's a session ID, extract it
|
|
|
|
|
let sessionId: Buffer | undefined;
|
|
|
|
|
let sessionIdKey: string | undefined;
|
|
|
|
|
let isResumption = false;
|
|
|
|
|
let resumedDomain: string | undefined;
|
|
|
|
|
|
|
|
|
|
if (sessionIDLength > 0) {
|
|
|
|
|
sessionId = Buffer.from(buffer.slice(offset + 1, offset + 1 + sessionIDLength));
|
|
|
|
|
|
|
|
|
|
// Convert sessionId to a string key for our cache
|
|
|
|
|
sessionIdKey = sessionId.toString('hex');
|
|
|
|
|
|
|
|
|
|
if (enableLogging) {
|
|
|
|
|
console.log(`Session ID: ${sessionIdKey}`);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Check if this is a session resumption attempt
|
|
|
|
|
if (tlsSessionCache.has(sessionIdKey)) {
|
|
|
|
|
const cachedInfo = tlsSessionCache.get(sessionIdKey)!;
|
|
|
|
|
resumedDomain = cachedInfo.domain;
|
|
|
|
|
isResumption = true;
|
|
|
|
|
|
|
|
|
|
if (enableLogging) {
|
|
|
|
|
console.log(`TLS Session Resumption detected for domain: ${resumedDomain}`);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
offset += 1 + sessionIDLength; // Skip session ID
|
|
|
|
|
|
|
|
|
|
// Cipher suites
|
|
|
|
@ -194,6 +291,10 @@ function extractSNI(buffer: Buffer, enableLogging: boolean = false): string | un
|
|
|
|
|
return undefined;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Variables to track session tickets
|
|
|
|
|
let hasSessionTicket = false;
|
|
|
|
|
let sessionTicketId: string | undefined;
|
|
|
|
|
|
|
|
|
|
// Parse extensions
|
|
|
|
|
while (offset + 4 <= extensionsEnd) {
|
|
|
|
|
const extensionType = buffer.readUInt16BE(offset);
|
|
|
|
@ -204,6 +305,33 @@ function extractSNI(buffer: Buffer, enableLogging: boolean = false): string | un
|
|
|
|
|
|
|
|
|
|
offset += 4;
|
|
|
|
|
|
|
|
|
|
// Check for Session Ticket extension (type 0x0023)
|
|
|
|
|
if (extensionType === 0x0023 && extensionLength > 0) {
|
|
|
|
|
hasSessionTicket = true;
|
|
|
|
|
|
|
|
|
|
// Extract a hash of the ticket for tracking
|
|
|
|
|
if (extensionLength > 16) { // Ensure we have enough bytes to create a meaningful ID
|
|
|
|
|
const ticketBytes = buffer.slice(offset, offset + Math.min(16, extensionLength));
|
|
|
|
|
sessionTicketId = ticketBytes.toString('hex');
|
|
|
|
|
|
|
|
|
|
if (enableLogging) {
|
|
|
|
|
console.log(`Session Ticket found, ID: ${sessionTicketId}`);
|
|
|
|
|
|
|
|
|
|
// Check if this is a known session ticket
|
|
|
|
|
if (tlsSessionCache.has(`ticket:${sessionTicketId}`)) {
|
|
|
|
|
const cachedInfo = tlsSessionCache.get(`ticket:${sessionTicketId}`);
|
|
|
|
|
console.log(`TLS Session Ticket Resumption detected for domain: ${cachedInfo?.domain}`);
|
|
|
|
|
|
|
|
|
|
// Set isResumption and resumedDomain if not already set
|
|
|
|
|
if (!isResumption && !resumedDomain) {
|
|
|
|
|
isResumption = true;
|
|
|
|
|
resumedDomain = cachedInfo?.domain;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (extensionType === 0x0000) {
|
|
|
|
|
// SNI extension
|
|
|
|
|
if (offset + 2 > buffer.length) {
|
|
|
|
@ -245,7 +373,43 @@ function extractSNI(buffer: Buffer, enableLogging: boolean = false): string | un
|
|
|
|
|
|
|
|
|
|
const serverName = buffer.toString('utf8', offset, offset + nameLen);
|
|
|
|
|
if (enableLogging) console.log(`Extracted SNI: ${serverName}`);
|
|
|
|
|
return serverName;
|
|
|
|
|
|
|
|
|
|
// Store the session ID to domain mapping for future resumptions
|
|
|
|
|
if (sessionIdKey && sessionId && serverName) {
|
|
|
|
|
tlsSessionCache.set(sessionIdKey, {
|
|
|
|
|
domain: serverName,
|
|
|
|
|
sessionId: sessionId,
|
|
|
|
|
ticketTimestamp: Date.now()
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
if (enableLogging) {
|
|
|
|
|
console.log(`Stored session ${sessionIdKey} for domain ${serverName}`);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Also store session ticket information if present
|
|
|
|
|
if (sessionTicketId && serverName) {
|
|
|
|
|
tlsSessionCache.set(`ticket:${sessionTicketId}`, {
|
|
|
|
|
domain: serverName,
|
|
|
|
|
ticketId: sessionTicketId,
|
|
|
|
|
ticketTimestamp: Date.now()
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
if (enableLogging) {
|
|
|
|
|
console.log(`Stored session ticket ${sessionTicketId} for domain ${serverName}`);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Return the complete extraction result
|
|
|
|
|
return {
|
|
|
|
|
serverName,
|
|
|
|
|
sessionId,
|
|
|
|
|
sessionIdKey,
|
|
|
|
|
sessionTicketId,
|
|
|
|
|
isResumption,
|
|
|
|
|
resumedDomain,
|
|
|
|
|
hasSessionTicket
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
offset += nameLen;
|
|
|
|
@ -257,13 +421,46 @@ function extractSNI(buffer: Buffer, enableLogging: boolean = false): string | un
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (enableLogging) console.log('No SNI extension found');
|
|
|
|
|
return undefined;
|
|
|
|
|
|
|
|
|
|
// Even without SNI, we might be dealing with a session resumption
|
|
|
|
|
if (isResumption && resumedDomain) {
|
|
|
|
|
return {
|
|
|
|
|
serverName: resumedDomain, // Use the domain from previous session
|
|
|
|
|
sessionId,
|
|
|
|
|
sessionIdKey,
|
|
|
|
|
sessionTicketId,
|
|
|
|
|
hasSessionTicket,
|
|
|
|
|
isResumption: true,
|
|
|
|
|
resumedDomain
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Return a basic result with just the session info
|
|
|
|
|
return {
|
|
|
|
|
isResumption,
|
|
|
|
|
sessionId,
|
|
|
|
|
sessionIdKey,
|
|
|
|
|
sessionTicketId,
|
|
|
|
|
hasSessionTicket,
|
|
|
|
|
resumedDomain
|
|
|
|
|
};
|
|
|
|
|
} catch (err) {
|
|
|
|
|
console.log(`Error extracting SNI: ${err}`);
|
|
|
|
|
return undefined;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Legacy wrapper for extractSNIInfo to maintain backward compatibility
|
|
|
|
|
* @param buffer - Buffer containing the TLS ClientHello
|
|
|
|
|
* @param enableLogging - Whether to enable detailed logging
|
|
|
|
|
* @returns The server name if found, otherwise undefined
|
|
|
|
|
*/
|
|
|
|
|
function extractSNI(buffer: Buffer, enableLogging: boolean = false): string | undefined {
|
|
|
|
|
const result = extractSNIInfo(buffer, enableLogging);
|
|
|
|
|
return result?.serverName;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Helper: Check if a port falls within any of the given port ranges
|
|
|
|
|
const isPortInRanges = (port: number, ranges: Array<{ from: number; to: number }>): boolean => {
|
|
|
|
|
return ranges.some((range) => port >= range.from && port <= range.to);
|
|
|
|
@ -502,25 +699,16 @@ export class PortProxy {
|
|
|
|
|
this.cleanupConnection(record, 'client_closed');
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
// Update activity on data transfer
|
|
|
|
|
// Special handler for TLS handshake detection with NetworkProxy
|
|
|
|
|
socket.on('data', (chunk: Buffer) => {
|
|
|
|
|
this.updateActivity(record);
|
|
|
|
|
|
|
|
|
|
// Check for potential TLS renegotiation or reconnection packets
|
|
|
|
|
// Check for TLS handshake packets (ContentType.handshake)
|
|
|
|
|
if (chunk.length > 0 && chunk[0] === 22) {
|
|
|
|
|
// ContentType.handshake
|
|
|
|
|
if (this.settings.enableDetailedLogging) {
|
|
|
|
|
console.log(
|
|
|
|
|
`[${connectionId}] Detected potential TLS handshake data while connected to NetworkProxy`
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Let the NetworkProxy handle the TLS renegotiation
|
|
|
|
|
// Just update the activity timestamp to prevent timeouts
|
|
|
|
|
record.lastActivity = Date.now();
|
|
|
|
|
console.log(`[${connectionId}] Detected potential TLS handshake with NetworkProxy, updating activity`);
|
|
|
|
|
this.updateActivity(record);
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
// Update activity on data transfer from the proxy socket
|
|
|
|
|
proxySocket.on('data', () => this.updateActivity(record));
|
|
|
|
|
|
|
|
|
|
if (this.settings.enableDetailedLogging) {
|
|
|
|
@ -776,6 +964,82 @@ export class PortProxy {
|
|
|
|
|
return this.initiateCleanupOnce(record, 'write_error');
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Set up the renegotiation listener *before* piping if this is a TLS connection with SNI
|
|
|
|
|
if (serverName && record.isTLS) {
|
|
|
|
|
// This listener handles TLS renegotiation detection
|
|
|
|
|
socket.on('data', (renegChunk) => {
|
|
|
|
|
if (renegChunk.length > 0 && renegChunk.readUInt8(0) === 22) {
|
|
|
|
|
// Always update activity timestamp for any handshake packet
|
|
|
|
|
this.updateActivity(record);
|
|
|
|
|
|
|
|
|
|
try {
|
|
|
|
|
// Extract all TLS information including session resumption data
|
|
|
|
|
const sniInfo = extractSNIInfo(renegChunk, this.settings.enableTlsDebugLogging);
|
|
|
|
|
let newSNI = sniInfo?.serverName;
|
|
|
|
|
|
|
|
|
|
// Handle session resumption - if we recognize the session ID, we know what domain it belongs to
|
|
|
|
|
if (sniInfo?.isResumption && sniInfo.resumedDomain) {
|
|
|
|
|
console.log(`[${connectionId}] Rehandshake with session resumption for domain: ${sniInfo.resumedDomain}`);
|
|
|
|
|
newSNI = sniInfo.resumedDomain;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// IMPORTANT: If we can't extract an SNI from renegotiation, we MUST allow it through
|
|
|
|
|
if (newSNI === undefined) {
|
|
|
|
|
console.log(`[${connectionId}] Rehandshake detected without SNI, allowing it through.`);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Check if the SNI has changed
|
|
|
|
|
if (newSNI !== serverName) {
|
|
|
|
|
console.log(`[${connectionId}] Rehandshake with different SNI: ${newSNI} vs original ${serverName}`);
|
|
|
|
|
|
|
|
|
|
// Allow if the new SNI matches existing domain config or find a new matching config
|
|
|
|
|
let allowed = false;
|
|
|
|
|
|
|
|
|
|
if (record.domainConfig) {
|
|
|
|
|
allowed = record.domainConfig.domains.some(d => plugins.minimatch(newSNI, d));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (!allowed) {
|
|
|
|
|
const newDomainConfig = this.settings.domainConfigs.find((config) =>
|
|
|
|
|
config.domains.some((d) => plugins.minimatch(newSNI, d))
|
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
if (newDomainConfig) {
|
|
|
|
|
const effectiveAllowedIPs = [
|
|
|
|
|
...newDomainConfig.allowedIPs,
|
|
|
|
|
...(this.settings.defaultAllowedIPs || []),
|
|
|
|
|
];
|
|
|
|
|
const effectiveBlockedIPs = [
|
|
|
|
|
...(newDomainConfig.blockedIPs || []),
|
|
|
|
|
...(this.settings.defaultBlockedIPs || []),
|
|
|
|
|
];
|
|
|
|
|
|
|
|
|
|
allowed = isGlobIPAllowed(record.remoteIP, effectiveAllowedIPs, effectiveBlockedIPs);
|
|
|
|
|
|
|
|
|
|
if (allowed) {
|
|
|
|
|
record.domainConfig = newDomainConfig;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (allowed) {
|
|
|
|
|
console.log(`[${connectionId}] Updated domain for connection from ${record.remoteIP} to: ${newSNI}`);
|
|
|
|
|
record.lockedDomain = newSNI;
|
|
|
|
|
} else {
|
|
|
|
|
console.log(`[${connectionId}] Rehandshake SNI ${newSNI} not allowed. Terminating connection.`);
|
|
|
|
|
this.initiateCleanupOnce(record, 'sni_mismatch');
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
console.log(`[${connectionId}] Rehandshake with same SNI: ${newSNI}`);
|
|
|
|
|
}
|
|
|
|
|
} catch (err) {
|
|
|
|
|
console.log(`[${connectionId}] Error processing renegotiation: ${err}. Allowing to continue.`);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Now set up piping for future data and resume the socket
|
|
|
|
|
socket.pipe(targetSocket);
|
|
|
|
|
targetSocket.pipe(socket);
|
|
|
|
@ -809,7 +1073,83 @@ export class PortProxy {
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
} else {
|
|
|
|
|
// No pending data, so just set up piping
|
|
|
|
|
// Set up the renegotiation listener *before* piping if this is a TLS connection with SNI
|
|
|
|
|
if (serverName && record.isTLS) {
|
|
|
|
|
// This listener handles TLS renegotiation detection
|
|
|
|
|
socket.on('data', (renegChunk) => {
|
|
|
|
|
if (renegChunk.length > 0 && renegChunk.readUInt8(0) === 22) {
|
|
|
|
|
// Always update activity timestamp for any handshake packet
|
|
|
|
|
this.updateActivity(record);
|
|
|
|
|
|
|
|
|
|
try {
|
|
|
|
|
// Extract all TLS information including session resumption data
|
|
|
|
|
const sniInfo = extractSNIInfo(renegChunk, this.settings.enableTlsDebugLogging);
|
|
|
|
|
let newSNI = sniInfo?.serverName;
|
|
|
|
|
|
|
|
|
|
// Handle session resumption - if we recognize the session ID, we know what domain it belongs to
|
|
|
|
|
if (sniInfo?.isResumption && sniInfo.resumedDomain) {
|
|
|
|
|
console.log(`[${connectionId}] Rehandshake with session resumption for domain: ${sniInfo.resumedDomain}`);
|
|
|
|
|
newSNI = sniInfo.resumedDomain;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// IMPORTANT: If we can't extract an SNI from renegotiation, we MUST allow it through
|
|
|
|
|
if (newSNI === undefined) {
|
|
|
|
|
console.log(`[${connectionId}] Rehandshake detected without SNI, allowing it through.`);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Check if the SNI has changed
|
|
|
|
|
if (newSNI !== serverName) {
|
|
|
|
|
console.log(`[${connectionId}] Rehandshake with different SNI: ${newSNI} vs original ${serverName}`);
|
|
|
|
|
|
|
|
|
|
// Allow if the new SNI matches existing domain config or find a new matching config
|
|
|
|
|
let allowed = false;
|
|
|
|
|
|
|
|
|
|
if (record.domainConfig) {
|
|
|
|
|
allowed = record.domainConfig.domains.some(d => plugins.minimatch(newSNI, d));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (!allowed) {
|
|
|
|
|
const newDomainConfig = this.settings.domainConfigs.find((config) =>
|
|
|
|
|
config.domains.some((d) => plugins.minimatch(newSNI, d))
|
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
if (newDomainConfig) {
|
|
|
|
|
const effectiveAllowedIPs = [
|
|
|
|
|
...newDomainConfig.allowedIPs,
|
|
|
|
|
...(this.settings.defaultAllowedIPs || []),
|
|
|
|
|
];
|
|
|
|
|
const effectiveBlockedIPs = [
|
|
|
|
|
...(newDomainConfig.blockedIPs || []),
|
|
|
|
|
...(this.settings.defaultBlockedIPs || []),
|
|
|
|
|
];
|
|
|
|
|
|
|
|
|
|
allowed = isGlobIPAllowed(record.remoteIP, effectiveAllowedIPs, effectiveBlockedIPs);
|
|
|
|
|
|
|
|
|
|
if (allowed) {
|
|
|
|
|
record.domainConfig = newDomainConfig;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (allowed) {
|
|
|
|
|
console.log(`[${connectionId}] Updated domain for connection from ${record.remoteIP} to: ${newSNI}`);
|
|
|
|
|
record.lockedDomain = newSNI;
|
|
|
|
|
} else {
|
|
|
|
|
console.log(`[${connectionId}] Rehandshake SNI ${newSNI} not allowed. Terminating connection.`);
|
|
|
|
|
this.initiateCleanupOnce(record, 'sni_mismatch');
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
console.log(`[${connectionId}] Rehandshake with same SNI: ${newSNI}`);
|
|
|
|
|
}
|
|
|
|
|
} catch (err) {
|
|
|
|
|
console.log(`[${connectionId}] Error processing renegotiation: ${err}. Allowing to continue.`);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Now set up piping
|
|
|
|
|
socket.pipe(targetSocket);
|
|
|
|
|
targetSocket.pipe(socket);
|
|
|
|
|
socket.resume(); // Resume the socket after piping is established
|
|
|
|
@ -846,31 +1186,8 @@ export class PortProxy {
|
|
|
|
|
record.pendingData = [];
|
|
|
|
|
record.pendingDataSize = 0;
|
|
|
|
|
|
|
|
|
|
// Add the renegotiation listener for SNI validation
|
|
|
|
|
if (serverName) {
|
|
|
|
|
socket.on('data', (renegChunk: Buffer) => {
|
|
|
|
|
if (renegChunk.length > 0 && renegChunk.readUInt8(0) === 22) {
|
|
|
|
|
try {
|
|
|
|
|
// Try to extract SNI from potential renegotiation
|
|
|
|
|
const newSNI = extractSNI(renegChunk, this.settings.enableTlsDebugLogging);
|
|
|
|
|
if (newSNI && newSNI !== record.lockedDomain) {
|
|
|
|
|
console.log(
|
|
|
|
|
`[${connectionId}] Rehandshake detected with different SNI: ${newSNI} vs locked ${record.lockedDomain}. Terminating connection.`
|
|
|
|
|
);
|
|
|
|
|
this.initiateCleanupOnce(record, 'sni_mismatch');
|
|
|
|
|
} else if (newSNI && this.settings.enableDetailedLogging) {
|
|
|
|
|
console.log(
|
|
|
|
|
`[${connectionId}] Rehandshake detected with same SNI: ${newSNI}. Allowing.`
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
} catch (err) {
|
|
|
|
|
console.log(
|
|
|
|
|
`[${connectionId}] Error processing potential renegotiation: ${err}. Allowing connection to continue.`
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
// Renegotiation detection is now handled before piping is established
|
|
|
|
|
// This ensures the data listener receives all packets properly
|
|
|
|
|
|
|
|
|
|
// Set connection timeout with simpler logic
|
|
|
|
|
if (record.cleanupTimer) {
|
|
|
|
@ -886,13 +1203,12 @@ export class PortProxy {
|
|
|
|
|
}
|
|
|
|
|
// No cleanup timer for immortal connections
|
|
|
|
|
}
|
|
|
|
|
// For TLS keep-alive connections, use an aggressive timeout to ensure
|
|
|
|
|
// certificates are regularly refreshed even in chained proxy scenarios
|
|
|
|
|
// For TLS keep-alive connections, use a more generous timeout now that
|
|
|
|
|
// we've fixed the renegotiation handling issue that was causing certificate problems
|
|
|
|
|
else if (record.hasKeepAlive && record.isTLS) {
|
|
|
|
|
// Use a much shorter timeout for TLS connections to ensure certificate contexts are refreshed frequently
|
|
|
|
|
// This prevents issues with stale certificates in browser tabs that have been idle for a while
|
|
|
|
|
// 30 minutes is aggressive enough to handle multi-proxy chains without causing too many reconnects
|
|
|
|
|
const tlsKeepAliveTimeout = 30 * 60 * 1000; // 30 minutes for TLS keep-alive - dramatically reduced from 8 hours
|
|
|
|
|
// Use a longer timeout for TLS connections now that renegotiation handling is fixed
|
|
|
|
|
// This reduces unnecessary reconnections while still ensuring certificate freshness
|
|
|
|
|
const tlsKeepAliveTimeout = 4 * 60 * 60 * 1000; // 4 hours for TLS keep-alive - increased from 30 minutes
|
|
|
|
|
const safeTimeout = ensureSafeTimeout(tlsKeepAliveTimeout);
|
|
|
|
|
|
|
|
|
|
record.cleanupTimer = setTimeout(() => {
|
|
|
|
@ -1063,34 +1379,34 @@ export class PortProxy {
|
|
|
|
|
// For TLS keep-alive connections after sleep/long inactivity, force close
|
|
|
|
|
// to make browser establish a new connection with fresh certificate context
|
|
|
|
|
if (record.isTLS && record.tlsHandshakeComplete) {
|
|
|
|
|
// Much more aggressive timeout (20 minutes) to ensure reliable operation in chained proxy scenarios
|
|
|
|
|
if (timeDiff > 20 * 60 * 1000) {
|
|
|
|
|
// If inactive for more than 20 minutes (reduced from 4 hours)
|
|
|
|
|
// More generous timeout now that we've fixed the renegotiation handling
|
|
|
|
|
if (timeDiff > 2 * 60 * 60 * 1000) {
|
|
|
|
|
// If inactive for more than 2 hours (increased from 20 minutes)
|
|
|
|
|
console.log(
|
|
|
|
|
`[${record.id}] TLS connection inactive for ${plugins.prettyMs(timeDiff)}. ` +
|
|
|
|
|
`Closing to force new connection with fresh certificate.`
|
|
|
|
|
);
|
|
|
|
|
return this.initiateCleanupOnce(record, 'certificate_refresh_needed');
|
|
|
|
|
} else if (timeDiff > 10 * 60 * 1000) {
|
|
|
|
|
// For shorter but still significant inactivity (10+ minutes), be more aggressive with refresh
|
|
|
|
|
} else if (timeDiff > 30 * 60 * 1000) {
|
|
|
|
|
// For shorter but still significant inactivity (30+ minutes), refresh TLS state
|
|
|
|
|
console.log(
|
|
|
|
|
`[${record.id}] TLS connection inactive for ${plugins.prettyMs(timeDiff)}. ` +
|
|
|
|
|
`Aggressively refreshing TLS state to prevent certificate issues in proxy chains.`
|
|
|
|
|
`Refreshing TLS state.`
|
|
|
|
|
);
|
|
|
|
|
this.refreshTlsStateAfterSleep(record);
|
|
|
|
|
|
|
|
|
|
// Add an additional check in 5 minutes if no activity
|
|
|
|
|
// Add an additional check in 15 minutes if no activity
|
|
|
|
|
const refreshCheckId = record.id;
|
|
|
|
|
const refreshCheck = setTimeout(() => {
|
|
|
|
|
const currentRecord = this.connectionRecords.get(refreshCheckId);
|
|
|
|
|
if (currentRecord && Date.now() - currentRecord.lastActivity > 5 * 60 * 1000) {
|
|
|
|
|
if (currentRecord && Date.now() - currentRecord.lastActivity > 15 * 60 * 1000) {
|
|
|
|
|
console.log(
|
|
|
|
|
`[${refreshCheckId}] No activity detected after TLS refresh. ` +
|
|
|
|
|
`Closing connection to ensure certificate freshness.`
|
|
|
|
|
);
|
|
|
|
|
this.initiateCleanupOnce(currentRecord, 'tls_refresh_verification_failed');
|
|
|
|
|
}
|
|
|
|
|
}, 5 * 60 * 1000);
|
|
|
|
|
}, 15 * 60 * 1000);
|
|
|
|
|
|
|
|
|
|
// Make sure timeout doesn't keep the process alive
|
|
|
|
|
if (refreshCheck.unref) {
|
|
|
|
@ -1133,9 +1449,9 @@ export class PortProxy {
|
|
|
|
|
const connectionAge = Date.now() - record.incomingStartTime;
|
|
|
|
|
const hourInMs = 60 * 60 * 1000;
|
|
|
|
|
|
|
|
|
|
// For TLS browser connections, use a much more aggressive timeout
|
|
|
|
|
// to avoid certificate issues, especially in chained proxy scenarios
|
|
|
|
|
if (record.isTLS && record.hasKeepAlive && connectionAge > 45 * 60 * 1000) { // 45 minutes instead of 12 hours
|
|
|
|
|
// For TLS browser connections, use a more generous timeout now that
|
|
|
|
|
// we've fixed the renegotiation handling issues
|
|
|
|
|
if (record.isTLS && record.hasKeepAlive && connectionAge > 8 * hourInMs) { // 8 hours instead of 45 minutes
|
|
|
|
|
console.log(
|
|
|
|
|
`[${record.id}] Long-lived TLS connection (${plugins.prettyMs(connectionAge)}). ` +
|
|
|
|
|
`Closing to ensure proper certificate handling on browser reconnect in proxy chain.`
|
|
|
|
@ -1602,6 +1918,12 @@ export class PortProxy {
|
|
|
|
|
// Save domain config in connection record
|
|
|
|
|
connectionRecord.domainConfig = domainConfig;
|
|
|
|
|
|
|
|
|
|
// Always set the lockedDomain, even for non-SNI connections
|
|
|
|
|
if (serverName) {
|
|
|
|
|
connectionRecord.lockedDomain = serverName;
|
|
|
|
|
console.log(`[${connectionId}] Locked connection to domain: ${serverName}`);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// IP validation is skipped if allowedIPs is empty
|
|
|
|
|
if (domainConfig) {
|
|
|
|
|
const effectiveAllowedIPs: string[] = [
|
|
|
|
@ -1769,7 +2091,17 @@ export class PortProxy {
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
serverName = extractSNI(chunk, this.settings.enableTlsDebugLogging) || '';
|
|
|
|
|
// Extract all TLS information including session resumption
|
|
|
|
|
const sniInfo = extractSNIInfo(chunk, this.settings.enableTlsDebugLogging);
|
|
|
|
|
|
|
|
|
|
if (sniInfo?.isResumption && sniInfo.resumedDomain) {
|
|
|
|
|
// This is a session resumption with a known domain
|
|
|
|
|
serverName = sniInfo.resumedDomain;
|
|
|
|
|
console.log(`[${connectionId}] TLS Session resumption detected for domain: ${serverName}`);
|
|
|
|
|
} else {
|
|
|
|
|
// Normal SNI extraction
|
|
|
|
|
serverName = sniInfo?.serverName || '';
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Lock the connection to the negotiated SNI.
|
|
|
|
@ -2086,6 +2418,9 @@ export class PortProxy {
|
|
|
|
|
console.log('PortProxy shutting down...');
|
|
|
|
|
this.isShuttingDown = true;
|
|
|
|
|
|
|
|
|
|
// Stop the session cleanup timer
|
|
|
|
|
stopSessionCleanupTimer();
|
|
|
|
|
|
|
|
|
|
// Stop accepting new connections
|
|
|
|
|
const closeServerPromises: Promise<void>[] = this.netServers.map(
|
|
|
|
|
(server) =>
|
|
|
|
|