| // 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> |
| #include <shlobj.h> |
| #include <time.h> |
| #include <string> |
| |
| #include "base/file_path.h" |
| #include "base/logging.h" |
| #include "base/pe_image.h" |
| #include "base/scoped_comptr_win.h" |
| #include "base/scoped_handle.h" |
| #include "base/string_util.h" |
| #include "base/time.h" |
| #include "base/win_util.h" |
| |
| namespace file_util { |
| |
| namespace { |
| |
| // Helper for NormalizeFilePath(), defined below. |
| bool DevicePathToDriveLetterPath(const FilePath& device_path, |
| FilePath* drive_letter_path) { |
| // 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."; |
| return false; |
| } |
| |
| // The drive mapping is a sequence of null terminated strings. |
| // The last string is empty. |
| wchar_t* drive_map_ptr = drive_mapping; |
| wchar_t device_name[MAX_PATH]; |
| wchar_t drive[] = L" :"; |
| |
| // For each string in the drive mapping, get the junction that links |
| // to it. If that junction is a prefix of |device_path|, then we |
| // know that |drive| is the real path prefix. |
| while(*drive_map_ptr) { |
| drive[0] = drive_map_ptr[0]; // Copy the drive letter. |
| |
| if (QueryDosDevice(drive, device_name, MAX_PATH) && |
| StartsWith(device_path.value(), device_name, true)) { |
| *drive_letter_path = FilePath(drive + |
| device_path.value().substr(wcslen(device_name))); |
| return true; |
| } |
| // Move to the next drive letter string, which starts one |
| // increment after the '\0' that terminates the current string. |
| while(*drive_map_ptr++); |
| } |
| |
| // No drive matched. The path does not start with a device junction. |
| *drive_letter_path = device_path; |
| return true; |
| } |
| |
| // Build a security descriptor with the weakest possible file permissions. |
| bool InitLooseSecurityDescriptor(SECURITY_ATTRIBUTES *sa, |
| SECURITY_DESCRIPTOR *sd) { |
| DWORD last_error; |
| |
| if (!InitializeSecurityDescriptor(sd, SECURITY_DESCRIPTOR_REVISION)) { |
| last_error = GetLastError(); |
| LOG(ERROR) << "InitializeSecurityDescriptor failed: GetLastError() = " |
| << last_error; |
| return false; |
| } |
| |
| if (!SetSecurityDescriptorDacl(sd, |
| TRUE, // bDaclPresent: Add one to |sd|. |
| NULL, // pDacl: NULL means allow all access. |
| FALSE // bDaclDefaulted: Not defaulted. |
| )) { |
| last_error = GetLastError(); |
| LOG(ERROR) << "SetSecurityDescriptorDacl() failed: GetLastError() = " |
| << last_error; |
| return false; |
| } |
| |
| if (!SetSecurityDescriptorGroup(sd, |
| NULL, // pGroup: No no primary group. |
| FALSE // bGroupDefaulted: Not defaulted. |
| )) { |
| last_error = GetLastError(); |
| LOG(ERROR) << "SetSecurityDescriptorGroup() failed: GetLastError() = " |
| << last_error; |
| return false; |
| } |
| |
| if (!SetSecurityDescriptorSacl(sd, |
| FALSE, // bSaclPresent: No SACL. |
| NULL, |
| FALSE |
| )) { |
| last_error = GetLastError(); |
| LOG(ERROR) << "SetSecurityDescriptorSacl() failed: GetLastError() = " |
| << last_error; |
| return false; |
| } |
| |
| sa->nLength = sizeof(SECURITY_ATTRIBUTES); |
| sa->lpSecurityDescriptor = sd; |
| sa->bInheritHandle = TRUE; |
| return true; |
| } |
| |
| } // namespace |
| |
| std::wstring GetDirectoryFromPath(const std::wstring& path) { |
| wchar_t path_buffer[MAX_PATH]; |
| wchar_t* file_ptr = NULL; |
| if (GetFullPathName(path.c_str(), MAX_PATH, path_buffer, &file_ptr) == 0) |
| return L""; |
| |
| std::wstring::size_type length = |
| file_ptr ? file_ptr - path_buffer : path.length(); |
| std::wstring directory(path, 0, length); |
| return FilePath(directory).StripTrailingSeparators().value(); |
| } |
| |
| bool AbsolutePath(FilePath* path) { |
| wchar_t file_path_buf[MAX_PATH]; |
| if (!_wfullpath(file_path_buf, path->value().c_str(), MAX_PATH)) |
| return false; |
| *path = FilePath(file_path_buf); |
| return true; |
| } |
| |
| int CountFilesCreatedAfter(const FilePath& path, |
| const base::Time& comparison_time) { |
| int file_count = 0; |
| FILETIME comparison_filetime(comparison_time.ToFileTime()); |
| |
| WIN32_FIND_DATA find_file_data; |
| // All files in given dir |
| std::wstring filename_spec = path.Append(L"*").value(); |
| HANDLE find_handle = FindFirstFile(filename_spec.c_str(), &find_file_data); |
| if (find_handle != INVALID_HANDLE_VALUE) { |
| do { |
| // Don't count current or parent directories. |
| if ((wcscmp(find_file_data.cFileName, L"..") == 0) || |
| (wcscmp(find_file_data.cFileName, L".") == 0)) |
| continue; |
| |
| long result = CompareFileTime(&find_file_data.ftCreationTime, |
| &comparison_filetime); |
| // File was created after or on comparison time |
| if ((result == 1) || (result == 0)) |
| ++file_count; |
| } while (FindNextFile(find_handle, &find_file_data)); |
| FindClose(find_handle); |
| } |
| |
| return file_count; |
| } |
| |
| bool Delete(const FilePath& path, bool recursive) { |
| if (path.value().length() >= MAX_PATH) |
| return false; |
| |
| if (!recursive) { |
| // If not recursing, then first check to see if |path| is a directory. |
| // If it is, then remove it with RemoveDirectory. |
| FileInfo 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; |
| } |
| |
| // 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()); |
| |
| 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); |
| // 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); |
| } |
| |
| bool DeleteAfterReboot(const FilePath& path) { |
| 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) { |
| // 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; |
| } |
| |
| bool ReplaceFile(const FilePath& from_path, const FilePath& to_path) { |
| // Make sure that the target file exists. |
| HANDLE target_file = ::CreateFile( |
| to_path.value().c_str(), |
| 0, |
| FILE_SHARE_READ | FILE_SHARE_WRITE, |
| NULL, |
| CREATE_NEW, |
| FILE_ATTRIBUTE_NORMAL, |
| NULL); |
| if (target_file != INVALID_HANDLE_VALUE) |
| ::CloseHandle(target_file); |
| // 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) { |
| // 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) { |
| // 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) { |
| 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 (win_util::GetWinVersion() >= win_util::WINVERSION_VISTA) |
| CreateDirectory(to_path); |
| else |
| ShellCopy(from_path, to_path, false); |
| } |
| |
| FilePath directory = from_path.Append(L"*.*"); |
| return ShellCopy(directory, to_path, false); |
| } |
| |
| bool CopyAndDeleteDirectory(const FilePath& from_path, |
| const FilePath& to_path) { |
| 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) { |
| return (GetFileAttributes(path.value().c_str()) != INVALID_FILE_ATTRIBUTES); |
| } |
| |
| bool PathIsWritable(const FilePath& path) { |
| HANDLE dir = |
| CreateFile(path.value().c_str(), FILE_ADD_FILE, |
| FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, |
| NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL); |
| |
| if (dir == INVALID_HANDLE_VALUE) |
| return false; |
| |
| CloseHandle(dir); |
| return true; |
| } |
| |
| bool DirectoryExists(const FilePath& path) { |
| DWORD fileattr = GetFileAttributes(path.value().c_str()); |
| if (fileattr != INVALID_FILE_ATTRIBUTES) |
| return (fileattr & FILE_ATTRIBUTE_DIRECTORY) != 0; |
| return false; |
| } |
| |
| bool GetFileCreationLocalTimeFromHandle(HANDLE file_handle, |
| LPSYSTEMTIME creation_time) { |
| if (!file_handle) |
| return false; |
| |
| FILETIME utc_filetime; |
| if (!GetFileTime(file_handle, &utc_filetime, NULL, NULL)) |
| return false; |
| |
| FILETIME local_filetime; |
| if (!FileTimeToLocalFileTime(&utc_filetime, &local_filetime)) |
| return false; |
| |
| return !!FileTimeToSystemTime(&local_filetime, creation_time); |
| } |
| |
| bool GetFileCreationLocalTime(const std::wstring& filename, |
| LPSYSTEMTIME creation_time) { |
| ScopedHandle file_handle( |
| CreateFile(filename.c_str(), GENERIC_READ, |
| FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, |
| OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL)); |
| return GetFileCreationLocalTimeFromHandle(file_handle.Get(), creation_time); |
| } |
| |
| bool ResolveShortcut(FilePath* path) { |
| HRESULT result; |
| ScopedComPtr<IShellLink> i_shell_link; |
| bool is_resolved = false; |
| |
| // Get pointer to the IShellLink interface |
| result = i_shell_link.CreateInstance(CLSID_ShellLink, NULL, |
| CLSCTX_INPROC_SERVER); |
| if (SUCCEEDED(result)) { |
| ScopedComPtr<IPersistFile> persist; |
| // Query IShellLink for the IPersistFile interface |
| result = persist.QueryFrom(i_shell_link); |
| if (SUCCEEDED(result)) { |
| WCHAR temp_path[MAX_PATH]; |
| // Load the shell link |
| result = persist->Load(path->value().c_str(), STGM_READ); |
| if (SUCCEEDED(result)) { |
| // Try to find the target of a shortcut |
| result = i_shell_link->Resolve(0, SLR_NO_UI); |
| if (SUCCEEDED(result)) { |
| result = i_shell_link->GetPath(temp_path, MAX_PATH, |
| NULL, SLGP_UNCPRIORITY); |
| *path = FilePath(temp_path); |
| is_resolved = true; |
| } |
| } |
| } |
| } |
| |
| return is_resolved; |
| } |
| |
| bool CreateShortcutLink(const wchar_t *source, const wchar_t *destination, |
| const wchar_t *working_dir, const wchar_t *arguments, |
| const wchar_t *description, const wchar_t *icon, |
| int icon_index, const wchar_t* app_id) { |
| // Length of arguments and description must be less than MAX_PATH. |
| DCHECK(lstrlen(arguments) < MAX_PATH); |
| DCHECK(lstrlen(description) < MAX_PATH); |
| |
| ScopedComPtr<IShellLink> i_shell_link; |
| ScopedComPtr<IPersistFile> i_persist_file; |
| |
| // Get pointer to the IShellLink interface |
| HRESULT result = i_shell_link.CreateInstance(CLSID_ShellLink, NULL, |
| CLSCTX_INPROC_SERVER); |
| if (FAILED(result)) |
| return false; |
| |
| // Query IShellLink for the IPersistFile interface |
| result = i_persist_file.QueryFrom(i_shell_link); |
| if (FAILED(result)) |
| return false; |
| |
| if (FAILED(i_shell_link->SetPath(source))) |
| return false; |
| |
| if (working_dir && FAILED(i_shell_link->SetWorkingDirectory(working_dir))) |
| return false; |
| |
| if (arguments && FAILED(i_shell_link->SetArguments(arguments))) |
| return false; |
| |
| if (description && FAILED(i_shell_link->SetDescription(description))) |
| return false; |
| |
| if (icon && FAILED(i_shell_link->SetIconLocation(icon, icon_index))) |
| return false; |
| |
| if (app_id && (win_util::GetWinVersion() >= win_util::WINVERSION_WIN7)) { |
| ScopedComPtr<IPropertyStore> property_store; |
| if (FAILED(property_store.QueryFrom(i_shell_link))) |
| return false; |
| |
| if (!win_util::SetAppIdForPropertyStore(property_store, app_id)) |
| return false; |
| } |
| |
| result = i_persist_file->Save(destination, TRUE); |
| return SUCCEEDED(result); |
| } |
| |
| |
| bool UpdateShortcutLink(const wchar_t *source, const wchar_t *destination, |
| const wchar_t *working_dir, const wchar_t *arguments, |
| const wchar_t *description, const wchar_t *icon, |
| int icon_index, const wchar_t* app_id) { |
| // Length of arguments and description must be less than MAX_PATH. |
| DCHECK(lstrlen(arguments) < MAX_PATH); |
| DCHECK(lstrlen(description) < MAX_PATH); |
| |
| // Get pointer to the IPersistFile interface and load existing link |
| ScopedComPtr<IShellLink> i_shell_link; |
| if (FAILED(i_shell_link.CreateInstance(CLSID_ShellLink, NULL, |
| CLSCTX_INPROC_SERVER))) |
| return false; |
| |
| ScopedComPtr<IPersistFile> i_persist_file; |
| if (FAILED(i_persist_file.QueryFrom(i_shell_link))) |
| return false; |
| |
| if (FAILED(i_persist_file->Load(destination, STGM_READWRITE))) |
| return false; |
| |
| if (source && FAILED(i_shell_link->SetPath(source))) |
| return false; |
| |
| if (working_dir && FAILED(i_shell_link->SetWorkingDirectory(working_dir))) |
| return false; |
| |
| if (arguments && FAILED(i_shell_link->SetArguments(arguments))) |
| return false; |
| |
| if (description && FAILED(i_shell_link->SetDescription(description))) |
| return false; |
| |
| if (icon && FAILED(i_shell_link->SetIconLocation(icon, icon_index))) |
| return false; |
| |
| if (app_id && win_util::GetWinVersion() >= win_util::WINVERSION_WIN7) { |
| ScopedComPtr<IPropertyStore> property_store; |
| if (FAILED(property_store.QueryFrom(i_shell_link))) |
| return false; |
| |
| if (!win_util::SetAppIdForPropertyStore(property_store, app_id)) |
| return false; |
| } |
| |
| HRESULT result = i_persist_file->Save(destination, TRUE); |
| return SUCCEEDED(result); |
| } |
| |
| bool TaskbarPinShortcutLink(const wchar_t* shortcut) { |
| // "Pin to taskbar" is only supported after Win7. |
| if (win_util::GetWinVersion() < win_util::WINVERSION_WIN7) |
| return false; |
| |
| int result = reinterpret_cast<int>(ShellExecute(NULL, L"taskbarpin", shortcut, |
| NULL, NULL, 0)); |
| return result > 32; |
| } |
| |
| bool TaskbarUnpinShortcutLink(const wchar_t* shortcut) { |
| // "Unpin from taskbar" is only supported after Win7. |
| if (win_util::GetWinVersion() < win_util::WINVERSION_WIN7) |
| return false; |
| |
| int result = reinterpret_cast<int>(ShellExecute(NULL, L"taskbarunpin", |
| shortcut, NULL, NULL, 0)); |
| return result > 32; |
| } |
| |
| bool GetTempDir(FilePath* path) { |
| wchar_t temp_path[MAX_PATH + 1]; |
| DWORD path_len = ::GetTempPath(MAX_PATH, temp_path); |
| if (path_len >= MAX_PATH || path_len <= 0) |
| return false; |
| // TODO(evanm): the old behavior of this function was to always strip the |
| // trailing slash. We duplicate this here, but it shouldn't be necessary |
| // when everyone is using the appropriate FilePath APIs. |
| *path = FilePath(temp_path).StripTrailingSeparators(); |
| return true; |
| } |
| |
| bool GetShmemTempDir(FilePath* path) { |
| return GetTempDir(path); |
| } |
| |
| bool CreateTemporaryFile(FilePath* path) { |
| FilePath temp_file; |
| |
| if (!GetTempDir(path)) |
| return false; |
| |
| if (CreateTemporaryFileInDir(*path, &temp_file)) { |
| *path = temp_file; |
| return true; |
| } |
| |
| return false; |
| } |
| |
| FILE* CreateAndOpenTemporaryShmemFile(FilePath* path) { |
| return CreateAndOpenTemporaryFile(path); |
| } |
| |
| // On POSIX we have semantics to create and open a temporary file |
| // atomically. |
| // TODO(jrg): is there equivalent call to use on Windows instead of |
| // going 2-step? |
| FILE* CreateAndOpenTemporaryFileInDir(const FilePath& dir, FilePath* path) { |
| if (!CreateTemporaryFileInDir(dir, path)) { |
| return NULL; |
| } |
| // Open file in binary mode, to avoid problems with fwrite. On Windows |
| // it replaces \n's with \r\n's, which may surprise you. |
| // Reference: http://msdn.microsoft.com/en-us/library/h9t88zwz(VS.71).aspx |
| return OpenFile(*path, "wb+"); |
| } |
| |
| bool CreateTemporaryFileInDir(const FilePath& dir, |
| FilePath* temp_file) { |
| wchar_t temp_name[MAX_PATH + 1]; |
| |
| if (!GetTempFileName(dir.value().c_str(), L"", 0, temp_name)) { |
| PLOG(WARNING) << "Failed to get temporary file name in " << dir.value(); |
| return false; |
| } |
| |
| DWORD path_len = GetLongPathName(temp_name, temp_name, MAX_PATH); |
| if (path_len > MAX_PATH + 1 || path_len == 0) { |
| PLOG(WARNING) << "Failed to get long path name for " << temp_name; |
| return false; |
| } |
| |
| std::wstring temp_file_str; |
| temp_file_str.assign(temp_name, path_len); |
| *temp_file = FilePath(temp_file_str); |
| return true; |
| } |
| |
| bool CreateTemporaryDirInDir(const FilePath& base_dir, |
| const FilePath::StringType& prefix, |
| bool loosen_permissions, |
| FilePath* new_dir) { |
| SECURITY_ATTRIBUTES sa; |
| SECURITY_DESCRIPTOR sd; |
| |
| LPSECURITY_ATTRIBUTES directory_security_attributes = NULL; |
| if (loosen_permissions) { |
| if (InitLooseSecurityDescriptor(&sa, &sd)) |
| directory_security_attributes = &sa; |
| else |
| LOG(ERROR) << "Failed to init security attributes, fall back to NULL."; |
| } |
| |
| FilePath path_to_create; |
| srand(static_cast<uint32>(time(NULL))); |
| |
| int count = 0; |
| while (count < 50) { |
| // Try create a new temporary directory with random generated name. If |
| // the one exists, keep trying another path name until we reach some limit. |
| path_to_create = base_dir; |
| |
| std::wstring new_dir_name; |
| new_dir_name.assign(prefix); |
| new_dir_name.append(IntToWString(rand() % kint16max)); |
| |
| path_to_create = path_to_create.Append(new_dir_name); |
| if (::CreateDirectory(path_to_create.value().c_str(), |
| directory_security_attributes)) |
| break; |
| count++; |
| } |
| |
| if (count == 50) { |
| return false; |
| } |
| |
| *new_dir = path_to_create; |
| |
| return true; |
| } |
| |
| bool CreateNewTempDirectory(const FilePath::StringType& prefix, |
| FilePath* new_temp_path) { |
| FilePath system_temp_dir; |
| if (!GetTempDir(&system_temp_dir)) |
| return false; |
| |
| return CreateTemporaryDirInDir(system_temp_dir, |
| prefix, |
| false, |
| new_temp_path); |
| } |
| |
| bool CreateDirectory(const FilePath& full_path) { |
| return file_util::CreateDirectoryExtraLogging(full_path, LOG(INFO)); |
| } |
| |
| // TODO(skerner): Extra logging has been added to understand crbug/35198 . |
| // Remove it once we get a log from a user who can reproduce the issue. |
| bool CreateDirectoryExtraLogging(const FilePath& full_path, |
| std::ostream& log) { |
| log << "Enter CreateDirectory: full_path = " << full_path.value() |
| << std::endl; |
| // If the path exists, we've succeeded if it's a directory, failed otherwise. |
| const wchar_t* full_path_str = full_path.value().c_str(); |
| DWORD fileattr = ::GetFileAttributes(full_path_str); |
| log << "::GetFileAttributes() returned " << fileattr << std::endl; |
| if (fileattr == INVALID_FILE_ATTRIBUTES) { |
| DWORD fileattr_error = GetLastError(); |
| log << "::GetFileAttributes() failed. GetLastError() = " |
| << fileattr_error << std::endl; |
| } else { |
| if ((fileattr & FILE_ATTRIBUTE_DIRECTORY) != 0) { |
| log << "CreateDirectory(" << full_path_str << "), " |
| << "directory already exists." << std::endl; |
| return true; |
| } else { |
| log << "CreateDirectory(" << full_path_str << "), " |
| << "conflicts with existing file." << std::endl; |
| } |
| } |
| |
| // Invariant: Path does not exist as file or directory. |
| |
| // Attempt to create the parent recursively. This will immediately return |
| // true if it already exists, otherwise will create all required parent |
| // directories starting with the highest-level missing parent. |
| FilePath parent_path(full_path.DirName()); |
| if (parent_path.value() == full_path.value()) { |
| log << "Can't create directory: parent_path " << parent_path.value() |
| << " should not equal full_path " << full_path.value() |
| << std::endl; |
| return false; |
| } |
| if (!CreateDirectory(parent_path)) { |
| log << "Failed to create one of the parent directories: " |
| << parent_path.value() << std::endl; |
| return false; |
| } |
| |
| log << "About to call ::CreateDirectory() with full_path_str = " |
| << full_path_str << std::endl; |
| if (!::CreateDirectory(full_path_str, NULL)) { |
| DWORD error_code = ::GetLastError(); |
| log << "CreateDirectory() gave last error " << error_code << std::endl; |
| |
| DWORD fileattr = GetFileAttributes(full_path.value().c_str()); |
| if (fileattr == INVALID_FILE_ATTRIBUTES) { |
| DWORD fileattr_error = ::GetLastError(); |
| log << "GetFileAttributes() failed, GetLastError() = " |
| << fileattr_error << std::endl; |
| } else { |
| log << "GetFileAttributes() returned " << fileattr << std::endl; |
| log << "Is the path a directory: " |
| << ((fileattr & FILE_ATTRIBUTE_DIRECTORY) != 0) << std::endl; |
| } |
| if (error_code == ERROR_ALREADY_EXISTS && DirectoryExists(full_path)) { |
| // This error code doesn't indicate whether we were racing with someone |
| // creating the same directory, or a file with the same path, therefore |
| // we check. |
| log << "Race condition? Directory already exists: " |
| << full_path.value() << std::endl; |
| return true; |
| } else { |
| DWORD dir_exists_error = ::GetLastError(); |
| log << "Failed to create directory " << full_path_str << std::endl; |
| log << "GetLastError() for DirectoryExists() is " |
| << dir_exists_error << std::endl; |
| return false; |
| } |
| } else { |
| log << "::CreateDirectory() succeeded." << std::endl; |
| return true; |
| } |
| } |
| |
| bool GetFileInfo(const FilePath& file_path, FileInfo* results) { |
| WIN32_FILE_ATTRIBUTE_DATA attr; |
| if (!GetFileAttributesEx(file_path.value().c_str(), |
| GetFileExInfoStandard, &attr)) { |
| return false; |
| } |
| |
| ULARGE_INTEGER size; |
| size.HighPart = attr.nFileSizeHigh; |
| size.LowPart = attr.nFileSizeLow; |
| results->size = size.QuadPart; |
| |
| results->is_directory = |
| (attr.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0; |
| results->last_modified = base::Time::FromFileTime(attr.ftLastWriteTime); |
| |
| return true; |
| } |
| |
| bool SetLastModifiedTime(const FilePath& file_path, base::Time last_modified) { |
| FILETIME timestamp(last_modified.ToFileTime()); |
| ScopedHandle file_handle( |
| CreateFile(file_path.value().c_str(), FILE_WRITE_ATTRIBUTES, |
| FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, |
| OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL)); |
| BOOL ret = SetFileTime(file_handle.Get(), NULL, ×tamp, ×tamp); |
| return ret != 0; |
| } |
| |
| FILE* OpenFile(const FilePath& filename, const char* mode) { |
| std::wstring w_mode = ASCIIToWide(std::string(mode)); |
| return _wfsopen(filename.value().c_str(), w_mode.c_str(), _SH_DENYNO); |
| } |
| |
| FILE* OpenFile(const std::string& filename, const char* mode) { |
| return _fsopen(filename.c_str(), mode, _SH_DENYNO); |
| } |
| |
| int ReadFile(const FilePath& filename, char* data, int size) { |
| ScopedHandle file(CreateFile(filename.value().c_str(), |
| GENERIC_READ, |
| FILE_SHARE_READ | FILE_SHARE_WRITE, |
| NULL, |
| OPEN_EXISTING, |
| FILE_FLAG_SEQUENTIAL_SCAN, |
| NULL)); |
| if (file == INVALID_HANDLE_VALUE) |
| return -1; |
| |
| DWORD read; |
| if (::ReadFile(file, data, size, &read, NULL) && |
| static_cast<int>(read) == size) |
| return read; |
| return -1; |
| } |
| |
| int WriteFile(const FilePath& filename, const char* data, int size) { |
| ScopedHandle file(CreateFile(filename.value().c_str(), |
| GENERIC_WRITE, |
| 0, |
| NULL, |
| CREATE_ALWAYS, |
| 0, |
| NULL)); |
| if (file == INVALID_HANDLE_VALUE) { |
| LOG(WARNING) << "CreateFile failed for path " << filename.value() << |
| " error code=" << GetLastError() << |
| " error text=" << win_util::FormatLastWin32Error(); |
| return -1; |
| } |
| |
| DWORD written; |
| BOOL result = ::WriteFile(file, data, size, &written, NULL); |
| if (result && static_cast<int>(written) == size) |
| return written; |
| |
| if (!result) { |
| // WriteFile failed. |
| LOG(WARNING) << "writing file " << filename.value() << |
| " failed, error code=" << GetLastError() << |
| " description=" << win_util::FormatLastWin32Error(); |
| } else { |
| // Didn't write all the bytes. |
| LOG(WARNING) << "wrote" << written << " bytes to " << |
| filename.value() << " expected " << size; |
| } |
| return -1; |
| } |
| |
| bool RenameFileAndResetSecurityDescriptor(const FilePath& source_file_path, |
| const FilePath& target_file_path) { |
| // The parameters to SHFileOperation must be terminated with 2 NULL chars. |
| std::wstring source = source_file_path.value(); |
| std::wstring target = target_file_path.value(); |
| |
| source.append(1, L'\0'); |
| target.append(1, L'\0'); |
| |
| SHFILEOPSTRUCT move_info = {0}; |
| move_info.wFunc = FO_MOVE; |
| move_info.pFrom = source.c_str(); |
| move_info.pTo = target.c_str(); |
| move_info.fFlags = FOF_SILENT | FOF_NOCONFIRMATION | FOF_NOERRORUI | |
| FOF_NOCONFIRMMKDIR | FOF_NOCOPYSECURITYATTRIBS; |
| |
| if (0 != SHFileOperation(&move_info)) |
| return false; |
| |
| return true; |
| } |
| |
| // Gets the current working directory for the process. |
| bool GetCurrentDirectory(FilePath* dir) { |
| wchar_t system_buffer[MAX_PATH]; |
| system_buffer[0] = 0; |
| DWORD len = ::GetCurrentDirectory(MAX_PATH, system_buffer); |
| if (len == 0 || len > MAX_PATH) |
| return false; |
| // TODO(evanm): the old behavior of this function was to always strip the |
| // trailing slash. We duplicate this here, but it shouldn't be necessary |
| // when everyone is using the appropriate FilePath APIs. |
| std::wstring dir_str(system_buffer); |
| *dir = FilePath(dir_str).StripTrailingSeparators(); |
| return true; |
| } |
| |
| // Sets the current working directory for the process. |
| bool SetCurrentDirectory(const FilePath& directory) { |
| BOOL ret = ::SetCurrentDirectory(directory.value().c_str()); |
| return ret != 0; |
| } |
| |
| /////////////////////////////////////////////// |
| // FileEnumerator |
| |
| FileEnumerator::FileEnumerator(const FilePath& root_path, |
| bool recursive, |
| FileEnumerator::FILE_TYPE file_type) |
| : recursive_(recursive), |
| file_type_(file_type), |
| is_in_find_op_(false), |
| find_handle_(INVALID_HANDLE_VALUE) { |
| // INCLUDE_DOT_DOT must not be specified if recursive. |
| DCHECK(!(recursive && (INCLUDE_DOT_DOT & file_type_))); |
| pending_paths_.push(root_path); |
| } |
| |
| FileEnumerator::FileEnumerator(const FilePath& root_path, |
| bool recursive, |
| FileEnumerator::FILE_TYPE file_type, |
| const FilePath::StringType& pattern) |
| : recursive_(recursive), |
| file_type_(file_type), |
| is_in_find_op_(false), |
| pattern_(pattern), |
| find_handle_(INVALID_HANDLE_VALUE) { |
| // INCLUDE_DOT_DOT must not be specified if recursive. |
| DCHECK(!(recursive && (INCLUDE_DOT_DOT & file_type_))); |
| pending_paths_.push(root_path); |
| } |
| |
| FileEnumerator::~FileEnumerator() { |
| if (find_handle_ != INVALID_HANDLE_VALUE) |
| FindClose(find_handle_); |
| } |
| |
| void FileEnumerator::GetFindInfo(FindInfo* info) { |
| DCHECK(info); |
| |
| if (!is_in_find_op_) |
| return; |
| |
| memcpy(info, &find_data_, sizeof(*info)); |
| } |
| |
| bool FileEnumerator::IsDirectory(const FindInfo& info) { |
| return (info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0; |
| } |
| |
| // static |
| FilePath FileEnumerator::GetFilename(const FindInfo& find_info) { |
| return FilePath(find_info.cFileName); |
| } |
| |
| FilePath FileEnumerator::Next() { |
| if (!is_in_find_op_) { |
| if (pending_paths_.empty()) |
| return FilePath(); |
| |
| // The last find FindFirstFile operation is done, prepare a new one. |
| root_path_ = pending_paths_.top(); |
| pending_paths_.pop(); |
| |
| // Start a new find operation. |
| FilePath src = root_path_; |
| |
| if (pattern_.empty()) |
| src = src.Append(L"*"); // No pattern = match everything. |
| else |
| src = src.Append(pattern_); |
| |
| find_handle_ = FindFirstFile(src.value().c_str(), &find_data_); |
| is_in_find_op_ = true; |
| |
| } else { |
| // Search for the next file/directory. |
| if (!FindNextFile(find_handle_, &find_data_)) { |
| FindClose(find_handle_); |
| find_handle_ = INVALID_HANDLE_VALUE; |
| } |
| } |
| |
| if (INVALID_HANDLE_VALUE == find_handle_) { |
| is_in_find_op_ = false; |
| |
| // This is reached when we have finished a directory and are advancing to |
| // the next one in the queue. We applied the pattern (if any) to the files |
| // in the root search directory, but for those directories which were |
| // matched, we want to enumerate all files inside them. This will happen |
| // when the handle is empty. |
| pattern_ = FilePath::StringType(); |
| |
| return Next(); |
| } |
| |
| FilePath cur_file(find_data_.cFileName); |
| if (ShouldSkip(cur_file)) |
| return Next(); |
| |
| // Construct the absolute filename. |
| cur_file = root_path_.Append(cur_file); |
| |
| if (find_data_.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) { |
| if (recursive_) { |
| // If |cur_file| is a directory, and we are doing recursive searching, add |
| // it to pending_paths_ so we scan it after we finish scanning this |
| // directory. |
| pending_paths_.push(cur_file); |
| } |
| return (file_type_ & FileEnumerator::DIRECTORIES) ? cur_file : Next(); |
| } |
| return (file_type_ & FileEnumerator::FILES) ? cur_file : Next(); |
| } |
| |
| /////////////////////////////////////////////// |
| // MemoryMappedFile |
| |
| MemoryMappedFile::MemoryMappedFile() |
| : file_(INVALID_HANDLE_VALUE), |
| file_mapping_(INVALID_HANDLE_VALUE), |
| data_(NULL), |
| length_(INVALID_FILE_SIZE) { |
| } |
| |
| bool MemoryMappedFile::MapFileToMemoryInternal() { |
| if (file_ == INVALID_HANDLE_VALUE) |
| return false; |
| |
| length_ = ::GetFileSize(file_, NULL); |
| if (length_ == INVALID_FILE_SIZE) |
| return false; |
| |
| // length_ value comes from GetFileSize() above. GetFileSize() returns DWORD, |
| // therefore the cast here is safe. |
| file_mapping_ = ::CreateFileMapping(file_, NULL, PAGE_READONLY, |
| 0, static_cast<DWORD>(length_), NULL); |
| if (file_mapping_ == INVALID_HANDLE_VALUE) |
| return false; |
| |
| data_ = static_cast<uint8*>( |
| ::MapViewOfFile(file_mapping_, FILE_MAP_READ, 0, 0, length_)); |
| return data_ != NULL; |
| } |
| |
| void MemoryMappedFile::CloseHandles() { |
| if (data_) |
| ::UnmapViewOfFile(data_); |
| if (file_mapping_ != INVALID_HANDLE_VALUE) |
| ::CloseHandle(file_mapping_); |
| if (file_ != INVALID_HANDLE_VALUE) |
| ::CloseHandle(file_); |
| |
| data_ = NULL; |
| file_mapping_ = file_ = INVALID_HANDLE_VALUE; |
| length_ = INVALID_FILE_SIZE; |
| } |
| |
| bool HasFileBeenModifiedSince(const FileEnumerator::FindInfo& find_info, |
| const base::Time& cutoff_time) { |
| long result = CompareFileTime(&find_info.ftLastWriteTime, |
| &cutoff_time.ToFileTime()); |
| return result == 1 || result == 0; |
| } |
| |
| bool NormalizeFilePath(const FilePath& path, FilePath* real_path) { |
| ScopedHandle path_handle( |
| ::CreateFile(path.value().c_str(), |
| GENERIC_READ, |
| FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, |
| NULL, |
| OPEN_EXISTING, |
| FILE_ATTRIBUTE_NORMAL, |
| NULL)); |
| if (path_handle == INVALID_HANDLE_VALUE) |
| return false; |
| |
| // In Vista, GetFinalPathNameByHandle() would give us the real path |
| // from a file handle. If we ever deprecate XP, consider changing the |
| // code below to a call to GetFinalPathNameByHandle(). The method this |
| // function uses is explained in the following msdn article: |
| // http://msdn.microsoft.com/en-us/library/aa366789(VS.85).aspx |
| DWORD file_size_high = 0; |
| DWORD file_size_low = ::GetFileSize(path_handle.Get(), &file_size_high); |
| if (file_size_low == 0 && file_size_high == 0) { |
| // It is not possible to map an empty file. |
| LOG(ERROR) << "NormalizeFilePath failed: Empty file."; |
| return false; |
| } |
| |
| // Create a file mapping object. Can't easily use MemoryMappedFile, because |
| // we only map the first byte, and need direct access to the handle. |
| ScopedHandle file_map_handle( |
| ::CreateFileMapping(path_handle.Get(), |
| NULL, |
| PAGE_READONLY, |
| 0, |
| 1, // Just one byte. No need to look at the data. |
| NULL)); |
| |
| if (file_map_handle == INVALID_HANDLE_VALUE) |
| return false; |
| |
| // Use a view of the file to get the path to the file. |
| void* file_view = MapViewOfFile( |
| file_map_handle.Get(), FILE_MAP_READ, 0, 0, 1); |
| if (!file_view) |
| return false; |
| |
| bool success = false; |
| |
| // The expansion of |path| into a full path may make it longer. |
| // GetMappedFileName() will fail if the result is longer than MAX_PATH. |
| // Pad a bit to be safe. If kMaxPathLength is ever changed to be less |
| // than MAX_PATH, it would be nessisary to test that GetMappedFileName() |
| // not return kMaxPathLength. This would mean that only part of the |
| // path fit in |mapped_file_path|. |
| const int kMaxPathLength = MAX_PATH + 10; |
| wchar_t mapped_file_path[kMaxPathLength]; |
| if (::GetMappedFileName(GetCurrentProcess(), |
| file_view, |
| mapped_file_path, |
| kMaxPathLength)) { |
| // GetMappedFileName() will return a path that starts with |
| // "\Device\Harddisk...". Helper DevicePathToDriveLetterPath() |
| // will find a drive letter which maps to the path's device, so |
| // that we return a path starting with a drive letter. |
| FilePath mapped_file(mapped_file_path); |
| success = DevicePathToDriveLetterPath(mapped_file, real_path); |
| } |
| UnmapViewOfFile(file_view); |
| return success; |
| } |
| |
| bool PreReadImage(const wchar_t* file_path, size_t size_to_read, |
| size_t step_size) { |
| HMODULE dll_module = LoadLibraryExW( |
| file_path, |
| NULL, |
| LOAD_WITH_ALTERED_SEARCH_PATH | DONT_RESOLVE_DLL_REFERENCES); |
| |
| if (!dll_module) |
| return false; |
| |
| PEImage pe_image(dll_module); |
| PIMAGE_NT_HEADERS nt_headers = pe_image.GetNTHeaders(); |
| size_t actual_size_to_read = size_to_read ? size_to_read : |
| nt_headers->OptionalHeader.SizeOfImage; |
| volatile uint8* touch = reinterpret_cast<uint8*>(dll_module); |
| size_t offset = 0; |
| while (offset < actual_size_to_read) { |
| uint8 unused = *(touch + offset); |
| offset += step_size; |
| } |
| FreeLibrary(dll_module); |
| return true; |
| } |
| |
| } // namespace file_util |
