| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 <windows.h> | 5 #include <windows.h> |
| 6 | 6 |
| 7 #include "base/win/scoped_handle.h" | 7 #include "base/win/scoped_handle.h" |
| 8 #include "sandbox/win/src/win_utils.h" | 8 #include "sandbox/win/src/win_utils.h" |
| 9 #include "sandbox/win/tests/common/test_utils.h" | 9 #include "sandbox/win/tests/common/test_utils.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
| 11 | 11 |
| 12 TEST(WinUtils, IsReparsePoint) { | 12 TEST(WinUtils, IsReparsePoint) { |
| 13 using sandbox::IsReparsePoint; | 13 using sandbox::IsReparsePoint; |
| 14 | 14 |
| 15 // Create a temp file because we need write access to it. | 15 // Create a temp file because we need write access to it. |
| 16 wchar_t temp_directory[MAX_PATH]; | 16 wchar_t temp_directory[MAX_PATH]; |
| 17 wchar_t my_folder[MAX_PATH]; | 17 wchar_t my_folder[MAX_PATH]; |
| 18 ASSERT_NE(::GetTempPath(MAX_PATH, temp_directory), 0u); | 18 ASSERT_NE(::GetTempPath(MAX_PATH, temp_directory), 0u); |
| 19 ASSERT_NE(::GetTempFileName(temp_directory, L"test", 0, my_folder), 0u); | 19 ASSERT_NE(::GetTempFileName(temp_directory, L"test", 0, my_folder), 0u); |
| 20 | 20 |
| 21 // Delete the file and create a directory instead. | 21 // Delete the file and create a directory instead. |
| 22 ASSERT_TRUE(::DeleteFile(my_folder)); | 22 ASSERT_TRUE(::DeleteFile(my_folder)); |
| 23 ASSERT_TRUE(::CreateDirectory(my_folder, NULL)); | 23 ASSERT_TRUE(::CreateDirectory(my_folder, NULL)); |
| 24 | 24 |
| 25 bool result = true; | 25 bool result = true; |
| 26 EXPECT_EQ(ERROR_SUCCESS, IsReparsePoint(my_folder, &result)); | 26 EXPECT_EQ(ERROR_SUCCESS, IsReparsePoint(my_folder, &result)); |
| 27 EXPECT_FALSE(result); | 27 EXPECT_FALSE(result); |
| 28 | 28 |
| 29 // We have to fix Bug 32224 to pass this test. | 29 // We have to fix Bug 32224 to pass this test. |
| 30 std::wstring not_found = std::wstring(my_folder) + L"\\foo\\bar"; | 30 base::string16 not_found = base::string16(my_folder) + L"\\foo\\bar"; |
| 31 // EXPECT_EQ(ERROR_PATH_NOT_FOUND, IsReparsePoint(not_found, &result)); | 31 // EXPECT_EQ(ERROR_PATH_NOT_FOUND, IsReparsePoint(not_found, &result)); |
| 32 | 32 |
| 33 std::wstring new_file = std::wstring(my_folder) + L"\\foo"; | 33 base::string16 new_file = base::string16(my_folder) + L"\\foo"; |
| 34 EXPECT_EQ(ERROR_SUCCESS, IsReparsePoint(new_file, &result)); | 34 EXPECT_EQ(ERROR_SUCCESS, IsReparsePoint(new_file, &result)); |
| 35 EXPECT_FALSE(result); | 35 EXPECT_FALSE(result); |
| 36 | 36 |
| 37 // Replace the directory with a reparse point to %temp%. | 37 // Replace the directory with a reparse point to %temp%. |
| 38 HANDLE dir = ::CreateFile(my_folder, FILE_ALL_ACCESS, | 38 HANDLE dir = ::CreateFile(my_folder, FILE_ALL_ACCESS, |
| 39 FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, | 39 FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, |
| 40 OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL); | 40 OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL); |
| 41 EXPECT_NE(INVALID_HANDLE_VALUE, dir); | 41 EXPECT_NE(INVALID_HANDLE_VALUE, dir); |
| 42 | 42 |
| 43 std::wstring temp_dir_nt = std::wstring(L"\\??\\") + temp_directory; | 43 base::string16 temp_dir_nt = base::string16(L"\\??\\") + temp_directory; |
| 44 EXPECT_TRUE(SetReparsePoint(dir, temp_dir_nt.c_str())); | 44 EXPECT_TRUE(SetReparsePoint(dir, temp_dir_nt.c_str())); |
| 45 | 45 |
| 46 EXPECT_EQ(ERROR_SUCCESS, IsReparsePoint(new_file, &result)); | 46 EXPECT_EQ(ERROR_SUCCESS, IsReparsePoint(new_file, &result)); |
| 47 EXPECT_TRUE(result); | 47 EXPECT_TRUE(result); |
| 48 | 48 |
| 49 EXPECT_TRUE(DeleteReparsePoint(dir)); | 49 EXPECT_TRUE(DeleteReparsePoint(dir)); |
| 50 EXPECT_TRUE(::CloseHandle(dir)); | 50 EXPECT_TRUE(::CloseHandle(dir)); |
| 51 EXPECT_TRUE(::RemoveDirectory(my_folder)); | 51 EXPECT_TRUE(::RemoveDirectory(my_folder)); |
| 52 } | 52 } |
| 53 | 53 |
| 54 TEST(WinUtils, SameObject) { | 54 TEST(WinUtils, SameObject) { |
| 55 using sandbox::SameObject; | 55 using sandbox::SameObject; |
| 56 | 56 |
| 57 // Create a temp file because we need write access to it. | 57 // Create a temp file because we need write access to it. |
| 58 wchar_t temp_directory[MAX_PATH]; | 58 wchar_t temp_directory[MAX_PATH]; |
| 59 wchar_t my_folder[MAX_PATH]; | 59 wchar_t my_folder[MAX_PATH]; |
| 60 ASSERT_NE(::GetTempPath(MAX_PATH, temp_directory), 0u); | 60 ASSERT_NE(::GetTempPath(MAX_PATH, temp_directory), 0u); |
| 61 ASSERT_NE(::GetTempFileName(temp_directory, L"test", 0, my_folder), 0u); | 61 ASSERT_NE(::GetTempFileName(temp_directory, L"test", 0, my_folder), 0u); |
| 62 | 62 |
| 63 // Delete the file and create a directory instead. | 63 // Delete the file and create a directory instead. |
| 64 ASSERT_TRUE(::DeleteFile(my_folder)); | 64 ASSERT_TRUE(::DeleteFile(my_folder)); |
| 65 ASSERT_TRUE(::CreateDirectory(my_folder, NULL)); | 65 ASSERT_TRUE(::CreateDirectory(my_folder, NULL)); |
| 66 | 66 |
| 67 std::wstring folder(my_folder); | 67 base::string16 folder(my_folder); |
| 68 std::wstring file_name = folder + L"\\foo.txt"; | 68 base::string16 file_name = folder + L"\\foo.txt"; |
| 69 const ULONG kSharing = FILE_SHARE_WRITE | FILE_SHARE_READ | FILE_SHARE_DELETE; | 69 const ULONG kSharing = FILE_SHARE_WRITE | FILE_SHARE_READ | FILE_SHARE_DELETE; |
| 70 base::win::ScopedHandle file(CreateFile( | 70 base::win::ScopedHandle file(CreateFile( |
| 71 file_name.c_str(), GENERIC_WRITE, kSharing, NULL, CREATE_ALWAYS, | 71 file_name.c_str(), GENERIC_WRITE, kSharing, NULL, CREATE_ALWAYS, |
| 72 FILE_FLAG_DELETE_ON_CLOSE, NULL)); | 72 FILE_FLAG_DELETE_ON_CLOSE, NULL)); |
| 73 | 73 |
| 74 EXPECT_TRUE(file.IsValid()); | 74 EXPECT_TRUE(file.IsValid()); |
| 75 std::wstring file_name_nt1 = std::wstring(L"\\??\\") + file_name; | 75 base::string16 file_name_nt1 = base::string16(L"\\??\\") + file_name; |
| 76 std::wstring file_name_nt2 = std::wstring(L"\\??\\") + folder + L"\\FOO.txT"; | 76 base::string16 file_name_nt2 = |
| 77 base::string16(L"\\??\\") + folder + L"\\FOO.txT"; |
| 77 EXPECT_TRUE(SameObject(file.Get(), file_name_nt1.c_str())); | 78 EXPECT_TRUE(SameObject(file.Get(), file_name_nt1.c_str())); |
| 78 EXPECT_TRUE(SameObject(file.Get(), file_name_nt2.c_str())); | 79 EXPECT_TRUE(SameObject(file.Get(), file_name_nt2.c_str())); |
| 79 | 80 |
| 80 file.Close(); | 81 file.Close(); |
| 81 EXPECT_TRUE(::RemoveDirectory(my_folder)); | 82 EXPECT_TRUE(::RemoveDirectory(my_folder)); |
| 82 } | 83 } |
| OLD | NEW |