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 #include "base/file_util.h" | 5 #include "base/file_util.h" |
6 | 6 |
7 #include <windows.h> | 7 #include <windows.h> |
8 #include <propvarutil.h> | 8 #include <propvarutil.h> |
9 #include <psapi.h> | 9 #include <psapi.h> |
10 #include <shellapi.h> | 10 #include <shellapi.h> |
(...skipping 114 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
125 } while (FindNextFile(find_handle, &find_file_data)); | 125 } while (FindNextFile(find_handle, &find_file_data)); |
126 FindClose(find_handle); | 126 FindClose(find_handle); |
127 } | 127 } |
128 | 128 |
129 return file_count; | 129 return file_count; |
130 } | 130 } |
131 | 131 |
132 bool Delete(const FilePath& path, bool recursive) { | 132 bool Delete(const FilePath& path, bool recursive) { |
133 base::ThreadRestrictions::AssertIOAllowed(); | 133 base::ThreadRestrictions::AssertIOAllowed(); |
134 | 134 |
135 if (path.value().length() >= MAX_PATH) | 135 base::PlatformFileInfo file_info; |
136 return false; | 136 if (!GetFileInfo(path, &file_info)) |
| 137 return true; |
137 | 138 |
138 if (!recursive) { | 139 if (!file_info.is_directory) |
139 // If not recursing, then first check to see if |path| is a directory. | 140 return (DeleteFile(path.value().c_str()) != 0); |
140 // If it is, then remove it with RemoveDirectory. | |
141 base::PlatformFileInfo file_info; | |
142 if (GetFileInfo(path, &file_info) && file_info.is_directory) | |
143 return RemoveDirectory(path.value().c_str()) != 0; | |
144 | 141 |
145 // Otherwise, it's a file, wildcard or non-existant. Try DeleteFile first | 142 if (!recursive) |
146 // because it should be faster. If DeleteFile fails, then we fall through | 143 return RemoveDirectory(path.value().c_str()) != 0; |
147 // to SHFileOperation, which will do the right thing. | 144 |
148 if (DeleteFile(path.value().c_str()) != 0) | 145 // Matches posix version of iterating directories. |
149 return true; | 146 bool success = true; |
| 147 std::stack<FilePath::StringType> directories; |
| 148 directories.push(path.value()); |
| 149 FileEnumerator traversal(path, true, static_cast<FileEnumerator::FILE_TYPE>( |
| 150 FileEnumerator::FILES | FileEnumerator::DIRECTORIES)); |
| 151 |
| 152 for (FilePath current = traversal.Next(); success && !current.empty(); |
| 153 current = traversal.Next()) { |
| 154 FileEnumerator::FindInfo info; |
| 155 traversal.GetFindInfo(&info); |
| 156 if (traversal.IsDirectory(info)) { |
| 157 directories.push(current.value()); |
| 158 } else { |
| 159 // This is needed because DeleteFile can fail for read-only files. |
| 160 uint32 attributes = ::GetFileAttributes(current.value().c_str()); |
| 161 if (attributes != INVALID_FILE_ATTRIBUTES && |
| 162 attributes & FILE_ATTRIBUTE_READONLY) |
| 163 ::SetFileAttributes(current.value().c_str(), FILE_ATTRIBUTE_NORMAL); |
| 164 success = (DeleteFile(current.value().c_str()) != 0); |
| 165 } |
150 } | 166 } |
151 | 167 |
152 // SHFILEOPSTRUCT wants the path to be terminated with two NULLs, | 168 while (success && !directories.empty()) { |
153 // so we have to use wcscpy because wcscpy_s writes non-NULLs | 169 FilePath dir = FilePath(directories.top()); |
154 // into the rest of the buffer. | 170 directories.pop(); |
155 wchar_t double_terminated_path[MAX_PATH + 1] = {0}; | 171 success = (RemoveDirectory(dir.value().c_str()) != 0); |
156 #pragma warning(suppress:4996) // don't complain about wcscpy deprecation | 172 } |
157 wcscpy(double_terminated_path, path.value().c_str()); | 173 return success; |
158 | |
159 SHFILEOPSTRUCT file_operation = {0}; | |
160 file_operation.wFunc = FO_DELETE; | |
161 file_operation.pFrom = double_terminated_path; | |
162 file_operation.fFlags = FOF_NOERRORUI | FOF_SILENT | FOF_NOCONFIRMATION; | |
163 if (!recursive) | |
164 file_operation.fFlags |= FOF_NORECURSION | FOF_FILESONLY; | |
165 int err = SHFileOperation(&file_operation); | |
166 // Some versions of Windows return ERROR_FILE_NOT_FOUND (0x2) when deleting | |
167 // an empty directory and some return 0x402 when they should be returning | |
168 // ERROR_FILE_NOT_FOUND. MSDN says Vista and up won't return 0x402. | |
169 return (err == 0 || err == ERROR_FILE_NOT_FOUND || err == 0x402); | |
170 } | 174 } |
171 | 175 |
172 bool DeleteAfterReboot(const FilePath& path) { | 176 bool DeleteAfterReboot(const FilePath& path) { |
173 base::ThreadRestrictions::AssertIOAllowed(); | 177 base::ThreadRestrictions::AssertIOAllowed(); |
174 | 178 |
175 if (path.value().length() >= MAX_PATH) | 179 if (path.value().length() >= MAX_PATH) |
176 return false; | 180 return false; |
177 | 181 |
178 return MoveFileEx(path.value().c_str(), NULL, | 182 return MoveFileEx(path.value().c_str(), NULL, |
179 MOVEFILE_DELAY_UNTIL_REBOOT | | 183 MOVEFILE_DELAY_UNTIL_REBOOT | |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
218 // When writing to a network share, we may not be able to change the ACLs. | 222 // When writing to a network share, we may not be able to change the ACLs. |
219 // Ignore ACL errors then (REPLACEFILE_IGNORE_MERGE_ERRORS). | 223 // Ignore ACL errors then (REPLACEFILE_IGNORE_MERGE_ERRORS). |
220 return ::ReplaceFile(to_path.value().c_str(), | 224 return ::ReplaceFile(to_path.value().c_str(), |
221 from_path.value().c_str(), NULL, | 225 from_path.value().c_str(), NULL, |
222 REPLACEFILE_IGNORE_MERGE_ERRORS, NULL, NULL) ? true : false; | 226 REPLACEFILE_IGNORE_MERGE_ERRORS, NULL, NULL) ? true : false; |
223 } | 227 } |
224 | 228 |
225 bool CopyFile(const FilePath& from_path, const FilePath& to_path) { | 229 bool CopyFile(const FilePath& from_path, const FilePath& to_path) { |
226 base::ThreadRestrictions::AssertIOAllowed(); | 230 base::ThreadRestrictions::AssertIOAllowed(); |
227 | 231 |
228 // NOTE: I suspect we could support longer paths, but that would involve | |
229 // analyzing all our usage of files. | |
230 if (from_path.value().length() >= MAX_PATH || | |
231 to_path.value().length() >= MAX_PATH) { | |
232 return false; | |
233 } | |
234 return (::CopyFile(from_path.value().c_str(), to_path.value().c_str(), | 232 return (::CopyFile(from_path.value().c_str(), to_path.value().c_str(), |
235 false) != 0); | 233 false) != 0); |
236 } | 234 } |
237 | 235 |
238 bool ShellCopy(const FilePath& from_path, const FilePath& to_path, | |
239 bool recursive) { | |
240 base::ThreadRestrictions::AssertIOAllowed(); | |
241 | |
242 // NOTE: I suspect we could support longer paths, but that would involve | |
243 // analyzing all our usage of files. | |
244 if (from_path.value().length() >= MAX_PATH || | |
245 to_path.value().length() >= MAX_PATH) { | |
246 return false; | |
247 } | |
248 | |
249 // SHFILEOPSTRUCT wants the path to be terminated with two NULLs, | |
250 // so we have to use wcscpy because wcscpy_s writes non-NULLs | |
251 // into the rest of the buffer. | |
252 wchar_t double_terminated_path_from[MAX_PATH + 1] = {0}; | |
253 wchar_t double_terminated_path_to[MAX_PATH + 1] = {0}; | |
254 #pragma warning(suppress:4996) // don't complain about wcscpy deprecation | |
255 wcscpy(double_terminated_path_from, from_path.value().c_str()); | |
256 #pragma warning(suppress:4996) // don't complain about wcscpy deprecation | |
257 wcscpy(double_terminated_path_to, to_path.value().c_str()); | |
258 | |
259 SHFILEOPSTRUCT file_operation = {0}; | |
260 file_operation.wFunc = FO_COPY; | |
261 file_operation.pFrom = double_terminated_path_from; | |
262 file_operation.pTo = double_terminated_path_to; | |
263 file_operation.fFlags = FOF_NOERRORUI | FOF_SILENT | FOF_NOCONFIRMATION | | |
264 FOF_NOCONFIRMMKDIR; | |
265 if (!recursive) | |
266 file_operation.fFlags |= FOF_NORECURSION | FOF_FILESONLY; | |
267 | |
268 return (SHFileOperation(&file_operation) == 0); | |
269 } | |
270 | |
271 bool CopyDirectory(const FilePath& from_path, const FilePath& to_path, | 236 bool CopyDirectory(const FilePath& from_path, const FilePath& to_path, |
272 bool recursive) { | 237 bool recursive) { |
273 base::ThreadRestrictions::AssertIOAllowed(); | 238 base::ThreadRestrictions::AssertIOAllowed(); |
274 | 239 |
| 240 FilePath real_to_path = to_path; |
| 241 if (PathExists(real_to_path)) { |
| 242 if (!AbsolutePath(&real_to_path)) |
| 243 return false; |
| 244 } else { |
| 245 real_to_path = real_to_path.DirName(); |
| 246 if (!AbsolutePath(&real_to_path)) |
| 247 return false; |
| 248 } |
| 249 FilePath real_from_path = from_path; |
| 250 if (!AbsolutePath(&real_from_path)) |
| 251 return false; |
| 252 if (real_to_path.value().size() >= real_from_path.value().size() && |
| 253 real_to_path.value().compare(0, real_from_path.value().size(), |
| 254 real_from_path.value()) == 0) |
| 255 return false; |
| 256 |
| 257 bool success = true; |
| 258 FileEnumerator::FILE_TYPE traverse_type = |
| 259 static_cast<FileEnumerator::FILE_TYPE>(FileEnumerator::FILES); |
275 if (recursive) | 260 if (recursive) |
276 return ShellCopy(from_path, to_path, true); | 261 traverse_type = static_cast<FileEnumerator::FILE_TYPE>( |
| 262 traverse_type | FileEnumerator::DIRECTORIES); |
| 263 FileEnumerator traversal(from_path, recursive, traverse_type); |
277 | 264 |
278 // The following code assumes that from path is a directory. | 265 // |to_path| may not exist yet, start the loop with |to_path|. |
279 DCHECK(DirectoryExists(from_path)); | 266 FilePath current = from_path; |
280 | 267 if (!PathExists(from_path)) { |
281 // Instead of creating a new directory, we copy the old one to include the | 268 LOG(ERROR) << "CopyDirectory() couldn't get info on source directory: " |
282 // security information of the folder as part of the copy. | 269 << from_path.value(); |
283 if (!PathExists(to_path)) { | 270 success = false; |
284 // Except that Vista fails to do that, and instead do a recursive copy if | |
285 // the target directory doesn't exist. | |
286 if (base::win::GetVersion() >= base::win::VERSION_VISTA) | |
287 CreateDirectory(to_path); | |
288 else | |
289 ShellCopy(from_path, to_path, false); | |
290 } | 271 } |
291 | 272 |
292 FilePath directory = from_path.Append(L"*.*"); | 273 FilePath from_path_base = from_path; |
293 return ShellCopy(directory, to_path, false); | 274 if (recursive && DirectoryExists(to_path)) { |
| 275 // If the destination already exists and is a directory, then the |
| 276 // top level of source needs to be copied. |
| 277 from_path_base = from_path.DirName(); |
| 278 } |
| 279 |
| 280 // The Windows version of this function assumes that non-recursive calls |
| 281 // will always have a directory for from_path. |
| 282 |
| 283 DCHECK(recursive || DirectoryExists(from_path)); |
| 284 while (success && !current.empty()) { |
| 285 // current is the source path, including from_path, so paste |
| 286 // the suffix after from_path onto to_path to create the target_path. |
| 287 FilePath::StringType suffix( |
| 288 ¤t.value().c_str()[from_path_base.value().size()]); |
| 289 // Strip the leading '\' (if any). |
| 290 if (!suffix.empty()) { |
| 291 DCHECK_EQ('\\', suffix[0]); |
| 292 suffix.erase(0, 1); |
| 293 } |
| 294 const FilePath target_path = to_path.Append(suffix); |
| 295 |
| 296 if (DirectoryExists(current)) { |
| 297 if (::CreateDirectory(target_path.value().c_str(), NULL) == 0) { |
| 298 if (!DirectoryExists(target_path)) { |
| 299 LOG(ERROR) << "CopyDirectory() couldn't create directory: " << |
| 300 target_path.value(); |
| 301 success = false; |
| 302 } |
| 303 } |
| 304 } else { |
| 305 if (!CopyFile(current, target_path)) { |
| 306 LOG(ERROR) << "CopyDirectory() couldn't create file: " << |
| 307 target_path.value(); |
| 308 success = false; |
| 309 } |
| 310 } |
| 311 current = traversal.Next(); |
| 312 } |
| 313 return success; |
294 } | 314 } |
295 | 315 |
296 bool CopyAndDeleteDirectory(const FilePath& from_path, | 316 bool CopyAndDeleteDirectory(const FilePath& from_path, |
297 const FilePath& to_path) { | 317 const FilePath& to_path) { |
298 base::ThreadRestrictions::AssertIOAllowed(); | 318 base::ThreadRestrictions::AssertIOAllowed(); |
299 if (CopyDirectory(from_path, to_path, true)) { | 319 if (CopyDirectory(from_path, to_path, true)) { |
300 if (Delete(from_path, true)) { | 320 if (Delete(from_path, true)) { |
301 return true; | 321 return true; |
302 } | 322 } |
303 // Like Move, this function is not transactional, so we just | 323 // Like Move, this function is not transactional, so we just |
304 // leave the copied bits behind if deleting from_path fails. | 324 // leave the copied bits behind if deleting from_path fails. |
305 // If to_path exists previously then we have already overwritten | 325 // If to_path exists previously then we have already overwritten |
306 // it by now, we don't get better off by deleting the new bits. | 326 // it by now, we don't get better off by deleting the new bits. |
307 } | 327 } |
308 return false; | 328 return false; |
309 } | 329 } |
310 | 330 |
311 | |
312 bool PathExists(const FilePath& path) { | 331 bool PathExists(const FilePath& path) { |
313 base::ThreadRestrictions::AssertIOAllowed(); | 332 base::ThreadRestrictions::AssertIOAllowed(); |
314 return (GetFileAttributes(path.value().c_str()) != INVALID_FILE_ATTRIBUTES); | 333 return (GetFileAttributes(path.value().c_str()) != INVALID_FILE_ATTRIBUTES); |
315 } | 334 } |
316 | 335 |
317 bool PathIsWritable(const FilePath& path) { | 336 bool PathIsWritable(const FilePath& path) { |
318 base::ThreadRestrictions::AssertIOAllowed(); | 337 base::ThreadRestrictions::AssertIOAllowed(); |
319 HANDLE dir = | 338 HANDLE dir = |
320 CreateFile(path.value().c_str(), FILE_ADD_FILE, kFileShareAll, | 339 CreateFile(path.value().c_str(), FILE_ADD_FILE, kFileShareAll, |
321 NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL); | 340 NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL); |
(...skipping 803 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1125 uint8 unused = *(touch + offset); | 1144 uint8 unused = *(touch + offset); |
1126 offset += step_size; | 1145 offset += step_size; |
1127 } | 1146 } |
1128 FreeLibrary(dll_module); | 1147 FreeLibrary(dll_module); |
1129 } | 1148 } |
1130 | 1149 |
1131 return true; | 1150 return true; |
1132 } | 1151 } |
1133 | 1152 |
1134 } // namespace file_util | 1153 } // namespace file_util |
OLD | NEW |