masalachai
/
epp-client
mirror of https://github.com/masalachai/epp-client.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Return error instead of panicking on unexpected EOF

This commit is contained in:
Dirkjan Ochtman 2021-12-01 13:34:21 +01:00 committed by masalachai
parent 48ec1e03ce
commit d8bcf690a4
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
epp-client/src/connection/registry.rs
View File

@ -4,7 +4,7 @@ use rustls::{OwnedTrustAnchor, RootCertStore};
use std::convert::TryInto; use std::convert::TryInto;
use std::sync::Arc; use std::sync::Arc;
use std::{error::Error, io as stdio, net::ToSocketAddrs}; use std::{error::Error, io as stdio, net::ToSocketAddrs};
use std::{str, u32}; use std::{io, str, u32};
use tokio::{io::AsyncReadExt, io::AsyncWriteExt, net::TcpStream}; use tokio::{io::AsyncReadExt, io::AsyncWriteExt, net::TcpStream};
use tokio_rustls::{client::TlsStream, rustls::ClientConfig, TlsConnector}; use tokio_rustls::{client::TlsStream, rustls::ClientConfig, TlsConnector};
@ -76,7 +76,11 @@ impl EppConnection {
debug!("{}: Total read: {} bytes", self.registry, read_size); debug!("{}: Total read: {} bytes", self.registry, read_size);
if read == 0 { if read == 0 {
panic!("{}: Unexpected eof", self.registry) return Err(io::Error::new(
io::ErrorKind::UnexpectedEof,
format!("{}: unexpected eof", self.registry),
)
.into());
} else if read_size >= message_size { } else if read_size >= message_size {
break; break;
} }