]> git.lizzy.rs Git - rust.git/blob - src/libstd/sys/windows/pipe.rs
ca7985aa35bf8f94ba2af6a2066ce5bdc4b47661
[rust.git] / src / libstd / sys / windows / pipe.rs
1 // Copyright 2014 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
4 //
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
10
11 //! Named pipes implementation for windows
12 //!
13 //! If are unfortunate enough to be reading this code, I would like to first
14 //! apologize. This was my first encounter with windows named pipes, and it
15 //! didn't exactly turn out very cleanly. If you, too, are new to named pipes,
16 //! read on as I'll try to explain some fun things that I ran into.
17 //!
18 //! # Unix pipes vs Named pipes
19 //!
20 //! As with everything else, named pipes on windows are pretty different from
21 //! unix pipes on unix. On unix, you use one "server pipe" to accept new client
22 //! pipes. So long as this server pipe is active, new children pipes can
23 //! connect. On windows, you instead have a number of "server pipes", and each
24 //! of these server pipes can throughout their lifetime be attached to a client
25 //! or not. Once attached to a client, a server pipe may then disconnect at a
26 //! later date.
27 //!
28 //! # Accepting clients
29 //!
30 //! As with most other I/O interfaces, our Listener/Acceptor/Stream interfaces
31 //! are built around the unix flavors. This means that we have one "server
32 //! pipe" to which many clients can connect. In order to make this compatible
33 //! with the windows model, each connected client consumes ownership of a server
34 //! pipe, and then a new server pipe is created for the next client.
35 //!
36 //! Note that the server pipes attached to clients are never given back to the
37 //! listener for recycling. This could possibly be implemented with a channel so
38 //! the listener half can re-use server pipes, but for now I err'd on the simple
39 //! side of things. Each stream accepted by a listener will destroy the server
40 //! pipe after the stream is dropped.
41 //!
42 //! This model ends up having a small race or two, and you can find more details
43 //! on the `native_accept` method.
44 //!
45 //! # Simultaneous reads and writes
46 //!
47 //! In testing, I found that two simultaneous writes and two simultaneous reads
48 //! on a pipe ended up working out just fine, but problems were encountered when
49 //! a read was executed simultaneously with a write. After some googling around,
50 //! it sounded like named pipes just weren't built for this kind of interaction,
51 //! and the suggested solution was to use overlapped I/O.
52 //!
53 //! I don't really know what overlapped I/O is, but my basic understanding after
54 //! reading about it is that you have an external Event which is used to signal
55 //! I/O completion, passed around in some OVERLAPPED structures. As to what this
56 //! is, I'm not exactly sure.
57 //!
58 //! This problem implies that all named pipes are created with the
59 //! FILE_FLAG_OVERLAPPED option. This means that all of their I/O is
60 //! asynchronous. Each I/O operation has an associated OVERLAPPED structure, and
61 //! inside of this structure is a HANDLE from CreateEvent. After the I/O is
62 //! determined to be pending (may complete in the future), the
63 //! GetOverlappedResult function is used to block on the event, waiting for the
64 //! I/O to finish.
65 //!
66 //! This scheme ended up working well enough. There were two snags that I ran
67 //! into, however:
68 //!
69 //! * Each UnixStream instance needs its own read/write events to wait on. These
70 //!   can't be shared among clones of the same stream because the documentation
71 //!   states that it unsets the event when the I/O is started (would possibly
72 //!   corrupt other events simultaneously waiting). For convenience's sake,
73 //!   these events are lazily initialized.
74 //!
75 //! * Each server pipe needs to be created with FILE_FLAG_OVERLAPPED in addition
76 //!   to all pipes created through `connect`. Notably this means that the
77 //!   ConnectNamedPipe function is nonblocking, implying that the Listener needs
78 //!   to have yet another event to do the actual blocking.
79 //!
80 //! # Conclusion
81 //!
82 //! The conclusion here is that I probably don't know the best way to work with
83 //! windows named pipes, but the solution here seems to work well enough to get
84 //! the test suite passing (the suite is in libstd), and that's good enough for
85 //! me!
86
87 use alloc::arc::Arc;
88 use libc;
89 use c_str::CString;
90 use mem;
91 use ptr;
92 use sync::atomic;
93 use rustrt::mutex;
94 use io::{mod, IoError, IoResult};
95 use prelude::*;
96
97 use sys_common::{mod, eof};
98
99 use super::{c, os, timer, to_utf16, decode_error_detailed};
100
101 struct Event(libc::HANDLE);
102
103 impl Event {
104     fn new(manual_reset: bool, initial_state: bool) -> IoResult<Event> {
105         let event = unsafe {
106             libc::CreateEventW(ptr::null_mut(),
107                                manual_reset as libc::BOOL,
108                                initial_state as libc::BOOL,
109                                ptr::null())
110         };
111         if event as uint == 0 {
112             Err(super::last_error())
113         } else {
114             Ok(Event(event))
115         }
116     }
117
118     fn handle(&self) -> libc::HANDLE { let Event(handle) = *self; handle }
119 }
120
121 impl Drop for Event {
122     fn drop(&mut self) {
123         unsafe { let _ = libc::CloseHandle(self.handle()); }
124     }
125 }
126
127 struct Inner {
128     handle: libc::HANDLE,
129     lock: mutex::NativeMutex,
130     read_closed: atomic::AtomicBool,
131     write_closed: atomic::AtomicBool,
132 }
133
134 impl Inner {
135     fn new(handle: libc::HANDLE) -> Inner {
136         Inner {
137             handle: handle,
138             lock: unsafe { mutex::NativeMutex::new() },
139             read_closed: atomic::AtomicBool::new(false),
140             write_closed: atomic::AtomicBool::new(false),
141         }
142     }
143 }
144
145 impl Drop for Inner {
146     fn drop(&mut self) {
147         unsafe {
148             let _ = libc::FlushFileBuffers(self.handle);
149             let _ = libc::CloseHandle(self.handle);
150         }
151     }
152 }
153
154 unsafe fn pipe(name: *const u16, init: bool) -> libc::HANDLE {
155     libc::CreateNamedPipeW(
156         name,
157         libc::PIPE_ACCESS_DUPLEX |
158             if init {libc::FILE_FLAG_FIRST_PIPE_INSTANCE} else {0} |
159             libc::FILE_FLAG_OVERLAPPED,
160         libc::PIPE_TYPE_BYTE | libc::PIPE_READMODE_BYTE |
161             libc::PIPE_WAIT,
162         libc::PIPE_UNLIMITED_INSTANCES,
163         65536,
164         65536,
165         0,
166         ptr::null_mut()
167     )
168 }
169
170 pub fn await(handle: libc::HANDLE, deadline: u64,
171              events: &[libc::HANDLE]) -> IoResult<uint> {
172     use libc::consts::os::extra::{WAIT_FAILED, WAIT_TIMEOUT, WAIT_OBJECT_0};
173
174     // If we've got a timeout, use WaitForSingleObject in tandem with CancelIo
175     // to figure out if we should indeed get the result.
176     let ms = if deadline == 0 {
177         libc::INFINITE as u64
178     } else {
179         let now = timer::now();
180         if deadline < now {0} else {deadline - now}
181     };
182     let ret = unsafe {
183         c::WaitForMultipleObjects(events.len() as libc::DWORD,
184                                   events.as_ptr(),
185                                   libc::FALSE,
186                                   ms as libc::DWORD)
187     };
188     match ret {
189         WAIT_FAILED => Err(super::last_error()),
190         WAIT_TIMEOUT => unsafe {
191             let _ = c::CancelIo(handle);
192             Err(sys_common::timeout("operation timed out"))
193         },
194         n => Ok((n - WAIT_OBJECT_0) as uint)
195     }
196 }
197
198 fn epipe() -> IoError {
199     IoError {
200         kind: io::EndOfFile,
201         desc: "the pipe has ended",
202         detail: None,
203     }
204 }
205
206 ////////////////////////////////////////////////////////////////////////////////
207 // Unix Streams
208 ////////////////////////////////////////////////////////////////////////////////
209
210 pub struct UnixStream {
211     inner: Arc<Inner>,
212     write: Option<Event>,
213     read: Option<Event>,
214     read_deadline: u64,
215     write_deadline: u64,
216 }
217
218 impl UnixStream {
219     fn try_connect(p: *const u16) -> Option<libc::HANDLE> {
220         // Note that most of this is lifted from the libuv implementation.
221         // The idea is that if we fail to open a pipe in read/write mode
222         // that we try afterwards in just read or just write
223         let mut result = unsafe {
224             libc::CreateFileW(p,
225                 libc::GENERIC_READ | libc::GENERIC_WRITE,
226                 0,
227                 ptr::null_mut(),
228                 libc::OPEN_EXISTING,
229                 libc::FILE_FLAG_OVERLAPPED,
230                 ptr::null_mut())
231         };
232         if result != libc::INVALID_HANDLE_VALUE {
233             return Some(result)
234         }
235
236         let err = unsafe { libc::GetLastError() };
237         if err == libc::ERROR_ACCESS_DENIED as libc::DWORD {
238             result = unsafe {
239                 libc::CreateFileW(p,
240                     libc::GENERIC_READ | libc::FILE_WRITE_ATTRIBUTES,
241                     0,
242                     ptr::null_mut(),
243                     libc::OPEN_EXISTING,
244                     libc::FILE_FLAG_OVERLAPPED,
245                     ptr::null_mut())
246             };
247             if result != libc::INVALID_HANDLE_VALUE {
248                 return Some(result)
249             }
250         }
251         let err = unsafe { libc::GetLastError() };
252         if err == libc::ERROR_ACCESS_DENIED as libc::DWORD {
253             result = unsafe {
254                 libc::CreateFileW(p,
255                     libc::GENERIC_WRITE | libc::FILE_READ_ATTRIBUTES,
256                     0,
257                     ptr::null_mut(),
258                     libc::OPEN_EXISTING,
259                     libc::FILE_FLAG_OVERLAPPED,
260                     ptr::null_mut())
261             };
262             if result != libc::INVALID_HANDLE_VALUE {
263                 return Some(result)
264             }
265         }
266         None
267     }
268
269     pub fn connect(addr: &CString, timeout: Option<u64>) -> IoResult<UnixStream> {
270         let addr = try!(to_utf16(addr.as_str()));
271         let start = timer::now();
272         loop {
273             match UnixStream::try_connect(addr.as_ptr()) {
274                 Some(handle) => {
275                     let inner = Inner::new(handle);
276                     let mut mode = libc::PIPE_TYPE_BYTE |
277                                    libc::PIPE_READMODE_BYTE |
278                                    libc::PIPE_WAIT;
279                     let ret = unsafe {
280                         libc::SetNamedPipeHandleState(inner.handle,
281                                                       &mut mode,
282                                                       ptr::null_mut(),
283                                                       ptr::null_mut())
284                     };
285                     return if ret == 0 {
286                         Err(super::last_error())
287                     } else {
288                         Ok(UnixStream {
289                             inner: Arc::new(inner),
290                             read: None,
291                             write: None,
292                             read_deadline: 0,
293                             write_deadline: 0,
294                         })
295                     }
296                 }
297                 None => {}
298             }
299
300             // On windows, if you fail to connect, you may need to call the
301             // `WaitNamedPipe` function, and this is indicated with an error
302             // code of ERROR_PIPE_BUSY.
303             let code = unsafe { libc::GetLastError() };
304             if code as int != libc::ERROR_PIPE_BUSY as int {
305                 return Err(super::last_error())
306             }
307
308             match timeout {
309                 Some(timeout) => {
310                     let now = timer::now();
311                     let timed_out = (now - start) >= timeout || unsafe {
312                         let ms = (timeout - (now - start)) as libc::DWORD;
313                         libc::WaitNamedPipeW(addr.as_ptr(), ms) == 0
314                     };
315                     if timed_out {
316                         return Err(sys_common::timeout("connect timed out"))
317                     }
318                 }
319
320                 // An example I found on Microsoft's website used 20
321                 // seconds, libuv uses 30 seconds, hence we make the
322                 // obvious choice of waiting for 25 seconds.
323                 None => {
324                     if unsafe { libc::WaitNamedPipeW(addr.as_ptr(), 25000) } == 0 {
325                         return Err(super::last_error())
326                     }
327                 }
328             }
329         }
330     }
331
332     pub fn handle(&self) -> libc::HANDLE { self.inner.handle }
333
334     fn read_closed(&self) -> bool {
335         self.inner.read_closed.load(atomic::SeqCst)
336     }
337
338     fn write_closed(&self) -> bool {
339         self.inner.write_closed.load(atomic::SeqCst)
340     }
341
342     fn cancel_io(&self) -> IoResult<()> {
343         match unsafe { c::CancelIoEx(self.handle(), ptr::null_mut()) } {
344             0 if os::errno() == libc::ERROR_NOT_FOUND as uint => {
345                 Ok(())
346             }
347             0 => Err(super::last_error()),
348             _ => Ok(())
349         }
350     }
351
352     pub fn read(&mut self, buf: &mut [u8]) -> IoResult<uint> {
353         if self.read.is_none() {
354             self.read = Some(try!(Event::new(true, false)));
355         }
356
357         let mut bytes_read = 0;
358         let mut overlapped: libc::OVERLAPPED = unsafe { mem::zeroed() };
359         overlapped.hEvent = self.read.as_ref().unwrap().handle();
360
361         // Pre-flight check to see if the reading half has been closed. This
362         // must be done before issuing the ReadFile request, but after we
363         // acquire the lock.
364         //
365         // See comments in close_read() about why this lock is necessary.
366         let guard = unsafe { self.inner.lock.lock() };
367         if self.read_closed() {
368             return Err(eof())
369         }
370
371         // Issue a nonblocking requests, succeeding quickly if it happened to
372         // succeed.
373         let ret = unsafe {
374             libc::ReadFile(self.handle(),
375                            buf.as_ptr() as libc::LPVOID,
376                            buf.len() as libc::DWORD,
377                            &mut bytes_read,
378                            &mut overlapped)
379         };
380         if ret != 0 { return Ok(bytes_read as uint) }
381
382         // If our errno doesn't say that the I/O is pending, then we hit some
383         // legitimate error and return immediately.
384         if os::errno() != libc::ERROR_IO_PENDING as uint {
385             return Err(super::last_error())
386         }
387
388         // Now that we've issued a successful nonblocking request, we need to
389         // wait for it to finish. This can all be done outside the lock because
390         // we'll see any invocation of CancelIoEx. We also call this in a loop
391         // because we're woken up if the writing half is closed, we just need to
392         // realize that the reading half wasn't closed and we go right back to
393         // sleep.
394         drop(guard);
395         loop {
396             // Process a timeout if one is pending
397             let wait_succeeded = await(self.handle(), self.read_deadline,
398                                        &[overlapped.hEvent]);
399
400             let ret = unsafe {
401                 libc::GetOverlappedResult(self.handle(),
402                                           &mut overlapped,
403                                           &mut bytes_read,
404                                           libc::TRUE)
405             };
406             // If we succeeded, or we failed for some reason other than
407             // CancelIoEx, return immediately
408             if ret != 0 { return Ok(bytes_read as uint) }
409             if os::errno() != libc::ERROR_OPERATION_ABORTED as uint {
410                 return Err(super::last_error())
411             }
412
413             // If the reading half is now closed, then we're done. If we woke up
414             // because the writing half was closed, keep trying.
415             if wait_succeeded.is_err() {
416                 return Err(sys_common::timeout("read timed out"))
417             }
418             if self.read_closed() {
419                 return Err(eof())
420             }
421         }
422     }
423
424     pub fn write(&mut self, buf: &[u8]) -> IoResult<()> {
425         if self.write.is_none() {
426             self.write = Some(try!(Event::new(true, false)));
427         }
428
429         let mut offset = 0;
430         let mut overlapped: libc::OVERLAPPED = unsafe { mem::zeroed() };
431         overlapped.hEvent = self.write.as_ref().unwrap().handle();
432
433         while offset < buf.len() {
434             let mut bytes_written = 0;
435
436             // This sequence below is quite similar to the one found in read().
437             // Some careful looping is done to ensure that if close_write() is
438             // invoked we bail out early, and if close_read() is invoked we keep
439             // going after we woke up.
440             //
441             // See comments in close_read() about why this lock is necessary.
442             let guard = unsafe { self.inner.lock.lock() };
443             if self.write_closed() {
444                 return Err(epipe())
445             }
446             let ret = unsafe {
447                 libc::WriteFile(self.handle(),
448                                 buf[offset..].as_ptr() as libc::LPVOID,
449                                 (buf.len() - offset) as libc::DWORD,
450                                 &mut bytes_written,
451                                 &mut overlapped)
452             };
453             let err = os::errno();
454             drop(guard);
455
456             if ret == 0 {
457                 if err != libc::ERROR_IO_PENDING as uint {
458                     return Err(decode_error_detailed(err as i32))
459                 }
460                 // Process a timeout if one is pending
461                 let wait_succeeded = await(self.handle(), self.write_deadline,
462                                            &[overlapped.hEvent]);
463                 let ret = unsafe {
464                     libc::GetOverlappedResult(self.handle(),
465                                               &mut overlapped,
466                                               &mut bytes_written,
467                                               libc::TRUE)
468                 };
469                 // If we weren't aborted, this was a legit error, if we were
470                 // aborted, then check to see if the write half was actually
471                 // closed or whether we woke up from the read half closing.
472                 if ret == 0 {
473                     if os::errno() != libc::ERROR_OPERATION_ABORTED as uint {
474                         return Err(super::last_error())
475                     }
476                     if !wait_succeeded.is_ok() {
477                         let amt = offset + bytes_written as uint;
478                         return if amt > 0 {
479                             Err(IoError {
480                                 kind: io::ShortWrite(amt),
481                                 desc: "short write during write",
482                                 detail: None,
483                             })
484                         } else {
485                             Err(sys_common::timeout("write timed out"))
486                         }
487                     }
488                     if self.write_closed() {
489                         return Err(epipe())
490                     }
491                     continue // retry
492                 }
493             }
494             offset += bytes_written as uint;
495         }
496         Ok(())
497     }
498
499     pub fn close_read(&mut self) -> IoResult<()> {
500         // On windows, there's no actual shutdown() method for pipes, so we're
501         // forced to emulate the behavior manually at the application level. To
502         // do this, we need to both cancel any pending requests, as well as
503         // prevent all future requests from succeeding. These two operations are
504         // not atomic with respect to one another, so we must use a lock to do
505         // so.
506         //
507         // The read() code looks like:
508         //
509         //      1. Make sure the pipe is still open
510         //      2. Submit a read request
511         //      3. Wait for the read request to finish
512         //
513         // The race this lock is preventing is if another thread invokes
514         // close_read() between steps 1 and 2. By atomically executing steps 1
515         // and 2 with a lock with respect to close_read(), we're guaranteed that
516         // no thread will erroneously sit in a read forever.
517         let _guard = unsafe { self.inner.lock.lock() };
518         self.inner.read_closed.store(true, atomic::SeqCst);
519         self.cancel_io()
520     }
521
522     pub fn close_write(&mut self) -> IoResult<()> {
523         // see comments in close_read() for why this lock is necessary
524         let _guard = unsafe { self.inner.lock.lock() };
525         self.inner.write_closed.store(true, atomic::SeqCst);
526         self.cancel_io()
527     }
528
529     pub fn set_timeout(&mut self, timeout: Option<u64>) {
530         let deadline = timeout.map(|a| timer::now() + a).unwrap_or(0);
531         self.read_deadline = deadline;
532         self.write_deadline = deadline;
533     }
534     pub fn set_read_timeout(&mut self, timeout: Option<u64>) {
535         self.read_deadline = timeout.map(|a| timer::now() + a).unwrap_or(0);
536     }
537     pub fn set_write_timeout(&mut self, timeout: Option<u64>) {
538         self.write_deadline = timeout.map(|a| timer::now() + a).unwrap_or(0);
539     }
540 }
541
542 impl Clone for UnixStream {
543     fn clone(&self) -> UnixStream {
544         UnixStream {
545             inner: self.inner.clone(),
546             read: None,
547             write: None,
548             read_deadline: 0,
549             write_deadline: 0,
550         }
551     }
552 }
553
554 ////////////////////////////////////////////////////////////////////////////////
555 // Unix Listener
556 ////////////////////////////////////////////////////////////////////////////////
557
558 pub struct UnixListener {
559     handle: libc::HANDLE,
560     name: CString,
561 }
562
563 impl UnixListener {
564     pub fn bind(addr: &CString) -> IoResult<UnixListener> {
565         // Although we technically don't need the pipe until much later, we
566         // create the initial handle up front to test the validity of the name
567         // and such.
568         let addr_v = try!(to_utf16(addr.as_str()));
569         let ret = unsafe { pipe(addr_v.as_ptr(), true) };
570         if ret == libc::INVALID_HANDLE_VALUE {
571             Err(super::last_error())
572         } else {
573             Ok(UnixListener { handle: ret, name: addr.clone() })
574         }
575     }
576
577     pub fn listen(self) -> IoResult<UnixAcceptor> {
578         Ok(UnixAcceptor {
579             listener: self,
580             event: try!(Event::new(true, false)),
581             deadline: 0,
582             inner: Arc::new(AcceptorState {
583                 abort: try!(Event::new(true, false)),
584                 closed: atomic::AtomicBool::new(false),
585             }),
586         })
587     }
588
589     pub fn handle(&self) -> libc::HANDLE {
590         self.handle
591     }
592 }
593
594 impl Drop for UnixListener {
595     fn drop(&mut self) {
596         unsafe { let _ = libc::CloseHandle(self.handle); }
597     }
598 }
599
600 pub struct UnixAcceptor {
601     inner: Arc<AcceptorState>,
602     listener: UnixListener,
603     event: Event,
604     deadline: u64,
605 }
606
607 struct AcceptorState {
608     abort: Event,
609     closed: atomic::AtomicBool,
610 }
611
612 impl UnixAcceptor {
613     pub fn accept(&mut self) -> IoResult<UnixStream> {
614         // This function has some funky implementation details when working with
615         // unix pipes. On windows, each server named pipe handle can be
616         // connected to a one or zero clients. To the best of my knowledge, a
617         // named server is considered active and present if there exists at
618         // least one server named pipe for it.
619         //
620         // The model of this function is to take the current known server
621         // handle, connect a client to it, and then transfer ownership to the
622         // UnixStream instance. The next time accept() is invoked, it'll need a
623         // different server handle to connect a client to.
624         //
625         // Note that there is a possible race here. Once our server pipe is
626         // handed off to a `UnixStream` object, the stream could be closed,
627         // meaning that there would be no active server pipes, hence even though
628         // we have a valid `UnixAcceptor`, no one can connect to it. For this
629         // reason, we generate the next accept call's server pipe at the end of
630         // this function call.
631         //
632         // This provides us an invariant that we always have at least one server
633         // connection open at a time, meaning that all connects to this acceptor
634         // should succeed while this is active.
635         //
636         // The actual implementation of doing this is a little tricky. Once a
637         // server pipe is created, a client can connect to it at any time. I
638         // assume that which server a client connects to is nondeterministic, so
639         // we also need to guarantee that the only server able to be connected
640         // to is the one that we're calling ConnectNamedPipe on. This means that
641         // we have to create the second server pipe *after* we've already
642         // accepted a connection. In order to at least somewhat gracefully
643         // handle errors, this means that if the second server pipe creation
644         // fails that we disconnect the connected client and then just keep
645         // using the original server pipe.
646         let handle = self.listener.handle;
647
648         // If we've had an artificial call to close_accept, be sure to never
649         // proceed in accepting new clients in the future
650         if self.inner.closed.load(atomic::SeqCst) { return Err(eof()) }
651
652         let name = try!(to_utf16(self.listener.name.as_str()));
653
654         // Once we've got a "server handle", we need to wait for a client to
655         // connect. The ConnectNamedPipe function will block this thread until
656         // someone on the other end connects. This function can "fail" if a
657         // client connects after we created the pipe but before we got down
658         // here. Thanks windows.
659         let mut overlapped: libc::OVERLAPPED = unsafe { mem::zeroed() };
660         overlapped.hEvent = self.event.handle();
661         if unsafe { libc::ConnectNamedPipe(handle, &mut overlapped) == 0 } {
662             let mut err = unsafe { libc::GetLastError() };
663
664             if err == libc::ERROR_IO_PENDING as libc::DWORD {
665                 // Process a timeout if one is pending
666                 let wait_succeeded = await(handle, self.deadline,
667                                            &[self.inner.abort.handle(),
668                                              overlapped.hEvent]);
669
670                 // This will block until the overlapped I/O is completed. The
671                 // timeout was previously handled, so this will either block in
672                 // the normal case or succeed very quickly in the timeout case.
673                 let ret = unsafe {
674                     let mut transfer = 0;
675                     libc::GetOverlappedResult(handle,
676                                               &mut overlapped,
677                                               &mut transfer,
678                                               libc::TRUE)
679                 };
680                 if ret == 0 {
681                     if wait_succeeded.is_ok() {
682                         err = unsafe { libc::GetLastError() };
683                     } else {
684                         return Err(sys_common::timeout("accept timed out"))
685                     }
686                 } else {
687                     // we succeeded, bypass the check below
688                     err = libc::ERROR_PIPE_CONNECTED as libc::DWORD;
689                 }
690             }
691             if err != libc::ERROR_PIPE_CONNECTED as libc::DWORD {
692                 return Err(super::last_error())
693             }
694         }
695
696         // Now that we've got a connected client to our handle, we need to
697         // create a second server pipe. If this fails, we disconnect the
698         // connected client and return an error (see comments above).
699         let new_handle = unsafe { pipe(name.as_ptr(), false) };
700         if new_handle == libc::INVALID_HANDLE_VALUE {
701             let ret = Err(super::last_error());
702             // If our disconnection fails, then there's not really a whole lot
703             // that we can do, so panic
704             let err = unsafe { libc::DisconnectNamedPipe(handle) };
705             assert!(err != 0);
706             return ret;
707         } else {
708             self.listener.handle = new_handle;
709         }
710
711         // Transfer ownership of our handle into this stream
712         Ok(UnixStream {
713             inner: Arc::new(Inner::new(handle)),
714             read: None,
715             write: None,
716             read_deadline: 0,
717             write_deadline: 0,
718         })
719     }
720
721     pub fn set_timeout(&mut self, timeout: Option<u64>) {
722         self.deadline = timeout.map(|i| i + timer::now()).unwrap_or(0);
723     }
724
725     pub fn close_accept(&mut self) -> IoResult<()> {
726         self.inner.closed.store(true, atomic::SeqCst);
727         let ret = unsafe {
728             c::SetEvent(self.inner.abort.handle())
729         };
730         if ret == 0 {
731             Err(super::last_error())
732         } else {
733             Ok(())
734         }
735     }
736
737     pub fn handle(&self) -> libc::HANDLE {
738         self.listener.handle()
739     }
740 }
741
742 impl Clone for UnixAcceptor {
743     fn clone(&self) -> UnixAcceptor {
744         let name = to_utf16(self.listener.name.as_str()).ok().unwrap();
745         UnixAcceptor {
746             inner: self.inner.clone(),
747             event: Event::new(true, false).ok().unwrap(),
748             deadline: 0,
749             listener: UnixListener {
750                 name: self.listener.name.clone(),
751                 handle: unsafe {
752                     let p = pipe(name.as_ptr(), false) ;
753                     assert!(p != libc::INVALID_HANDLE_VALUE as libc::HANDLE);
754                     p
755                 },
756             },
757         }
758     }
759 }