Directory.EnumerateFiles => UnauthorizedAccessException

Question

有在.NET 4.0中一个不错的新方法，通过列举的方式流获取目录中的文件。

这里的问题是，如果一个人希望枚举所有文件中的一个可能不会提前知道哪些文件或文件夹的访问保护，可以抛出UnauthorizedAccessException。

要重现，一个可以运行这个片段：

foreach (var file in Directory.EnumerateFiles(@"c:\", "*", SearchOption.AllDirectories)) 
{ 
    // whatever 
} 

在此之前，.NET方法存在，有可能大致实现通过实现字符串数组返回方法上的递归迭代器相同的效果。但它不像新的.NET方法那么懒。

所以，我们要做什么呢？在使用这种方法时，UnauthorizedAccessException可以被抑制还是生活中的事实？

在我看来，该方法应该有一个接受过载的作用，以应付任何异常。

Answer 1

这个问题与上述答案是没有照顾子目录中的异常。这将是一个更好的方式来处理这些异常，所以你除了那些所有子目录中得到的所有文件抛出访问异常：

/// <summary> 
    /// A safe way to get all the files in a directory and sub directory without crashing on UnauthorizedException or PathTooLongException 
    /// </summary> 
    /// <param name="rootPath">Starting directory</param> 
    /// <param name="patternMatch">Filename pattern match</param> 
    /// <param name="searchOption">Search subdirectories or only top level directory for files</param> 
    /// <returns>List of files</returns> 
    public static IEnumerable<string> GetDirectoryFiles(string rootPath, string patternMatch, SearchOption searchOption) 
    { 
     var foundFiles = Enumerable.Empty<string>(); 

     if (searchOption == SearchOption.AllDirectories) 
     { 
      try 
      { 
       IEnumerable<string> subDirs = Directory.EnumerateDirectories(rootPath); 
       foreach (string dir in subDirs) 
       { 
        foundFiles = foundFiles.Concat(GetDirectoryFiles(dir, patternMatch, searchOption)); // Add files in subdirectories recursively to the list 
       } 
      } 
      catch (UnauthorizedAccessException) { } 
      catch (PathTooLongException) {} 
     } 

     try 
     { 
      foundFiles = foundFiles.Concat(Directory.EnumerateFiles(rootPath, patternMatch)); // Add files from the current directory 
     } 
     catch (UnauthorizedAccessException) { } 

     return foundFiles; 
    } 

Answer 2

我知道这是MoveNext抛出异常。

我试图编写一个方法，安全地遍历一个序列并尝试忽略MoveNext异常。但是我不确定是否MoveNext在引发异常时前进，所以这可能是无限循环。这也是一个坏主意，因为我们会依赖实现细节。

但它只是太好玩了！

public static IEnumerable<T> SafeWalk<T> (this IEnumerable<T> source) 
{ 
    var enumerator = source.GetEnumerator(); 
    bool? hasCurrent = null; 

    do { 
     try { 
      hasCurrent = enumerator.MoveNext(); 
     } catch { 
      hasCurrent = null; // we're not sure 
     } 

     if (hasCurrent ?? false) // if not sure, do not return value 
      yield return enumerator.Current; 

    } while (hasCurrent ?? true); // if not sure, continue walking 
} 

foreach (var file in Directory.EnumerateFiles("c:\\", "*", SearchOption.AllDirectories) 
           .SafeWalk()) 
{ 
    // ... 
} 

这只会如果满足下列条件，有关框架的实现这个迭代器的真实工作（见FileSystemEnumerableIterator<TSource>在反射器供参考）：

• MoveNext前进的位置，当它失败;
• 当MoveNext在最后一个元素上失败时，后续调用将返回false而不是抛出异常;
• 此行为是不同版本的.NET Framework的一致;
• 我没有犯任何逻辑或语法错误。

即使它工作，请不要在生产中使用它！
但我真的想知道它是否。

Answer 3

我不能得到上述工作，但这里是我的实现，我已经测试了C：\用户在“Win7的”框，因为如果有这方面的“讨厌”迪尔斯：

SafeWalk.EnumerateFiles(@"C:\users", "*.jpg", SearchOption.AllDirectories).Take(10) 

类：

public static class SafeWalk 
{ 
    public static IEnumerable<string> EnumerateFiles(string path, string searchPattern, SearchOption searchOpt) 
    { 
     try 
     { 
      var dirFiles = Enumerable.Empty<string>(); 
      if(searchOpt == SearchOption.AllDirectories) 
      { 
       dirFiles = Directory.EnumerateDirectories(path) 
            .SelectMany(x => EnumerateFiles(x, searchPattern, searchOpt)); 
      } 
      return dirFiles.Concat(Directory.EnumerateFiles(path, searchPattern)); 
     } 
     catch(UnauthorizedAccessException ex) 
     { 
      return Enumerable.Empty<string>(); 
     } 
    } 
} 

Answer 4

基于strudso的答案，但作为两个FileInfo和DirectoryInfo扩展方法。

public static IEnumerable<FileInfo> EnumerateFilesSafe(this DirectoryInfo dir, string filter = "*.*", SearchOption opt = SearchOption.TopDirectoryOnly) 
{ 
    var retval = Enumerable.Empty<FileInfo>(); 

    try { retval = dir.EnumerateFiles(filter); } 
    catch { Debug.WriteLine("{0} Inaccessable.", dir.FullName); } 

    if (opt == SearchOption.AllDirectories) 
     retval = retval.Concat(dir.EnumerateDirectoriesSafe(opt: opt).SelectMany(x => x.EnumerateFilesSafe(filter, opt))); 

    return retval; 
} 

public static IEnumerable<DirectoryInfo> EnumerateDirectoriesSafe(this DirectoryInfo dir, string filter = "*.*", SearchOption opt = SearchOption.TopDirectoryOnly) 
{ 
    var retval = Enumerable.Empty<DirectoryInfo>(); 

    try { retval = dir.EnumerateDirectories(filter); } 
    catch { Debug.WriteLine("{0} Inaccessable.", dir.FullName); } 

    if (opt == SearchOption.AllDirectories) 
     retval = retval.Concat(retval.SelectMany(x => x.EnumerateDirectoriesSafe(filter, opt))); 

    return retval; 
}