use std::io;
use std::mem;
use std::os::unix::io::{RawFd};
use nix::unistd::pipe2;
use nix::fcntl::OFlag;
use libc;
use libc::{c_void, size_t};
use error::{result, Error};
use error::ErrorCode::CreatePipe;
#[derive(Debug)]
pub struct Pipe(RawFd, RawFd);
#[derive(Debug)]
pub struct PipeReader(RawFd);
#[derive(Debug)]
pub struct PipeWriter(RawFd);
#[derive(Debug)]
pub enum PipeHolder {
Reader(PipeReader),
Writer(PipeWriter),
}
impl Pipe {
pub fn new() -> Result<Pipe, Error> {
let (rd, wr) = try!(result(CreatePipe, pipe2(OFlag::O_CLOEXEC)));
Ok(Pipe(rd, wr))
}
pub fn split(self) -> (PipeReader, PipeWriter) {
let Pipe(rd, wr) = self;
mem::forget(self);
(PipeReader(rd), PipeWriter(wr))
}
}
impl Drop for Pipe {
fn drop(&mut self) {
let Pipe(x, y) = *self;
unsafe {
libc::close(x);
libc::close(y);
}
}
}
impl PipeReader {
pub fn into_fd(self) -> RawFd {
let PipeReader(fd) = self;
mem::forget(self);
return fd;
}
}
impl PipeWriter {
pub fn into_fd(self) -> RawFd {
let PipeWriter(fd) = self;
mem::forget(self);
return fd;
}
}
impl Drop for PipeReader {
fn drop(&mut self) {
unsafe { libc::close(self.0) };
}
}
impl Drop for PipeWriter {
fn drop(&mut self) {
unsafe { libc::close(self.0) };
}
}
impl io::Read for PipeReader {
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
let ret = unsafe {
libc::read(self.0,
buf.as_mut_ptr() as *mut c_void,
buf.len() as size_t)
};
if ret < 0 {
return Err(io::Error::last_os_error());
}
Ok(ret as usize)
}
}
impl io::Write for PipeWriter {
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
let ret = unsafe {
libc::write(self.0,
buf.as_ptr() as *const c_void,
buf.len() as size_t)
};
if ret < 0 {
return Err(io::Error::last_os_error());
}
Ok(ret as usize)
}
fn flush(&mut self) -> io::Result<()> { Ok(()) }
}