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   EXPECT_EQ(ERROR_NOT_A_REPARSE_POINT, IsReparsePoint(my_folder));
  26
  27   base::string16 not_found = base::string16(my_folder) + L"\\foo\\bar";
  28   EXPECT_EQ(ERROR_NOT_A_REPARSE_POINT, IsReparsePoint(not_found));
  29
  30   base::string16 new_file = base::string16(my_folder) + L"\\foo";
  31   EXPECT_EQ(ERROR_NOT_A_REPARSE_POINT, IsReparsePoint(new_file));
  32
  33   // Replace the directory with a reparse point to %temp%.
  34   HANDLE dir = ::CreateFile(my_folder, FILE_ALL_ACCESS,
  35                             FILE_SHARE_READ | FILE_SHARE_WRITE, NULL,
  36                             OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
  37   EXPECT_NE(INVALID_HANDLE_VALUE, dir);
  38
  39   base::string16 temp_dir_nt = base::string16(L"\\??\\") + temp_directory;
  40   EXPECT_TRUE(SetReparsePoint(dir, temp_dir_nt.c_str()));
  41
  42   EXPECT_EQ(ERROR_SUCCESS, IsReparsePoint(new_file));
  43
  44   EXPECT_TRUE(DeleteReparsePoint(dir));
  45   EXPECT_TRUE(::CloseHandle(dir));
  46   EXPECT_TRUE(::RemoveDirectory(my_folder));
  47 }
  48
  49 TEST(WinUtils, SameObject) {
  50   using sandbox::SameObject;
  51
  52   // Create a temp file because we need write access to it.
  53   wchar_t temp_directory[MAX_PATH];
  54   wchar_t my_folder[MAX_PATH];
  55   ASSERT_NE(::GetTempPath(MAX_PATH, temp_directory), 0u);
  56   ASSERT_NE(::GetTempFileName(temp_directory, L"test", 0, my_folder), 0u);
  57
  58   // Delete the file and create a directory instead.
  59   ASSERT_TRUE(::DeleteFile(my_folder));
  60   ASSERT_TRUE(::CreateDirectory(my_folder, NULL));
  61
  62   base::string16 folder(my_folder);
  63   base::string16 file_name = folder + L"\\foo.txt";
  64   const ULONG kSharing = FILE_SHARE_WRITE | FILE_SHARE_READ | FILE_SHARE_DELETE;
  65   base::win::ScopedHandle file(CreateFile(
  66       file_name.c_str(), GENERIC_WRITE, kSharing, NULL, CREATE_ALWAYS,
  67       FILE_FLAG_DELETE_ON_CLOSE, NULL));
  68
  69   EXPECT_TRUE(file.IsValid());
  70   base::string16 file_name_nt1 = base::string16(L"\\??\\") + file_name;
  71   base::string16 file_name_nt2 =
  72       base::string16(L"\\??\\") + folder + L"\\FOO.txT";
  73   EXPECT_TRUE(SameObject(file.Get(), file_name_nt1.c_str()));
  74   EXPECT_TRUE(SameObject(file.Get(), file_name_nt2.c_str()));
  75
  76   file.Close();
  77   EXPECT_TRUE(::RemoveDirectory(my_folder));
  78 }
  79
  80 TEST(WinUtils, IsPipe) {
  81   using sandbox::IsPipe;
  82
  83   base::string16 pipe_name = L"\\??\\pipe\\mypipe";
  84   EXPECT_TRUE(IsPipe(pipe_name));
  85
  86   pipe_name = L"\\??\\PiPe\\mypipe";
  87   EXPECT_TRUE(IsPipe(pipe_name));
  88
  89   pipe_name = L"\\??\\pipe";
  90   EXPECT_FALSE(IsPipe(pipe_name));
  91
  92   pipe_name = L"\\??\\_pipe_\\mypipe";
  93   EXPECT_FALSE(IsPipe(pipe_name));
  94
  95   pipe_name = L"\\??\\ABCD\\mypipe";
  96   EXPECT_FALSE(IsPipe(pipe_name));
  97
  98
  99   // Written as two strings to prevent trigraph '?' '?' '/'.
 100   pipe_name = L"/?" L"?/pipe/mypipe";
 101   EXPECT_FALSE(IsPipe(pipe_name));
 102
 103   pipe_name = L"\\XX\\pipe\\mypipe";
 104   EXPECT_FALSE(IsPipe(pipe_name));
 105
 106   pipe_name = L"\\Device\\NamedPipe\\mypipe";
 107   EXPECT_FALSE(IsPipe(pipe_name));
 108 }