15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <windows.h> 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/command_line.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/test/multiprocess_test.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/scoped_handle.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/scoped_process_information.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/startup_information.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/windows_version.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/multiprocess_func_list.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const wchar_t kSectionName[] = L"EventTestSection"; 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const size_t kSectionSize = 4096; 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)MULTIPROCESS_TEST_MAIN(FireInheritedEvents) { 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE section = ::OpenFileMappingW(PAGE_READWRITE, false, kSectionName); 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE* events = reinterpret_cast<HANDLE*>(::MapViewOfFile(section, 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PAGE_READWRITE, 0, 0, kSectionSize)); 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This event should not be valid because it wasn't explicitly inherited. 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (::SetEvent(events[1])) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return -1; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This event should be valid because it was explicitly inherited. 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!::SetEvent(events[0])) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return -1; 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class StartupInformationTest : public base::MultiProcessTest {}; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Verify that only the explicitly specified event is inherited. 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(StartupInformationTest, InheritStdOut) { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (base::win::GetVersion() < base::win::VERSION_VISTA) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::win::StartupInformation startup_info; 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE section = ::CreateFileMappingW(INVALID_HANDLE_VALUE, NULL, 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PAGE_READWRITE, 0, kSectionSize, 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kSectionName); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(section); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE* events = reinterpret_cast<HANDLE*>(::MapViewOfFile(section, 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE_MAP_READ | FILE_MAP_WRITE, 0, 0, kSectionSize)); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make two inheritable events. 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SECURITY_ATTRIBUTES security_attributes = { sizeof(security_attributes), 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, true }; 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events[0] = ::CreateEvent(&security_attributes, false, false, NULL); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(events[0]); 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events[1] = ::CreateEvent(&security_attributes, false, false, NULL); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(events[1]); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(startup_info.InitializeProcThreadAttributeList(1)); 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(startup_info.UpdateProcThreadAttribute( 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PROC_THREAD_ATTRIBUTE_HANDLE_LIST, &events[0], 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sizeof(events[0]))); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::wstring cmd_line = 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this->MakeCmdLine("FireInheritedEvents", false).GetCommandLineString(); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 67f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) PROCESS_INFORMATION temp_process_info = {}; 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(::CreateProcess(NULL, const_cast<wchar_t*>(cmd_line.c_str()), 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, NULL, true, EXTENDED_STARTUPINFO_PRESENT, 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, NULL, startup_info.startup_info(), 71f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) &temp_process_info)) << ::GetLastError(); 72f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) base::win::ScopedProcessInformation process_info(temp_process_info); 73f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Only the first event should be signalled 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(WAIT_OBJECT_0, ::WaitForMultipleObjects(2, events, false, 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4000)); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 79