From ec9ade938e9e4aa710f4351e48a8fda1037352aa Mon Sep 17 00:00:00 2001 From: Alex Crichton Date: Thu, 24 Apr 2014 18:48:21 -0700 Subject: std: Add close_{read,write}() methods to I/O Two new methods were added to TcpStream and UnixStream: fn close_read(&mut self) -> IoResult<()>; fn close_write(&mut self) -> IoResult<()>; These two methods map to shutdown()'s behavior (the system call on unix), closing the reading or writing half of a duplex stream. These methods are primarily added to allow waking up a pending read in another task. By closing the reading half of a connection, all pending readers will be woken up and will return with EndOfFile. The close_write() method was added for symmetry with close_read(), and I imagine that it will be quite useful at some point. Implementation-wise, librustuv got the short end of the stick this time. The native versions just delegate to the shutdown() syscall (easy). The uv versions can leverage uv_shutdown() for tcp/unix streams, but only for closing the writing half. Closing the reading half is done through some careful dancing to wake up a pending reader. As usual, windows likes to be different from unix. The windows implementation uses shutdown() for sockets, but shutdown() is not available for named pipes. Instead, CancelIoEx was used with same fancy synchronization to make sure everyone knows what's up. cc #11165 --- src/libstd/rt/rtio.rs | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'src/libstd/rt') diff --git a/src/libstd/rt/rtio.rs b/src/libstd/rt/rtio.rs index fe9f4932a2a..c5afe7887ad 100644 --- a/src/libstd/rt/rtio.rs +++ b/src/libstd/rt/rtio.rs @@ -221,6 +221,7 @@ pub trait RtioTcpStream : RtioSocket { fn letdie(&mut self) -> IoResult<()>; fn clone(&self) -> Box; fn close_write(&mut self) -> IoResult<()>; + fn close_read(&mut self) -> IoResult<()>; } pub trait RtioSocket { @@ -274,6 +275,9 @@ pub trait RtioPipe { fn read(&mut self, buf: &mut [u8]) -> IoResult; fn write(&mut self, buf: &[u8]) -> IoResult<()>; fn clone(&self) -> Box; + + fn close_write(&mut self) -> IoResult<()>; + fn close_read(&mut self) -> IoResult<()>; } pub trait RtioUnixListener { -- cgit 1.4.1-3-g733a5