Remove Sync bound from TlsConnector io arg

This commit is contained in:
Martin Algesten
2022-05-01 09:33:43 +02:00
parent 049b5a5acd
commit 4675d748e9
6 changed files with 31 additions and 13 deletions

View File

@@ -34,7 +34,7 @@ impl TlsConnector for PassThrough {
fn connect( fn connect(
&self, &self,
_dns_name: &str, _dns_name: &str,
io: Box<dyn ReadWrite + Sync>, io: Box<dyn ReadWrite>,
) -> Result<Box<dyn ReadWrite>, Error> { ) -> Result<Box<dyn ReadWrite>, Error> {
if self.handshake_fail { if self.handshake_fail {
let io_err = io::Error::new(io::ErrorKind::InvalidData, PassThroughError); let io_err = io::Error::new(io::ErrorKind::InvalidData, PassThroughError);

View File

@@ -56,10 +56,11 @@ impl TlsConnector for MbedTlsConnector {
fn connect( fn connect(
&self, &self,
_dns_name: &str, _dns_name: &str,
io: Box<dyn ReadWrite + Sync>, io: Box<dyn ReadWrite>,
) -> Result<Box<dyn ReadWrite>, Error> { ) -> Result<Box<dyn ReadWrite>, Error> {
let mut ctx = self.context.lock().unwrap(); let mut ctx = self.context.lock().unwrap();
match ctx.establish(io, None) { let sync = SyncIo(Mutex::new(io));
match ctx.establish(sync, None) {
Err(_) => { Err(_) => {
let io_err = io::Error::new(io::ErrorKind::InvalidData, MbedTlsError); let io_err = io::Error::new(io::ErrorKind::InvalidData, MbedTlsError);
return Err(io_err.into()); return Err(io_err.into());
@@ -69,6 +70,27 @@ impl TlsConnector for MbedTlsConnector {
} }
} }
struct SyncIo(Mutex<Box<dyn ReadWrite>>);
impl io::Read for SyncIo {
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
let mut lock = self.0.lock().unwrap();
lock.read(buf)
}
}
impl io::Write for SyncIo {
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
let mut lock = self.0.lock().unwrap();
lock.write(buf)
}
fn flush(&mut self) -> io::Result<()> {
let mut lock = self.0.lock().unwrap();
lock.flush()
}
}
struct MbedTlsStream { struct MbedTlsStream {
context: Arc<Mutex<Context>>, //tcp_stream: TcpStream, context: Arc<Mutex<Context>>, //tcp_stream: TcpStream,
} }

View File

@@ -364,7 +364,7 @@ pub(crate) fn default_tls_config() -> std::sync::Arc<dyn TlsConnector> {
fn connect( fn connect(
&self, &self,
_dns_name: &str, _dns_name: &str,
_io: Box<dyn ReadWrite + Sync>, _io: Box<dyn ReadWrite>,
) -> Result<Box<dyn ReadWrite>, crate::error::Error> { ) -> Result<Box<dyn ReadWrite>, crate::error::Error> {
Err(ErrorKind::UnknownScheme Err(ErrorKind::UnknownScheme
.msg("cannot make HTTPS request because no TLS backend is configured")) .msg("cannot make HTTPS request because no TLS backend is configured"))

View File

@@ -11,11 +11,7 @@ pub(crate) fn default_tls_config() -> std::sync::Arc<dyn TlsConnector> {
} }
impl TlsConnector for native_tls::TlsConnector { impl TlsConnector for native_tls::TlsConnector {
fn connect( fn connect(&self, dns_name: &str, io: Box<dyn ReadWrite>) -> Result<Box<dyn ReadWrite>, Error> {
&self,
dns_name: &str,
io: Box<dyn ReadWrite + Sync>,
) -> Result<Box<dyn ReadWrite>, Error> {
let stream = let stream =
native_tls::TlsConnector::connect(self, dns_name, io).map_err(|e| match e { native_tls::TlsConnector::connect(self, dns_name, io).map_err(|e| match e {
native_tls::HandshakeError::Failure(e) => ErrorKind::ConnectionFailed native_tls::HandshakeError::Failure(e) => ErrorKind::ConnectionFailed
@@ -31,7 +27,7 @@ impl TlsConnector for native_tls::TlsConnector {
} }
#[cfg(feature = "native-tls")] #[cfg(feature = "native-tls")]
impl ReadWrite for native_tls::TlsStream<Box<dyn ReadWrite + Sync>> { impl ReadWrite for native_tls::TlsStream<Box<dyn ReadWrite>> {
fn socket(&self) -> Option<&TcpStream> { fn socket(&self) -> Option<&TcpStream> {
self.get_ref().socket() self.get_ref().socket()
} }

View File

@@ -27,7 +27,7 @@ fn is_close_notify(e: &std::io::Error) -> bool {
false false
} }
struct RustlsStream(rustls::StreamOwned<rustls::ClientConnection, Box<dyn ReadWrite + Sync>>); struct RustlsStream(rustls::StreamOwned<rustls::ClientConnection, Box<dyn ReadWrite>>);
impl ReadWrite for RustlsStream { impl ReadWrite for RustlsStream {
fn socket(&self) -> Option<&TcpStream> { fn socket(&self) -> Option<&TcpStream> {
@@ -97,7 +97,7 @@ impl TlsConnector for Arc<rustls::ClientConfig> {
fn connect( fn connect(
&self, &self,
dns_name: &str, dns_name: &str,
mut io: Box<dyn ReadWrite + Sync>, mut io: Box<dyn ReadWrite>,
) -> Result<Box<dyn ReadWrite>, Error> { ) -> Result<Box<dyn ReadWrite>, Error> {
let sni = rustls::ServerName::try_from(dns_name) let sni = rustls::ServerName::try_from(dns_name)
.map_err(|e| ErrorKind::Dns.msg(format!("parsing '{}'", dns_name)).src(e))?; .map_err(|e| ErrorKind::Dns.msg(format!("parsing '{}'", dns_name)).src(e))?;

View File

@@ -42,7 +42,7 @@ pub trait TlsConnector: Send + Sync {
fn connect( fn connect(
&self, &self,
dns_name: &str, dns_name: &str,
io: Box<dyn ReadWrite + Sync>, io: Box<dyn ReadWrite>,
) -> Result<Box<dyn ReadWrite>, crate::error::Error>; ) -> Result<Box<dyn ReadWrite>, crate::error::Error>;
} }