Moving away from shell api to support long path names on windows for filesystem.

base/file_util_win.cc

 // Copyright (c) 2010 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 "base/file_util.h"
 
 #include <windows.h>
 #include <propvarutil.h>
 #include <psapi.h>
 #include <shellapi.h>
@@ skipping 15 matching lines @@
 #include "base/win/scoped_comptr.h"
 #include "base/win/windows_version.h"
 
 namespace file_util {
 
 namespace {
 
 const DWORD kFileShareAll =
     FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE;
 
+bool RemoveReadOnlyAttribute(const FilePath& path) {
+  // This helper for DeleteFile which can fail for read-only files.
+  DWORD attributes = ::GetFileAttributes(path.value().c_str());
+  if (attributes == INVALID_FILE_ATTRIBUTES)
+      return false;
+  if (attributes & FILE_ATTRIBUTE_READONLY) {
+    ::SetFileAttributes(path.value().c_str(),
+                        attributes &~ FILE_ATTRIBUTE_READONLY);
+  }
+  return true;
+}
+
 // Helper for NormalizeFilePath(), defined below.
 bool DevicePathToDriveLetterPath(const FilePath& device_path,
                                  FilePath* drive_letter_path) {
   base::ThreadRestrictions::AssertIOAllowed();
 
   // Get the mapping of drive letters to device paths.
   const int kDriveMappingSize = 1024;
   wchar_t drive_mapping[kDriveMappingSize] = {'\0'};
   if (!::GetLogicalDriveStrings(kDriveMappingSize - 1, drive_mapping)) {
     LOG(ERROR) << "Failed to get drive mapping.";
@@ skipping 79 matching lines @@
     } while (FindNextFile(find_handle,  &find_file_data));
     FindClose(find_handle);
   }
 
   return file_count;
 }
 
 bool Delete(const FilePath& path, bool recursive) {
   base::ThreadRestrictions::AssertIOAllowed();
 
-  if (path.value().length() >= MAX_PATH)
-    return false;
+  base::PlatformFileInfo file_info;
+  if (!GetFileInfo(path, &file_info))
+    return true;
 
-  if (!recursive) {
-    // If not recursing, then first check to see if |path| is a directory.
-    // If it is, then remove it with RemoveDirectory.
-    base::PlatformFileInfo file_info;
-    if (GetFileInfo(path, &file_info) && file_info.is_directory)
-      return RemoveDirectory(path.value().c_str()) != 0;
-
-    // Otherwise, it's a file, wildcard or non-existant. Try DeleteFile first
-    // because it should be faster. If DeleteFile fails, then we fall through
-    // to SHFileOperation, which will do the right thing.
-    if (DeleteFile(path.value().c_str()) != 0)
-      return true;
+  if (!file_info.is_directory) {
+    if (!RemoveReadOnlyAttribute(path))
+      return false;
+    return (DeleteFile(path.value().c_str()) != 0);
   }
 
-  // SHFILEOPSTRUCT wants the path to be terminated with two NULLs,
-  // so we have to use wcscpy because wcscpy_s writes non-NULLs
-  // into the rest of the buffer.
-  wchar_t double_terminated_path[MAX_PATH + 1] = {0};
-#pragma warning(suppress:4996)  // don't complain about wcscpy deprecation
-  wcscpy(double_terminated_path, path.value().c_str());
+  if (!recursive)
+    return RemoveDirectory(path.value().c_str()) != 0;
 
-  SHFILEOPSTRUCT file_operation = {0};
-  file_operation.wFunc = FO_DELETE;
-  file_operation.pFrom = double_terminated_path;
-  file_operation.fFlags = FOF_NOERRORUI | FOF_SILENT | FOF_NOCONFIRMATION;
-  if (!recursive)
-    file_operation.fFlags |= FOF_NORECURSION | FOF_FILESONLY;
-  int err = SHFileOperation(&file_operation);
+  // Matches posix version of iterating directories.
+  bool success = true;
+  std::stack<FilePath::StringType> directories;
+  directories.push(path.value());
+  FileEnumerator traversal(path, true, static_cast<FileEnumerator::FILE_TYPE>(
+      FileEnumerator::FILES | FileEnumerator::DIRECTORIES));
 
-  // Since we're passing flags to the operation telling it to be silent,
-  // it's possible for the operation to be aborted/cancelled without err
-  // being set (although MSDN doesn't give any scenarios for how this can
-  // happen).  See MSDN for SHFileOperation and SHFILEOPTSTRUCT.
-  if (file_operation.fAnyOperationsAborted)
-    return false;
-
-  // Some versions of Windows return ERROR_FILE_NOT_FOUND (0x2) when deleting
-  // an empty directory and some return 0x402 when they should be returning
-  // ERROR_FILE_NOT_FOUND. MSDN says Vista and up won't return 0x402.
-  return (err == 0 || err == ERROR_FILE_NOT_FOUND || err == 0x402);
+  for (FilePath current = traversal.Next(); success && !current.empty();
+       current = traversal.Next()) {
+    FileEnumerator::FindInfo info;
+    traversal.GetFindInfo(&info);
+    if (traversal.IsDirectory(info)) {
+      directories.push(current.value());
+    } else {
+      if (!RemoveReadOnlyAttribute(current))
+        return false;
+      success = (DeleteFile(current.value().c_str()) != 0);
+    }
+  }
+  while (success && !directories.empty()) {
+    FilePath dir = FilePath(directories.top());
+    directories.pop();
+    success = (RemoveDirectory(dir.value().c_str()) != 0);
+  }
+  return success;
 }
 
 bool DeleteAfterReboot(const FilePath& path) {
   base::ThreadRestrictions::AssertIOAllowed();
 
   if (path.value().length() >= MAX_PATH)
     return false;
 
   return MoveFileEx(path.value().c_str(), NULL,
                     MOVEFILE_DELAY_UNTIL_REBOOT |
                         MOVEFILE_REPLACE_EXISTING) != FALSE;
 }
 
 bool Move(const FilePath& from_path, const FilePath& to_path) {
   base::ThreadRestrictions::AssertIOAllowed();
 
-  // NOTE: I suspect we could support longer paths, but that would involve
-  // analyzing all our usage of files.
-  if (from_path.value().length() >= MAX_PATH ||
-      to_path.value().length() >= MAX_PATH) {
-    return false;
-  }
   if (MoveFileEx(from_path.value().c_str(), to_path.value().c_str(),
                  MOVEFILE_COPY_ALLOWED | MOVEFILE_REPLACE_EXISTING) != 0)
     return true;
   if (DirectoryExists(from_path)) {
     // MoveFileEx fails if moving directory across volumes. We will simulate
     // the move by using Copy and Delete. Ideally we could check whether
     // from_path and to_path are indeed in different volumes.
     return CopyAndDeleteDirectory(from_path, to_path);
   }
   return false;
@@ skipping 16 matching lines @@
   // When writing to a network share, we may not be able to change the ACLs.
   // Ignore ACL errors then (REPLACEFILE_IGNORE_MERGE_ERRORS).
   return ::ReplaceFile(to_path.value().c_str(),
       from_path.value().c_str(), NULL,
       REPLACEFILE_IGNORE_MERGE_ERRORS, NULL, NULL) ? true : false;
 }
 
 bool CopyFile(const FilePath& from_path, const FilePath& to_path) {
   base::ThreadRestrictions::AssertIOAllowed();
 
-  // NOTE: I suspect we could support longer paths, but that would involve
-  // analyzing all our usage of files.
-  if (from_path.value().length() >= MAX_PATH ||
-      to_path.value().length() >= MAX_PATH) {
-    return false;
-  }
   return (::CopyFile(from_path.value().c_str(), to_path.value().c_str(),
                      false) != 0);
 }
 
-bool ShellCopy(const FilePath& from_path, const FilePath& to_path,
-               bool recursive) {
-  base::ThreadRestrictions::AssertIOAllowed();
-
-  // NOTE: I suspect we could support longer paths, but that would involve
-  // analyzing all our usage of files.
-  if (from_path.value().length() >= MAX_PATH ||
-      to_path.value().length() >= MAX_PATH) {
-    return false;
-  }
-
-  // SHFILEOPSTRUCT wants the path to be terminated with two NULLs,
-  // so we have to use wcscpy because wcscpy_s writes non-NULLs
-  // into the rest of the buffer.
-  wchar_t double_terminated_path_from[MAX_PATH + 1] = {0};
-  wchar_t double_terminated_path_to[MAX_PATH + 1] = {0};
-#pragma warning(suppress:4996)  // don't complain about wcscpy deprecation
-  wcscpy(double_terminated_path_from, from_path.value().c_str());
-#pragma warning(suppress:4996)  // don't complain about wcscpy deprecation
-  wcscpy(double_terminated_path_to, to_path.value().c_str());
-
-  SHFILEOPSTRUCT file_operation = {0};
-  file_operation.wFunc = FO_COPY;
-  file_operation.pFrom = double_terminated_path_from;
-  file_operation.pTo = double_terminated_path_to;
-  file_operation.fFlags = FOF_NOERRORUI | FOF_SILENT | FOF_NOCONFIRMATION |
-                          FOF_NOCONFIRMMKDIR;
-  if (!recursive)
-    file_operation.fFlags |= FOF_NORECURSION | FOF_FILESONLY;
-
-  return (SHFileOperation(&file_operation) == 0);
-}
-
 bool CopyDirectory(const FilePath& from_path, const FilePath& to_path,
                    bool recursive) {
   base::ThreadRestrictions::AssertIOAllowed();
+  FilePath real_to_path = to_path;
+  if (PathExists(real_to_path)) {
+    if (!AbsolutePath(&real_to_path))

[kinuko, 2010/12/22 23:20:49]

Shouldn't we call the startWith checks before calling AbsolutePath (if we don't
modify AbsolutePath)?

Might have been better to keep the FilePath::StartsWith...() method as well.

   if (StartsWith(path_, kExtendedPathPrefix, false) ||
        StartsWith(path_, kUNCExtendedPathPrefix, false))

[Kavita Kanetkar, 2010/12/23 03:14:28]

Done.

+      return false;
+  } else {
+    real_to_path = real_to_path.DirName();
+    if (!AbsolutePath(&real_to_path))
+      return false;
+  }
+  FilePath real_from_path = from_path;
+  if (!AbsolutePath(&real_from_path))
+    return false;
+  if (real_to_path.value().size() >= real_from_path.value().size() &&
+      real_to_path.value().compare(0, real_from_path.value().size(),
+      real_from_path.value()) == 0)
+    return false;
 
-  if (recursive)
-    return ShellCopy(from_path, to_path, true);
-
-  // The following code assumes that from path is a directory.
-  DCHECK(DirectoryExists(from_path));
-
-  // Instead of creating a new directory, we copy the old one to include the
-  // security information of the folder as part of the copy.
-  if (!PathExists(to_path)) {
-    // Except that Vista fails to do that, and instead do a recursive copy if
-    // the target directory doesn't exist.
-    if (base::win::GetVersion() >= base::win::VERSION_VISTA)
-      CreateDirectory(to_path);
-    else
-      ShellCopy(from_path, to_path, false);
+  if (!PathExists(from_path)) {
+    LOG(ERROR) << "CopyDirectory() couldn't get info on source directory: "
+               << from_path.value();
+    return false;
   }
 
-  FilePath directory = from_path.Append(L"*.*");
-  return ShellCopy(directory, to_path, false);
+  FilePath from_path_base = from_path;
+  if (recursive && DirectoryExists(to_path)) {
+    // If the destination already exists and is a directory, then the
+    // top level of source needs to be copied.
+    from_path_base = from_path.DirName();
+  }
+
+  // This matches previous shell implementation that assumed that
+  // non-recursive calls will always have a directory for from_path.
+  DCHECK(recursive || DirectoryExists(from_path));
+
+  bool success = true;
+  FileEnumerator::FILE_TYPE traverse_type =
+      static_cast<FileEnumerator::FILE_TYPE>(FileEnumerator::FILES);

[kinuko, 2010/12/22 23:20:49]

(repeating michael's question) is this static_cast necessary?

+  if (recursive)
+    traverse_type = static_cast<FileEnumerator::FILE_TYPE>(
+        traverse_type | FileEnumerator::DIRECTORIES);
+  FileEnumerator traversal(from_path, recursive, traverse_type);
+  FilePath current = from_path;
+  while (success && !current.empty()) {
+    // current is the source path, including from_path, so paste
+    // the suffix after from_path onto to_path to create the target_path.
+    FilePath::StringType relative_path_suffix(

[kinuko, 2010/12/22 23:20:49]

I think michael meant suffix -> relative_path rather than relative_path_suffix.
Also can you update the comment to match the code?

[Kavita Kanetkar, 2010/12/23 03:14:28]

I liked suffix var name better. But changed since you both want it changed.

On 2010/12/22 23:20:49, kinuko wrote:

+        &current.value().c_str()[from_path_base.value().size()]);
+
+    // Strip the leading '\' (if any).
+    if (!relative_path_suffix.empty()) {
+      DCHECK_EQ('\\', relative_path_suffix[0]);
+      relative_path_suffix.erase(0, 1);
+    }
+    const FilePath target_path = to_path.Append(relative_path_suffix);
+
+    FileEnumerator::FindInfo info;
+    traversal.GetFindInfo(&info);
+    if (traversal.IsDirectory(info)) {

[kinuko, 2010/12/22 23:20:49]

FileEnumerator::IsDirectory(info) would be better?

IsDirectory is a static method.

[Kavita Kanetkar, 2010/12/23 03:14:28]

I needed to replace this with DirectoryExists() need to ask current whether it's
a directory, current is assigned to traversal.Next() at the end.

On 2010/12/22 23:20:49, kinuko wrote:

+      if (::CreateDirectory(target_path.value().c_str(), NULL) == 0) {
+        if (!DirectoryExists(target_path)) {
+          LOG(ERROR) << "CopyDirectory() couldn't create directory: "
+                     << target_path.value();
+          success = false;
+        }
+      }
+    } else {
+      if (!CopyFile(current, target_path)) {
+        LOG(ERROR) << "CopyDirectory() couldn't create file: "
+                   << target_path.value();
+        success = false;
+      }
+    }
+    current = traversal.Next();
+  }
+  return success;
 }
 
 bool CopyAndDeleteDirectory(const FilePath& from_path,
                             const FilePath& to_path) {
   base::ThreadRestrictions::AssertIOAllowed();
   if (CopyDirectory(from_path, to_path, true)) {
     if (Delete(from_path, true)) {
       return true;
     }
     // Like Move, this function is not transactional, so we just
     // leave the copied bits behind if deleting from_path fails.
     // If to_path exists previously then we have already overwritten
     // it by now, we don't get better off by deleting the new bits.
   }
   return false;
 }
 
-
 bool PathExists(const FilePath& path) {
   base::ThreadRestrictions::AssertIOAllowed();
   return (GetFileAttributes(path.value().c_str()) != INVALID_FILE_ATTRIBUTES);
 }
 
 bool PathIsWritable(const FilePath& path) {
   base::ThreadRestrictions::AssertIOAllowed();
   HANDLE dir =
       CreateFile(path.value().c_str(), FILE_ADD_FILE, kFileShareAll,
                  NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
@@ skipping 803 matching lines @@
       uint8 unused = *(touch + offset);
       offset += step_size;
     }
     FreeLibrary(dll_module);
   }
 
   return true;
 }
 
 }  // namespace file_util