From 1db416150d4895b72b62881ec533b17e98a20e54 Mon Sep 17 00:00:00 2001
From: Wolfgang Bumiller <w.bumiller@proxmox.com>
Date: Wed, 13 Feb 2019 14:02:27 +0100
Subject: [PATCH] tools/fs: let ReadDir iterator yield ReadDirEntry

Signed-off-by: Wolfgang Bumiller <w.bumiller@proxmox.com>
---
 src/tools/fs.rs | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/tools/fs.rs b/src/tools/fs.rs
index 8ab33477..447583b9 100644
--- a/src/tools/fs.rs
+++ b/src/tools/fs.rs
@@ -2,7 +2,7 @@
 
 use std::borrow::{Borrow, BorrowMut};
 use std::ops::{Deref, DerefMut};
-use std::os::unix::io::RawFd;
+use std::os::unix::io::{AsRawFd, RawFd};
 
 use failure::Error;
 use nix::dir;
@@ -72,13 +72,17 @@ impl ReadDirEntry {
 /// Wrapper over a pair of `nix::dir::Dir` and `nix::dir::Iter`, returned by `read_subdir()`.
 pub struct ReadDir {
     iter: Tied<Dir, Iterator<Item = nix::Result<dir::Entry>>>,
+    dir_fd: RawFd,
 }
 
 impl Iterator for ReadDir {
-    type Item = Result<dir::Entry, Error>;
+    type Item = Result<ReadDirEntry, Error>;
 
     fn next(&mut self) -> Option<Self::Item> {
-        self.iter.next().map(|res| res.map_err(|e| Error::from(e)))
+        self.iter.next().map(|res| {
+            res.map(|entry| ReadDirEntry { entry, parent_fd: self.dir_fd })
+                .map_err(|e| Error::from(e))
+        })
     }
 }
 
@@ -89,10 +93,11 @@ pub fn read_subdir<P: ?Sized + nix::NixPath>(dirfd: RawFd, path: &P) -> Result<R
     use nix::sys::stat::Mode;
 
     let dir = Dir::openat(dirfd, path, OFlag::O_RDONLY, Mode::empty())?;
+    let fd = dir.as_raw_fd();
     let iter = Tied::new(dir, |dir| {
         Box::new(unsafe { (*dir).iter() }) as Box<Iterator<Item = nix::Result<dir::Entry>>>
     });
-    Ok(ReadDir { iter })
+    Ok(ReadDir { iter, dir_fd: fd })
 }
 
 /// Scan through a directory with a regular expression. This is simply a shortcut filtering the