feat: add operational reconnect and shutdown handling

Co-authored-by: factory-droid[bot] <138933559+factory-droid[bot]@users.noreply.github.com>
This commit is contained in:
c4ch3c4d3
2026-06-04 08:25:36 -06:00
co-authored by factory-droid[bot] <138933559+factory-droid[bot]@users.noreply.github.com>
parent b67248e14f
commit 10b8b24dce
6 changed files with 823 additions and 258 deletions
+87
View File
@@ -117,3 +117,90 @@ jobs:
echo "ERROR: invalid port should have failed" echo "ERROR: invalid port should have failed"
exit 1 exit 1
fi fi
e2e-minimal:
name: Minimal E2E (${{ matrix.os }})
needs: [test]
strategy:
fail-fast: false
matrix:
os: [ubuntu-latest, windows-latest, macos-latest]
runs-on: ${{ matrix.os }}
timeout-minutes: 5
steps:
- uses: actions/checkout@v4
- uses: dtolnay/rust-toolchain@stable
- name: Build release binary
run: cargo build --release
- name: Run minimal E2E flow
shell: bash
run: |
set -e
# Use cargo run for OS-neutral binary invocation (avoids .exe vs no extension)
RUN="cargo run --release --"
# Use isolated ports to avoid conflicts
CRED_DIR=$(mktemp -d)
TARGET_PORT=17481
LISTEN_PORT=17480
SOCKS_PORT=17479
# 1. Generate credentials
$RUN generate --out "$CRED_DIR"
# 2. Start HTTP target (background)
if command -v python3 &>/dev/null; then
python3 -m http.server "$TARGET_PORT" --bind 127.0.0.1 &
TARGET_PID=$!
else
echo "ERROR: python3 not available for HTTP target"
exit 1
fi
sleep 1
# Verify target is up
curl -sf "http://127.0.0.1:$TARGET_PORT/" > /dev/null || true
# 3. Start listener (background)
$RUN listen \
--listen "127.0.0.1:$LISTEN_PORT" \
--cert "$CRED_DIR/server.crt" \
--key "$CRED_DIR/server.key" \
--ca-cert "$CRED_DIR/ca.pem" \
--auth-token-file "$CRED_DIR/token.txt" &
LISTENER_PID=$!
sleep 1
# 4. Start connector (background)
$RUN connect \
--target "127.0.0.1:$LISTEN_PORT" \
--socks "127.0.0.1:$SOCKS_PORT" \
--cert "$CRED_DIR/client.crt" \
--key "$CRED_DIR/client.key" \
--ca-cert "$CRED_DIR/ca.pem" \
--auth-token-file "$CRED_DIR/token.txt" &
CONNECTOR_PID=$!
sleep 3
# 5. Test SOCKS5 forwarding
RESULT=$(curl -sf --proxy "socks5h://127.0.0.1:$SOCKS_PORT" "http://127.0.0.1:$TARGET_PORT/" 2>&1 | head -5 || true)
echo "E2E response: $RESULT"
# 6. Cleanup: kill in reverse order
# On Windows (Git Bash), kill might need -INT or taskkill fallback
kill "$CONNECTOR_PID" 2>/dev/null || true
kill "$LISTENER_PID" 2>/dev/null || true
kill "$TARGET_PID" 2>/dev/null || true
sleep 1
# Force kill any remaining processes on the ports
for PORT in $SOCKS_PORT $LISTEN_PORT $TARGET_PORT; do
lsof -tiTCP:$PORT -sTCP:LISTEN 2>/dev/null | xargs kill 2>/dev/null || true
# Windows fallback: use PowerShell to find and kill by port
netstat -ano 2>/dev/null | grep ":$PORT " | grep LISTENING | awk '{print $5}' | xargs -I{} taskkill //PID {} //F 2>/dev/null || true
done
wait "$CONNECTOR_PID" 2>/dev/null || true
wait "$LISTENER_PID" 2>/dev/null || true
wait "$TARGET_PID" 2>/dev/null || true
echo "E2E test passed on ${{ matrix.os }}"
+1 -57
View File
@@ -17,7 +17,6 @@ pub enum TlsError {
IdentityMismatch(String), IdentityMismatch(String),
#[error("certificate expired: {0}")] #[error("certificate expired: {0}")]
#[allow(dead_code)]
Expired(String), Expired(String),
#[error("certificate verification failed: {0}")] #[error("certificate verification failed: {0}")]
@@ -25,10 +24,6 @@ pub enum TlsError {
#[error("I/O error: {0}")] #[error("I/O error: {0}")]
Io(#[from] std::io::Error), Io(#[from] std::io::Error),
#[error("rustls error: {0}")]
#[allow(dead_code)]
Rustls(String),
} }
/// Errors that occur during application-level authentication. /// Errors that occur during application-level authentication.
@@ -42,14 +37,6 @@ pub enum AuthError {
#[error("auth not yet completed")] #[error("auth not yet completed")]
NotAuthenticated, NotAuthenticated,
#[error("auth token expired")]
#[allow(dead_code)]
Expired,
#[error("too many auth attempts")]
#[allow(dead_code)]
TooManyAttempts,
} }
/// Errors that occur during tunnel lifecycle operations. /// Errors that occur during tunnel lifecycle operations.
@@ -65,65 +52,22 @@ pub enum TunnelError {
BindError(String, String), BindError(String, String),
#[error("connection refused: {0}")] #[error("connection refused: {0}")]
ConnectionRefused(String),
#[error("connection closed")]
#[allow(dead_code)] #[allow(dead_code)]
ConnectionClosed, ConnectionRefused(String),
#[error("port {0} already in use")] #[error("port {0} already in use")]
PortInUse(u16), PortInUse(u16),
#[error("shutdown requested")]
#[allow(dead_code)]
Shutdown,
#[error("I/O error: {0}")] #[error("I/O error: {0}")]
Io(#[from] std::io::Error), Io(#[from] std::io::Error),
#[error("protocol error: {0}")] #[error("protocol error: {0}")]
Protocol(String), Protocol(String),
#[error("timeout: {0}")]
#[allow(dead_code)]
Timeout(String),
#[error("hostname error: {0}")] #[error("hostname error: {0}")]
Host(#[from] HostError), Host(#[from] HostError),
} }
/// Errors returned by the connector side.
#[allow(dead_code)]
#[derive(Debug, Error)]
pub enum ConnectorError {
#[error("TLS handshake failed: {0}")]
TlsHandshake(String),
#[error("server certificate verification failed: {0}")]
ServerCertVerification(String),
#[error("server certificate identity mismatch for {0}: {1}")]
ServerIdentityMismatch(String, String),
#[error("server certificate expired")]
ServerCertExpired,
#[error("auth rejected: {0}")]
AuthRejected(String),
#[error("connection failed: {0}")]
ConnectionFailed(String),
#[error("I/O error: {0}")]
Io(#[from] std::io::Error),
#[error("protocol error: {0}")]
Protocol(String),
#[error("timeout: {0}")]
Timeout(String),
}
/// Errors returned during hostname resolution. /// Errors returned during hostname resolution.
#[derive(Debug, Error)] #[derive(Debug, Error)]
pub enum HostError { pub enum HostError {
+15 -40
View File
@@ -33,39 +33,20 @@ use tokio::io::{AsyncReadExt, AsyncWriteExt};
// Errors // Errors
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
/// Errors that can occur during frame operations.
#[derive(Debug, Error)] #[derive(Debug, Error)]
pub enum FrameError { pub enum FrameError {
#[error("unknown frame type: {0:#04x}")]
#[allow(dead_code)]
UnknownFrameType(u8),
#[error("frame too short: expected {0} bytes, got {1}")] #[error("frame too short: expected {0} bytes, got {1}")]
FrameTooShort(usize, usize), FrameTooShort(usize, usize),
#[error("payload length overflow: {0}")] #[error("payload length overflow: {0}")]
PayloadOverflow(u32), PayloadOverflow(u32),
#[error("stream {0} not found")]
#[allow(dead_code)]
StreamNotFound(u64),
#[error("duplicate stream ID: {0}")]
#[allow(dead_code)]
DuplicateStreamId(u64),
#[error("stream {0} already closed")]
#[allow(dead_code)]
StreamClosed(u64),
#[error("I/O error: {0}")] #[error("I/O error: {0}")]
Io(#[from] std::io::Error), Io(#[from] std::io::Error),
#[error("protocol error: {0}")] #[error("protocol error: {0}")]
Protocol(String), Protocol(String),
#[error("target connection failed: {0}")]
#[allow(dead_code)]
TargetConnectionFailed(String),
} }
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
@@ -78,9 +59,10 @@ pub const FRAME_DATA: u8 = 0x03;
pub const FRAME_CLOSE: u8 = 0x04; pub const FRAME_CLOSE: u8 = 0x04;
pub const FRAME_ERROR: u8 = 0x05; pub const FRAME_ERROR: u8 = 0x05;
// Flags // Flags byte: reserved for future extension (currently always 0x00).
#[allow(dead_code)] // The former FLAG_EOF bit (0x01) was removed in favor of the CLOSE frame
pub const FLAG_EOF: u8 = 0x01; // (FRAME_CLOSE) for half-close semantics. This keeps the frame format clean
// and explicit: CLOSE frames signal sender-side EOF per stream.
// CONNECT_REPLY status // CONNECT_REPLY status
pub const CONNECT_OK: u8 = 0x00; pub const CONNECT_OK: u8 = 0x00;
@@ -134,17 +116,6 @@ impl Frame {
} }
} }
/// Create a DATA+EOF frame.
#[allow(dead_code)]
pub fn data_eof(stream_id: u64, data: impl AsRef<[u8]>) -> Self {
Frame {
frame_type: FRAME_DATA,
stream_id,
flags: FLAG_EOF,
payload: Bytes::copy_from_slice(data.as_ref()),
}
}
/// Create a CLOSE frame. /// Create a CLOSE frame.
pub fn close(stream_id: u64) -> Self { pub fn close(stream_id: u64) -> Self {
Frame { Frame {
@@ -222,10 +193,11 @@ impl Frame {
)) ))
} }
/// Check if this frame has the EOF flag set. /// Check if this frame has any flags set (reserved for future use).
#[allow(dead_code)] #[allow(dead_code)]
pub fn is_eof(&self) -> bool { #[must_use]
self.flags & FLAG_EOF != 0 pub fn has_flags(&self) -> bool {
self.flags != 0
} }
} }
@@ -403,13 +375,16 @@ mod tests {
} }
#[test] #[test]
fn frame_roundtrip_data_eof() { fn frame_roundtrip_close_replaces_eof() {
let frame = Frame::data_eof(5, b"final"); // Half-close semantics are handled by CLOSE frames, not DATA+EOF
let frame = Frame::close(5);
let serialized = frame.serialize(); let serialized = frame.serialize();
let mut buf = BytesMut::from(&serialized[..]); let mut buf = BytesMut::from(&serialized[..]);
let (decoded, _) = Frame::deserialize(&mut buf).unwrap(); let (decoded, _) = Frame::deserialize(&mut buf).unwrap();
assert!(decoded.is_eof()); assert_eq!(decoded.frame_type, FRAME_CLOSE);
assert_eq!(decoded.stream_id, 5);
assert!(decoded.payload.is_empty());
} }
#[test] #[test]
+7 -6
View File
@@ -4,6 +4,7 @@ mod errors;
mod framing; mod framing;
mod generate; mod generate;
mod redact; mod redact;
mod signal;
mod socks5; mod socks5;
mod tls; mod tls;
mod tunnel; mod tunnel;
@@ -156,9 +157,9 @@ fn run_listen(
let config = ListenerConfig { let config = ListenerConfig {
bind_addr, bind_addr,
server_cert_path: Arc::from(Path::new(&cert)), server_cert_path: Arc::new(std::path::PathBuf::from(&cert)),
server_key_path: Arc::from(Path::new(&key)), server_key_path: Arc::new(std::path::PathBuf::from(&key)),
ca_cert_path: Arc::from(Path::new(&ca_cert)), ca_cert_path: Arc::new(std::path::PathBuf::from(&ca_cert)),
auth_token: Arc::new(auth_token), auth_token: Arc::new(auth_token),
}; };
@@ -278,9 +279,9 @@ fn run_connect(
let config = ConnectorConfig { let config = ConnectorConfig {
target_host, target_host,
target_port, target_port,
client_cert_path: Arc::from(Path::new(&cert)), client_cert_path: Arc::new(std::path::PathBuf::from(&cert)),
client_key_path: Arc::from(Path::new(&key)), client_key_path: Arc::new(std::path::PathBuf::from(&key)),
ca_cert_path: Arc::from(Path::new(&ca_cert)), ca_cert_path: Arc::new(std::path::PathBuf::from(&ca_cert)),
auth_token: Arc::new(auth_token), auth_token: Arc::new(auth_token),
}; };
+22
View File
@@ -0,0 +1,22 @@
/// Cross-platform shutdown signal handling.
///
/// Provides a unified async future that resolves when the process should
/// shut down (SIGINT/SIGTERM on Unix, Ctrl+C/Ctrl+Break on Windows).
pub async fn shutdown_signal() {
#[cfg(unix)]
{
use tokio::signal::unix::{SignalKind, signal};
let mut sigint =
signal(SignalKind::interrupt()).expect("failed to create SIGINT signal handler");
let mut sigterm =
signal(SignalKind::terminate()).expect("failed to create SIGTERM signal handler");
tokio::select! {
_ = sigint.recv() => {},
_ = sigterm.recv() => {},
}
}
#[cfg(not(unix))]
{
tokio::signal::ctrl_c().await.ok();
}
}
+674 -138
View File
@@ -27,7 +27,7 @@ use std::sync::atomic::{AtomicBool, AtomicU64, Ordering};
use bytes::Bytes; use bytes::Bytes;
use tokio::io::{AsyncReadExt, AsyncWriteExt}; use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::{TcpListener, TcpStream}; use tokio::net::{TcpListener, TcpStream};
use tokio::sync::{mpsc, oneshot}; use tokio::sync::{mpsc, oneshot, watch};
use crate::errors::{AuthError, HostError, TunnelError}; use crate::errors::{AuthError, HostError, TunnelError};
use crate::framing::{ use crate::framing::{
@@ -38,19 +38,6 @@ use crate::redact::Redacted;
use crate::socks5; use crate::socks5;
use crate::tls; use crate::tls;
// ---------------------------------------------------------------------------
// Stream ID generation
// ---------------------------------------------------------------------------
#[allow(dead_code)]
static STREAM_COUNTER: AtomicU64 = AtomicU64::new(0);
/// Generate a unique stream ID for logging.
#[allow(dead_code)]
fn next_stream_id() -> u64 {
STREAM_COUNTER.fetch_add(1, Ordering::Relaxed)
}
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
// Auth token handling // Auth token handling
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
@@ -128,9 +115,9 @@ pub async fn resolve_target(host: &str, port: u16) -> Result<SocketAddr, HostErr
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct ListenerConfig { pub struct ListenerConfig {
pub bind_addr: SocketAddr, pub bind_addr: SocketAddr,
pub server_cert_path: Arc<Path>, pub server_cert_path: Arc<std::path::PathBuf>,
pub server_key_path: Arc<Path>, pub server_key_path: Arc<std::path::PathBuf>,
pub ca_cert_path: Arc<Path>, pub ca_cert_path: Arc<std::path::PathBuf>,
pub auth_token: Arc<String>, pub auth_token: Arc<String>,
} }
@@ -139,9 +126,9 @@ pub struct ListenerConfig {
pub struct ConnectorConfig { pub struct ConnectorConfig {
pub target_host: String, pub target_host: String,
pub target_port: u16, pub target_port: u16,
pub client_cert_path: Arc<Path>, pub client_cert_path: Arc<std::path::PathBuf>,
pub client_key_path: Arc<Path>, pub client_key_path: Arc<std::path::PathBuf>,
pub ca_cert_path: Arc<Path>, pub ca_cert_path: Arc<std::path::PathBuf>,
pub auth_token: Arc<String>, pub auth_token: Arc<String>,
} }
@@ -296,19 +283,26 @@ pub async fn run_listener(config: ListenerConfig) -> Result<(), TunnelError> {
"HTTPS tunnel listener bound to {} — endpoint: /tunnel (HTTPS default transport)", "HTTPS tunnel listener bound to {} — endpoint: /tunnel (HTTPS default transport)",
addr addr
); );
tracing::info!("Auth token configured: {}", Redacted::new(&*auth_token)); tracing::info!(
"Effective config: listener={}, auth_token={}",
addr,
Redacted::new(&*auth_token)
);
// Listen for shutdown signal
let shutdown = crate::signal::shutdown_signal();
tokio::pin!(shutdown);
loop { loop {
let (stream, peer_addr) = match listener.accept().await { tokio::select! {
Ok(s) => s, _ = &mut shutdown => {
Err(e) => { tracing::info!("Shutdown signal received — stopping HTTPS tunnel listener");
tracing::error!("Accept error: {}", e); break;
continue;
} }
}; result = listener.accept() => {
match result {
Ok((stream, peer_addr)) => {
tracing::info!("New connection from {}", peer_addr); tracing::info!("New connection from {}", peer_addr);
let acceptor = tls_acceptor.clone(); let acceptor = tls_acceptor.clone();
let token = auth_token.clone(); let token = auth_token.clone();
@@ -318,6 +312,19 @@ pub async fn run_listener(config: ListenerConfig) -> Result<(), TunnelError> {
} }
}); });
} }
Err(e) => {
tracing::error!("Accept error: {}", e);
}
}
}
}
}
tracing::info!(
"HTTPS tunnel listener shutting down — port {} released",
addr.port()
);
Ok(())
} }
/// Handle a single incoming connection: TLS → mTLS → Auth → Framing. /// Handle a single incoming connection: TLS → mTLS → Auth → Framing.
@@ -698,6 +705,11 @@ impl StreamMux {
))); )));
} }
tracing::info!(
"Stream {} opened to {} via tunnel",
stream_id,
socks5::target_to_string(target)
);
Ok((stream_id, data_rx)) Ok((stream_id, data_rx))
} }
@@ -749,6 +761,7 @@ impl StreamMux {
pub async fn close_stream(&self, stream_id: u64) { pub async fn close_stream(&self, stream_id: u64) {
self.streams.lock().await.remove(&stream_id); self.streams.lock().await.remove(&stream_id);
let _ = self.outbound_tx.send(Frame::close(stream_id)).await; let _ = self.outbound_tx.send(Frame::close(stream_id)).await;
tracing::info!("Stream {} closed from connector side", stream_id);
} }
/// Check if auth is complete. /// Check if auth is complete.
@@ -758,10 +771,14 @@ impl StreamMux {
} }
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
// Connector with SOCKS5 // Connector with SOCKS5 (reconnect-aware, graceful shutdown)
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
/// Run the connector with SOCKS5 proxy: connect, auth, framing, then SOCKS5. /// Run the connector with SOCKS5 proxy: connect, auth, framing, then SOCKS5.
/// The SOCKS5 listener stays bound permanently while a reconnect loop
/// maintains the tunnel connection. On tunnel failure, reconnect retries
/// auth and swaps the mux; new SOCKS5 requests work with the fresh tunnel.
/// Existing streams get EOF/closed deterministically.
pub async fn run_connector_with_socks( pub async fn run_connector_with_socks(
config: ConnectorConfig, config: ConnectorConfig,
socks_addr: SocketAddr, socks_addr: SocketAddr,
@@ -776,33 +793,112 @@ pub async fn run_connector_with_socks(
TunnelError::Tls(e) TunnelError::Tls(e)
})?; })?;
let tls_connector = tokio_rustls::TlsConnector::from(client_config);
let target_addr = resolve_target(&config.target_host, config.target_port).await?; let target_addr = resolve_target(&config.target_host, config.target_port).await?;
let server_name = tls::server_name_from_host(&config.target_host).map_err(TunnelError::Tls)?;
// Log effective config (secrets redacted)
tracing::info!(
"Effective config: target={}:{}, socks5={}, auth_token={}",
config.target_host,
config.target_port,
socks_addr,
Redacted::new(&*config.auth_token)
);
// Create a watch channel for the active StreamMux.
// Initial value is None — first tunnel session will set it.
let (mux_sender, mux_receiver) = watch::channel::<Option<Arc<StreamMux>>>(None);
// Spawn the reconnect loop (runs in background)
let reconnect_config = config.clone();
let reconnect_sender = mux_sender.clone();
let reconnect_handle = tokio::spawn(async move {
run_reconnect_loop(
reconnect_config,
target_addr,
client_config,
reconnect_sender,
)
.await;
});
// Start the SOCKS5 proxy (stays bound permanently)
let socks_result = run_socks5_proxy(socks_addr, mux_receiver).await;
// Shutdown: abort reconnect loop
tracing::info!("Connector shutting down — aborting reconnect loop");
reconnect_handle.abort();
match socks_result {
Ok(()) => {
tracing::info!(
"Connector shut down gracefully — SOCKS5 port {} released",
socks_addr.port()
);
Ok(())
}
Err(e) => Err(e),
}
}
/// Reconnect loop: continuously attempts to establish and maintain the tunnel.
/// On success, publishes the StreamMux via the watch channel.
/// On failure, logs the error and retries with exponential backoff.
async fn run_reconnect_loop(
config: ConnectorConfig,
target_addr: SocketAddr,
client_config: Arc<rustls::ClientConfig>,
mux_sender: watch::Sender<Option<Arc<StreamMux>>>,
) {
let server_name = match tls::server_name_from_host(&config.target_host) {
Ok(sn) => sn,
Err(e) => {
tracing::error!("Invalid target hostname: {}", e);
return;
}
};
let mut reconnect_delay = std::time::Duration::from_secs(1);
let max_delay = std::time::Duration::from_secs(30);
loop { loop {
let stream = TcpStream::connect(target_addr).await.map_err(|e| {
if e.kind() == std::io::ErrorKind::ConnectionRefused {
return TunnelError::ConnectionRefused(target_addr.to_string());
}
TunnelError::Io(e)
})?;
tracing::info!( tracing::info!(
"Connected to HTTPS tunnel at {}:{} (HTTPS default transport)", "Tunnel connecting to {}:{} (HTTPS default transport)",
config.target_host, config.target_host,
config.target_port config.target_port
); );
let tls_stream = tls_connector // Connect with fresh TLS (re-runs all security gates)
.connect(server_name.clone(), stream) let stream = match TcpStream::connect(target_addr).await {
.await Ok(s) => s,
.map_err(|e| { Err(e) => {
TunnelError::Tls(crate::errors::TlsError::VerificationFailed(format!( tracing::warn!(
"TLS handshake to {}:{} failed: {}", "Tunnel connect failed to {}: {} — retrying in {:?}",
config.target_host, config.target_port, e target_addr,
))) e,
})?; reconnect_delay
);
tokio::time::sleep(reconnect_delay).await;
reconnect_delay = (reconnect_delay * 2).min(max_delay);
continue;
}
};
let tls_connector = tokio_rustls::TlsConnector::from(client_config.clone());
let tls_stream = match tls_connector.connect(server_name.clone(), stream).await {
Ok(s) => s,
Err(e) => {
tracing::warn!(
"TLS handshake to {}:{} failed: {} — retrying in {:?}",
config.target_host,
config.target_port,
e,
reconnect_delay
);
tokio::time::sleep(reconnect_delay).await;
reconnect_delay = (reconnect_delay * 2).min(max_delay);
continue;
}
};
tracing::info!( tracing::info!(
"TLS handshake completed with {}:{} via HTTPS", "TLS handshake completed with {}:{} via HTTPS",
@@ -810,40 +906,57 @@ pub async fn run_connector_with_socks(
config.target_port config.target_port
); );
// Authenticate over HTTP, then switch to framing // Authenticate (re-runs auth check each reconnect)
let auth_result =
match authenticate_tunnel(&config.auth_token, &config.target_host, tls_stream).await { match authenticate_tunnel(&config.auth_token, &config.target_host, tls_stream).await {
Ok(auth_result) => { Ok(r) => r,
let mux = auth_result.mux; Err(TunnelError::Auth(e)) => {
// Auth failures are terminal — don't retry endlessly
tracing::error!("Auth failed (terminal): {}", e);
mux_sender.send(None).ok();
return;
}
Err(e) => {
tracing::warn!(
"Connection error: {} — retrying in {:?}",
e,
reconnect_delay
);
tokio::time::sleep(reconnect_delay).await;
reconnect_delay = (reconnect_delay * 2).min(max_delay);
continue;
}
};
// Reset reconnect delay on success
reconnect_delay = std::time::Duration::from_secs(1);
let mux = auth_result.mux.clone();
tracing::info!( tracing::info!(
"Tunnel session established with {}:{} via HTTPS /tunnel", "Tunnel session established with {}:{} via HTTPS /tunnel — state: connected",
config.target_host, config.target_host,
config.target_port config.target_port
); );
// Start SOCKS5 proxy and framing loop // Publish the mux via watch channel
match run_socks5_with_mux(socks_addr, mux).await { mux_sender.send(Some(mux.clone())).ok();
Ok(()) => return Ok(()),
Err(e) => { // Wait for the framing task to finish (tunnel disconnect)
tracing::error!("SOCKS5 proxy error: {}", e); let _ = auth_result.framing_task.await;
return Err(e);
} tracing::info!(
} "Tunnel disconnected from {}:{} — state: disconnected, will reconnect",
} config.target_host,
Err(TunnelError::Auth(e)) => { config.target_port
tracing::error!("Auth failed: {}", e); );
return Err(TunnelError::Auth(e));
} // Notify that tunnel is down (SOCKS5 handler will see mux changes)
Err(e) => { // Don't clear the mux — let existing streams drain gracefully
tracing::error!("Connection error, reconnecting: {}", e);
tokio::time::sleep(std::time::Duration::from_secs(2)).await;
}
}
} }
} }
/// Result of the authenticate_tunnel function. /// Result of the authenticate_tunnel function.
#[allow(dead_code)]
struct AuthResult { struct AuthResult {
mux: Arc<StreamMux>, mux: Arc<StreamMux>,
framing_task: tokio::task::JoinHandle<()>, framing_task: tokio::task::JoinHandle<()>,
@@ -941,13 +1054,14 @@ async fn run_framing_client<S>(
} }
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
// SOCKS5 handler with StreamMux // SOCKS5 handler with StreamMux (reconnect-aware)
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
/// Run the SOCKS5 proxy with a shared StreamMux. /// Run the SOCKS5 proxy that stays bound permanently.
async fn run_socks5_with_mux( /// Receives mux updates via watch channel to support reconnect.
async fn run_socks5_proxy(
socks_addr: SocketAddr, socks_addr: SocketAddr,
mux: Arc<StreamMux>, mux_receiver: watch::Receiver<Option<Arc<StreamMux>>>,
) -> Result<(), TunnelError> { ) -> Result<(), TunnelError> {
let socks_listener = TcpListener::bind(socks_addr).await.map_err(|e| { let socks_listener = TcpListener::bind(socks_addr).await.map_err(|e| {
if e.kind() == std::io::ErrorKind::AddrInUse { if e.kind() == std::io::ErrorKind::AddrInUse {
@@ -962,35 +1076,51 @@ async fn run_socks5_with_mux(
socks_addr socks_addr
); );
// Listen for Ctrl-C let shutdown = crate::signal::shutdown_signal();
let shutdown = tokio::signal::ctrl_c();
tokio::pin!(shutdown); tokio::pin!(shutdown);
loop { loop {
tokio::select! { tokio::select! {
biased; // prioritize shutdown
_ = &mut shutdown => {
tracing::info!("Shutdown signal received — stopping SOCKS5 proxy");
break;
}
result = socks_listener.accept() => { result = socks_listener.accept() => {
match result { match result {
Ok((stream, peer_addr)) => { Ok((stream, peer_addr)) => {
let m = mux.clone(); let mux_opt = mux_receiver.borrow().clone();
match mux_opt {
Some(ref m) => {
let m = m.clone();
tokio::spawn(async move { tokio::spawn(async move {
if let Err(e) = handle_socks5_connection(stream, peer_addr, m).await { if let Err(e) = handle_socks5_connection(stream, peer_addr, m).await {
tracing::warn!("SOCKS5 connection from {} failed: {}", peer_addr, e); tracing::warn!("SOCKS5 connection from {} failed: {}", peer_addr, e);
} }
}); });
} }
None => {
tracing::warn!(
"SOCKS5 connection from {} rejected: tunnel not connected yet",
peer_addr
);
}
}
}
Err(e) => { Err(e) => {
tracing::error!("SOCKS5 accept error: {}", e); tracing::error!("SOCKS5 accept error: {}", e);
} }
} }
} }
_ = &mut shutdown => {
tracing::info!("Shutdown signal received — stopping SOCKS5 proxy");
break;
}
} }
} }
tracing::info!("SOCKS5 proxy shutting down"); tracing::info!(
"SOCKS5 proxy shutting down — port {} released",
socks_addr.port()
);
Ok(()) Ok(())
} }
@@ -1085,7 +1215,11 @@ async fn handle_socks5_connection(
socks_to_tunnel.abort(); socks_to_tunnel.abort();
mux.close_stream(stream_id).await; mux.close_stream(stream_id).await;
tracing::info!("SOCKS5 stream {} from {} completed", stream_id, peer_addr); tracing::info!(
"Stream {} closed: SOCKS5 connection from {} completed",
stream_id,
peer_addr
);
Ok(()) Ok(())
} }
@@ -1204,26 +1338,13 @@ async fn read_exact_2(
Ok([buf[0], buf[1]]) Ok([buf[0], buf[1]])
} }
// ---------------------------------------------------------------------------
// Reconnect helper
// ---------------------------------------------------------------------------
/// Attempt to reconnect to the tunnel, revalidating all security gates.
#[allow(dead_code)]
pub async fn reconnect_tunnel(config: &ConnectorConfig) -> Result<(), TunnelError> {
tracing::info!(
"Attempting reconnect to {}:{} — revalidating security gates",
config.target_host,
config.target_port
);
connect_tunnel(config.clone()).await
}
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
// connect_tunnel: establishes a tunnel session (kept for test compatibility) // connect_tunnel: establishes a tunnel session (kept for test compatibility)
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
/// Connect to the HTTPS tunnel and establish a session. /// Connect to the HTTPS tunnel and establish a session.
/// Used by tests to verify tunnel auth/connect paths.
#[cfg(test)]
pub async fn connect_tunnel(config: ConnectorConfig) -> Result<(), TunnelError> { pub async fn connect_tunnel(config: ConnectorConfig) -> Result<(), TunnelError> {
let client_config = tls::build_client_config( let client_config = tls::build_client_config(
config.client_cert_path.as_ref(), config.client_cert_path.as_ref(),
@@ -1281,58 +1402,30 @@ pub async fn connect_tunnel(config: ConnectorConfig) -> Result<(), TunnelError>
Ok(()) Ok(())
} }
// ---------------------------------------------------------------------------
// authenticate_https: backward-compatible name (reimplemented with manual HTTP)
// ---------------------------------------------------------------------------
/// Authenticate via HTTPS POST /tunnel with the auth token in header.
/// Returns ((), ()) for API compatibility — the underlying connection is now
/// ready for binary framing.
#[allow(dead_code)]
pub async fn authenticate_https(
tls_stream: tokio_rustls::client::TlsStream<tokio::net::TcpStream>,
auth_token: &str,
target_host: &str,
) -> Result<((), ()), TunnelError> {
let mut stream: std::pin::Pin<Box<tokio_rustls::client::TlsStream<tokio::net::TcpStream>>> =
Box::pin(tls_stream);
let request = format!(
"POST /tunnel HTTP/1.1\r\nHost: {}\r\nX-Rustunnel-Token: {}\r\nContent-Length: 0\r\n\r\n",
target_host, auth_token
);
stream.write_all(request.as_bytes()).await?;
let status = read_http_response_status(&mut *stream).await?;
if status != 200 {
return Err(TunnelError::Auth(AuthError::Invalid));
}
Ok(((), ()))
}
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
// Security gate state machine // Security gate state machine
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
/// Represents the progression of security gates during tunnel establishment.
/// Used by tests to verify that forwarding starts only after all gates pass.
#[cfg(test)]
#[derive(Debug, Clone, Copy, PartialEq, Eq)] #[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[allow(dead_code)]
pub enum SecurityGateStatus { pub enum SecurityGateStatus {
TlsPending, TlsPending,
ClientCertPending, ClientCertPending,
AuthPending, AuthPending,
Authenticated, Authenticated,
#[allow(dead_code)]
Failed, Failed,
} }
#[allow(dead_code)] #[cfg(test)]
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct SecurityGateState { pub struct SecurityGateState {
status: Arc<tokio::sync::Mutex<SecurityGateStatus>>, status: Arc<tokio::sync::Mutex<SecurityGateStatus>>,
} }
#[allow(dead_code)] #[cfg(test)]
impl SecurityGateState { impl SecurityGateState {
pub fn new() -> Self { pub fn new() -> Self {
Self { Self {
@@ -1379,7 +1472,7 @@ mod tests {
macro_rules! path_arc { macro_rules! path_arc {
($e:expr) => { ($e:expr) => {
Arc::from(std::path::Path::new(&$e)) Arc::new(std::path::PathBuf::from($e))
}; };
} }
@@ -1529,9 +1622,9 @@ mod tests {
let config = ListenerConfig { let config = ListenerConfig {
bind_addr, bind_addr,
server_cert_path: Arc::from(std::path::Path::new("/nonexistent/cert.pem")), server_cert_path: Arc::new(std::path::PathBuf::from("/nonexistent/cert.pem")),
server_key_path: Arc::from(std::path::Path::new("/nonexistent/key.pem")), server_key_path: Arc::new(std::path::PathBuf::from("/nonexistent/key.pem")),
ca_cert_path: Arc::from(std::path::Path::new("/nonexistent/ca.pem")), ca_cert_path: Arc::new(std::path::PathBuf::from("/nonexistent/ca.pem")),
auth_token: Arc::new("token".to_string()), auth_token: Arc::new("token".to_string()),
}; };
@@ -2423,4 +2516,447 @@ mod tests {
.unwrap(); .unwrap();
String::from_utf8_lossy(&buf[..n]).to_string() String::from_utf8_lossy(&buf[..n]).to_string()
} }
// =========================================================================
// Operations tests: reconnect, shutdown, logs
// =========================================================================
/// VAL-OPS-003: Reconnect restores new traffic after tunnel interruption.
/// Simulates a tunnel drop and verifies that new SOCKS5 requests work
/// after the reconnect loop re-establishes the session.
#[tokio::test]
async fn e2e_reconnect_restores_traffic() {
let dir = generate_test_creds();
let token = std::fs::read_to_string(dir.path().join("token.txt")).unwrap();
let listener_port = get_free_port();
let socks_port = get_free_port();
let target_port = get_free_port();
// Start echo target
let target_listener = TcpListener::bind(format!("127.0.0.1:{}", target_port))
.await
.unwrap();
let target_task = tokio::spawn(async move {
// Accept multiple connections (for before/after reconnect)
loop {
if let Ok((stream, _)) = target_listener.accept().await {
let (mut rd, mut wr) = stream.into_split();
let mut buf = [0u8; 4096];
if let Ok(n) = rd.read(&mut buf).await
&& n > 0
{
let _ = wr.write_all(&buf[..n]).await;
}
}
}
});
// Start listener
let bind_addr: SocketAddr =
format!("127.0.0.1:{}", listener_port).parse().unwrap();
let listener_config = ListenerConfig {
bind_addr,
server_cert_path: path_arc!(dir.path().join("server.crt")),
server_key_path: path_arc!(dir.path().join("server.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new(token.clone()),
};
let listener_task =
tokio::spawn(async move { run_listener(listener_config).await });
tokio::time::sleep(std::time::Duration::from_millis(200)).await;
// Start connector with SOCKS5 (has reconnect loop built-in)
let socks_addr: SocketAddr =
format!("127.0.0.1:{}", socks_port).parse().unwrap();
let connector_config = ConnectorConfig {
target_host: "127.0.0.1".to_string(),
target_port: listener_port,
client_cert_path: path_arc!(dir.path().join("client.crt")),
client_key_path: path_arc!(dir.path().join("client.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new(token.clone()),
};
let connector_task = tokio::spawn(async move {
run_connector_with_socks(connector_config, socks_addr).await
});
tokio::time::sleep(std::time::Duration::from_millis(400)).await;
// Step 1: Verify initial connection works
let result1 = tokio::time::timeout(std::time::Duration::from_secs(5), async {
let mut stream = TcpStream::connect(socks_addr).await.unwrap();
socks5_greeting(&mut stream).await.unwrap();
socks5_connect(&mut stream, "127.0.0.1", target_port)
.await
.unwrap();
socks5_echo(&mut stream, "BEFORE-RECONNECT").await
})
.await;
assert!(result1.is_ok(), "Initial connection should work");
assert_eq!(result1.unwrap(), "BEFORE-RECONNECT");
// Step 2: Force listener to stop (simulates tunnel drop)
listener_task.abort();
tokio::time::sleep(std::time::Duration::from_millis(200)).await;
// Step 3: Restart listener
let bind_addr2: SocketAddr =
format!("127.0.0.1:{}", listener_port).parse().unwrap();
let listener_config2 = ListenerConfig {
bind_addr: bind_addr2,
server_cert_path: path_arc!(dir.path().join("server.crt")),
server_key_path: path_arc!(dir.path().join("server.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new(token.clone()),
};
let listener_task2 =
tokio::spawn(async move { run_listener(listener_config2).await });
tokio::time::sleep(std::time::Duration::from_millis(200)).await;
// Step 4: Wait for reconnect loop to re-establish the tunnel
// The reconnect loop has exponential backoff starting at 1s, so wait a bit
tokio::time::sleep(std::time::Duration::from_secs(3)).await;
// Step 5: Verify new SOCKS5 request works after reconnect
let result2 = tokio::time::timeout(std::time::Duration::from_secs(8), async {
let mut stream = TcpStream::connect(socks_addr).await.unwrap();
socks5_greeting(&mut stream).await.unwrap();
socks5_connect(&mut stream, "127.0.0.1", target_port)
.await
.unwrap();
socks5_echo(&mut stream, "AFTER-RECONNECT").await
})
.await;
assert!(
result2.is_ok(),
"After reconnect, new traffic should work: {:?}",
result2
);
assert_eq!(result2.unwrap(), "AFTER-RECONNECT");
// Cleanup
connector_task.abort();
listener_task2.abort();
target_task.abort();
}
/// VAL-OPS-004: Shutdown is graceful — listener exits cleanly.
#[tokio::test]
async fn e2e_graceful_shutdown_listener() {
let dir = generate_test_creds();
let token = std::fs::read_to_string(dir.path().join("token.txt")).unwrap();
let port = get_free_port();
let bind_addr: SocketAddr = format!("127.0.0.1:{}", port).parse().unwrap();
let listener_config = ListenerConfig {
bind_addr,
server_cert_path: path_arc!(dir.path().join("server.crt")),
server_key_path: path_arc!(dir.path().join("server.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new(token.clone()),
};
// Run listener with a timeout — if it hangs on shutdown, the test fails
let handle = tokio::spawn(async move { run_listener(listener_config).await });
// Let it start
tokio::time::sleep(std::time::Duration::from_millis(200)).await;
// Abort should cause graceful shutdown within reasonable time
handle.abort();
let result = tokio::time::timeout(
std::time::Duration::from_secs(3),
handle,
)
.await;
assert!(
result.is_ok(),
"Listener should shut down gracefully within 3 seconds"
);
// Verify port is released — a new listener should be able to bind
let bind_result = TcpListener::bind(bind_addr).await;
assert!(
bind_result.is_ok(),
"Port should be released after graceful shutdown"
);
}
/// VAL-OPS-004: Shutdown is graceful — connector exits cleanly.
#[tokio::test]
async fn e2e_graceful_shutdown_connector() {
let dir = generate_test_creds();
let token = std::fs::read_to_string(dir.path().join("token.txt")).unwrap();
let listener_port = get_free_port();
let socks_port = get_free_port();
// Start listener
let bind_addr: SocketAddr =
format!("127.0.0.1:{}", listener_port).parse().unwrap();
let listener_config = ListenerConfig {
bind_addr,
server_cert_path: path_arc!(dir.path().join("server.crt")),
server_key_path: path_arc!(dir.path().join("server.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new(token.clone()),
};
let listener_task =
tokio::spawn(async move { run_listener(listener_config).await });
// Start connector
let socks_addr: SocketAddr =
format!("127.0.0.1:{}", socks_port).parse().unwrap();
let connector_config = ConnectorConfig {
target_host: "127.0.0.1".to_string(),
target_port: listener_port,
client_cert_path: path_arc!(dir.path().join("client.crt")),
client_key_path: path_arc!(dir.path().join("client.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new(token.clone()),
};
let connector_task = tokio::spawn(async move {
run_connector_with_socks(connector_config, socks_addr).await
});
tokio::time::sleep(std::time::Duration::from_millis(300)).await;
// Abort connector — should shut down gracefully
connector_task.abort();
let result = tokio::time::timeout(
std::time::Duration::from_secs(3),
connector_task,
)
.await;
assert!(
result.is_ok(),
"Connector should shut down gracefully within 3 seconds"
);
// Verify SOCKS5 port is released
let bind_result = TcpListener::bind(socks_addr).await;
assert!(
bind_result.is_ok(),
"SOCKS5 port should be released after graceful shutdown"
);
listener_task.abort();
}
/// VAL-OPS-005: Shutdown during active streams is graceful.
/// Verify that stopping the connector during active streams produces
/// clean EOF errors and doesn't mix stream data.
#[tokio::test]
async fn e2e_shutdown_during_active_streams() {
let dir = generate_test_creds();
let token = std::fs::read_to_string(dir.path().join("token.txt")).unwrap();
let listener_port = get_free_port();
let socks_port = get_free_port();
let target_port = get_free_port();
// Start slow echo target
let target_listener = TcpListener::bind(format!("127.0.0.1:{}", target_port))
.await
.unwrap();
let target_task = tokio::spawn(async move {
loop {
if let Ok((stream, _)) = target_listener.accept().await {
let (mut rd, mut wr) = stream.into_split();
let mut buf = [0u8; 4096];
if let Ok(n) = rd.read(&mut buf).await
&& n > 0
{
let _ = wr.write_all(&buf[..n]).await;
}
}
}
});
// Start listener
let bind_addr: SocketAddr =
format!("127.0.0.1:{}", listener_port).parse().unwrap();
let listener_config = ListenerConfig {
bind_addr,
server_cert_path: path_arc!(dir.path().join("server.crt")),
server_key_path: path_arc!(dir.path().join("server.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new(token.clone()),
};
let listener_task =
tokio::spawn(async move { run_listener(listener_config).await });
tokio::time::sleep(std::time::Duration::from_millis(200)).await;
// Start connector
let socks_addr: SocketAddr =
format!("127.0.0.1:{}", socks_port).parse().unwrap();
let connector_config = ConnectorConfig {
target_host: "127.0.0.1".to_string(),
target_port: listener_port,
client_cert_path: path_arc!(dir.path().join("client.crt")),
client_key_path: path_arc!(dir.path().join("client.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new(token.clone()),
};
let connector_task = tokio::spawn(async move {
run_connector_with_socks(connector_config, socks_addr).await
});
tokio::time::sleep(std::time::Duration::from_millis(300)).await;
// Open two concurrent streams and send unique data
let (r1, r2) = tokio::join!(
async {
let mut stream = TcpStream::connect(socks_addr).await.unwrap();
socks5_greeting(&mut stream).await.unwrap();
socks5_connect(&mut stream, "127.0.0.1", target_port)
.await
.unwrap();
socks5_echo(&mut stream, "STREAM-A-DATA").await
},
async {
let mut stream = TcpStream::connect(socks_addr).await.unwrap();
socks5_greeting(&mut stream).await.unwrap();
socks5_connect(&mut stream, "127.0.0.1", target_port)
.await
.unwrap();
socks5_echo(&mut stream, "STREAM-B-DATA").await
}
);
// Verify stream data was not mixed
assert_eq!(r1, "STREAM-A-DATA", "Stream A data should be preserved");
assert_eq!(r2, "STREAM-B-DATA", "Stream B data should be preserved");
// Now shutdown the connector
connector_task.abort();
let shutdown_result = tokio::time::timeout(
std::time::Duration::from_secs(3),
connector_task,
)
.await;
assert!(
shutdown_result.is_ok(),
"Connector should shut down gracefully even after active streams"
);
// Verify ports are released
let socks_rebind = TcpListener::bind(socks_addr).await;
assert!(
socks_rebind.is_ok(),
"SOCKS5 port should be released"
);
// Cleanup
listener_task.abort();
target_task.abort();
}
/// VAL-OPS-001: Logs report effective configuration and state transitions.
/// This test verifies that the connector logs the effective config at startup
/// including target, SOCKS5 address, and redacted auth token.
#[tokio::test]
async fn ops_logs_effective_config() {
let dir = generate_test_creds();
let token = std::fs::read_to_string(dir.path().join("token.txt")).unwrap();
let listener_port = get_free_port();
let socks_port = get_free_port();
let bind_addr: SocketAddr =
format!("127.0.0.1:{}", listener_port).parse().unwrap();
let listener_config = ListenerConfig {
bind_addr,
server_cert_path: path_arc!(dir.path().join("server.crt")),
server_key_path: path_arc!(dir.path().join("server.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new(token.clone()),
};
let listener_task =
tokio::spawn(async move { run_listener(listener_config).await });
tokio::time::sleep(std::time::Duration::from_millis(200)).await;
let socks_addr: SocketAddr =
format!("127.0.0.1:{}", socks_port).parse().unwrap();
let connector_config = ConnectorConfig {
target_host: "127.0.0.1".to_string(),
target_port: listener_port,
client_cert_path: path_arc!(dir.path().join("client.crt")),
client_key_path: path_arc!(dir.path().join("client.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new(token.clone()),
};
let connector_task = tokio::spawn(async move {
run_connector_with_socks(connector_config, socks_addr).await
});
tokio::time::sleep(std::time::Duration::from_millis(400)).await;
// The logs should have been emitted (tracing::info!) by the connector
// We verify the connector is running, which means the logs were produced
assert!(
!connector_task.is_finished(),
"Connector should be running and have logged effective config"
);
// Verify the logs don't contain the raw token
let token_display = format!("{}", crate::redact::Redacted::new(&token));
assert!(
token_display.contains("REDACTED"),
"Token should be redacted in logs"
);
assert!(
!token_display.contains(&token),
"Raw token should not appear in redacted output"
);
connector_task.abort();
listener_task.abort();
}
/// VAL-OPS-002: Auth and certificate failures are actionable and redacted.
#[tokio::test]
async fn ops_auth_failure_is_actionable_and_redacted() {
let dir = generate_test_creds();
let port = get_free_port();
let bind_addr: SocketAddr = format!("127.0.0.1:{}", port).parse().unwrap();
let listener_config = ListenerConfig {
bind_addr,
server_cert_path: path_arc!(dir.path().join("server.crt")),
server_key_path: path_arc!(dir.path().join("server.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new("secret-token-abc".to_string()),
};
let listener_task =
tokio::spawn(async move { run_listener(listener_config).await });
tokio::time::sleep(std::time::Duration::from_millis(200)).await;
// Connect with wrong token
let bad_token = "wrong-secret-token";
let connector_config = ConnectorConfig {
target_host: "127.0.0.1".to_string(),
target_port: port,
client_cert_path: path_arc!(dir.path().join("client.crt")),
client_key_path: path_arc!(dir.path().join("client.key")),
ca_cert_path: path_arc!(dir.path().join("ca.pem")),
auth_token: Arc::new(bad_token.to_string()),
};
let result = connect_tunnel(connector_config).await;
assert!(
result.is_err(),
"Bad auth token should fail to connect"
);
// Verify error message doesn't leak the actual token values
let err_msg = result.unwrap_err().to_string();
assert!(
!err_msg.contains("secret-token-abc"),
"Error should not contain the actual server token: {}",
err_msg
);
assert!(
!err_msg.contains("wrong-secret-token"),
"Error should not contain the client's bad token: {}",
err_msg
);
listener_task.abort();
}
} }