| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "chrome_elf/create_file/chrome_create_file.h" | 5 #include "chrome_elf/create_file/chrome_create_file.h" |
| 6 | 6 |
| 7 #include <windows.h> | 7 #include <windows.h> |
| 8 | 8 |
| 9 #include <bitset> | 9 #include <bitset> |
| 10 #include <string> | 10 #include <string> |
| (...skipping 224 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 235 file_handle.Set(CreateFileNTDLL(path.value().c_str(), | 235 file_handle.Set(CreateFileNTDLL(path.value().c_str(), |
| 236 GENERIC_WRITE, | 236 GENERIC_WRITE, |
| 237 FILE_SHARE_READ, | 237 FILE_SHARE_READ, |
| 238 NULL, | 238 NULL, |
| 239 CREATE_ALWAYS, | 239 CREATE_ALWAYS, |
| 240 FILE_ATTRIBUTE_NORMAL | flag, | 240 FILE_ATTRIBUTE_NORMAL | flag, |
| 241 NULL)); | 241 NULL)); |
| 242 } | 242 } |
| 243 | 243 |
| 244 | 244 |
| 245 EXPECT_FALSE(file_handle == INVALID_HANDLE_VALUE); | 245 EXPECT_TRUE(file_handle.IsValid()); |
| 246 ::WriteFile(file_handle, kTestData, buffer_size, &bytes_written, NULL); | 246 ::WriteFile(file_handle.Get(), kTestData, buffer_size, &bytes_written, |
| 247 NULL); |
| 247 EXPECT_EQ(buffer_size, bytes_written); | 248 EXPECT_EQ(buffer_size, bytes_written); |
| 248 } | 249 } |
| 249 | 250 |
| 250 void DoReadCheck(const base::FilePath& path, DWORD flag, bool is_system) { | 251 void DoReadCheck(const base::FilePath& path, DWORD flag, bool is_system) { |
| 251 base::win::ScopedHandle file_handle; | 252 base::win::ScopedHandle file_handle; |
| 252 const char kTestData[] = "0123456789"; | 253 const char kTestData[] = "0123456789"; |
| 253 int buffer_size = sizeof(kTestData) - 1; | 254 int buffer_size = sizeof(kTestData) - 1; |
| 254 DWORD bytes_read; | 255 DWORD bytes_read; |
| 255 char read_buffer[10]; | 256 char read_buffer[10]; |
| 256 | 257 |
| 257 if (is_system) { | 258 if (is_system) { |
| 258 file_handle.Set(::CreateFileW(path.value().c_str(), | 259 file_handle.Set(::CreateFileW(path.value().c_str(), |
| 259 GENERIC_READ, | 260 GENERIC_READ, |
| 260 0, | 261 0, |
| 261 NULL, | 262 NULL, |
| 262 OPEN_ALWAYS, | 263 OPEN_ALWAYS, |
| 263 FILE_ATTRIBUTE_NORMAL | flag, | 264 FILE_ATTRIBUTE_NORMAL | flag, |
| 264 NULL)); | 265 NULL)); |
| 265 } else { | 266 } else { |
| 266 file_handle.Set(CreateFileNTDLL(path.value().c_str(), | 267 file_handle.Set(CreateFileNTDLL(path.value().c_str(), |
| 267 GENERIC_READ, | 268 GENERIC_READ, |
| 268 0, | 269 0, |
| 269 NULL, | 270 NULL, |
| 270 OPEN_ALWAYS, | 271 OPEN_ALWAYS, |
| 271 FILE_ATTRIBUTE_NORMAL | flag, | 272 FILE_ATTRIBUTE_NORMAL | flag, |
| 272 NULL)); | 273 NULL)); |
| 273 } | 274 } |
| 274 | 275 |
| 275 EXPECT_FALSE(file_handle == INVALID_HANDLE_VALUE); | 276 EXPECT_TRUE(file_handle.IsValid()); |
| 276 ::ReadFile(file_handle, read_buffer, buffer_size, &bytes_read, NULL); | 277 ::ReadFile(file_handle.Get(), read_buffer, buffer_size, &bytes_read, NULL); |
| 277 EXPECT_EQ(buffer_size, bytes_read); | 278 EXPECT_EQ(buffer_size, bytes_read); |
| 278 EXPECT_EQ(0, memcmp(kTestData, read_buffer, bytes_read)); | 279 EXPECT_EQ(0, memcmp(kTestData, read_buffer, bytes_read)); |
| 279 } | 280 } |
| 280 | 281 |
| 281 void RunChecks(DWORD flag, bool check_reads) { | 282 void RunChecks(DWORD flag, bool check_reads) { |
| 282 // Make sure we can write to this file handle when called via the system. | 283 // Make sure we can write to this file handle when called via the system. |
| 283 base::FilePath junk_path_1 = temp_dir_.path().Append(L"junk_1.txt"); | 284 base::FilePath junk_path_1 = temp_dir_.path().Append(L"junk_1.txt"); |
| 284 base::FilePath junk_path_2 = temp_dir_.path().Append(L"junk_2.txt"); | 285 base::FilePath junk_path_2 = temp_dir_.path().Append(L"junk_2.txt"); |
| 285 DoWriteCheck(junk_path_1, flag, true); | 286 DoWriteCheck(junk_path_1, flag, true); |
| 286 DoWriteCheck(junk_path_2, flag, false); | 287 DoWriteCheck(junk_path_2, flag, false); |
| (...skipping 112 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 399 DoReadCheck(file_name, FILE_ATTRIBUTE_NORMAL, false); | 400 DoReadCheck(file_name, FILE_ATTRIBUTE_NORMAL, false); |
| 400 } | 401 } |
| 401 | 402 |
| 402 TEST_F(ChromeCreateFileTest, ReadWriteFromThunk) { | 403 TEST_F(ChromeCreateFileTest, ReadWriteFromThunk) { |
| 403 base::FilePath file_name = temp_dir_.path().Append(L"some_file.txt"); | 404 base::FilePath file_name = temp_dir_.path().Append(L"some_file.txt"); |
| 404 DoWriteCheck(file_name, FILE_ATTRIBUTE_NORMAL, false); | 405 DoWriteCheck(file_name, FILE_ATTRIBUTE_NORMAL, false); |
| 405 DoReadCheck(file_name, FILE_ATTRIBUTE_NORMAL, false); | 406 DoReadCheck(file_name, FILE_ATTRIBUTE_NORMAL, false); |
| 406 } | 407 } |
| 407 | 408 |
| 408 } // namespace | 409 } // namespace |
| OLD | NEW |