docker/pgcat
1
0
Fork 0
mirror of https://github.com/postgresml/pgcat.git synced 2026-03-27 18:56:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Don't unwrap stream.peer_addr() (#562)

Browse Source
This commit is contained in:
Zain Kabani
2023-08-25 13:33:39 -04:00
committed by GitHub
parent be549f3faa
commit ffe820497f
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/client.rs
View File

@@ -348,7 +348,15 @@ pub async fn startup_tls(
) -> Result<Client<ReadHalf<TlsStream<TcpStream>>, WriteHalf<TlsStream<TcpStream>>>, Error> { ) -> Result<Client<ReadHalf<TlsStream<TcpStream>>, WriteHalf<TlsStream<TcpStream>>>, Error> {
// Negotiate TLS. // Negotiate TLS.
let tls = Tls::new()?; let tls = Tls::new()?;
let addr = stream.peer_addr().unwrap(); let addr = match stream.peer_addr() {
Ok(addr) => addr,
Err(err) => {
return Err(Error::SocketError(format!(
"Failed to get peer address: {:?}",
err
)));
}
};
let mut stream = match tls.acceptor.accept(stream).await { let mut stream = match tls.acceptor.accept(stream).await {
Ok(stream) => stream, Ok(stream) => stream,