1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/security/sandbox/win/src/handle_inheritance_test.cc Wed Dec 31 06:09:35 2014 +0100 1.3 @@ -0,0 +1,54 @@ 1.4 +// Copyright (c) 2012 The Chromium Authors. All rights reserved. 1.5 +// Use of this source code is governed by a BSD-style license that can be 1.6 +// found in the LICENSE file. 1.7 + 1.8 +#include <stdio.h> 1.9 + 1.10 +#include "base/file_util.h" 1.11 +#include "base/win/windows_version.h" 1.12 +#include "sandbox/win/tests/common/controller.h" 1.13 +#include "testing/gtest/include/gtest/gtest.h" 1.14 + 1.15 +namespace sandbox { 1.16 + 1.17 +SBOX_TESTS_COMMAND int HandleInheritanceTests_PrintToStdout(int argc, 1.18 + wchar_t** argv) { 1.19 + printf("Example output to stdout\n"); 1.20 + return SBOX_TEST_SUCCEEDED; 1.21 +} 1.22 + 1.23 +TEST(HandleInheritanceTests, TestStdoutInheritance) { 1.24 + wchar_t temp_directory[MAX_PATH]; 1.25 + wchar_t temp_file_name[MAX_PATH]; 1.26 + ASSERT_NE(::GetTempPath(MAX_PATH, temp_directory), 0u); 1.27 + ASSERT_NE(::GetTempFileName(temp_directory, L"test", 0, temp_file_name), 0u); 1.28 + 1.29 + SECURITY_ATTRIBUTES attrs = {}; 1.30 + attrs.nLength = sizeof(attrs); 1.31 + attrs.lpSecurityDescriptor = NULL; 1.32 + attrs.bInheritHandle = TRUE; 1.33 + HANDLE file_handle = CreateFile( 1.34 + temp_file_name, GENERIC_WRITE, 1.35 + FILE_SHARE_WRITE | FILE_SHARE_READ | FILE_SHARE_DELETE, 1.36 + &attrs, OPEN_EXISTING, 0, NULL); 1.37 + EXPECT_NE(file_handle, INVALID_HANDLE_VALUE); 1.38 + 1.39 + TestRunner runner; 1.40 + EXPECT_EQ(SBOX_ALL_OK, runner.GetPolicy()->SetStdoutHandle(file_handle)); 1.41 + int result = runner.RunTest(L"HandleInheritanceTests_PrintToStdout"); 1.42 + EXPECT_EQ(SBOX_TEST_SUCCEEDED, result); 1.43 + EXPECT_TRUE(::CloseHandle(file_handle)); 1.44 + 1.45 + std::string data; 1.46 + EXPECT_TRUE(base::ReadFileToString(base::FilePath(temp_file_name), &data)); 1.47 + // Redirection uses a feature that was added in Windows Vista. 1.48 + if (base::win::GetVersion() >= base::win::VERSION_VISTA) { 1.49 + EXPECT_EQ("Example output to stdout\r\n", data); 1.50 + } else { 1.51 + EXPECT_EQ("", data); 1.52 + } 1.53 + 1.54 + EXPECT_TRUE(::DeleteFile(temp_file_name)); 1.55 +} 1.56 + 1.57 +}