tuxmain/rust-rpxy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

workaround for failure to bind tcp/quic listeners

Browse source
This commit is contained in:
Jun Kurihara 2022-07-06 21:16:46 +09:00
commit 1c4e450f1d
No known key found for this signature in database
GPG key ID: 48ADFD173ED22B03
4 changed files with 51 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/proxy/proxy_main.rs
View file

@ -1,6 +1,6 @@
// use super::proxy_handler::handle_request;
use super::Backends;
use crate::{error::*, globals::Globals, log::*};
use crate::{constants::GET_LISTENER_RETRY_TIMEOUT_SEC, error::*, globals::Globals, log::*};
use hyper::{
client::connect::Connect, server::conn::Http, service::service_fn, Body, Client, Request,
};
@ -77,9 +77,27 @@ where
});
}
// Work around to forcibly get tcp listener for "address already in use"
async fn try_bind_tcp_listener(&self) -> Result<TcpListener> {
let fut = async {
loop {
if let Ok(listener) = TcpListener::bind(&self.listening_on).await {
break listener;
}
}
};
tokio::time::timeout(
tokio::time::Duration::from_secs(GET_LISTENER_RETRY_TIMEOUT_SEC),
fut,
)
.await
.map_err(|_e| anyhow!("Failed to get listener"))
}
async fn start_without_tls(self, server: Http<LocalExecutor>) -> Result<()> {
let listener_service = async {
let tcp_listener = TcpListener::bind(&self.listening_on).await?;
// let tcp_listener = TcpListener::bind(&self.listening_on).await?;
let tcp_listener = self.try_bind_tcp_listener().await?;
info!(
"Start TCP proxy serving with HTTP request for configured host names: {:?}",
tcp_listener.local_addr()?