Share connection semaphore across listeners (review #010 C4)

This commit is contained in:
glm-5.3-flash committed 2026-09-13 07:46:58 +00:00
1 parent 9e26295cf7
commit 1bbb9c237c
6 files changed
+183 -25

No files matched your search

+1 -1
View File
@@ -117,7 +117,7 @@ Configuration uses TOML and is split into **static** (requires restart) and
| `shutdown_timeout_secs` | `30` | Graceful shutdown timeout | | `shutdown_timeout_secs` | `30` | Graceful shutdown timeout |
| `connection_idle_timeout_secs` | `60` | Server-side idle timeout for client TLS connections (prevents FD exhaustion from abandoned connections) | | `connection_idle_timeout_secs` | `60` | Server-side idle timeout for client TLS connections (prevents FD exhaustion from abandoned connections) |
| `tls_handshake_timeout_secs` | `10` | Max seconds to complete the TLS handshake; stalled handshakes are closed (prevents slowloris FD/permit exhaustion) | | `tls_handshake_timeout_secs` | `10` | Max seconds to complete the TLS handshake; stalled handshakes are closed (prevents slowloris FD/permit exhaustion) |
| `max_connections` | `1024` | Max concurrent client TLS connections (backpressure via semaphore) | | `max_connections` | `1024` | Max concurrent client TLS connections, shared across all listeners (global semaphore) |
| `logging.level` | `"info"` | Log level | | `logging.level` | `"info"` | Log level |
| `logging.format` | `"text"` | Log format (`"text"` or `"json"`) | | `logging.format` | `"text"` | Log format (`"text"` or `"json"`) |
| `logging.log_file_path` | (not set) | Path to log file for fail2ban | | `logging.log_file_path` | (not set) | Path to log file for fail2ban |
+5 -3
View File
@@ -92,7 +92,7 @@ Immutable after startup. Changes require a process restart.
| `shutdown_timeout_secs` | `u64` | Maximum seconds to wait for in-flight requests during graceful shutdown (default: `30`) | | `shutdown_timeout_secs` | `u64` | Maximum seconds to wait for in-flight requests during graceful shutdown (default: `30`) |
| `connection_idle_timeout_secs` | `u64` | Server-side idle timeout for client TLS connections. Idle HTTP/2 connections are closed after this duration (with keep-alive pings at 15s intervals to detect dead peers). HTTP/1.1 connections are closed if the client doesn't send a complete request header within this duration. Prevents FD exhaustion from abandoned connections (default: `60`; must be > 0; see review #007 C1) | | `connection_idle_timeout_secs` | `u64` | Server-side idle timeout for client TLS connections. Idle HTTP/2 connections are closed after this duration (with keep-alive pings at 15s intervals to detect dead peers). HTTP/1.1 connections are closed if the client doesn't send a complete request header within this duration. Prevents FD exhaustion from abandoned connections (default: `60`; must be > 0; see review #007 C1) |
| `tls_handshake_timeout_secs` | `u64` | Maximum seconds a client may take to complete the TLS handshake. Stalled handshakes (e.g. crawlers/slowloris clients that open a TCP connection but never send a ClientHello) are closed after this duration, releasing the FD and connection slot. Without it, a stalled handshake holds an FD + connection semaphore permit indefinitely (default: `10`; must be > 0; see review #010 C3) | | `tls_handshake_timeout_secs` | `u64` | Maximum seconds a client may take to complete the TLS handshake. Stalled handshakes (e.g. crawlers/slowloris clients that open a TCP connection but never send a ClientHello) are closed after this duration, releasing the FD and connection slot. Without it, a stalled handshake holds an FD + connection semaphore permit indefinitely (default: `10`; must be > 0; see review #010 C3) |
| `max_connections` | `usize` | Maximum number of concurrent client TLS connections. When the limit is reached, new connections wait in the OS TCP backlog until a slot frees (default: `1024`; must be > 0; see review #007 C2) | | `max_connections` | `usize` | Maximum number of concurrent client TLS connections **process-wide**. The connection semaphore is shared across all HTTPS listeners, so this is a global cap — not a per-listener cap. When the limit is reached, new connections wait in the OS TCP backlog until a slot frees (default: `1024`; must be > 0; see review #007 C2, review #010 C4) |
| `logging` | `LoggingConfig` | Logging configuration (see below) | | `logging` | `LoggingConfig` | Logging configuration (see below) |
**LoggingConfig** (nested in `[logging]` TOML section): **LoggingConfig** (nested in `[logging]` TOML section):
@@ -316,7 +316,7 @@ health_check_port = 9900 # Local health check (0 to disable)
admin_key_path = "/etc/reverse-proxy/admin-key" # Empty string to disable admin_key_path = "/etc/reverse-proxy/admin-key" # Empty string to disable
# connection_idle_timeout_secs = 60 # Server-side idle timeout (default: 60) # connection_idle_timeout_secs = 60 # Server-side idle timeout (default: 60)
# tls_handshake_timeout_secs = 10 # TLS handshake timeout (default: 10) # tls_handshake_timeout_secs = 10 # TLS handshake timeout (default: 10)
# max_connections = 1024 # Max concurrent TLS connections (default: 1024) # max_connections = 1024 # Max concurrent TLS connections, process-wide across all listeners (default: 1024)
[logging] [logging]
level = "info" level = "info"
@@ -463,7 +463,9 @@ On startup, the config is validated:
the server-side idle timeout, reintroducing the FD exhaustion bug from the server-side idle timeout, reintroducing the FD exhaustion bug from
review #007 C1. review #007 C1.
22. `max_connections` must be > 0. A zero value would deadlock the connection 22. `max_connections` must be > 0. A zero value would deadlock the connection
semaphore, preventing any client connection from being accepted. semaphore, preventing any client connection from being accepted. The
semaphore is shared across all listeners (review #010 C4), so
`max_connections` is the process-wide cap rather than per-listener.
23. `tls_handshake_timeout_secs` must be > 0. A zero value would immediately 23. `tls_handshake_timeout_secs` must be > 0. A zero value would immediately
kill every TLS handshake, preventing any client connection from kill every TLS handshake, preventing any client connection from
completing (review #010 C3). completing (review #010 C3).
+22 -15
View File
@@ -24,9 +24,9 @@ fixes:
default 10s) wraps tls_acceptor.accept() in src/server.rs; stalled default 10s) wraps tls_acceptor.accept() in src/server.rs; stalled
handshakes release FD + permit. handshakes release FD + permit.
- >- - >-
C4: OPEN — connection semaphore is per-listener (src/server.rs:338), so C4: FIXED 2026-09-13 — connection semaphore is shared across all
the effective cap is max_connections × listeners; sequence before C2 so listeners (ConnectionSemaphore, src/server.rs); max_connections is now a
the RLIMIT cross-check uses the real FD budget. process-wide cap, not max_connections × listeners.
- >- - >-
C2: OPEN — no startup cross-check that max_connections fits under C2: OPEN — no startup cross-check that max_connections fits under
RLIMIT_NOFILE with headroom. Sequenced after C3/C4. RLIMIT_NOFILE with headroom. Sequenced after C3/C4.
@@ -187,11 +187,13 @@ Notes from implementation:
(default 10s) wrapping the accept in `tokio::time::timeout`. This (default 10s) wrapping the accept in `tokio::time::timeout`. This
was the likely actual FD-exhaustion vector for slow/held crawler was the likely actual FD-exhaustion vector for slow/held crawler
handshakes, and a slowloris amplifier. handshakes, and a slowloris amplifier.
- **C4 (open)**: `conn_sem` is per-listener (`main.rs` spawns one - ~~**C4 (open)**~~ — **FIXED 2026-09-13**: `conn_sem` was per-listener
`serve_https_listener` per listener, each creating its own (`main.rs` spawns one `serve_https_listener` per listener, each creating
semaphore), so the effective connection cap is its own semaphore), so the effective connection cap was
`max_connections × listeners`. Any C2 RLIMIT cross-check must `max_connections × listeners`. Fixed via a single `ConnectionSemaphore`
account for this (or the semaphore should be shared). (`Arc<Semaphore>` wrapper, src/server.rs) created in `main.rs` and shared
by every listener; `max_connections` is now a process-wide cap. This is
the topology the C2 RLIMIT cross-check should assume.
### Original finding (pre-fix, preserved for context) ### Original finding (pre-fix, preserved for context)
@@ -265,7 +267,7 @@ observed limit).
Residual risk after mitigation: none identified for FD exhaustion at Residual risk after mitigation: none identified for FD exhaustion at
current traffic (peak concurrent connections observed ≪ 800); the code current traffic (peak concurrent connections observed ≪ 800); the code
findings C4/C2 remain the durable fix (C1 + C3 landed 2026-09-13). findings C2 remains the durable fix (C1 + C3 + C4 landed 2026-09-13).
## Traffic-analysis side note (from the same investigation) ## Traffic-analysis side note (from the same investigation)
@@ -298,13 +300,18 @@ behavior, which is exactly what made the EMFILE state reachable.
future is dropped, releasing the TCP FD and the semaphore permit; the future is dropped, releasing the TCP FD and the semaphore permit; the
idle watchdog never needs to run for a stalled handshake. Closes the idle watchdog never needs to run for a stalled handshake. Closes the
crawler slow-handshake vector described under "Trigger conditions". crawler slow-handshake vector described under "Trigger conditions".
3. Land C4 (shared connection semaphore across listeners) so 3. ~~Land C4 (shared connection semaphore across listeners) so
`max_connections` is a global cap rather than per-listener. Sequenced `max_connections` is a global cap rather than per-listener~~ — DONE
before C2 because the RLIMIT cross-check's FD budget depends on the 2026-09-13. `serve_https_listener()` no longer creates its own semaphore;
final semaphore topology (shared vs per-listener). it receives an `Arc<ConnectionSemaphore>` (new public wrapper in
src/server.rs) built once in `main.rs` and cloned into every listener
task. `max_connections` is now the process-wide concurrent TLS connection
cap; the effective cap no longer scales with listener count. This is the
topology C2's RLIMIT cross-check should assume.
4. Land C2 (RLIMIT cross-check at startup) with a prominent warning or 4. Land C2 (RLIMIT cross-check at startup) with a prominent warning or
hard validation error. Must account for C4 (per-listener semaphore hard validation error. The FD budget assumes a shared semaphore
multiplication → shared after C4 lands) when computing the FD budget. (landed, see step 3): connection FDs are capped at `max_connections`
process-wide.
Lower urgency after M1: with the deploy baseline (`nofile 8192`, Lower urgency after M1: with the deploy baseline (`nofile 8192`,
`max_connections 800`) the ceiling is ~10% of the limit, so C2 is a `max_connections 800`) the ceiling is ~10% of the limit, so C2 is a
validation guard, not an active exposure. validation guard, not an active exposure.
+10 -2
View File
@@ -17,7 +17,9 @@ use reverse_proxy::health;
use reverse_proxy::logging; use reverse_proxy::logging;
use reverse_proxy::proxy::{build_router, create_http_client, create_https_client, ProxyState}; use reverse_proxy::proxy::{build_router, create_http_client, create_https_client, ProxyState};
use reverse_proxy::rate_limit::{start_eviction_task, RateLimiter}; use reverse_proxy::rate_limit::{start_eviction_task, RateLimiter};
use reverse_proxy::server::{drain_in_flight, serve_https_listener, InFlightCounter}; use reverse_proxy::server::{
drain_in_flight, serve_https_listener, ConnectionSemaphore, InFlightCounter,
};
use reverse_proxy::shutdown::GracefulShutdown; use reverse_proxy::shutdown::GracefulShutdown;
use reverse_proxy::tls::acceptor::{setup_tls, TlsMode}; use reverse_proxy::tls::acceptor::{setup_tls, TlsMode};
use reverse_proxy::tls::redirect; use reverse_proxy::tls::redirect;
@@ -217,6 +219,12 @@ async fn run_server(loaded_config: cli::LoadedConfig, config_path: &str) -> Resu
let app = build_router(proxy_state.clone(), config_arc.clone(), rate_limiter); let app = build_router(proxy_state.clone(), config_arc.clone(), rate_limiter);
let in_flight = InFlightCounter::new(); let in_flight = InFlightCounter::new();
let conn_sem = ConnectionSemaphore::new(loaded_config.static_config.max_connections);
info!(
max_connections = conn_sem.max_connections(),
"global connection semaphore created (shared across all listeners)"
);
let mut https_server_handles = Vec::new(); let mut https_server_handles = Vec::new();
@@ -237,7 +245,7 @@ async fn run_server(loaded_config: cli::LoadedConfig, config_path: &str) -> Resu
std::time::Duration::from_secs( std::time::Duration::from_secs(
loaded_config.static_config.tls_handshake_timeout_secs, loaded_config.static_config.tls_handshake_timeout_secs,
), ),
loaded_config.static_config.max_connections, conn_sem.clone(),
)); ));
info!( info!(
+55 -3
View File
@@ -171,6 +171,33 @@ impl InFlightCounter {
} }
} }
/// Connection semaphore shared across all HTTPS listeners so that
/// `max_connections` acts as a global cap rather than a per-listener cap
/// (review #010 C4).
pub struct ConnectionSemaphore {
semaphore: Arc<Semaphore>,
max_connections: usize,
}
impl ConnectionSemaphore {
pub fn new(max_connections: usize) -> Arc<Self> {
Arc::new(Self {
semaphore: Arc::new(Semaphore::new(max_connections)),
max_connections,
})
}
pub fn max_connections(&self) -> usize {
self.max_connections
}
pub async fn acquire_owned(
&self,
) -> Result<tokio::sync::OwnedSemaphorePermit, tokio::sync::AcquireError> {
self.semaphore.clone().acquire_owned().await
}
}
#[derive(Debug)] #[derive(Debug)]
struct IdleState { struct IdleState {
last_activity: Mutex<Instant>, last_activity: Mutex<Instant>,
@@ -354,10 +381,9 @@ pub async fn serve_https_listener(
in_flight: Arc<InFlightCounter>, in_flight: Arc<InFlightCounter>,
connection_idle_timeout: Duration, connection_idle_timeout: Duration,
tls_handshake_timeout: Duration, tls_handshake_timeout: Duration,
max_connections: usize, conn_sem: Arc<ConnectionSemaphore>,
) { ) {
let local_addr = tcp_listener.local_addr(); let local_addr = tcp_listener.local_addr();
let conn_sem = Arc::new(Semaphore::new(max_connections));
let accept_errors = Arc::new(AcceptErrorReporter::new()); let accept_errors = Arc::new(AcceptErrorReporter::new());
loop { loop {
@@ -374,7 +400,6 @@ pub async fn serve_https_listener(
let tls_acceptor = tls_acceptor.clone(); let tls_acceptor = tls_acceptor.clone();
let router = router.clone(); let router = router.clone();
let in_flight = in_flight.clone(); let in_flight = in_flight.clone();
let conn_sem = conn_sem.clone();
let permit = match conn_sem.acquire_owned().await { let permit = match conn_sem.acquire_owned().await {
Ok(permit) => permit, Ok(permit) => permit,
@@ -799,6 +824,33 @@ mod tests {
assert_eq!(state.in_flight(), 0); assert_eq!(state.in_flight(), 0);
} }
#[tokio::test]
async fn connection_semaphore_reports_cap() {
let sem = ConnectionSemaphore::new(7);
assert_eq!(sem.max_connections(), 7);
}
#[tokio::test]
async fn connection_semaphore_permits_are_shared_across_clones() {
let sem = ConnectionSemaphore::new(2);
let p1 = sem.acquire_owned().await.unwrap();
let p2 = sem.acquire_owned().await.unwrap();
let clone = Arc::clone(&sem);
let blocked = tokio::time::timeout(Duration::from_millis(100), clone.acquire_owned()).await;
assert!(
blocked.is_err(),
"third acquire through a clone must block once the shared pool of 2 is exhausted"
);
drop(p1);
drop(p2);
let p3 = sem.acquire_owned().await.expect("permits released, acquire succeeds");
drop(p3);
}
#[test] #[test]
fn idle_state_touch_updates_last_activity() { fn idle_state_touch_updates_last_activity() {
let state = IdleState::new(); let state = IdleState::new();
+90 -1
View File
@@ -1176,6 +1176,21 @@ mod idle_timeout_tests {
Arc<InFlightCounter>, Arc<InFlightCounter>,
tokio::task::JoinHandle<()>, tokio::task::JoinHandle<()>,
tokio::sync::watch::Sender<bool>, tokio::sync::watch::Sender<bool>,
) {
let conn_sem = reverse_proxy::server::ConnectionSemaphore::new(1024);
spawn_test_https_server(idle_timeout, handshake_timeout, upstream, conn_sem).await
}
async fn spawn_test_https_server(
idle_timeout: Duration,
handshake_timeout: Duration,
upstream: String,
conn_sem: Arc<reverse_proxy::server::ConnectionSemaphore>,
) -> (
std::net::SocketAddr,
Arc<InFlightCounter>,
tokio::task::JoinHandle<()>,
tokio::sync::watch::Sender<bool>,
) { ) {
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap(); let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
let addr = listener.local_addr().unwrap(); let addr = listener.local_addr().unwrap();
@@ -1195,7 +1210,7 @@ mod idle_timeout_tests {
in_flight_clone, in_flight_clone,
idle_timeout, idle_timeout,
handshake_timeout, handshake_timeout,
1024, conn_sem,
) )
.await; .await;
}); });
@@ -1539,4 +1554,78 @@ mod idle_timeout_tests {
let _ = upstream.shutdown_tx.send(()); let _ = upstream.shutdown_tx.send(());
} }
// Reproducer for review #010 C4: the connection semaphore must be shared
// across listeners. Two listeners with a shared cap of 2 must admit only
// 2 concurrent connections total — the third connection, hitting either
// listener, must wait until a slot frees.
#[tokio::test]
async fn connection_semaphore_is_shared_across_listeners() {
let conn_sem = reverse_proxy::server::ConnectionSemaphore::new(2);
let upstream = helpers::http_test_helper::TestUpstream::spawn_ok().await;
let upstream_addr = format!("127.0.0.1:{}", upstream.addr.port());
let (addr1, _in_flight1, _h1, _s1) = spawn_test_https_server(
Duration::from_secs(60),
Duration::from_secs(10),
upstream_addr.clone(),
conn_sem.clone(),
)
.await;
let (addr2, _in_flight2, _h2, _s2) = spawn_test_https_server(
Duration::from_secs(60),
Duration::from_secs(10),
upstream_addr,
conn_sem.clone(),
)
.await;
// Two live connections saturate the shared pool: one per listener.
let tls1 = connect_tls(addr1).await;
let tls2 = connect_tls(addr2).await;
// A third connection (racing against listener 1) must not complete its
// handshake: the accept loop parks it on the shared semaphore before
// TLS ever starts.
let (third_tx, third_rx) = tokio::sync::oneshot::channel();
let third_task = tokio::spawn(async move {
let stream = connect_tls(addr1).await;
let _ = third_tx.send(stream);
});
tokio::time::sleep(Duration::from_millis(500)).await;
assert!(
third_rx.is_empty(),
"third connection must be capped by the shared semaphore"
);
// Release one slot; the parked third connection must now get through.
drop(tls2);
let mut tls3 = tokio::time::timeout(Duration::from_secs(5), third_rx)
.await
.expect("third connection should complete after a permit is released")
.expect("third connection task should not panic");
tls3.write_all(b"GET / HTTP/1.1\r\nHost: test.local\r\nConnection: close\r\n\r\n")
.await
.unwrap();
let mut buf = vec![0u8; 4096];
let n = tokio::time::timeout(Duration::from_secs(5), tls3.read(&mut buf))
.await
.expect("timeout waiting for response")
.expect("read error");
let response = String::from_utf8_lossy(&buf[..n]);
assert!(
response.starts_with("HTTP/1.1 200 OK"),
"expected a real 200 round-trip on the capped connection, got: {response}"
);
third_task.abort();
drop(tls1);
drop(tls3);
let _ = upstream.shutdown_tx.send(());
}
} }