| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <windows.h> | 5 #include <windows.h> |
| 6 | 6 |
| 7 #include <fstream> | 7 #include <fstream> |
| 8 | 8 |
| 9 #include "base/base_paths.h" | 9 #include "base/base_paths.h" |
| 10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 42 wchar_t text_content_1[] = L"delete me"; | 42 wchar_t text_content_1[] = L"delete me"; |
| 43 wchar_t text_content_2[] = L"delete me as well"; | 43 wchar_t text_content_2[] = L"delete me as well"; |
| 44 }; | 44 }; |
| 45 | 45 |
| 46 // Delete a tree without key path. Everything should be deleted. | 46 // Delete a tree without key path. Everything should be deleted. |
| 47 TEST_F(DeleteTreeWorkItemTest, DeleteTreeNoKeyPath) { | 47 TEST_F(DeleteTreeWorkItemTest, DeleteTreeNoKeyPath) { |
| 48 // Create tree to be deleted. | 48 // Create tree to be deleted. |
| 49 base::FilePath dir_name_delete(temp_dir_.path()); | 49 base::FilePath dir_name_delete(temp_dir_.path()); |
| 50 dir_name_delete = dir_name_delete.AppendASCII("to_be_delete"); | 50 dir_name_delete = dir_name_delete.AppendASCII("to_be_delete"); |
| 51 file_util::CreateDirectory(dir_name_delete); | 51 file_util::CreateDirectory(dir_name_delete); |
| 52 ASSERT_TRUE(file_util::PathExists(dir_name_delete)); | 52 ASSERT_TRUE(base::PathExists(dir_name_delete)); |
| 53 | 53 |
| 54 base::FilePath dir_name_delete_1(dir_name_delete); | 54 base::FilePath dir_name_delete_1(dir_name_delete); |
| 55 dir_name_delete_1 = dir_name_delete_1.AppendASCII("1"); | 55 dir_name_delete_1 = dir_name_delete_1.AppendASCII("1"); |
| 56 file_util::CreateDirectory(dir_name_delete_1); | 56 file_util::CreateDirectory(dir_name_delete_1); |
| 57 ASSERT_TRUE(file_util::PathExists(dir_name_delete_1)); | 57 ASSERT_TRUE(base::PathExists(dir_name_delete_1)); |
| 58 | 58 |
| 59 base::FilePath dir_name_delete_2(dir_name_delete); | 59 base::FilePath dir_name_delete_2(dir_name_delete); |
| 60 dir_name_delete_2 = dir_name_delete_2.AppendASCII("2"); | 60 dir_name_delete_2 = dir_name_delete_2.AppendASCII("2"); |
| 61 file_util::CreateDirectory(dir_name_delete_2); | 61 file_util::CreateDirectory(dir_name_delete_2); |
| 62 ASSERT_TRUE(file_util::PathExists(dir_name_delete_2)); | 62 ASSERT_TRUE(base::PathExists(dir_name_delete_2)); |
| 63 | 63 |
| 64 base::FilePath file_name_delete_1(dir_name_delete_1); | 64 base::FilePath file_name_delete_1(dir_name_delete_1); |
| 65 file_name_delete_1 = file_name_delete_1.AppendASCII("File_1.txt"); | 65 file_name_delete_1 = file_name_delete_1.AppendASCII("File_1.txt"); |
| 66 CreateTextFile(file_name_delete_1.value(), text_content_1); | 66 CreateTextFile(file_name_delete_1.value(), text_content_1); |
| 67 ASSERT_TRUE(file_util::PathExists(file_name_delete_1)); | 67 ASSERT_TRUE(base::PathExists(file_name_delete_1)); |
| 68 | 68 |
| 69 base::FilePath file_name_delete_2(dir_name_delete_2); | 69 base::FilePath file_name_delete_2(dir_name_delete_2); |
| 70 file_name_delete_2 = file_name_delete_2.AppendASCII("File_2.txt"); | 70 file_name_delete_2 = file_name_delete_2.AppendASCII("File_2.txt"); |
| 71 CreateTextFile(file_name_delete_2.value(), text_content_1); | 71 CreateTextFile(file_name_delete_2.value(), text_content_1); |
| 72 ASSERT_TRUE(file_util::PathExists(file_name_delete_2)); | 72 ASSERT_TRUE(base::PathExists(file_name_delete_2)); |
| 73 | 73 |
| 74 // Test Do(). | 74 // Test Do(). |
| 75 base::ScopedTempDir temp_dir; | 75 base::ScopedTempDir temp_dir; |
| 76 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 76 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| 77 | 77 |
| 78 std::vector<base::FilePath> key_files; | 78 std::vector<base::FilePath> key_files; |
| 79 scoped_ptr<DeleteTreeWorkItem> work_item( | 79 scoped_ptr<DeleteTreeWorkItem> work_item( |
| 80 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), | 80 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), |
| 81 key_files)); | 81 key_files)); |
| 82 EXPECT_TRUE(work_item->Do()); | 82 EXPECT_TRUE(work_item->Do()); |
| 83 | 83 |
| 84 // everything should be gone | 84 // everything should be gone |
| 85 EXPECT_FALSE(file_util::PathExists(file_name_delete_1)); | 85 EXPECT_FALSE(base::PathExists(file_name_delete_1)); |
| 86 EXPECT_FALSE(file_util::PathExists(file_name_delete_2)); | 86 EXPECT_FALSE(base::PathExists(file_name_delete_2)); |
| 87 EXPECT_FALSE(file_util::PathExists(dir_name_delete)); | 87 EXPECT_FALSE(base::PathExists(dir_name_delete)); |
| 88 | 88 |
| 89 work_item->Rollback(); | 89 work_item->Rollback(); |
| 90 // everything should come back | 90 // everything should come back |
| 91 EXPECT_TRUE(file_util::PathExists(file_name_delete_1)); | 91 EXPECT_TRUE(base::PathExists(file_name_delete_1)); |
| 92 EXPECT_TRUE(file_util::PathExists(file_name_delete_2)); | 92 EXPECT_TRUE(base::PathExists(file_name_delete_2)); |
| 93 EXPECT_TRUE(file_util::PathExists(dir_name_delete)); | 93 EXPECT_TRUE(base::PathExists(dir_name_delete)); |
| 94 } | 94 } |
| 95 | 95 |
| 96 | 96 |
| 97 // Delete a tree with keypath but not in use. Everything should be gone. | 97 // Delete a tree with keypath but not in use. Everything should be gone. |
| 98 // Rollback should bring back everything | 98 // Rollback should bring back everything |
| 99 TEST_F(DeleteTreeWorkItemTest, DeleteTree) { | 99 TEST_F(DeleteTreeWorkItemTest, DeleteTree) { |
| 100 // Create tree to be deleted | 100 // Create tree to be deleted |
| 101 base::FilePath dir_name_delete(temp_dir_.path()); | 101 base::FilePath dir_name_delete(temp_dir_.path()); |
| 102 dir_name_delete = dir_name_delete.AppendASCII("to_be_delete"); | 102 dir_name_delete = dir_name_delete.AppendASCII("to_be_delete"); |
| 103 file_util::CreateDirectory(dir_name_delete); | 103 file_util::CreateDirectory(dir_name_delete); |
| 104 ASSERT_TRUE(file_util::PathExists(dir_name_delete)); | 104 ASSERT_TRUE(base::PathExists(dir_name_delete)); |
| 105 | 105 |
| 106 base::FilePath dir_name_delete_1(dir_name_delete); | 106 base::FilePath dir_name_delete_1(dir_name_delete); |
| 107 dir_name_delete_1 = dir_name_delete_1.AppendASCII("1"); | 107 dir_name_delete_1 = dir_name_delete_1.AppendASCII("1"); |
| 108 file_util::CreateDirectory(dir_name_delete_1); | 108 file_util::CreateDirectory(dir_name_delete_1); |
| 109 ASSERT_TRUE(file_util::PathExists(dir_name_delete_1)); | 109 ASSERT_TRUE(base::PathExists(dir_name_delete_1)); |
| 110 | 110 |
| 111 base::FilePath dir_name_delete_2(dir_name_delete); | 111 base::FilePath dir_name_delete_2(dir_name_delete); |
| 112 dir_name_delete_2 = dir_name_delete_2.AppendASCII("2"); | 112 dir_name_delete_2 = dir_name_delete_2.AppendASCII("2"); |
| 113 file_util::CreateDirectory(dir_name_delete_2); | 113 file_util::CreateDirectory(dir_name_delete_2); |
| 114 ASSERT_TRUE(file_util::PathExists(dir_name_delete_2)); | 114 ASSERT_TRUE(base::PathExists(dir_name_delete_2)); |
| 115 | 115 |
| 116 base::FilePath file_name_delete_1(dir_name_delete_1); | 116 base::FilePath file_name_delete_1(dir_name_delete_1); |
| 117 file_name_delete_1 = file_name_delete_1.AppendASCII("File_1.txt"); | 117 file_name_delete_1 = file_name_delete_1.AppendASCII("File_1.txt"); |
| 118 CreateTextFile(file_name_delete_1.value(), text_content_1); | 118 CreateTextFile(file_name_delete_1.value(), text_content_1); |
| 119 ASSERT_TRUE(file_util::PathExists(file_name_delete_1)); | 119 ASSERT_TRUE(base::PathExists(file_name_delete_1)); |
| 120 | 120 |
| 121 base::FilePath file_name_delete_2(dir_name_delete_2); | 121 base::FilePath file_name_delete_2(dir_name_delete_2); |
| 122 file_name_delete_2 = file_name_delete_2.AppendASCII("File_2.txt"); | 122 file_name_delete_2 = file_name_delete_2.AppendASCII("File_2.txt"); |
| 123 CreateTextFile(file_name_delete_2.value(), text_content_1); | 123 CreateTextFile(file_name_delete_2.value(), text_content_1); |
| 124 ASSERT_TRUE(file_util::PathExists(file_name_delete_2)); | 124 ASSERT_TRUE(base::PathExists(file_name_delete_2)); |
| 125 | 125 |
| 126 // test Do() | 126 // test Do() |
| 127 base::ScopedTempDir temp_dir; | 127 base::ScopedTempDir temp_dir; |
| 128 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 128 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| 129 | 129 |
| 130 std::vector<base::FilePath> key_files(1, file_name_delete_1); | 130 std::vector<base::FilePath> key_files(1, file_name_delete_1); |
| 131 scoped_ptr<DeleteTreeWorkItem> work_item( | 131 scoped_ptr<DeleteTreeWorkItem> work_item( |
| 132 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), | 132 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), |
| 133 key_files)); | 133 key_files)); |
| 134 EXPECT_TRUE(work_item->Do()); | 134 EXPECT_TRUE(work_item->Do()); |
| 135 | 135 |
| 136 // everything should be gone | 136 // everything should be gone |
| 137 EXPECT_FALSE(file_util::PathExists(file_name_delete_1)); | 137 EXPECT_FALSE(base::PathExists(file_name_delete_1)); |
| 138 EXPECT_FALSE(file_util::PathExists(file_name_delete_2)); | 138 EXPECT_FALSE(base::PathExists(file_name_delete_2)); |
| 139 EXPECT_FALSE(file_util::PathExists(dir_name_delete)); | 139 EXPECT_FALSE(base::PathExists(dir_name_delete)); |
| 140 | 140 |
| 141 work_item->Rollback(); | 141 work_item->Rollback(); |
| 142 // everything should come back | 142 // everything should come back |
| 143 EXPECT_TRUE(file_util::PathExists(file_name_delete_1)); | 143 EXPECT_TRUE(base::PathExists(file_name_delete_1)); |
| 144 EXPECT_TRUE(file_util::PathExists(file_name_delete_2)); | 144 EXPECT_TRUE(base::PathExists(file_name_delete_2)); |
| 145 EXPECT_TRUE(file_util::PathExists(dir_name_delete)); | 145 EXPECT_TRUE(base::PathExists(dir_name_delete)); |
| 146 } | 146 } |
| 147 | 147 |
| 148 // Delete a tree with key_path in use. Everything should still be there. | 148 // Delete a tree with key_path in use. Everything should still be there. |
| 149 TEST_F(DeleteTreeWorkItemTest, DeleteTreeInUse) { | 149 TEST_F(DeleteTreeWorkItemTest, DeleteTreeInUse) { |
| 150 // Create tree to be deleted | 150 // Create tree to be deleted |
| 151 base::FilePath dir_name_delete(temp_dir_.path()); | 151 base::FilePath dir_name_delete(temp_dir_.path()); |
| 152 dir_name_delete = dir_name_delete.AppendASCII("to_be_delete"); | 152 dir_name_delete = dir_name_delete.AppendASCII("to_be_delete"); |
| 153 file_util::CreateDirectory(dir_name_delete); | 153 file_util::CreateDirectory(dir_name_delete); |
| 154 ASSERT_TRUE(file_util::PathExists(dir_name_delete)); | 154 ASSERT_TRUE(base::PathExists(dir_name_delete)); |
| 155 | 155 |
| 156 base::FilePath dir_name_delete_1(dir_name_delete); | 156 base::FilePath dir_name_delete_1(dir_name_delete); |
| 157 dir_name_delete_1 = dir_name_delete_1.AppendASCII("1"); | 157 dir_name_delete_1 = dir_name_delete_1.AppendASCII("1"); |
| 158 file_util::CreateDirectory(dir_name_delete_1); | 158 file_util::CreateDirectory(dir_name_delete_1); |
| 159 ASSERT_TRUE(file_util::PathExists(dir_name_delete_1)); | 159 ASSERT_TRUE(base::PathExists(dir_name_delete_1)); |
| 160 | 160 |
| 161 base::FilePath dir_name_delete_2(dir_name_delete); | 161 base::FilePath dir_name_delete_2(dir_name_delete); |
| 162 dir_name_delete_2 = dir_name_delete_2.AppendASCII("2"); | 162 dir_name_delete_2 = dir_name_delete_2.AppendASCII("2"); |
| 163 file_util::CreateDirectory(dir_name_delete_2); | 163 file_util::CreateDirectory(dir_name_delete_2); |
| 164 ASSERT_TRUE(file_util::PathExists(dir_name_delete_2)); | 164 ASSERT_TRUE(base::PathExists(dir_name_delete_2)); |
| 165 | 165 |
| 166 base::FilePath file_name_delete_1(dir_name_delete_1); | 166 base::FilePath file_name_delete_1(dir_name_delete_1); |
| 167 file_name_delete_1 = file_name_delete_1.AppendASCII("File_1.txt"); | 167 file_name_delete_1 = file_name_delete_1.AppendASCII("File_1.txt"); |
| 168 CreateTextFile(file_name_delete_1.value(), text_content_1); | 168 CreateTextFile(file_name_delete_1.value(), text_content_1); |
| 169 ASSERT_TRUE(file_util::PathExists(file_name_delete_1)); | 169 ASSERT_TRUE(base::PathExists(file_name_delete_1)); |
| 170 | 170 |
| 171 base::FilePath file_name_delete_2(dir_name_delete_2); | 171 base::FilePath file_name_delete_2(dir_name_delete_2); |
| 172 file_name_delete_2 = file_name_delete_2.AppendASCII("File_2.txt"); | 172 file_name_delete_2 = file_name_delete_2.AppendASCII("File_2.txt"); |
| 173 CreateTextFile(file_name_delete_2.value(), text_content_1); | 173 CreateTextFile(file_name_delete_2.value(), text_content_1); |
| 174 ASSERT_TRUE(file_util::PathExists(file_name_delete_2)); | 174 ASSERT_TRUE(base::PathExists(file_name_delete_2)); |
| 175 | 175 |
| 176 // Create a key path file. | 176 // Create a key path file. |
| 177 base::FilePath key_path(dir_name_delete); | 177 base::FilePath key_path(dir_name_delete); |
| 178 key_path = key_path.AppendASCII("key_file.exe"); | 178 key_path = key_path.AppendASCII("key_file.exe"); |
| 179 | 179 |
| 180 wchar_t exe_full_path_str[MAX_PATH]; | 180 wchar_t exe_full_path_str[MAX_PATH]; |
| 181 ::GetModuleFileNameW(NULL, exe_full_path_str, MAX_PATH); | 181 ::GetModuleFileNameW(NULL, exe_full_path_str, MAX_PATH); |
| 182 base::FilePath exe_full_path(exe_full_path_str); | 182 base::FilePath exe_full_path(exe_full_path_str); |
| 183 | 183 |
| 184 base::CopyFile(exe_full_path, key_path); | 184 base::CopyFile(exe_full_path, key_path); |
| 185 ASSERT_TRUE(file_util::PathExists(key_path)); | 185 ASSERT_TRUE(base::PathExists(key_path)); |
| 186 | 186 |
| 187 VLOG(1) << "copy ourself from " << exe_full_path.value() | 187 VLOG(1) << "copy ourself from " << exe_full_path.value() |
| 188 << " to " << key_path.value(); | 188 << " to " << key_path.value(); |
| 189 | 189 |
| 190 // Run the key path file to keep it in use. | 190 // Run the key path file to keep it in use. |
| 191 STARTUPINFOW si = {sizeof(si)}; | 191 STARTUPINFOW si = {sizeof(si)}; |
| 192 PROCESS_INFORMATION pi = {0}; | 192 PROCESS_INFORMATION pi = {0}; |
| 193 ASSERT_TRUE( | 193 ASSERT_TRUE( |
| 194 ::CreateProcessW(NULL, const_cast<wchar_t*>(key_path.value().c_str()), | 194 ::CreateProcessW(NULL, const_cast<wchar_t*>(key_path.value().c_str()), |
| 195 NULL, NULL, FALSE, CREATE_NO_WINDOW | CREATE_SUSPENDED, | 195 NULL, NULL, FALSE, CREATE_NO_WINDOW | CREATE_SUSPENDED, |
| 196 NULL, NULL, &si, &pi)); | 196 NULL, NULL, &si, &pi)); |
| 197 | 197 |
| 198 // test Do(). | 198 // test Do(). |
| 199 { | 199 { |
| 200 base::ScopedTempDir temp_dir; | 200 base::ScopedTempDir temp_dir; |
| 201 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 201 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| 202 | 202 |
| 203 std::vector<base::FilePath> key_paths(1, key_path); | 203 std::vector<base::FilePath> key_paths(1, key_path); |
| 204 scoped_ptr<DeleteTreeWorkItem> work_item( | 204 scoped_ptr<DeleteTreeWorkItem> work_item( |
| 205 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), | 205 WorkItem::CreateDeleteTreeWorkItem(dir_name_delete, temp_dir.path(), |
| 206 key_paths)); | 206 key_paths)); |
| 207 | 207 |
| 208 // delete should fail as file in use. | 208 // delete should fail as file in use. |
| 209 EXPECT_FALSE(work_item->Do()); | 209 EXPECT_FALSE(work_item->Do()); |
| 210 } | 210 } |
| 211 | 211 |
| 212 // verify everything is still there. | 212 // verify everything is still there. |
| 213 EXPECT_TRUE(file_util::PathExists(key_path)); | 213 EXPECT_TRUE(base::PathExists(key_path)); |
| 214 EXPECT_TRUE(file_util::PathExists(file_name_delete_1)); | 214 EXPECT_TRUE(base::PathExists(file_name_delete_1)); |
| 215 EXPECT_TRUE(file_util::PathExists(file_name_delete_2)); | 215 EXPECT_TRUE(base::PathExists(file_name_delete_2)); |
| 216 | 216 |
| 217 TerminateProcess(pi.hProcess, 0); | 217 TerminateProcess(pi.hProcess, 0); |
| 218 // make sure the handle is closed. | 218 // make sure the handle is closed. |
| 219 WaitForSingleObject(pi.hProcess, INFINITE); | 219 WaitForSingleObject(pi.hProcess, INFINITE); |
| 220 } | 220 } |
| OLD | NEW |