Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/files/file_util.h" | 5 #include "base/files/file_util.h" |
| 6 | 6 |
| 7 #include <windows.h> | 7 #include <windows.h> |
| 8 #include <io.h> | 8 #include <io.h> |
| 9 #include <psapi.h> | 9 #include <psapi.h> |
| 10 #include <shellapi.h> | 10 #include <shellapi.h> |
| 11 #include <shlobj.h> | 11 #include <shlobj.h> |
| 12 #include <stddef.h> | 12 #include <stddef.h> |
| 13 #include <stdint.h> | 13 #include <stdint.h> |
| 14 #include <time.h> | 14 #include <time.h> |
| 15 #include <winsock2.h> | 15 #include <winsock2.h> |
| 16 | 16 |
| 17 #include <algorithm> | 17 #include <algorithm> |
| 18 #include <limits> | 18 #include <limits> |
| 19 #include <string> | 19 #include <string> |
| 20 | 20 |
| 21 #include "base/files/file_enumerator.h" | 21 #include "base/files/file_enumerator.h" |
| 22 #include "base/files/file_path.h" | 22 #include "base/files/file_path.h" |
| 23 #include "base/guid.h" | |
| 23 #include "base/logging.h" | 24 #include "base/logging.h" |
| 24 #include "base/macros.h" | 25 #include "base/macros.h" |
| 25 #include "base/metrics/histogram.h" | 26 #include "base/metrics/histogram.h" |
| 26 #include "base/process/process_handle.h" | 27 #include "base/process/process_handle.h" |
| 27 #include "base/rand_util.h" | 28 #include "base/rand_util.h" |
| 28 #include "base/strings/string_number_conversions.h" | 29 #include "base/strings/string_number_conversions.h" |
| 29 #include "base/strings/string_util.h" | 30 #include "base/strings/string_util.h" |
| 30 #include "base/strings/utf_string_conversions.h" | 31 #include "base/strings/utf_string_conversions.h" |
| 31 #include "base/threading/thread_restrictions.h" | 32 #include "base/threading/thread_restrictions.h" |
| 32 #include "base/time/time.h" | 33 #include "base/time/time.h" |
| (...skipping 300 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 333 } | 334 } |
| 334 // Open file in binary mode, to avoid problems with fwrite. On Windows | 335 // Open file in binary mode, to avoid problems with fwrite. On Windows |
| 335 // it replaces \n's with \r\n's, which may surprise you. | 336 // it replaces \n's with \r\n's, which may surprise you. |
| 336 // Reference: http://msdn.microsoft.com/en-us/library/h9t88zwz(VS.71).aspx | 337 // Reference: http://msdn.microsoft.com/en-us/library/h9t88zwz(VS.71).aspx |
| 337 return OpenFile(*path, "wb+"); | 338 return OpenFile(*path, "wb+"); |
| 338 } | 339 } |
| 339 | 340 |
| 340 bool CreateTemporaryFileInDir(const FilePath& dir, FilePath* temp_file) { | 341 bool CreateTemporaryFileInDir(const FilePath& dir, FilePath* temp_file) { |
| 341 ThreadRestrictions::AssertIOAllowed(); | 342 ThreadRestrictions::AssertIOAllowed(); |
| 342 | 343 |
| 343 wchar_t temp_name[MAX_PATH + 1]; | 344 // Use GUID instead of ::GetTempFileName() to generate unique file names. |
| 345 // "Due to the algorithm used to generate file names, GetTempFileName can | |
| 346 // perform poorly when creating a large number of files with the same prefix. | |
| 347 // In such cases, it is recommended that you construct unique file names based | |
| 348 // on GUIDs." | |
| 349 // https://msdn.microsoft.com/en-us/library/windows/desktop/aa364991(v=vs.85). aspx | |
| 350 FilePath temp_name = dir.Append(ASCIIToUTF16(base::GenerateGUID()) + L".tmp"); | |
|
gab
2017/04/06 20:19:12
Actually, we should probably loop this while it ex
chengx
2017/04/06 21:00:31
Acknowledged.
| |
| 344 | 351 |
| 345 if (!GetTempFileName(dir.value().c_str(), L"", 0, temp_name)) { | 352 File file(temp_name, File::FLAG_CREATE | File::FLAG_READ | File::FLAG_WRITE); |
| 353 if (!file.IsValid()) { | |
| 346 DPLOG(WARNING) << "Failed to get temporary file name in " | 354 DPLOG(WARNING) << "Failed to get temporary file name in " |
| 347 << UTF16ToUTF8(dir.value()); | 355 << UTF16ToUTF8(dir.value()); |
| 348 return false; | 356 return false; |
| 349 } | 357 } |
| 358 file.Close(); | |
| 350 | 359 |
| 351 wchar_t long_temp_name[MAX_PATH + 1]; | 360 wchar_t long_temp_name[MAX_PATH + 1]; |
| 352 DWORD long_name_len = GetLongPathName(temp_name, long_temp_name, MAX_PATH); | 361 DWORD long_name_len = |
| 362 GetLongPathName(temp_name.value().c_str(), long_temp_name, MAX_PATH); | |
| 353 if (long_name_len > MAX_PATH || long_name_len == 0) { | 363 if (long_name_len > MAX_PATH || long_name_len == 0) { |
| 354 // GetLongPathName() failed, but we still have a temporary file. | 364 // GetLongPathName() failed, but we still have a temporary file. |
| 355 *temp_file = FilePath(temp_name); | 365 *temp_file = std::move(temp_name); |
| 356 return true; | 366 return true; |
| 357 } | 367 } |
| 358 | 368 |
| 359 FilePath::StringType long_temp_name_str; | 369 FilePath::StringType long_temp_name_str; |
| 360 long_temp_name_str.assign(long_temp_name, long_name_len); | 370 long_temp_name_str.assign(long_temp_name, long_name_len); |
| 361 *temp_file = FilePath(long_temp_name_str); | 371 *temp_file = FilePath(std::move(long_temp_name_str)); |
| 362 return true; | 372 return true; |
| 363 } | 373 } |
| 364 | 374 |
| 365 bool CreateTemporaryDirInDir(const FilePath& base_dir, | 375 bool CreateTemporaryDirInDir(const FilePath& base_dir, |
| 366 const FilePath::StringType& prefix, | 376 const FilePath::StringType& prefix, |
| 367 FilePath* new_dir) { | 377 FilePath* new_dir) { |
| 368 ThreadRestrictions::AssertIOAllowed(); | 378 ThreadRestrictions::AssertIOAllowed(); |
| 369 | 379 |
| 370 FilePath path_to_create; | 380 FilePath path_to_create; |
| 371 | 381 |
| (...skipping 470 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 842 // Like Move, this function is not transactional, so we just | 852 // Like Move, this function is not transactional, so we just |
| 843 // leave the copied bits behind if deleting from_path fails. | 853 // leave the copied bits behind if deleting from_path fails. |
| 844 // If to_path exists previously then we have already overwritten | 854 // If to_path exists previously then we have already overwritten |
| 845 // it by now, we don't get better off by deleting the new bits. | 855 // it by now, we don't get better off by deleting the new bits. |
| 846 } | 856 } |
| 847 return false; | 857 return false; |
| 848 } | 858 } |
| 849 | 859 |
| 850 } // namespace internal | 860 } // namespace internal |
| 851 } // namespace base | 861 } // namespace base |
| OLD | NEW |