aleym_core 0.1.0-alpha.1

Extensible news aggregation and knowledge-base engine (Core Library Component of Aleym)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

use std::pin::Pin;
use std::sync::Arc;

use super::{AsyncStream, tcp};
use crate::net::NetworkError;

pub enum AlpnProtocols {
	#[cfg(feature = "net_protocol_http2")]
	H2,
	Fallback,
}

pub trait Transport: super::tcp::Transport {
	fn tls_config(&self) -> Arc<tokio_rustls::rustls::ClientConfig>;

	async fn connect(
		&self,
		host: String,
		port: u16,
	) -> Result<(tokio_rustls::TlsStream<Pin<Box<dyn AsyncStream>>>, AlpnProtocols), NetworkError> {
		let tcp_stream = tcp::Transport::connect(self, &host, port).await?;

		let tls_stream = tokio_rustls::TlsConnector::from(Arc::clone(&self.tls_config()))
			.connect(tokio_rustls::rustls::pki_types::ServerName::try_from(host)?, tcp_stream)
			.await?;

		let alpn_protocol = {
			let (_, tls_client_connection) = tls_stream.get_ref();

			match tls_client_connection.alpn_protocol() {
				#[cfg(feature = "net_protocol_http2")]
				Some(b"h2") => AlpnProtocols::H2,
				_ => AlpnProtocols::Fallback,
			}
		};

		Ok((tokio_rustls::TlsStream::Client(tls_stream), alpn_protocol))
	}
}

#[cfg(any(feature = "net_protocol_http1", feature = "net_protocol_http2"))]
impl<T: Transport> crate::net::protocols::http::Https for T {}