integrity_level_test.cc 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113
  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. #include <windows.h>
  5. #include <atlsecurity.h>
  6. #include "base/process/process_info.h"
  7. #include "base/win/access_token.h"
  8. #include "base/win/windows_version.h"
  9. #include "sandbox/win/src/sandbox.h"
  10. #include "sandbox/win/src/sandbox_factory.h"
  11. #include "sandbox/win/src/sandbox_policy.h"
  12. #include "sandbox/win/tests/common/controller.h"
  13. #include "testing/gtest/include/gtest/gtest.h"
  14. #include "third_party/abseil-cpp/absl/types/optional.h"
  15. namespace sandbox {
  16. SBOX_TESTS_COMMAND int CheckUntrustedIntegrityLevel(int argc, wchar_t** argv) {
  17. return (base::GetCurrentProcessIntegrityLevel() == base::UNTRUSTED_INTEGRITY)
  18. ? SBOX_TEST_SUCCEEDED
  19. : SBOX_TEST_FAILED;
  20. }
  21. SBOX_TESTS_COMMAND int CheckLowIntegrityLevel(int argc, wchar_t** argv) {
  22. return (base::GetCurrentProcessIntegrityLevel() == base::LOW_INTEGRITY)
  23. ? SBOX_TEST_SUCCEEDED
  24. : SBOX_TEST_FAILED;
  25. }
  26. SBOX_TESTS_COMMAND int CheckIntegrityLevel(int argc, wchar_t** argv) {
  27. absl::optional<base::win::AccessToken> token =
  28. base::win::AccessToken::FromEffective();
  29. if (!token)
  30. return SBOX_TEST_FAILED;
  31. if (token->IntegrityLevel() == SECURITY_MANDATORY_LOW_RID)
  32. return SBOX_TEST_SUCCEEDED;
  33. return SBOX_TEST_DENIED;
  34. }
  35. std::unique_ptr<TestRunner> LowILRealRunner() {
  36. auto runner = std::make_unique<TestRunner>(
  37. JobLevel::kLockdown, USER_INTERACTIVE, USER_INTERACTIVE);
  38. runner->SetTimeout(INFINITE);
  39. runner->GetPolicy()->SetAlternateDesktop(true);
  40. runner->GetPolicy()->SetIntegrityLevel(INTEGRITY_LEVEL_LOW);
  41. return runner;
  42. }
  43. TEST(IntegrityLevelTest, TestLowILReal) {
  44. auto runner = LowILRealRunner();
  45. EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner->RunTest(L"CheckIntegrityLevel"));
  46. runner = LowILRealRunner();
  47. runner->SetTestState(BEFORE_REVERT);
  48. EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner->RunTest(L"CheckIntegrityLevel"));
  49. }
  50. std::unique_ptr<TestRunner> LowILDelayedRunner() {
  51. auto runner = std::make_unique<TestRunner>(
  52. JobLevel::kLockdown, USER_INTERACTIVE, USER_INTERACTIVE);
  53. runner->SetTimeout(INFINITE);
  54. runner->GetPolicy()->SetDelayedIntegrityLevel(INTEGRITY_LEVEL_LOW);
  55. return runner;
  56. }
  57. TEST(DelayedIntegrityLevelTest, TestLowILDelayed) {
  58. auto runner = LowILDelayedRunner();
  59. EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner->RunTest(L"CheckIntegrityLevel"));
  60. runner = LowILDelayedRunner();
  61. runner->SetTestState(BEFORE_REVERT);
  62. EXPECT_EQ(SBOX_TEST_DENIED, runner->RunTest(L"CheckIntegrityLevel"));
  63. }
  64. TEST(IntegrityLevelTest, TestNoILChange) {
  65. TestRunner runner(JobLevel::kLockdown, USER_INTERACTIVE, USER_INTERACTIVE);
  66. runner.SetTimeout(INFINITE);
  67. EXPECT_EQ(SBOX_TEST_DENIED, runner.RunTest(L"CheckIntegrityLevel"));
  68. }
  69. TEST(IntegrityLevelTest, TestUntrustedIL) {
  70. TestRunner runner(JobLevel::kLockdown, USER_RESTRICTED_SAME_ACCESS,
  71. USER_LOCKDOWN);
  72. runner.GetPolicy()->SetIntegrityLevel(INTEGRITY_LEVEL_LOW);
  73. runner.GetPolicy()->SetDelayedIntegrityLevel(INTEGRITY_LEVEL_UNTRUSTED);
  74. runner.GetPolicy()->SetLockdownDefaultDacl();
  75. runner.SetTimeout(INFINITE);
  76. EXPECT_EQ(SBOX_TEST_SUCCEEDED,
  77. runner.RunTest(L"CheckUntrustedIntegrityLevel"));
  78. }
  79. TEST(IntegrityLevelTest, TestLowIL) {
  80. TestRunner runner(JobLevel::kLockdown, USER_RESTRICTED_SAME_ACCESS,
  81. USER_LOCKDOWN);
  82. runner.GetPolicy()->SetIntegrityLevel(INTEGRITY_LEVEL_LOW);
  83. runner.GetPolicy()->SetDelayedIntegrityLevel(INTEGRITY_LEVEL_LOW);
  84. runner.GetPolicy()->SetLockdownDefaultDacl();
  85. runner.SetTimeout(INFINITE);
  86. EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"CheckLowIntegrityLevel"));
  87. }
  88. } // namespace sandbox