sandbox/win/src/win_utils_unittest.cc

   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
  12 TEST(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
  54 TEST(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
  85 TEST(WinUtils, IsPipe) {
  86   using sandbox::IsPipe;
  87
  88   base::string16 pipe_name = L"\\??\\pipe\\mypipe";
  89   EXPECT_TRUE(IsPipe(pipe_name));
  90
  91   pipe_name = L"\\??\\PiPe\\mypipe";
  92   EXPECT_TRUE(IsPipe(pipe_name));
  93
  94   pipe_name = L"\\??\\pipe";
  95   EXPECT_FALSE(IsPipe(pipe_name));
  96
  97   pipe_name = L"\\??\\_pipe_\\mypipe";
  98   EXPECT_FALSE(IsPipe(pipe_name));
  99
 100   pipe_name = L"\\??\\ABCD\\mypipe";
 101   EXPECT_FALSE(IsPipe(pipe_name));
 102
 103
 104   // Written as two strings to prevent trigraph '?' '?' '/'.
 105   pipe_name = L"/?" L"?/pipe/mypipe";
 106   EXPECT_FALSE(IsPipe(pipe_name));
 107
 108   pipe_name = L"\\XX\\pipe\\mypipe";
 109   EXPECT_FALSE(IsPipe(pipe_name));
 110
 111   pipe_name = L"\\Device\\NamedPipe\\mypipe";
 112   EXPECT_FALSE(IsPipe(pipe_name));
 113 }