OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 // This file defines helper methods used to schedule files for deletion | 5 // This file defines helper methods used to schedule files for deletion |
6 // on next reboot. The code here is heavily borrowed and simplified from | 6 // on next reboot. The code here is heavily borrowed and simplified from |
7 // http://code.google.com/p/omaha/source/browse/trunk/common/file.cc and | 7 // http://code.google.com/p/omaha/source/browse/trunk/common/file.cc and |
8 // http://code.google.com/p/omaha/source/browse/trunk/common/utils.cc | 8 // http://code.google.com/p/omaha/source/browse/trunk/common/utils.cc |
9 // | 9 // |
10 // This implementation really is not fast, so do not use it where that will | 10 // This implementation really is not fast, so do not use it where that will |
11 // matter. | 11 // matter. |
12 | 12 |
13 #include "chrome/installer/util/delete_after_reboot_helper.h" | 13 #include "chrome/installer/util/delete_after_reboot_helper.h" |
14 | 14 |
15 #include <string> | 15 #include <string> |
16 #include <vector> | 16 #include <vector> |
17 | 17 |
18 #include "base/file_util.h" | 18 #include "base/file_util.h" |
| 19 #include "base/files/file_enumerator.h" |
19 #include "base/win/registry.h" | 20 #include "base/win/registry.h" |
20 #include "base/string_util.h" | 21 #include "base/string_util.h" |
21 | 22 |
22 // The moves-pending-reboot is a MULTISZ registry key in the HKLM part of the | 23 // The moves-pending-reboot is a MULTISZ registry key in the HKLM part of the |
23 // registry. | 24 // registry. |
24 const wchar_t kSessionManagerKey[] = | 25 const wchar_t kSessionManagerKey[] = |
25 L"SYSTEM\\CurrentControlSet\\Control\\Session Manager"; | 26 L"SYSTEM\\CurrentControlSet\\Control\\Session Manager"; |
26 const wchar_t kPendingFileRenameOps[] = L"PendingFileRenameOperations"; | 27 const wchar_t kPendingFileRenameOps[] = L"PendingFileRenameOperations"; |
27 | 28 |
28 namespace { | 29 namespace { |
(...skipping 80 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
109 } | 110 } |
110 // Confirm it is a directory | 111 // Confirm it is a directory |
111 if (!(dir_attributes & FILE_ATTRIBUTE_DIRECTORY)) { | 112 if (!(dir_attributes & FILE_ATTRIBUTE_DIRECTORY)) { |
112 LOG(ERROR) << "Scheduled directory is not a directory: " << dir_name; | 113 LOG(ERROR) << "Scheduled directory is not a directory: " << dir_name; |
113 return false; | 114 return false; |
114 } | 115 } |
115 | 116 |
116 // First schedule all the normal files for deletion. | 117 // First schedule all the normal files for deletion. |
117 { | 118 { |
118 bool success = true; | 119 bool success = true; |
119 file_util::FileEnumerator file_enum(base::FilePath(dir_name), false, | 120 base::FileEnumerator file_enum(base::FilePath(dir_name), false, |
120 file_util::FileEnumerator::FILES); | 121 base::FileEnumerator::FILES); |
121 for (base::FilePath file = file_enum.Next(); !file.empty(); | 122 for (base::FilePath file = file_enum.Next(); !file.empty(); |
122 file = file_enum.Next()) { | 123 file = file_enum.Next()) { |
123 success = ScheduleFileSystemEntityForDeletion(file.value().c_str()); | 124 success = ScheduleFileSystemEntityForDeletion(file.value().c_str()); |
124 if (!success) { | 125 if (!success) { |
125 LOG(ERROR) << "Failed to schedule file for deletion: " << file.value(); | 126 LOG(ERROR) << "Failed to schedule file for deletion: " << file.value(); |
126 return false; | 127 return false; |
127 } | 128 } |
128 } | 129 } |
129 } | 130 } |
130 | 131 |
131 // Then recurse to all the subdirectories. | 132 // Then recurse to all the subdirectories. |
132 { | 133 { |
133 bool success = true; | 134 bool success = true; |
134 file_util::FileEnumerator dir_enum(base::FilePath(dir_name), false, | 135 base::FileEnumerator dir_enum(base::FilePath(dir_name), false, |
135 file_util::FileEnumerator::DIRECTORIES); | 136 base::FileEnumerator::DIRECTORIES); |
136 for (base::FilePath sub_dir = dir_enum.Next(); !sub_dir.empty(); | 137 for (base::FilePath sub_dir = dir_enum.Next(); !sub_dir.empty(); |
137 sub_dir = dir_enum.Next()) { | 138 sub_dir = dir_enum.Next()) { |
138 success = ScheduleDirectoryForDeletion(sub_dir.value().c_str()); | 139 success = ScheduleDirectoryForDeletion(sub_dir.value().c_str()); |
139 if (!success) { | 140 if (!success) { |
140 LOG(ERROR) << "Failed to schedule subdirectory for deletion: " | 141 LOG(ERROR) << "Failed to schedule subdirectory for deletion: " |
141 << sub_dir.value(); | 142 << sub_dir.value(); |
142 return false; | 143 return false; |
143 } | 144 } |
144 } | 145 } |
145 } | 146 } |
(...skipping 237 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
383 ERROR_SUCCESS); | 384 ERROR_SUCCESS); |
384 } | 385 } |
385 std::vector<char> buffer; | 386 std::vector<char> buffer; |
386 StringArrayToMultiSZBytes(strings_to_keep, &buffer); | 387 StringArrayToMultiSZBytes(strings_to_keep, &buffer); |
387 DCHECK_GT(buffer.size(), 0U); | 388 DCHECK_GT(buffer.size(), 0U); |
388 if (buffer.empty()) | 389 if (buffer.empty()) |
389 return false; | 390 return false; |
390 return (session_manager_key.WriteValue(kPendingFileRenameOps, &buffer[0], | 391 return (session_manager_key.WriteValue(kPendingFileRenameOps, &buffer[0], |
391 buffer.size(), REG_MULTI_SZ) == ERROR_SUCCESS); | 392 buffer.size(), REG_MULTI_SZ) == ERROR_SUCCESS); |
392 } | 393 } |
OLD | NEW |