hex/wstunnel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(server): Add tcp-keepalive on in-coming server connections

Browse source
This commit is contained in:
Σrebe - Romain GERARD 2024-05-01 15:07:45 +02:00
parent 421a5a230c
commit 69c372490a
No known key found for this signature in database
GPG key ID: 7A42B4B97E0332F4
2 changed files with 11 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/tunnel/server.rs
View file

@ -7,6 +7,7 @@ use http_body_util::{BodyStream, Either, StreamBody};
use std::cmp::min;
use std::fmt::Debug;
use std::future::Future;
use std::net::{IpAddr, SocketAddr};
use std::ops::Deref;
use std::pin::Pin;
@ -25,6 +26,7 @@ use hyper_util::rt::TokioExecutor;
use jsonwebtoken::TokenData;
use once_cell::sync::Lazy;
use parking_lot::Mutex;
use socket2::SockRef;
use crate::restrictions::config_reloader::RestrictionsRulesReloader;
use crate::restrictions::types::{
@ -727,8 +729,8 @@ pub async fn run_server(server_config: Arc<WsServerConfig>, restrictions: Restri
// Bind server and run forever to serve incoming connections.
let mut restrictions = RestrictionsRulesReloader::new(restrictions, server_config.restriction_config.clone())?;
let listener = TcpListener::bind(&server_config.bind).await?;
let mut await_config_reload = Box::pin(restrictions.reload_notifier());
let listener = TcpListener::bind(&server_config.bind).await?;
loop {
let cnx = select! {
@ -751,7 +753,10 @@ pub async fn run_server(server_config: Arc<WsServerConfig>, restrictions: Restri
continue;
}
};
let _ = stream.set_nodelay(true);
if let Err(err) = tcp::configure_socket(SockRef::from(&stream), &None) {
warn!("Error while configuring server socket {:?}", err);
}
let span = span!(
Level::INFO,