Index: base/test_file_util_win.cc |
=================================================================== |
--- base/test_file_util_win.cc (revision 0) |
+++ base/test_file_util_win.cc (revision 0) |
@@ -0,0 +1,100 @@ |
+// Copyright (c) 2008 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "base/test_file_util.h" |
+ |
+#include <windows.h> |
+ |
+#include <vector> |
+ |
+#include "base/file_util.h" |
+ |
+namespace file_util { |
+ |
+bool EvictFileFromSystemCache(const wchar_t* file) { |
+ // Request exclusive access to the file and overwrite it with no buffering. |
+ win_util::ScopedHandle hfile( |
+ CreateFile(file, GENERIC_READ | GENERIC_WRITE, 0, NULL, |
+ OPEN_EXISTING, FILE_FLAG_NO_BUFFERING, |
+ NULL)); |
+ if (!hfile) |
+ return false; |
+ |
+ // Execute in chunks. It could be optimized. We want to do few of these since |
+ // these opterations will be slow without the cache. |
+ char buffer[4096]; |
+ int total_bytes = 0; |
+ DWORD bytes_read; |
+ for (;;) { |
+ bytes_read = 0; |
+ ReadFile(hfile, buffer, sizeof(buffer), &bytes_read, NULL); |
+ if (bytes_read == 0) |
+ break; |
+ |
+ SetFilePointer(hfile, total_bytes, 0, FILE_BEGIN); |
+ if (!WriteFile(hfile, buffer, bytes_read, &bytes_read, NULL)) |
+ return false; |
+ total_bytes += bytes_read; |
+ } |
+ return true; |
+} |
+ |
+// Like CopyFileNoCache but recursively copies all files and subdirectories |
+// in the given input directory to the output directory. |
+bool CopyRecursiveDirNoCache(const std::wstring& source_dir, |
+ const std::wstring& dest_dir) { |
+ // Try to create the directory if it doesn't already exist. |
+ if (!CreateDirectory(dest_dir)) { |
+ if (GetLastError() != ERROR_ALREADY_EXISTS) |
+ return false; |
+ } |
+ |
+ std::vector<std::wstring> files_copied; |
+ |
+ std::wstring src(source_dir); |
+ file_util::AppendToPath(&src, L"*"); |
+ |
+ WIN32_FIND_DATA fd; |
+ HANDLE fh = FindFirstFile(src.c_str(), &fd); |
+ if (fh == INVALID_HANDLE_VALUE) |
+ return false; |
+ |
+ do { |
+ std::wstring cur_file(fd.cFileName); |
+ if (cur_file == L"." || cur_file == L"..") |
+ continue; // Skip these special entries. |
+ |
+ std::wstring cur_source_path(source_dir); |
+ file_util::AppendToPath(&cur_source_path, cur_file); |
+ |
+ std::wstring cur_dest_path(dest_dir); |
+ file_util::AppendToPath(&cur_dest_path, cur_file); |
+ |
+ if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) { |
+ // Recursively copy a subdirectory. We stripped "." and ".." already. |
+ if (!CopyRecursiveDirNoCache(cur_source_path, cur_dest_path)) { |
+ FindClose(fh); |
+ return false; |
+ } |
+ } else { |
+ // Copy the file. |
+ if (!::CopyFile(cur_source_path.c_str(), cur_dest_path.c_str(), false)) { |
+ FindClose(fh); |
+ return false; |
+ } |
+ |
+ // We don't check for errors from this function, often, we are copying |
+ // files that are in the repository, and they will have read-only set. |
+ // This will prevent us from evicting from the cache, but these don't |
+ // matter anyway. |
+ EvictFileFromSystemCache(cur_dest_path.c_str()); |
+ } |
+ } while (FindNextFile(fh, &fd)); |
+ |
+ FindClose(fh); |
+ return true; |
+} |
+ |
+} // namespace file_util |
+ |