microsd 0.1.0

Light‐weight systemd auxiliars
Documentation 
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

use std::iter::FusedIterator;
use std::ops::Range;
use std::os::fd::RawFd;

use nix::unistd::close;

use crate::Error;
use crate::fd::Fd;

/// Listen file descriptor iterator for undifferentiated [`Fd`].
#[must_use]
#[derive(Debug)]
pub struct ListenFds(Range<RawFd>);

impl ListenFds {
	pub(crate) const START: RawFd = 3;

	/// Construct new [`ListenFds`] instance from a file descriptor number.
	///
	/// # Safety
	///
	/// This function is only safe to use if `num` unowned system file
	/// descriptors are available starting from descriptor number `3`
	/// ([`SD_LISTEN_FDS_START`](https://www.freedesktop.org/software/systemd/man/latest/SD_LISTEN_FDS_START.html)).
	///
	/// It is intended to be used with the number provided through the
	/// `LISTEN_FDS` environment variable by the session manager.
	#[inline]
	pub const unsafe fn new(num: u32) -> Self {
		Self(Range { start: Self::START, end: Self::START + num as RawFd })
	}
}

impl Drop for ListenFds {
	fn drop(&mut self) {
		// Close remaining file descriptors
		for fd in &mut self.0 {
			let res = close(fd);
			debug_assert!(res.is_ok());
		}
	}
}

impl Iterator for ListenFds {
	type Item = Result<Fd, Error>;

	#[inline]
	fn next(&mut self) -> Option<Self::Item> {
		self.0.next().map(|fd| unsafe { Fd::try_from(fd) })
	}

	#[inline]
	fn size_hint(&self) -> (usize, Option<usize>) {
		self.0.size_hint()
	}
}

impl DoubleEndedIterator for ListenFds {
	#[inline]
	fn next_back(&mut self) -> Option<Self::Item> {
		self.0.next_back().map(|fd| unsafe { Fd::try_from(fd) })
	}
}

impl ExactSizeIterator for ListenFds {}
impl FusedIterator for ListenFds {}