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 <string> 6 7 #include "base/file_util.h" 8 #include "base/files/file_path.h" 9 #include "base/files/scoped_temp_dir.h" 10 #include "base/platform_file.h" 11 #include "base/win/scoped_handle.h" 12 #include "chrome/installer/util/logging_installer.h" 13 #include "testing/gtest/include/gtest/gtest.h" 14 15 TEST(LoggingInstallerTest, TestTruncate) { 16 const std::string test_data(installer::kMaxInstallerLogFileSize + 1, 'a'); 17 18 base::ScopedTempDir temp_dir; 19 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 20 21 base::FilePath temp_file = temp_dir.path().Append(L"temp"); 22 EXPECT_EQ(test_data.size(), 23 file_util::WriteFile(temp_file, &test_data[0], test_data.size())); 24 ASSERT_TRUE(base::PathExists(temp_file)); 25 26 int64 file_size = 0; 27 EXPECT_TRUE(file_util::GetFileSize(temp_file, &file_size)); 28 EXPECT_EQ(test_data.size(), file_size); 29 30 EXPECT_EQ(installer::LOGFILE_TRUNCATED, 31 installer::TruncateLogFileIfNeeded(temp_file)); 32 33 EXPECT_TRUE(file_util::GetFileSize(temp_file, &file_size)); 34 EXPECT_EQ(installer::kTruncatedInstallerLogFileSize , file_size); 35 36 // Check that the temporary file was deleted. 37 EXPECT_FALSE(base::PathExists(temp_file.Append(L".tmp"))); 38 } 39 40 TEST(LoggingInstallerTest, TestTruncationNotNeeded) { 41 const std::string test_data(installer::kMaxInstallerLogFileSize, 'a'); 42 43 base::ScopedTempDir temp_dir; 44 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 45 46 base::FilePath temp_file = temp_dir.path().Append(L"temp"); 47 EXPECT_EQ(test_data.size(), 48 file_util::WriteFile(temp_file, &test_data[0], test_data.size())); 49 ASSERT_TRUE(base::PathExists(temp_file)); 50 51 int64 file_size = 0; 52 EXPECT_TRUE(file_util::GetFileSize(temp_file, &file_size)); 53 EXPECT_EQ(test_data.size(), file_size); 54 55 EXPECT_EQ(installer::LOGFILE_UNTOUCHED, 56 installer::TruncateLogFileIfNeeded(temp_file)); 57 EXPECT_TRUE(base::PathExists(temp_file)); 58 EXPECT_TRUE(file_util::GetFileSize(temp_file, &file_size)); 59 EXPECT_EQ(test_data.size(), file_size); 60 } 61 62 TEST(LoggingInstallerTest, TestInUseNeedsTruncation) { 63 const std::string test_data(installer::kMaxInstallerLogFileSize + 1, 'a'); 64 65 base::ScopedTempDir temp_dir; 66 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 67 68 base::FilePath temp_file = temp_dir.path().Append(L"temp"); 69 EXPECT_EQ(test_data.size(), 70 file_util::WriteFile(temp_file, &test_data[0], test_data.size())); 71 ASSERT_TRUE(base::PathExists(temp_file)); 72 int64 file_size = 0; 73 EXPECT_TRUE(file_util::GetFileSize(temp_file, &file_size)); 74 EXPECT_EQ(test_data.size(), file_size); 75 76 // Prevent the log file from being moved or deleted. 77 const int file_flags = base::PLATFORM_FILE_OPEN | 78 base::PLATFORM_FILE_READ | 79 base::PLATFORM_FILE_EXCLUSIVE_READ; 80 base::win::ScopedHandle temp_platform_file( 81 base::CreatePlatformFile(temp_file, file_flags, NULL, NULL)); 82 ASSERT_TRUE(temp_platform_file.IsValid()); 83 84 EXPECT_EQ(installer::LOGFILE_UNTOUCHED, 85 installer::TruncateLogFileIfNeeded(temp_file)); 86 EXPECT_TRUE(base::PathExists(temp_file)); 87 EXPECT_TRUE(file_util::GetFileSize(temp_file, &file_size)); 88 EXPECT_EQ(test_data.size(), file_size); 89 } 90 91 TEST(LoggingInstallerTest, TestMoveFailsNeedsTruncation) { 92 const std::string test_data(installer::kMaxInstallerLogFileSize + 1, 'a'); 93 94 base::ScopedTempDir temp_dir; 95 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 96 97 base::FilePath temp_file = temp_dir.path().Append(L"temp"); 98 EXPECT_EQ(test_data.size(), 99 file_util::WriteFile(temp_file, &test_data[0], test_data.size())); 100 ASSERT_TRUE(base::PathExists(temp_file)); 101 int64 file_size = 0; 102 EXPECT_TRUE(file_util::GetFileSize(temp_file, &file_size)); 103 EXPECT_EQ(test_data.size(), file_size); 104 105 // Create an inconvenient, non-deletable file in the location that 106 // TruncateLogFileIfNeeded would like to move the log file to. 107 const int file_flags = base::PLATFORM_FILE_CREATE | 108 base::PLATFORM_FILE_READ | 109 base::PLATFORM_FILE_EXCLUSIVE_READ; 110 base::FilePath temp_file_move_dest( 111 temp_file.value() + FILE_PATH_LITERAL(".tmp")); 112 base::win::ScopedHandle temp_move_destination_file( 113 base::CreatePlatformFile(temp_file_move_dest, file_flags, NULL, NULL)); 114 ASSERT_TRUE(temp_move_destination_file.IsValid()); 115 116 EXPECT_EQ(installer::LOGFILE_DELETED, 117 installer::TruncateLogFileIfNeeded(temp_file)); 118 EXPECT_FALSE(base::PathExists(temp_file)); 119 } 120
