Index: chrome/browser/win/jumplist_file_util.cc |
diff --git a/chrome/browser/win/jumplist_file_util.cc b/chrome/browser/win/jumplist_file_util.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..8170926437513342575d2a53ac66c7aee699cb51 |
--- /dev/null |
+++ b/chrome/browser/win/jumplist_file_util.cc |
@@ -0,0 +1,117 @@ |
+// Copyright (c) 2017 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "chrome/browser/win/jumplist_file_util.h" |
+ |
+#include <Shlwapi.h> |
+#include <windows.h> |
+ |
+#include "base/files/file_enumerator.h" |
+#include "base/metrics/histogram_macros.h" |
+#include "base/threading/thread_restrictions.h" |
+ |
+FolderDeleteResult DeleteFiles(const base::FilePath& path, |
+ const base::FilePath::StringType& pattern, |
+ const int max_file_deleted) { |
+ int success_count = 0; |
+ int failure_count = 0; |
+ FolderDeleteResult delete_status = SUCCEED; |
+ |
+ base::FileEnumerator traversal( |
+ path, false, |
+ base::FileEnumerator::FILES | base::FileEnumerator::DIRECTORIES, pattern); |
+ for (base::FilePath current = traversal.Next(); !current.empty(); |
+ current = traversal.Next()) { |
+ // Try to clear the read-only bit if we find it. |
+ base::FileEnumerator::FileInfo info = traversal.GetInfo(); |
+ if (info.find_data().dwFileAttributes & FILE_ATTRIBUTE_READONLY) { |
+ SetFileAttributes( |
+ current.value().c_str(), |
+ info.find_data().dwFileAttributes & ~FILE_ATTRIBUTE_READONLY); |
+ } |
+ |
+ if (info.IsDirectory()) { |
+ // JumpListIcons{,Old} directories shouldn't have sub-directories. |
+ // If any of them does for unknown reasons, don't delete them. Instead, |
+ // increment the failure count and record this information. |
+ delete_status = FAIL_SUBDIRECTORY_EXISTS; |
+ failure_count++; |
+ } else if (!::DeleteFile(current.value().c_str())) { |
+ failure_count++; |
+ } else { |
+ success_count++; |
+ } |
+ // If it deletes max_file_deleted files with any attempt failures, record |
+ // this information in |delete_status|. |
+ if (success_count >= max_file_deleted) { |
+ // The desired max number of files have been deleted. |
+ if (failure_count) |
grt (UTC plus 2)
2017/03/23 11:24:58
return failure_count ? FAIL_DELETE_MAX_FILES_WITH_
chengx
2017/03/23 21:16:11
Done.
|
+ delete_status = FAIL_DELETE_MAX_FILES_WITH_ERRORS; |
+ return delete_status; |
+ } |
+ if (failure_count >= max_file_deleted) { |
+ // The desired max number of failures have been hit. |
+ return FAIL_MAX_DELETE_FAILURES; |
+ } |
+ } |
+ return delete_status; |
+} |
+ |
+FolderDeleteResult DeleteDirectoryContent(const base::FilePath& path, |
+ const int max_file_deleted) { |
+ base::ThreadRestrictions::AssertIOAllowed(); |
+ |
+ if (path.empty()) |
+ return SUCCEED; |
+ |
+ // For JumpListIcons{,Old} directories, since their names are shorter than |
+ // MAX_PATH, hitting the code in the if-block below is unexpected. |
+ if (path.value().length() >= MAX_PATH) { |
grt (UTC plus 2)
2017/03/23 11:24:57
nit: omit braces
chengx
2017/03/23 21:16:12
Done.
|
+ return FAIL_INVALID_FILE_PATH; |
+ } |
+ |
+ DWORD attr = GetFileAttributes(path.value().c_str()); |
+ // We're done if we can't find the path. |
+ if (attr == INVALID_FILE_ATTRIBUTES) |
+ return SUCCEED; |
+ // Try to clear the read-only bit if we find it. |
+ if ((attr & FILE_ATTRIBUTE_READONLY) && |
+ !SetFileAttributes(path.value().c_str(), |
+ attr & ~FILE_ATTRIBUTE_READONLY)) { |
+ return FAIL_READ_ONLY_DIRECTORY; |
+ } |
+ |
+ // If |path| is a file, simply delete it. However, since JumpListIcons{,Old} |
+ // are directories, hitting the code inside the if-block below is unexpected. |
+ if (!(attr & FILE_ATTRIBUTE_DIRECTORY)) { |
+ ::DeleteFile(path.value().c_str()); |
+ return FAIL_DELETE_SINGLE_FILE; |
+ } |
+ |
+ // If |path| is a directory, delete at most |max_file_deleted| files in it. |
+ return DeleteFiles(path, L"*", max_file_deleted); |
+} |
+ |
+FolderDeleteResult DeleteDirectory(const base::FilePath& path, |
+ const int max_file_deleted) { |
+ base::ThreadRestrictions::AssertIOAllowed(); |
+ // Delete at most |max_file_deleted| files in |path|. |
+ FolderDeleteResult delete_status = |
+ DeleteDirectoryContent(path, max_file_deleted); |
+ // Since DeleteDirectoryContent() can only delete at most |max_file_deleted| |
+ // files, its return value cannot indicate if |path| is empty or not. |
+ // Instead, use PathIsDirectoryEmpty to check if |path| is empty and remove it |
+ // if it is. |
+ if (::PathIsDirectoryEmpty(path.value().c_str()) && |
+ !::RemoveDirectory(path.value().c_str())) |
+ delete_status = FAIL_REMOVE_RAW_DIRECTORY; |
grt (UTC plus 2)
2017/03/23 11:24:58
nit: braces around this due to multi-line conditio
chengx
2017/03/23 21:16:12
Done.
|
+ return delete_status; |
+} |
+ |
+void DeleteDirectoryAndLogResults(const base::FilePath& path, |
+ const int max_file_deleted) { |
+ FolderDeleteResult delete_status = DeleteDirectory(path, max_file_deleted); |
+ UMA_HISTOGRAM_ENUMERATION("WinJumplist.DeleteStatusJumpListIconsOld", |
+ delete_status, END); |
+} |