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

27
src/proxy/proxy_tls.rs
View file

@ -1,5 +1,5 @@
use super::proxy_main::{LocalExecutor, Proxy};
use crate::{constants::CERTS_WATCH_DELAY_SECS, error::*, log::*};
use crate::{constants::*, error::*, log::*};
#[cfg(feature = "h3")]
use futures::StreamExt;
use futures::{future::FutureExt, join, select};
@ -99,8 +99,8 @@ where
let server_crypto = backend_serve.get_tls_server_config().unwrap();
let server_config_h3 = quinn::ServerConfig::with_crypto(Arc::new(server_crypto));
let (endpoint, incoming) =
quinn::Endpoint::server(server_config_h3, self.listening_on).unwrap();
let (endpoint, incoming) = self.try_bind_quic_listener(server_config_h3).await?;
// quinn::Endpoint::server(server_config_h3, self.listening_on).unwrap();
info!(
"Start UDP proxy serving with HTTP/3 request for configured host names: {:?}",
endpoint.local_addr()?
@ -181,6 +181,27 @@ where
}
}
// Work around to forcibly get quic listener for "address already in use"
#[cfg(feature = "h3")]
async fn try_bind_quic_listener(
&self,
server_config: quinn::ServerConfig,
) -> Result<(quinn::Endpoint, quinn::Incoming)> {
let fut = async {
loop {
if let Ok(listener) = quinn::Endpoint::server(server_config.clone(), self.listening_on) {
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"))
}
fn fetch_server_crypto(&self, server_name: &str) -> Option<ServerConfig> {
let backend_serve = if let Some(backend_serve) = self.backends.apps.get(server_name) {
backend_serve