Skip to content

Commit 7c05356

Browse files
bors[bot]yshui
bors[bot]
and
yshui
authored
Merge #148
148: Make sure into_raw_fd doesn't close the file descriptor r=stjepang a=yshui Closes #147 Co-authored-by: Yuxuan Shui <[email protected]> Co-authored-by: yshui <[email protected]>
2 parents 9bf06ce + 8e2bf24 commit 7c05356

File tree

8 files changed

+54
-55
lines changed

8 files changed

+54
-55
lines changed

‎src/net/driver/mod.rs

+25-12
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,7 @@ pub struct IoHandle<T: Evented> {
183183
entry: Arc<Entry>,
184184

185185
/// The I/O event source.
186-
source: T,
186+
source: Option<T>,
187187
}
188188

189189
impl<T: Evented> IoHandle<T> {
@@ -196,13 +196,13 @@ impl<T: Evented> IoHandle<T> {
196196
entry: REACTOR
197197
.register(&source)
198198
.expect("cannot register an I/O event source"),
199-
source,
199+
source: Some(source),
200200
}
201201
}
202202

203203
/// Returns a reference to the inner I/O event source.
204204
pub fn get_ref(&self) -> &T {
205-
&self.source
205+
self.source.as_ref().unwrap()
206206
}
207207

208208
/// Polls the I/O handle for reading.
@@ -286,13 +286,26 @@ impl<T: Evented> IoHandle<T> {
286286

287287
Ok(())
288288
}
289+
290+
/// Deregisters and returns the inner I/O source.
291+
///
292+
/// This method is typically used to convert `IoHandle`s to raw file descriptors/handles.
293+
pub fn into_inner(mut self) -> T {
294+
let source = self.source.take().unwrap();
295+
REACTOR
296+
.deregister(&source, &self.entry)
297+
.expect("cannot deregister I/O event source");
298+
source
299+
}
289300
}
290301

291302
impl<T: Evented> Drop for IoHandle<T> {
292303
fn drop(&mut self) {
293-
REACTOR
294-
.deregister(&self.source, &self.entry)
295-
.expect("cannot deregister I/O event source");
304+
if let Some(ref source) = self.source {
305+
REACTOR
306+
.deregister(source, &self.entry)
307+
.expect("cannot deregister I/O event source");
308+
}
296309
}
297310
}
298311

@@ -313,7 +326,7 @@ impl<T: Evented + std::io::Read + Unpin> AsyncRead for IoHandle<T> {
313326
) -> Poll<io::Result<usize>> {
314327
futures_core::ready!(Pin::new(&mut *self).poll_readable(cx)?);
315328

316-
match self.source.read(buf) {
329+
match self.source.as_mut().unwrap().read(buf) {
317330
Err(ref err) if err.kind() == io::ErrorKind::WouldBlock => {
318331
self.clear_readable(cx)?;
319332
Poll::Pending
@@ -334,7 +347,7 @@ where
334347
) -> Poll<io::Result<usize>> {
335348
futures_core::ready!(Pin::new(&mut *self).poll_readable(cx)?);
336349

337-
match (&self.source).read(buf) {
350+
match self.source.as_ref().unwrap().read(buf) {
338351
Err(ref err) if err.kind() == io::ErrorKind::WouldBlock => {
339352
self.clear_readable(cx)?;
340353
Poll::Pending
@@ -352,7 +365,7 @@ impl<T: Evented + std::io::Write + Unpin> AsyncWrite for IoHandle<T> {
352365
) -> Poll<io::Result<usize>> {
353366
futures_core::ready!(self.poll_writable(cx)?);
354367

355-
match self.source.write(buf) {
368+
match self.source.as_mut().unwrap().write(buf) {
356369
Err(ref err) if err.kind() == io::ErrorKind::WouldBlock => {
357370
self.clear_writable(cx)?;
358371
Poll::Pending
@@ -364,7 +377,7 @@ impl<T: Evented + std::io::Write + Unpin> AsyncWrite for IoHandle<T> {
364377
fn poll_flush(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<()>> {
365378
futures_core::ready!(self.poll_writable(cx)?);
366379

367-
match self.source.flush() {
380+
match self.source.as_mut().unwrap().flush() {
368381
Err(ref err) if err.kind() == io::ErrorKind::WouldBlock => {
369382
self.clear_writable(cx)?;
370383
Poll::Pending
@@ -389,7 +402,7 @@ where
389402
) -> Poll<io::Result<usize>> {
390403
futures_core::ready!(self.poll_writable(cx)?);
391404

392-
match (&self.source).write(buf) {
405+
match self.get_ref().write(buf) {
393406
Err(ref err) if err.kind() == io::ErrorKind::WouldBlock => {
394407
self.clear_writable(cx)?;
395408
Poll::Pending
@@ -401,7 +414,7 @@ where
401414
fn poll_flush(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<()>> {
402415
futures_core::ready!(self.poll_writable(cx)?);
403416

404-
match (&self.source).flush() {
417+
match self.get_ref().flush() {
405418
Err(ref err) if err.kind() == io::ErrorKind::WouldBlock => {
406419
self.clear_writable(cx)?;
407420
Poll::Pending

‎src/net/tcp/listener.rs

+2-8
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,6 @@ use crate::task::{Context, Poll};
5050
#[derive(Debug)]
5151
pub struct TcpListener {
5252
io_handle: IoHandle<mio::net::TcpListener>,
53-
54-
#[cfg(unix)]
55-
raw_fd: std::os::unix::io::RawFd,
5653
// #[cfg(windows)]
5754
// raw_socket: std::os::windows::io::RawSocket,
5855
}
@@ -87,7 +84,6 @@ impl TcpListener {
8784
Ok(mio_listener) => {
8885
#[cfg(unix)]
8986
let listener = TcpListener {
90-
raw_fd: mio_listener.as_raw_fd(),
9187
io_handle: IoHandle::new(mio_listener),
9288
};
9389

@@ -136,7 +132,6 @@ impl TcpListener {
136132

137133
#[cfg(unix)]
138134
let stream = TcpStream {
139-
raw_fd: mio_stream.as_raw_fd(),
140135
io_handle: IoHandle::new(mio_stream),
141136
};
142137

@@ -243,7 +238,6 @@ impl From<std::net::TcpListener> for TcpListener {
243238

244239
#[cfg(unix)]
245240
let listener = TcpListener {
246-
raw_fd: mio_listener.as_raw_fd(),
247241
io_handle: IoHandle::new(mio_listener),
248242
};
249243

@@ -273,7 +267,7 @@ cfg_if! {
273267
if #[cfg(any(unix, feature = "docs"))] {
274268
impl AsRawFd for TcpListener {
275269
fn as_raw_fd(&self) -> RawFd {
276-
self.raw_fd
270+
self.io_handle.get_ref().as_raw_fd()
277271
}
278272
}
279273

@@ -285,7 +279,7 @@ cfg_if! {
285279

286280
impl IntoRawFd for TcpListener {
287281
fn into_raw_fd(self) -> RawFd {
288-
self.raw_fd
282+
self.io_handle.into_inner().into_raw_fd()
289283
}
290284
}
291285
}

‎src/net/tcp/stream.rs

+2-7
Original file line numberDiff line numberDiff line change
@@ -51,9 +51,6 @@ use crate::task::{Context, Poll};
5151
#[derive(Debug)]
5252
pub struct TcpStream {
5353
pub(super) io_handle: IoHandle<mio::net::TcpStream>,
54-
55-
#[cfg(unix)]
56-
pub(super) raw_fd: std::os::unix::io::RawFd,
5754
// #[cfg(windows)]
5855
// pub(super) raw_socket: std::os::windows::io::RawSocket,
5956
}
@@ -103,7 +100,6 @@ impl TcpStream {
103100
let stream = mio::net::TcpStream::connect(&addr).map(|mio_stream| {
104101
#[cfg(unix)]
105102
let stream = TcpStream {
106-
raw_fd: mio_stream.as_raw_fd(),
107103
io_handle: IoHandle::new(mio_stream),
108104
};
109105

@@ -445,7 +441,6 @@ impl From<std::net::TcpStream> for TcpStream {
445441

446442
#[cfg(unix)]
447443
let stream = TcpStream {
448-
raw_fd: mio_stream.as_raw_fd(),
449444
io_handle: IoHandle::new(mio_stream),
450445
};
451446

@@ -475,7 +470,7 @@ cfg_if! {
475470
if #[cfg(any(unix, feature = "docs"))] {
476471
impl AsRawFd for TcpStream {
477472
fn as_raw_fd(&self) -> RawFd {
478-
self.raw_fd
473+
self.io_handle.get_ref().as_raw_fd()
479474
}
480475
}
481476

@@ -487,7 +482,7 @@ cfg_if! {
487482

488483
impl IntoRawFd for TcpStream {
489484
fn into_raw_fd(self) -> RawFd {
490-
self.raw_fd
485+
self.io_handle.into_inner().into_raw_fd()
491486
}
492487
}
493488
}

‎src/net/udp/mod.rs

+2-7
Original file line numberDiff line numberDiff line change
@@ -48,9 +48,6 @@ use crate::task::Poll;
4848
#[derive(Debug)]
4949
pub struct UdpSocket {
5050
io_handle: IoHandle<mio::net::UdpSocket>,
51-
52-
#[cfg(unix)]
53-
raw_fd: std::os::unix::io::RawFd,
5451
// #[cfg(windows)]
5552
// raw_socket: std::os::windows::io::RawSocket,
5653
}
@@ -82,7 +79,6 @@ impl UdpSocket {
8279
Ok(mio_socket) => {
8380
#[cfg(unix)]
8481
let socket = UdpSocket {
85-
raw_fd: mio_socket.as_raw_fd(),
8682
io_handle: IoHandle::new(mio_socket),
8783
};
8884

@@ -515,7 +511,6 @@ impl From<std::net::UdpSocket> for UdpSocket {
515511

516512
#[cfg(unix)]
517513
let socket = UdpSocket {
518-
raw_fd: mio_socket.as_raw_fd(),
519514
io_handle: IoHandle::new(mio_socket),
520515
};
521516

@@ -545,7 +540,7 @@ cfg_if! {
545540
if #[cfg(any(unix, feature = "docs"))] {
546541
impl AsRawFd for UdpSocket {
547542
fn as_raw_fd(&self) -> RawFd {
548-
self.raw_fd
543+
self.io_handle.get_ref().as_raw_fd()
549544
}
550545
}
551546

@@ -557,7 +552,7 @@ cfg_if! {
557552

558553
impl IntoRawFd for UdpSocket {
559554
fn into_raw_fd(self) -> RawFd {
560-
self.raw_fd
555+
self.io_handle.into_inner().into_raw_fd()
561556
}
562557
}
563558
}

‎src/os/unix/net/datagram.rs

+2-6
Original file line numberDiff line numberDiff line change
@@ -44,15 +44,12 @@ use crate::task::{blocking, Poll};
4444
pub struct UnixDatagram {
4545
#[cfg(not(feature = "docs"))]
4646
io_handle: IoHandle<mio_uds::UnixDatagram>,
47-
48-
raw_fd: RawFd,
4947
}
5048

5149
impl UnixDatagram {
5250
#[cfg(not(feature = "docs"))]
5351
fn new(socket: mio_uds::UnixDatagram) -> UnixDatagram {
5452
UnixDatagram {
55-
raw_fd: socket.as_raw_fd(),
5653
io_handle: IoHandle::new(socket),
5754
}
5855
}
@@ -362,15 +359,14 @@ impl From<std::os::unix::net::UnixDatagram> for UnixDatagram {
362359
fn from(datagram: std::os::unix::net::UnixDatagram) -> UnixDatagram {
363360
let mio_datagram = mio_uds::UnixDatagram::from_datagram(datagram).unwrap();
364361
UnixDatagram {
365-
raw_fd: mio_datagram.as_raw_fd(),
366362
io_handle: IoHandle::new(mio_datagram),
367363
}
368364
}
369365
}
370366

371367
impl AsRawFd for UnixDatagram {
372368
fn as_raw_fd(&self) -> RawFd {
373-
self.raw_fd
369+
self.io_handle.get_ref().as_raw_fd()
374370
}
375371
}
376372

@@ -383,6 +379,6 @@ impl FromRawFd for UnixDatagram {
383379

384380
impl IntoRawFd for UnixDatagram {
385381
fn into_raw_fd(self) -> RawFd {
386-
self.raw_fd
382+
self.io_handle.into_inner().into_raw_fd()
387383
}
388384
}

‎src/os/unix/net/listener.rs

+2-7
Original file line numberDiff line numberDiff line change
@@ -50,8 +50,6 @@ use crate::task::{blocking, Context, Poll};
5050
pub struct UnixListener {
5151
#[cfg(not(feature = "docs"))]
5252
io_handle: IoHandle<mio_uds::UnixListener>,
53-
54-
raw_fd: RawFd,
5553
}
5654

5755
impl UnixListener {
@@ -73,7 +71,6 @@ impl UnixListener {
7371
let listener = blocking::spawn(async move { mio_uds::UnixListener::bind(path) }).await?;
7472

7573
Ok(UnixListener {
76-
raw_fd: listener.as_raw_fd(),
7774
io_handle: IoHandle::new(listener),
7875
})
7976
}
@@ -102,7 +99,6 @@ impl UnixListener {
10299
Ok(Some((io, addr))) => {
103100
let mio_stream = mio_uds::UnixStream::from_stream(io)?;
104101
let stream = UnixStream {
105-
raw_fd: mio_stream.as_raw_fd(),
106102
io_handle: IoHandle::new(mio_stream),
107103
};
108104
Poll::Ready(Ok((stream, addr)))
@@ -214,15 +210,14 @@ impl From<std::os::unix::net::UnixListener> for UnixListener {
214210
fn from(listener: std::os::unix::net::UnixListener) -> UnixListener {
215211
let mio_listener = mio_uds::UnixListener::from_listener(listener).unwrap();
216212
UnixListener {
217-
raw_fd: mio_listener.as_raw_fd(),
218213
io_handle: IoHandle::new(mio_listener),
219214
}
220215
}
221216
}
222217

223218
impl AsRawFd for UnixListener {
224219
fn as_raw_fd(&self) -> RawFd {
225-
self.raw_fd
220+
self.io_handle.get_ref().as_raw_fd()
226221
}
227222
}
228223

@@ -235,6 +230,6 @@ impl FromRawFd for UnixListener {
235230

236231
impl IntoRawFd for UnixListener {
237232
fn into_raw_fd(self) -> RawFd {
238-
self.raw_fd
233+
self.io_handle.into_inner().into_raw_fd()
239234
}
240235
}

‎src/os/unix/net/stream.rs

+2-8
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,6 @@ use crate::task::{blocking, Context, Poll};
4242
pub struct UnixStream {
4343
#[cfg(not(feature = "docs"))]
4444
pub(super) io_handle: IoHandle<mio_uds::UnixStream>,
45-
46-
pub(super) raw_fd: RawFd,
4745
}
4846

4947
impl UnixStream {
@@ -71,7 +69,6 @@ impl UnixStream {
7169
let mut state = {
7270
match blocking::spawn(async move { mio_uds::UnixStream::connect(path) }).await {
7371
Ok(mio_stream) => State::Waiting(UnixStream {
74-
raw_fd: mio_stream.as_raw_fd(),
7572
io_handle: IoHandle::new(mio_stream),
7673
}),
7774
Err(err) => State::Error(err),
@@ -124,11 +121,9 @@ impl UnixStream {
124121
pub fn pair() -> io::Result<(UnixStream, UnixStream)> {
125122
let (a, b) = mio_uds::UnixStream::pair()?;
126123
let a = UnixStream {
127-
raw_fd: a.as_raw_fd(),
128124
io_handle: IoHandle::new(a),
129125
};
130126
let b = UnixStream {
131-
raw_fd: b.as_raw_fd(),
132127
io_handle: IoHandle::new(b),
133128
};
134129
Ok((a, b))
@@ -271,15 +266,14 @@ impl From<std::os::unix::net::UnixStream> for UnixStream {
271266
fn from(stream: std::os::unix::net::UnixStream) -> UnixStream {
272267
let mio_stream = mio_uds::UnixStream::from_stream(stream).unwrap();
273268
UnixStream {
274-
raw_fd: mio_stream.as_raw_fd(),
275269
io_handle: IoHandle::new(mio_stream),
276270
}
277271
}
278272
}
279273

280274
impl AsRawFd for UnixStream {
281275
fn as_raw_fd(&self) -> RawFd {
282-
self.raw_fd
276+
self.io_handle.get_ref().as_raw_fd()
283277
}
284278
}
285279

@@ -292,6 +286,6 @@ impl FromRawFd for UnixStream {
292286

293287
impl IntoRawFd for UnixStream {
294288
fn into_raw_fd(self) -> RawFd {
295-
self.raw_fd
289+
self.io_handle.into_inner().into_raw_fd()
296290
}
297291
}

0 commit comments

Comments
 (0)