sandbox/win/tools/finder/finder_fs.cc

   1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "sandbox/win/src/restricted_token.h"
   6 #include "sandbox/win/src/restricted_token_utils.h"
   7 #include "sandbox/win/tools/finder/finder.h"
   8
   9 DWORD Finder::ParseFileSystem(ATL::CString directory) {
  10   WIN32_FIND_DATA find_data;
  11   HANDLE find;
  12
  13   //Search for items in the directory.
  14   ATL::CString name_to_search = directory + L"\\*";
  15   find = ::FindFirstFile(name_to_search, &find_data);
  16   if (INVALID_HANDLE_VALUE == find) {
  17     DWORD error = ::GetLastError();
  18     Output(FS_ERR, error, directory);
  19     filesystem_stats_[BROKEN]++;
  20     return error;
  21   }
  22
  23   // parse all files or folders.
  24   do {
  25     if (_tcscmp(find_data.cFileName, L".") == 0 ||
  26         _tcscmp(find_data.cFileName, L"..") == 0)
  27       continue;
  28
  29     ATL::CString complete_name = directory + L"\\" + find_data.cFileName;
  30     TestFileAccess(complete_name);
  31
  32     // Call recursively the function if the path found is a directory.
  33     if ((find_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0) {
  34       ParseFileSystem(complete_name);
  35     }
  36   } while (::FindNextFile(find, &find_data) != 0);
  37
  38   DWORD err_code = ::GetLastError();
  39   ::FindClose(find);
  40
  41   if (ERROR_NO_MORE_FILES != err_code) {
  42     Output(FS_ERR, err_code, directory);
  43     filesystem_stats_[BROKEN]++;
  44     return err_code;
  45   }
  46
  47   return ERROR_SUCCESS;
  48 }
  49
  50 DWORD Finder::TestFileAccess(ATL::CString name) {
  51   Impersonater impersonate(token_handle_);
  52
  53   filesystem_stats_[PARSE]++;
  54
  55   HANDLE file;
  56   if (access_type_ & kTestForAll) {
  57     file = ::CreateFile(name.GetBuffer(),
  58                         GENERIC_ALL,
  59                         FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
  60                         NULL,
  61                         OPEN_EXISTING,
  62                         FILE_ATTRIBUTE_NORMAL,
  63                         NULL);
  64
  65     if (file != INVALID_HANDLE_VALUE) {
  66       filesystem_stats_[ALL]++;
  67       Output(FS, L"R/W", name.GetBuffer());
  68       ::CloseHandle(file);
  69       return GENERIC_ALL;
  70     } else if (::GetLastError() != ERROR_ACCESS_DENIED) {
  71       Output(FS_ERR, GetLastError(), name);
  72       filesystem_stats_[BROKEN]++;
  73     }
  74   }
  75
  76   if (access_type_ & kTestForWrite) {
  77     file = ::CreateFile(name.GetBuffer(),
  78                         GENERIC_WRITE,
  79                         FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
  80                         NULL,
  81                         OPEN_EXISTING,
  82                         FILE_ATTRIBUTE_NORMAL,
  83                         NULL);
  84
  85     if (file != INVALID_HANDLE_VALUE) {
  86       filesystem_stats_[WRITE]++;
  87       Output(FS, L"W", name);
  88       ::CloseHandle(file);
  89       return GENERIC_WRITE;
  90     } else if (::GetLastError() != ERROR_ACCESS_DENIED) {
  91       Output(FS_ERR, ::GetLastError(), name);
  92       filesystem_stats_[BROKEN]++;
  93     }
  94   }
  95
  96   if (access_type_ & kTestForRead) {
  97     file = ::CreateFile(name.GetBuffer(),
  98                         GENERIC_READ,
  99                         FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
 100                         NULL,
 101                         OPEN_EXISTING,
 102                         FILE_ATTRIBUTE_NORMAL,
 103                         NULL);
 104
 105     if (file != INVALID_HANDLE_VALUE) {
 106       filesystem_stats_[READ]++;
 107       Output(FS, L"R", name);
 108       ::CloseHandle(file);
 109       return GENERIC_READ;
 110     } else if (::GetLastError() != ERROR_ACCESS_DENIED) {
 111       Output(FS_ERR, GetLastError(), name);
 112       filesystem_stats_[BROKEN]++;
 113     }
 114   }
 115
 116   return 0;
 117 }