1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

// Copyright 2015-2018 Parity Technologies (UK) Ltd.
// This file is part of Parity.

// Parity is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Parity is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Parity.  If not, see <http://www.gnu.org/licenses/>.

use std::os::unix::io::{FromRawFd, RawFd, AsRawFd};
use std::{fs, io};

use crate::{Result, ErrorKind, map_err};

#[derive(Debug)]
pub struct EventedPipe {
	file: fs::File,
}

impl EventedPipe {
	pub fn from_fd(fd: RawFd) -> Result<Self> {
		unsafe {
			// get the flags for this fd
			let previous = map_err!(
				libc::fcntl(fd, libc::F_GETFL),
				ErrorKind::Fnctl(io::Error::last_os_error())
			)?;

			// add the O_NONBLOCK flag
			map_err!(
				libc::fcntl(fd, libc::F_SETFL, previous | libc::O_NONBLOCK),
				ErrorKind::Fnctl(io::Error::last_os_error())
			)?;

			Ok(EventedPipe {
				file: fs::File::from_raw_fd(fd),
			})
		}
	}
}

impl mio::Evented for EventedPipe {
	fn register(
		&self,
		poll: &mio::Poll,
		token: mio::Token,
		interest: mio::Ready,
		opts: mio::PollOpt,
	) -> io::Result<()> {
		mio::unix::EventedFd(&self.file.as_raw_fd()).register(poll, token, interest, opts)
	}

	fn reregister(
		&self,
		poll: &mio::Poll,
		token: mio::Token,
		interest: mio::Ready,
		opts: mio::PollOpt,
	) -> io::Result<()> {
		mio::unix::EventedFd(&self.file.as_raw_fd()).reregister(poll, token, interest, opts)
	}

	fn deregister(&self, poll: &mio::Poll) -> io::Result<()> {
		mio::unix::EventedFd(&self.file.as_raw_fd()).deregister(poll)
	}
}

impl io::Read for EventedPipe {
	fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
		self.file.read(buf)
	}
}

impl io::Write for EventedPipe {
	fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
		self.file.write(buf)
	}

	fn flush(&mut self) -> io::Result<()> {
		self.file.flush()
	}
}

impl io::Seek for EventedPipe {
	fn seek(&mut self, pos: io::SeekFrom) -> io::Result<u64> {
		self.file.seek(pos)
	}
}