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