|
|
|
@ -10,7 +10,7 @@ use std::{
|
|
|
|
|
collections::{BTreeMap, HashMap},
|
|
|
|
|
fs,
|
|
|
|
|
future::Future,
|
|
|
|
|
net::IpAddr,
|
|
|
|
|
net::{IpAddr, SocketAddr},
|
|
|
|
|
path::PathBuf,
|
|
|
|
|
sync::{Arc, Mutex, RwLock},
|
|
|
|
|
time::{Duration, Instant},
|
|
|
|
@ -39,6 +39,8 @@ pub struct Globals {
|
|
|
|
|
keypair: Arc<ruma::signatures::Ed25519KeyPair>,
|
|
|
|
|
dns_resolver: TokioAsyncResolver,
|
|
|
|
|
jwt_decoding_key: Option<jsonwebtoken::DecodingKey<'static>>,
|
|
|
|
|
federation_client: reqwest::Client,
|
|
|
|
|
default_client: reqwest::Client,
|
|
|
|
|
pub(super) server_signingkeys: Arc<dyn Tree>,
|
|
|
|
|
pub bad_event_ratelimiter: Arc<RwLock<HashMap<Box<EventId>, RateLimitState>>>,
|
|
|
|
|
pub bad_signature_ratelimiter: Arc<RwLock<HashMap<Vec<String>, RateLimitState>>>,
|
|
|
|
@ -132,6 +134,17 @@ impl Globals {
|
|
|
|
|
.as_ref()
|
|
|
|
|
.map(|secret| jsonwebtoken::DecodingKey::from_secret(secret.as_bytes()).into_static());
|
|
|
|
|
|
|
|
|
|
let default_client = reqwest_client_builder(&config)?.build()?;
|
|
|
|
|
let name_override = Arc::clone(&tls_name_override);
|
|
|
|
|
let federation_client = reqwest_client_builder(&config)?
|
|
|
|
|
.resolve_fn(move |domain| {
|
|
|
|
|
let read_guard = name_override.read().unwrap();
|
|
|
|
|
let (override_name, port) = read_guard.get(&domain)?;
|
|
|
|
|
let first_name = override_name.get(0)?;
|
|
|
|
|
Some(SocketAddr::new(*first_name, *port))
|
|
|
|
|
})
|
|
|
|
|
.build()?;
|
|
|
|
|
|
|
|
|
|
let s = Self {
|
|
|
|
|
globals,
|
|
|
|
|
config,
|
|
|
|
@ -141,6 +154,8 @@ impl Globals {
|
|
|
|
|
})?,
|
|
|
|
|
actual_destination_cache: Arc::new(RwLock::new(WellKnownMap::new())),
|
|
|
|
|
tls_name_override,
|
|
|
|
|
federation_client,
|
|
|
|
|
default_client,
|
|
|
|
|
server_signingkeys,
|
|
|
|
|
jwt_decoding_key,
|
|
|
|
|
bad_event_ratelimiter: Arc::new(RwLock::new(HashMap::new())),
|
|
|
|
@ -163,17 +178,16 @@ impl Globals {
|
|
|
|
|
&self.keypair
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Returns a reqwest client which can be used to send requests.
|
|
|
|
|
pub fn reqwest_client(&self) -> Result<reqwest::ClientBuilder> {
|
|
|
|
|
let mut reqwest_client_builder = reqwest::Client::builder()
|
|
|
|
|
.connect_timeout(Duration::from_secs(30))
|
|
|
|
|
.timeout(Duration::from_secs(60 * 3))
|
|
|
|
|
.pool_max_idle_per_host(1);
|
|
|
|
|
if let Some(proxy) = self.config.proxy.to_proxy()? {
|
|
|
|
|
reqwest_client_builder = reqwest_client_builder.proxy(proxy);
|
|
|
|
|
}
|
|
|
|
|
/// Returns a reqwest client which can be used to send requests
|
|
|
|
|
pub fn default_client(&self) -> reqwest::Client {
|
|
|
|
|
// Client is cheap to clone (Arc wrapper) and avoids lifetime issues
|
|
|
|
|
self.default_client.clone()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Ok(reqwest_client_builder)
|
|
|
|
|
/// Returns a client used for resolving .well-knowns
|
|
|
|
|
pub fn federation_client(&self) -> reqwest::Client {
|
|
|
|
|
// Client is cheap to clone (Arc wrapper) and avoids lifetime issues
|
|
|
|
|
self.federation_client.clone()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[tracing::instrument(skip(self))]
|
|
|
|
@ -340,3 +354,15 @@ impl Globals {
|
|
|
|
|
r
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn reqwest_client_builder(config: &Config) -> Result<reqwest::ClientBuilder> {
|
|
|
|
|
let mut reqwest_client_builder = reqwest::Client::builder()
|
|
|
|
|
.connect_timeout(Duration::from_secs(30))
|
|
|
|
|
.timeout(Duration::from_secs(60 * 3));
|
|
|
|
|
|
|
|
|
|
if let Some(proxy) = config.proxy.to_proxy()? {
|
|
|
|
|
reqwest_client_builder = reqwest_client_builder.proxy(proxy);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Ok(reqwest_client_builder)
|
|
|
|
|
}
|
|
|
|
|