| 1 | use std::error; | 
| 2 | use std::fmt; | 
|---|
| 3 | use std::io; | 
|---|
| 4 | use std::path::{Path, PathBuf}; | 
|---|
| 5 |  | 
|---|
| 6 | use crate::DirEntry; | 
|---|
| 7 |  | 
|---|
| 8 | /// An error produced by recursively walking a directory. | 
|---|
| 9 | /// | 
|---|
| 10 | /// This error type is a light wrapper around [`std::io::Error`]. In | 
|---|
| 11 | /// particular, it adds the following information: | 
|---|
| 12 | /// | 
|---|
| 13 | /// * The depth at which the error occurred in the file tree, relative to the | 
|---|
| 14 | /// root. | 
|---|
| 15 | /// * The path, if any, associated with the IO error. | 
|---|
| 16 | /// * An indication that a loop occurred when following symbolic links. In this | 
|---|
| 17 | /// case, there is no underlying IO error. | 
|---|
| 18 | /// | 
|---|
| 19 | /// To maintain good ergonomics, this type has a | 
|---|
| 20 | /// [`impl From<Error> for std::io::Error`][impl] defined which preserves the original context. | 
|---|
| 21 | /// This allows you to use an [`io::Result`] with methods in this crate if you don't care about | 
|---|
| 22 | /// accessing the underlying error data in a structured form. | 
|---|
| 23 | /// | 
|---|
| 24 | /// [`std::io::Error`]: https://doc.rust-lang.org/stable/std/io/struct.Error.html | 
|---|
| 25 | /// [`io::Result`]: https://doc.rust-lang.org/stable/std/io/type.Result.html | 
|---|
| 26 | /// [impl]: struct.Error.html#impl-From%3CError%3E | 
|---|
| 27 | #[ derive(Debug)] | 
|---|
| 28 | pub struct Error { | 
|---|
| 29 | depth: usize, | 
|---|
| 30 | inner: ErrorInner, | 
|---|
| 31 | } | 
|---|
| 32 |  | 
|---|
| 33 | #[ derive(Debug)] | 
|---|
| 34 | enum ErrorInner { | 
|---|
| 35 | Io { path: Option<PathBuf>, err: io::Error }, | 
|---|
| 36 | Loop { ancestor: PathBuf, child: PathBuf }, | 
|---|
| 37 | } | 
|---|
| 38 |  | 
|---|
| 39 | impl Error { | 
|---|
| 40 | /// Returns the path associated with this error if one exists. | 
|---|
| 41 | /// | 
|---|
| 42 | /// For example, if an error occurred while opening a directory handle, | 
|---|
| 43 | /// the error will include the path passed to [`std::fs::read_dir`]. | 
|---|
| 44 | /// | 
|---|
| 45 | /// [`std::fs::read_dir`]: https://doc.rust-lang.org/stable/std/fs/fn.read_dir.html | 
|---|
| 46 | pub fn path(&self) -> Option<&Path> { | 
|---|
| 47 | match self.inner { | 
|---|
| 48 | ErrorInner::Io { path: None, .. } => None, | 
|---|
| 49 | ErrorInner::Io { path: Some(ref path), .. } => Some(path), | 
|---|
| 50 | ErrorInner::Loop { ref child, .. } => Some(child), | 
|---|
| 51 | } | 
|---|
| 52 | } | 
|---|
| 53 |  | 
|---|
| 54 | /// Returns the path at which a cycle was detected. | 
|---|
| 55 | /// | 
|---|
| 56 | /// If no cycle was detected, [`None`] is returned. | 
|---|
| 57 | /// | 
|---|
| 58 | /// A cycle is detected when a directory entry is equivalent to one of | 
|---|
| 59 | /// its ancestors. | 
|---|
| 60 | /// | 
|---|
| 61 | /// To get the path to the child directory entry in the cycle, use the | 
|---|
| 62 | /// [`path`] method. | 
|---|
| 63 | /// | 
|---|
| 64 | /// [`None`]: https://doc.rust-lang.org/stable/std/option/enum.Option.html#variant.None | 
|---|
| 65 | /// [`path`]: struct.Error.html#path | 
|---|
| 66 | pub fn loop_ancestor(&self) -> Option<&Path> { | 
|---|
| 67 | match self.inner { | 
|---|
| 68 | ErrorInner::Loop { ref ancestor, .. } => Some(ancestor), | 
|---|
| 69 | _ => None, | 
|---|
| 70 | } | 
|---|
| 71 | } | 
|---|
| 72 |  | 
|---|
| 73 | /// Returns the depth at which this error occurred relative to the root. | 
|---|
| 74 | /// | 
|---|
| 75 | /// The smallest depth is `0` and always corresponds to the path given to | 
|---|
| 76 | /// the [`new`] function on [`WalkDir`]. Its direct descendents have depth | 
|---|
| 77 | /// `1`, and their descendents have depth `2`, and so on. | 
|---|
| 78 | /// | 
|---|
| 79 | /// [`new`]: struct.WalkDir.html#method.new | 
|---|
| 80 | /// [`WalkDir`]: struct.WalkDir.html | 
|---|
| 81 | pub fn depth(&self) -> usize { | 
|---|
| 82 | self.depth | 
|---|
| 83 | } | 
|---|
| 84 |  | 
|---|
| 85 | /// Inspect the original [`io::Error`] if there is one. | 
|---|
| 86 | /// | 
|---|
| 87 | /// [`None`] is returned if the [`Error`] doesn't correspond to an | 
|---|
| 88 | /// [`io::Error`]. This might happen, for example, when the error was | 
|---|
| 89 | /// produced because a cycle was found in the directory tree while | 
|---|
| 90 | /// following symbolic links. | 
|---|
| 91 | /// | 
|---|
| 92 | /// This method returns a borrowed value that is bound to the lifetime of the [`Error`]. To | 
|---|
| 93 | /// obtain an owned value, the [`into_io_error`] can be used instead. | 
|---|
| 94 | /// | 
|---|
| 95 | /// > This is the original [`io::Error`] and is _not_ the same as | 
|---|
| 96 | /// > [`impl From<Error> for std::io::Error`][impl] which contains additional context about the | 
|---|
| 97 | /// error. | 
|---|
| 98 | /// | 
|---|
| 99 | /// # Example | 
|---|
| 100 | /// | 
|---|
| 101 | /// ```rust,no_run | 
|---|
| 102 | /// use std::io; | 
|---|
| 103 | /// use std::path::Path; | 
|---|
| 104 | /// | 
|---|
| 105 | /// use walkdir::WalkDir; | 
|---|
| 106 | /// | 
|---|
| 107 | /// for entry in WalkDir::new( "foo") { | 
|---|
| 108 | ///     match entry { | 
|---|
| 109 | ///         Ok(entry) => println!( "{}", entry.path().display()), | 
|---|
| 110 | ///         Err(err) => { | 
|---|
| 111 | ///             let path = err.path().unwrap_or(Path::new( "")).display(); | 
|---|
| 112 | ///             println!( "failed to access entry {}", path); | 
|---|
| 113 | ///             if let Some(inner) = err.io_error() { | 
|---|
| 114 | ///                 match inner.kind() { | 
|---|
| 115 | ///                     io::ErrorKind::InvalidData => { | 
|---|
| 116 | ///                         println!( | 
|---|
| 117 | /// "entry contains invalid data: {}", | 
|---|
| 118 | ///                             inner) | 
|---|
| 119 | ///                     } | 
|---|
| 120 | ///                     io::ErrorKind::PermissionDenied => { | 
|---|
| 121 | ///                         println!( | 
|---|
| 122 | /// "Missing permission to read entry: {}", | 
|---|
| 123 | ///                             inner) | 
|---|
| 124 | ///                     } | 
|---|
| 125 | ///                     _ => { | 
|---|
| 126 | ///                         println!( | 
|---|
| 127 | /// "Unexpected error occurred: {}", | 
|---|
| 128 | ///                             inner) | 
|---|
| 129 | ///                     } | 
|---|
| 130 | ///                 } | 
|---|
| 131 | ///             } | 
|---|
| 132 | ///         } | 
|---|
| 133 | ///     } | 
|---|
| 134 | /// } | 
|---|
| 135 | /// ``` | 
|---|
| 136 | /// | 
|---|
| 137 | /// [`None`]: https://doc.rust-lang.org/stable/std/option/enum.Option.html#variant.None | 
|---|
| 138 | /// [`io::Error`]: https://doc.rust-lang.org/stable/std/io/struct.Error.html | 
|---|
| 139 | /// [`From`]: https://doc.rust-lang.org/stable/std/convert/trait.From.html | 
|---|
| 140 | /// [`Error`]: struct.Error.html | 
|---|
| 141 | /// [`into_io_error`]: struct.Error.html#method.into_io_error | 
|---|
| 142 | /// [impl]: struct.Error.html#impl-From%3CError%3E | 
|---|
| 143 | pub fn io_error(&self) -> Option<&io::Error> { | 
|---|
| 144 | match self.inner { | 
|---|
| 145 | ErrorInner::Io { ref err, .. } => Some(err), | 
|---|
| 146 | ErrorInner::Loop { .. } => None, | 
|---|
| 147 | } | 
|---|
| 148 | } | 
|---|
| 149 |  | 
|---|
| 150 | /// Similar to [`io_error`] except consumes self to convert to the original | 
|---|
| 151 | /// [`io::Error`] if one exists. | 
|---|
| 152 | /// | 
|---|
| 153 | /// [`io_error`]: struct.Error.html#method.io_error | 
|---|
| 154 | /// [`io::Error`]: https://doc.rust-lang.org/stable/std/io/struct.Error.html | 
|---|
| 155 | pub fn into_io_error(self) -> Option<io::Error> { | 
|---|
| 156 | match self.inner { | 
|---|
| 157 | ErrorInner::Io { err, .. } => Some(err), | 
|---|
| 158 | ErrorInner::Loop { .. } => None, | 
|---|
| 159 | } | 
|---|
| 160 | } | 
|---|
| 161 |  | 
|---|
| 162 | pub(crate) fn from_path( | 
|---|
| 163 | depth: usize, | 
|---|
| 164 | pb: PathBuf, | 
|---|
| 165 | err: io::Error, | 
|---|
| 166 | ) -> Self { | 
|---|
| 167 | Error { depth, inner: ErrorInner::Io { path: Some(pb), err } } | 
|---|
| 168 | } | 
|---|
| 169 |  | 
|---|
| 170 | pub(crate) fn from_entry(dent: &DirEntry, err: io::Error) -> Self { | 
|---|
| 171 | Error { | 
|---|
| 172 | depth: dent.depth(), | 
|---|
| 173 | inner: ErrorInner::Io { | 
|---|
| 174 | path: Some(dent.path().to_path_buf()), | 
|---|
| 175 | err, | 
|---|
| 176 | }, | 
|---|
| 177 | } | 
|---|
| 178 | } | 
|---|
| 179 |  | 
|---|
| 180 | pub(crate) fn from_io(depth: usize, err: io::Error) -> Self { | 
|---|
| 181 | Error { depth, inner: ErrorInner::Io { path: None, err } } | 
|---|
| 182 | } | 
|---|
| 183 |  | 
|---|
| 184 | pub(crate) fn from_loop( | 
|---|
| 185 | depth: usize, | 
|---|
| 186 | ancestor: &Path, | 
|---|
| 187 | child: &Path, | 
|---|
| 188 | ) -> Self { | 
|---|
| 189 | Error { | 
|---|
| 190 | depth, | 
|---|
| 191 | inner: ErrorInner::Loop { | 
|---|
| 192 | ancestor: ancestor.to_path_buf(), | 
|---|
| 193 | child: child.to_path_buf(), | 
|---|
| 194 | }, | 
|---|
| 195 | } | 
|---|
| 196 | } | 
|---|
| 197 | } | 
|---|
| 198 |  | 
|---|
| 199 | impl error::Error for Error { | 
|---|
| 200 | #[ allow(deprecated)] | 
|---|
| 201 | fn description(&self) -> &str { | 
|---|
| 202 | match self.inner { | 
|---|
| 203 | ErrorInner::Io { ref err: &Error, .. } => err.description(), | 
|---|
| 204 | ErrorInner::Loop { .. } => "file system loop found", | 
|---|
| 205 | } | 
|---|
| 206 | } | 
|---|
| 207 |  | 
|---|
| 208 | fn cause(&self) -> Option<&dyn error::Error> { | 
|---|
| 209 | self.source() | 
|---|
| 210 | } | 
|---|
| 211 |  | 
|---|
| 212 | fn source(&self) -> Option<&(dyn error::Error + 'static)> { | 
|---|
| 213 | match self.inner { | 
|---|
| 214 | ErrorInner::Io { ref err: &Error, .. } => Some(err), | 
|---|
| 215 | ErrorInner::Loop { .. } => None, | 
|---|
| 216 | } | 
|---|
| 217 | } | 
|---|
| 218 | } | 
|---|
| 219 |  | 
|---|
| 220 | impl fmt::Display for Error { | 
|---|
| 221 | fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { | 
|---|
| 222 | match self.inner { | 
|---|
| 223 | ErrorInner::Io { path: None, ref err: &Error } => err.fmt(f), | 
|---|
| 224 | ErrorInner::Io { path: Some(ref path: &PathBuf), ref err: &Error } => write!( | 
|---|
| 225 | f, | 
|---|
| 226 | "IO error for operation on {} : {} ", | 
|---|
| 227 | path.display(), | 
|---|
| 228 | err | 
|---|
| 229 | ), | 
|---|
| 230 | ErrorInner::Loop { ref ancestor: &PathBuf, ref child: &PathBuf } => write!( | 
|---|
| 231 | f, | 
|---|
| 232 | "File system loop found: \ | 
|---|
| 233 |                  {}  points to an ancestor {} ", | 
|---|
| 234 | child.display(), | 
|---|
| 235 | ancestor.display() | 
|---|
| 236 | ), | 
|---|
| 237 | } | 
|---|
| 238 | } | 
|---|
| 239 | } | 
|---|
| 240 |  | 
|---|
| 241 | impl From<Error> for io::Error { | 
|---|
| 242 | /// Convert the [`Error`] to an [`io::Error`], preserving the original | 
|---|
| 243 | /// [`Error`] as the ["inner error"]. Note that this also makes the display | 
|---|
| 244 | /// of the error include the context. | 
|---|
| 245 | /// | 
|---|
| 246 | /// This is different from [`into_io_error`] which returns the original | 
|---|
| 247 | /// [`io::Error`]. | 
|---|
| 248 | /// | 
|---|
| 249 | /// [`Error`]: struct.Error.html | 
|---|
| 250 | /// [`io::Error`]: https://doc.rust-lang.org/stable/std/io/struct.Error.html | 
|---|
| 251 | /// ["inner error"]: https://doc.rust-lang.org/std/io/struct.Error.html#method.into_inner | 
|---|
| 252 | /// [`into_io_error`]: struct.WalkDir.html#method.into_io_error | 
|---|
| 253 | fn from(walk_err: Error) -> io::Error { | 
|---|
| 254 | let kind: ErrorKind = match walk_err { | 
|---|
| 255 | Error { inner: ErrorInner::Io { ref err: &Error, .. }, .. } => err.kind(), | 
|---|
| 256 | Error { inner: ErrorInner::Loop { .. }, .. } => { | 
|---|
| 257 | io::ErrorKind::Other | 
|---|
| 258 | } | 
|---|
| 259 | }; | 
|---|
| 260 | io::Error::new(kind, error:walk_err) | 
|---|
| 261 | } | 
|---|
| 262 | } | 
|---|
| 263 |  | 
|---|