Directory.EnumerateFiles => UnauthorizedAccessException

Question

.NET 4.0에는 열거 방식을 통해 스트리밍 방식으로 디렉터리에 파일을 가져 오는 멋진 새로운 방법이 있습니다.

여기에서 문제는 모든 파일을 열거하기를 원한다면 어떤 파일이나 폴더가 액세스 보호되어 있는지 미리 알지 못하고 UnauthorizedAccessException을 throw 할 수 있다는 것입니다.

한 막 단편을 실행할 수 있고, 재현이 .NET 방법은 대략 방법 복귀 문자열 배열 재귀 반복기를 구현하여 동일한 효과를 얻을 수 있었다 존재

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

전에. 그러나 새로운 .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

나는 예외를 throw하는 것이 MoveNext이라는 것을 알고 있습니다.

시퀀스를 안전 도보로 처리하고 MoveNext 예외를 무시하는 메서드를 작성하려고했습니다. 그러나 MoveNext 진도 위치가 예외를 throw 할 경우 확실하지 않아서 무한 루프가 될 수도 있습니다. 우리가 구현 세부 사항에 의존하기 때문에 나쁜 생각이기도합니다.

그러나 그것은 단지 재미 있습니다!

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이 마지막 요소에서 실패하면 후속 호출은 예외를 throw하는 대신 false을 반환합니다.
• 이 동작은 .NET Framework의 여러 버전에서 일관됩니다.
• 논리 또는 구문 오류를 만들지 않았습니다.

작동하는 경우에도 프로덕션 환경에서는 사용하지 마십시오.
하지만 실제로 그렇게되는지 궁금합니다.

Answer 3

나는 위의 내 구현 여기에 작동하지만 가져올 수 없습니다, 전 C에 그것을 테스트했습니다는 "Win7에"상자에 사용자 \ 때문에 모든 "불쾌한"DIRS이있는 경우 :

을

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

등급 : FileInfo 및 DirectoryInfo 모두

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의 답변에 따라

하지만 같은 확장 방법.

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; 
}