scoped_process_information_unittest.cc 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175
  1. // Copyright (c) 2012 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 "base/win/scoped_process_information.h"
  5. #include <windows.h>
  6. #include <string>
  7. #include "base/command_line.h"
  8. #include "base/process/kill.h"
  9. #include "base/process/process.h"
  10. #include "base/strings/string_util.h"
  11. #include "base/test/multiprocess_test.h"
  12. #include "base/win/windows_version.h"
  13. #include "testing/multiprocess_func_list.h"
  14. namespace {
  15. const DWORD kProcessId = 4321;
  16. const DWORD kThreadId = 1234;
  17. const HANDLE kProcessHandle = reinterpret_cast<HANDLE>(7651);
  18. const HANDLE kThreadHandle = reinterpret_cast<HANDLE>(1567);
  19. void MockCreateProcess(base::win::ScopedProcessInformation* process_info) {
  20. PROCESS_INFORMATION process_information = {};
  21. process_information.dwProcessId = kProcessId;
  22. process_information.dwThreadId = kThreadId;
  23. process_information.hProcess = kProcessHandle;
  24. process_information.hThread = kThreadHandle;
  25. process_info->Set(process_information);
  26. }
  27. } // namespace
  28. class ScopedProcessInformationTest : public base::MultiProcessTest {
  29. protected:
  30. void DoCreateProcess(const std::string& main_id,
  31. PROCESS_INFORMATION* process_handle);
  32. };
  33. MULTIPROCESS_TEST_MAIN(ReturnSeven) {
  34. return 7;
  35. }
  36. MULTIPROCESS_TEST_MAIN(ReturnNine) {
  37. return 9;
  38. }
  39. void ScopedProcessInformationTest::DoCreateProcess(
  40. const std::string& main_id,
  41. PROCESS_INFORMATION* process_handle) {
  42. base::CommandLine::StringType cmd_line =
  43. MakeCmdLine(main_id).GetCommandLineString();
  44. STARTUPINFO startup_info = {};
  45. startup_info.cb = sizeof(startup_info);
  46. EXPECT_TRUE(::CreateProcess(nullptr, std::data(cmd_line), nullptr, nullptr,
  47. false, 0, nullptr, nullptr, &startup_info,
  48. process_handle));
  49. }
  50. TEST_F(ScopedProcessInformationTest, InitiallyInvalid) {
  51. base::win::ScopedProcessInformation process_info;
  52. ASSERT_FALSE(process_info.IsValid());
  53. }
  54. TEST_F(ScopedProcessInformationTest, Receive) {
  55. base::win::ScopedProcessInformation process_info;
  56. MockCreateProcess(&process_info);
  57. EXPECT_TRUE(process_info.IsValid());
  58. EXPECT_EQ(kProcessId, process_info.process_id());
  59. EXPECT_EQ(kThreadId, process_info.thread_id());
  60. EXPECT_EQ(kProcessHandle, process_info.process_handle());
  61. EXPECT_EQ(kThreadHandle, process_info.thread_handle());
  62. process_info.Take();
  63. }
  64. TEST_F(ScopedProcessInformationTest, TakeProcess) {
  65. base::win::ScopedProcessInformation process_info;
  66. MockCreateProcess(&process_info);
  67. HANDLE process = process_info.TakeProcessHandle();
  68. EXPECT_EQ(kProcessHandle, process);
  69. EXPECT_EQ(nullptr, process_info.process_handle());
  70. EXPECT_EQ(0u, process_info.process_id());
  71. EXPECT_TRUE(process_info.IsValid());
  72. process_info.Take();
  73. }
  74. TEST_F(ScopedProcessInformationTest, TakeThread) {
  75. base::win::ScopedProcessInformation process_info;
  76. MockCreateProcess(&process_info);
  77. HANDLE thread = process_info.TakeThreadHandle();
  78. EXPECT_EQ(kThreadHandle, thread);
  79. EXPECT_EQ(nullptr, process_info.thread_handle());
  80. EXPECT_EQ(0u, process_info.thread_id());
  81. EXPECT_TRUE(process_info.IsValid());
  82. process_info.Take();
  83. }
  84. TEST_F(ScopedProcessInformationTest, TakeBoth) {
  85. base::win::ScopedProcessInformation process_info;
  86. MockCreateProcess(&process_info);
  87. process_info.TakeProcessHandle();
  88. process_info.TakeThreadHandle();
  89. EXPECT_FALSE(process_info.IsValid());
  90. process_info.Take();
  91. }
  92. TEST_F(ScopedProcessInformationTest, TakeWholeStruct) {
  93. base::win::ScopedProcessInformation process_info;
  94. MockCreateProcess(&process_info);
  95. PROCESS_INFORMATION to_discard = process_info.Take();
  96. EXPECT_EQ(kProcessId, to_discard.dwProcessId);
  97. EXPECT_EQ(kThreadId, to_discard.dwThreadId);
  98. EXPECT_EQ(kProcessHandle, to_discard.hProcess);
  99. EXPECT_EQ(kThreadHandle, to_discard.hThread);
  100. EXPECT_FALSE(process_info.IsValid());
  101. }
  102. TEST_F(ScopedProcessInformationTest, Duplicate) {
  103. if (base::win::GetVersion() <= base::win::Version::WIN7) {
  104. // Disabled on Windows 7 due to flakiness. https://crbug.com/1336879
  105. GTEST_SKIP();
  106. }
  107. PROCESS_INFORMATION temp_process_information;
  108. DoCreateProcess("ReturnSeven", &temp_process_information);
  109. base::win::ScopedProcessInformation process_info;
  110. process_info.Set(temp_process_information);
  111. base::win::ScopedProcessInformation duplicate;
  112. duplicate.DuplicateFrom(process_info);
  113. ASSERT_TRUE(process_info.IsValid());
  114. ASSERT_NE(0u, process_info.process_id());
  115. ASSERT_EQ(duplicate.process_id(), process_info.process_id());
  116. ASSERT_NE(0u, process_info.thread_id());
  117. ASSERT_EQ(duplicate.thread_id(), process_info.thread_id());
  118. // Validate that we have separate handles that are good.
  119. int exit_code = 0;
  120. base::Process process(process_info.TakeProcessHandle());
  121. ASSERT_TRUE(process.WaitForExit(&exit_code));
  122. ASSERT_EQ(7, exit_code);
  123. exit_code = 0;
  124. base::Process dup_process(duplicate.TakeProcessHandle());
  125. ASSERT_TRUE(dup_process.WaitForExit(&exit_code));
  126. ASSERT_EQ(7, exit_code);
  127. ASSERT_TRUE(::CloseHandle(process_info.TakeThreadHandle()));
  128. ASSERT_TRUE(::CloseHandle(duplicate.TakeThreadHandle()));
  129. }
  130. TEST_F(ScopedProcessInformationTest, Set) {
  131. base::win::ScopedProcessInformation base_process_info;
  132. MockCreateProcess(&base_process_info);
  133. PROCESS_INFORMATION base_struct = base_process_info.Take();
  134. base::win::ScopedProcessInformation process_info;
  135. process_info.Set(base_struct);
  136. EXPECT_EQ(kProcessId, process_info.process_id());
  137. EXPECT_EQ(kThreadId, process_info.thread_id());
  138. EXPECT_EQ(kProcessHandle, process_info.process_handle());
  139. EXPECT_EQ(kThreadHandle, process_info.thread_handle());
  140. base_struct = process_info.Take();
  141. }