use shutdown error values to throw exceptions in close stream
This commit is contained in:
@@ -160,20 +160,6 @@ impl StreamInstance {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Drop for StreamInstance {
|
|
||||||
fn drop(&mut self) {
|
|
||||||
match self {
|
|
||||||
StreamInstance::TcpStream(_, ref mut tcp_stream) => {
|
|
||||||
tcp_stream.shutdown(Shutdown::Both).unwrap_or(())
|
|
||||||
}
|
|
||||||
StreamInstance::TlsStream(_, ref mut tls_stream) => {
|
|
||||||
tls_stream.shutdown().unwrap_or(());
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl fmt::Debug for StreamInstance {
|
impl fmt::Debug for StreamInstance {
|
||||||
fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
|
fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
|
||||||
match self {
|
match self {
|
||||||
@@ -596,8 +582,18 @@ impl Stream {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
pub(crate) fn close(&mut self) {
|
pub(crate) fn close(&mut self) -> Result<(), std::io::Error> {
|
||||||
|
let result = match self.stream_inst.0.borrow_mut().stream_inst {
|
||||||
|
StreamInstance::TcpStream(_, ref mut tcp_stream) => {
|
||||||
|
tcp_stream.shutdown(Shutdown::Both)
|
||||||
|
},
|
||||||
|
StreamInstance::TlsStream(_, ref mut tls_stream) => {
|
||||||
|
tls_stream.shutdown()
|
||||||
|
}
|
||||||
|
_ => Ok(())
|
||||||
|
};
|
||||||
self.stream_inst.0.borrow_mut().stream_inst = StreamInstance::Null;
|
self.stream_inst.0.borrow_mut().stream_inst = StreamInstance::Null;
|
||||||
|
result
|
||||||
}
|
}
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
@@ -636,20 +632,6 @@ impl Stream {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn is_closed(&self) -> bool {
|
|
||||||
match self.stream_inst.0.borrow_mut().stream_inst {
|
|
||||||
StreamInstance::Null => true,
|
|
||||||
StreamInstance::TcpStream(_, ref mut tcp_stream) => {
|
|
||||||
let mut buf = [0;8];
|
|
||||||
match tcp_stream.peek(&mut buf) {
|
|
||||||
Ok(n_bytes) => n_bytes == 0,
|
|
||||||
Err(_) => true
|
|
||||||
}
|
|
||||||
},
|
|
||||||
_ => false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn unpause_stream(&mut self) {
|
fn unpause_stream(&mut self) {
|
||||||
let stream_inst = match self.stream_inst.0.borrow_mut().stream_inst {
|
let stream_inst = match self.stream_inst.0.borrow_mut().stream_inst {
|
||||||
StreamInstance::PausedPrologStream(ref put_back, ref mut stream_inst)
|
StreamInstance::PausedPrologStream(ref put_back, ref mut stream_inst)
|
||||||
|
|||||||
@@ -2704,7 +2704,12 @@ impl MachineState {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if !stream.is_stdin() && !stream.is_stdout() && !stream.is_stderr() {
|
if !stream.is_stdin() && !stream.is_stdout() && !stream.is_stderr() {
|
||||||
if stream.is_closed() {
|
let close_result = stream.close();
|
||||||
|
|
||||||
|
if let Some(ref alias) = stream.options().alias {
|
||||||
|
indices.stream_aliases.remove(alias);
|
||||||
|
}
|
||||||
|
if let Err(_) = close_result {
|
||||||
let stub = MachineError::functor_stub(
|
let stub = MachineError::functor_stub(
|
||||||
clause_name!("close"),
|
clause_name!("close"),
|
||||||
1,
|
1,
|
||||||
@@ -2721,12 +2726,6 @@ impl MachineState {
|
|||||||
),
|
),
|
||||||
stub,
|
stub,
|
||||||
));
|
));
|
||||||
} else {
|
|
||||||
stream.close();
|
|
||||||
|
|
||||||
if let Some(ref alias) = stream.options().alias {
|
|
||||||
indices.stream_aliases.remove(alias);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user