| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/test/test_file_util.h" | 5 #include "base/test/test_file_util.h" |
| 6 | 6 |
| 7 #include <windows.h> | 7 #include <windows.h> |
| 8 #include <aclapi.h> | 8 #include <aclapi.h> |
| 9 #include <shlwapi.h> | 9 #include <shlwapi.h> |
| 10 | 10 |
| (...skipping 118 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 129 PlatformThread::Sleep(kTimeout); | 129 PlatformThread::Sleep(kTimeout); |
| 130 } | 130 } |
| 131 return false; | 131 return false; |
| 132 } | 132 } |
| 133 | 133 |
| 134 bool EvictFileFromSystemCache(const FilePath& file) { | 134 bool EvictFileFromSystemCache(const FilePath& file) { |
| 135 // Request exclusive access to the file and overwrite it with no buffering. | 135 // Request exclusive access to the file and overwrite it with no buffering. |
| 136 base::win::ScopedHandle file_handle( | 136 base::win::ScopedHandle file_handle( |
| 137 CreateFile(file.value().c_str(), GENERIC_READ | GENERIC_WRITE, 0, NULL, | 137 CreateFile(file.value().c_str(), GENERIC_READ | GENERIC_WRITE, 0, NULL, |
| 138 OPEN_EXISTING, FILE_FLAG_NO_BUFFERING, NULL)); | 138 OPEN_EXISTING, FILE_FLAG_NO_BUFFERING, NULL)); |
| 139 if (!file_handle) | 139 if (!file_handle.IsValid()) |
| 140 return false; | 140 return false; |
| 141 | 141 |
| 142 // Get some attributes to restore later. | 142 // Get some attributes to restore later. |
| 143 BY_HANDLE_FILE_INFORMATION bhi = {0}; | 143 BY_HANDLE_FILE_INFORMATION bhi = {0}; |
| 144 CHECK(::GetFileInformationByHandle(file_handle, &bhi)); | 144 CHECK(::GetFileInformationByHandle(file_handle.Get(), &bhi)); |
| 145 | 145 |
| 146 // Execute in chunks. It could be optimized. We want to do few of these since | 146 // Execute in chunks. It could be optimized. We want to do few of these since |
| 147 // these operations will be slow without the cache. | 147 // these operations will be slow without the cache. |
| 148 | 148 |
| 149 // Allocate a buffer for the reads and the writes. | 149 // Allocate a buffer for the reads and the writes. |
| 150 char* buffer = reinterpret_cast<char*>(VirtualAlloc(NULL, | 150 char* buffer = reinterpret_cast<char*>(VirtualAlloc(NULL, |
| 151 kOneMB, | 151 kOneMB, |
| 152 MEM_COMMIT | MEM_RESERVE, | 152 MEM_COMMIT | MEM_RESERVE, |
| 153 PAGE_READWRITE)); | 153 PAGE_READWRITE)); |
| 154 | 154 |
| 155 // If the file size isn't a multiple of kOneMB, we'll need special | 155 // If the file size isn't a multiple of kOneMB, we'll need special |
| 156 // processing. | 156 // processing. |
| 157 bool file_is_aligned = true; | 157 bool file_is_aligned = true; |
| 158 int total_bytes = 0; | 158 int total_bytes = 0; |
| 159 DWORD bytes_read, bytes_written; | 159 DWORD bytes_read, bytes_written; |
| 160 for (;;) { | 160 for (;;) { |
| 161 bytes_read = 0; | 161 bytes_read = 0; |
| 162 ::ReadFile(file_handle, buffer, kOneMB, &bytes_read, NULL); | 162 ::ReadFile(file_handle.Get(), buffer, kOneMB, &bytes_read, NULL); |
| 163 if (bytes_read == 0) | 163 if (bytes_read == 0) |
| 164 break; | 164 break; |
| 165 | 165 |
| 166 if (bytes_read < kOneMB) { | 166 if (bytes_read < kOneMB) { |
| 167 // Zero out the remaining part of the buffer. | 167 // Zero out the remaining part of the buffer. |
| 168 // WriteFile will fail if we provide a buffer size that isn't a | 168 // WriteFile will fail if we provide a buffer size that isn't a |
| 169 // sector multiple, so we'll have to write the entire buffer with | 169 // sector multiple, so we'll have to write the entire buffer with |
| 170 // padded zeros and then use SetEndOfFile to truncate the file. | 170 // padded zeros and then use SetEndOfFile to truncate the file. |
| 171 ZeroMemory(buffer + bytes_read, kOneMB - bytes_read); | 171 ZeroMemory(buffer + bytes_read, kOneMB - bytes_read); |
| 172 file_is_aligned = false; | 172 file_is_aligned = false; |
| 173 } | 173 } |
| 174 | 174 |
| 175 // Move back to the position we just read from. | 175 // Move back to the position we just read from. |
| 176 // Note that SetFilePointer will also fail if total_bytes isn't sector | 176 // Note that SetFilePointer will also fail if total_bytes isn't sector |
| 177 // aligned, but that shouldn't happen here. | 177 // aligned, but that shouldn't happen here. |
| 178 DCHECK((total_bytes % kOneMB) == 0); | 178 DCHECK((total_bytes % kOneMB) == 0); |
| 179 SetFilePointer(file_handle, total_bytes, NULL, FILE_BEGIN); | 179 SetFilePointer(file_handle.Get(), total_bytes, NULL, FILE_BEGIN); |
| 180 if (!::WriteFile(file_handle, buffer, kOneMB, &bytes_written, NULL) || | 180 if (!::WriteFile(file_handle.Get(), buffer, kOneMB, &bytes_written, NULL) || |
| 181 bytes_written != kOneMB) { | 181 bytes_written != kOneMB) { |
| 182 BOOL freed = VirtualFree(buffer, 0, MEM_RELEASE); | 182 BOOL freed = VirtualFree(buffer, 0, MEM_RELEASE); |
| 183 DCHECK(freed); | 183 DCHECK(freed); |
| 184 NOTREACHED(); | 184 NOTREACHED(); |
| 185 return false; | 185 return false; |
| 186 } | 186 } |
| 187 | 187 |
| 188 total_bytes += bytes_read; | 188 total_bytes += bytes_read; |
| 189 | 189 |
| 190 // If this is false, then we just processed the last portion of the file. | 190 // If this is false, then we just processed the last portion of the file. |
| 191 if (!file_is_aligned) | 191 if (!file_is_aligned) |
| 192 break; | 192 break; |
| 193 } | 193 } |
| 194 | 194 |
| 195 BOOL freed = VirtualFree(buffer, 0, MEM_RELEASE); | 195 BOOL freed = VirtualFree(buffer, 0, MEM_RELEASE); |
| 196 DCHECK(freed); | 196 DCHECK(freed); |
| 197 | 197 |
| 198 if (!file_is_aligned) { | 198 if (!file_is_aligned) { |
| 199 // The size of the file isn't a multiple of 1 MB, so we'll have | 199 // The size of the file isn't a multiple of 1 MB, so we'll have |
| 200 // to open the file again, this time without the FILE_FLAG_NO_BUFFERING | 200 // to open the file again, this time without the FILE_FLAG_NO_BUFFERING |
| 201 // flag and use SetEndOfFile to mark EOF. | 201 // flag and use SetEndOfFile to mark EOF. |
| 202 file_handle.Set(NULL); | 202 file_handle.Set(NULL); |
| 203 file_handle.Set(CreateFile(file.value().c_str(), GENERIC_WRITE, 0, NULL, | 203 file_handle.Set(CreateFile(file.value().c_str(), GENERIC_WRITE, 0, NULL, |
| 204 OPEN_EXISTING, 0, NULL)); | 204 OPEN_EXISTING, 0, NULL)); |
| 205 CHECK_NE(SetFilePointer(file_handle, total_bytes, NULL, FILE_BEGIN), | 205 CHECK_NE(SetFilePointer(file_handle.Get(), total_bytes, NULL, FILE_BEGIN), |
| 206 INVALID_SET_FILE_POINTER); | 206 INVALID_SET_FILE_POINTER); |
| 207 CHECK(::SetEndOfFile(file_handle)); | 207 CHECK(::SetEndOfFile(file_handle.Get())); |
| 208 } | 208 } |
| 209 | 209 |
| 210 // Restore the file attributes. | 210 // Restore the file attributes. |
| 211 CHECK(::SetFileTime(file_handle, &bhi.ftCreationTime, &bhi.ftLastAccessTime, | 211 CHECK(::SetFileTime(file_handle.Get(), &bhi.ftCreationTime, |
| 212 &bhi.ftLastWriteTime)); | 212 &bhi.ftLastAccessTime, &bhi.ftLastWriteTime)); |
| 213 | 213 |
| 214 return true; | 214 return true; |
| 215 } | 215 } |
| 216 | 216 |
| 217 // Checks if the volume supports Alternate Data Streams. This is required for | 217 // Checks if the volume supports Alternate Data Streams. This is required for |
| 218 // the Zone Identifier implementation. | 218 // the Zone Identifier implementation. |
| 219 bool VolumeSupportsADS(const FilePath& path) { | 219 bool VolumeSupportsADS(const FilePath& path) { |
| 220 wchar_t drive[MAX_PATH] = {0}; | 220 wchar_t drive[MAX_PATH] = {0}; |
| 221 wcscpy_s(drive, MAX_PATH, path.value().c_str()); | 221 wcscpy_s(drive, MAX_PATH, path.value().c_str()); |
| 222 | 222 |
| (...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 273 DCHECK(info_ != NULL); | 273 DCHECK(info_ != NULL); |
| 274 DCHECK_NE(0u, length_); | 274 DCHECK_NE(0u, length_); |
| 275 } | 275 } |
| 276 | 276 |
| 277 FilePermissionRestorer::~FilePermissionRestorer() { | 277 FilePermissionRestorer::~FilePermissionRestorer() { |
| 278 if (!RestorePermissionInfo(path_, info_, length_)) | 278 if (!RestorePermissionInfo(path_, info_, length_)) |
| 279 NOTREACHED(); | 279 NOTREACHED(); |
| 280 } | 280 } |
| 281 | 281 |
| 282 } // namespace base | 282 } // namespace base |
| OLD | NEW |